From 8f8fe215c16f4ca521d3635184f3cb4cc4207f2f Mon Sep 17 00:00:00 2001 From: jx2lee Date: Wed, 18 Dec 2024 00:45:22 +0900 Subject: [PATCH] generated datamodel-codegen --- task_sdk/src/airflow/sdk/api/datamodels/_generated.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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