solana-agent 17.1.3__tar.gz → 17.1.4__tar.gz

{solana_agent-17.1.3 → solana_agent-17.1.4}/PKG-INFO

@@ -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 → solana_agent-17.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "17.1.3"
+version = "17.1.4"
 description = "Agentic IQ"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"

{solana_agent-17.1.3 → solana_agent-17.1.4}/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-17.1.3 → solana_agent-17.1.4}/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."""