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

hanzo_mcp/tools/todo/todo_write.py

@@ -3,13 +3,13 @@
 This module provides the TodoWrite tool for creating and managing a structured task list for a session.
 """
 
-from typing import Annotated, Literal, TypedDict, Unpack, final, override
+from typing import Unpack, Literal, Annotated, TypedDict, final, override
 
-from mcp.server.fastmcp import Context as MCPContext
-from mcp.server import FastMCP
 from pydantic import Field
+from mcp.server import FastMCP
+from mcp.server.fastmcp import Context as MCPContext
 
-from hanzo_mcp.tools.todo.base import TodoBaseTool, TodoStorage
+from hanzo_mcp.tools.todo.base import TodoStorage, TodoBaseTool
 
 
 class TodoItem(TypedDict):
@@ -369,8 +369,6 @@ When in doubt, use this tool. Being proactive with task management demonstrates
 
         @mcp_server.tool(name=self.name, description=self.description)
         async def todo_write(
-            session_id: SessionId,
-            todos: Todos,
-            ctx: MCPContext
+            session_id: SessionId, todos: Todos, ctx: MCPContext
         ) -> str:
             return await tool_self.call(ctx, session_id=session_id, todos=todos)

hanzo_mcp/tools/vector/__init__.py

@@ -7,30 +7,31 @@ Supported backends:
 - Infinity database (default) - High-performance local vector database
 """
 
+from mcp.server import FastMCP
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.permissions import PermissionManager
-from mcp.server import FastMCP
 
 # Try to import vector dependencies
 try:
-    from .infinity_store import InfinityVectorStore
-    from .project_manager import ProjectVectorManager
+    from .index_tool import IndexTool
     from .vector_index import VectorIndexTool
     from .vector_search import VectorSearchTool
-    from .index_tool import IndexTool
-    
+    from .infinity_store import InfinityVectorStore
+    from .project_manager import ProjectVectorManager
+
     VECTOR_AVAILABLE = True
-    
+
     def register_vector_tools(
         mcp_server: FastMCP,
         permission_manager: PermissionManager,
         vector_config: dict | None = None,
         enabled_tools: dict[str, bool] | None = None,
         search_paths: list[str] | None = None,
-        project_manager: 'ProjectVectorManager | None' = None,
+        project_manager: "ProjectVectorManager | None" = None,
     ) -> list[BaseTool]:
         """Register vector database tools with the MCP server.
-        
+
         Args:
             mcp_server: The FastMCP server instance
             permission_manager: Permission manager for access control
@@ -38,58 +39,67 @@ try:
             enabled_tools: Dictionary of individual tool enable states
             search_paths: Paths to search for projects (default: None, uses allowed paths)
             project_manager: Optional existing project manager to reuse
-            
+
         Returns:
             List of registered tools
         """
         if not vector_config or not vector_config.get("enabled", False):
             return []
-        
+
         # Check individual tool enablement
         tool_enabled = enabled_tools or {}
         tools = []
-        
+
         # Use provided project manager or create new one
         if project_manager is None:
             # Initialize project-aware vector manager
             store_config = vector_config.copy()
             project_manager = ProjectVectorManager(
                 global_db_path=store_config.get("data_path"),
-                embedding_model=store_config.get("embedding_model", "text-embedding-3-small"),
+                embedding_model=store_config.get(
+                    "embedding_model", "text-embedding-3-small"
+                ),
                 dimension=store_config.get("dimension", 1536),
             )
-            
+
             # Auto-detect projects from search paths for new manager
             if search_paths:
                 detected_projects = project_manager.detect_projects(search_paths)
                 import logging
+
                 logger = logging.getLogger(__name__)
-                logger.info(f"Detected {len(detected_projects)} projects with LLM.md files")
-        
+                logger.info(
+                    f"Detected {len(detected_projects)} projects with LLM.md files"
+                )
+
         # Register individual tools if enabled
         if tool_enabled.get("index", True):
             tools.append(IndexTool(permission_manager))
-        
+
         if tool_enabled.get("vector_index", True):
             tools.append(VectorIndexTool(permission_manager, project_manager))
-        
+
         if tool_enabled.get("vector_search", True):
             tools.append(VectorSearchTool(permission_manager, project_manager))
-        
+
         # Register with MCP server
         from hanzo_mcp.tools.common.base import ToolRegistry
+
         ToolRegistry.register_tools(mcp_server, tools)
-        
+
         return tools
 
 except ImportError:
     VECTOR_AVAILABLE = False
-    
+
     def register_vector_tools(*args, **kwargs) -> list[BaseTool]:
         """Vector tools not available - missing dependencies."""
         import logging
+
         logger = logging.getLogger(__name__)
-        logger.warning("Vector tools not available. Install infinity-embedded: pip install infinity-embedded")
+        logger.warning(
+            "Vector tools not available. Install infinity-embedded: pip install infinity-embedded"
+        )
         return []
 
 
@@ -99,10 +109,12 @@ __all__ = [
 ]
 
 if VECTOR_AVAILABLE:
-    __all__.extend([
-        "InfinityVectorStore",
-        "ProjectVectorManager",
-        "IndexTool",
-        "VectorIndexTool", 
-        "VectorSearchTool",
-    ])
+    __all__.extend(
+        [
+            "InfinityVectorStore",
+            "ProjectVectorManager",
+            "IndexTool",
+            "VectorIndexTool",
+            "VectorSearchTool",
+        ]
+    )