From 8fcc864fb764b8c86c7789c5fbbb6ae47c481527 Mon Sep 17 00:00:00 2001 From: Yongtao Huang Date: Tue, 2 Sep 2025 20:59:08 +0800 Subject: [PATCH] Post fix of #23224 (#25007) --- api/core/rag/retrieval/dataset_retrieval.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py index 11010c9d6..1c1019e18 100644 --- a/api/core/rag/retrieval/dataset_retrieval.py +++ b/api/core/rag/retrieval/dataset_retrieval.py @@ -534,7 +534,6 @@ class DatasetRetrieval: synchronize_session=False, ) ) - db.session.commit() else: query = db.session.query(DocumentSegment).where( DocumentSegment.index_node_id == document.metadata["doc_id"] @@ -596,7 +595,7 @@ class DatasetRetrieval: with flask_app.app_context(): with Session(db.engine) as session: dataset_stmt = select(Dataset).where(Dataset.id == dataset_id) - dataset = db.session.scalar(dataset_stmt) + dataset = session.scalar(dataset_stmt) if not dataset: return []