"""
LLM 配置文件
定义大语言模型的默认参数
"""

# ============================================================
# 模型配置
# ============================================================

# 默认模型
MODEL = 'deepseek-v3.2'

# 可选模型列表
AVAILABLE_MODELS = [
    'deepseek-v3.2',        # 最新版，支持思考模式
    'deepseek-v3.2-exp',    # 实验版
    'deepseek-v3.1',        # 上一版本
    'deepseek-r1',          # 推理模型 (总是思考)
    'deepseek-r1-0528',     # 推理模型升级版
    'deepseek-v3',          # 基础版 (不思考)
]


# ============================================================
# API 配置
# ============================================================

# API Base URL (北京地域)
API_BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1'


# ============================================================
# 生成参数
# ============================================================

# 是否启用思考模式 (仅 deepseek-v3.2/v3.1 等支持)
ENABLE_THINKING = False

# 温度 (0.0-2.0, 越高越随机)
TEMPERATURE = 1.0

# Top P 采样
TOP_P = 0.95

# 最大输出 Token 数
MAX_TOKENS = 4096


# ============================================================
# 角色配置
# ============================================================

# 默认角色
DEFAULT_ROLE = 'assistant'

# 预设角色 (角色名 -> 系统提示词)
ROLES = {
    # 通用助手
    'assistant': """  你是心镜助手！ 是一个内置在一个ai镜子里面的助手你会以简短的话来安慰状态不好的用户！  """,
}


# ============================================================
# 上下文配置
# ============================================================

# 系统提示词 (默认使用角色配置)
SYSTEM_PROMPT = ROLES[DEFAULT_ROLE]

# 上下文长度限制
MAX_CONTEXT_LENGTH = 131072