浏览代码

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

### 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)
tags/v0.20.0
Yongteng Lei 3 个月前
父节点
当前提交
e9b14142a5
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 7 次插入6 次删除
  1. 3
    2
      rag/app/presentation.py
  2. 4
    4
      rag/svr/task_executor.py

+ 3
- 2
rag/app/presentation.py 查看文件

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

+ 4
- 4
rag/svr/task_executor.py 查看文件

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

正在加载...
取消
保存