estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update metadata_service.py (#23272)

Browse Source
This commit is contained in:
XiaoXiaoJiangYun
2025-08-02 12:15:23 +08:00
committed by GitHub
parent 46ba0a8781
commit b2c8718f35
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/services/metadata_service.py
View File

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