| 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"]) | ||||