hanzo-mcp 0.8.8__py3-none-any.whl → 0.8.13__py3-none-any.whl

hanzo_mcp/__init__.py

@@ -16,9 +16,7 @@ import os
 import warnings
 
 # Suppress litellm deprecation warnings about event loop
-warnings.filterwarnings(
-    "ignore", message="There is no current event loop", category=DeprecationWarning
-)
+warnings.filterwarnings("ignore", message="There is no current event loop", category=DeprecationWarning)
 
 if os.environ.get("HANZO_MCP_TRANSPORT") == "stdio":
     try:

hanzo_mcp/analytics/posthog_analytics.py

@@ -16,7 +16,6 @@ import traceback
 from typing import Any, Dict, TypeVar, Callable, Optional
 from datetime import datetime
 from dataclasses import dataclass
-from importlib.metadata import version, PackageNotFoundError
 
 # Try to import PostHog, but make it optional
 try:
@@ -27,12 +26,6 @@ except ImportError:
     POSTHOG_AVAILABLE = False
     Posthog = None
 
-# Get package version
-try:
-    MCP_VERSION = version("hanzo-mcp")
-except PackageNotFoundError:
-    MCP_VERSION = "0.8.7"  # Fallback version
-
 
 F = TypeVar("F", bound=Callable[..., Any])
 
@@ -105,7 +98,7 @@ class Analytics:
                 "timestamp": datetime.utcnow().isoformat(),
                 "platform": platform.system(),
                 "python_version": platform.python_version(),
-                "mcp_version": MCP_VERSION,
+                "mcp_version": "0.6.13",  # TODO: Get from package
                 **(properties or {}),
             }
 
@@ -144,9 +137,7 @@ class Analytics:
 
         self.capture("tool_used", properties)
 
-    def track_error(
-        self, error: Exception, context: Optional[Dict[str, Any]] = None
-    ) -> None:
+    def track_error(self, error: Exception, context: Optional[Dict[str, Any]] = None) -> None:
         """Track an error event."""
         if not self.config.capture_errors:
             return
@@ -166,9 +157,7 @@ class Analytics:
             return default
 
         try:
-            return self._client.feature_enabled(
-                flag_key, self.config.distinct_id, default=default
-            )
+            return self._client.feature_enabled(flag_key, self.config.distinct_id, default=default)
         except Exception:
             return default
 
@@ -178,9 +167,7 @@ class Analytics:
             return default
 
         try:
-            return self._client.get_feature_flag(
-                flag_key, self.config.distinct_id, default=default
-            )
+            return self._client.get_feature_flag(flag_key, self.config.distinct_id, default=default)
         except Exception:
             return default
 

hanzo_mcp/bridge.py

@@ -134,9 +134,7 @@ class ClaudeBridge(FastMCP):
             """
             logger.info(f"Bridge {self.config.instance_id}: Task delegation")
 
-            delegation_prompt = self._build_delegation_prompt(
-                task, requirements, constraints
-            )
+            delegation_prompt = self._build_delegation_prompt(task, requirements, constraints)
             result = await self._forward_to_claude(delegation_prompt)
 
             return {
@@ -178,9 +176,7 @@ class ClaudeBridge(FastMCP):
             }
 
         @self.tool()
-        async def share_context_with_claude(
-            key: str, value: Any, description: Optional[str] = None
-        ) -> bool:
+        async def share_context_with_claude(key: str, value: Any, description: Optional[str] = None) -> bool:
             """Share context with another Claude instance.
 
             Args:
@@ -263,9 +259,7 @@ class ClaudeBridge(FastMCP):
                 "shared_context_keys": list(self.shared_context.keys()),
             }
 
-    def _build_review_prompt(
-        self, code: str, description: str, focus_areas: Optional[List[str]]
-    ) -> str:
+    def _build_review_prompt(self, code: str, description: str, focus_areas: Optional[List[str]]) -> str:
         """Build a code review prompt."""
         prompt = f"""
         Please review the following code:
@@ -293,9 +287,7 @@ class ClaudeBridge(FastMCP):
 
         return prompt
 
-    def _build_delegation_prompt(
-        self, task: str, requirements: List[str], constraints: Optional[List[str]]
-    ) -> str:
+    def _build_delegation_prompt(self, task: str, requirements: List[str], constraints: Optional[List[str]]) -> str:
         """Build a task delegation prompt."""
         prompt = f"""
         Please complete the following task:
