diff --git a/task_sdk/src/airflow/sdk/api/datamodels/_generated.py b/task_sdk/src/airflow/sdk/api/datamodels/_generated.py index 5a103e78fc0ff..37062480865a5 100644 --- a/task_sdk/src/airflow/sdk/api/datamodels/_generated.py +++ b/task_sdk/src/airflow/sdk/api/datamodels/_generated.py @@ -144,6 +144,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 +156,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 +201,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