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

hanzo_mcp/config/tool_config.py

@@ -1,12 +1,13 @@
 """Tool configuration definitions for Hanzo AI."""
 
 from enum import Enum
-from typing import Dict, List, Optional
+from typing import Dict, List
 from dataclasses import dataclass
 
 
 class ToolCategory(str, Enum):
     """Categories of tools available in Hanzo AI."""
+
     FILESYSTEM = "filesystem"
     SHELL = "shell"
     JUPYTER = "jupyter"
@@ -19,13 +20,14 @@ class ToolCategory(str, Enum):
 @dataclass
 class ToolConfig:
     """Configuration for an individual tool."""
+
     name: str
     category: ToolCategory
     enabled: bool = True
     description: str = ""
     requires_permissions: bool = True
     cli_flag: str = ""
-    
+
     def __post_init__(self):
         """Generate CLI flag if not provided."""
         if not self.cli_flag:
@@ -39,138 +41,132 @@ TOOL_REGISTRY: Dict[str, ToolConfig] = {
         name="read",
         category=ToolCategory.FILESYSTEM,
         description="Read file contents with line numbers and truncation support",
-        cli_flag="--disable-read"
+        cli_flag="--disable-read",
     ),
     "write": ToolConfig(
-        name="write", 
+        name="write",
         category=ToolCategory.FILESYSTEM,
         description="Write content to files, create new files or overwrite existing ones",
-        cli_flag="--disable-write"
+        cli_flag="--disable-write",
     ),
     "edit": ToolConfig(
         name="edit",
-        category=ToolCategory.FILESYSTEM, 
+        category=ToolCategory.FILESYSTEM,
         description="Make precise string replacements in files with validation",
-        cli_flag="--disable-edit"
+        cli_flag="--disable-edit",
     ),
     "multi_edit": ToolConfig(
         name="multi_edit",
         category=ToolCategory.FILESYSTEM,
-        description="Perform multiple edits to a single file in one operation", 
-        cli_flag="--disable-multi-edit"
+        description="Perform multiple edits to a single file in one operation",
+        cli_flag="--disable-multi-edit",
     ),
     "directory_tree": ToolConfig(
         name="directory_tree",
         category=ToolCategory.FILESYSTEM,
         description="Display directory structure as a tree",
-        cli_flag="--disable-directory-tree"
+        cli_flag="--disable-directory-tree",
     ),
     "grep": ToolConfig(
         name="grep",
         category=ToolCategory.FILESYSTEM,
         description="Fast content search using ripgrep or fallback Python implementation",
-        cli_flag="--disable-grep"
+        cli_flag="--disable-grep",
     ),
     "ast": ToolConfig(
-        name="ast", 
+        name="ast",
         category=ToolCategory.FILESYSTEM,
         description="Search source code with AST context using tree-sitter",
-        cli_flag="--disable-ast"
+        cli_flag="--disable-ast",
     ),
     "content_replace": ToolConfig(
         name="content_replace",
         category=ToolCategory.FILESYSTEM,
         description="Bulk text replacement across multiple files",
-        cli_flag="--disable-content-replace"
+        cli_flag="--disable-content-replace",
     ),
-    
     # Shell Tools (1)
     "run_command": ToolConfig(
         name="run_command",
         category=ToolCategory.SHELL,
         description="Execute shell commands with session support",
-        cli_flag="--disable-run-command"
+        cli_flag="--disable-run-command",
     ),
-    
     # Jupyter Tools (2)
     "notebook_read": ToolConfig(
         name="notebook_read",
         category=ToolCategory.JUPYTER,
         description="Read Jupyter notebook files (.ipynb)",
-        cli_flag="--disable-notebook-read"
+        cli_flag="--disable-notebook-read",
     ),
     "notebook_edit": ToolConfig(
         name="notebook_edit",
         category=ToolCategory.JUPYTER,
         description="Edit Jupyter notebook cells (replace, insert, delete)",
-        cli_flag="--disable-notebook-edit"
+        cli_flag="--disable-notebook-edit",
     ),
