From ce3dd019c3a9d8d1886769a3184a2ba00a81ec5e Mon Sep 17 00:00:00 2001 From: Jay Xu Date: Mon, 11 Aug 2025 17:07:49 +0800 Subject: [PATCH] Fix broken data stream when writing image file (#9354) ### What problem does this PR solve? fix "broken data stream when writing image file", just log warning and ignore Close #8379 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- rag/svr/task_executor.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index ba34103d9..f089b0f4e 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -304,7 +304,11 @@ async def build_chunks(task, progress_callback): converted_image = d["image"].convert("RGB") d["image"].close() # Close original image d["image"] = converted_image - d["image"].save(output_buffer, format='JPEG') + try: + d["image"].save(output_buffer, format='JPEG') + except OSError as e: + logging.warning( + "Saving image of chunk {}/{}/{} got exception, ignore: {}".format(task["location"], task["name"], d["id"], str(e))) async with minio_limiter: await trio.to_thread.run_sync(lambda: STORAGE_IMPL.put(task["kb_id"], d["id"], output_buffer.getvalue()))