Skip to content

D
dify
Commit 18ca6d0f authored by jyong's avatar jyong
Browse files
Options

check thread block

parent d3622fb7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
api/core/tool/dataset_multi_retriever_tool.py
View file @ 18ca6d0f
...@@ -95,6 +95,7 @@ class DatasetMultiRetrieverTool(BaseTool): ...@@ -95,6 +95,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