mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-27 04:20:09 +00:00
zeekay
a3b58c011f
- Fix import sorting (I001) in all packages - Fix bare except (E722) in lsp_tool.py - Fix duplicate set item (B033) in refactor_tool.py
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
"""Todo tools for Hanzo MCP.
|
|
|
|
This package provides task management tools:
|
|
- TodoTool: Unified todo management with add, update, remove, list, clear operations
|
|
"""
|
|
|
|
from mcp.server import FastMCP
|
|
|
|
from hanzo_tools.core import BaseTool, ToolRegistry
|
|
from hanzo_tools.todo.base import TodoStorage, TodoBaseTool
|
|
from hanzo_tools.todo.todo_tool import TodoTool
|
|
|
|
__all__ = [
|
|
"TodoTool",
|
|
"TodoStorage",
|
|
"TodoBaseTool",
|
|
"get_todo_tools",
|
|
"register_todo_tools",
|
|
"TOOLS",
|
|
]
|
|
|
|
|
|
def get_todo_tools() -> list[BaseTool]:
|
|
"""Create instances of all todo tools.
|
|
|
|
Returns:
|
|
List of todo tool instances
|
|
"""
|
|
return [TodoTool()]
|
|
|
|
|
|
def register_todo_tools(
|
|
mcp_server: FastMCP,
|
|
enabled_tools: dict[str, bool] | None = None,
|
|
) -> list[BaseTool]:
|
|
"""Register todo tools with the MCP server.
|
|
|
|
Args:
|
|
mcp_server: The FastMCP server instance
|
|
enabled_tools: Dictionary of individual tool enable states (default: None)
|
|
|
|
Returns:
|
|
List of registered tools
|
|
"""
|
|
tools = get_todo_tools()
|
|
ToolRegistry.register_tools(mcp_server, tools)
|
|
return tools
|
|
|
|
|
|
# TOOLS list for entry point discovery
|
|
TOOLS = [TodoTool]
|