File size: 2,270 Bytes
068bc7f b64b6b0 068bc7f 359acf6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | """Tool registry for Stack 2.9 tools."""
from __future__ import annotations
from typing import Any
from .base import BaseTool
class ToolRegistry:
"""Singleton registry mapping tool names to tool instances."""
_instance: ToolRegistry | None = None
_tools: dict[str, BaseTool] = {}
def __new__(cls) -> ToolRegistry:
if cls._instance is None:
cls._instance = super().__new__(cls)
cls._tools = {}
return cls._instance
def register(self, tool: BaseTool) -> None:
"""Register a tool instance by name."""
if not tool.name:
raise ValueError("Tool must have a non-empty name")
self._tools[tool.name] = tool
def get(self, name: str) -> BaseTool | None:
"""Retrieve a registered tool by name."""
return self._tools.get(name)
def list(self) -> list[str]:
"""List all registered tool names."""
return list(self._tools.keys())
def list_tools(self) -> dict[str, dict[str, Any]]:
"""List all registered tools with their info.
Returns a dict mapping tool name to info dict with keys:
- name: str
- description: str
- input_schema: dict
"""
result = {}
for name, tool in self._tools.items():
schema = tool.input_schema
if callable(schema):
schema = schema()
result[name] = {
"name": tool.name,
"description": tool.description,
"input_schema": schema,
}
return result
def call(self, name: str, input_data: dict[str, Any]) -> Any:
"""Convenience: get tool and call it in one step."""
tool = self.get(name)
if tool is None:
raise KeyError(f"Tool not found: {name}")
return tool.call(input_data)
def unregister(self, name: str) -> bool:
"""Remove a tool from the registry. Returns True if it existed."""
if name in self._tools:
del self._tools[name]
return True
return False
def get_registry() -> ToolRegistry:
"""Get the global ToolRegistry instance."""
return ToolRegistry()
# Global registry instance
tool_registry = ToolRegistry()
|