jandres/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-08 07:14:14 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

feat: Implement partial update for document metadata, allowing merging of new values with existing ones. (#28390)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
GuanMu
2025-11-21 12:58:20 +08:00
committed by GitHub
parent 06466cb73a
commit 5f61ca5e6f
6 changed files with 185 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
api/services/entities/knowledge_entities/knowledge_entities.py
View File

@@ -158,6 +158,7 @@ class MetadataDetail(BaseModel):
class DocumentMetadataOperation(BaseModel):
document_id: str
metadata_list: list[MetadataDetail]
partial_update: bool = False
class MetadataOperationData(BaseModel):

19
api/services/metadata_service.py
View File

@@ -206,7 +206,10 @@ class MetadataService:
document = DocumentService.get_document(dataset.id, operation.document_id)
if document is None:
raise ValueError("Document not found.")
doc_metadata = {}
if operation.partial_update:
doc_metadata = copy.deepcopy(document.doc_metadata) if document.doc_metadata else {}
else:
doc_metadata = {}
for metadata_value in operation.metadata_list:
doc_metadata[metadata_value.name] = metadata_value.value
if dataset.built_in_field_enabled:
@@ -219,9 +222,21 @@ class MetadataService:
db.session.add(document)
db.session.commit()
# deal metadata binding
db.session.query(DatasetMetadataBinding).filter_by(document_id=operation.document_id).delete()
if not operation.partial_update:
db.session.query(DatasetMetadataBinding).filter_by(document_id=operation.document_id).delete()
current_user, current_tenant_id = current_account_with_tenant()
for metadata_value in operation.metadata_list:
# check if binding already exists
if operation.partial_update:
existing_binding = (
db.session.query(DatasetMetadataBinding)
.filter_by(document_id=operation.document_id, metadata_id=metadata_value.id)
.first()
)
if existing_binding:
continue
dataset_metadata_binding = DatasetMetadataBinding(
tenant_id=current_tenant_id,
dataset_id=dataset.id,