Fix: early return when update doc in sdk (#7907)

### What problem does this PR solve?

Fix early return when update doc. #7886

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Yongteng Lei 2025-05-28 19:20:27 +08:00 committed by GitHub
parent 1239f5afc8
commit b95747be4c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -306,27 +306,27 @@ def update_doc(tenant_id, dataset_id, document_id):
return get_error_data_result( return get_error_data_result(
f"`chunk_method` {req['chunk_method']} doesn't exist" f"`chunk_method` {req['chunk_method']} doesn't exist"
) )
if doc.parser_id.lower() == req["chunk_method"].lower():
return get_result()
if doc.type == FileType.VISUAL or re.search(r"\.(ppt|pptx|pages)$", doc.name): if doc.type == FileType.VISUAL or re.search(r"\.(ppt|pptx|pages)$", doc.name):
return get_error_data_result(message="Not supported yet!") return get_error_data_result(message="Not supported yet!")
e = DocumentService.update_by_id( if doc.parser_id.lower() != req["chunk_method"].lower():
doc.id, e = DocumentService.update_by_id(
{ doc.id,
"parser_id": req["chunk_method"], {
"progress": 0, "parser_id": req["chunk_method"],
"progress_msg": "", "progress": 0,
"run": TaskStatus.UNSTART.value, "progress_msg": "",
}, "run": TaskStatus.UNSTART.value,
) },
if not e: )
return get_error_data_result(message="Document not found!") if not e:
req["parser_config"] = get_parser_config( return get_error_data_result(message="Document not found!")
req["chunk_method"], req.get("parser_config") if not req.get("parser_config"):
) req["parser_config"] = get_parser_config(
DocumentService.update_parser_config(doc.id, req["parser_config"]) req["chunk_method"], req.get("parser_config")
)
DocumentService.update_parser_config(doc.id, req["parser_config"])
if doc.token_num > 0: if doc.token_num > 0:
e = DocumentService.increment_chunk_num( e = DocumentService.increment_chunk_num(
doc.id, doc.id,