hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-13 12:56:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat/tools/gitlab (#10407)

Browse Source
This commit is contained in:
Leo.Wang
2024-11-08 09:53:03 +08:00
committed by GitHub
parent 0e8ab0588f
commit c9f785e00f
7 changed files with 320 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/rag/extractor/word_extractor.py
View File

@@ -28,7 +28,6 @@ logger = logging.getLogger(__name__)
class WordExtractor(BaseExtractor):
"""Load docx files.
Args:
file_path: Path to the file to load.
"""
@@ -51,9 +50,9 @@ class WordExtractor(BaseExtractor):
self.web_path = self.file_path
# TODO: use a better way to handle the file
self.temp_file = tempfile.NamedTemporaryFile() # noqa: SIM115
self.temp_file.write(r.content)
self.file_path = self.temp_file.name
with tempfile.NamedTemporaryFile(delete=False) as self.temp_file:
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")