solana-agent 2.0.0__py3-none-any.whl → 2.1.0__py3-none-any.whl

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,

{solana_agent-2.0.0.dist-info → solana_agent-2.1.0.dist-info}/METADATA

@@ -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.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
+solana_agent/ai.py,sha256=TM1oqJtPE5cfIxtRiy2uyTRgH2ux4o7Fy_0P3HUkcss,44006
+solana_agent-2.1.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-2.1.0.dist-info/METADATA,sha256=pw_VcbC-0q3OEH2Tx2xLEASOpPyuh-KKPjYvAhaKTCo,4769
+solana_agent-2.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-2.1.0.dist-info/RECORD,,

solana_agent-2.0.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-solana_agent/__init__.py,sha256=zpfnWqANd3OHGWm7NCF5Y6m01BWG4NkNk8SK9Ex48nA,18
-solana_agent/ai.py,sha256=4D2g2B-30Qw_1uKuD2s_anBg8CUL95j0Eusc69bs4L4,42730
-solana_agent-2.0.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-2.0.0.dist-info/METADATA,sha256=BAHvcEQ1JFVWfrZBf0BKvd7dDPl43SjWkYhBkv1UTMQ,4726
-solana_agent-2.0.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-2.0.0.dist-info/RECORD,,