Skip to content

ClientTool

autogen.beta.tools.final.client_tool.ClientTool #

ClientTool(schema)

Bases: Tool

Source code in autogen/beta/tools/final/client_tool.py 
21
22
def __init__(self, schema: dict[str, Any]) -> None:
    self.schema = FunctionToolSchema.from_dict(schema)

schema instance-attribute #

schema = from_dict(schema)

schemas async #

schemas(context)
Source code in autogen/beta/tools/final/client_tool.py 
24
25
async def schemas(self, context: "Context") -> list[FunctionToolSchema]:
    return [self.schema]

register #

register(stack, context, *, middleware=())
Source code in autogen/beta/tools/final/client_tool.py 
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
def register(
    self,
    stack: "ExitStack",
    context: "Context",
    *,
    middleware: Iterable["BaseMiddleware"] = (),
) -> None:
    execution: ToolExecution = self
    for mw in middleware:
        execution = partial(mw.on_tool_execution, execution)

    async def execute(event: "ToolCall", context: "Context") -> None:
        result = await execution(event, context)
        await context.send(result)

    stack.enter_context(
        context.stream.where((ToolCall.name == self.schema.function.name) & ClientToolCall.not_()).sub_scope(
            execute
        ),
    )