solana-agent 17.1.2__py3-none-any.whl → 17.1.4__py3-none-any.whl

solana_agent/client/solana_agent.py

@@ -112,26 +112,31 @@ class SolanaAgent(SolanaAgentInterface):
         )
 
     def register_tool(self, tool: Tool) -> bool:
-        """Register a custom tool for use by agents.
+        """
+        Register a tool with the agent system.
 
         Args:
-            tool: Tool implementation following the Tool interface
+            tool: Tool instance to register
 
         Returns:
-            bool: True if registration was successful
-
-        Example:
-            ```python
-            from solana_agent import SolanaAgent
-            from my_tools import CustomTool
-
-            agent = SolanaAgent(config=config)
-            tool = CustomTool()
-            agent.register_tool(tool)
-            ```
+            True if successful, False
         """
+
         try:
-            return self.query_service.agent_service.tool_registry.register_tool(tool)
+            print(f"Attempting to register tool: {tool.name}")
+            success = self.query_service.agent_service.tool_registry.register_tool(
+                tool)
+            if success:
+                print(f"Tool {tool.name} registered successfully")
+                # Get all agents and assign the tool to them
+                agents = self.query_service.agent_service.get_all_ai_agents()
+                for agent_name in agents:
+                    print(f"Assigning {tool.name} to agent {agent_name}")
+                    self.query_service.agent_service.assign_tool_for_agent(
+                        agent_name, tool.name)
+            return success
         except Exception as e:
-            print(f"Error registering tool: {str(e)}")
+            print(f"Error in register_tool: {str(e)}")
+            import traceback
+            print(traceback.format_exc())
             return False

solana_agent/plugins/registry.py

@@ -20,9 +20,14 @@ class ToolRegistry(ToolRegistryInterface):
 
     def register_tool(self, tool: Tool) -> bool:
         """Register a tool with this registry."""
-        self._tools[tool.name] = tool
-        print(f"Registered tool: {tool.name}")
-        return True
+        try:
+            self._tools[tool.name] = tool
+            print(f"Successfully registered tool: {tool.name}")
+            print(f"Current tools in registry: {list(self._tools.keys())}")
+            return True
+        except Exception as e:
+            print(f"Error registering tool: {str(e)}")
+            return False
 
     def get_tool(self, tool_name: str) -> Optional[Tool]:
         """Get a tool by name."""
@@ -31,7 +36,8 @@ class ToolRegistry(ToolRegistryInterface):
     def assign_tool_to_agent(self, agent_name: str, tool_name: str) -> bool:
         """Give an agent access to a specific tool."""
         if tool_name not in self._tools:
-            print(f"Error: Tool {tool_name} is not registered")
+            print(
+                f"Error: Tool {tool_name} is not registered. Available tools: {list(self._tools.keys())}")
             return False
 
         if agent_name not in self._agent_tools:
@@ -39,13 +45,17 @@ class ToolRegistry(ToolRegistryInterface):
 
         if tool_name not in self._agent_tools[agent_name]:
             self._agent_tools[agent_name].append(tool_name)
+            print(
+                f"Successfully assigned tool {tool_name} to agent {agent_name}")
+            print(
+                f"Agent {agent_name} now has access to: {self._agent_tools[agent_name]}")
 
         return True
 
     def get_agent_tools(self, agent_name: str) -> List[Dict[str, Any]]:
         """Get all tools available to an agent."""
         tool_names = self._agent_tools.get(agent_name, [])
-        return [
+        tools = [
             {
                 "name": name,
                 "description": self._tools[name].description,
@@ -53,6 +63,9 @@ class ToolRegistry(ToolRegistryInterface):
             }
             for name in tool_names if name in self._tools
         ]
+        print(
+            f"Tools available to agent {agent_name}: {[t['name'] for t in tools]}")
+        return tools
 
     def list_all_tools(self) -> List[str]:
         """List all registered tools."""

solana_agent/repositories/memory.py

@@ -32,7 +32,7 @@ class MemoryRepository(MemoryProvider):
         elif zep_api_key and zep_base_url:
             self.zep = AsyncZep(api_key=zep_api_key, base_url=zep_base_url)
         else:
-            self.zep = AsyncZep(base_url="http://localhost:8000")
+            self.zep = None
 
     async def store(self, user_id: str, messages: List[Dict[str, Any]]) -> None:
         """Store messages in both Zep and MongoDB."""
@@ -55,6 +55,9 @@ class MemoryRepository(MemoryProvider):
             print(f"MongoDB storage error: {e}")
 
         # Store in Zep with role-based format
+        if not self.zep:
+            return
+
         try:
             try:
                 await self.zep.user.add(user_id=user_id)
@@ -82,6 +85,9 @@ class MemoryRepository(MemoryProvider):
 
     async def retrieve(self, user_id: str) -> str:
         """Retrieve memory context from Zep only."""
+        if not self.zep:
+            return ""
+
         try:
             memory = await self.zep.memory.get(session_id=user_id)
 
@@ -101,6 +107,9 @@ class MemoryRepository(MemoryProvider):
         except Exception as e:
             print(f"MongoDB deletion error: {e}")
 
+        if not self.zep:
+            return
+
         try:
             await self.zep.memory.delete(session_id=user_id)
             await self.zep.user.delete(user_id=user_id)

{solana_agent-17.1.2.dist-info → solana_agent-17.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 17.1.2
+Version: 17.1.4
 Summary: Agentic IQ
 License: MIT
 Keywords: ai,openai,ai agents,agi
@@ -43,7 +43,7 @@ Power your business using Solana Agent!
 ## Features
 
 * Seamless text and audio streaming with real-time multi-modal processing
-* Persistent memory that preserves context across all agent interactions
+* Persistent memory that preserves context across all agent interactions (optional)
 * Intelligent query routing to agents with optimal domain expertise
 * Unified value system ensuring brand-aligned agent responses
 * Powerful tool integration using standard Python packages
@@ -54,7 +54,7 @@ Power your business using Solana Agent!
 * [Python](https://python.org) - Programming Language
 * [OpenAI](https://openai.com) - LLMs
 * [MongoDB](https://mongodb.com) - Database
-* [Zep](https://getzep.com) - Conversational Memory
+* [Zep](https://getzep.com) - Conversational Memory (optional)
 
 ## Installation
 
@@ -84,7 +84,11 @@ config = {
         "database": "solana_agent"
     },
     "openai": {
-        "api_key": "your-openai-key",
+        "api_key": "your-openai-api-key",
+    },
+    "zep": { # optional
+        "api_key": "your-zep-api-key",
+        "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
     },
     "agents": [
         {

{solana_agent-17.1.2.dist-info → solana_agent-17.1.4.dist-info}/RECORD

@@ -3,7 +3,7 @@ solana_agent/adapters/__init__.py,sha256=tiEEuuy0NF3ngc_tGEcRTt71zVI58v3dYY9RvMr
 solana_agent/adapters/llm_adapter.py,sha256=hi2JYj6CvhNLJxznIv_7Ef7Y0mk7aztHd_OMYe2scMQ,6034
 solana_agent/adapters/mongodb_adapter.py,sha256=qqEFbY_v1XGyFXBmwd5HSXSSHnA9wWo-Hm1vGEyIG0k,2718
 solana_agent/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/client/solana_agent.py,sha256=no4TH0G44gGVft3GJKLHf_L_JDAwUkzHk-hwxZZeAxk,5010
+solana_agent/client/solana_agent.py,sha256=Q9vnsoezsdhe6-T_tMb7Gr-697D1Bo2qIpr1-ytP1ak,5361
 solana_agent/domains/__init__.py,sha256=HiC94wVPRy-QDJSSRywCRrhrFfTBeHjfi5z-QfZv46U,168
 solana_agent/domains/agent.py,sha256=Ak_hD5gTCzRqAHLmqtxnny0Xki1qAKR7RzLW9LOQBTg,2930
 solana_agent/domains/routing.py,sha256=UDlgTjUoC9xIBVYu_dnf9-KG_bBgdEXAv_UtDOrYo0w,650
@@ -21,17 +21,17 @@ solana_agent/interfaces/services/query.py,sha256=bspnm-CN6zjRWnlFnkl34qo0EIW5m2T
 solana_agent/interfaces/services/routing.py,sha256=gohkt5f9uYDLpu4iDVDk9yj8js9P56R6QHSIDNylgwA,438
 solana_agent/plugins/__init__.py,sha256=coZdgJKq1ExOaj6qB810i3rEhbjdVlrkN76ozt_Ojgo,193
 solana_agent/plugins/manager.py,sha256=GWwhfMBn9THwVn7biOvVa25GLthCA1ilWIoDkt5hXNI,5084
-solana_agent/plugins/registry.py,sha256=dRKWoOEqiU7OLsjpBWf4VJfDQYZdJPjW5AKxeITmVMA,2283
+solana_agent/plugins/registry.py,sha256=CyZpfjLxV3cVtDnXvCVcMgXg_XEijDDsKuJ0bqG64cA,2900
 solana_agent/plugins/tools/__init__.py,sha256=c0z7ij42gs94_VJrcn4Y8gUlTxMhsFNY6ahIsNswdLk,231
 solana_agent/plugins/tools/auto_tool.py,sha256=Z3CcOzwdXpzciH-5yphhd9qt1b9owTxhwC-dYmPF6B0,1489
 solana_agent/repositories/__init__.py,sha256=fP83w83CGzXLnSdq-C5wbw9EhWTYtqE2lQTgp46-X_4,163
 solana_agent/repositories/agent.py,sha256=e1rnsQiigkKwJNLKro86a3b6TBiky3GMfmCRc5b_jPw,3187
-solana_agent/repositories/memory.py,sha256=GABGwaz00thjviHewLvb18NeKE8dkBROxy_stsiiWrE,4722
+solana_agent/repositories/memory.py,sha256=DrhaVxlE-iAOmX0sfDCqgdPJauYvrnPd7rmVlf6_HGE,4822
 solana_agent/services/__init__.py,sha256=ab_NXJmwYUCmCrCzuTlZ47bJZINW0Y0F5jfQ9OovidU,163
 solana_agent/services/agent.py,sha256=j0aI_BGaY5Nhkb9ga0r4BqI3qMKu5TOc4QPQsU3NHyQ,17000
 solana_agent/services/query.py,sha256=rm7XlTCcp5NeorIaLUdr7rdxtWCgg1Q1qe5YuI1bumo,11246
 solana_agent/services/routing.py,sha256=TPJ2Pas4acE93QzMEV6ZP670OtTNrVEPa76fz6urEV4,4996
-solana_agent-17.1.2.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-17.1.2.dist-info/METADATA,sha256=NRDlTqIZjUxvhy6Z3liDI5IubC-XJ2VV5S-9vzpzpjw,4547
-solana_agent-17.1.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-17.1.2.dist-info/RECORD,,
+solana_agent-17.1.4.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-17.1.4.dist-info/METADATA,sha256=o_zPKHGGkqqQpbwtnfjPwlFhgBQWW_Ytupx0rA6Bx7k,4720
+solana_agent-17.1.4.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-17.1.4.dist-info/RECORD,,