solana-agent 2.0.1__tar.gz → 2.1.0__tar.gz

{solana_agent-2.0.1 → solana_agent-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 2.0.1
+Version: 2.1.0
 Summary: Build self-learning AI Agents
 License: MIT
 Keywords: ai,openai,ai agents
@@ -23,6 +23,7 @@ Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: pymongo (>=4.11.1,<5.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: zep-cloud (>=2.4.0,<3.0.0)
+Requires-Dist: zep-python (>=2.0.2,<3.0.0)
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent
 Description-Content-Type: text/markdown
 

{solana_agent-2.0.1 → solana_agent-2.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "2.0.1"
+version = "2.1.0"
 description = "Build self-learning AI Agents"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -22,6 +22,7 @@ openai = "^1.64.0"
 pydantic = "^2.10.6"
 pymongo = "^4.11.1"
 zep-cloud = "^2.4.0"
+zep-python = "^2.0.2"
 requests = "^2.32.3"
 pinecone = "^6.0.1"
 pandas = "^2.2.3"

{solana_agent-2.0.1 → solana_agent-2.1.0}/solana_agent/ai.py

@@ -9,8 +9,10 @@ from pymongo import MongoClient
 from openai import OpenAI
 import inspect
 import requests
-from zep_cloud.client import AsyncZep
-from zep_cloud.client import Zep
+from zep_cloud.client import AsyncZep as AsyncZepCloud
+from zep_cloud.client import Zep as ZepCloud
+from zep_python.client import Zep
+from zep_python.client import AsyncZep
 from zep_cloud.types import Message
 from pinecone import Pinecone
 
@@ -58,6 +60,7 @@ class AI:
         instructions: str,
         database: Any,
         zep_api_key: str = None,
+        zep_base_url: str = None,
         perplexity_api_key: str = None,
         grok_api_key: str = None,
         pinecone_api_key: str = None,
@@ -74,6 +77,7 @@ class AI:
             instructions (str): Base behavioral instructions for the AI
             database (Any): Database instance for message/thread storage
             zep_api_key (str, optional): API key for Zep memory storage. Defaults to None
+            zep_base_url (str, optional): Base URL for Zep API. Defaults to None
             perplexity_api_key (str, optional): API key for Perplexity search. Defaults to None
             grok_api_key (str, optional): API key for X/Twitter search via Grok. Defaults to None
             pinecone_api_key (str, optional): API key for Pinecone. Defaults to None
@@ -113,8 +117,15 @@ class AI:
         self._instructions = instructions + " " + memory_instructions
         self._database: MongoDatabase = database
         self._accumulated_value_queue = asyncio.Queue()
-        self._zep = AsyncZep(api_key=zep_api_key) if zep_api_key else None
-        self._sync_zep = Zep(api_key=zep_api_key) if zep_api_key else None
+        if zep_api_key and not zep_base_url:
+            self._zep = AsyncZepCloud(api_key=zep_api_key)
+            self._sync_zep = ZepCloud(api_key=zep_api_key)
+        elif zep_api_key and zep_base_url:
+            self._zep = AsyncZep(api_key=zep_api_key, base_url=zep_base_url)
+            self._sync_zep = Zep(api_key=zep_api_key, base_url=zep_base_url)
+        else:
+            self._zep = None
+            self._sync_zep = None
         self._perplexity_api_key = perplexity_api_key
         self._grok_api_key = grok_api_key
         self._gemini_api_key = gemini_api_key