solana-agent 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl

solana_agent/ai.py

@@ -13,7 +13,7 @@ import inspect
 import requests
 from zep_cloud.client import AsyncZep
 from zep_cloud.client import Zep
-from zep_cloud.types import Message, RoleType
+from zep_cloud.types import Message
 import pandas as pd
 from pinecone import Pinecone
 
@@ -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:
@@ -692,12 +696,12 @@ class AI:
             messages = [
                 Message(
                     role="user",
-                    role_type=RoleType["user"],
+                    role_type="user",
                     content=user_text,
                 ),
                 Message(
                     role="assistant",
-                    role_type=RoleType["assistant"],
+                    role_type="assistant",
                     content=full_response,
                 ),
             ]
@@ -806,12 +810,12 @@ class AI:
             messages = [
                 Message(
                     role="user",
-                    role_type=RoleType["user"],
+                    role_type="user",
                     content=transcript,
                 ),
                 Message(
                     role="assistant",
-                    role_type=RoleType["assistant"],
+                    role_type="assistant",
                     content=full_response,
                 ),
             ]

{solana_agent-0.0.11.dist-info → solana_agent-0.0.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 0.0.11
+Version: 0.0.13
 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.13.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=-01z87VNJ6kr-WzyPXCinSIlYZsaWgJWSkN7f5iDJvM,33538
+solana_agent-0.0.13.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-0.0.13.dist-info/METADATA,sha256=oWgOt9pc9th3N748QxEdxabQ1GS3l1KXfAZ72J9R4tg,5577
+solana_agent-0.0.13.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+solana_agent-0.0.13.dist-info/RECORD,,

solana_agent-0.0.11.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=RVT-1F16Z-RYOmn8rdWtqffIqyRvtahL1NG1LSYj6-s,33406
-solana_agent-0.0.11.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-0.0.11.dist-info/METADATA,sha256=RPS-y92uBtkcvYR2-cJo2YZd_46i9B66dnE7B3ReBds,5577
-solana_agent-0.0.11.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-solana_agent-0.0.11.dist-info/RECORD,,