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

hanzo_mcp/tools/mcp/mcp_add.py

@@ -8,6 +8,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
 
@@ -131,6 +133,9 @@ Use 'mcp_stats' to see all added servers and their status.
 """
 
     @override
+    @auto_timeout("mcp_add")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/mcp/mcp_remove.py

@@ -5,6 +5,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.mcp.mcp_add import McpAddTool
 from hanzo_mcp.tools.common.context import create_tool_context
@@ -64,6 +66,9 @@ Use 'mcp_stats' to see all servers before removing.
 """
 
     @override
+    @auto_timeout("mcp_remove")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/mcp/mcp_stats.py

@@ -4,6 +4,8 @@ from typing import Unpack, TypedDict, final, override
 
 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.mcp.mcp_add import McpAddTool
 from hanzo_mcp.tools.common.context import create_tool_context
@@ -47,6 +49,9 @@ Example:
 """
 
     @override
+    @auto_timeout("mcp_stats")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/mcp/mcp_tool.py

@@ -20,6 +20,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
 
@@ -270,6 +272,9 @@ mcp --action enable --name github
 Status: {enabled} enabled, {running} running"""
 
     @override
+    @auto_timeout("mcp")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/memory/knowledge_tools.py

@@ -7,6 +7,8 @@ supporting hierarchical organization (session, project, global).
 from typing import Any, Dict, List, Optional, final, override
 
 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
@@ -69,6 +71,9 @@ recall_facts(queries=["company policies"], scope="global", limit=5)
 """
 
     @override
+    @auto_timeout("knowledge_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,
@@ -190,6 +195,9 @@ store_facts(facts=["Company founded in 2020"], scope="global", kb_name="company_
 """
 
     @override
+    @auto_timeout("knowledge_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,
@@ -286,6 +294,9 @@ summarize_to_memory(content="Company guidelines...", topic="Guidelines", scope="
 """
 
     @override
+    @auto_timeout("knowledge_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,
@@ -390,6 +401,9 @@ manage_knowledge_bases(action="delete", kb_name="old_docs")
 """
 
     @override
+    @auto_timeout("knowledge_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/memory/memory_tools.py

@@ -7,6 +7,8 @@ The hanzo-memory package provides embedded database and vector search capabiliti
 from typing import Dict, List, Optional, final, override
 
 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
@@ -69,6 +71,9 @@ recall_memories(queries=["coding standards"], scope="global")
 """
 
     @override
+    @auto_timeout("memory_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,
@@ -155,6 +160,9 @@ create_memories(statements=["User prefers dark mode", "User works in Python"])
 """
 
     @override
+    @auto_timeout("memory_tools")
+
+
     async def call(self, ctx: MCPContext, statements: List[str]) -> str:
         """Create new memories.
 
@@ -219,6 +227,9 @@ update_memories(updates=[
 """
 
     @override
+    @auto_timeout("memory_tools")
+
+
     async def call(self, ctx: MCPContext, updates: List[Dict[str, str]]) -> str:
         """Update memories.
 
@@ -283,6 +294,9 @@ delete_memories(ids=["mem_1", "mem_2"])
 """
 
     @override
+    @auto_timeout("memory_tools")
+
+
     async def call(self, ctx: MCPContext, ids: List[str]) -> str:
         """Delete memories.
 
@@ -345,6 +359,9 @@ manage_memories(
 """
 
     @override
+    @auto_timeout("memory_tools")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/search/find_tool.py

@@ -11,6 +11,7 @@ from dataclasses import dataclass
 
 from hanzo_mcp.types import MCPResourceDocument
 from hanzo_mcp.tools.common.base import BaseTool
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 
 # Check if ffind command is available
 try:
@@ -375,7 +376,8 @@ class FindTool(BaseTool):
             }
         )
 
-    async def call(self, **kwargs) -> str:
+    @auto_timeout("find")
+    async def call(self, ctx=None, **kwargs) -> str:
         """Tool interface for MCP - converts result to JSON string."""
         result = await self.run(**kwargs)
         return result.to_json_string()

hanzo_mcp/tools/search/unified_search.py

@@ -14,6 +14,7 @@ from dataclasses import dataclass
 
 from hanzo_mcp.types import MCPResourceDocument
 from hanzo_mcp.tools.common.base import BaseTool
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 
 # Import memory tools if available
 try:
@@ -349,7 +350,8 @@ class UnifiedSearch(BaseTool):
 
         return MCPResourceDocument(data=response_data)
 
-    async def call(self, **kwargs) -> str:
+    @auto_timeout("search")
+    async def call(self, ctx=None, **kwargs) -> str:
         """Tool interface for MCP - converts result to JSON string."""
         result = await self.run(**kwargs)
         return result.to_json_string()

hanzo_mcp/tools/shell/base_process.py

@@ -183,11 +183,13 @@ class BaseProcessTool(BaseTool):
         else:
             if output.startswith("Command failed"):
                 raise RuntimeError(output)
+            # Get configurable max tokens from environment, default to 25000
+            max_tokens = int(os.environ.get("HANZO_MCP_MAX_RESPONSE_TOKENS", "25000"))
             # Truncate output to prevent token limit issues
             return truncate_response(
                 output,
-                max_tokens=25000,
-                truncation_message="\n\n[Command output truncated due to token limit. Output may be available in logs or files.]",
+                max_tokens=max_tokens,
+                truncation_message=f"\n\n[Command output truncated due to {max_tokens} token limit. Output may be available in logs or files. Set HANZO_MCP_MAX_RESPONSE_TOKENS env var to adjust limit.]",
             )
 
     async def execute_background(

hanzo_mcp/tools/shell/bash_tool.py

@@ -8,6 +8,7 @@ from pathlib import Path
 from mcp.server import FastMCP
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from hanzo_mcp.tools.shell.base_process import BaseScriptTool
 
 
@@ -30,6 +31,7 @@ class BashTool(BaseScriptTool):
         ) -> str:
             return await tool_self.run(ctx, command=command, cwd=cwd, env=env, timeout=timeout)
 
+    @auto_timeout("bash")
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/shell/logs.py

@@ -6,6 +6,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
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -105,6 +107,9 @@ Use run_command with 'tail -f' for continuous monitoring.
 """
 
     @override
+    @auto_timeout("logs")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/npx.py

@@ -7,6 +7,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
@@ -103,6 +105,9 @@ For long-running servers, use npx_background instead.
 """
 
     @override
+    @auto_timeout("npx")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/npx_background.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
@@ -120,6 +122,9 @@ Use 'processes' to list running processes and 'pkill' to stop them.
 """
 
     @override
+    @auto_timeout("npx")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/npx_tool.py

@@ -4,6 +4,8 @@ from typing import Optional, 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.shell.base_process import BaseBinaryTool
@@ -88,6 +90,9 @@ npx json-server db.json  # Auto-backgrounds if needed"""
         ) -> str:
             return await tool_self.run(ctx, package=package, args=args, cwd=cwd, yes=yes)
 
