airflow.providers.common.ai.example_dags.example_agent_capabilities

Example DAGs demonstrating pydantic-ai capabilities via agent_params.

Capabilities (https://ai.pydantic.dev/capabilities/) are pydantic-ai’s composable units for thinking, web search, image generation, MCP, and more. AgentOperator forwards anything in agent_params to the underlying Agent(...) constructor, so capabilities work today without operator-level support. A first-class capabilities= kwarg is on the roadmap.

Functions

example_agent_capabilities_thinking()

example_agent_capabilities_web_search()

example_agent_capabilities_composed()

Module Contents

airflow.providers.common.ai.example_dags.example_agent_capabilities.example_agent_capabilities_thinking()[source]
airflow.providers.common.ai.example_dags.example_agent_capabilities.example_agent_capabilities_composed()[source]

Was this entry helpful?