Skip to content

Toolkit

autogen.beta.tools.final.toolkit.Toolkit #

Toolkit(*tools, name=None, middleware=())

Bases: Tool

Source code in autogen/beta/tools/final/toolkit.py 
27
28
29
30
31
32
33
34
35
36
37
38
def __init__(
    self,
    *tools: Tool | Callable[..., Any],
    name: str | None = None,
    middleware: Iterable[ToolMiddleware] = (),
) -> None:
    self._middleware: tuple[ToolMiddleware, ...] = tuple(middleware)
    self.name = name or self.__class__.__name__

    self._tools: dict[str, Tool] = {}
    for t in tools:
        self._add_tool(t)

name instance-attribute #

name = name or __name__

tools property #

tools

set_provider #

set_provider(provider)
Source code in autogen/beta/tools/final/toolkit.py 
44
45
46
def set_provider(self, provider: Provider) -> None:
    for t in self.tools:
        t.set_provider(provider)

tool #

tool(function: Callable[..., Any], *, name: str | None = None, description: str | None = None, schema: FunctionParameters | None = None, sync_to_thread: bool = True, middleware: Iterable[ToolMiddleware] = ()) -> FunctionTool

tool(function: None = None, *, name: str | None = None, description: str | None = None, schema: FunctionParameters | None = None, sync_to_thread: bool = True, middleware: Iterable[ToolMiddleware] = ()) -> Callable[[Callable[..., Any]], FunctionTool]

tool(function=None, *, name=None, description=None, schema=None, sync_to_thread=True, middleware=())
Source code in autogen/beta/tools/final/toolkit.py 
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
def tool(
    self,
    function: Callable[..., Any] | None = None,
    *,
    name: str | None = None,
    description: str | None = None,
    schema: FunctionParameters | None = None,
    sync_to_thread: bool = True,
    middleware: Iterable[ToolMiddleware] = (),
) -> FunctionTool | Callable[[Callable[..., Any]], FunctionTool]:
    def make_tool(f: Callable[..., Any]) -> FunctionTool:
        t = tool(
            f,
            name=name,
            description=description,
            schema=schema,
            sync_to_thread=sync_to_thread,
            middleware=middleware,
        )
        self._add_tool(t)
        return t

    if function:
        return make_tool(function)

    return make_tool

schemas async #

schemas(context)
Source code in autogen/beta/tools/final/toolkit.py 
116
117
118
119
120
async def schemas(self, context: "Context") -> Iterable[ToolSchema]:
    schemas: list[ToolSchema] = []
    for t in self.tools:
        schemas.extend(await t.schemas(context))
    return schemas

register #

register(stack, context, *, middleware=())
Source code in autogen/beta/tools/final/toolkit.py 
122
123
124
125
126
127
128
129
130
def register(
    self,
    stack: "ExitStack",
    context: "Context",
    *,
    middleware: Iterable["BaseMiddleware"] = (),
) -> None:
    for t in self.tools:
        t.register(stack, context, middleware=middleware)