hanzo-mcp 0.8.11__py3-none-any.whl → 0.8.14__py3-none-any.whl

hanzo_mcp/tools/agent/cli_agent_base.py

@@ -97,9 +97,7 @@ class CLIAgentBase(BaseTool):
 
         # Check if installed
         if not self.is_installed():
-            error_msg = (
-                f"{self.provider_name} CLI ({self.command_name}) is not installed. "
-            )
+            error_msg = f"{self.provider_name} CLI ({self.command_name}) is not installed. "
             error_msg += f"Please install it first: https://github.com/anthropics/{self.command_name}"
             await tool_ctx.error(error_msg)
             return f"Error: {error_msg}"
@@ -115,15 +113,11 @@ class CLIAgentBase(BaseTool):
         cli_args = self.get_cli_args(prompt, **kwargs)
 
         # Log command
-        await tool_ctx.info(
-            f"Executing {self.provider_name}: {self.command_name} {' '.join(cli_args[:3])}..."
-        )
+        await tool_ctx.info(f"Executing {self.provider_name}: {self.command_name} {' '.join(cli_args[:3])}...")
 
         try:
             # Create temp file for prompt if needed
-            with tempfile.NamedTemporaryFile(
-                mode="w", suffix=".txt", delete=False
-            ) as f:
+            with tempfile.NamedTemporaryFile(mode="w", suffix=".txt", delete=False) as f:
                 f.write(prompt)
                 prompt_file = f.name
 
@@ -131,12 +125,7 @@ class CLIAgentBase(BaseTool):
             if "--prompt-file" in cli_args:
                 # Replace placeholder with actual file
                 cli_args = [
-                    (
-                        arg.replace("--prompt-file", prompt_file)
-                        if arg == "--prompt-file"
-                        else arg
-                    )
-                    for arg in cli_args
+                    (arg.replace("--prompt-file", prompt_file) if arg == "--prompt-file" else arg) for arg in cli_args
                 ]
 
             # Execute command
@@ -151,13 +140,9 @@ class CLIAgentBase(BaseTool):
 
             # Send prompt via stdin if not using file
             if "--prompt-file" not in cli_args:
-                stdout, stderr = await asyncio.wait_for(
-                    process.communicate(input=prompt.encode()), timeout=timeout
-                )
+                stdout, stderr = await asyncio.wait_for(process.communicate(input=prompt.encode()), timeout=timeout)
             else:
-                stdout, stderr = await asyncio.wait_for(
-                    process.communicate(), timeout=timeout
-                )
+                stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=timeout)
 
             # Clean up temp file
             try:
@@ -175,9 +160,7 @@ class CLIAgentBase(BaseTool):
             return result
 
         except asyncio.TimeoutError:
-            await tool_ctx.error(
-                f"{self.provider_name} timed out after {timeout} seconds"
-            )
+            await tool_ctx.error(f"{self.provider_name} timed out after {timeout} seconds")
             return f"Error: Command timed out after {timeout} seconds"
         except Exception as e:
             await tool_ctx.error(f"{self.provider_name} error: {str(e)}")

hanzo_mcp/tools/agent/cli_tools.py

