gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor: Improve the try logic for upload_to_minio (#9735)

Browse source 
### What problem does this PR solve?

Improve the try logic for upload_to_minio

### Type of change

- [x] Refactoring
This commit is contained in:
Stephen Hu 2025-08-28 09:35:29 +08:00 • committed by GitHub
parent a1633e0a2f
commit c461261f0b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rag/svr/task_executor.py
View file

@ -293,8 +293,7 @@ async def build_chunks(task, progress_callback):
docs.append(d) docs.append(d)
return return
output_buffer = BytesIO() with BytesIO() as output_buffer:
try:
if isinstance(d["image"], bytes): if isinstance(d["image"], bytes):
output_buffer.write(d["image"]) output_buffer.write(d["image"])
output_buffer.seek(0) output_buffer.seek(0)
@ -317,8 +316,6 @@ async def build_chunks(task, progress_callback):
d["image"].close() d["image"].close()
del d["image"] # Remove image reference del d["image"] # Remove image reference
docs.append(d) docs.append(d)
finally:
output_buffer.close() # Ensure BytesIO is always closed
except Exception: except Exception:
logging.exception( logging.exception(
"Saving image of chunk {}/{}/{} got exception".format(task["location"], task["name"], d["id"])) "Saving image of chunk {}/{}/{} got exception".format(task["location"], task["name"], d["id"]))