瀏覽代碼

fix: fix the incorrect plaintext file key when saving (#10429)

tags/1.0.0-beta.1
Yingchun Lai 9 月之前
父節點
當前提交
53bb37b749
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 2 行新增3 行删除
  1. 2
    3
      api/core/rag/extractor/pdf_extractor.py

+ 2
- 3
api/core/rag/extractor/pdf_extractor.py 查看文件

self._file_cache_key = file_cache_key self._file_cache_key = file_cache_key


def extract(self) -> list[Document]: def extract(self) -> list[Document]:
plaintext_file_key = ""
plaintext_file_exists = False plaintext_file_exists = False
if self._file_cache_key: if self._file_cache_key:
try: try:
text = "\n\n".join(text_list) text = "\n\n".join(text_list)


# save plaintext file for caching # save plaintext file for caching
if not plaintext_file_exists and plaintext_file_key:
storage.save(plaintext_file_key, text.encode("utf-8"))
if not plaintext_file_exists and self._file_cache_key:
storage.save(self._file_cache_key, text.encode("utf-8"))


return documents return documents



Loading…
取消
儲存