From b95747be4ce47ece6d2aaa14c307749c01445638 Mon Sep 17 00:00:00 2001 From: Yongteng Lei Date: Wed, 28 May 2025 19:20:27 +0800 Subject: [PATCH] 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) --- api/apps/sdk/doc.py | 34 +++++++++++++++++----------------- 1 file changed, 17 insertions(+), 17 deletions(-) diff --git a/api/apps/sdk/doc.py b/api/apps/sdk/doc.py index f76065c1c..5fe91e961 100644 --- a/api/apps/sdk/doc.py +++ b/api/apps/sdk/doc.py @@ -306,27 +306,27 @@ def update_doc(tenant_id, dataset_id, document_id): return get_error_data_result( 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): return get_error_data_result(message="Not supported yet!") - e = DocumentService.update_by_id( - doc.id, - { - "parser_id": req["chunk_method"], - "progress": 0, - "progress_msg": "", - "run": TaskStatus.UNSTART.value, - }, - ) - if not e: - return get_error_data_result(message="Document not found!") - req["parser_config"] = get_parser_config( - req["chunk_method"], req.get("parser_config") - ) - DocumentService.update_parser_config(doc.id, req["parser_config"]) + if doc.parser_id.lower() != req["chunk_method"].lower(): + e = DocumentService.update_by_id( + doc.id, + { + "parser_id": req["chunk_method"], + "progress": 0, + "progress_msg": "", + "run": TaskStatus.UNSTART.value, + }, + ) + if not e: + return get_error_data_result(message="Document not found!") + if not req.get("parser_config"): + req["parser_config"] = get_parser_config( + req["chunk_method"], req.get("parser_config") + ) + DocumentService.update_parser_config(doc.id, req["parser_config"]) if doc.token_num > 0: e = DocumentService.increment_chunk_num( doc.id,