LQAgentPlatform/config/model_setting.yaml

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

# 可用模型列表（必须与 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
  - shutian_qwen3_6_27b    # 蜀天Qwen3.6-27B

  # Embedding 模型
  - 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: "文档二级分类，蜀天122B"

  # 文档分类 - 三级分类（需要高精度行级分类）
  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"

  # 查询提取（从审查条文中提取查询关键词）
  query_extract:
    model: shutian_qwen3_5_35b
    enable_thinking: false
    description: "查询提取，从审查条文中提取实体和关键词，蜀天35B"

  # 相关性判断（判断两个文本是否相关）
  relevance_judge:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "相关性判断，判断文本是否可作为审查引用依据，蜀天122B"

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

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

  # 语义逻辑检查
  semantic_logic_check:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "语义逻辑审查，蜀天122B"

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

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

  # 非参数合规性检查（安全相关/强制性条文）
  non_parameter_compliance_check:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "非参数合规性检查，安全相关/强制性条文知识库，蜀天122B"

  # 参数合规性检查（实体概念/工程术语）
  parameter_compliance_check:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "参数合规性检查，实体概念/工程术语知识库，蜀天122B"

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

  # 目录完整性审查（对比实际目录与标准目录）
  catalog_integrity_review:
    model: shutian_qwen3_5_122b
    enable_thinking: true
    description: "目录完整性审查，对比OCR提取目录与标准目录，找出缺失项，蜀天122B"

  # ============================================================
  # 施工方案编写模块（construction_write）
  # 说明：编写模块各功能可用的模型集中在此分组，新增编写功能请在此处添加。
  # 当前编写模块的 LLM 入口集中在 outline_generator._call_llm()。
  # ============================================================

  # 章节内容生成（模板受限校订模式）
  outline_chapter_revise:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "施工方案章节模板受限校订，蜀天122B"

  # 施工方案大纲生成（SSE流式）
  write_outline_generate:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "施工方案大纲流式生成，蜀天122B"

  # 施工方案内容补全生成（SSE流式）
  write_content_generate:
    model: shutian_qwen3_5_122b
    enable_thinking: false
    description: "施工方案内容补全流式生成，蜀天122B"

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

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