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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 0.0.10
+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.10 → solana_agent-0.0.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "0.0.10"
+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.10 → solana_agent-0.0.12}/solana_agent/ai.py

@@ -47,7 +47,7 @@ class MongoDatabase:
         self._client = AsyncIOMotorClient(db_url)
         self._db = self._client[db_name]
         self._threads = self._db["threads"]
-        self._messages = self._db["messages"]
+        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})
@@ -58,11 +58,15 @@ class MongoDatabase:
 
     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({})
 
+    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:
     def __init__(
@@ -125,7 +129,7 @@ class AI:
                        ] 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)