generated datamodel-codegen

apache · Dec 18, 2024 · 8f8fe21 · 8f8fe21
1 parent 7e95d16
commit 8f8fe21
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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