estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

deal db session in celery worker (#17549)

Browse Source
This commit is contained in:
Jyong
2025-04-07 20:31:26 +08:00
committed by GitHub
parent da7f8ad936
commit c8145ce581
26 changed files with 104 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/tasks/add_document_to_index_task.py
View File

@@ -4,7 +4,6 @@ import time
import click
from celery import shared_task # type: ignore
from werkzeug.exceptions import NotFound
from core.rag.index_processor.constant.index_type import IndexType
from core.rag.index_processor.index_processor_factory import IndexProcessorFactory
@@ -28,7 +27,9 @@ def add_document_to_index_task(dataset_document_id: str):
dataset_document = db.session.query(DatasetDocument).filter(DatasetDocument.id == dataset_document_id).first()
if not dataset_document:
raise NotFound("Document not found")
logging.info(click.style("Document not found: {}".format(dataset_document_id), fg="red"))
db.session.close()
return
if dataset_document.indexing_status != "completed":
return