DESKTOP-72TV0V4\caoxiaozhu
1afa88e812
- 新增 loop.py Agent 运行循环 - 优化 memory.py 记忆模块 - 扩展 api/routes.py 接口 - 更新 tools 模块：builtin.py, manager.py, __init__.py - 新增 .env.example 配置示例 - 更新 requirements.txt 依赖 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
111 lines
2.9 KiB
Python
111 lines
2.9 KiB
Python
"""Tool manager for loading and managing tools."""
|
|
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from nanobot.agent.tools.registry import ToolRegistry
|
|
|
|
from agents.tools.builtin import get_builtin_tools
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ToolManager:
|
|
"""Manages tools for the agent."""
|
|
|
|
def __init__(self, workspace: Path | None = None, use_sandbox: bool = False):
|
|
"""Initialize tool manager.
|
|
|
|
Args:
|
|
workspace: Optional workspace path
|
|
use_sandbox: Whether to use sandbox for shell execution (recommended for untrusted code)
|
|
"""
|
|
self.workspace = workspace
|
|
self.use_sandbox = use_sandbox
|
|
self.registry = ToolRegistry()
|
|
self._load_builtin_tools()
|
|
|
|
def _load_builtin_tools(self) -> None:
|
|
"""Load all built-in tools."""
|
|
tools = get_builtin_tools(self.workspace, use_sandbox=self.use_sandbox)
|
|
for tool in tools:
|
|
self.registry.register(tool)
|
|
logger.info(f"Loaded {len(tools)} built-in tools (sandbox: {self.use_sandbox})")
|
|
|
|
def register_tool(self, tool: Any) -> None:
|
|
"""Register a custom tool.
|
|
|
|
Args:
|
|
tool: Tool instance to register
|
|
"""
|
|
self.registry.register(tool)
|
|
logger.info(f"Registered tool: {tool.name}")
|
|
|
|
def unregister_tool(self, name: str) -> None:
|
|
"""Unregister a tool.
|
|
|
|
Args:
|
|
name: Tool name to unregister
|
|
"""
|
|
self.registry.unregister(name)
|
|
logger.info(f"Unregistered tool: {name}")
|
|
|
|
def get_tool(self, name: str) -> Any:
|
|
"""Get a tool by name.
|
|
|
|
Args:
|
|
name: Tool name
|
|
|
|
Returns:
|
|
Tool instance or None
|
|
"""
|
|
return self.registry.get(name)
|
|
|
|
def has_tool(self, name: str) -> bool:
|
|
"""Check if a tool is registered.
|
|
|
|
Args:
|
|
name: Tool name
|
|
|
|
Returns:
|
|
True if tool exists
|
|
"""
|
|
return self.registry.has(name)
|
|
|
|
def list_tools(self) -> list[str]:
|
|
"""List all registered tool names.
|
|
|
|
Returns:
|
|
List of tool names
|
|
"""
|
|
return self.registry.tool_names
|
|
|
|
def get_tool_definitions(self) -> list[dict[str, Any]]:
|
|
"""Get all tool definitions in OpenAI format.
|
|
|
|
Returns:
|
|
List of tool schemas
|
|
"""
|
|
return self.registry.get_definitions()
|
|
|
|
async def execute_tool(self, name: str, params: dict[str, Any]) -> str:
|
|
"""Execute a tool by name.
|
|
|
|
Args:
|
|
name: Tool name
|
|
params: Tool parameters
|
|
|
|
Returns:
|
|
Tool execution result
|
|
"""
|
|
return await self.registry.execute(name, params)
|
|
|
|
def __len__(self) -> int:
|
|
"""Get number of registered tools."""
|
|
return len(self.registry)
|
|
|
|
def __contains__(self, name: str) -> bool:
|
|
"""Check if tool is registered."""
|
|
return name in self.registry
|