fix dataset_id and index_node_id idx missed in document_segments tabl… (

#7681)
langgenius · Aug 27, 2024 · 35431bc · 35431bc
1 parent 7b7576a
commit 35431bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/core/indexing_runner.py b/api/core/indexing_runner.py
@@ -720,6 +720,7 @@ def _process_keyword_index(flask_app, dataset_id, document_id, documents):
                 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 @@ def _process_chunk(self, flask_app, index_processor, chunk_documents, dataset, d
             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({