Ver Fonte

fix(extractor): temporary file (#10543)

-LAN- há 5 meses atrás
pai
commit
a1543b7da0
1 ficheiros alterados com 3 adições e 3 exclusões
  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")