diff --git a/api/core/tools/utils/yaml_utils.py b/api/core/tools/utils/yaml_utils.py index 22e4d3d12..3526647b4 100644 --- a/api/core/tools/utils/yaml_utils.py +++ b/api/core/tools/utils/yaml_utils.py @@ -4,6 +4,7 @@ import os import yaml from yaml import YAMLError +logger = logging.getLogger(__name__) def load_yaml_file(file_path: str, ignore_error: bool = False) -> dict: """ @@ -24,11 +25,11 @@ def load_yaml_file(file_path: str, ignore_error: bool = False) -> dict: except Exception as e: raise YAMLError(f'Failed to load YAML file {file_path}: {e}') except FileNotFoundError as e: - logging.debug(f'Failed to load YAML file {file_path}: {e}') + logger.debug(f'Failed to load YAML file {file_path}: {e}') return {} except Exception as e: if ignore_error: - logging.warning(f'Failed to load YAML file {file_path}: {e}') + logger.warning(f'Failed to load YAML file {file_path}: {e}') return {} else: raise e