Maker-Checker：先写后查

{"passed": false, "feedback": "太笼统。加上具体景点名和时间。"}

"""maker_checker_travel.py — 旅游行程的 Maker-Checker"""
from __future__ import annotations

import json
from dataclasses import dataclass
from typing import Any


# ── 极简运行时（和项目 runtime 同构，但自包含可直接跑） ──────────

@dataclass(frozen=True)
class Message:
    role: str
    content: str

class MockLLM:
    """按顺序返回预设回复的假模型。"""
    def __init__(self, responses: list[str]):
        self._responses = list(responses)
        self._idx = 0
    def complete(self, messages: list[Message]) -> str:
        if self._idx >= len(self._responses):
            raise RuntimeError("MockLLM 回复用完了")
        out = self._responses[self._idx]
        self._idx += 1
        return out

@dataclass(frozen=True)
class CheckResult:
    passed: bool
    feedback: str


# ── Maker-Checker 核心逻辑 ──────────────────────────────

def maker_checker(
    maker: MockLLM,
    checker: MockLLM,
    *,
    task: str,
    max_rounds: int = 3,
) -> str:
    """
    1) Maker 写草稿
    2) Checker 返回 {passed, feedback}
    3) 没通过 → Maker 带着反馈修订
    4) 到 max_rounds 还没通过，返回最后一版
    """
    draft: str | None = None
    for round_idx in range(max_rounds):
        # ── Maker 阶段 ──
        if draft is None:
            maker_prompt = task
        else:
            maker_prompt = (
                f"任务：\n{task}\n\n"
                f"请根据反馈修订下面的草稿：\n{draft}"
            )
        draft = maker.complete([
            Message(role="system", content="你是行程规划师。写出最佳草稿。"),
            Message(role="user", content=maker_prompt),
        ])
        print(f"[轮次 {round_idx + 1}] Maker 草稿：{draft[:80]}...")

        # ── Checker 阶段 ──
        check_prompt = (
            f"任务：\n{task}\n\n草稿：\n{draft}\n\n"
            "按 rubric 检查，返回 JSON：{{\"passed\": true/false, \"feedback\": \"...\"}}"
        )
        raw = checker.complete([
            Message(role="system", content="你是行程审查员。严格按标准评估。只返回 JSON。"),
            Message(role="user", content=check_prompt),
        ])
        result = _parse_check(raw)
        print(f"[轮次 {round_idx + 1}] Checker：passed={result.passed}, feedback={result.feedback}")

        if result.passed:
            return draft

    print(f"[警告] {max_rounds} 轮后仍未通过，返回最后草稿")
    return draft or ""


def _parse_check(raw: str) -> CheckResult:
    obj = json.loads(raw)
    return CheckResult(
        passed=bool(obj["passed"]),
        feedback=str(obj.get("feedback", "")),
    )


# ── 运行 ────────────────────────────────────────────────

def main() -> None:
    task = (
        "用户要求：杭州一日游，偏好轻松步行，下午可能下雨。\n"
        "Rubric：\n"
        "1. 至少 3 个景点，含室内备选\n"
        "2. 时间安排具体到小时\n"
        "3. 提到下雨应对\n"
        "4. 提到交通方式"
    )

    maker = MockLLM([
        # 第 1 轮草稿：太笼统
        "上午去西湖，下午去灵隐寺，晚上河坊街。建议带伞。",
        # 第 2 轮草稿：修订后更好
        (
            "09:00 西湖断桥（步行 40 分钟）→ "
            "10:30 中国茶叶博物馆（室内，步行 30 分钟）→ "
            "12:00 龙井村午餐 → "
            "14:00 如遇下雨转中国丝绸博物馆（室内）/ 晴天继续太子湾公园 → "
            "17:00 河坊街小吃。全程公交+步行，备折叠伞。"
        ),
    ])

    checker = MockLLM([
        # 第 1 轮检查：不通过
        json.dumps({
            "passed": False,
            "feedback": "缺少具体时间安排；未提供室内备选；交通方式不明确。"
        }, ensure_ascii=False),
        # 第 2 轮检查：通过
        json.dumps({
            "passed": True,
            "feedback": ""
        }),
    ])

    result = maker_checker(maker, checker, task=task, max_rounds=3)
    print(f"\n最终行程：\n{result}")


if __name__ == "__main__":
    main()

[轮次 1] Maker 草稿：上午去西湖，下午去灵隐寺，晚上河坊街。建议带伞。...
[轮次 1] Checker：passed=False, feedback=缺少具体时间安排；未提供室内备选；交通方式不明确。
[轮次 2] Maker 草稿：09:00 西湖断桥（步行 40 分钟）→ 10:30 中国茶叶博物馆（室内，步行 30 分钟）→ 12:00 龙...
[轮次 2] Checker：passed=True, feedback=

最终行程：
09:00 西湖断桥（步行 40 分钟）→ 10:30 中国茶叶博物馆（室内，步行 30 分钟）→ 12:00 龙井村午餐 → 14:00 如遇下雨转中国丝绸博物馆（室内）/ 晴天继续太子湾公园 → 17:00 河坊街小吃。全程公交+步行，备折叠伞。

总调用数 = 轮次数 × 2（一次 Maker + 一次 Checker）
最好情况 = 2 次调用（第 1 轮就通过）
最坏情况 = max_rounds × 2