fix bugs

Author: truemagic-coder

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "0.0.10"
+version = "0.0.11"
 description = "Build self-learning AI Agents"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"

solana_agent/ai.py

@@ -46,22 +46,22 @@ class MongoDatabase:
     def __init__(self, db_url: str, db_name: str):
         self._client = AsyncIOMotorClient(db_url)
         self._db = self._client[db_name]
-        self._threads = self._db["threads"]
-        self._messages = self._db["messages"]
+        self.threads = self._db["threads"]
+        self.messages = self._db["messages"]
 
     async def save_thread_id(self, user_id: str, thread_id: str):
-        await self._threads.insert_one({"thread_id": thread_id, "user_id": user_id})
+        await self.threads.insert_one({"thread_id": thread_id, "user_id": user_id})
 
     async def get_thread_id(self, user_id: str) -> Optional[str]:
-        document = await self._threads.find_one({"user_id": user_id})
+        document = await self.threads.find_one({"user_id": user_id})
         return document["thread_id"] if document else None
 
     async def save_message(self, user_id: str, metadata: Dict[str, Any]):
         metadata["user_id"] = user_id
-        await self._messages.insert_one(metadata)
+        await self.messages.insert_one(metadata)
 
     async def delete_all_threads(self):
-        await self._threads.delete_many({})
+        await self.threads.delete_many({})
 
 
 class AI:
@@ -125,7 +125,7 @@ def __init__(
                        ] if code_interpreter else []
         self._tool_handlers = {}
         self._assistant_id = None
-        self._database = database
+        self._database: MongoDatabase = database
         self._accumulated_value_queue = asyncio.Queue()
         self._zep = (
             AsyncZep(api_key=zep_api_key)