blob: 8aad82afaeebc5d42d00dbc30a4f7adecdc26f3c (
plain)
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
|
from abc import ABC, abstractmethod
from dataclasses import dataclass
class ToolError(Exception):
pass
@dataclass
class ToolResult:
content: str
class Tool(ABC):
@property
@abstractmethod
def name(self) -> str: ...
@property
@abstractmethod
def description(self) -> str: ...
@property
@abstractmethod
def schema(self) -> dict[str, object]: ...
@abstractmethod
def execute(self, arguments: dict[str, object]) -> ToolResult: ...
class ToolRegistry:
def __init__(self) -> None:
self.tools: dict[str, Tool] = {}
def register(self, tool: Tool) -> None:
self.tools[tool.name] = tool
def get(self, name: str) -> Tool | None:
return self.tools.get(name)
def all(self) -> list[Tool]:
return list(self.tools.values())
def schemas(self) -> list[dict[str, object]]:
return [
{
"type": "function",
"function": {
"name": tool.name,
"description": tool.description,
"parameters": tool.schema,
},
}
for tool in self.tools.values()
]
|