+    @auto_timeout("npx")
+
+
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/shell/open.py

@@ -8,6 +8,8 @@ from pathlib import Path
 from urllib.parse import urlparse
 
 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
@@ -26,6 +28,9 @@ class OpenTool(BaseTool):
         async def open(path: str, ctx: MCPContext) -> str:
             return await tool_self.run(ctx, path)
 
+    @auto_timeout("open")
+
+
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(ctx, params["path"])

hanzo_mcp/tools/shell/pkill.py

@@ -7,6 +7,8 @@ import psutil
 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
@@ -106,6 +108,9 @@ Examples:
 """
 
     @override
+    @auto_timeout("pkill")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/process_tool.py

@@ -4,6 +4,8 @@ import signal
 from typing import Optional, override
 
 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
@@ -132,6 +134,9 @@ process --action logs --id bash_ghi789 --lines 50"""
         ) -> str:
             return await tool_self.run(ctx, action=action, id=id, signal_type=signal_type, lines=lines)
 
+    @auto_timeout("process")
+
+
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/shell/processes.py

@@ -7,6 +7,8 @@ import psutil
 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
@@ -90,6 +92,9 @@ Examples:
 """
 
     @override
+    @auto_timeout("processes")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/run_background.py

@@ -10,6 +10,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
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -181,6 +183,9 @@ Examples:
 """
 
     @override
+    @auto_timeout("run")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/run_command.py

@@ -9,6 +9,7 @@ from pydantic import Field
 from mcp.server import FastMCP
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from hanzo_mcp.tools.shell.base import ShellBaseTool
 from hanzo_mcp.tools.common.base import handle_connection_errors
 from hanzo_mcp.tools.common.context import create_tool_context
@@ -269,6 +270,7 @@ Important:
         return tool_ctx
 
     @override
