From 93e3751eb056711ed58584835aab8fcbe66fce45 Mon Sep 17 00:00:00 2001
From: water <672684719@qq.com>
Date: Mon, 24 Aug 2026 16:52:06 +0800
Subject: [PATCH] fix(db_cleanup): remove invalid dag_id column for
 task_reschedule

TaskReschedule table was migrated in Airflow 3.0 (migration 0063)
to use ti_id (FK to task_instance.id) instead of dag_id/task_id/run_id.
The dag_id column was dropped. db_cleanup.py still referenced it,
causing "column base.dag_id does not exist" errors when running
`airflow db clean` with --dag-id filter.

Fixes #72020
---
 airflow-core/src/airflow/utils/db_cleanup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/airflow-core/src/airflow/utils/db_cleanup.py b/airflow-core/src/airflow/utils/db_cleanup.py
index 1a79dbc9e30ce..9c153ab2967cf 100644
--- a/airflow-core/src/airflow/utils/db_cleanup.py
+++ b/airflow-core/src/airflow/utils/db_cleanup.py
@@ -202,7 +202,7 @@ def readable_config(self):
         recency_column_name="expires_at",
         dag_id_column_name="dag_id",
     ),
-    _TableConfig(table_name="task_reschedule", recency_column_name="start_date", dag_id_column_name="dag_id"),
+    _TableConfig(table_name="task_reschedule", recency_column_name="start_date"),
     _TableConfig(table_name="xcom", recency_column_name="timestamp", dag_id_column_name="dag_id"),
     _TableConfig(table_name="_xcom_archive", recency_column_name="timestamp", dag_id_column_name="dag_id"),
     _TableConfig(table_name="callback_request", recency_column_name="created_at"),