airflow.providers.common.ai.example_dags.example_agent¶

Example DAGs demonstrating AgentOperator, @task.agent, and toolsets.

Attributes¶

SQLToolset

Classes¶

Analysis

Structured analysis output for the agent example.

Functions¶

`example_agent_operator_sql`()
`example_agent_operator_hook`()
`example_agent_decorator`()
`example_agent_structured_output`()
`example_agent_chain`()
`example_agent_operator_hitl_review`()	AgentOperator with HITL review — a human approves output via hitl-review plugin UI.
`example_agent_operator_code_mode`()
`example_agent_session`()	Resume a conversation across runs via `message_history`.

Module Contents¶

airflow.providers.common.ai.example_dags.example_agent.SQLToolset = None[source]¶

class airflow.providers.common.ai.example_dags.example_agent.Analysis(/, **data)[source]¶

Bases: pydantic.BaseModel

Structured analysis output for the agent example.

summary: str[source]¶

top_items: list[str][source]¶

row_count: int[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_operator_sql()[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_operator_hook()[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_decorator()[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_structured_output()[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_chain()[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_operator_hitl_review()[source]¶: AgentOperator with HITL review — a human approves output via hitl-review plugin UI.

airflow.providers.common.ai.example_dags.example_agent.example_agent_operator_code_mode()[source]¶

airflow.providers.common.ai.example_dags.example_agent.example_agent_session()[source]¶

Resume a conversation across runs via message_history.

The agent step seeds itself with the prior transcript and re-emits the updated transcript to XCom (key message_history). Loading and storing that transcript under a session key is the DAG’s job – here, a JSON file in object storage keyed by session_id. Swap the path for s3:// / gs:// in a deployment.