jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Fix getting extension in files

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-28 17:03:56 -03:00
commit f7e8526a06
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/components/data/File.py
View file

@ -1,3 +1,4 @@
from pathlib import Path
from typing import Any, Dict, List from typing import Any, Dict, List
from langflow.base.data.utils import TEXT_FILE_TYPES, parse_text_file_to_record from langflow.base.data.utils import TEXT_FILE_TYPES, parse_text_file_to_record
@ -26,7 +27,8 @@ class FileComponent(CustomComponent):
def load_file(self, path: str, silent_errors: bool = False) -> Record: def load_file(self, path: str, silent_errors: bool = False) -> Record:
resolved_path = self.resolve_path(path) resolved_path = self.resolve_path(path)
extension = resolved_path.split(".")[-1] path_obj = Path(resolved_path)
extension = path_obj.suffix[1:].lower()
if extension == "doc": if extension == "doc":
raise ValueError("doc files are not supported. Please save as .docx") raise ValueError("doc files are not supported. Please save as .docx")
if extension not in TEXT_FILE_TYPES: if extension not in TEXT_FILE_TYPES: