zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 15:31:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix document progress not set to fail when embedding model error (#16381)

Browse Source
This commit is contained in:
Wang Qi
2026-06-26 16:11:54 +08:00
committed by GitHub
parent 8081a77c7c
commit 985e3c1db5
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -28,7 +28,7 @@ from api.db.db_models import Task, Document, Knowledgebase, Tenant
from api.db.services.common_service import CommonService
from api.db.services.document_service import DocumentService
from common.misc_utils import get_uuid
from common.time_utils import current_timestamp
from common.time_utils import current_timestamp, get_format_time
from common.constants import StatusEnum, TaskStatus, MAXIMUM_PAGE_NUMBER, MAXIMUM_TASK_PAGE_NUMBER
from deepdoc.parser.excel_parser import RAGFlowExcelParser
from rag.utils.redis_conn import REDIS_CONN
@@ -138,6 +138,7 @@ class TaskService(CommonService):
).where(cls.model.id == docs[0]["id"]).execute()
if docs[0]["retry_count"] >= 3:
DocumentService.update_by_id(docs[0]["doc_id"], {"progress": -1, "run": TaskStatus.FAIL.value, "update_time": current_timestamp(), "update_date": get_format_time()})
return None
return docs[0]
@@ -347,6 +348,11 @@ class TaskService(CommonService):
if begin_at is not None:
process_duration = (datetime.now() - begin_at).total_seconds()
cls.model.update(process_duration=process_duration).where(cls.model.id == id).execute()
if info.get("progress") == -1:
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)
@classmethod
@DB.connection_context()