Procházet zdrojové kódy

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 před 4 měsíci
rodič
revize
be712714af
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 2 přidání a 3 odebrání
  1. 2
    3
      rag/svr/task_executor.py

+ 2
- 3
rag/svr/task_executor.py Zobrazit soubor

@@ -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

Načítá se…
Zrušit
Uložit