-    
     # Todo Tools (2)
     "todo_read": ToolConfig(
         name="todo_read",
         category=ToolCategory.TODO,
         description="Read the current todo list for a session",
-        cli_flag="--disable-todo-read"
+        cli_flag="--disable-todo-read",
     ),
     "todo_write": ToolConfig(
-        name="todo_write", 
+        name="todo_write",
         category=ToolCategory.TODO,
         description="Create and manage structured task lists",
-        cli_flag="--disable-todo-write"
+        cli_flag="--disable-todo-write",
     ),
-    
     # Agent Tools (3)
     "dispatch_agent": ToolConfig(
         name="dispatch_agent",
         category=ToolCategory.AGENT,
         enabled=False,  # Disabled by default
         description="Delegate tasks to sub-agents for concurrent/specialized processing",
-        cli_flag="--enable-dispatch-agent"
+        cli_flag="--enable-dispatch-agent",
     ),
     "swarm": ToolConfig(
         name="swarm",
         category=ToolCategory.AGENT,
         enabled=False,  # Disabled by default
         description="Execute multiple agent tasks in parallel across different files",
-        cli_flag="--enable-swarm"
+        cli_flag="--enable-swarm",
     ),
     "hierarchical_swarm": ToolConfig(
         name="hierarchical_swarm",
         category=ToolCategory.AGENT,
         enabled=False,  # Disabled by default
         description="Execute hierarchical agent swarms with Claude Code integration",
-        cli_flag="--enable-hierarchical-swarm"
+        cli_flag="--enable-hierarchical-swarm",
     ),
-    
     # Common Tools (3)
     "think": ToolConfig(
         name="think",
         category=ToolCategory.COMMON,
         description="Provide structured thinking space for complex reasoning",
-        cli_flag="--disable-think"
+        cli_flag="--disable-think",
     ),
     "batch": ToolConfig(
         name="batch",
         category=ToolCategory.COMMON,
         description="Execute multiple tools in parallel or serial",
-        cli_flag="--disable-batch"
+        cli_flag="--disable-batch",
     ),
-    
     # Vector Tools (2)
     "vector_index": ToolConfig(
         name="vector_index",
         category=ToolCategory.VECTOR,
         enabled=False,  # Disabled by default
         description="Index documents in local vector database for semantic search",
-        cli_flag="--enable-vector-index"
+        cli_flag="--enable-vector-index",
     ),
     "vector_search": ToolConfig(
         name="vector_search",
         category=ToolCategory.VECTOR,
         enabled=False,  # Disabled by default
         description="Search documents using semantic similarity in vector database",
-        cli_flag="--enable-vector-search"
+        cli_flag="--enable-vector-search",
     ),
 }
 
@@ -208,4 +204,6 @@ def disable_tool(tool_name: str) -> bool:
 
 def is_tool_enabled(tool_name: str) -> bool:
     """Check if a tool is enabled."""
-    return TOOL_REGISTRY.get(tool_name, ToolConfig("", ToolCategory.COMMON, False)).enabled
+    return TOOL_REGISTRY.get(
+        tool_name, ToolConfig("", ToolCategory.COMMON, False)
+    ).enabled

hanzo_mcp/dev_server.py

@@ -1,12 +1,10 @@
 """Development server with hot reload for Hanzo AI."""
 
+import time
 import asyncio
 import logging
-import os
-import sys
-import time
+from typing import Set, Optional
 from pathlib import Path
-from typing import Optional, Set
 
 import watchdog.events
 import watchdog.observers
@@ -17,77 +15,87 @@ from hanzo_mcp.server import HanzoMCPServer
 
 class MCPReloadHandler(FileSystemEventHandler):
     """Handler for file system events that triggers MCP server reload."""
