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

Post fix of #23224 (#25007)

Browse source
This commit is contained in:
Yongtao Huang 2025-09-02 20:59:08 +08:00 • committed by GitHub
commit 8fcc864fb7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/core/rag/retrieval/dataset_retrieval.py
View file

@ -534,7 +534,6 @@ class DatasetRetrieval:
synchronize_session=False, synchronize_session=False,
) )
) )
db.session.commit()
else: else:
query = db.session.query(DocumentSegment).where( query = db.session.query(DocumentSegment).where(
DocumentSegment.index_node_id == document.metadata["doc_id"] DocumentSegment.index_node_id == document.metadata["doc_id"]
@ -596,7 +595,7 @@ class DatasetRetrieval:
with flask_app.app_context(): with flask_app.app_context():
with Session(db.engine) as session: with Session(db.engine) as session:
dataset_stmt = select(Dataset).where(Dataset.id == dataset_id) dataset_stmt = select(Dataset).where(Dataset.id == dataset_id)
dataset = db.session.scalar(dataset_stmt) dataset = session.scalar(dataset_stmt)
if not dataset: if not dataset:
return [] return []