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>
100 lines
2.3 KiB
Python
100 lines
2.3 KiB
Python
"""Tools module for X-Agents.
|
|
|
|
This module provides tool infrastructure for the agent system.
|
|
It wraps and extends the nanobot tool implementation.
|
|
"""
|
|
|
|
from nanobot.agent.tools.base import Tool
|
|
from nanobot.agent.tools.registry import ToolRegistry
|
|
|
|
from agents.tools.builtin import (
|
|
get_builtin_tools,
|
|
ReadFileTool,
|
|
WriteFileTool,
|
|
ListDirectoryTool,
|
|
SearchTool,
|
|
WebSearchTool,
|
|
CalculatorTool,
|
|
GetTimeTool,
|
|
BashTool,
|
|
)
|
|
from agents.tools.manager import ToolManager
|
|
|
|
|
|
def create_default_registry(use_sandbox: bool = False) -> ToolRegistry:
|
|
"""Create a tool registry with default tools.
|
|
|
|
Args:
|
|
use_sandbox: Whether to use sandbox for shell execution
|
|
|
|
Returns:
|
|
Tool registry with built-in tools
|
|
"""
|
|
registry = ToolRegistry()
|
|
# Register built-in tools
|
|
for tool in get_builtin_tools(use_sandbox=use_sandbox):
|
|
registry.register(tool)
|
|
return registry
|
|
|
|
|
|
# Import sandbox tools from nanobot (optional)
|
|
try:
|
|
from nanobot.agent.tools.sandbox_execution import (
|
|
SandboxType,
|
|
SandboxCodeExecutionTool,
|
|
SandboxBashTool,
|
|
get_sandbox_tools,
|
|
)
|
|
from nanobot.agent.tools.bwrap_sandbox import (
|
|
BwrapSandbox,
|
|
get_bwrap_sandbox,
|
|
execute_in_bwrap,
|
|
)
|
|
from nanobot.agent.tools.gvisor_sandbox import (
|
|
GvisorSandbox,
|
|
get_gvisor_sandbox,
|
|
execute_in_gvisor,
|
|
)
|
|
SANDBOX_AVAILABLE = True
|
|
except ImportError as e:
|
|
SandboxType = None
|
|
SandboxCodeExecutionTool = None
|
|
SandboxBashTool = None
|
|
get_sandbox_tools = None
|
|
BwrapSandbox = None
|
|
get_bwrap_sandbox = None
|
|
execute_in_bwrap = None
|
|
GvisorSandbox = None
|
|
get_gvisor_sandbox = None
|
|
execute_in_gvisor = None
|
|
SANDBOX_AVAILABLE = False
|
|
|
|
|
|
__all__ = [
|
|
"Tool",
|
|
"ToolRegistry",
|
|
"ToolManager",
|
|
"create_default_registry",
|
|
"get_builtin_tools",
|
|
"ReadFileTool",
|
|
"WriteFileTool",
|
|
"ListDirectoryTool",
|
|
"SearchTool",
|
|
"WebSearchTool",
|
|
"CalculatorTool",
|
|
"GetTimeTool",
|
|
"BashTool",
|
|
# Sandbox tools
|
|
"SANDBOX_AVAILABLE",
|
|
"SandboxType",
|
|
"SandboxCodeExecutionTool",
|
|
"SandboxBashTool",
|
|
"get_sandbox_tools",
|
|
"BwrapSandbox",
|
|
"GvisorSandbox",
|
|
"get_bwrap_sandbox",
|
|
"get_gvisor_sandbox",
|
|
"execute_in_bwrap",
|
|
"execute_in_gvisor",
|
|
]
|