airflow.providers.common.ai.toolsets.logging¶
Logging wrapper toolset for pydantic-ai tool calls.
Classes¶
Wrap a toolset to log each tool call with timing. |
Module Contents¶
- class airflow.providers.common.ai.toolsets.logging.LoggingToolset[source]¶
Bases:
pydantic_ai.toolsets.wrapper.WrapperToolset[Any]Wrap a toolset to log each tool call with timing.
- logger: airflow.sdk.types.Logger | logging.Logger[source]¶
- async call_tool(name, tool_args, ctx, tool)[source]¶
Call a tool with the given arguments.
- Args:
name: The name of the tool to call. tool_args: The arguments to pass to the tool. ctx: The run context. tool: The tool definition returned by [get_tools][pydantic_ai.toolsets.AbstractToolset.get_tools] that was called.