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

refactor: Update FileComponent to handle missing file path

Browse source 
Update the FileComponent in File.py to handle the case where no file path is provided. If the path is empty, raise a ValueError with the message "Please, upload a file to use this component." This change improves the error handling and user experience of the component.
This commit is contained in:
ogabrielluiz 2024-06-13 01:00:07 -03:00
commit 1ac7555150
1 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,9 +1,10 @@
from pathlib import Path from pathlib import Path
from langflow.custom import Component
from langflow.inputs import FileInput, BoolInput
from langflow.template import Output
from langflow.schema import Data
from langflow.base.data.utils import TEXT_FILE_TYPES, parse_text_file_to_data from langflow.base.data.utils import TEXT_FILE_TYPES, parse_text_file_to_data
from langflow.custom import Component
from langflow.inputs import BoolInput, FileInput
from langflow.schema import Data
from langflow.template import Output
class FileComponent(Component): class FileComponent(Component):
@ -15,6 +16,7 @@ class FileComponent(Component):
FileInput( FileInput(
name="path", name="path",
display_name="Path", display_name="Path",
file_types=TEXT_FILE_TYPES,
info=f"Supported file types: {', '.join(TEXT_FILE_TYPES)}", info=f"Supported file types: {', '.join(TEXT_FILE_TYPES)}",
), ),
BoolInput( BoolInput(
@ -30,11 +32,12 @@ class FileComponent(Component):
] ]
def load_file(self) -> Data: def load_file(self) -> Data:
path = self.path if not self.path:
raise ValueError("Please, upload a file to use this component.")
resolved_path = self.resolve_path(self.path)
silent_errors = self.silent_errors silent_errors = self.silent_errors
resolved_path = Path(path).resolve() extension = Path(resolved_path).suffix[1:].lower()
extension = resolved_path.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")