solana-agent 0.0.11__tar.gz → 0.0.12__tar.gz

{solana_agent-0.0.11 → solana_agent-0.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 0.0.11
+Version: 0.0.12
 Summary: Build self-learning AI Agents
 License: MIT
 Keywords: ai,openai,ai agents
@@ -22,7 +22,7 @@ Requires-Dist: pandas (>=2.2.3,<3.0.0)
 Requires-Dist: pinecone (>=6.0.1,<7.0.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
-Requires-Dist: zep-cloud (>=2.3.1,<3.0.0)
+Requires-Dist: zep-cloud (>=2.4.0,<3.0.0)
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent
 Description-Content-Type: text/markdown
 

{solana_agent-0.0.11 → solana_agent-0.0.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "0.0.11"
+version = "0.0.12"
 description = "Build self-learning AI Agents"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -21,7 +21,7 @@ python = ">=3.9,<4.0"
 openai = "^1.61.1"
 pydantic = "^2.10.6"
 motor = "^3.7.0"
-zep-cloud = "^2.3.1"
+zep-cloud = "^2.4.0"
 requests = "^2.32.3"
 pandas = "^2.2.3"
 pinecone = "^6.0.1"

{solana_agent-0.0.11 → solana_agent-0.0.12}/solana_agent/ai.py

@@ -46,14 +46,14 @@ 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._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]):
@@ -61,7 +61,11 @@ class MongoDatabase:
         await self.messages.insert_one(metadata)
 
     async def delete_all_threads(self):
-        await self.threads.delete_many({})
+        await self._threads.delete_many({})
+
+    async def clear_user_history(self, user_id: str):
+        await self.messages.delete_many({"user_id": user_id})
+        await self._threads.delete_one({"user_id": user_id})
 
 
 class AI: