ailine/frontend/src/config.py

"""
前端配置管理模块
集中管理所有配置项，支持环境变量覆盖
需要类型转换的配置在此处理
"""

import os
from dataclasses import dataclass
from typing import Optional
from dotenv import load_dotenv

# 加载 .env 文件
load_dotenv()


# ========== 辅助函数：类型转换 ==========
def _get_str(key: str) -> str | None:
    """获取字符串配置"""
    return os.getenv(key)


def _get_int(key: str, default: int = 0) -> int:
    """获取整数配置，自动转换"""
    value = os.getenv(key)
    if value is not None:
        try:
            return int(value)
        except (ValueError, TypeError):
            pass
    return default


def _get_bool(key: str, default: bool = False) -> bool:
    """获取布尔配置，自动转换"""
    value = os.getenv(key)
    if value is not None:
        return value.lower() in ("true", "1", "yes", "on")
    return default


@dataclass
class FrontendConfig:
    """前端配置类 - 统一管理所有配置项"""
    
    # ==================== API 配置 ====================
    api_base: str = ""
    
    # ==================== 页面配置（固定值，无需环境变量） ====================
    page_title: str = "AI 个人助手"
    page_icon: str = "🤖"
    layout: str = "wide"
    
    # ==================== 模型配置（固定值，无需环境变量） ====================
    default_model: str = "zhipu"
    model_options: Optional[dict] = None
    
    # ==================== 用户配置（固定值，无需环境变量） ====================
    default_user_id: str = "default_user"
    
    # ==================== 历史记录配置（固定值，无需环境变量） ====================
    history_limit: int = 50
    summary_max_length: int = 30
    
    # ==================== 流式响应配置（固定值，无需环境变量） ====================
    stream_timeout: int = 120
    
    # ==================== 日志配置 ====================
    log_level: str = ""
    debug: bool = False
    
    def __post_init__(self):
        """初始化后处理 - 设置默认值和加载环境变量"""
        if self.model_options is None:
            self.model_options = {
                "zhipu": "智谱 GLM-4.7-Flash（在线）",
                "local": "本地 llama.cpp（Gemma-4）",
                "deepseek": "DeepSeek V3.2（在线）"
            }
        
        # 从环境变量加载配置（优先级最高）
        self._load_from_env()
    
    def _load_from_env(self):
        """从环境变量加载配置（仅加载必要的配置项）"""
        # API 地址（移除 /chat 后缀）
        api_url = _get_str("API_URL")
        if api_url:
            self.api_base = api_url.replace("/chat", "").rstrip("/")
        
        # 日志配置
        log_level = _get_str("LOG_LEVEL")
        if log_level:
            self.log_level = log_level.upper()
        
        self.debug = _get_bool("DEBUG", False)


# 全局配置实例（单例模式）
config = FrontendConfig()