Implement AI orchestration wedding demo
This commit is contained in:
15
backend/app/ai/tool_registry.py
Normal file
15
backend/app/ai/tool_registry.py
Normal file
@@ -0,0 +1,15 @@
|
||||
from app.domain.schemas import ToolDefinition
|
||||
|
||||
|
||||
class ToolRegistry:
|
||||
def __init__(self, tools: list[ToolDefinition]) -> None:
|
||||
self._tools = {tool.name: tool for tool in tools}
|
||||
|
||||
def list(self) -> list[ToolDefinition]:
|
||||
return list(self._tools.values())
|
||||
|
||||
def get(self, name: str) -> ToolDefinition | None:
|
||||
return self._tools.get(name)
|
||||
|
||||
def names(self) -> set[str]:
|
||||
return set(self._tools)
|
||||
Reference in New Issue
Block a user