From add33cb5e646da37a48cc7e9bf798a8dda5fcf58 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 16 Jan 2024 00:59:28 +0800 Subject: [PATCH] fix SQL slow query (#2043) Co-authored-by: jyong --- api/core/tool/dataset_multi_retriever_tool.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/core/tool/dataset_multi_retriever_tool.py b/api/core/tool/dataset_multi_retriever_tool.py index 97cf7c9dfa..c9ca7eb04a 100644 --- a/api/core/tool/dataset_multi_retriever_tool.py +++ b/api/core/tool/dataset_multi_retriever_tool.py @@ -94,6 +94,7 @@ class DatasetMultiRetrieverTool(BaseTool): document_context_list = [] index_node_ids = [document.metadata['doc_id'] for document in all_documents] segments = DocumentSegment.query.filter( + DocumentSegment.dataset_id.in_(self.dataset_ids), DocumentSegment.completed_at.isnot(None), DocumentSegment.status == 'completed', DocumentSegment.enabled == True,