generated datamodel-codegen

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