@@ -320,9 +312,7 @@ class ClaudeBridge(FastMCP):
 
         return prompt
 
-    def _build_opinion_prompt(
-        self, question: str, options: Optional[List[str]], criteria: Optional[List[str]]
-    ) -> str:
+    def _build_opinion_prompt(self, question: str, options: Optional[List[str]], criteria: Optional[List[str]]) -> str:
         """Build an opinion request prompt."""
         prompt = f"""
         I need your opinion on the following:
@@ -334,7 +324,7 @@ class ClaudeBridge(FastMCP):
             prompt += f"""
         
         Options to consider:
-        {chr(10).join(f"{i+1}. {opt}" for i, opt in enumerate(options))}
+        {chr(10).join(f"{i + 1}. {opt}" for i, opt in enumerate(options))}
         """
 
         if criteria:
@@ -351,9 +341,7 @@ class ClaudeBridge(FastMCP):
 
         return prompt
 
-    async def _forward_to_claude(
-        self, prompt: str, context: Optional[str] = None
-    ) -> str:
+    async def _forward_to_claude(self, prompt: str, context: Optional[str] = None) -> str:
         """Forward a request to the target Claude instance.
 
         In production, this would make an actual API call to the Claude instance.
@@ -365,9 +353,7 @@ class ClaudeBridge(FastMCP):
             full_prompt = f"Context: {context}\n\n{prompt}"
 
         # Log the forwarding
-        logger.info(
-            f"Forwarding from instance {self.config.source_instance} to {self.config.target_instance}"
-        )
+        logger.info(f"Forwarding from instance {self.config.source_instance} to {self.config.target_instance}")
         logger.debug(f"Prompt: {full_prompt[:200]}...")
 
         # In production, this would:
@@ -440,9 +426,7 @@ async def run_bridge_server(config: BridgeConfig):
 
 def main():
     """Main entry point for the bridge."""
-    parser = argparse.ArgumentParser(
-        description="MCP Bridge for Claude-to-Claude communication"
-    )
+    parser = argparse.ArgumentParser(description="MCP Bridge for Claude-to-Claude communication")
     parser.add_argument(
         "--target-port",
         type=int,

hanzo_mcp/cli.py

@@ -62,13 +62,12 @@ def main() -> None:
     # Avoid importing hanzo_mcp package just to get version (it can have side-effects).
     try:
         from importlib.metadata import version as _pkg_version  # py3.8+
+
         _version = _pkg_version("hanzo-mcp")
     except Exception:
         _version = "unknown"
 
-    parser = argparse.ArgumentParser(
-        description="MCP server implementing Hanzo AI capabilities"
-    )
+    parser = argparse.ArgumentParser(description="MCP server implementing Hanzo AI capabilities")
 
     parser.add_argument("--version", action="version", version=f"hanzo-mcp {_version}")
 
@@ -242,12 +241,8 @@ def main() -> None:
     port: int = cast(int, args.port)
     log_level: str = cast(str, args.log_level)
     project_dir: str | None = cast(str | None, args.project_dir)
-    allowed_paths: list[str] = (
-        cast(list[str], args.allowed_paths) if args.allowed_paths else []
-    )
-    project_paths: list[str] = (
-        cast(list[str], args.project_paths) if args.project_paths else []
-    )
+    allowed_paths: list[str] = cast(list[str], args.allowed_paths) if args.allowed_paths else []
+    project_paths: list[str] = cast(list[str], args.project_paths) if args.project_paths else []
 
     # Handle project_dir parameter (add to both allowed_paths and project_paths)
     if project_dir:
@@ -257,9 +252,7 @@ def main() -> None:
             project_paths.append(project_dir)
 
     if install:
-        install_claude_desktop_config(
-            name, allowed_paths, disable_write_tools, disable_search_tools, host, port
-        )
+        install_claude_desktop_config(name, allowed_paths, disable_write_tools, disable_search_tools, host, port)
         return
 
     # Get logger
@@ -288,9 +281,9 @@ def main() -> None:
             format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
         )
 
