-LAN- 5 месяцев назад
Родитель
Сommit
a1543b7da0
1 измененных файлов с 3 добавлено и 3 удалено
  1. 3 3
      api/core/rag/extractor/word_extractor.py

+ 3 - 3
api/core/rag/extractor/word_extractor.py

@@ -50,9 +50,9 @@ class WordExtractor(BaseExtractor):
 
             self.web_path = self.file_path
             # TODO: use a better way to handle the file
-            with tempfile.NamedTemporaryFile(delete=False) as self.temp_file:
-                self.temp_file.write(r.content)
-                self.file_path = self.temp_file.name
+            self.temp_file = tempfile.NamedTemporaryFile()  # noqa: SIM115
+            self.temp_file.write(r.content)
+            self.file_path = self.temp_file.name
         elif not os.path.isfile(self.file_path):
             raise ValueError(f"File path {self.file_path} is not a valid file or url")