From b2c8718f358107ae4d63e11b4132c8b97ebf4570 Mon Sep 17 00:00:00 2001
From: XiaoXiaoJiangYun <33343036+XiaoXiaoJiangYun@users.noreply.github.com>
Date: Sat, 2 Aug 2025 12:15:23 +0800
Subject: [PATCH] Update metadata_service.py (#23272)

---
 api/services/metadata_service.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/services/metadata_service.py b/api/services/metadata_service.py
index 59a30040e..2a83588f4 100644
--- a/api/services/metadata_service.py
+++ b/api/services/metadata_service.py
@@ -143,7 +143,6 @@ class MetadataService:
         lock_key = f"dataset_metadata_lock_{dataset.id}"
         try:
             MetadataService.knowledge_base_metadata_lock_check(dataset.id, None)
-            dataset.built_in_field_enabled = True
             db.session.add(dataset)
             documents = DocumentService.get_working_documents_by_dataset_id(dataset.id)
             if documents:
@@ -159,6 +158,7 @@ class MetadataService:
                     doc_metadata[BuiltInField.source.value] = MetadataDataSource[document.data_source_type].value
                     document.doc_metadata = doc_metadata
                     db.session.add(document)
+            dataset.built_in_field_enabled = True
             db.session.commit()
         except Exception:
             logging.exception("Enable built-in field failed")
@@ -172,7 +172,6 @@ class MetadataService:
         lock_key = f"dataset_metadata_lock_{dataset.id}"
         try:
             MetadataService.knowledge_base_metadata_lock_check(dataset.id, None)
-            dataset.built_in_field_enabled = False
             db.session.add(dataset)
             documents = DocumentService.get_working_documents_by_dataset_id(dataset.id)
             document_ids = []
@@ -190,6 +189,7 @@ class MetadataService:
                     document.doc_metadata = doc_metadata
                     db.session.add(document)
                     document_ids.append(document.id)
+            dataset.built_in_field_enabled = False
             db.session.commit()
         except Exception:
             logging.exception("Disable built-in field failed")