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

hanzo_mcp/tools/shell/uvx.py

@@ -135,9 +135,7 @@ For long-running servers, use uvx_background instead.
 
             try:
                 # Run installation
-                install_result = subprocess.run(
-                    install_cmd, shell=True, capture_output=True, text=True, timeout=60
-                )
+                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!")
@@ -146,9 +144,7 @@ For long-running servers, use uvx_background instead.
                     import os
 
                     home = os.path.expanduser("~")
-                    os.environ["PATH"] = (
-                        f"{home}/.cargo/bin:{os.environ.get('PATH', '')}"
-                    )
+                    os.environ["PATH"] = f"{home}/.cargo/bin:{os.environ.get('PATH', '')}"
 
                     # Check again
                     if not shutil.which("uvx"):
@@ -197,9 +193,7 @@ curl -LsSf https://astral.sh/uv/install.sh | sh"""
 
         try:
             # Execute command
-            result = subprocess.run(
-                cmd, capture_output=True, text=True, timeout=timeout, check=True
-            )
+            result = subprocess.run(cmd, capture_output=True, text=True, timeout=timeout, check=True)
 
             output = []
             if result.stdout:
@@ -207,11 +201,7 @@ curl -LsSf https://astral.sh/uv/install.sh | sh"""
             if result.stderr:
                 output.append(f"\nSTDERR:\n{result.stderr}")
 
-            return (
-                "\n".join(output)
-                if output
-                else "Command completed successfully with no output."
-            )
+            return "\n".join(output) if output else "Command completed successfully with no output."
 
         except subprocess.TimeoutExpired:
             return f"Error: Command timed out after {timeout} seconds. Use uvx_background for long-running processes."

hanzo_mcp/tools/shell/uvx_background.py

@@ -156,9 +156,7 @@ Use 'processes' to list running processes and 'pkill' to stop them.
 
             try:
                 # Run installation
-                install_result = subprocess.run(
-                    install_cmd, shell=True, capture_output=True, text=True, timeout=60
-                )
+                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!")
@@ -167,9 +165,7 @@ Use 'processes' to list running processes and 'pkill' to stop them.
                     import os
 
                     home = os.path.expanduser("~")
-                    os.environ["PATH"] = (
-                        f"{home}/.cargo/bin:{os.environ.get('PATH', '')}"
-                    )
+                    os.environ["PATH"] = f"{home}/.cargo/bin:{os.environ.get('PATH', '')}"
 
                     # Check again
                     if not shutil.which("uvx"):

hanzo_mcp/tools/shell/uvx_tool.py

@@ -86,9 +86,7 @@ uvx jupyter lab --port 8888  # Auto-backgrounds if needed"""
             cwd: Optional[str] = None,
             python: Optional[str] = None,
         ) -> str:
-            return await tool_self.run(
-                ctx, package=package, args=args, cwd=cwd, python=python
-            )
+            return await tool_self.run(ctx, package=package, args=args, cwd=cwd, python=python)
 
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""

hanzo_mcp/tools/shell/zsh_tool.py

@@ -29,9 +29,7 @@ class ZshTool(BaseScriptTool):
             env: Optional[dict[str, str]] = None,
             timeout: Optional[int] = None,
         ) -> str:
-            return await tool_self.run(
-                ctx, command=command, cwd=cwd, env=env, timeout=timeout
-            )
+            return await tool_self.run(ctx, command=command, cwd=cwd, env=env, timeout=timeout)
 
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
@@ -79,12 +77,12 @@ zsh "npm run dev" --cwd ./frontend  # Auto-backgrounds if needed"""
                     return path
             # Fall back to bash if no zsh found
             return "bash"
-        
+
         # On Unix-like systems, check for zsh
         zsh_path = shutil.which("zsh")
         if zsh_path:
             return zsh_path
-        
+
         # Fall back to bash if zsh not found
         return "bash"
 
@@ -124,14 +122,12 @@ zsh "npm run dev" --cwd ./frontend  # Auto-backgrounds if needed"""
         # Check if zsh is available
         if not shutil.which("zsh") and platform.system() != "Windows":
             return "Error: Zsh is not installed. Please install zsh first."
-        
+
         # Prepare working directory
         work_dir = Path(cwd).resolve() if cwd else Path.cwd()
 
         # Use execute_sync which has auto-backgrounding
