Skip to content

D
dify
Unverified Commit add33cb5 authored by Jyong's avatar Jyong Committed by GitHub
Browse files
Options

fix SQL slow query (#2043) 

Co-authored-by: 's avatarjyong <jyong@dify.ai>
parent 83105d0d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
api/core/tool/dataset_multi_retriever_tool.py
View file @ add33cb5
...@@ -94,6 +94,7 @@ class DatasetMultiRetrieverTool(BaseTool): ...@@ -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,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment