From 5623839c710b3437343d3516995fdcea987d7bdb Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 22 Aug 2023 17:59:24 +0800 Subject: [PATCH] update document segment (#950) Co-authored-by: jyong --- api/core/docstore/dataset_docstore.py | 1 + api/core/indexing_runner.py | 1 + 2 files changed, 2 insertions(+) diff --git a/api/core/docstore/dataset_docstore.py b/api/core/docstore/dataset_docstore.py index 24dc6194e2..ca5f172ece 100644 --- a/api/core/docstore/dataset_docstore.py +++ b/api/core/docstore/dataset_docstore.py @@ -103,6 +103,7 @@ class DatesetDocumentStore: content=doc.page_content, word_count=len(doc.page_content), tokens=tokens, + enabled=False, created_by=self._user_id, ) if 'answer' in doc.metadata and doc.metadata['answer']: diff --git a/api/core/indexing_runner.py b/api/core/indexing_runner.py index ef6e9001c7..98248874e2 100644 --- a/api/core/indexing_runner.py +++ b/api/core/indexing_runner.py @@ -691,6 +691,7 @@ class IndexingRunner: DocumentSegment.status == "indexing" ).update({ DocumentSegment.status: "completed", + DocumentSegment.enabled: True, DocumentSegment.completed_at: datetime.datetime.utcnow() })