airflow.providers.common.ai.utils.logging

Logging utilities for pydantic-ai agent runs.

Functions

log_run_summary(logger, result)

Log model name, token usage, and tool call sequence from an agent run.

wrap_toolsets_for_logging(toolsets, logger)

Wrap each toolset in a LoggingToolset.

Module Contents

airflow.providers.common.ai.utils.logging.log_run_summary(logger, result)[source]

Log model name, token usage, and tool call sequence from an agent run.

airflow.providers.common.ai.utils.logging.wrap_toolsets_for_logging(toolsets, logger)[source]

Wrap each toolset in a LoggingToolset.

Was this entry helpful?