jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit c8145ce581
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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