airflow.providers.common.ai.utils.tool_metrics¶

Count calls to the toolsets this provider ships, by toolset, agent framework and outcome.

Attributes¶

Framework

Functions¶

`calling_framework`(name)	Attribute the tool calls made inside the block to agent framework `name`.
`current_framework`()	Return the framework the current tool call is attributed to, if an adapter set one.
`record_tool_call`(toolset, outcome)	Count one call to a toolset.

Module Contents¶

airflow.providers.common.ai.utils.tool_metrics.Framework[source]¶

airflow.providers.common.ai.utils.tool_metrics.calling_framework(name)[source]¶

Attribute the tool calls made inside the block to agent framework name.

airflow.providers.common.ai.utils.tool_metrics.current_framework()[source]¶

Return the framework the current tool call is attributed to, if an adapter set one.

airflow.providers.common.ai.utils.tool_metrics.record_tool_call(toolset, outcome)[source]¶

Count one call to a toolset.

Tags stay low-cardinality: the toolset class, the framework and the outcome, never arguments, connection IDs, table names or paths. A call made outside any adapter is a Pydantic AI agent’s, such as AgentOperator’s.