airflow.providers.common.ai.example_dags.example_agent

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

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.

Module Contents

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.

Was this entry helpful?