diff --git a/task_sdk/src/airflow/sdk/api/datamodels/_generated.py b/task_sdk/src/airflow/sdk/api/datamodels/_generated.py index 5a103e78fc0ff..d85199e822a32 100644 --- a/task_sdk/src/airflow/sdk/api/datamodels/_generated.py +++ b/task_sdk/src/airflow/sdk/api/datamodels/_generated.py @@ -17,7 +17,7 @@ # generated by datamodel-codegen: # filename: http://0.0.0.0:9091/execution/openapi.json -# version: 0.26.3 +# version: 0.26.4 from __future__ import annotations @@ -26,7 +26,7 @@ from typing import Annotated, Any, Literal from uuid import UUID -from pydantic import BaseModel, ConfigDict, Field +from pydantic import BaseModel, ConfigDict, Field, RootModel class ConnectionResponse(BaseModel): @@ -69,6 +69,10 @@ class IntermediateTIState(str, Enum): DEFERRED = "deferred" +class JsonValue(RootModel[Any]): + root: Any + + class TIDeferredStatePayload(BaseModel): """ Schema for updating TaskInstance to a deferred state. @@ -144,6 +148,9 @@ class VariableResponse(BaseModel): Variable schema for responses with fields that are needed for Runtime. """ + model_config = ConfigDict( + extra="forbid", + ) key: Annotated[str, Field(title="Key")] value: Annotated[str | None, Field(title="Value")] = None @@ -153,6 +160,9 @@ class XComResponse(BaseModel): XCom schema for responses with fields that are needed for Runtime. """ + model_config = ConfigDict( + extra="forbid", + ) key: Annotated[str, Field(title="Key")] value: Annotated[Any, Field(title="Value")] @@ -195,6 +205,9 @@ class TIRunContext(BaseModel): Response schema for TaskInstance run context. """ + model_config = ConfigDict( + extra="forbid", + ) dag_run: DagRun variables: Annotated[list[VariableResponse] | None, Field(title="Variables")] = None connections: Annotated[list[ConnectionResponse] | None, Field(title="Connections")] = None