hanzo-mcp 0.6.10__py3-none-any.whl → 0.6.13__py3-none-any.whl

hanzo_mcp/tools/mcp/__init__.py

@@ -1,6 +1,6 @@
 """MCP management tools."""
 
-from hanzo_mcp.tools.mcp.mcp_unified import UnifiedMCPTool
+from hanzo_mcp.tools.mcp.mcp_tool import MCPTool
 
 # Legacy imports
 from hanzo_mcp.tools.mcp.mcp_add import McpAddTool
@@ -8,7 +8,7 @@ from hanzo_mcp.tools.mcp.mcp_remove import McpRemoveTool
 from hanzo_mcp.tools.mcp.mcp_stats import McpStatsTool
 
 __all__ = [
-    "UnifiedMCPTool",
+    "MCPTool",
     "McpAddTool",
     "McpRemoveTool",
     "McpStatsTool",

hanzo_mcp/tools/mcp/{mcp_unified.py → mcp_tool.py}

@@ -74,7 +74,7 @@ ConfigValue = Annotated[
 AutoStart = Annotated[
     bool,
     Field(
-        description="Auto-start server when Hanzo MCP starts",
+        description="Auto-start server when Hanzo AI starts",
         default=True,
     ),
 ]
@@ -93,8 +93,8 @@ class MCPParams(TypedDict, total=False):
 
 
 @final
-class UnifiedMCPTool(BaseTool):
-    """Unified tool for managing MCP servers."""
+class MCPTool(BaseTool):
+    """Tool for managing MCP servers."""
     
     # Config file
     CONFIG_FILE = Path.home() / ".hanzo" / "mcp" / "servers.json"

hanzo_mcp/tools/shell/__init__.py

@@ -1,4 +1,4 @@
-"""Shell tools package for Hanzo MCP.
+"""Shell tools package for Hanzo AI.
 
 This package provides tools for executing shell commands and scripts.
 """
@@ -8,11 +8,11 @@ from mcp.server import FastMCP
 from hanzo_mcp.tools.common.base import BaseTool, ToolRegistry
 from hanzo_mcp.tools.common.permissions import PermissionManager
 
-# Import unified tools
-from hanzo_mcp.tools.shell.bash_unified import bash_tool
-from hanzo_mcp.tools.shell.npx_unified import npx_tool
-from hanzo_mcp.tools.shell.uvx_unified import uvx_tool
-from hanzo_mcp.tools.shell.process_unified import process_tool
+# Import tools
+from hanzo_mcp.tools.shell.bash_tool import bash_tool
+from hanzo_mcp.tools.shell.npx_tool import npx_tool
+from hanzo_mcp.tools.shell.uvx_tool import uvx_tool
+from hanzo_mcp.tools.shell.process_tool import process_tool
 from hanzo_mcp.tools.shell.open import open_tool
 
 # Export all tool classes

hanzo_mcp/tools/shell/base_process.py

@@ -130,7 +130,8 @@ class BaseProcessTool(BaseTool):
         """
         # Check permissions if manager is available
         if self.permission_manager and cwd:
-            self.permission_manager.check_permission(cwd)
+            if not self.permission_manager.is_path_allowed(str(cwd)):
+                raise PermissionError(f"Access denied to path: {cwd}")
         
         # Get command arguments
         cmd_args = self.get_command_args(command, **kwargs)
@@ -179,7 +180,8 @@ class BaseProcessTool(BaseTool):
         """
         # Check permissions if manager is available
         if self.permission_manager and cwd:
-            self.permission_manager.check_permission(cwd)
+            if not self.permission_manager.is_path_allowed(str(cwd)):
+                raise PermissionError(f"Access denied to path: {cwd}")
         
         # Generate process ID and log file
         process_id = f"{self.get_tool_name()}_{uuid.uuid4().hex[:8]}"

hanzo_mcp/tools/shell/bash_session_executor.py

@@ -1,10 +1,11 @@
-"""Bash session executor for Hanzo MCP.
+"""Bash session executor for Hanzo AI.
 
 This module provides a BashSessionExecutor class that replaces the old CommandExecutor
 implementation with the new BashSession-based approach for better persistent execution.
 """
 
 import asyncio
+import logging
 import os
 import shlex
 import subprocess
@@ -64,16 +65,18 @@ class BashSessionExecutor:
         if data is not None:
             try:
                 import json
-
+                logger = logging.getLogger(__name__)
                 if isinstance(data, (dict, list)):
                     data_str = json.dumps(data)
                 else:
                     data_str = str(data)
-                print(f"DEBUG: {message}: {data_str}")
+                logger.debug(f"{message}: {data_str}")
             except Exception:
-                print(f"DEBUG: {message}: {data}")
+                logger = logging.getLogger(__name__)
+                logger.debug(f"{message}: {data}")
         else:
-            print(f"DEBUG: {message}")
+            logger = logging.getLogger(__name__)
+            logger.debug(f"{message}")
 
     def allow_command(self, command: str) -> None:
         """Allow a specific command that might otherwise be excluded.

hanzo_mcp/tools/shell/{bash_unified.py → bash_tool.py}

@@ -27,7 +27,7 @@ class BashTool(BaseScriptTool):
             action: str = "run",
             cwd: Optional[str] = None,
             env: Optional[dict[str, str]] = None,
-            timeout: Optional[int] = None,
+            timeout: Optional[int] = None
         ) -> str:
             return await tool_self.run(
                 ctx,

hanzo_mcp/tools/shell/command_executor.py

@@ -1,4 +1,4 @@
-"""Command executor tools for Hanzo MCP.
+"""Command executor tools for Hanzo AI.
 
 This module provides tools for executing shell commands and scripts with
 comprehensive error handling, permissions checking, and progress tracking.
@@ -22,7 +22,7 @@ from hanzo_mcp.tools.shell.base import CommandResult
 
 @final
 class CommandExecutor:
-    """Command executor tools for Hanzo MCP.
+    """Command executor tools for Hanzo AI.
 
     This class provides tools for executing shell commands and scripts with
     comprehensive error handling, permissions checking, and progress tracking.
@@ -183,16 +183,18 @@ class CommandExecutor:
         if data is not None:
             try:
                 import json
-
+                logger = logging.getLogger(__name__)
                 if isinstance(data, (dict, list)):
                     data_str = json.dumps(data)
                 else:
                     data_str = str(data)
-                print(f"DEBUG: {message}: {data_str}", file=sys.stderr)
+                logger.debug(f"{message}: {data_str}")
             except Exception:
-                print(f"DEBUG: {message}: {data}", file=sys.stderr)
+                logger = logging.getLogger(__name__)
+                logger.debug(f"{message}: {data}")
         else:
-            print(f"DEBUG: {message}", file=sys.stderr)
+            logger = logging.getLogger(__name__)
+            logger.debug(f"{message}")
 
     def is_command_allowed(self, command: str) -> bool:
         """Check if a command is allowed based on exclusion lists.

hanzo_mcp/tools/shell/{npx_unified.py → npx_tool.py}

@@ -99,7 +99,7 @@ npx --action background json-server db.json"""
             args: str = "",
             action: str = "run",
             cwd: Optional[str] = None,
-            yes: bool = True,
+            yes: bool = True
         ) -> str:
             return await tool_self.run(
                 ctx,

hanzo_mcp/tools/shell/open.py

@@ -24,8 +24,8 @@ class OpenTool(BaseTool):
         
         @server.tool(name=self.name, description=self.description)
         async def open(
-            ctx: MCPContext,
-            path: str
+            path: str,
+            ctx: MCPContext
         ) -> str:
             return await tool_self.run(ctx, path)
     

hanzo_mcp/tools/shell/{process_unified.py → process_tool.py}

@@ -128,7 +128,7 @@ process --action logs --id bash_ghi789 --lines 50"""
             action: str = "list",
             id: Optional[str] = None,
             signal_type: str = "TERM",
-            lines: int = 100,
+            lines: int = 100
         ) -> str:
             return await tool_self.run(
                 ctx,

hanzo_mcp/tools/shell/run_command_windows.py

@@ -123,7 +123,7 @@ Important notes:
 git commit -m \"$(cat <<'EOF'
    Commit message here.
 
-   🤖 Generated with [Hanzo MCP](https://github.com/SDGLBL/hanzo-mcp)
+   🤖 Generated with [Hanzo AI](https://github.com/SDGLBL/hanzo-mcp)
 
    EOF
    )\"

hanzo_mcp/tools/shell/uvx.py

@@ -129,11 +129,56 @@ For long-running servers, use uvx_background instead.
 
         # Check if uvx is available
         if not shutil.which("uvx"):
-            return """Error: uvx is not installed. Install it with:
+            await tool_ctx.info("uvx not found, attempting to install...")
+            
+            # Try to auto-install uvx
+            install_cmd = "curl -LsSf https://astral.sh/uv/install.sh | sh"
+            
+            try:
+                # Run installation
+                install_result = subprocess.run(
+                    install_cmd,
+                    shell=True,
+                    capture_output=True,
+                    text=True,
+                    timeout=60
+                )
+                
+                if install_result.returncode == 0:
+                    await tool_ctx.info("uvx installed successfully!")
+                    
+                    # Add to PATH for current session
+                    import os
+                    home = os.path.expanduser("~")
+                    os.environ["PATH"] = f"{home}/.cargo/bin:{os.environ.get('PATH', '')}"
+                    
+                    # Check again
+                    if not shutil.which("uvx"):
+                        return """Error: uvx installed but not found in PATH. 
+Please add ~/.cargo/bin to your PATH and restart your shell.
+
+Add to ~/.zshrc or ~/.bashrc:
+export PATH="$HOME/.cargo/bin:$PATH"
+"""
+                else:
+                    return f"""Error: Failed to install uvx automatically.
+                    
+Install manually with:
 curl -LsSf https://astral.sh/uv/install.sh | sh
 
 Or on macOS:
-brew install uv"""
+brew install uv
+
+Error details: {install_result.stderr}"""
+                    
+            except subprocess.TimeoutExpired:
+                return """Error: Installation timed out. Install uvx manually with:
+curl -LsSf https://astral.sh/uv/install.sh | sh"""
+            except Exception as e:
+                return f"""Error: Failed to auto-install uvx: {str(e)}
+                
+Install manually with:
+curl -LsSf https://astral.sh/uv/install.sh | sh"""
 
         # Build command
         cmd = ["uvx"]

hanzo_mcp/tools/shell/uvx_background.py

@@ -150,11 +150,56 @@ Use 'processes' to list running processes and 'pkill' to stop them.
 
         # Check if uvx is available
         if not shutil.which("uvx"):
-            return """Error: uvx is not installed. Install it with:
+            await tool_ctx.info("uvx not found, attempting to install...")
+            
+            # Try to auto-install uvx
+            install_cmd = "curl -LsSf https://astral.sh/uv/install.sh | sh"
+            
+            try:
+                # Run installation
+                install_result = subprocess.run(
+                    install_cmd,
+                    shell=True,
+                    capture_output=True,
+                    text=True,
+                    timeout=60
+                )
+                
+                if install_result.returncode == 0:
+                    await tool_ctx.info("uvx installed successfully!")
+                    
+                    # Add to PATH for current session
+                    import os
+                    home = os.path.expanduser("~")
+                    os.environ["PATH"] = f"{home}/.cargo/bin:{os.environ.get('PATH', '')}"
+                    
+                    # Check again
+                    if not shutil.which("uvx"):
+                        return """Error: uvx installed but not found in PATH. 
+Please add ~/.cargo/bin to your PATH and restart your shell.
+
+Add to ~/.zshrc or ~/.bashrc:
+export PATH="$HOME/.cargo/bin:$PATH"
+"""
+                else:
+                    return f"""Error: Failed to install uvx automatically.
+                    
+Install manually with:
 curl -LsSf https://astral.sh/uv/install.sh | sh
 
 Or on macOS:
-brew install uv"""
+brew install uv
+
+Error details: {install_result.stderr}"""
+                    
+            except subprocess.TimeoutExpired:
+                return """Error: Installation timed out. Install uvx manually with:
+curl -LsSf https://astral.sh/uv/install.sh | sh"""
+            except Exception as e:
+                return f"""Error: Failed to auto-install uvx: {str(e)}
+                
+Install manually with:
+curl -LsSf https://astral.sh/uv/install.sh | sh"""
 
         # Build command
         cmd = ["uvx"]

hanzo_mcp/tools/shell/{uvx_unified.py → uvx_tool.py}

@@ -99,7 +99,7 @@ uvx --action background jupyter lab --port 8888"""
             args: str = "",
             action: str = "run",
             cwd: Optional[str] = None,
-            python: Optional[str] = None,
+            python: Optional[str] = None
         ) -> str:
             return await tool_self.run(
                 ctx,

hanzo_mcp/tools/todo/__init__.py

@@ -1,19 +1,17 @@
-"""Todo tools package for Hanzo MCP.
+"""Todo tools package for Hanzo AI.
 
-This package provides tools for managing todo lists across different Claude Desktop sessions,
-using in-memory storage to maintain separate task lists for each conversation.
+This package provides a unified todo management tool for organizing tasks
+within Claude Desktop sessions.
 """
 
 from mcp.server import FastMCP
 
 from hanzo_mcp.tools.common.base import BaseTool, ToolRegistry
-from hanzo_mcp.tools.todo.todo_read import TodoReadTool
-from hanzo_mcp.tools.todo.todo_write import TodoWriteTool
+from hanzo_mcp.tools.todo.todo import TodoTool
 
 # Export all tool classes
 __all__ = [
-    "TodoReadTool",
-    "TodoWriteTool",
+    "TodoTool",
     "get_todo_tools",
     "register_todo_tools",
 ]
@@ -26,8 +24,7 @@ def get_todo_tools() -> list[BaseTool]:
         List of todo tool instances
     """
     return [
-        TodoReadTool(),
-        TodoWriteTool(),
+        TodoTool(),
     ]
 
 
@@ -44,23 +41,21 @@ def register_todo_tools(
     Returns:
         List of registered tools
     """
-    # Define tool mapping
+    # Define tool mapping - single unified todo tool
     tool_classes = {
-        "todo_read": TodoReadTool,
-        "todo_write": TodoWriteTool,
+        "todo": TodoTool,
     }
-    
+
     tools = []
-    
+
     if enabled_tools:
         # Use individual tool configuration
-        for tool_name, enabled in enabled_tools.items():
-            if enabled and tool_name in tool_classes:
-                tool_class = tool_classes[tool_name]
-                tools.append(tool_class())
+        # Support both old names and new name for backward compatibility
+        if enabled_tools.get("todo", True) or enabled_tools.get("todo_read", True) or enabled_tools.get("todo_write", True):
+            tools.append(TodoTool())
     else:
         # Use all tools (backward compatibility)
         tools = get_todo_tools()
-    
+
     ToolRegistry.register_tools(mcp_server, tools)
     return tools

hanzo_mcp/tools/todo/todo.py

@@ -260,6 +260,27 @@ todo --filter in_progress
             self.write_todos([])
             return f"Cleared all {count} todo(s)"
 
+    @override
     def register(self, mcp_server) -> None:
         """Register this tool with the MCP server."""
-        pass
+        tool_self = self  # Create a reference to self for use in the closure
+
+        @mcp_server.tool(name=self.name, description=self.description)
+        async def todo(
+            action: Action = "list",
+            content: Content = None,
+            id: TodoId = None,
+            status: Status = None,
+            priority: Priority = None,
+            filter: Filter = None,
+            ctx: MCPContext = None,
+        ) -> str:
+            return await tool_self.call(
+                ctx,
+                action=action,
+                content=content,
+                id=id,
+                status=status,
+                priority=priority,
+                filter=filter,
+            )

hanzo_mcp/tools/vector/__init__.py

@@ -1,4 +1,4 @@
-"""Vector database tools for Hanzo MCP.
+"""Vector database tools for Hanzo AI.
 
 This package provides tools for working with local vector databases for semantic search,
 document indexing, and retrieval-augmented generation (RAG) workflows.
@@ -62,7 +62,9 @@ try:
             # Auto-detect projects from search paths for new manager
             if search_paths:
                 detected_projects = project_manager.detect_projects(search_paths)
-                print(f"Detected {len(detected_projects)} projects with LLM.md files")
+                import logging
+                logger = logging.getLogger(__name__)
+                logger.info(f"Detected {len(detected_projects)} projects with LLM.md files")
         
         # Register individual tools if enabled
         if tool_enabled.get("index", True):
@@ -85,7 +87,9 @@ except ImportError:
     
     def register_vector_tools(*args, **kwargs) -> list[BaseTool]:
         """Vector tools not available - missing dependencies."""
-        print("Warning: Vector tools not available. Install infinity-embedded: pip install infinity-embedded")
+        import logging
+        logger = logging.getLogger(__name__)
+        logger.warning("Vector tools not available. Install infinity-embedded: pip install infinity-embedded")
         return []
 
 

hanzo_mcp/tools/vector/ast_analyzer.py

@@ -94,7 +94,9 @@ class ASTAnalyzer:
                 # Python parser
                 self.parsers['python'] = tree_sitter.Language(tspython.language())
             except Exception as e:
-                print(f"Warning: Could not initialize Python parser: {e}")
+                import logging
+                logger = logging.getLogger(__name__)
+                logger.warning(f"Could not initialize Python parser: {e}")
     
     def analyze_file(self, file_path: str) -> Optional[FileAST]:
         """Analyze a file and extract AST information and symbols.
@@ -127,7 +129,9 @@ class ASTAnalyzer:
                 return self._analyze_generic_file(file_path, content, file_hash, language)
                 
         except Exception as e:
-            print(f"Error analyzing file {file_path}: {e}")
+            import logging
+            logger = logging.getLogger(__name__)
+            logger.error(f"Error analyzing file {file_path}: {e}")
             return None
     
     def _detect_language(self, path: Path) -> Optional[str]:
@@ -194,9 +198,13 @@ class ASTAnalyzer:
                 ast_nodes = self._extract_tree_sitter_nodes(ts_tree.root_node, content)
             
         except SyntaxError as e:
-            print(f"Syntax error in {file_path}: {e}")
+            import logging
+            logger = logging.getLogger(__name__)
+            logger.error(f"Syntax error in {file_path}: {e}")
         except Exception as e:
-            print(f"Error parsing Python file {file_path}: {e}")
+            import logging
+            logger = logging.getLogger(__name__)
+            logger.error(f"Error parsing Python file {file_path}: {e}")
         
         return FileAST(
             file_path=file_path,

hanzo_mcp/tools/vector/infinity_store.py

@@ -1,4 +1,4 @@
-"""Infinity vector database integration for Hanzo MCP."""
+"""Infinity vector database integration for Hanzo AI."""
 
 import json
 import hashlib
@@ -45,7 +45,7 @@ class SymbolSearchResult:
 
 @dataclass
 class UnifiedSearchResult:
-    """Unified search result combining text, vector, and symbol search."""
+    """Search result combining text, vector, and symbol search."""
     type: str  # 'document', 'symbol', 'reference'
     content: str
     file_path: str
@@ -528,7 +528,9 @@ class InfinityVectorStore:
             return file_ast
             
         except Exception as e:
-            print(f"Error searching AST nodes: {e}")
+            import logging
+            logger = logging.getLogger(__name__)
+            logger.error(f"Error searching AST nodes: {e}")
             return None
     
     def get_file_references(self, file_path: str) -> List[Dict[str, Any]]:
@@ -555,7 +557,9 @@ class InfinityVectorStore:
             return references
             
         except Exception as e:
-            print(f"Error getting file references: {e}")
+            import logging
+            logger = logging.getLogger(__name__)
+            logger.error(f"Error getting file references: {e}")
             return []
     
     def search(
@@ -803,7 +807,9 @@ class InfinityVectorStore:
             
             return True
         except Exception as e:
-            print(f"Error clearing vector store: {e}")
+            import logging
+            logger = logging.getLogger(__name__)
+            logger.error(f"Error clearing vector store: {e}")
             return False
     
     async def index_document(

hanzo_mcp/tools/vector/project_manager.py

@@ -1,4 +1,4 @@
-"""Project-aware vector database management for Hanzo MCP."""
+"""Project-aware vector database management for Hanzo AI."""
 
 import os
 from pathlib import Path
@@ -294,7 +294,9 @@ class ProjectVectorManager:
             project_name = project_names[i]
             if isinstance(result, Exception):
                 # Log error but continue
-                print(f"Error searching project {project_name}: {result}")
+                import logging
+                logger = logging.getLogger(__name__)
+                logger.error(f"Error searching project {project_name}: {result}")
                 combined_results[project_name] = []
             else:
                 combined_results[project_name] = result