@@ -55,6 +55,7 @@ Timeout = Annotated[
 
 class CLIToolParams(TypedDict, total=False):
     """Common parameters for CLI tools."""
+
     prompt: str
     model: Optional[str]
     working_dir: Optional[str]
@@ -63,7 +64,7 @@ class CLIToolParams(TypedDict, total=False):
 
 class BaseCLITool(BaseTool):
     """Base class for CLI tool implementations."""
-    
+
     def __init__(
         self,
         permission_manager: Optional[PermissionManager] = None,
@@ -71,7 +72,7 @@ class BaseCLITool(BaseTool):
         api_key_env: Optional[str] = None,
     ):
         """Initialize CLI tool.
-        
+
         Args:
             permission_manager: Permission manager for access control
             default_model: Default model to use
@@ -80,21 +81,21 @@ class BaseCLITool(BaseTool):
         self.permission_manager = permission_manager
         self.default_model = default_model
         self.api_key_env = api_key_env
-        
+
     def get_auth_env(self) -> dict[str, str]:
         """Get authentication environment variables."""
         env = os.environ.copy()
-        
+
         # Add API key if configured
         if self.api_key_env and self.api_key_env in os.environ:
             env[self.api_key_env] = os.environ[self.api_key_env]
-        
+
         # Add Hanzo API key for unified auth
         if "HANZO_API_KEY" in os.environ:
             env["HANZO_API_KEY"] = os.environ["HANZO_API_KEY"]
-            
+
         return env
-    
+
     async def execute_cli(
         self,
         command: list[str],
@@ -103,20 +104,20 @@ class BaseCLITool(BaseTool):
         timeout: int = 300,
     ) -> str:
         """Execute CLI command with proper error handling.
-        
+
         Args:
             command: Command and arguments
             input_text: Optional stdin input
             working_dir: Working directory
             timeout: Timeout in seconds
-            
+
         Returns:
             Command output
         """
         try:
             # Set up environment with auth
             env = self.get_auth_env()
-            
+
             # Execute command
             process = await asyncio.create_subprocess_exec(
                 *command,
@@ -126,33 +127,33 @@ class BaseCLITool(BaseTool):
                 cwd=working_dir,
                 env=env,
             )
-            
+
             # Send input and get output
             stdout, stderr = await asyncio.wait_for(
                 process.communicate(input_text.encode() if input_text else None),
                 timeout=timeout,
             )
-            
+
             # Check for errors
             if process.returncode != 0:
                 error_msg = stderr.decode() if stderr else "Unknown error"
                 return f"Error: {error_msg}"
-            
+
             return stdout.decode()
-            
+
         except asyncio.TimeoutError:
             return f"Error: Command timed out after {timeout} seconds"
         except Exception as e:
             return f"Error executing command: {str(e)}"
-    
+
     def register(self, mcp_server: FastMCP) -> None:
         """Register this tool with the MCP server.
-        
+
         Args:
             mcp_server: The FastMCP server instance
         """
         tool_self = self  # Create a reference to self for use in the closure
-        
+
         @mcp_server.tool(name=self.name, description=self.description)
         async def tool_wrapper(
             prompt: str,
@@ -169,33 +170,33 @@ class BaseCLITool(BaseTool):
 
 class ClaudeCLITool(BaseCLITool):
     """Claude CLI tool (also available as 'cc' alias)."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="claude-3-5-sonnet-20241022",
             api_key_env="ANTHROPIC_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "claude"
-    
+
     @property
     def description(self) -> str:
         return "Execute Claude CLI for AI assistance using Anthropic's models"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt: str = params.get("prompt", "")
         model: Optional[str] = params.get("model") or self.default_model
         working_dir: Optional[str] = params.get("working_dir")
         timeout: int = params.get("timeout", 300)
-        
+
         # Build command
         command: list[str] = ["claude"]
         if model:
             command.extend(["--model", model])
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -207,11 +208,11 @@ class ClaudeCLITool(BaseCLITool):
 
 class ClaudeCodeCLITool(ClaudeCLITool):
     """Claude Code CLI tool (cc alias)."""
-    
+
     @property
     def name(self) -> str:
         return "cc"
-    
+
     @property
     def description(self) -> str:
         return "Claude Code CLI (alias for claude)"
@@ -219,34 +220,34 @@ class ClaudeCodeCLITool(ClaudeCLITool):
 
 class CodexCLITool(BaseCLITool):
     """OpenAI Codex/GPT-4 CLI tool."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="gpt-4-turbo",
             api_key_env="OPENAI_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "codex"
-    
+
     @property
     def description(self) -> str:
         return "Execute OpenAI Codex/GPT-4 CLI for code generation and AI assistance"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt: str = params.get("prompt", "")
         model: Optional[str] = params.get("model") or self.default_model
         working_dir: Optional[str] = params.get("working_dir")
         timeout: int = params.get("timeout", 300)
-        
+
         # Build command (using openai CLI or custom wrapper)
         command: list[str] = ["openai", "api", "chat.completions.create"]
         if model:
             command.extend(["-m", model])
         command.extend(["-g", "user", prompt])
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -257,34 +258,34 @@ class CodexCLITool(BaseCLITool):
 
 class GeminiCLITool(BaseCLITool):
     """Google Gemini CLI tool."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="gemini-1.5-pro",
             api_key_env="GEMINI_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "gemini"
-    
+
     @property
     def description(self) -> str:
         return "Execute Google Gemini CLI for multimodal AI assistance"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt: str = params.get("prompt", "")
         model: Optional[str] = params.get("model") or self.default_model
         working_dir: Optional[str] = params.get("working_dir")
         timeout: int = params.get("timeout", 300)
-        
+
         # Build command
         command: list[str] = ["gemini"]
         if model:
             command.extend(["--model", model])
         command.append(prompt)
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -295,34 +296,34 @@ class GeminiCLITool(BaseCLITool):
 
 class GrokCLITool(BaseCLITool):
     """xAI Grok CLI tool."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="grok-4",
             api_key_env="XAI_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "grok"
-    
+
     @property
     def description(self) -> str:
         return "Execute xAI Grok CLI for real-time AI assistance"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt: str = params.get("prompt", "")
         model: Optional[str] = params.get("model") or self.default_model
         working_dir: Optional[str] = params.get("working_dir")
         timeout: int = params.get("timeout", 300)
-        
+
         # Build command
         command: list[str] = ["grok"]
         if model:
             command.extend(["--model", model])
         command.append(prompt)
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -333,34 +334,34 @@ class GrokCLITool(BaseCLITool):
 
 class OpenHandsCLITool(BaseCLITool):
     """OpenHands (OpenDevin) CLI tool."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="claude-3-5-sonnet-20241022",
             api_key_env="OPENAI_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "openhands"
-    
+
     @property
     def description(self) -> str:
         return "Execute OpenHands (OpenDevin) for autonomous coding assistance"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt = params.get("prompt", "")
         model = params.get("model") or self.default_model
         working_dir: str = params.get("working_dir") or os.getcwd()
         timeout: int = params.get("timeout", 600)  # 10 minutes for OpenHands
-        
+
         # Build command
         command: list[str] = ["openhands", "run", prompt]
         if model:
             command.extend(["--model", model])
         command.extend(["--workspace", working_dir])
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -371,11 +372,11 @@ class OpenHandsCLITool(BaseCLITool):
 
 class OpenHandsShortCLITool(OpenHandsCLITool):
     """OpenHands CLI tool (oh alias)."""
-    
+
     @property
     def name(self) -> str:
         return "oh"
-    
+
     @property
     def description(self) -> str:
         return "OpenHands CLI (alias for openhands)"
@@ -383,34 +384,34 @@ class OpenHandsShortCLITool(OpenHandsCLITool):
 
 class HanzoDevCLITool(BaseCLITool):
     """Hanzo Dev AI coding assistant."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="claude-3-5-sonnet-20241022",
             api_key_env="HANZO_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "hanzo_dev"
-    
+
     @property
     def description(self) -> str:
         return "Execute Hanzo Dev for AI-powered code editing and development"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt = params.get("prompt", "")
         model = params.get("model") or self.default_model
         working_dir: str = params.get("working_dir") or os.getcwd()
         timeout: int = params.get("timeout", 600)
-        
+
         # Build command
         command: list[str] = ["dev"]
         if model:
             command.extend(["--model", model])
         command.extend(["--prompt", prompt])
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -421,31 +422,31 @@ class HanzoDevCLITool(BaseCLITool):
 
 class ClineCLITool(BaseCLITool):
     """Cline (formerly Claude Engineer) CLI tool."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="claude-3-5-sonnet-20241022",
             api_key_env="ANTHROPIC_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "cline"
-    
+
     @property
     def description(self) -> str:
         return "Execute Cline for autonomous coding with Claude"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt = params.get("prompt", "")
         working_dir: str = params.get("working_dir") or os.getcwd()
         timeout: int = params.get("timeout", 600)
-        
+
         # Build command
         command: list[str] = ["cline", prompt]
         command.extend(["--no-interactive"])  # Non-interactive mode for batch
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -456,28 +457,28 @@ class ClineCLITool(BaseCLITool):
 
 class AiderCLITool(BaseCLITool):
     """Aider AI pair programming tool."""
-    
+
     def __init__(self, permission_manager: Optional[PermissionManager] = None):
         super().__init__(
             permission_manager=permission_manager,
             default_model="gpt-4-turbo",
             api_key_env="OPENAI_API_KEY",
         )
-    
+
     @property
     def name(self) -> str:
         return "aider"
-    
+
     @property
     def description(self) -> str:
         return "Execute Aider for AI pair programming"
-    
+
     async def call(self, ctx: Context[Any, Any, Any], **params: Any) -> str:
         prompt = params.get("prompt", "")
         model = params.get("model") or self.default_model
         working_dir: str = params.get("working_dir") or os.getcwd()
         timeout: int = params.get("timeout", 600)
-        
+
         # Build command
         command: list[str] = ["aider"]
         if model:
@@ -485,7 +486,7 @@ class AiderCLITool(BaseCLITool):
         command.extend(["--message", prompt])
         command.extend(["--yes"])  # Auto-approve changes
         command.extend(["--no-stream"])  # No streaming for batch
-        
+
         # Execute
         return await self.execute_cli(
             command,
@@ -499,11 +500,11 @@ def register_cli_tools(
     permission_manager: Optional[PermissionManager] = None,
 ) -> list[BaseTool]:
     """Register all CLI tools with the MCP server.
-    
+
     Args:
         mcp_server: The FastMCP server instance
         permission_manager: Permission manager for access control
-        
+
     Returns:
         List of registered CLI tools
     """
@@ -519,11 +520,11 @@ def register_cli_tools(
         ClineCLITool(permission_manager),
         AiderCLITool(permission_manager),
     ]
-    
+
     # Register each tool
     for tool in tools:
         tool.register(mcp_server)
-    
+
     return tools
 
 
@@ -540,4 +541,4 @@ __all__ = [
     "ClineCLITool",
     "AiderCLITool",
     "register_cli_tools",
-]
+]

hanzo_mcp/tools/agent/code_auth.py

@@ -322,9 +322,7 @@ class CodeAuthManager:
             api_key = None
             if parent_info.get("has_keyring"):
                 try:
-                    api_key = keyring.get_password(
-                        f"hanzo-{parent_info['provider']}", parent_account
-                    )
+                    api_key = keyring.get_password(f"hanzo-{parent_info['provider']}", parent_account)
                 except Exception:
                     pass
 

hanzo_mcp/tools/agent/code_auth_tool.py

@@ -117,9 +117,7 @@ Providers: claude, openai, azure, deepseek, google, groq"""
             model = params.get("model")
             description = params.get("description")
 
-            success, msg = self.auth_manager.create_account(
-                account, provider, api_key, model, description
-            )
+            success, msg = self.auth_manager.create_account(account, provider, api_key, model, description)
             return msg
 
         elif action == "login":
@@ -148,9 +146,7 @@ Providers: claude, openai, azure, deepseek, google, groq"""
             parent_account = params.get("parent_account")
 
             # Try to create agent account
-            success, result = self.auth_manager.create_agent_account(
-                agent_id, provider, parent_account
-            )
+            success, result = self.auth_manager.create_agent_account(agent_id, provider, parent_account)
 
             if success:
                 # Get credentials

hanzo_mcp/tools/agent/critic_tool.py

@@ -117,9 +117,7 @@ class AutoCritic:
     ) -> str:
         """Perform automated critical review."""
         review_func = self.review_patterns.get(review_type, self._review_general)
