From f7e8526a06392f89739a4422e1c058d9ed8ec950 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 28 Mar 2024 17:03:56 -0300 Subject: [PATCH] Fix getting extension in files --- src/backend/base/langflow/components/data/File.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/data/File.py b/src/backend/base/langflow/components/data/File.py index 4ffed5300..1e9d3e29e 100644 --- a/src/backend/base/langflow/components/data/File.py +++ b/src/backend/base/langflow/components/data/File.py @@ -1,3 +1,4 @@ +from pathlib import Path from typing import Any, Dict, List 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: 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": raise ValueError("doc files are not supported. Please save as .docx") if extension not in TEXT_FILE_TYPES: