浏览代码

fix SQL slow query (#2043)

Co-authored-by: jyong <jyong@dify.ai>
Jyong 1 年之前
父节点
当前提交
add33cb5e6
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      api/core/tool/dataset_multi_retriever_tool.py

+ 1 - 0
api/core/tool/dataset_multi_retriever_tool.py

@@ -94,6 +94,7 @@ class DatasetMultiRetrieverTool(BaseTool):
         document_context_list = []
         document_context_list = []
         index_node_ids = [document.metadata['doc_id'] for document in all_documents]
         index_node_ids = [document.metadata['doc_id'] for document in all_documents]
         segments = DocumentSegment.query.filter(
         segments = DocumentSegment.query.filter(
+            DocumentSegment.dataset_id.in_(self.dataset_ids),
             DocumentSegment.completed_at.isnot(None),
             DocumentSegment.completed_at.isnot(None),
             DocumentSegment.status == 'completed',
             DocumentSegment.status == 'completed',
             DocumentSegment.enabled == True,
             DocumentSegment.enabled == True,