From a316e6bc1b7cf92ed942c00edf38dc61063019ac Mon Sep 17 00:00:00 2001
From: yongtenglei <yongtengrey@outlook.com>
Date: Wed, 12 Nov 2025 11:49:03 +0800
Subject: [PATCH] ruff check

---
 rag/svr/task_executor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py
index 5f56646ef..af8dfc186 100644
--- a/rag/svr/task_executor.py
+++ b/rag/svr/task_executor.py
@@ -193,7 +193,7 @@ async def collect():
     canceled = False
     if msg.get("doc_id", "") in [GRAPH_RAPTOR_FAKE_DOC_ID, CANVAS_DEBUG_DOC_ID]:
         task = msg
-        if task["task_type"] in ["graphrag", "raptor", "mindmap"]:
+        if task["task_type"] in PIPELINE_SPECIAL_PROGRESS_FREEZE_TASK_TYPES:
             task = TaskService.get_task(msg["id"], msg["doc_ids"])
             if task:
                 task["doc_id"] = msg["doc_id"]