|
@@ -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,
|