diff --git a/task_sdk/src/airflow/sdk/api/datamodels/_generated.py b/task_sdk/src/airflow/sdk/api/datamodels/_generated.py index 5a103e78fc0ff..2e891e00eb102 100644 --- a/task_sdk/src/airflow/sdk/api/datamodels/_generated.py +++ b/task_sdk/src/airflow/sdk/api/datamodels/_generated.py @@ -34,6 +34,8 @@ class ConnectionResponse(BaseModel): Connection schema for responses with fields that are needed for Runtime. """ + model_config = ConfigDict(extra="forbid") + conn_id: Annotated[str, Field(title="Conn Id")] conn_type: Annotated[str, Field(title="Conn Type")] host: Annotated[str | None, Field(title="Host")] = None @@ -144,6 +146,8 @@ 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 +157,8 @@ 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,8 @@ 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