72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
"""
|
||
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
|