56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""Beaver 工具注册表。
|
||
|
||
这层只做三件事：
|
||
1. 注册工具
|
||
2. 按名称查找工具
|
||
3. 导出 provider 可消费的 tool schemas
|
||
|
||
不要把执行逻辑塞进这里。
|
||
执行属于 runtime/executor，那样边界更清晰。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Iterable
|
||
|
||
from beaver.tools.base import BaseTool, ToolSpec
|
||
|
||
|
||
class ToolRegistry:
|
||
"""统一维护当前 runtime 可用的工具集合。"""
|
||
|
||
def __init__(self) -> None:
|
||
self._tools: dict[str, BaseTool] = {}
|
||
|
||
def register(self, tool: BaseTool, *, replace: bool = False) -> None:
|
||
"""注册一个工具。
|
||
|
||
默认不允许重名覆盖，避免 loader/runtime 不小心把同名工具静默冲掉。
|
||
"""
|
||
|
||
name = tool.spec.name
|
||
if not replace and name in self._tools:
|
||
raise ValueError(f"Tool '{name}' is already registered")
|
||
self._tools[name] = tool
|
||
|
||
def register_many(self, tools: Iterable[BaseTool], *, replace: bool = False) -> None:
|
||
for tool in tools:
|
||
self.register(tool, replace=replace)
|
||
|
||
def get(self, name: str) -> BaseTool | None:
|
||
return self._tools.get(name)
|
||
|
||
def require(self, name: str) -> BaseTool:
|
||
tool = self.get(name)
|
||
if tool is None:
|
||
raise KeyError(f"Unknown tool '{name}'")
|
||
return tool
|
||
|
||
def list_specs(self) -> list[ToolSpec]:
|
||
return [tool.spec for tool in self._tools.values()]
|
||
|
||
def export_provider_schemas(self) -> list[dict]:
|
||
"""导出给 provider 的函数工具 schema 列表。"""
|
||
|
||
return [spec.to_provider_schema() for spec in self.list_specs()]
|