瀏覽代碼

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

### 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>
tags/v0.20.0
Stephen Hu 4 月之前
父節點
當前提交
be712714af
No account linked to committer's email address
共有 1 個文件被更改,包括 2 次插入3 次删除
  1. 2
    3
      rag/svr/task_executor.py

+ 2
- 3
rag/svr/task_executor.py 查看文件

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

Loading…
取消
儲存