Spaces:
Sleeping
Sleeping
| """全局配置存储模块，提供配置文件的加载和缓存功能，API密钥和base_url从环境变量加载。""" | |
| import os | |
| from typing import Any, Dict, Optional | |
| import yaml | |
| class ConfigManager: | |
| """配置管理器，使用单例模式缓存配置，API密钥和base_url从环境变量加载。""" | |
| _instance = None | |
| _config: Optional[Dict[str, Any]] = None | |
| def __new__(cls): | |
| if cls._instance is None: | |
| cls._instance = super().__new__(cls) | |
| return cls._instance | |
| def get_config(self) -> Dict[str, Any]: | |
| """获取配置，如果未加载则自动加载。 | |
| Returns: | |
| 包含所有配置信息的字典 | |
| """ | |
| if self._config is None: | |
| self._config = self._load_config() | |
| return self._config | |
| def _get_environment(self) -> str: | |
| """获取当前环境类型。 | |
| Returns: | |
| 环境类型：'prod' 或 'dev' | |
| """ | |
| return os.getenv("ENVIRONMENT", "dev").lower() | |
| def _get_config_path(self) -> str: | |
| """根据环境获取配置文件路径。 | |
| Returns: | |
| 配置文件路径 | |
| """ | |
| env = self._get_environment() | |
| if env == "prod": | |
| return "config/app_config_prod.yaml" | |
| return "config/app_config_dev.yaml" | |
| def _load_config(self) -> Dict[str, Any]: | |
| """加载配置文件，并覆盖API密钥和base_url为环境变量值。 | |
| Returns: | |
| 从YAML文件加载的配置字典，API密钥和base_url从环境变量覆盖 | |
| """ | |
| config_path = self._get_config_path() | |
| try: | |
| with open(config_path, "r", encoding="utf-8") as file: | |
| config = yaml.safe_load(file) | |
| # 添加环境信息到配置中 | |
| config["environment"] = self._get_environment() | |
| # 从环境变量覆盖API密钥和base_url | |
| self._override_api_configs(config) | |
| return config | |
| except FileNotFoundError as exc: | |
| raise FileNotFoundError(f"配置文件未找到: {config_path}") from exc | |
| except yaml.YAMLError as exc: | |
| raise ValueError(f"配置文件格式错误: {exc}") from exc | |
| def _override_api_configs(self, config: Dict[str, Any]) -> None: | |
| """从环境变量覆盖API密钥和base_url配置。 | |
| Args: | |
| config: 配置字典 | |
| """ | |
| # QA LLM 主模型 | |
| if "qa-llm" in config and "main" in config["qa-llm"]: | |
| main_config = config["qa-llm"]["main"] | |
| if os.getenv("QA_LLM_MAIN_API_KEY"): | |
| main_config["api_key"] = os.getenv("QA_LLM_MAIN_API_KEY") | |
| if os.getenv("QA_LLM_MAIN_BASE_URL"): | |
| main_config["base_url"] = os.getenv("QA_LLM_MAIN_BASE_URL") | |
| # QA LLM 备用模型 | |
| if "qa-llm" in config and "backup" in config["qa-llm"]: | |
| backup_config = config["qa-llm"]["backup"] | |
| if os.getenv("QA_LLM_BACKUP_API_KEY"): | |
| backup_config["api_key"] = os.getenv("QA_LLM_BACKUP_API_KEY") | |
| if os.getenv("QA_LLM_BACKUP_BASE_URL"): | |
| backup_config["base_url"] = os.getenv("QA_LLM_BACKUP_BASE_URL") | |
| # Rewrite LLM 备用模型 (GPT-4o) | |
| if "rewrite-llm" in config and "backup" in config["rewrite-llm"]: | |
| backup_config = config["rewrite-llm"]["backup"] | |
| if os.getenv("REWRITE_LLM_BACKUP_API_KEY"): | |
| backup_config["api_key"] = os.getenv("REWRITE_LLM_BACKUP_API_KEY") | |
| if os.getenv("REWRITE_LLM_BACKUP_BASE_URL"): | |
| backup_config["base_url"] = os.getenv("REWRITE_LLM_BACKUP_BASE_URL") | |
| # Rewrite LLM 主模型 | |
| if "rewrite-llm" in config and "main" in config["rewrite-llm"]: | |
| main_config = config["rewrite-llm"]["main"] | |
| if os.getenv("REWRITE_LLM_MAIN_API_KEY"): | |
| main_config["api_key"] = os.getenv("REWRITE_LLM_MAIN_API_KEY") | |
| if os.getenv("REWRITE_LLM_MAIN_BASE_URL"): | |
| main_config["base_url"] = os.getenv("REWRITE_LLM_MAIN_BASE_URL") | |
| # 全局配置管理器实例 | |
| _config_manager = ConfigManager() | |
| def get_model_config() -> Dict[str, Any]: | |
| """获取模型配置。 | |
| Returns: | |
| 包含所有配置信息的字典 | |
| """ | |
| return _config_manager.get_config() | |