# 模型功能配置
# 按功能模块配置使用的模型及参数

# 可用模型列表（必须与 model_handler.py 中的模型类型名称一致）
# 参考: foundation/ai/models/model_handler.py
available_models:
  # DashScope 系列
  - qwen3_5_35b_a3b        # DashScope Qwen3.5-35B-A3B（默认兜底模型）
  - qwen3_5_27b            # DashScope Qwen3.5-27B
  - qwen3_5_122b_a10b      # DashScope Qwen3.5-122B-A10B

  # 豆包系列
  - doubao                 # 豆包模型
  - doubao-1.5-pro-256k    # 豆包1.5 Pro
  - doubao-1.5-lite-32k    # 豆包1.5 Lite

  # DeepSeek 系列
  - deepseek               # DeepSeek 模型
  - deepseek-v3            # DeepSeek V3

  # 本地模型系列
  - lq_qwen3_8b            # 本地Qwen3-8B
  - lq_qwen3_8b_lq_lora    # 本地Qwen3-8B-lq-lora
  - lq_qwen3_4b            # 本地Qwen3-4B
  - qwen_local_14b         # 本地Qwen3-14B

  # 蜀天算力系列
  - shutian_qwen3_5_122b   # 蜀天Qwen3.5-122B
  - shutian_qwen3_8b       # 蜀天Qwen3-8B
  - shutian_qwen3_5_35b    # 蜀天Qwen3.5-35B

  # Embedding 模型
  - lq_qwen3_8b_emd        # 本地Qwen3-Embedding-8B
  - siliconflow_embed      # 硅基流动Embedding
  - shutian_qwen3_embed    # 蜀天Embedding

  # Reranker 模型
  - lq_bge_reranker_v2_m3  # BGE-reranker-v2-m3
  - shutian_qwen3_reranker # 蜀天Reranker

# 功能模块模型配置
model_settings:
  # 文档分类 - 一级分类（使用完整章节标题进行分类）
  doc_classification_primary:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "文档一级分类，使用章节标题，蜀天35B"

  # 文档分类 - 二级分类（需要快速响应）
  doc_classification_secondary:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "文档二级分类，蜀天35B"

  # 文档分类 - 三级分类（需要高精度行级分类）
  doc_classification_tertiary:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "文档三级分类，蜀天122B"

  # 文档分类 - 三级分类复杂段落（可选更强的模型）
  doc_classification_tertiary_complex:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "文档三级分类-复杂段落，蜀天122B"

  # 完整性审查 - 内容生成
  completeness_review_generate:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "完整性审查内容生成，蜀天122B"

  # 完整性审查 - 分类识别
  completeness_review_classify:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "完整性审查快速分类，蜀天35B"

  # RAG 检索 - 查询理解
  rag_query_understand:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "RAG查询理解，蜀天35B"

  # RAG 检索 - 答案生成
  rag_answer_generate:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "RAG答案生成，蜀天122B"

  # 敏感信息检查
  sensitive_check:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "敏感信息快速检查，蜀天35B"

  # 语法检查
  grammar_check:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "语法快速检查，蜀天35B"

  # 时效性审查
  timeliness_review:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "时效性审查，蜀天35B"

  # 规范性审查（引用匹配）
  reference_review:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "规范性审查（引用匹配），蜀天35B"

  # 时效规范审查抽取（目录提取）
  directory_extraction:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "时效规范审查抽取（目录提取），蜀天35B"

  # Embedding 模型（用于相似度计算）
  embedding:
    model: lq_qwen3_8b_emd
    description: "文本Embedding向量生成"

# 默认配置（当功能未指定时使用）
default:
  model: shutian_qwen3_5_35b
  enable_thinking: false