-    # If no allowed paths are specified, use the current directory
+    # If no allowed paths are specified, use the home directory
     if not allowed_paths:
-        allowed_paths = [os.getcwd()]
+        allowed_paths = [os.path.expanduser("~")]
 
     # Run in dev mode if requested
     if dev:
@@ -404,9 +397,7 @@ def install_claude_desktop_config(
         args.append("--disable-search-tools")
 
     # Create config object
-    config: dict[str, Any] = {
-        "mcpServers": {name: {"command": script_path.as_posix(), "args": args}}
-    }
+    config: dict[str, Any] = {"mcpServers": {name: {"command": script_path.as_posix(), "args": args}}}
 
     # Check if the file already exists
     if config_file.exists():

hanzo_mcp/cli_enhanced.py

@@ -344,10 +344,7 @@ def apply_cli_overrides(args: argparse.Namespace) -> Dict[str, Any]:
         vector_config["enabled"] = True
     if hasattr(args, "vector_store_path") and args.vector_store_path:
         vector_config["data_path"] = args.vector_store_path
-    if (
-        hasattr(args, "embedding_model")
-        and args.embedding_model != "text-embedding-3-small"
-    ):
+    if hasattr(args, "embedding_model") and args.embedding_model != "text-embedding-3-small":
         vector_config["embedding_model"] = args.embedding_model
 
     if vector_config:
@@ -380,9 +377,7 @@ def list_tools(settings: HanzoMCPSettings) -> None:
 
     logger.info(f"\nTotal: {len(TOOL_REGISTRY)} tools")
     enabled_count = len(settings.get_enabled_tools())
-    logger.info(
-        f"Enabled: {enabled_count}, Disabled: {len(TOOL_REGISTRY) - enabled_count}"
-    )
+    logger.info(f"Enabled: {enabled_count}, Disabled: {len(TOOL_REGISTRY) - enabled_count}")
 
 
 def main() -> None:
@@ -442,15 +437,11 @@ def main() -> None:
         agent_base_url=settings.agent.base_url,
         agent_max_iterations=settings.agent.max_iterations,
         agent_max_tool_uses=settings.agent.max_tool_uses,
-        enable_agent_tool=settings.agent.enabled
-        or settings.is_tool_enabled("dispatch_agent"),
+        enable_agent_tool=settings.agent.enabled or settings.is_tool_enabled("dispatch_agent"),
         disable_write_tools=not any(
-            settings.is_tool_enabled(t)
-            for t in ["write", "edit", "multi_edit", "content_replace"]
-        ),
-        disable_search_tools=not any(
-            settings.is_tool_enabled(t) for t in ["grep", "grep_ast"]
+            settings.is_tool_enabled(t) for t in ["write", "edit", "multi_edit", "content_replace"]
         ),
+        disable_search_tools=not any(settings.is_tool_enabled(t) for t in ["grep", "grep_ast"]),
         host=settings.server.host,
         port=settings.server.port,
         enabled_tools=settings.enabled_tools,  # Pass individual tool configuration

hanzo_mcp/cli_plugin.py

@@ -53,9 +53,7 @@ Examples:
             create_plugin_template(output_dir, args.name)
             print(f"\n✅ Plugin template created successfully!")
             print(f"\nTo use your plugin:")
-            print(
-                f"1. Edit the tool implementation in {output_dir / f'{args.name}_tool.py'}"
-            )
+            print(f"1. Edit the tool implementation in {output_dir / f'{args.name}_tool.py'}")
             print(f"2. Restart Hanzo MCP to load the plugin")
             print(f"3. Add '{args.name}' to your mode's tool list")
         except Exception as e:
@@ -81,13 +79,9 @@ Examples:
                     print(f"\n  {name}:")
                     print(f"    Source: {plugin.source_path}")
                     if plugin.metadata:
-                        print(
-                            f"    Version: {plugin.metadata.get('version', 'unknown')}"
-                        )
+                        print(f"    Version: {plugin.metadata.get('version', 'unknown')}")
                         print(f"    Author: {plugin.metadata.get('author', 'unknown')}")
-                        print(
-                            f"    Description: {plugin.metadata.get('description', '')}"
-                        )
+                        print(f"    Description: {plugin.metadata.get('description', '')}")
         except Exception as e:
             print(f"❌ Error listing plugins: {e}", file=sys.stderr)
             sys.exit(1)

hanzo_mcp/config/settings.py

@@ -53,9 +53,7 @@ class ProjectConfig:
     tasks: List[Dict[str, Any]] = field(default_factory=list)
     enabled_tools: Dict[str, bool] = field(default_factory=dict)
     disabled_tools: List[str] = field(default_factory=list)
-    mcp_servers: List[str] = field(
-        default_factory=list
-    )  # Names of enabled MCP servers for this project
+    mcp_servers: List[str] = field(default_factory=list)  # Names of enabled MCP servers for this project
 
 
 @dataclass
@@ -109,9 +107,7 @@ class HanzoMCPSettings:
     # MCP Hub configuration
     mcp_servers: Dict[str, MCPServerConfig] = field(default_factory=dict)
     hub_enabled: bool = False
-    trusted_servers: List[str] = field(
-        default_factory=list
-    )  # Whitelist of trusted server names
+    trusted_servers: List[str] = field(default_factory=list)  # Whitelist of trusted server names
 
     # Project-specific configurations
     projects: Dict[str, ProjectConfig] = field(default_factory=dict)
@@ -123,9 +119,7 @@ class HanzoMCPSettings:
     def __post_init__(self):
         """Initialize default tool states if not specified."""
         if not self.enabled_tools:
-            self.enabled_tools = {
-                name: config.enabled for name, config in TOOL_REGISTRY.items()
-            }
+            self.enabled_tools = {name: config.enabled for name, config in TOOL_REGISTRY.items()}
 
         # Apply disabled_tools list to enabled_tools dict
         for tool_name in self.disabled_tools:
@@ -139,9 +133,7 @@ class HanzoMCPSettings:
             return False
         return self.enabled_tools.get(
             tool_name,
-            TOOL_REGISTRY.get(
-                tool_name, type("obj", (object,), {"enabled": False})
-            ).enabled,
+            TOOL_REGISTRY.get(tool_name, type("obj", (object,), {"enabled": False})).enabled,
         )
 
     def enable_tool(self, tool_name: str) -> bool:
@@ -399,11 +391,7 @@ def _load_from_env() -> Dict[str, Any]:
         config.setdefault("agent", {})["enabled"] = True
 
     # Check for MODE/PERSONALITY/HANZO_MODE
-    if (
-        mode := os.environ.get("HANZO_MODE")
-        or os.environ.get("PERSONALITY")
-        or os.environ.get("MODE")
-    ):
+    if mode := os.environ.get("HANZO_MODE") or os.environ.get("PERSONALITY") or os.environ.get("MODE"):
         config["active_mode"] = mode
 
     # Check for other environment overrides
@@ -496,9 +484,7 @@ def save_settings(settings: HanzoMCPSettings, global_config: bool = True) -> Pat
     return config_path
 
 
-def _merge_config(
-    base_settings: HanzoMCPSettings, config_dict: Dict[str, Any]
-) -> HanzoMCPSettings:
+def _merge_config(base_settings: HanzoMCPSettings, config_dict: Dict[str, Any]) -> HanzoMCPSettings:
     """Merge configuration dictionary into settings object."""
     # Convert to dict, merge, then convert back
     base_dict = asdict(base_settings)

hanzo_mcp/config/tool_config.py

@@ -105,7 +105,7 @@ TOOL_REGISTRY: Dict[str, ToolConfig] = {
         description="Edit Jupyter notebook cells (replace, insert, delete)",
         cli_flag="--disable-notebook-edit",
     ),
-    # Task Management Tools (2)
+    # Todo Tools (2)
     "todo_read": ToolConfig(
         name="todo_read",
         category=ToolCategory.TODO,
@@ -204,6 +204,4 @@ 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