Переглянути джерело

fix(extractor): temporary file (#10543)

tags/0.11.1
-LAN- 11 місяці тому
джерело
коміт
a1543b7da0
Аккаунт користувача з таким Email не знайдено
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")


Завантаження…
Відмінити
Зберегти