-        output = await self.execute_sync(
-            command, cwd=work_dir, env=env, timeout=timeout
-        )
+        output = await self.execute_sync(command, cwd=work_dir, env=env, timeout=timeout)
         return output if output else "Command completed successfully (no output)"
 
 
@@ -152,12 +148,12 @@ class ShellTool(BaseScriptTool):
             # Also check if .zshrc exists
             if (Path.home() / ".zshrc").exists():
                 return "zsh"
-        
+
         # Check for user's preferred shell
         user_shell = os.environ.get("SHELL", "")
         if user_shell and Path(user_shell).exists():
             return user_shell
-        
+
         # Default to bash
         return "bash"
 
@@ -173,9 +169,7 @@ class ShellTool(BaseScriptTool):
             env: Optional[dict[str, str]] = None,
             timeout: Optional[int] = None,
         ) -> str:
-            return await tool_self.run(
-                ctx, command=command, cwd=cwd, env=env, timeout=timeout
-            )
+            return await tool_self.run(ctx, command=command, cwd=cwd, env=env, timeout=timeout)
 
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
@@ -249,12 +243,10 @@ shell "npm run dev" --cwd ./frontend  # Auto-backgrounds if needed"""
 
         # Add shell info to output if verbose
         shell_name = os.path.basename(self._best_shell)
-        
+
         # Use execute_sync which has auto-backgrounding
-        output = await self.execute_sync(
-            command, cwd=work_dir, env=env, timeout=timeout
-        )
-        
+        output = await self.execute_sync(command, cwd=work_dir, env=env, timeout=timeout)
+
         if output:
             return output
         else:
@@ -263,4 +255,4 @@ shell "npm run dev" --cwd ./frontend  # Auto-backgrounds if needed"""
 
 # Create tool instances
 zsh_tool = ZshTool()
-shell_tool = ShellTool()  # Smart shell that prefers zsh
+shell_tool = ShellTool()  # Smart shell that prefers zsh

hanzo_mcp/tools/todo/todo.py

@@ -161,9 +161,7 @@ todo --filter in_progress
                     priority_icon = {"high": "🔴", "medium": "🟡", "low": "🟢"}.get(
                         todo.get("priority", "medium"), "⚪"
                     )
-                    output.append(
-                        f"{priority_icon} [{todo['id'][:8]}] {todo['content']}"
-                    )
+                    output.append(f"{priority_icon} [{todo['id'][:8]}] {todo['content']}")
 
         # Summary
         output.append(

hanzo_mcp/tools/todo/todo_read.py

@@ -14,9 +14,7 @@ from hanzo_mcp.tools.todo.base import TodoStorage, TodoBaseTool
 
 SessionId = Annotated[
     str | int | float,
-    Field(
-        description="Unique identifier for the Claude Desktop session (generate using timestamp command)"
-    ),
+    Field(description="Unique identifier for the Claude Desktop session (generate using timestamp command)"),
 ]
 
 
@@ -109,13 +107,9 @@ Usage:
 
             # Log status
             if todos:
-                await tool_ctx.info(
-                    f"Found {len(todos)} todos for session {session_id}"
-                )
+                await tool_ctx.info(f"Found {len(todos)} todos for session {session_id}")
             else:
-                await tool_ctx.info(
-                    f"No todos found for session {session_id} (returning empty list)"
-                )
+                await tool_ctx.info(f"No todos found for session {session_id} (returning empty list)")
 
             # Return todos as JSON string
             result = json.dumps(todos, indent=2)

hanzo_mcp/tools/todo/todo_write.py

@@ -309,9 +309,7 @@ When in doubt, use this tool. Being proactive with task management demonstrates
 
             # Log storage stats
             session_count = TodoStorage.get_session_count()
-            await tool_ctx.info(
-                f"Successfully stored todos. Total active sessions: {session_count}"
-            )
+            await tool_ctx.info(f"Successfully stored todos. Total active sessions: {session_count}")
 
             # Provide feedback about the todos
             if todos:
@@ -328,23 +326,15 @@ When in doubt, use this tool. Being proactive with task management demonstrates
                 # Create summary
                 summary_parts = []
                 if status_counts:
-                    status_summary = ", ".join(
-                        [f"{count} {status}" for status, count in status_counts.items()]
-                    )
+                    status_summary = ", ".join([f"{count} {status}" for status, count in status_counts.items()])
                     summary_parts.append(f"Status: {status_summary}")
 
                 if priority_counts:
-                    priority_summary = ", ".join(
-                        [
-                            f"{count} {priority}"
-                            for priority, count in priority_counts.items()
-                        ]
-                    )
+                    priority_summary = ", ".join([f"{count} {priority}" for priority, count in priority_counts.items()])
                     summary_parts.append(f"Priority: {priority_summary}")
 
-                summary = (
-                    f"Successfully stored {len(todos)} todos for session {session_id}.\n"
-                    + "; ".join(summary_parts)
+                summary = f"Successfully stored {len(todos)} todos for session {session_id}.\n" + "; ".join(
+                    summary_parts
                 )
 
                 return summary
@@ -368,7 +358,5 @@ When in doubt, use this tool. Being proactive with task management demonstrates
         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_write(
-            session_id: SessionId, todos: Todos, ctx: MCPContext
-        ) -> str:
+        async def todo_write(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

@@ -56,9 +56,7 @@ try:
             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),
             )
 
@@ -68,9 +66,7 @@ try:
                 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):