-    
+
     def __init__(self, restart_callback, ignore_patterns: Optional[Set[str]] = None):
         """Initialize the reload handler.
-        
+
         Args:
             restart_callback: Function to call when files change
             ignore_patterns: Set of patterns to ignore
         """
         self.restart_callback = restart_callback
         self.ignore_patterns = ignore_patterns or {
-            "__pycache__", ".pyc", ".pyo", ".git", ".pytest_cache",
-            ".mypy_cache", ".ruff_cache", ".coverage", "*.log",
-            ".env", ".venv", "venv", "node_modules"
+            "__pycache__",
+            ".pyc",
+            ".pyo",
+            ".git",
+            ".pytest_cache",
+            ".mypy_cache",
+            ".ruff_cache",
+            ".coverage",
+            "*.log",
+            ".env",
+            ".venv",
+            "venv",
+            "node_modules",
         }
         self.last_reload = 0
         self.reload_delay = 0.5  # Debounce delay in seconds
-    
+
     def should_ignore(self, path: str) -> bool:
         """Check if a path should be ignored."""
         path_obj = Path(path)
-        
+
         # Check against ignore patterns
         for pattern in self.ignore_patterns:
             if pattern in str(path_obj):
                 return True
             if path_obj.name.endswith(pattern):
                 return True
-        
+
         # Only watch Python files and config files
         if path_obj.is_file():
             allowed_extensions = {".py", ".json", ".yaml", ".yml", ".toml"}
             if path_obj.suffix not in allowed_extensions:
                 return True
-        
+
         return False
-    
+
     def on_any_event(self, event):
         """Handle any file system event."""
         if event.is_directory:
             return
-        
+
         if self.should_ignore(event.src_path):
             return
-        
+
         # Debounce rapid changes
         current_time = time.time()
         if current_time - self.last_reload < self.reload_delay:
             return
-        
+
         self.last_reload = current_time
-        
+
         logger = logging.getLogger(__name__)
         logger.info(f"\n🔄 File changed: {event.src_path}")
         logger.info("🔄 Reloading MCP server...")
-        
+
         self.restart_callback()
 
 
 class DevServer:
     """Development server with hot reload capability."""
