diff --git a/airflow-core/src/airflow/api_fastapi/execution_api/datamodels/taskinstance.py b/airflow-core/src/airflow/api_fastapi/execution_api/datamodels/taskinstance.py index 7fa329e0f32b5..f519d246d98ac 100644 --- a/airflow-core/src/airflow/api_fastapi/execution_api/datamodels/taskinstance.py +++ b/airflow-core/src/airflow/api_fastapi/execution_api/datamodels/taskinstance.py @@ -41,7 +41,7 @@ TaskInstanceState as TIState, TerminalTIState, ) -from airflow.utils.types import DagRunType +from airflow.utils.types import DagRunTriggeredByType, DagRunType AwareDatetimeAdapter = TypeAdapter(AwareDatetime) @@ -300,6 +300,8 @@ class DagRun(StrictBaseModel): state: DagRunState conf: Annotated[dict[str, Any], Field(default_factory=dict)] consumed_asset_events: list[AssetEventDagRunReference] + triggered_by: DagRunTriggeredByType | None = None + triggering_user_name: str | None = None class TIRunContext(BaseModel): diff --git a/task-sdk/src/airflow/sdk/api/datamodels/_generated.py b/task-sdk/src/airflow/sdk/api/datamodels/_generated.py index 8e4c4207b2742..c46dd54c0e93c 100644 --- a/task-sdk/src/airflow/sdk/api/datamodels/_generated.py +++ b/task-sdk/src/airflow/sdk/api/datamodels/_generated.py @@ -143,6 +143,21 @@ class DagRunStateResponse(BaseModel): state: DagRunState +class DagRunTriggeredByType(str, Enum): + """ + Class with DagRun triggered by types. + """ + + ASSET = "asset" + BACKFILL = "backfill" + CLI = "cli" + OPERATOR = "operator" + REST_API = "rest_api" + TEST = "test" + TIMETABLE = "timetable" + UI = "ui" + + class DagRunType(str, Enum): """ Class with DagRun types. @@ -540,6 +555,8 @@ class DagRun(BaseModel): state: DagRunState conf: Annotated[dict[str, Any] | None, Field(title="Conf")] = None consumed_asset_events: Annotated[list[AssetEventDagRunReference], Field(title="Consumed Asset Events")] + triggered_by: Annotated[DagRunTriggeredByType | None, Field(title="Triggered By")] = None + triggering_user_name: Annotated[str | None, Field(title="Triggering User Name")] = None class HITLDetailRequest(BaseModel): @@ -583,9 +600,9 @@ class TIRunContext(BaseModel): max_tries: Annotated[int, Field(title="Max Tries")] variables: Annotated[list[VariableResponse] | None, Field(title="Variables")] = None connections: Annotated[list[ConnectionResponse] | None, Field(title="Connections")] = None - upstream_map_indexes: Annotated[ - dict[str, int | list[int] | None] | None, Field(title="Upstream Map Indexes") - ] = None + upstream_map_indexes: Annotated[dict[str, int | list[int] | None] | None, Field(title="Upstream Map Indexes")] = ( + None + ) next_method: Annotated[str | None, Field(title="Next Method")] = None next_kwargs: Annotated[dict[str, Any] | str | None, Field(title="Next Kwargs")] = None xcom_keys_to_clear: Annotated[list[str] | None, Field(title="Xcom Keys To Clear")] = None