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

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

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

{solana_agent-17.1.3 → solana_agent-17.1.5}/pyproject.toml

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

{solana_agent-17.1.3 → solana_agent-17.1.5}/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.5}/solana_agent/factories/agent_factory.py

@@ -72,6 +72,7 @@ class SolanaAgentFactory:
             agent_repository=agent_repo,
             llm_provider=llm_adapter,
             organization_mission=organization_mission,
+            config=config,
         )
 
         # Debug the agent service tool registry

{solana_agent-17.1.3 → solana_agent-17.1.5}/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 → solana_agent-17.1.5}/solana_agent/services/agent.py

@@ -12,7 +12,8 @@ from typing import AsyncGenerator, Dict, List, Literal, Optional, Any, Union
 from solana_agent.interfaces.services.agent import AgentService as AgentServiceInterface
 from solana_agent.interfaces.providers.llm import LLMProvider
 from solana_agent.interfaces.repositories.agent import AgentRepository
-from solana_agent.interfaces.plugins.plugins import ToolRegistry
+from solana_agent.interfaces.plugins.plugins import ToolRegistry as ToolRegistryInterface
+from solana_agent.plugins.registry import ToolRegistry
 from solana_agent.domains.agent import AIAgent, OrganizationMission
 
 
@@ -25,7 +26,7 @@ class AgentService(AgentServiceInterface):
         agent_repository: AgentRepository,
         organization_mission: Optional[OrganizationMission] = None,
         config: Optional[Dict[str, Any]] = None,
-        tool_registry: Optional[ToolRegistry] = None,
+        tool_registry: Optional[ToolRegistryInterface] = None,
     ):
         """Initialize the agent service.
 
@@ -34,6 +35,7 @@ class AgentService(AgentServiceInterface):
             agent_repository: Repository for agent data
             organization_mission: Optional organization mission and values
             config: Optional service configuration
+            tool_registry: Optional tool registry
         """
         self.llm_provider = llm_provider
         self.agent_repository = agent_repository
@@ -45,9 +47,9 @@ class AgentService(AgentServiceInterface):
         if tool_registry:
             self.tool_registry = tool_registry
         else:
-            # Import the concrete implementation
-            from solana_agent.plugins.registry import ToolRegistry as ConcreteToolRegistry
-            self.tool_registry = ConcreteToolRegistry()
+            config = self.config
+            print("Initializing tool registry with config")
+            self.tool_registry = ToolRegistry(config=config)
 
         # Will be set by factory if plugin system is enabled
         self.plugin_manager = None