Browse Source

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 months ago
parent
commit
be712714af
No account linked to committer's email address
1 changed files with 2 additions and 3 deletions
  1. 2
    3
      rag/svr/task_executor.py

+ 2
- 3
rag/svr/task_executor.py View File

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
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 = doc.run == TaskStatus.CANCEL.value or doc.progress < 0
canceled = TaskService.do_cancel(task["id"])
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

Loading…
Cancel
Save