|
@@ -44,7 +44,6 @@ class DatasetIndexToolCallbackHandler:
|
|
DocumentSegment.index_node_id == document.metadata["doc_id"]
|
|
DocumentSegment.index_node_id == document.metadata["doc_id"]
|
|
)
|
|
)
|
|
|
|
|
|
- # if 'dataset_id' in document.metadata:
|
|
|
|
if "dataset_id" in document.metadata:
|
|
if "dataset_id" in document.metadata:
|
|
query = query.filter(DocumentSegment.dataset_id == document.metadata["dataset_id"])
|
|
query = query.filter(DocumentSegment.dataset_id == document.metadata["dataset_id"])
|
|
|
|
|