From 1eb6b88d9126262b6c9987f72351c27ca1b4d27b Mon Sep 17 00:00:00 2001
From: Amogh <amoghrajesh1999@gmail.com>
Date: Thu, 14 Nov 2024 10:05:29 +0530
Subject: [PATCH] Remove ORM references from datamodels for assets

---
 airflow/api_fastapi/core_api/datamodels/assets.py      | 6 +++---
 airflow/api_fastapi/core_api/openapi/v1-generated.yaml | 6 +++---
 airflow/ui/openapi-gen/requests/schemas.gen.ts         | 9 +++------
 airflow/ui/openapi-gen/requests/types.gen.ts           | 6 +++---
 4 files changed, 12 insertions(+), 15 deletions(-)

diff --git a/airflow/api_fastapi/core_api/datamodels/assets.py b/airflow/api_fastapi/core_api/datamodels/assets.py
index 498dc4edae6fd..16258a8b9a05a 100644
--- a/airflow/api_fastapi/core_api/datamodels/assets.py
+++ b/airflow/api_fastapi/core_api/datamodels/assets.py
@@ -23,7 +23,7 @@
 
 
 class DagScheduleAssetReference(BaseModel):
-    """Serializable version of the DagScheduleAssetReference ORM SqlAlchemyModel."""
+    """DAG schedule reference serializer for assets."""
 
     dag_id: str
     created_at: datetime
@@ -31,7 +31,7 @@ class DagScheduleAssetReference(BaseModel):
 
 
 class TaskOutletAssetReference(BaseModel):
-    """Serializable version of the TaskOutletAssetReference ORM SqlAlchemyModel."""
+    """Task outlet reference serializer for assets."""
 
     dag_id: str
     task_id: str
@@ -40,7 +40,7 @@ class TaskOutletAssetReference(BaseModel):
 
 
 class AssetAliasSchema(BaseModel):
-    """Serializable version of the AssetAliasSchema ORM SqlAlchemyModel."""
+    """Asset alias serializer for assets."""
 
     id: int
     name: str
diff --git a/airflow/api_fastapi/core_api/openapi/v1-generated.yaml b/airflow/api_fastapi/core_api/openapi/v1-generated.yaml
index 6efa1e445856b..57bcd876c3e64 100644
--- a/airflow/api_fastapi/core_api/openapi/v1-generated.yaml
+++ b/airflow/api_fastapi/core_api/openapi/v1-generated.yaml
@@ -3550,7 +3550,7 @@ components:
       - id
       - name
       title: AssetAliasSchema
-      description: Serializable version of the AssetAliasSchema ORM SqlAlchemyModel.
+      description: Asset alias serializer for assets.
     AssetCollectionResponse:
       properties:
         assets:
@@ -4688,7 +4688,7 @@ components:
       - created_at
       - updated_at
       title: DagScheduleAssetReference
-      description: Serializable version of the DagScheduleAssetReference ORM SqlAlchemyModel.
+      description: DAG schedule reference serializer for assets.
     DagStatsCollectionResponse:
       properties:
         dags:
@@ -5588,7 +5588,7 @@ components:
       - created_at
       - updated_at
       title: TaskOutletAssetReference
-      description: Serializable version of the TaskOutletAssetReference ORM SqlAlchemyModel.
+      description: Task outlet reference serializer for assets.
     TaskResponse:
       properties:
         task_id:
diff --git a/airflow/ui/openapi-gen/requests/schemas.gen.ts b/airflow/ui/openapi-gen/requests/schemas.gen.ts
index b3339e78dad88..cb64ef7cbd524 100644
--- a/airflow/ui/openapi-gen/requests/schemas.gen.ts
+++ b/airflow/ui/openapi-gen/requests/schemas.gen.ts
@@ -103,8 +103,7 @@ export const $AssetAliasSchema = {
   type: "object",
   required: ["id", "name"],
   title: "AssetAliasSchema",
-  description:
-    "Serializable version of the AssetAliasSchema ORM SqlAlchemyModel.",
+  description: "Asset alias serializer for assets.",
 } as const;
 
 export const $AssetCollectionResponse = {
@@ -1849,8 +1848,7 @@ export const $DagScheduleAssetReference = {
   type: "object",
   required: ["dag_id", "created_at", "updated_at"],
   title: "DagScheduleAssetReference",
-  description:
-    "Serializable version of the DagScheduleAssetReference ORM SqlAlchemyModel.",
+  description: "DAG schedule reference serializer for assets.",
 } as const;
 
 export const $DagStatsCollectionResponse = {
@@ -3164,8 +3162,7 @@ export const $TaskOutletAssetReference = {
   type: "object",
   required: ["dag_id", "task_id", "created_at", "updated_at"],
   title: "TaskOutletAssetReference",
-  description:
-    "Serializable version of the TaskOutletAssetReference ORM SqlAlchemyModel.",
+  description: "Task outlet reference serializer for assets.",
 } as const;
 
 export const $TaskResponse = {
diff --git a/airflow/ui/openapi-gen/requests/types.gen.ts b/airflow/ui/openapi-gen/requests/types.gen.ts
index 6c036f76bf299..dd12f8b048b02 100644
--- a/airflow/ui/openapi-gen/requests/types.gen.ts
+++ b/airflow/ui/openapi-gen/requests/types.gen.ts
@@ -22,7 +22,7 @@ export type AppBuilderViewResponse = {
 };
 
 /**
- * Serializable version of the AssetAliasSchema ORM SqlAlchemyModel.
+ * Asset alias serializer for assets.
  */
 export type AssetAliasSchema = {
   id: number;
@@ -417,7 +417,7 @@ export type DagRunType =
   | "asset_triggered";
 
 /**
- * Serializable version of the DagScheduleAssetReference ORM SqlAlchemyModel.
+ * DAG schedule reference serializer for assets.
  */
 export type DagScheduleAssetReference = {
   dag_id: string;
@@ -769,7 +769,7 @@ export type TaskInstanceStateCount = {
 };
 
 /**
- * Serializable version of the TaskOutletAssetReference ORM SqlAlchemyModel.
+ * Task outlet reference serializer for assets.
  */
 export type TaskOutletAssetReference = {
   dag_id: string;