hanzo-mcp 0.7.7__py3-none-any.whl → 0.8.1__py3-none-any.whl

hanzo_mcp/tools/agent/swarm_tool_v1_deprecated.py

@@ -4,25 +4,31 @@ This module implements the SwarmTool that enables both parallel execution of mul
 agent instances and hierarchical workflows with specialized roles.
 """
 
-import asyncio
-import json
 import os
-from typing import Annotated, Any, TypedDict, Unpack, final, override, Optional, Dict, List
-from pathlib import Path
+import asyncio
+from typing import (
+    Any,
+    Dict,
+    List,
+    Unpack,
+    Optional,
+    TypedDict,
+    final,
+    override,
+)
 
 from mcp.server import FastMCP
 from mcp.server.fastmcp import Context as MCPContext
-from pydantic import Field
 
-from hanzo_mcp.tools.agent.agent_tool import AgentTool
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
+from hanzo_mcp.tools.agent.agent_tool import AgentTool
 from hanzo_mcp.tools.common.permissions import PermissionManager
 
 
 class AgentNode(TypedDict):
     """Node in the agent network.
-    
+
     Attributes:
         id: Unique identifier for this agent
         query: The specific query/task for this agent
@@ -32,6 +38,7 @@ class AgentNode(TypedDict):
         receives_from: Optional list of agent IDs this agent receives input from
         file_path: Optional specific file for the agent to work on
     """
+
     id: str
     query: str
     model: Optional[str]
@@ -43,12 +50,13 @@ class AgentNode(TypedDict):
 
 class SwarmConfig(TypedDict):
     """Configuration for an agent network.
-    
+
     Attributes:
         agents: Dictionary of agent configurations keyed by ID
         entry_point: ID of the first agent to execute (optional, defaults to finding roots)
         topology: Optional topology type (tree, dag, pipeline, star, mesh)
     """
+
     agents: Dict[str, AgentNode]
     entry_point: Optional[str]
     topology: Optional[str]
@@ -56,13 +64,14 @@ class SwarmConfig(TypedDict):
 
 class SwarmToolParams(TypedDict):
     """Parameters for the SwarmTool.
-    
+
     Attributes:
         config: Agent network configuration
         query: Initial query to send to entry point agent(s)
         context: Optional context shared by all agents
         max_concurrent: Maximum number of concurrent agents (default: 10)
     """
+
     config: SwarmConfig
     query: str
     context: Optional[str]
@@ -72,17 +81,17 @@ class SwarmToolParams(TypedDict):
 @final
 class SwarmTool(BaseTool):
     """Tool for executing multiple agent tasks in parallel.
-    
+
     The SwarmTool enables efficient parallel processing of multiple files or tasks
     by spawning independent agent instances for each task.
     """
-    
+
     @property
     @override
     def name(self) -> str:
         """Get the tool name."""
         return "swarm"
-    
+
     @property
     @override
     def description(self) -> str:
@@ -160,7 +169,7 @@ Models can be specified as:
 - Short: 'claude-3-5-sonnet', 'gpt-4o', 'gemini-1.5-pro'
 - CLI tools: 'claude_cli', 'codex_cli', 'gemini_cli', 'grok_cli'
 """
-    
+
     def __init__(
         self,
         permission_manager: PermissionManager,
@@ -172,7 +181,7 @@ Models can be specified as:
         agent_max_tool_uses: int = 30,
     ):
         """Initialize the swarm tool.
-        
+
         Args:
             permission_manager: Permission manager for access control
             model: Optional model name override (defaults to Claude Sonnet)
@@ -185,13 +194,18 @@ Models can be specified as:
         self.permission_manager = permission_manager
         # Default to latest Claude Sonnet if no model specified
         from hanzo_mcp.tools.agent.code_auth import get_latest_claude_model
+
         self.model = model or f"anthropic/{get_latest_claude_model()}"
-        self.api_key = api_key or os.environ.get("ANTHROPIC_API_KEY") or os.environ.get("CLAUDE_API_KEY")
+        self.api_key = (
+            api_key
+            or os.environ.get("ANTHROPIC_API_KEY")
+            or os.environ.get("CLAUDE_API_KEY")
+        )
         self.base_url = base_url
         self.max_tokens = max_tokens
         self.agent_max_iterations = agent_max_iterations
         self.agent_max_tool_uses = agent_max_tool_uses
-        
+
     @override
     async def call(
         self,
@@ -199,48 +213,50 @@ Models can be specified as:
         **params: Unpack[SwarmToolParams],
     ) -> str:
         """Execute the swarm tool.
