zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 15:31:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: handle partial upload success in document batch upload (#16438)

Browse Source
This commit is contained in:
euvre
2026-06-29 13:06:14 +08:00
committed by GitHub
parent d56c17b1e6
commit a339e8a579
2 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
api/apps/restful_apis/document_api.py
View File

@@ -608,7 +608,11 @@ async def _upload_local_documents(kb, tenant_id):
parent_path=form.get("parent_path"),
parser_config_override=parser_config_override,
)
if err:
# Handle partial success: some files uploaded successfully, some had errors
is_partial_success = err and files
if err and not is_partial_success:
msg = "\n".join(err)
logging.error(msg)
return get_error_data_result(message=msg, code=RetCode.SERVER_ERROR)
@@ -622,10 +626,17 @@ async def _upload_local_documents(kb, tenant_id):
return_raw_files = request.args.get("return_raw_files", "false").lower() == "true"
if return_raw_files:
return get_result(data=files)
doc_data = files
else:
doc_data = [map_doc_keys_with_run_status(doc, run_status="0") for doc in files]
renamed_doc_list = [map_doc_keys_with_run_status(doc, run_status="0") for doc in files]
return get_result(data=renamed_doc_list)
# For partial success, include error message along with successful uploads
if is_partial_success:
msg = "\n".join(err)
logging.warning(f"Partial upload success: {len(files)} succeeded, {len(err)} failed - {msg}")
return construct_json_result(code=RetCode.SERVER_ERROR, message=msg, data=doc_data)
return get_result(data=doc_data)
@manager.route("/datasets/<dataset_id>/documents", methods=["GET"]) # noqa: F821