-    
+
     def __init__(
         self,
         name: str = "hanzo-dev",
         allowed_paths: Optional[list[str]] = None,
         project_paths: Optional[list[str]] = None,
         project_dir: Optional[str] = None,
-        **kwargs
+        **kwargs,
     ):
         """Initialize the development server.
-        
+
         Args:
             name: Server name
             allowed_paths: Allowed paths for the server
@@ -103,7 +111,7 @@ class DevServer:
         self.server_process = None
         self.observer = None
         self.running = False
-    
+
     def create_server(self) -> HanzoMCPServer:
         """Create a new MCP server instance."""
         return HanzoMCPServer(
@@ -111,76 +119,80 @@ class DevServer:
             allowed_paths=self.allowed_paths,
             project_paths=self.project_paths,
             project_dir=self.project_dir,
-            **self.server_kwargs
+            **self.server_kwargs,
         )
-    
+
     def start_file_watcher(self):
         """Start watching for file changes."""
         # Watch the hanzo_mcp package directory
         package_dir = Path(__file__).parent
-        
+
         # Create observer and handler
         self.observer = watchdog.observers.Observer()
         handler = MCPReloadHandler(self.restart_server)
-        
+
         # Watch the package directory
         self.observer.schedule(handler, str(package_dir), recursive=True)
-        
+
         # Also watch any project directories
         if self.project_dir:
             self.observer.schedule(handler, self.project_dir, recursive=True)
-        
+
         for path in self.allowed_paths:
             if Path(path).is_dir() and path not in [str(package_dir), self.project_dir]:
                 self.observer.schedule(handler, path, recursive=True)
-        
+
         self.observer.start()
         logger = logging.getLogger(__name__)
         logger.info(f"👀 Watching for changes in: {package_dir}")
         if self.project_dir:
             logger.info(f"👀 Also watching: {self.project_dir}")
-    
+
     def stop_file_watcher(self):
         """Stop the file watcher."""
         if self.observer and self.observer.is_alive():
             self.observer.stop()
             self.observer.join(timeout=2)
-    
+
     def restart_server(self):
         """Restart the MCP server."""
         # Since MCP servers run in the same process, we need to handle this differently
         # For now, we'll log a message indicating a restart is needed
         logger = logging.getLogger(__name__)
-        logger.warning("\n⚠️  Server restart required. Please restart the MCP client to reload changes.")
-        logger.info("💡 Tip: In development, consider using the MCP test client for easier reloading.")
-    
+        logger.warning(
+            "\n⚠️  Server restart required. Please restart the MCP client to reload changes."
+        )
+        logger.info(
+            "💡 Tip: In development, consider using the MCP test client for easier reloading."
+        )
+
     async def run_async(self, transport: str = "stdio"):
         """Run the development server asynchronously."""
         self.running = True
-        
+
         logger = logging.getLogger(__name__)
         logger.info(f"\n🚀 Starting Hanzo AI in development mode...")
         logger.info(f"🔧 Hot reload enabled - watching for file changes")
         logger.info(f"📁 Project: {self.project_dir or 'current directory'}")
         logger.info(f"🌐 Transport: {transport}\n")
-        
+
         # Start file watcher
         self.start_file_watcher()
-        
+
         try:
             # Create and run server
             server = self.create_server()
-            
+
             # Run the server (this will block)
             server.run(transport=transport)
-            
+
         except KeyboardInterrupt:
             logger.info("\n\n🛑 Shutting down development server...")
         finally:
             self.running = False
             self.stop_file_watcher()
             logger.info("👋 Development server stopped")
-    
+
     def run(self, transport: str = "stdio"):
         """Run the development server."""
         try:
@@ -193,48 +205,35 @@ class DevServer:
 def run_dev_server():
     """Entry point for development server."""
     import argparse
-    
-    parser = argparse.ArgumentParser(description="Run Hanzo AI in development mode with hot reload")
-    parser.add_argument(
-        "--name",
-        type=str,
-        default="hanzo-dev",
-        help="Name of the MCP server"
+
+    parser = argparse.ArgumentParser(
+        description="Run Hanzo AI in development mode with hot reload"
     )
     parser.add_argument(
-        "--project-dir",
-        type=str,
-        help="Project directory to serve"
+        "--name", type=str, default="hanzo-dev", help="Name of the MCP server"
     )
+    parser.add_argument("--project-dir", type=str, help="Project directory to serve")
     parser.add_argument(
         "--allowed-path",
         type=str,
         action="append",
         dest="allowed_paths",
-        help="Additional allowed paths (can be specified multiple times)"
+        help="Additional allowed paths (can be specified multiple times)",
     )
     parser.add_argument(
         "--transport",
         type=str,
         default="stdio",
         choices=["stdio", "sse"],
-        help="Transport type (default: stdio)"
+        help="Transport type (default: stdio)",
     )
     parser.add_argument(
-        "--host",
-        type=str,
-        default="127.0.0.1",
-        help="Host for SSE transport"
+        "--host", type=str, default="127.0.0.1", help="Host for SSE transport"
     )
-    parser.add_argument(
-        "--port",
-        type=int,
-        default=3000,
-        help="Port for SSE transport"
-    )
-    
+    parser.add_argument("--port", type=int, default=3000, help="Port for SSE transport")
+
     args = parser.parse_args()
-    
+
     # Create and run dev server
     dev_server = DevServer(
         name=args.name,
@@ -243,9 +242,9 @@ def run_dev_server():
         host=args.host,
         port=args.port,
     )
-    
+
     dev_server.run(transport=args.transport)
 
 
 if __name__ == "__main__":
-    run_dev_server()
+    run_dev_server()

hanzo_mcp/prompts/__init__.py

@@ -2,25 +2,36 @@ import os
 
 from mcp.server import FastMCP
 
-from hanzo_mcp.prompts.compact_conversation import COMPACT_CONVERSATION_PROMPT
-from hanzo_mcp.prompts.create_release import CREATE_RELEASE_PROMPT
-from hanzo_mcp.prompts.project_system import PROJECT_SYSTEM_PROMPT
-from hanzo_mcp.prompts.project_todo_reminder import (
-    PROJECT_TODO_EMPTY_REMINDER,
-    get_project_todo_reminder,
-)
 from hanzo_mcp.prompts.utils import (
-    get_directory_structure,
-    get_git_info,
     get_os_info,
+    get_git_info,
+    get_directory_structure,
 )
 from hanzo_mcp.prompts.tool_explorer import (
-    TOOL_EXPLORER_PROMPT,
-    FILESYSTEM_TOOLS_HELP,
     AGENT_TOOLS_HELP,
     SHELL_TOOLS_HELP,
     BATCH_TOOL_EXAMPLES,
-    create_tool_category_prompt,
+    TOOL_EXPLORER_PROMPT,
+    FILESYSTEM_TOOLS_HELP,
+)
+from hanzo_mcp.prompts.create_release import CREATE_RELEASE_PROMPT
+from hanzo_mcp.prompts.project_system import PROJECT_SYSTEM_PROMPT
+from hanzo_mcp.prompts.enhanced_prompts import (
+    QUICK_START_PROMPT,
+    LSP_TOOLS_HELP_PROMPT,
+    PAGINATION_GUIDE_PROMPT,
+    PERFORMANCE_TIPS_PROMPT,
+    MEMORY_VECTOR_HELP_PROMPT,
+    CONFIGURATION_GUIDE_PROMPT,
+    DATABASE_TOOLS_HELP_PROMPT,
+    NETWORK_AGENT_GUIDE_PROMPT,
+    TROUBLESHOOTING_GUIDE_PROMPT,
+    SECURITY_BEST_PRACTICES_PROMPT,
+)
+from hanzo_mcp.prompts.compact_conversation import COMPACT_CONVERSATION_PROMPT
+from hanzo_mcp.prompts.project_todo_reminder import (
+    PROJECT_TODO_EMPTY_REMINDER,
+    get_project_todo_reminder,
 )
 
 CONTINUE_FROM_LAST_SESSION_PROMPT = """<system-reminder>
