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

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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "2.0.0"
+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.0 → 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
@@ -101,11 +105,27 @@ class AI:
         """
         self._client = OpenAI(api_key=openai_api_key, base_url=openai_base_url) if openai_base_url else OpenAI(
             api_key=openai_api_key)
-        self._instructions = instructions
+        memory_instructions = """
+            You are a highly intelligent, context-aware conversational AI. When a user sends a query or statement, you should not only process the current input but also retrieve and integrate relevant context from their previous interactions. Use the memory data to:
+            - Infer nuances in the user's intent.
+            - Recall previous topics, preferences, or facts that might be relevant.
+            - Provide a thoughtful, clear, and structured response.
+            - Clarify ambiguous queries by relating them to known user history.
+
+            Always be concise and ensure that your response maintains coherence across the conversation while respecting the user's context and previous data.
+        """
+        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
@@ -766,11 +786,12 @@ class AI:
                 messages=[
                     {
                         "role": "system",
-                        "content": self._instructions,
+                        "content": self._instructions + f" Memory: {memory}",
+
                     },
                     {
                         "role": "user",
-                        "content": f"Query: {user_text}, Memory: {memory}",
+                        "content": user_text,
                     },
                 ],
                 tools=self._tools,
@@ -927,11 +948,11 @@ class AI:
                 messages=[
                     {
                         "role": "system",
-                        "content": self._instructions,
+                        "content": self._instructions + f" Memory: {memory}",
                     },
                     {
                         "role": "user",
-                        "content": f"Query: {transcript}, Memory: {memory}",
+                        "content":  transcript,
                     },
                 ],
                 tools=self._tools,