|
@@ -88,7 +88,7 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
|
|
|
DocumentSegment.dataset_id.in_(self.dataset_ids),
|
|
|
DocumentSegment.completed_at.isnot(None),
|
|
|
DocumentSegment.status == "completed",
|
|
|
- DocumentSegment.enabled == True, # noqa
|
|
|
+ DocumentSegment.enabled == True,
|
|
|
DocumentSegment.index_node_id.in_(index_node_ids),
|
|
|
).all()
|
|
|
|
|
@@ -109,8 +109,8 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
|
|
|
dataset = Dataset.query.filter_by(id=segment.dataset_id).first()
|
|
|
document = Document.query.filter(
|
|
|
Document.id == segment.document_id,
|
|
|
- Document.enabled == True, # noqa
|
|
|
- Document.archived == False, # noqa
|
|
|
+ Document.enabled == True,
|
|
|
+ Document.archived == False,
|
|
|
).first()
|
|
|
if dataset and document:
|
|
|
source = {
|