zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-18 21:57:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix cancel ingest task, it will be stilling running and show internal server error (#836) (#16945)

Browse Source
This commit is contained in:
Wang Qi
2026-07-15 18:33:51 +08:00
committed by GitHub
parent 3b26688d68
commit 2223a514de
3 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/db/services/document_service.py
View File

@@ -917,7 +917,7 @@ class DocumentService(CommonService):
info["run"] = TaskStatus.RUNNING.value
# keep the doc in DONE state when keep_progress=True for GraphRAG, RAPTOR and Mindmap tasks
cls.update_by_id(doc_id, info)
cls.model.update(info).where((cls.model.id == doc_id) & ((cls.model.run.is_null(True)) | (cls.model.run != TaskStatus.CANCEL.value))).execute()
@classmethod
@DB.connection_context()

4
api/db/services/task_service.py
View File

@@ -421,7 +421,9 @@ class TaskService(CommonService):
doc_info = {"progress": -1, "run": TaskStatus.FAIL.value, "update_time": current_timestamp(), "update_date": get_format_time()}
if info.get("progress_msg"):
doc_info["progress_msg"] = trim_header_by_lines((task.progress_msg or "") + "\n" + info["progress_msg"], TASK_MAX_LOG_LENGTH)
DocumentService.update_by_id(task.doc_id, doc_info)
DocumentService.model.update(doc_info).where(
(DocumentService.model.id == task.doc_id) & ((DocumentService.model.run.is_null(True)) | (DocumentService.model.run != TaskStatus.CANCEL.value))
).execute()
@classmethod
@DB.connection_context()