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

Refactor file parsing in FileComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 17:07:44 -03:00
commit dced59cef8
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/components/data/File.py
View file

@ -1,7 +1,7 @@
from typing import Any, Dict, Optional
from langflow import CustomComponent
from langflow.base.data.utils import parse_file_to_record
from langflow.base.data.utils import TEXT_FILE_TYPES, parse_text_file_to_record
from langflow.schema import Record
@ -25,4 +25,7 @@ class FileComponent(CustomComponent):
silent_errors: bool = False,
) -> Optional[Record]:
resolved_path = self.resolve_path(path)
return parse_file_to_record(resolved_path, silent_errors)
extension = resolved_path.split(".")[-1]
if extension not in TEXT_FILE_TYPES:
raise ValueError(f"Unsupported file type: {extension}")
return parse_text_file_to_record(resolved_path, silent_errors)