+    @auto_timeout("run_command")
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/run_command_windows.py

@@ -8,6 +8,8 @@ from typing import Any, Annotated, 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.shell.base import ShellBaseTool
@@ -194,6 +196,9 @@ Important:
         return tool_ctx
 
     @override
+    @auto_timeout("run_command_windows")
+
+
     async def call(self, ctx: MCPContext, **params: Any) -> str:
         """Execute the tool with the given parameters.
 

hanzo_mcp/tools/shell/streaming_command.py

@@ -1,3 +1,5 @@
+
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 """Streaming command execution with disk-based logging and session management."""
 
 import os
@@ -173,6 +175,9 @@ class StreamingCommandTool(BaseProcessTool):
 
         return None
 
+    @auto_timeout("streaming_command")
+
+
     async def call(self, ctx: Any, **kwargs) -> Dict[str, Any]:
         """MCP tool entry point.
 

hanzo_mcp/tools/shell/uvx.py

@@ -7,6 +7,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
@@ -100,6 +102,9 @@ For long-running servers, use uvx_background instead.
 """
 
     @override
+    @auto_timeout("uvx")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/uvx_background.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
@@ -119,6 +121,9 @@ Use 'processes' to list running processes and 'pkill' to stop them.
 """
 
     @override
+    @auto_timeout("uvx")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/shell/uvx_tool.py

@@ -4,6 +4,8 @@ from typing import Optional, 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.shell.base_process import BaseBinaryTool
@@ -88,6 +90,9 @@ uvx jupyter lab --port 8888  # Auto-backgrounds if needed"""
         ) -> str:
             return await tool_self.run(ctx, package=package, args=args, cwd=cwd, python=python)
 
+    @auto_timeout("uvx")
+
+
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/shell/zsh_tool.py

@@ -9,6 +9,7 @@ from pathlib import Path
 from mcp.server import FastMCP
 from mcp.server.fastmcp import Context as MCPContext
 
+from hanzo_mcp.tools.common.auto_timeout import auto_timeout
 from hanzo_mcp.tools.shell.base_process import BaseScriptTool
 
 
@@ -31,6 +32,7 @@ class ZshTool(BaseScriptTool):
         ) -> str:
             return await tool_self.run(ctx, command=command, cwd=cwd, env=env, timeout=timeout)
 
+    @auto_timeout("zsh")
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(
@@ -171,6 +173,7 @@ class ShellTool(BaseScriptTool):
         ) -> str:
             return await tool_self.run(ctx, command=command, cwd=cwd, env=env, timeout=timeout)
 
+    @auto_timeout("shell")
     async def call(self, ctx: MCPContext, **params) -> str:
         """Call the tool with arguments."""
         return await self.run(

hanzo_mcp/tools/todo/todo.py

@@ -16,6 +16,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.todo.base import TodoBaseTool
 
 # Parameter types
@@ -104,6 +106,9 @@ todo --filter in_progress
 """
 
     @override
+    @auto_timeout("todo")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/todo/todo_read.py

@@ -8,6 +8,8 @@ from typing 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.todo.base import TodoStorage, TodoBaseTool
@@ -66,6 +68,9 @@ Usage:
 - If no todos exist yet for the session, an empty list will be returned"""
 
     @override
+    @auto_timeout("todo_read")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/todo/todo_write.py

@@ -7,6 +7,8 @@ from typing import Unpack, Literal, 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.todo.base import TodoStorage, TodoBaseTool
@@ -254,6 +256,9 @@ The assistant did not use the todo list because this is a single command executi
 When in doubt, use this tool. Being proactive with task management demonstrates attentiveness and ensures you complete all requirements successfully."""
 
     @override
+    @auto_timeout("todo_write")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/vector/index_tool.py

@@ -8,6 +8,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
 from hanzo_mcp.tools.common.permissions import PermissionManager
@@ -105,6 +107,9 @@ Usage:
 - index --force: Force re-indexing of all files"""
 
     @override
+    @auto_timeout("index")
+
+
     async def call(
         self,
         ctx: MCPContext,

hanzo_mcp/tools/vector/vector.py

@@ -15,6 +15,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.permissions import PermissionManager
 from hanzo_mcp.tools.vector.project_manager import ProjectVectorManager
@@ -131,6 +133,9 @@ vector --action clear --path ./old_code
 """
 
     @override
+    @auto_timeout("vector")
+
+
     async def call(
         self,
         ctx: MCPContext,