-        
+
         Args:
             ctx: MCP context
             **params: Tool parameters
-            
+
         Returns:
             Combined results from all agents
         """
         tool_ctx = create_tool_context(ctx)
         await tool_ctx.set_tool_info(self.name)
-        
+
         # Extract parameters
         agents = params.get("agents", [])
         manager_query = params.get("manager_query")
         reviewer_query = params.get("reviewer_query")
         common_context = params.get("common_context", "")
         max_concurrent = params.get("max_concurrent", 10)
-        
+
         if not agents:
             await tool_ctx.error("No agents provided")
             return "Error: At least one agent must be provided."
-        
+
         # Extract parameters
         config = params.get("config", {})
         initial_query = params.get("query", "")
         context = params.get("context", "")
-        
+
         agents_config = config.get("agents", {})
         entry_point = config.get("entry_point")
-        
-        await tool_ctx.info(f"Starting swarm execution with {len(agents_config)} agents")
-        
+
+        await tool_ctx.info(
+            f"Starting swarm execution with {len(agents_config)} agents"
+        )
+
         # Build agent network
         agent_instances = {}
         agent_results = {}
         execution_queue = asyncio.Queue()
         completed_agents = set()
-        
+
         # Create agent instances
         for agent_id, agent_config in agents_config.items():
             model = agent_config.get("model", self.model)
-            
+
             # Support CLI tools
             cli_tools = {
                 "claude_cli": self._get_cli_tool("claude_cli"),
@@ -248,7 +264,7 @@ Models can be specified as:
                 "gemini_cli": self._get_cli_tool("gemini_cli"),
                 "grok_cli": self._get_cli_tool("grok_cli"),
             }
-            
+
             if model in cli_tools:
                 agent = cli_tools[model]
             else:
@@ -262,9 +278,9 @@ Models can be specified as:
                     max_iterations=self.agent_max_iterations,
                     max_tool_uses=self.agent_max_tool_uses,
                 )
-            
+
             agent_instances[agent_id] = agent
-        
+
         # Find entry points (agents with no incoming connections)
         if entry_point:
             await execution_queue.put((entry_point, initial_query, {}))
@@ -276,83 +292,94 @@ Models can be specified as:
                     # Check if any other agent connects to this one
                     has_incoming = False
                     for other_config in agents_config.values():
-                        if other_config.get("connections") and agent_id in other_config["connections"]:
+                        if (
+                            other_config.get("connections")
+                            and agent_id in other_config["connections"]
+                        ):
                             has_incoming = True
                             break
                     if not has_incoming:
                         roots.append(agent_id)
-            
+
             if not roots:
                 await tool_ctx.error("No entry point found in agent network")
                 return "Error: Could not determine entry point for agent network"
-            
+
             for root in roots:
                 await execution_queue.put((root, initial_query, {}))
-        
+
         # Execute agents in network order
-        async def execute_agent(agent_id: str, query: str, inputs: Dict[str, str]) -> str:
+        async def execute_agent(
+            agent_id: str, query: str, inputs: Dict[str, str]
+        ) -> str:
             """Execute a single agent in the network."""
             async with semaphore:
                 try:
                     agent_config = agents_config[agent_id]
                     agent = agent_instances[agent_id]
-                    
-                    await tool_ctx.info(f"Executing agent: {agent_id} ({agent_config.get('role', 'Agent')})")
-                    
+
+                    await tool_ctx.info(
+                        f"Executing agent: {agent_id} ({agent_config.get('role', 'Agent')})"
+                    )
+
                     # Build prompt with context and inputs
                     prompt_parts = []
-                    
+
                     # Add role context
                     if agent_config.get("role"):
                         prompt_parts.append(f"Your role: {agent_config['role']}")
-                    
+
                     # Add shared context
                     if context:
                         prompt_parts.append(f"Context:\n{context}")
-                    
+
                     # Add inputs from connected agents
                     if inputs:
                         prompt_parts.append("Input from previous agents:")
                         for input_agent, input_result in inputs.items():
-                            prompt_parts.append(f"\n--- From {input_agent} ---\n{input_result}")
-                    
+                            prompt_parts.append(
+                                f"\n--- From {input_agent} ---\n{input_result}"
+                            )
+
                     # Add file context if specified
                     if agent_config.get("file_path"):
-                        prompt_parts.append(f"\nFile to work on: {agent_config['file_path']}")
-                    
+                        prompt_parts.append(
+                            f"\nFile to work on: {agent_config['file_path']}"
+                        )
+
                     # Add the main query
                     prompt_parts.append(f"\nTask: {agent_config['query']}")
-                    
+
                     # Combine query with initial query if this is entry point
-                    if query and query != agent_config['query']:
+                    if query and query != agent_config["query"]:
                         prompt_parts.append(f"\nMain objective: {query}")
-                    
+
                     full_prompt = "\n\n".join(prompt_parts)
-                    
+
                     # Execute the agent
                     result = await agent.call(ctx, prompts=full_prompt)
-                    
+
                     await tool_ctx.info(f"Agent {agent_id} completed")
                     return result
-                    
+
                 except Exception as e:
                     error_msg = f"Agent {agent_id} failed: {str(e)}"
                     await tool_ctx.error(error_msg)
                     return f"Error: {error_msg}"
-        
+
         # Process agent network
         running_tasks = set()
-        
+
         while not execution_queue.empty() or running_tasks:
             # Start new tasks up to concurrency limit
             while not execution_queue.empty() and len(running_tasks) < max_concurrent:
                 agent_id, query, inputs = await execution_queue.get()
-                
+
                 if agent_id not in completed_agents:
                     # Check if all dependencies are met
                     agent_config = agents_config[agent_id]
                     receives_from = agent_config.get("receives_from", [])
-                    
+
                     # Collect inputs from dependencies
                     ready = True
                     for dep in receives_from:
@@ -363,38 +390,42 @@ Models can be specified as:
                             break
                         else:
                             inputs[dep] = agent_results[dep]
-                    
+
                     if ready:
                         # Execute agent
-                        task = asyncio.create_task(execute_agent(agent_id, query, inputs))
+                        task = asyncio.create_task(
+                            execute_agent(agent_id, query, inputs)
+                        )
                         running_tasks.add(task)
-                        
+
                         async def handle_completion(task, agent_id=agent_id):
                             result = await task
                             agent_results[agent_id] = result
                             completed_agents.add(agent_id)
                             running_tasks.discard(task)
-                            
+
                             # Queue connected agents
                             agent_config = agents_config[agent_id]
                             connections = agent_config.get("connections", [])
                             for next_agent in connections:
                                 if next_agent in agents_config:
-                                    await execution_queue.put((next_agent, "", {agent_id: result}))
-                        
+                                    await execution_queue.put(
+                                        (next_agent, "", {agent_id: result})
+                                    )
+
                         asyncio.create_task(handle_completion(task))
-            
+
             # Wait a bit if we're at capacity
             if running_tasks:
                 await asyncio.sleep(0.1)
-        
+
         # Wait for all tasks to complete
         if running_tasks:
             await asyncio.gather(*running_tasks, return_exceptions=True)
-        
+
         # Format results
         return self._format_network_results(agents_config, agent_results, entry_point)
-    
+
     def _normalize_model(self, model: str) -> str:
         """Normalize model names to full format."""
         model_map = {
@@ -406,64 +437,75 @@ Models can be specified as:
             "gemini-1.5-flash": "google/gemini-1.5-flash",
         }
         return model_map.get(model, model)
-    
+
     def _get_cli_tool(self, tool_name: str):
         """Get CLI tool instance."""
         # Import here to avoid circular imports
         if tool_name == "claude_cli":
             from hanzo_mcp.tools.agent.claude_cli_tool import ClaudeCLITool
+
             return ClaudeCLITool(self.permission_manager)
         elif tool_name == "codex_cli":
             from hanzo_mcp.tools.agent.codex_cli_tool import CodexCLITool
+
             return CodexCLITool(self.permission_manager)
         elif tool_name == "gemini_cli":
             from hanzo_mcp.tools.agent.gemini_cli_tool import GeminiCLITool
+
             return GeminiCLITool(self.permission_manager)
         elif tool_name == "grok_cli":
             from hanzo_mcp.tools.agent.grok_cli_tool import GrokCLITool
+
             return GrokCLITool(self.permission_manager)
         return None
-    
+
     def _format_network_results(
         self,
         agents_config: Dict[str, Any],
         results: Dict[str, str],
-        entry_point: Optional[str]
+        entry_point: Optional[str],
     ) -> str:
         """Format results from agent network execution."""
         output = ["Agent Network Execution Results"]
         output.append("=" * 80)
         output.append(f"Total agents: {len(agents_config)}")
         output.append(f"Completed: {len(results)}")
-        output.append(f"Failed: {len([r for r in results.values() if r.startswith('Error:')])}")
-        
+        output.append(
+            f"Failed: {len([r for r in results.values() if r.startswith('Error:')])}"
+        )
+
         if entry_point:
             output.append(f"Entry point: {entry_point}")
-        
+
         output.append("\nExecution Flow:")
         output.append("-" * 40)
-        
+
         # Show results in execution order
         def format_agent_tree(agent_id: str, level: int = 0) -> List[str]:
             lines = []
             indent = "  " * level
-            
+
             if agent_id in agents_config:
                 config = agents_config[agent_id]
                 role = config.get("role", "Agent")
                 model = config.get("model", "default")
-                
-                status = "✅" if agent_id in results and not results[agent_id].startswith("Error:") else "❌"
+
+                status = (
+                    "✅"
+                    if agent_id in results
+                    and not results[agent_id].startswith("Error:")
+                    else "❌"
+                )
                 lines.append(f"{indent}{status} {agent_id} ({role}) [{model}]")
-                
+
                 # Show connections
                 connections = config.get("connections", [])
                 for conn in connections:
                     if conn in agents_config:
                         lines.extend(format_agent_tree(conn, level + 1))
-            
+
             return lines
-        
+
         # Start from entry point or roots
         if entry_point:
             output.extend(format_agent_tree(entry_point))
@@ -478,39 +520,39 @@ Models can be specified as:
                         break
                 if not has_incoming:
                     roots.append(agent_id)
-            
+
             for root in roots:
                 output.extend(format_agent_tree(root))
-        
+
         # Detailed results
         output.append("\n\nDetailed Results:")
         output.append("=" * 80)
-        
+
         for agent_id, result in results.items():
             config = agents_config.get(agent_id, {})
             role = config.get("role", "Agent")
-            
+
             output.append(f"\n### {agent_id} ({role})")
             output.append("-" * 40)
-            
+
             if result.startswith("Error:"):
                 output.append(result)
             else:
                 # Show first part of result
-                lines = result.split('\n')
+                lines = result.split("\n")
                 preview_lines = lines[:10]
                 output.extend(preview_lines)
-                
+
                 if len(lines) > 10:
                     output.append(f"... ({len(lines) - 10} more lines)")
-        
+
         return "\n".join(output)
-    
+
     @override
     def register(self, mcp_server: FastMCP) -> None:
         """Register this swarm tool with the MCP server."""
         tool_self = self
-        
+
         @mcp_server.tool(name=self.name, description=self.description)
         async def swarm(
             ctx: MCPContext,
@@ -523,13 +565,13 @@ Models can be specified as:
             typed_config = SwarmConfig(
                 agents=config.get("agents", {}),
                 entry_point=config.get("entry_point"),
-                topology=config.get("topology")
+                topology=config.get("topology"),
             )
-            
+
             return await tool_self.call(
                 ctx,
                 config=typed_config,
                 query=query,
                 context=context,
-                max_concurrent=max_concurrent
-            )
+                max_concurrent=max_concurrent,
+            )

hanzo_mcp/tools/agent/tool_adapter.py

@@ -5,14 +5,14 @@ formats, making MCP tools available to the OpenAI API, and processing tool input
 and outputs for agent execution.
 """
 
