mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-27 04:20:09 +00:00
zeekay
259d437061
Fix 21 auto-fixable errors: import sorting (I001), unused import (F401), and empty f-strings (F541) in 13 files across tool packages.
84 lines
2.1 KiB
Python
84 lines
2.1 KiB
Python
"""Agent tools for Hanzo AI.
|
|
|
|
Provides consolidated agent tools:
|
|
- agent: Multi-agent orchestration (run, dag, swarm, consensus, dispatch)
|
|
- zen: Hanzo Zen guidance for decisions
|
|
- review: Code review tool
|
|
|
|
Install:
|
|
pip install hanzo-tools-agent
|
|
|
|
Usage:
|
|
from hanzo_tools.agent import register_tools, TOOLS, AgentTool
|
|
|
|
# Register with MCP server
|
|
register_tools(mcp_server)
|
|
|
|
Consensus: https://github.com/luxfi/consensus
|
|
"""
|
|
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Core tools
|
|
from .zen_tool import ZenTool
|
|
from .agent_tool import AgentTool
|
|
|
|
# Legacy imports for backwards compatibility
|
|
from .critic_tool import CriticTool # Legacy - use reasoning.critic instead
|
|
from .review_tool import ReviewTool
|
|
from .grok_cli_tool import GrokCLITool
|
|
from .cli_agent_base import CLIAgentBase
|
|
from .codex_cli_tool import CodexCLITool
|
|
from .claude_cli_tool import ClaudeCLITool
|
|
from .gemini_cli_tool import GeminiCLITool
|
|
|
|
# Export list for tool discovery
|
|
TOOLS = [
|
|
AgentTool, # Multi-agent orchestration
|
|
ZenTool, # Hanzo Zen guidance
|
|
ReviewTool, # Code review
|
|
]
|
|
|
|
__all__ = [
|
|
"register_tools",
|
|
"TOOLS",
|
|
# Primary tools
|
|
"AgentTool",
|
|
"ZenTool",
|
|
"ReviewTool",
|
|
# Legacy (for backwards compatibility)
|
|
"CriticTool",
|
|
"CLIAgentBase",
|
|
"ClaudeCLITool",
|
|
"CodexCLITool",
|
|
"GeminiCLITool",
|
|
"GrokCLITool",
|
|
]
|
|
|
|
|
|
def register_tools(mcp_server, enabled_tools: dict[str, bool] | None = None):
|
|
"""Register agent tools with the MCP server."""
|
|
from hanzo_tools.core import ToolRegistry
|
|
|
|
enabled = enabled_tools or {}
|
|
registered = []
|
|
|
|
for tool_class in TOOLS:
|
|
tool_name = (
|
|
tool_class.name
|
|
if hasattr(tool_class, "name")
|
|
else tool_class.__name__.lower()
|
|
)
|
|
|
|
if enabled.get(tool_name, True):
|
|
try:
|
|
tool = tool_class()
|
|
ToolRegistry.register_tool(mcp_server, tool)
|
|
registered.append(tool)
|
|
except Exception as e:
|
|
logger.warning(f"Failed to register {tool_name}: {e}")
|
|
|
|
return registered
|