The Tool System

1class BaseTool(ABC):
2    """Base class for all tools"""
3
4    @property
5    def name(self) -> str:
6        """Tool name, e.g. 'Read', 'Bash'"""
7        ...
8
9    @property
10    def description(self) -> str:
11        """Tool description — how the LLM understands its capabilities"""
12        ...
13
14    def get_input_schema(self) -> dict:
15        """Parameter definition in JSON Schema format"""
16        ...
17
18    async def execute(self, params, context) -> ToolResult:
19        """Execute the tool, return the result"""
20        ...

1class ToolRegistry:
2    """Tool Registry — unified management for all tools"""
3
4    def register(self, tool: BaseTool):
5        """Register a new tool"""
6        self._tools[tool.name] = tool
7
8    def get_tool(self, name: str) -> BaseTool:
9        """Get a tool by name"""
10        return self._tools[name]
11
12    def get_tool_schemas(self) -> list[dict]:
13        """Get JSON Schemas for all tools (for the LLM to see)"""
14        return [tool.to_json_schema() for tool in self._tools.values()]