@@ -97,9 +93,7 @@ except ImportError:
         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 []
 
 

hanzo_mcp/tools/vector/ast_analyzer.py

@@ -130,9 +130,7 @@ class ASTAnalyzer:
                 return self._analyze_python_file(file_path, content, file_hash)
             else:
                 # Generic analysis for other languages
-                return self._analyze_generic_file(
-                    file_path, content, file_hash, language
-                )
+                return self._analyze_generic_file(file_path, content, file_hash, language)
 
         except Exception as e:
             import logging
@@ -177,9 +175,7 @@ class ASTAnalyzer:
 
         return language_map.get(extension)
 
-    def _analyze_python_file(
-        self, file_path: str, content: str, file_hash: str
-    ) -> FileAST:
+    def _analyze_python_file(self, file_path: str, content: str, file_hash: str) -> FileAST:
         """Analyze Python file using both AST and tree-sitter."""
         symbols = []
         ast_nodes = []
@@ -228,9 +224,7 @@ class ASTAnalyzer:
             dependencies=dependencies,
         )
 
-    def _analyze_generic_file(
-        self, file_path: str, content: str, file_hash: str, language: str
-    ) -> FileAST:
+    def _analyze_generic_file(self, file_path: str, content: str, file_hash: str, language: str) -> FileAST:
         """Generic analysis for non-Python files."""
         # For now, just basic line-based analysis
         # Could be enhanced with language-specific parsers
@@ -248,11 +242,7 @@ class ASTAnalyzer:
 
             # Basic function detection (works for many C-style languages)
             if language in ["javascript", "typescript", "java", "cpp", "c"]:
-                if (
-                    "function " in line
-                    or line.startswith("def ")
-                    or " function(" in line
-                ):
+                if "function " in line or line.startswith("def ") or " function(" in line:
                     # Extract function name
                     parts = line.split()
                     for j, part in enumerate(parts):
@@ -399,9 +389,7 @@ class PythonSymbolExtractor(ast.NodeVisitor):
 
         # Extract base classes
         bases = [self._get_name(base) for base in node.bases]
-        signature = (
-            f"class {node.name}({', '.join(bases)})" if bases else f"class {node.name}"
-        )
+        signature = f"class {node.name}({', '.join(bases)})" if bases else f"class {node.name}"
 
         symbol = Symbol(
             name=node.name,
@@ -441,9 +429,7 @@ class PythonSymbolExtractor(ast.NodeVisitor):
 
         for alias in node.names:
             if alias.name != "*":
-                import_item = (
-                    f"{node.module}.{alias.name}" if node.module else alias.name
-                )
+                import_item = f"{node.module}.{alias.name}" if node.module else alias.name
                 self.imports.append(import_item)
 
     def visit_Assign(self, node):

hanzo_mcp/tools/vector/git_ingester.py

@@ -153,16 +153,12 @@ class GitIngester:
         )
         return result.stdout.strip()
 
-    def _get_repository_files(
-        self, repo_path: Path, patterns: Optional[List[str]] = None
-    ) -> List[Path]:
+    def _get_repository_files(self, repo_path: Path, patterns: Optional[List[str]] = None) -> List[Path]:
         """Get list of files in repository matching patterns."""
         # Use git ls-files to respect .gitignore
         cmd = ["git", "ls-files"]
 
