diff --git a/airflow/api_connexion/openapi/v1.yaml b/airflow/api_connexion/openapi/v1.yaml
index f8f7e986215c8..afd878de74f6f 100644
--- a/airflow/api_connexion/openapi/v1.yaml
+++ b/airflow/api_connexion/openapi/v1.yaml
@@ -890,7 +890,9 @@ paths:
           content:
             application/json:
               schema:
-                $ref: '#/components/schemas/DAGRun'
+                anyOf:
+                  - $ref: '#/components/schemas/DAGRun'
+                  - $ref: '#/components/schemas/TaskInstanceCollection'
         '400':
           $ref: '#/components/responses/BadRequest'
         '401':
diff --git a/airflow/www/static/js/types/api-generated.ts b/airflow/www/static/js/types/api-generated.ts
index 2380e7fef5cba..f28b5c8a15ae1 100644
--- a/airflow/www/static/js/types/api-generated.ts
+++ b/airflow/www/static/js/types/api-generated.ts
@@ -3002,7 +3002,8 @@ export interface operations {
       /** Success. */
       200: {
         content: {
-          "application/json": components["schemas"]["DAGRun"];
+          "application/json": Partial<components["schemas"]["DAGRun"]> &
+            Partial<components["schemas"]["TaskInstanceCollection"]>;
         };
       };
       400: components["responses"]["BadRequest"];