@@ -140,6 +151,77 @@ def register_all_prompts(
         """
         return BATCH_TOOL_EXAMPLES
 
+    # Enhanced prompts for better discoverability
+    @mcp_server.prompt(name="Quick start guide")
+    def quick_start() -> str:
+        """
+        Common workflows and recipes for getting started quickly.
+        """
+        return QUICK_START_PROMPT
+
+    @mcp_server.prompt(name="Pagination guide")
+    def pagination_guide() -> str:
+        """
+        How to use pagination for large result sets.
+        """
+        return PAGINATION_GUIDE_PROMPT
+
+    @mcp_server.prompt(name="Memory and vector tools help")
+    def memory_vector_help() -> str:
+        """
+        Guide for semantic search and memory tools.
+        """
+        return MEMORY_VECTOR_HELP_PROMPT
+
+    @mcp_server.prompt(name="Database tools help")
+    def database_help() -> str:
+        """
+        SQL and graph database operations guide.
+        """
+        return DATABASE_TOOLS_HELP_PROMPT
+
+    @mcp_server.prompt(name="LSP tools help")
+    def lsp_help() -> str:
+        """
+        Language Server Protocol features and code intelligence.
+        """
+        return LSP_TOOLS_HELP_PROMPT
+
+    @mcp_server.prompt(name="Configuration guide")
+    def config_guide() -> str:
+        """
+        How to configure tools, presets, and settings.
+        """
+        return CONFIGURATION_GUIDE_PROMPT
+
+    @mcp_server.prompt(name="Network agent guide")
+    def network_guide() -> str:
+        """
+        Distributed AI orchestration with network/swarm tools.
+        """
+        return NETWORK_AGENT_GUIDE_PROMPT
+
+    @mcp_server.prompt(name="Performance tips")
+    def performance_tips() -> str:
+        """
+        Optimization strategies for better performance.
+        """
+        return PERFORMANCE_TIPS_PROMPT
+
+    @mcp_server.prompt(name="Security best practices")
+    def security_practices() -> str:
+        """
+        Safe usage patterns and security guidelines.
+        """
+        return SECURITY_BEST_PRACTICES_PROMPT
+
+    @mcp_server.prompt(name="Troubleshooting guide")
+    def troubleshooting() -> str:
+        """
+        Common issues and their solutions.
+        """
+        return TROUBLESHOOTING_GUIDE_PROMPT
+
     if projects is None:
         return