"""
Few-shot CoT 提示词模板。

三种模式对比：
  A. 无 CoT      → 直接要答案
  B. 零样本 CoT  → 只说「请一步一步思考」，不给示例
  C. 少样本 CoT  → 先给 1～2 个「问题 + 推理过程」示例，再答新题
"""

SYSTEM_DIRECT = (
    "你是一名逻辑推理助手。"
    "你必须严格遵守：只输出一行结论，格式为「最终答案：xxx」。"
    "禁止输出任何推理过程、分步说明或解释。"
)

USER_DIRECT_TEMPLATE = "{question}"

SYSTEM_ZERO_SHOT_COT = (
    "你是一名严谨的逻辑推理助手。"
    "解决问题时必须先写出完整推理过程，再给出最终答案。"
    "请按「第一步 / 第二步 / …」组织推理，最后一行必须以「最终答案：」开头。"
)

USER_ZERO_SHOT_COT_TEMPLATE = (
    "{question}\n\n"
    "请一步一步思考，并把每一步推理过程写出来。"
)

SYSTEM_FEW_SHOT_COT = (
    "你是一名严谨的逻辑推理助手。"
    "我会先给你若干个「示例问题 + 示例推理」，请你严格模仿示例的推理格式来回答新问题。"
    "要求：\n"
    "1. 必须按「第一步 / 第二步 / …」分步写出；\n"
    "2. 遇到年龄题，要先算年龄差，不要直接把「一半」套用到现在；\n"
    "3. 最后一行必须以「最终答案：」开头。"
)

USER_FEW_SHOT_COT_TEMPLATE = "{question}"