| @@ -720,6 +720,7 @@ class IndexingRunner: | |||
| document_ids = [document.metadata['doc_id'] for document in documents] | |||
| db.session.query(DocumentSegment).filter( | |||
| DocumentSegment.document_id == document_id, | |||
| DocumentSegment.dataset_id == dataset_id, | |||
| DocumentSegment.index_node_id.in_(document_ids), | |||
| DocumentSegment.status == "indexing" | |||
| ).update({ | |||
| @@ -751,6 +752,7 @@ class IndexingRunner: | |||
| document_ids = [document.metadata['doc_id'] for document in chunk_documents] | |||
| db.session.query(DocumentSegment).filter( | |||
| DocumentSegment.document_id == dataset_document.id, | |||
| DocumentSegment.dataset_id == dataset.id, | |||
| DocumentSegment.index_node_id.in_(document_ids), | |||
| DocumentSegment.status == "indexing" | |||
| ).update({ | |||