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

hanzo_mcp/core/model_registry.py

@@ -15,7 +15,7 @@ from dataclasses import field, dataclass
 
 class ModelProvider(Enum):
     """Enumeration of AI model providers."""
-    
+
     ANTHROPIC = "anthropic"
     OPENAI = "openai"
     GOOGLE = "google"
@@ -30,7 +30,7 @@ class ModelProvider(Enum):
 @dataclass(frozen=True)
 class ModelConfig:
     """Configuration for a single AI model."""
-    
+
     full_name: str
     provider: ModelProvider
     aliases: Set[str] = field(default_factory=set)
@@ -46,16 +46,16 @@ class ModelConfig:
 
 class ModelRegistry:
     """Centralized registry for all AI models.
-    
+
     Thread-safe singleton implementation ensuring single source of truth
     for model configurations across the codebase.
     """
-    
+
     _instance: Optional[ModelRegistry] = None
     _lock = threading.Lock()
     _models: Dict[str, ModelConfig] = {}
     _initialized = False
-    
+
     def __new__(cls) -> ModelRegistry:
         """Thread-safe singleton pattern."""
         if cls._instance is None:
@@ -63,7 +63,7 @@ class ModelRegistry:
                 if cls._instance is None:
                     cls._instance = super().__new__(cls)
         return cls._instance
-    
+
     def __init__(self) -> None:
         """Initialize model registry once."""
         if not self._initialized:
@@ -71,225 +71,253 @@ class ModelRegistry:
                 if not self._initialized:
                     self._initialize_models()
                     self._initialized = True
-    
+
     def _initialize_models(self) -> None:
         """Initialize all model configurations."""
         # Claude models
