20 lines
463 B
Python
20 lines
463 B
Python
|
from typing import Dict, List
|
||
|
|
|
||
|
|
from app.tools.base import ToolDefinition
|
||
|
|
|
||
|
|
|
||
|
|
class ToolRegistry:
|
||
|
|
_tools: Dict[str, ToolDefinition] = {}
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def register(cls, definition: ToolDefinition) -> None:
|
||
|
|
cls._tools[definition.name] = definition
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get(cls, name: str) -> ToolDefinition:
|
||
|
|
return cls._tools[name]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def list(cls) -> List[ToolDefinition]:
|
||
|
|
return list(cls._tools.values())
|