mirror of
https://github.com/langgenius/dify.git
synced 2025-11-14 02:12:34 +00:00
1b0f92a331
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
34 lines
997 B
Python
34 lines
997 B
Python
import logging
|
|
from functools import lru_cache
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import yaml # type: ignore
|
|
from yaml import YAMLError
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _load_yaml_file(*, file_path: str):
|
|
if not file_path or not Path(file_path).exists():
|
|
raise FileNotFoundError(f"File not found: {file_path}")
|
|
|
|
with open(file_path, encoding="utf-8") as yaml_file:
|
|
try:
|
|
yaml_content = yaml.safe_load(yaml_file)
|
|
return yaml_content
|
|
except Exception as e:
|
|
raise YAMLError(f"Failed to load YAML file {file_path}: {e}") from e
|
|
|
|
|
|
@lru_cache(maxsize=128)
|
|
def load_yaml_file_cached(file_path: str) -> Any:
|
|
"""
|
|
Cached version of load_yaml_file for static configuration files.
|
|
Only use for files that don't change during runtime (e.g., position files)
|
|
|
|
:param file_path: the path of the YAML file
|
|
:return: an object of the YAML content
|
|
"""
|
|
return _load_yaml_file(file_path=file_path)
|