+# Import litellm with warnings suppressed
+import warnings
+
 from openai.types import FunctionParameters
 from openai.types.chat import ChatCompletionToolParam
 
-# Import litellm with warnings suppressed
-import warnings
 with warnings.catch_warnings():
     warnings.simplefilter("ignore", DeprecationWarning)
-    import litellm
 
 from hanzo_mcp.tools.common.base import BaseTool
 
@@ -80,20 +80,30 @@ def supports_parallel_function_calling(model: str) -> bool:
     # based on known models that support parallel function calling
     parallel_capable_models = {
         # OpenAI models that support parallel function calling
-        "gpt-4-turbo", "gpt-4-turbo-preview", "gpt-4-turbo-2024-04-09",
-        "gpt-4o", "gpt-4o-mini", "gpt-4o-2024-05-13", "gpt-4o-2024-08-06",
-        "gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-3.5-turbo-1106",
+        "gpt-4-turbo",
+        "gpt-4-turbo-preview",
+        "gpt-4-turbo-2024-04-09",
+        "gpt-4o",
+        "gpt-4o-mini",
+        "gpt-4o-2024-05-13",
+        "gpt-4o-2024-08-06",
+        "gpt-3.5-turbo",
+        "gpt-3.5-turbo-0125",
+        "gpt-3.5-turbo-1106",
         # Anthropic models with tool support
-        "claude-3-opus", "claude-3-sonnet", "claude-3-haiku",
-        "claude-3-5-sonnet", "claude-3-5-sonnet-20241022",
+        "claude-3-opus",
+        "claude-3-sonnet",
+        "claude-3-haiku",
+        "claude-3-5-sonnet",
+        "claude-3-5-sonnet-20241022",
     }
