hanzo-mcp 0.8.15__py3-none-any.whl → 0.9.1__py3-none-any.whl

hanzo_mcp/tools/database/sql.py

@@ -15,6 +15,8 @@ from typing import (
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.permissions import PermissionManager
 from hanzo_mcp.tools.database.database_manager import DatabaseManager
@@ -100,6 +102,9 @@ sql --action stats --table users
 """
 
     @override
+    @auto_timeout("sql")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/database/sql_query.py

@@ -6,6 +6,7 @@ from typing import Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -89,6 +90,7 @@ Examples:
 Note: Use sql_search for text search operations."""
 
     @override
+    @auto_timeout("sql_query")
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/database/sql_search.py

@@ -6,6 +6,8 @@ from typing import Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -107,6 +109,9 @@ Examples:
 Use sql_query for complex queries with joins, conditions, etc."""
 
     @override
+    @auto_timeout("sql_search")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/database/sql_stats.py

@@ -6,6 +6,8 @@ from typing import Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -75,6 +77,9 @@ Examples:
 """
 
     @override
+    @auto_timeout("sql_stats")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/editor/neovim_command.py

@@ -9,6 +9,8 @@ from typing import List, Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -128,6 +130,9 @@ Note: Requires Neovim to be installed.
 """
 
     @override
+    @auto_timeout("neovim_command")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/editor/neovim_edit.py

@@ -8,6 +8,8 @@ from typing import Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -125,6 +127,9 @@ Note: Requires Neovim to be installed and available in PATH.
 """
 
     @override
+    @auto_timeout("neovim_edit")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/editor/neovim_session.py

@@ -11,6 +11,8 @@ from datetime import datetime
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 
@@ -114,6 +116,9 @@ Note: Requires Neovim to be installed.
 """
 
     @override
+    @auto_timeout("neovim_session")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/ast_tool.py

@@ -15,6 +15,7 @@ from grep_ast.grep_ast import TreeContext
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 
 Pattern = Annotated[
     str,
@@ -97,6 +98,7 @@ ast "def test_" ./tests
 Searches code structure intelligently, understanding syntax and providing semantic context."""
 
     @override
+    @auto_timeout("ast")
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/batch_search.py

@@ -21,6 +21,8 @@ from typing_extensions import Unpack, Annotated, TypedDict, final, override
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -539,6 +541,9 @@ Perfect for comprehensive code analysis and refactoring tasks."""
         return all_results
 
     @override
+    @auto_timeout("batch_search")
+
+
     async def call(self, ctx: MCPContext, **params: Unpack[BatchSearchParams]) -> str:
         """Execute batch search with multiple queries in parallel."""
         import time

hanzo_mcp/tools/filesystem/content_replace.py

@@ -9,6 +9,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -101,6 +103,9 @@ Can be run in dry-run mode to preview changes without applying them.
 Only works within allowed directories."""
 
     @override
+    @auto_timeout("content_replace")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/diff.py

@@ -5,6 +5,8 @@ from typing import override
 from pathlib import Path
 
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
@@ -200,6 +202,9 @@ diff a.json b.json --ignore-whitespace"""
                 show_line_numbers=show_line_numbers,
             )
 
+    @auto_timeout("diff")
+
+
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/filesystem/directory_tree.py

@@ -8,6 +8,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.truncate import truncate_response
@@ -85,6 +87,9 @@ indented list for readability. By default, common development directories like
 requested. Only works within allowed directories."""
 
     @override
+    @auto_timeout("directory_tree")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/directory_tree_paginated.py

@@ -19,6 +19,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -116,6 +118,9 @@ Use the cursor field to continue from where the previous request left off.
 Returns nextCursor if more entries are available."""
 
     @override
+    @auto_timeout("directory_tree_paginated")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/edit.py

@@ -9,6 +9,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -88,6 +90,9 @@ Usage:
 - ALWAYS prefer editing existing files in the codebase. NEVER write new files unless explicitly required."""
 
     @override
+    @auto_timeout("edit")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/find.py

@@ -23,6 +23,8 @@ from pathlib import Path
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
 
 # Parameter types
@@ -147,6 +149,9 @@ Fast, intuitive file content search."""
         return self._available_backends
 
     @override
+    @auto_timeout("find")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/find_files.py

@@ -6,6 +6,8 @@ from typing import Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -138,6 +140,9 @@ For database search, use 'sql_search' or 'vector_search'.
 """
 
     @override
+    @auto_timeout("find_files")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/git_search.py

@@ -8,6 +8,8 @@ from typing import Unpack, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -153,6 +155,9 @@ Examples:
 """
 
     @override
+    @auto_timeout("git_search")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/grep.py

@@ -14,6 +14,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.context import ToolContext
@@ -353,6 +355,9 @@ When you are doing an open ended search that may require multiple rounds of glob
             return f"Error searching file contents: {str(e)}"
 
     @override
+    @auto_timeout("grep")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/multi_edit.py

@@ -9,6 +9,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -132,6 +134,9 @@ If you want to create a new file, use:
 - Subsequent edits: normal edit operations on the created content"""
 
     @override
+    @auto_timeout("multi_edit")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/read.py

@@ -8,6 +8,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.truncate import truncate_response
@@ -91,6 +93,9 @@ Usage:
 - When reading multiple files, you MUST use the batch tool to read them all at once"""
 
     @override
+    @auto_timeout("read")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/rules_tool.py

@@ -9,6 +9,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -74,6 +76,9 @@ The tool returns the contents of all found configuration files to help
 understand project-specific requirements and preferences."""
 
     @override
+    @auto_timeout("rules")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/search_tool.py

@@ -27,6 +27,8 @@ from dataclasses import dataclass
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -528,6 +530,9 @@ This is the recommended search tool for comprehensive results."""
         return results
 
     @override
+    @auto_timeout("search")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/symbols_tool.py

@@ -23,6 +23,8 @@ from pydantic import Field
 from grep_ast.grep_ast import TreeContext
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
 
 # Parameter types
@@ -126,6 +128,9 @@ symbols --action list --path ./src --symbol-type class
 Finds code structures (functions, classes, methods) with full context."""
 
     @override
+    @auto_timeout("symbols")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/tree.py

@@ -9,6 +9,8 @@ from pathlib import Path
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
 
 # Parameter types
@@ -95,6 +97,9 @@ tree --dirs-only
 tree --pattern "*.py" --show-size"""
 
     @override
+    @auto_timeout("tree")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/filesystem/watch.py

@@ -7,6 +7,8 @@ from pathlib import Path
 from datetime import datetime
 
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
@@ -41,6 +43,9 @@ class WatchTool(BaseTool):
                 duration=duration,
             )
 
+    @auto_timeout("watch")
+
+
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/filesystem/write.py

@@ -8,6 +8,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.filesystem.base import FilesystemBaseTool
@@ -72,6 +74,9 @@ Usage:
 - NEVER proactively create documentation files (*.md) or README files. Only create documentation files if explicitly requested by the User."""
 
     @override
+    @auto_timeout("write")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/jupyter/jupyter.py

@@ -16,6 +16,8 @@ import nbformat
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.jupyter.base import JupyterBaseTool
 
 # Parameter types
@@ -111,6 +113,9 @@ jupyter --action create "new.ipynb"
 """
 
     @override
+    @auto_timeout("jupyter")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/jupyter/notebook_edit.py

@@ -9,6 +9,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.jupyter.base import JupyterBaseTool
@@ -96,6 +98,9 @@ class NoteBookEditTool(JupyterBaseTool):
         return "Completely replaces the contents of a specific cell in a Jupyter notebook (.ipynb file) with new source. Jupyter notebooks are interactive documents that combine code, text, and visualizations, commonly used for data analysis and scientific computing. The notebook_path parameter must be an absolute path, not a relative path. The cell_number is 0-indexed. Use edit_mode=insert to add a new cell at the index specified by cell_number. Use edit_mode=delete to delete the cell at the index specified by cell_number."
 
     @override
+    @auto_timeout("notebook_edit")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/jupyter/notebook_read.py

@@ -9,6 +9,8 @@ from pathlib import Path
 
 from pydantic import Field
 from mcp.server import FastMCP
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.jupyter.base import JupyterBaseTool
@@ -56,6 +58,9 @@ class NotebookReadTool(JupyterBaseTool):
         return "Reads a Jupyter notebook (.ipynb file) and returns all of the cells with their outputs. Jupyter notebooks are interactive documents that combine code, text, and visualizations, commonly used for data analysis and scientific computing. The notebook_path parameter must be an absolute path, not a relative path."
 
     @override
+    @auto_timeout("notebook_read")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/llm/consensus_tool.py

@@ -15,6 +15,7 @@ from typing import (
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.llm.llm_tool import LLMTool
 from hanzo_mcp.tools.common.context import create_tool_context, ToolContext
@@ -155,6 +156,7 @@ The tool will:
 """
 
     @override
+    @auto_timeout("consensus")
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/llm/llm_manage.py

@@ -7,6 +7,8 @@ from pathlib import Path
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.llm.llm_tool import LLMTool
 from hanzo_mcp.tools.common.context import create_tool_context
@@ -110,6 +112,9 @@ Providers are automatically detected based on environment variables:
 """
 
     @override
+    @auto_timeout("llm_manage")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/llm/llm_tool.py

@@ -21,6 +21,7 @@ from mcp.server.fastmcp import Context as MCPContext
 
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 
 # Check if litellm is available
 try:
@@ -263,6 +264,7 @@ llm --action models --provider openai
 Available: {", ".join(available) if available else "None"}"""
 
     @override
+    @auto_timeout("llm")
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/llm/llm_unified.py

@@ -19,6 +19,8 @@ from pathlib import Path
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.common.context import create_tool_context
 
@@ -263,6 +265,9 @@ llm --action models --provider openai
 Available: {", ".join(available) if available else "None"}"""
 
     @override
+    @auto_timeout("llm_unified")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/llm/provider_tools.py

@@ -5,6 +5,8 @@ from typing import Dict, Unpack, Optional, Annotated, TypedDict, final, override
 from pydantic import Field
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
+
 from hanzo_mcp.tools.common.base import BaseTool
 from hanzo_mcp.tools.llm.llm_tool import LLMTool
 
@@ -98,6 +100,9 @@ class BaseProviderTool(BaseTool):
         return model
 
     @override
+    @auto_timeout("provider_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/lsp/lsp_tool.py

@@ -1,3 +1,5 @@
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 """Language Server Protocol (LSP) tool for code intelligence.
 
 This tool provides on-demand LSP configuration and installation for various
@@ -472,6 +474,9 @@ class LSPTool(BaseTool):
 
         return MCPResourceDocument(data=result)
 
+    @auto_timeout("lsp")
+
+
     async def call(self, **kwargs) -> str:
         """Tool interface for MCP - converts result to JSON string."""
         result = await self.run(**kwargs)