-        result = subprocess.run(
-            cmd, cwd=repo_path, capture_output=True, text=True, check=True
-        )
+        result = subprocess.run(cmd, cwd=repo_path, capture_output=True, text=True, check=True)
 
         files = []
         for line in result.stdout.strip().split("\n"):
@@ -178,9 +174,7 @@ class GitIngester:
 
         return files
 
-    def _get_commit_history(
-        self, repo_path: Path, branch: str = "HEAD", max_commits: int = 1000
-    ) -> List[GitCommit]:
+    def _get_commit_history(self, repo_path: Path, branch: str = "HEAD", max_commits: int = 1000) -> List[GitCommit]:
         """Get commit history for the repository."""
         # Get commit list with basic info
         result = subprocess.run(
@@ -222,9 +216,7 @@ class GitIngester:
 
         return commits
 
-    def _get_commit_files(
-        self, repo_path: Path, commit_hash: str
-    ) -> List[Dict[str, str]]:
+    def _get_commit_files(self, repo_path: Path, commit_hash: str) -> List[Dict[str, str]]:
         """Get list of files changed in a commit."""
         result = subprocess.run(
             ["git", "show", "--name-status", "--format=", commit_hash],
@@ -275,21 +267,15 @@ class GitIngester:
             if history:
                 metadata["first_commit"] = history[-1]["hash"]
                 metadata["last_commit"] = history[0]["hash"]
-                metadata["last_modified"] = datetime.fromtimestamp(
-                    history[0]["timestamp"]
-                ).isoformat()
+                metadata["last_modified"] = datetime.fromtimestamp(history[0]["timestamp"]).isoformat()
 
         # Add blame information if requested
         if include_blame:
             blame_data = self._get_file_blame(repo_path, relative_path)
-            metadata["unique_authors"] = len(
-                set(b["author"] for b in blame_data.values())
-            )
+            metadata["unique_authors"] = len(set(b["author"] for b in blame_data.values()))
 
         # Index the file content
-        doc_ids = self.vector_store.add_file(
-            str(file_path), chunk_size=1000, chunk_overlap=200, metadata=metadata
-        )
+        doc_ids = self.vector_store.add_file(str(file_path), chunk_size=1000, chunk_overlap=200, metadata=metadata)
         results["files_indexed"] += 1
 
         # Perform AST analysis for supported languages
@@ -309,9 +295,7 @@ class GitIngester:
             except Exception as e:
                 logger.warning(f"AST analysis failed for {file_path}: {e}")
 
-    def _get_file_history(
-        self, repo_path: Path, file_path: Path
-    ) -> List[Dict[str, Any]]:
+    def _get_file_history(self, repo_path: Path, file_path: Path) -> List[Dict[str, Any]]:
         """Get commit history for a specific file."""
         result = subprocess.run(
             [
@@ -346,9 +330,7 @@ class GitIngester:
 
         return history
 
-    def _get_file_blame(
-        self, repo_path: Path, file_path: Path
-    ) -> Dict[int, Dict[str, Any]]:
+    def _get_file_blame(self, repo_path: Path, file_path: Path) -> Dict[int, Dict[str, Any]]:
         """Get blame information for a file."""
         result = subprocess.run(
             ["git", "blame", "--line-porcelain", "--", str(file_path)],
@@ -448,9 +430,7 @@ Message: {commit.message}
             text=True,
         )
 
-        remote_url = (
-            remote_result.stdout.strip() if remote_result.returncode == 0 else None
-        )
+        remote_url = remote_result.stdout.strip() if remote_result.returncode == 0 else None
 
         # Create repository summary document
         repo_doc = f"""Repository: {repo_path.name}

hanzo_mcp/tools/vector/index_tool.py

@@ -151,13 +151,9 @@ Usage:
             if not force:
                 stats = await vector_store.get_stats()
                 if stats and stats.get("document_count", 0) > 0:
-                    await tool_ctx.info(
-                        "Project already indexed, use --force to re-index"
-                    )
+                    await tool_ctx.info("Project already indexed, use --force to re-index")
                     if show_stats:
-                        return self._format_stats(
-                            stats, abs_path, time.time() - start_time
-                        )
+                        return self._format_stats(stats, abs_path, time.time() - start_time)
                     return "Project is already indexed. Use --force to re-index."
 
             # Prepare file patterns
@@ -268,9 +264,7 @@ Usage:
                     except Exception as e:
                         errors.append(f"{file_path}: {str(e)}")
 
-            await tool_ctx.info(
-                f"Indexed {indexed_files} files ({total_size / 1024 / 1024:.1f} MB)"
-            )
+            await tool_ctx.info(f"Indexed {indexed_files} files ({total_size / 1024 / 1024:.1f} MB)")
 
             # Index git history if requested
             git_stats = {}

hanzo_mcp/tools/vector/infinity_store.py

@@ -11,9 +11,10 @@ try:
 
     INFINITY_AVAILABLE = True
 except ImportError:
-    # infinity_embedded not available, functionality will be limited
-    INFINITY_AVAILABLE = False
-    infinity_embedded = None  # type: ignore
+    # Use mock implementation when infinity_embedded is not available
+    from . import mock_infinity as infinity_embedded
+
+    INFINITY_AVAILABLE = True  # Mock is always available
 
 from .ast_analyzer import Symbol, FileAST, ASTAnalyzer, create_symbol_embedding_text
 
@@ -78,9 +79,7 @@ class InfinityVectorStore:
             dimension: Vector dimension (must match embedding model)
         """
         if not INFINITY_AVAILABLE:
-            raise ImportError(
-                "infinity_embedded is required for vector store functionality"
-            )
+            raise ImportError("infinity_embedded is required for vector store functionality")
 
         # Set up data path
         if data_path:
@@ -171,17 +170,13 @@ class InfinityVectorStore:
                     "source_file": {"type": "varchar"},
                     "target_file": {"type": "varchar"},
                     "symbol_name": {"type": "varchar"},
-                    "reference_type": {
-                        "type": "varchar"
-                    },  # import, call, inheritance, etc.
+                    "reference_type": {"type": "varchar"},  # import, call, inheritance, etc.
                     "line_number": {"type": "integer"},
                     "metadata": {"type": "varchar"},  # JSON string
                 },
             )
 
-    def _generate_doc_id(
-        self, content: str, file_path: str = "", chunk_index: int = 0
-    ) -> str:
+    def _generate_doc_id(self, content: str, file_path: str = "", chunk_index: int = 0) -> str:
         """Generate a unique document ID."""
         content_hash = hashlib.sha256(content.encode()).hexdigest()[:16]
         path_hash = hashlib.sha256(file_path.encode()).hexdigest()[:8]
@@ -533,11 +528,7 @@ class InfinityVectorStore:
             FileAST object if file found, None otherwise
         """
         try:
-            results = (
-                self.ast_table.output(["*"])
-                .filter(f"file_path = '{file_path}'")
-                .to_pl()
-            )
+            results = self.ast_table.output(["*"]).filter(f"file_path = '{file_path}'").to_pl()
 
             if len(results) == 0:
                 return None
@@ -576,11 +567,7 @@ class InfinityVectorStore:
             List of reference information
         """
         try:
-            results = (
-                self.references_table.output(["*"])
-                .filter(f"target_file = '{file_path}'")
-                .to_pl()
-            )
+            results = self.references_table.output(["*"]).filter(f"target_file = '{file_path}'").to_pl()
 
             references = []
             for row in results.iter_rows(named=True):
@@ -695,11 +682,7 @@ class InfinityVectorStore:
         """
         try:
             # Get count first
-            results = (
-                self.documents_table.output(["id"])
-                .filter(f"file_path = '{file_path}'")
-                .to_pl()
-            )
+            results = self.documents_table.output(["id"]).filter(f"file_path = '{file_path}'").to_pl()
             count = len(results)
 
             # Delete all documents for this file
@@ -725,9 +708,7 @@ class InfinityVectorStore:
                         metadata = json.loads(row["metadata"])
                         files[file_path] = {
                             "file_path": file_path,
-                            "file_name": metadata.get(
-                                "file_name", Path(file_path).name
-                            ),
+                            "file_name": metadata.get("file_name", Path(file_path).name),
                             "file_size": metadata.get("file_size", 0),
                             "total_chunks": metadata.get("total_chunks", 1),
                         }