-    
+
     # Extract model name without provider prefix
     model_name = model.split("/")[-1] if "/" in model else model
-    
+
     # Check if the base model name matches any known parallel-capable models
     for capable_model in parallel_capable_models:
         if model_name.startswith(capable_model):
             return True
-    
+
     return False

hanzo_mcp/tools/common/__init__.py

@@ -4,8 +4,8 @@ from mcp.server import FastMCP
 
 from hanzo_mcp.tools.common.base import BaseTool, ToolRegistry
 from hanzo_mcp.tools.common.batch_tool import BatchTool
-from hanzo_mcp.tools.common.thinking_tool import ThinkingTool
 from hanzo_mcp.tools.common.critic_tool import CriticTool
+from hanzo_mcp.tools.common.thinking_tool import ThinkingTool
 
 
 def register_thinking_tool(

hanzo_mcp/tools/common/base.py

@@ -7,19 +7,17 @@ behavior and provide a foundation for tool registration and management.
 
 import functools
 from abc import ABC, abstractmethod
-from collections.abc import Awaitable
 from typing import Any, Callable, final
+from collections.abc import Awaitable
 
 from mcp.server import FastMCP
 from mcp.server.fastmcp import Context as MCPContext
 
-
-from hanzo_mcp.tools.common.permissions import PermissionManager
 from hanzo_mcp.tools.common.validation import (
     ValidationResult,
     validate_path_parameter,
 )
-from hanzo_mcp.tools.common.truncate import truncate_response
+from hanzo_mcp.tools.common.permissions import PermissionManager
 
 
 def handle_connection_errors(
@@ -174,7 +172,7 @@ class ToolRegistry:
         # Check if tool is enabled before registering
         # Import here to avoid circular imports
         from hanzo_mcp.tools.common.tool_enable import ToolEnableTool
-        
+
         if ToolEnableTool.is_tool_enabled(tool.name):
             # Use the tool's register method which handles all the details
             tool.register(mcp_server)