DB works

ums/management/main.py

@@ -10,12 +10,17 @@
 
 # TEST ONLY
 
-from ums.utils import AgentMessage, RiddleData, RiddleDataType, RiddleSolution
+
 
 from typing import Union
 
 from fastapi import FastAPI
 
+from ums.utils import AgentMessage, RiddleData, RiddleDataType, RiddleSolution
+from ums.management.db import DB
+
+db = DB()
+
 app = FastAPI()
 
 @app.get("/")
@@ -43,6 +48,12 @@ def huhu():
 		solution="Otto",
 		explanation="Written in line 6 after 'Name:'"
 	)
+
+	#ins_count = db.add_message('from', 'to', ex)
+	db.set_processed(34234)
+
+
+	print(db.by_count(3))
 	
 	return ex