From 2f0ce80d193beefd0ee7fe7ecfd2fe3f0c9adca8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 3 Apr 2024 16:47:31 -0300 Subject: [PATCH] Fix error handling in DirectoryReader class --- .../custom/directory_reader/directory_reader.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/interface/custom/directory_reader/directory_reader.py b/src/backend/base/langflow/interface/custom/directory_reader/directory_reader.py index 9ffb9825d..83075eeb3 100644 --- a/src/backend/base/langflow/interface/custom/directory_reader/directory_reader.py +++ b/src/backend/base/langflow/interface/custom/directory_reader/directory_reader.py @@ -3,9 +3,11 @@ import os import zlib from pathlib import Path -from langflow.interface.custom.custom_component import CustomComponent from loguru import logger +from langflow.interface.custom.custom_component import CustomComponent +from langflow.interface.custom.custom_component import CustomComponent + class CustomComponentPathValueError(ValueError): pass @@ -198,7 +200,12 @@ class DirectoryReader: Process a file by validating its content and returning the result and content/error message. """ - file_content = self.read_file_content(file_path) + try: + file_content = self.read_file_content(file_path) + except Exception as exc: + logger.exception(exc) + logger.error(f"Error while reading file {file_path}: {str(exc)}") + return False, f"Could not read {file_path}" if file_content is None: return False, f"Could not read {file_path}"