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

langgenius · Jan 8, 2025 · 53bb37b · 53bb37b
1 parent d258627
commit 53bb37b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/api/core/rag/extractor/pdf_extractor.py b/api/core/rag/extractor/pdf_extractor.py
@@ -23,7 +23,6 @@ def __init__(self, file_path: str, file_cache_key: Optional[str] = None):
         self._file_cache_key = file_cache_key
 
     def extract(self) -> list[Document]:
-        plaintext_file_key = ""
         plaintext_file_exists = False
         if self._file_cache_key:
             try:
@@ -39,8 +38,8 @@ def extract(self) -> list[Document]:
         text = "\n\n".join(text_list)
 
         # 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