from pydantic import Field
from pydantic_settings import BaseSettings, SettingsConfigDict


class DeepSeekConfig(BaseSettings):
    """DeepSeek API 配置"""

    api_key: str = Field(
        default="sk-cfd2df92c8864ab999d66a615ee812c5", description="DeepSeek API Key"
    )
    reasoner_model: str = Field(
        default="deepseek-reasoner", description="DeepSeek 推理模型"
    )
    chat_model: str = Field(default="deepseek-chat", description="DeepSeek 对话模型")

    model_config = SettingsConfigDict(
        env_prefix="DEEPSEEK_", env_file=".env", case_sensitive=False, extra="ignore"
    )

    def get_model_map(self) -> dict:
        """获取模型映射字典，兼容旧代码"""
        return {
            "DeepSeek-R1": self.reasoner_model,
            "DeepSeek-V3": self.chat_model,
        }