-        return review_func(
-            work_description, code_snippets, file_paths, specific_concerns
-        )
+        return review_func(work_description, code_snippets, file_paths, specific_concerns)
 
     def _review_code_quality(
         self,
@@ -137,15 +135,11 @@ class AutoCritic:
             for snippet in code_snippets:
                 # Check for proper error handling
                 if "error" in snippet.lower() and "if err" not in snippet:
-                    issues.append(
-                        "❌ Missing error handling - always check errors in Go"
-                    )
+                    issues.append("❌ Missing error handling - always check errors in Go")
 
                 # Check for magic numbers
                 if any(char.isdigit() for char in snippet) and "const" not in snippet:
-                    suggestions.append(
-                        "💡 Consider extracting magic numbers to named constants"
-                    )
+                    suggestions.append("💡 Consider extracting magic numbers to named constants")
 
                 # Check for proper imports
                 if "import" in snippet:
@@ -160,9 +154,7 @@ class AutoCritic:
             suggestions.append("💡 Consider edge cases and error scenarios")
 
         if file_paths and len(file_paths) > 5:
-            suggestions.append(
-                "💡 Large number of files modified - consider breaking into smaller PRs"
-            )
+            suggestions.append("💡 Large number of files modified - consider breaking into smaller PRs")
 
         # Build response
         response = "🔍 CODE QUALITY REVIEW:\n\n"
@@ -173,9 +165,7 @@ class AutoCritic:
             response += "✅ No major code quality issues detected.\n\n"
 
         if suggestions:
-            response += (
-                "Suggestions for Improvement:\n" + "\n".join(suggestions) + "\n\n"
-            )
+            response += "Suggestions for Improvement:\n" + "\n".join(suggestions) + "\n\n"
 
         if specific_concerns:
             response += f"Regarding your concern: '{specific_concerns}'\n"
@@ -183,9 +173,7 @@ class AutoCritic:
                 response += "→ Imports look properly formatted. Ensure they're in the standard order: stdlib, external, internal.\n"
 
         response += "\nOverall: " + (
-            "⚠️ Address the issues before proceeding."
-            if issues
-            else "✅ Good work, but always room for improvement!"
+            "⚠️ Address the issues before proceeding." if issues else "✅ Good work, but always room for improvement!"
         )
 
         return response
@@ -269,9 +257,7 @@ Security Checklist:
         response = "🔍 COMPLETENESS REVIEW:\n\n"
 
         if "fix" in tasks_mentioned and "test" not in tasks_mentioned:
-            response += (
-                "❌ No mention of tests - have you verified the fix with tests?\n"
-            )
+            response += "❌ No mention of tests - have you verified the fix with tests?\n"
 
         if "import" in tasks_mentioned:
             response += "✓ Import fixes mentioned\n"
@@ -382,8 +368,6 @@ class CriticProtocol:
         except KeyError:
             review_enum = ReviewType.GENERAL
 
-        review = self.auto_critic.review(
-            review_enum, work_description, code_snippets, file_paths, specific_concerns
-        )
+        review = self.auto_critic.review(review_enum, work_description, code_snippets, file_paths, specific_concerns)
 
         return f"Review {self.review_count}/{self.max_reviews}:\n\n{review}"