solana-agent 17.1.10__py3-none-any.whl → 17.1.11__py3-none-any.whl

solana_agent/plugins/registry.py

@@ -42,15 +42,17 @@ class ToolRegistry(ToolRegistryInterface):
                 f"Error: Tool {tool_name} is not registered. Available tools: {list(self._tools.keys())}")
             return False
 
+        # Initialize agent's tool list if not exists
         if agent_name not in self._agent_tools:
-            self._agent_tools[agent_name] = []
+            self._agent_tools[agent_name] = [tool_name]
+        elif tool_name not in self._agent_tools[agent_name]:
+            # Add new tool to existing list
+            self._agent_tools[agent_name] = [
+                *self._agent_tools[agent_name], tool_name]
 
-        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]}")
+        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
 

solana_agent/services/agent.py

@@ -241,8 +241,6 @@ class AgentService(AgentServiceInterface):
             is_json = False
             text_buffer = ""
 
-            print("\n=== Starting Response Generation ===")
-
             # Generate and stream response
             async for chunk in self.llm_provider.generate_text(
                 prompt=query_text,
@@ -260,42 +258,23 @@ class AgentService(AgentServiceInterface):
                     try:
                         # Try to parse complete JSON
                         data = json.loads(json_buffer)
-                        print(
-                            f"Successfully parsed JSON: {json.dumps(data, indent=2)}")
 
                         # Valid JSON found, handle it
-                        if "tool_calls" in data:  # Now looking for tool_calls array
-                            tool_results = []
-                            async for tool_result in self._handle_multiple_tool_calls(
+                        if "tool_call" in data:
+                            # Process tool call with existing method
+                            response_text = await self._handle_tool_call(
                                 agent_name=agent_name,
                                 json_chunk=json_buffer
-                            ):
-                                tool_results.append(tool_result)
-
-                            # Combine results and create a new prompt with clear instructions
-                            tool_response = "\n".join(tool_results)
-                            process_prompt = f"""
-                                {tool_response}
-
-                                IMPORTANT INSTRUCTIONS:
-                                1. Maintain ALL factual details
-                                2. Include ALL statistics, numbers, and specific data points
-                                3. Use direct quotes where relevant
-                                4. Keep ALL source citations and references
-                                5. DO NOT omit or summarize away important details
-                                6. DO NOT add any information not present in the results
-                                7. DO NOT make any new tool calls or return JSON
-                            """
-
-                            # Process combined results through LLM with modified system prompt
-                            summary_system_prompt = self.get_agent_system_prompt(agent_name) + \
-                                "\n DO NOT make any tool calls or return JSON. Present ALL facts and maintain ALL details from the source material."
+                            )
+
+                            system_prompt = self.get_agent_system_prompt(agent_name) + \
+                                "\n DO NOT make any tool calls or return JSON."
 
                             # Collect all processed text first
                             processed_text = ""
                             async for processed_chunk in self.llm_provider.generate_text(
-                                prompt=process_prompt,
-                                system_prompt=summary_system_prompt,
+                                prompt=response_text,
+                                system_prompt=system_prompt,
                             ):
                                 processed_text += processed_chunk
                                 # For text output, yield chunks as they come
@@ -388,6 +367,30 @@ class AgentService(AgentServiceInterface):
             import traceback
             print(traceback.format_exc())
 
+    async def _handle_tool_call(
+        self,
+        agent_name: str,
+        json_chunk: str,
+    ) -> str:
+        """Handle tool calls and return formatted response."""
+        try:
+            data = json.loads(json_chunk)
+            if "tool_call" in data:
+                tool_data = data["tool_call"]
+                tool_name = tool_data.get("name")
+                parameters = tool_data.get("parameters", {})
+
+                if tool_name:
+                    result = self.execute_tool(
+                        agent_name, tool_name, parameters)
+                    if result.get("status") == "success":
+                        return result.get("result", "")
+                    else:
+                        return f"I apologize, but I encountered an issue: {result.get('message', 'Unknown error')}"
+            return json_chunk
+        except json.JSONDecodeError:
+            return json_chunk
+
     def _get_tool_usage_prompt(self, agent_name: str) -> str:
         """Generate JSON-based instructions for tool usage."""
         # Get tools assigned to this agent
@@ -399,94 +402,27 @@ class AgentService(AgentServiceInterface):
         available_tool_names = [tool.get("name", "") for tool in tools]
         tools_json = json.dumps(tools, indent=2)
 
-        # Create tool example if search is available
-        tool_example = ""
-        if "search_internet" in available_tool_names:
-            tool_example = """
-    For latest news query:
-    {
-        "tool_calls": [{
-            "name": "search_internet",
-            "parameters": {
-                "query": "latest Solana blockchain news March 2025"
-            }
-        }]
-    }"""
-
         return f"""
     AVAILABLE TOOLS:
     {tools_json}
     
     TOOL USAGE FORMAT:
     {{
-        "tool_calls": [{{
+        "tool_call": {{
             "name": "<one_of:{', '.join(available_tool_names)}>",
             "parameters": {{
                 // parameters as specified in tool definition above
             }}
-        }}]
+        }}
     }}
     
-    {tool_example if tool_example else ''}
-    
     RESPONSE RULES:
     1. For tool usage:
        - Only use tools from the AVAILABLE TOOLS list above
        - Follow the exact parameter format shown in the tool definition
-       - Include "March 2025" in any search queries for current information
     
     2. Format Requirements:
        - Return ONLY the JSON object for tool calls
        - No explanation text before or after
        - Use exact tool names as shown in AVAILABLE TOOLS
     """
-
-    async def _handle_multiple_tool_calls(
-        self,
-        agent_name: str,
-        json_chunk: str,
-    ) -> AsyncGenerator[str, None]:
-        """Handle multiple tool calls concurrently and yield results as they complete."""
-        try:
-            data = json.loads(json_chunk)
-            if "tool_calls" not in data:
-                yield json_chunk
-                return
-
-            tool_calls = data["tool_calls"]
-
-            if not isinstance(tool_calls, list):
-                print("Error: tool_calls is not a list")
-                yield "Error: 'tool_calls' must be an array of tool calls."
-                return
-
-            # Define individual tool execution coroutine
-            async def execute_single_tool(tool_info):
-                tool_name = tool_info.get("name")
-                parameters = tool_info.get("parameters", {})
-                print(f"\nExecuting tool: {tool_name}")
-                print(f"With parameters: {parameters}")
-
-                if not tool_name:
-                    return f"Error: Missing tool name in tool call."
-
-                result = self.execute_tool(agent_name, tool_name, parameters)
-
-                if result.get("status") == "success":
-                    return f"Result from {tool_name}: {result.get('result', '')}"
-                else:
-                    return f"Error from {tool_name}: {result.get('message', 'Unknown error')}"
-
-            # Execute all tool calls concurrently
-            tasks = [execute_single_tool(tool_call)
-                     for tool_call in tool_calls]
-            for task in asyncio.as_completed(tasks):
-                result = await task
-                yield result
-
-        except json.JSONDecodeError:
-            yield "Error: Could not parse tool calls JSON."
-        except Exception as e:
-            import traceback
-            print(traceback.format_exc())
-            yield f"Error processing tool calls: {str(e)}"

{solana_agent-17.1.10.dist-info → solana_agent-17.1.11.dist-info}/METADATA

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

{solana_agent-17.1.10.dist-info → solana_agent-17.1.11.dist-info}/RECORD

@@ -20,16 +20,16 @@ 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=Z41sW_E8vObg16gA9gB7IrcH5mRGFJMeNC879zJUDRY,2948
+solana_agent/plugins/registry.py,sha256=ZUeczZ9bkdm9G6NBMDpY7sglOW2IrJLiLhboMNAHors,3072
 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/memory.py,sha256=cDGoRz8FEkjwCE7j0XvA03-NL0TyROAt4_uwx288Th0,4790
 solana_agent/services/__init__.py,sha256=ab_NXJmwYUCmCrCzuTlZ47bJZINW0Y0F5jfQ9OovidU,163
-solana_agent/services/agent.py,sha256=fHE5deFvIREDY-99QDxG_JtiHBcYB5ontvKvi5I3oF8,19542
+solana_agent/services/agent.py,sha256=yyLAzVsqhYi8qhqoJ1O_IA3bkXxMDJ_-K5qVWKSrlNY,16641
 solana_agent/services/query.py,sha256=qXrvzAyMqESdF8QD3xYaz2vyfR7ndLpsh2TahYQ-LYg,10414
 solana_agent/services/routing.py,sha256=IPvBicgTYXqQ8iIRaatCsBGQVsOBGdAkq2i6U8hZlOY,6479
-solana_agent-17.1.10.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-17.1.10.dist-info/METADATA,sha256=EWDPkKhVIo-z3nbbcYO5FHQwoSVAsMU5MZyhXbGE8f8,4693
-solana_agent-17.1.10.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-17.1.10.dist-info/RECORD,,
+solana_agent-17.1.11.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-17.1.11.dist-info/METADATA,sha256=ZgNTbcRdnlM_QCFkLYhnSFtnG6OcZk5HGAr9-PKnF94,4693
+solana_agent-17.1.11.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-17.1.11.dist-info/RECORD,,