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

1 year ago · 35431bce0d
parent 7b7576ad55
commit 35431bce0d
1 changed files with 2 additions and 0 deletions
--- a/api/core/indexing_runner.py
+++ b/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({