fix SQL slow query (#2043)

Co-authored-by: jyong <jyong@dify.ai>
2 years ago · add33cb5e6
parent 83105d0d8f
commit add33cb5e6
1 changed files with 1 additions and 0 deletions
--- 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,