-        self._register(ModelConfig(
-            full_name="claude-3-5-sonnet-20241022",
-            provider=ModelProvider.ANTHROPIC,
-            aliases={"claude", "cc", "claude-code", "sonnet", "sonnet-4.1"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=200000,
-            max_output=8192,
-            api_key_env="ANTHROPIC_API_KEY",
-            cli_command="claude",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="claude-opus-4-1-20250805",
-            provider=ModelProvider.ANTHROPIC,
-            aliases={"opus", "opus-4.1", "claude-opus"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=200000,
-            max_output=8192,
-            api_key_env="ANTHROPIC_API_KEY",
-            cli_command="claude",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="claude-3-haiku-20240307",
-            provider=ModelProvider.ANTHROPIC,
-            aliases={"haiku", "claude-haiku"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=200000,
-            max_output=4096,
-            api_key_env="ANTHROPIC_API_KEY",
-            cli_command="claude",
-        ))
-        
+        self._register(
+            ModelConfig(
+                full_name="claude-3-5-sonnet-20241022",
+                provider=ModelProvider.ANTHROPIC,
+                aliases={"claude", "cc", "claude-code", "sonnet", "sonnet-4.1"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=200000,
+                max_output=8192,
+                api_key_env="ANTHROPIC_API_KEY",
+                cli_command="claude",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="claude-opus-4-1-20250805",
+                provider=ModelProvider.ANTHROPIC,
+                aliases={"opus", "opus-4.1", "claude-opus"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=200000,
+                max_output=8192,
+                api_key_env="ANTHROPIC_API_KEY",
+                cli_command="claude",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="claude-3-haiku-20240307",
+                provider=ModelProvider.ANTHROPIC,
+                aliases={"haiku", "claude-haiku"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=200000,
+                max_output=4096,
+                api_key_env="ANTHROPIC_API_KEY",
+                cli_command="claude",
+            )
+        )
+
         # OpenAI models
-        self._register(ModelConfig(
-            full_name="gpt-4-turbo",
-            provider=ModelProvider.OPENAI,
-            aliases={"gpt4", "gpt-4", "codex"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=128000,
-            max_output=4096,
-            api_key_env="OPENAI_API_KEY",
-            cli_command="openai",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="gpt-5-turbo",
-            provider=ModelProvider.OPENAI,
-            aliases={"gpt5", "gpt-5"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=256000,
-            max_output=16384,
-            api_key_env="OPENAI_API_KEY",
-            cli_command="openai",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="o1-preview",
-            provider=ModelProvider.OPENAI,
-            aliases={"o1", "openai-o1"},
-            supports_vision=False,
-            supports_tools=False,
-            context_window=128000,
-            max_output=32768,
-            api_key_env="OPENAI_API_KEY",
-            cli_command="openai",
-        ))
-        
+        self._register(
+            ModelConfig(
+                full_name="gpt-4-turbo",
+                provider=ModelProvider.OPENAI,
+                aliases={"gpt4", "gpt-4", "codex"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=128000,
+                max_output=4096,
+                api_key_env="OPENAI_API_KEY",
+                cli_command="openai",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="gpt-5-turbo",
+                provider=ModelProvider.OPENAI,
+                aliases={"gpt5", "gpt-5"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=256000,
+                max_output=16384,
+                api_key_env="OPENAI_API_KEY",
+                cli_command="openai",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="o1-preview",
+                provider=ModelProvider.OPENAI,
+                aliases={"o1", "openai-o1"},
+                supports_vision=False,
+                supports_tools=False,
+                context_window=128000,
+                max_output=32768,
+                api_key_env="OPENAI_API_KEY",
+                cli_command="openai",
+            )
+        )
+
         # Google models
-        self._register(ModelConfig(
-            full_name="gemini-2.0-flash-exp",
-            provider=ModelProvider.GOOGLE,
-            aliases={"gemini-2", "gemini-2.0", "gemini2"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=1000000,
-            max_output=8192,
-            api_key_env="GEMINI_API_KEY",
-            cli_command="gemini",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="gemini-exp-1206",
-            provider=ModelProvider.GOOGLE,
-            aliases={"gemini-2.5", "gemini-2.5-pro", "gemini-pro-2.5"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=2000000,
-            max_output=8192,
-            api_key_env="GEMINI_API_KEY",
-            cli_command="gemini",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="gemini-1.5-pro",
-            provider=ModelProvider.GOOGLE,
-            aliases={"gemini", "gemini-pro", "gemini-1.5"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=2000000,
-            max_output=8192,
-            api_key_env="GEMINI_API_KEY",
-            cli_command="gemini",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="gemini-1.5-flash",
-            provider=ModelProvider.GOOGLE,
-            aliases={"gemini-flash", "flash"},
-            supports_vision=True,
-            supports_tools=True,
-            context_window=1000000,
-            max_output=8192,
-            api_key_env="GEMINI_API_KEY",
-            cli_command="gemini",
-        ))
-        
+        self._register(
+            ModelConfig(
+                full_name="gemini-2.0-flash-exp",
+                provider=ModelProvider.GOOGLE,
+                aliases={"gemini-2", "gemini-2.0", "gemini2"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=1000000,
+                max_output=8192,
+                api_key_env="GEMINI_API_KEY",
+                cli_command="gemini",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="gemini-exp-1206",
+                provider=ModelProvider.GOOGLE,
+                aliases={"gemini-2.5", "gemini-2.5-pro", "gemini-pro-2.5"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=2000000,
+                max_output=8192,
+                api_key_env="GEMINI_API_KEY",
+                cli_command="gemini",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="gemini-1.5-pro",
+                provider=ModelProvider.GOOGLE,
+                aliases={"gemini", "gemini-pro", "gemini-1.5"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=2000000,
+                max_output=8192,
+                api_key_env="GEMINI_API_KEY",
+                cli_command="gemini",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="gemini-1.5-flash",
+                provider=ModelProvider.GOOGLE,
+                aliases={"gemini-flash", "flash"},
+                supports_vision=True,
+                supports_tools=True,
+                context_window=1000000,
+                max_output=8192,
+                api_key_env="GEMINI_API_KEY",
+                cli_command="gemini",
+            )
+        )
+
         # xAI models
-        self._register(ModelConfig(
-            full_name="grok-4",
-            provider=ModelProvider.XAI,
-            aliases={"grok", "xai-grok", "grok-2"},  # grok-2 for backward compat
-            supports_vision=True,  # Grok-4 supports multimodal
-            supports_tools=True,
-            context_window=128000,
-            max_output=8192,
-            api_key_env="XAI_API_KEY",
-            cli_command="grok",
-        ))
-        
+        self._register(
+            ModelConfig(
+                full_name="grok-4",
+                provider=ModelProvider.XAI,
+                aliases={"grok", "xai-grok", "grok-2"},  # grok-2 for backward compat
+                supports_vision=True,  # Grok-4 supports multimodal
+                supports_tools=True,
+                context_window=128000,
+                max_output=8192,
+                api_key_env="XAI_API_KEY",
+                cli_command="grok",
+            )
+        )
+
         # Ollama models
-        self._register(ModelConfig(
-            full_name="ollama/llama-3.2-3b",
-            provider=ModelProvider.OLLAMA,
-            aliases={"llama", "llama-3.2", "llama3"},
-            supports_vision=False,
-            supports_tools=False,
-            context_window=128000,
-            max_output=4096,
-            api_key_env=None,  # Local model
-            cli_command="ollama",
-        ))
-        
-        self._register(ModelConfig(
-            full_name="ollama/mistral:7b",
-            provider=ModelProvider.MISTRAL,
-            aliases={"mistral", "mistral-7b"},
-            supports_vision=False,
-            supports_tools=False,
-            context_window=32000,
-            max_output=4096,
-            api_key_env=None,  # Local model
-            cli_command="ollama",
-        ))
-        
+        self._register(
+            ModelConfig(
+                full_name="ollama/llama-3.2-3b",
+                provider=ModelProvider.OLLAMA,
+                aliases={"llama", "llama-3.2", "llama3"},
+                supports_vision=False,
+                supports_tools=False,
+                context_window=128000,
+                max_output=4096,
+                api_key_env=None,  # Local model
+                cli_command="ollama",
+            )
+        )
+
+        self._register(
+            ModelConfig(
+                full_name="ollama/mistral:7b",
+                provider=ModelProvider.MISTRAL,
+                aliases={"mistral", "mistral-7b"},
+                supports_vision=False,
+                supports_tools=False,
+                context_window=32000,
+                max_output=4096,
+                api_key_env=None,  # Local model
+                cli_command="ollama",
+            )
+        )
+
         # DeepSeek models
-        self._register(ModelConfig(
-            full_name="deepseek-coder-v2",
-            provider=ModelProvider.DEEPSEEK,
-            aliases={"deepseek", "deepseek-coder"},
-            supports_vision=False,
-            supports_tools=True,
-            context_window=128000,
-            max_output=8192,
-            api_key_env="DEEPSEEK_API_KEY",
-            cli_command="deepseek",
-        ))
-    
+        self._register(
+            ModelConfig(
+                full_name="deepseek-coder-v2",
+                provider=ModelProvider.DEEPSEEK,
+                aliases={"deepseek", "deepseek-coder"},
+                supports_vision=False,
+                supports_tools=True,
+                context_window=128000,
+                max_output=8192,
+                api_key_env="DEEPSEEK_API_KEY",
+                cli_command="deepseek",
+            )
+        )
+
     def _register(self, config: ModelConfig) -> None:
         """Register a model configuration.
-        
+
         Args:
             config: Model configuration to register
         """
         # Register by full name
         self._models[config.full_name] = config
-        
+
         # Register all aliases
         for alias in config.aliases:
             self._models[alias.lower()] = config
-    
+
     def get(self, model_name: str) -> Optional[ModelConfig]:
         """Get model configuration by name or alias.
-        
+
         Args:
             model_name: Model name or alias
-            
+
         Returns:
             Model configuration or None if not found
         """
         return self._models.get(model_name.lower())
-    
+
     def resolve(self, model_name: str) -> str:
         """Resolve model name or alias to full model name.
-        
+
         Args:
             model_name: Model name or alias
-            
+
         Returns:
             Full model name, or original if not found
         """
         config = self.get(model_name)
         return config.full_name if config else model_name
-    
+
     def get_by_provider(self, provider: ModelProvider) -> List[ModelConfig]:
         """Get all unique models for a specific provider.
-        
+
         Args:
             provider: Model provider
-            
+
         Returns:
             List of unique model configurations
         """
@@ -300,7 +328,7 @@ class ModelRegistry:
                 seen_names.add(config.full_name)
                 results.append(config)
         return results
-    
+
     def get_models_supporting(
         self,
         vision: Optional[bool] = None,
@@ -308,61 +336,61 @@ class ModelRegistry:
         streaming: Optional[bool] = None,
     ) -> List[ModelConfig]:
         """Get unique models supporting specific features.
-        
+
         Args:
             vision: Filter by vision support
             tools: Filter by tool support
             streaming: Filter by streaming support
-            
+
         Returns:
             List of unique matching model configurations
         """
         seen_names = set()
         results = []
-        
+
         for config in self._models.values():
             if config.full_name in seen_names:
                 continue
-                
+
             if vision is not None and config.supports_vision != vision:
                 continue
             if tools is not None and config.supports_tools != tools:
                 continue
             if streaming is not None and config.supports_streaming != streaming:
                 continue
-            
+
             seen_names.add(config.full_name)
             results.append(config)
-        
+
         return results
-    
+
     def get_api_key_env(self, model_name: str) -> Optional[str]:
         """Get the API key environment variable for a model.
-        
+
         Args:
             model_name: Model name or alias
-            
+
         Returns:
             Environment variable name or None
         """
         config = self.get(model_name)
         return config.api_key_env if config else None
-    
+
     def get_cli_command(self, model_name: str) -> Optional[str]:
         """Get the CLI command for a model.
-        
+
         Args:
             model_name: Model name or alias
-            
+
         Returns:
             CLI command or None
         """
         config = self.get(model_name)
         return config.cli_command if config else None
-    
+
     def list_all_models(self) -> List[str]:
         """List all unique model full names.
-        
+
         Returns:
             Sorted list of unique full model names
         """
@@ -370,10 +398,10 @@ class ModelRegistry:
         for config in self._models.values():
             seen_names.add(config.full_name)
         return sorted(list(seen_names))
-    
+
     def list_all_aliases(self) -> Dict[str, str]:
         """List all aliases and their full names.
-        
+
         Returns:
             Dictionary mapping aliases to full names
         """
@@ -391,10 +419,10 @@ registry = ModelRegistry()
 # Convenience functions
 def resolve_model(model_name: str) -> str:
     """Resolve model name or alias to full model name.
-    
+
     Args:
         model_name: Model name or alias
-        
+
     Returns:
         Full model name
     """
@@ -403,10 +431,10 @@ def resolve_model(model_name: str) -> str:
 
 def get_model_config(model_name: str) -> Optional[ModelConfig]:
     """Get model configuration.
-    
+
     Args:
         model_name: Model name or alias
-        
+
     Returns:
         Model configuration or None
     """
@@ -415,10 +443,10 @@ def get_model_config(model_name: str) -> Optional[ModelConfig]:
 
 def get_api_key_env(model_name: str) -> Optional[str]:
     """Get API key environment variable for model.
-    
+
     Args:
         model_name: Model name or alias
-        
+
     Returns:
         Environment variable name or None
     """
@@ -433,4 +461,4 @@ __all__ = [
     "resolve_model",
     "get_model_config",
     "get_api_key_env",
-]
+]

hanzo_mcp/dev_server.py

@@ -159,12 +159,8 @@ class DevServer:
         # 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."""
@@ -218,12 +214,8 @@ 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("--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",
@@ -239,9 +231,7 @@ def run_dev_server():
         choices=["stdio", "sse"],
         help="Transport type (default: stdio)",
     )
-    parser.add_argument(
-        "--host", type=str, default="127.0.0.1", help="Host for SSE transport"
-    )
+    parser.add_argument("--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")
 
     args = parser.parse_args()