pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor:improve the logic to check cancel (#8524)

Browse Source 
### What problem does this PR solve?

improve the logic to check cancel

### Type of change

- [x] Refactoring

---------

Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
This commit is contained in:
Stephen Hu
2025-06-27 10:22:53 +08:00
committed by GitHub
parent 938d8dd878
commit be712714af
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rag/svr/task_executor.py
View File

@ -46,7 +46,7 @@ import faulthandler
import numpy as np import numpy as np
from peewee import DoesNotExist from peewee import DoesNotExist
from api.db import LLMType, ParserType, TaskStatus from api.db import LLMType, ParserType
from api.db.services.document_service import DocumentService from api.db.services.document_service import DocumentService
from api.db.services.llm_service import LLMBundle from api.db.services.llm_service import LLMBundle
from api.db.services.task_service import TaskService from api.db.services.task_service import TaskService
@ -213,8 +213,7 @@ async def collect():
canceled = False canceled = False
task = TaskService.get_task(msg["id"]) task = TaskService.get_task(msg["id"])
if task: if task:
_, doc = DocumentService.get_by_id(task["doc_id"]) canceled = TaskService.do_cancel(task["id"])
canceled = doc.run == TaskStatus.CANCEL.value or doc.progress < 0
if not task or canceled: if not task or canceled:
state = "is unknown" if not task else "has been cancelled" state = "is unknown" if not task else "has been cancelled"
FAILED_TASKS += 1 FAILED_TASKS += 1