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()))