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

Fix: fixed invalid save() arguments for slide thumbnails (#8851)

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

Fixed invalid save() arguments for slide thumbnails.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Yongteng Lei 2025-07-15 17:19:45 +08:00 • committed by GitHub
parent aa4a725529
commit e9b14142a5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rag/app/presentation.py
View file

@ -43,8 +43,9 @@ class Ppt(PptParser):
with BytesIO() as buffered: with BytesIO() as buffered:
slide.get_thumbnail( slide.get_thumbnail(
0.5, 0.5).save( 0.5, 0.5).save(
buffered, drawing.imaging.ImageFormat.jpeg, quality=80) buffered, drawing.imaging.ImageFormat.jpeg)
imgs.append(Image.open(buffered)) buffered.seek(0)
imgs.append(Image.open(buffered).copy())
except RuntimeError as e: except RuntimeError as e:
raise RuntimeError(f'ppt parse error at page {i+1}, original error: {str(e)}') from e raise RuntimeError(f'ppt parse error at page {i+1}, original error: {str(e)}') from e
assert len(imgs) == len( assert len(imgs) == len(

8
rag/svr/task_executor.py
View file

@ -302,7 +302,7 @@ async def build_chunks(task, progress_callback):
d["image"].close() # Close original image d["image"].close() # Close original image
d["image"] = converted_image d["image"] = converted_image
d["image"].save(output_buffer, format='JPEG') d["image"].save(output_buffer, format='JPEG')
async with minio_limiter: async with minio_limiter:
await trio.to_thread.run_sync(lambda: STORAGE_IMPL.put(task["kb_id"], d["id"], output_buffer.getvalue())) await trio.to_thread.run_sync(lambda: STORAGE_IMPL.put(task["kb_id"], d["id"], output_buffer.getvalue()))
d["img_id"] = "{}-{}".format(task["kb_id"], d["id"]) d["img_id"] = "{}-{}".format(task["kb_id"], d["id"])
@ -631,7 +631,7 @@ async def do_handle_task(task):
nursery.start_soon(delete_image, task_dataset_id, chunk_id) nursery.start_soon(delete_image, task_dataset_id, chunk_id)
progress_callback(-1, msg=f"Chunk updates failed since task {task['id']} is unknown.") progress_callback(-1, msg=f"Chunk updates failed since task {task['id']} is unknown.")
return return
logging.info("Indexing doc({}), page({}-{}), chunks({}), elapsed: {:.2f}".format(task_document_name, task_from_page, logging.info("Indexing doc({}), page({}-{}), chunks({}), elapsed: {:.2f}".format(task_document_name, task_from_page,
task_to_page, len(chunks), task_to_page, len(chunks),
timer() - start_ts)) timer() - start_ts))
@ -723,8 +723,8 @@ async def report_status():
finally: finally:
redis_lock.release() redis_lock.release()
await trio.sleep(30) await trio.sleep(30)
async def task_manager(): async def task_manager():
try: try:
await handle_task() await handle_task()