Jelajahi Sumber

fix dataset_id and index_node_id idx missed in document_segments tabl… (#7681)

Jyong 8 bulan lalu
induk
melakukan
35431bce0d
1 mengubah file dengan 2 tambahan dan 0 penghapusan
  1. 2 0
      api/core/indexing_runner.py

+ 2 - 0
api/core/indexing_runner.py

@@ -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({