From 35431bce0d25afd1c7213f40dc4e58be62e0cf08 Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Tue, 27 Aug 2024 10:25:24 +0800
Subject: [PATCH] =?UTF-8?q?fix=20dataset=5Fid=20and=20index=5Fnode=5Fid=20?=
 =?UTF-8?q?idx=20missed=20in=20document=5Fsegments=20tabl=E2=80=A6=20(#768?=
 =?UTF-8?q?1)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/core/indexing_runner.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/core/indexing_runner.py b/api/core/indexing_runner.py
index dddf5567c1..062666ac6a 100644
--- 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({