solana-agent 17.1.3__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-17.1.3.dist-info → solana_agent-17.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 17.1.3
+Version: 17.1.4
 Summary: Agentic IQ
 License: MIT
 Keywords: ai,openai,ai agents,agi

{solana_agent-17.1.3.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,7 +21,7 @@ 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
@@ -31,7 +31,7 @@ solana_agent/services/__init__.py,sha256=ab_NXJmwYUCmCrCzuTlZ47bJZINW0Y0F5jfQ9Oo
 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.3.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-17.1.3.dist-info/METADATA,sha256=vrwcjIs5-jurOcjV-_yIrHRwezMJAnci0-8WNfzIjZE,4720
-solana_agent-17.1.3.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-17.1.3.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,,