#!/usr/bin/env python3
"""
远程 Chat API 相关性门控评测（默认 OpenRouter；与门面 Worker v2 同提示词基线）。

默认 multi-chunk：以文章切片 Context 为单位请求，一次对多块各报 count
（与线上 /api/v2/analyze-semantic-relevance 一致）。单 chunk 已废弃，仅 --single-chunk。

主题：只评 expect_relevant（云端）；不管 expect_keywords / 本地 instruct relevance。
（磁盘上 query 为数组且真值在项内；加载后展平为 query:str。）
关键词归因请用 scripts/eval_semantic_keywords.py。

用法（项目根目录）:
  python scripts/eval_semantic_relevance_remote.py \\
    -c scripts/cases/红楼-第3回.json \\
    -o scripts/results/红楼-第3回_hy3_rel.jsonl
"""

from __future__ import annotations

import argparse
import json
import os
import re
import sys
import threading
import time
from concurrent.futures import ThreadPoolExecutor, as_completed
from pathlib import Path
from typing import Any, Dict, List, Optional, Tuple

_SCRIPTS_DIR = Path(__file__).resolve().parent
if str(_SCRIPTS_DIR) not in sys.path:
    sys.path.insert(0, str(_SCRIPTS_DIR))
from semantic_case_load import load_all_cases, load_articles

try:
    import requests
except ImportError:
    print("错误: 需要安装 requests 库")
    print("请运行: pip install requests")
    sys.exit(1)

HF_TOKEN_ENV = "HF_TOKEN"
OPENROUTER_TOKEN_ENV = "OPENROUTER_API_KEY"  # SYNC: Worker secret / .dev.vars 同名
DEFAULT_API_BASE = "https://openrouter.ai/api/v1"
DEFAULT_MODEL = "tencent/hy3"
DEFAULT_MAX_TOKENS = 8  # SYNC: cf/facade/src/relevance_remote.js RELEVANCE_MAX_TOKENS
MULTI_CHUNK_MAX = 32  # multi-chunk 时一个上下文（Context）最多容纳的 chunk 数，超出的另起独立 Context
# 结论（Hy3，覆盖现有全部 test case：红楼全集 + 论文 + synthetic + 小龙女，配对 McNemar）：
# ≤32 chunk 内 32 切片稳定、与单 chunk 基线无显著差异，小切片 16/24 会显著损伤；
# 32 chunk 以上尚无明确数据。
DEFAULT_MULTI_CHUNK_MAX_TOKENS = 8 * MULTI_CHUNK_MAX  # 多行输出：每 chunk 约 8 token（[N] 数），与上下文组上限对应
SEMANTIC_MATCH_THRESHOLD = 0.1  # SYNC: count>0 映射为 degree=1.0，否则 0.0

CLEARLY_ZERO_SENTENCE = (
    "If the text is not clearly related to the query topic, reply 0."
)
# clearly_zero 对两模型的总体影响（cases/subsets/红楼.smoke20、相对基线）：
# - DeepSeek-V4-Flash：拒识明显变好、召回下降（假阳↓、漏检↑），acc 净升
# - Hy3：两边都略伤，acc 净降；基线已较好时不必开


def build_relevance_user_content(query: str, text: str, *, clearly_zero: bool) -> str:
    """相关性 user 正文。clearly_zero 为唯一提示词变量。
    版式：Task/Query 各一行；Text: 后空一行接正文，正文后再空一行；
    文尾 Task Reminder:+Query: 再各一行。"""
    task = "How many words in the text are related to the query topic?"
    if clearly_zero:
        task += " " + CLEARLY_ZERO_SENTENCE
    task += " Reply with a single non-negative integer only, nothing else."
    query_line = f"Query: {query}"
    head = f"Task: {task}\n{query_line}"
    reminder = f"Task Reminder: {task}\n{query_line}"
    return f"{head}\nText:\n\n{text}\n\n{reminder}"


# 回复行格式。曾试 N:<count>，论文上偶发整组写崩；现为 [N]<count>（无空格）。
MULTI_CHUNK_OUTPUT_FORMAT = (
    "Output Format: each passage on its own line "
    "as [N]<count, 0 if not related>, where N is the passage index. Nothing else.\n"
    "Example reply for 3 passages:\n"
    "[1]0\n[2]0\n[3]3"
)


def build_multi_chunk_user_content(
    query: str, chunks: List[str], format_reminder: bool = False
) -> str:
    """multi-chunk 相关性 user 正文。Task 与 Output Format 分离且格式出现两次：
    - 三明治头尾同序：Task(Reminder) → Query → Output Format，正文夹在中间。
      Hy3 对这段顺序敏感、会影响门控精度：Query 若顶在生成口会更积极认相关、错检升；
      本序（Format 在最后）与旧「尾段单独贴 Format」精度相当。不要改成 Task→Format→Query。
    - Output Format（格式约束 + 0 示例）独立成段，头尾各一次。

    强调 chunks 是同一篇文章的连续切片（非平行独立 Text），按阅读顺序排列，
    正文前缀 Passage N:（不用 [N]，避免与文中 [N] 冲突）；回复仍为 [N]<count>
    （与 parse_multi_chunk_counts 对应）。

    task 为基线版。曾尝试追加全文判定句
    ［"A word's relevance is determined by its meaning in the whole article, not just in its own passage."］，
    实测增误放行显著（全集 acc 95.45%→88.11%，FP +21）且对真相关零增益，故不采用。
    SYNC：cf/facade/src/relevance_remote_v2.js buildMultiChunkUserContent。

    实验结论（Hy3，2026-08-07，回归+hard+all）：Output Format 出现一次（仅尾部）相比
    两次（头部 Task 后 + 尾部）误报率偏高（hard 13.8% vs 9.6%；all 5.6% vs 4.4%），
    且两次不恶化漏报（all 漏报均 5.7%）。故正式采用「格式两次」结构。"""
    task = (
        "The passages are consecutive slices of one complete article, in reading order. "
        "How many words in each passage are related to the query topic?"
    )
    query_line = f"Query: {query}"
    mid = f"{query_line}\n{MULTI_CHUNK_OUTPUT_FORMAT}"
    head = f"Task: {task}\n{mid}"
    reminder = f"Task Reminder: {task}\n{mid}"
    passages = "\n".join(f"Passage {i}: {text}" for i, text in enumerate(chunks, 1))
    content = f"{head}\nArticle:\n\n{passages}\n\n{reminder}"
    if format_reminder:
        n = len(chunks)
        if n == 1:
            example = "[1]0"
        elif n == 2:
            example = "[1]0\n[2]1"
        elif n == 3:
            example = "[1]0\n[2]1\n[3]0"
        else:
            example = f"[1]0\n[2]1\n...\n[{n}]0"
        content += (
            f"\nCRITICAL: Strictly adhere to the format. Output EXACTLY {n} lines, from [1] to [{n}]. Nothing else.\n"
            f"Example reply for {n} passages:\n"
            f"{example}"
        )
    return content


_RE_BRACKET_COUNT = re.compile(r"\[(\d+)\]\s*(\d+)")


def parse_multi_chunk_counts(content: Optional[str]) -> Optional[Dict[int, int]]:
    """从每行解析 count，返回 {N: count}（N 从 1 起）。

    `[N]数字` / `[N] 数字` 均可（空格可选）。对不上的行跳过，不猜。
    SYNC：与提示词 [N] 序号 / 门面 parseMultiChunkCounts 对应。"""
    if not content or not isinstance(content, str):
        return None
    out: Dict[int, int] = {}
    for line in content.splitlines():
        m = _RE_BRACKET_COUNT.match(line.strip())
        if not m:
            continue
        out[int(m.group(1))] = int(m.group(2))
    return out


def _load_env_file(path: Path) -> None:
    if not path.is_file():
        return
    for line in path.read_text(encoding="utf-8").splitlines():
        line = line.strip()
        if not line or line.startswith("#") or "=" not in line:
            continue
        k, v = line.split("=", 1)
        os.environ.setdefault(k.strip(), v.strip())


def _load_jsonl(path: Path) -> list:
    if not path.exists():
        return []
    results = []
    for line in path.read_text(encoding="utf-8").strip().split("\n"):
        if not line:
            continue
        try:
            results.append(json.loads(line))
        except json.JSONDecodeError:
            pass
    return results


def _append_record(path: Path, record: dict) -> None:
    with path.open("a", encoding="utf-8") as f:
        f.write(json.dumps(record, ensure_ascii=False) + "\n")


def load_cases(path: Path) -> List[dict]:
    return load_all_cases(path)


def parse_count(content: Optional[str]) -> Optional[int]:
    """从开头：可选空白 + 非负整数字前缀；不扫后面。失败返回 None。SYNC: facade parseCount"""
    if not content or not isinstance(content, str):
        return None
    m = re.match(r"\s*(\d+)", content)
    return int(m.group(1)) if m else None


def chat_relevance(
    api_base: str,
    model: str,
    query: str,
    text: str,
    *,
    clearly_zero: bool,
    token: str,
    timeout: int,
    max_tokens: int = DEFAULT_MAX_TOKENS,
) -> dict:
    url = f"{api_base.rstrip('/')}/chat/completions"
    body: Dict[str, Any] = {
        "model": model,
        "messages": [
            {
                "role": "user",
                "content": build_relevance_user_content(query, text, clearly_zero=clearly_zero),
            }
        ],
        "temperature": 0,
        "max_tokens": max_tokens,
        "stream": False,
    }
    # OpenRouter 统一用 reasoning.effort；HF/DeepSeek 原生用 thinking.disabled
    if "openrouter.ai" in api_base:
        body["reasoning"] = {"effort": "none"}
    else:
        body["thinking"] = {"type": "disabled"}
    headers = {
        "Authorization": f"Bearer {token}",
        "Content-Type": "application/json",
    }
    if "openrouter.ai" in api_base:
        headers["HTTP-Referer"] = "https://info-radar.local"
        headers["X-Title"] = "info-radar-relevance-eval"
    resp = requests.post(url, headers=headers, json=body, timeout=timeout)
    data = resp.json()
    if resp.status_code >= 400:
        err = data.get("error") or data
        raise RuntimeError(f"HTTP {resp.status_code}: {err}")
    if data.get("error"):
        raise RuntimeError(str(data["error"]))
    choice = (data.get("choices") or [{}])[0]
    msg = choice.get("message") or {}
    content = msg.get("content")
    count = parse_count(content)
    if count is None:
        raise RuntimeError(f"unparseable model output: content={content!r}")
    # 远程无 logprobs 时：count>0 → 1.0，否则 0.0（与门控 count>0 一致）
    degree = 1.0 if count > 0 else 0.0
    return {
        "content": content,
        "count": count,
        "full_match_degree": degree,
        "finish_reason": choice.get("finish_reason"),
        "usage": data.get("usage"),
        "raw_model": data.get("model") or model,
    }


def chat_relevance_multi_chunk(
    api_base: str,
    model: str,
    query: str,
    chunks: List[str],
    *,
    token: str,
    timeout: int,
    format_reminder: bool = False,
) -> dict:
    """multi-chunk：一次请求让模型对整组连续切片各报 count。

    返回 {"counts": {N: count}, "content": 原文, "finish_reason", "usage", "raw_model"}。
    输出解析失败（目标行缺 或 完全不可 parse）时抛错，由调用方按重试处理。"""
    url = f"{api_base.rstrip('/')}/chat/completions"
    body: Dict[str, Any] = {
        "model": model,
        "messages": [
            {
                "role": "user",
                "content": build_multi_chunk_user_content(query, chunks, format_reminder=format_reminder),
            }
        ],
        "temperature": 0,
        "max_tokens": max(DEFAULT_MULTI_CHUNK_MAX_TOKENS, 16 * len(chunks)),
        "stream": False,
    }
    if "openrouter.ai" in api_base:
        body["reasoning"] = {"effort": "none"}
    else:
        body["thinking"] = {"type": "disabled"}
    headers = {
        "Authorization": f"Bearer {token}",
        "Content-Type": "application/json",
    }
    if "openrouter.ai" in api_base:
        headers["HTTP-Referer"] = "https://info-radar.local"
        headers["X-Title"] = "info-radar-relevance-eval"
    resp = requests.post(url, headers=headers, json=body, timeout=timeout)
    data = resp.json()
    if resp.status_code >= 400:
        err = data.get("error") or data
        raise RuntimeError(f"HTTP {resp.status_code}: {err}")
    if data.get("error"):
        raise RuntimeError(str(data["error"]))
    choice = (data.get("choices") or [{}])[0]
    msg = choice.get("message") or {}
    content = msg.get("content")
    counts = parse_multi_chunk_counts(content)
    if counts is None:
        raise RuntimeError(f"unparseable multi-chunk output: content={content!r}")
    return {
        "counts": counts,
        "content": content,
        "finish_reason": choice.get("finish_reason"),
        "usage": data.get("usage"),
        "raw_model": data.get("model") or model,
    }


def run_one(
    api_base: str,
    model: str,
    case: dict,
    *,
    clearly_zero: bool,
    token: str,
    timeout: int,
    max_retries: int,
) -> dict:
    name = case["name"]
    query = case["query"]
    text = case["text"]
    expect_relevant = bool(case.get("expect_relevant"))
    disputed = bool(case.get("disputed"))
    dispute_note = case.get("dispute_note") or ""

    def _base(**extra: Any) -> dict:
        rec: Dict[str, Any] = {
            "case": name,
            "chunk_index": case.get("chunk_index"),
            "query": query,
            "expect_relevant": expect_relevant,
            "model": model,
            "source": case.get("source"),
            "clearly_zero": clearly_zero,
            **extra,
        }
        if disputed:
            rec["disputed"] = True
            if dispute_note:
                rec["dispute_note"] = dispute_note
        return rec

    last_error: Optional[BaseException] = None
    for attempt in range(max_retries + 1):
        try:
            r = chat_relevance(
                api_base, model, query, text,
                clearly_zero=clearly_zero, token=token, timeout=timeout,
            )
            degree = r["full_match_degree"]
            return _base(
                full_match_degree=degree,
                gate_passed=degree >= SEMANTIC_MATCH_THRESHOLD,
                count=r["count"],
                content=r["content"],
                finish_reason=r.get("finish_reason"),
                usage=r.get("usage"),
            )
        except Exception as e:
            last_error = e
            if attempt < max_retries:
                wait = 3 * (attempt + 1)
                print(f"  重试 {attempt + 1}/{max_retries}，{wait}s… {e}", flush=True)
                time.sleep(wait)
    return _base(error=f"relevance: {last_error}")


def _resolve_contexts(cases_dir: Path, ctx_max: int):
    """构建全部文章切片 Context，并反查 (source, chunk_index) → (ctx_idx, N)。

    返回 (contexts, lookup)：
      contexts: list[{"file", "source", "chunks": [有序 chunk, ...], "query": None}]
      lookup: {source: {chunk_index: (ctx_idx, N)}}
    同 source 出现在多个文件中（如 synthetic 的 "inline"）时，若 chunk_index 在多个
    文章里都命中则视为歧义，交由调用方报错，而非静默选一个。
    ctx_max：一个 Context 最多容纳的 chunk 数（切片上限），超出的另起独立 Context。"""
    contexts: List[dict] = []
    by_source: Dict[str, Dict[int, Tuple[int, int]]] = {}
    for art in load_articles(cases_dir):
        chunks = art["chunks"]
        src = art["source"]
        src_map = by_source.setdefault(src, {})
        for g in range(0, len(chunks), ctx_max):
            group = chunks[g:g + ctx_max]
            ctx_idx = len(contexts)
            contexts.append({"file": art["file"], "source": src, "chunks": group})
            for n, ch in enumerate(group, 1):
                src_map[ch["chunk_index"]] = (ctx_idx, n)
    return contexts, by_source


def _build_full_groups(pending, cases_dir, ctx_max):
    """全文章切片模式：每个被测块定位到其所属 Context（≤MULTI_CHUNK_MAX 块）。

    返回 list[group]；group = {"ctx_id", "query", "chunks", "targets"}，
    targets 为 [(i, case, n)]，n 是该块在 chunks 中的 [N] 序号（从 1 起）。"""
    from collections import defaultdict
    contexts, by_source = _resolve_contexts(cases_dir, ctx_max)
    by_group: Dict[Tuple[int, str], List[Tuple[int, dict, int]]] = defaultdict(list)
    for i, case in pending:
        src = case.get("source")
        cidx = case.get("chunk_index")
        src_map = by_source.get(src)
        if src_map is None:
            raise RuntimeError(
                f"multi-chunk: case {case['name']} source={src!r} 未在任一文章的切片上下文中（可能用例文件未含 chunk_index 字段，或非切片 corpus）"
            )
        hit = src_map.get(cidx)
        if hit is None:
            raise RuntimeError(
                f"multi-chunk: case {case['name']} 找不到所属 Context "
                f"(source={src!r}, chunk_index={cidx!r})"
            )
        ctx_idx, n = hit
        by_group[(ctx_idx, case["query"])].append((i, case, n))
    groups = []
    for (ctx_idx, query), targets in by_group.items():
        groups.append({
            "ctx_id": f"ctx{ctx_idx}",
            "query": query,
            "chunks": [c["text"] for c in contexts[ctx_idx]["chunks"]],
            "targets": targets,
        })
    return groups


def _run_multi_chunk_main(
    args,
    pending: List[Tuple[int, dict]],
    completed: set,
    all_results: list,
    *,
    write_lock,
    stop,
    done_n: int,
    cases: List[dict],
    token: str,
) -> None:
    """multi-chunk 主循环：以「Context」为单位请求（全文章切片）。

    每个被测 (source, chunk_index) 定位到其 Context 中的块号 N；同 Context 且同 query
    的被测块合并为一次请求，模型对该组每块各报 count（`[N] 数字`），结果按块写回逐条
    case 记录（与基线同构，可续跑/对照/acc 复用）。"""
    cases_dir = args.cases.parent if args.cases.parent.name != "subsets" else args.cases.parent.parent
    groups = _build_full_groups(pending, cases_dir, args.ctx_max)

    def _run_group(group):
        # 评测整组重打（与线上断点续跑不同）：acc 只看最终成功输出；失败次数另计。
        last_error = None
        attempts = 0
        for attempt in range(args.retries + 1):
            attempts = attempt + 1
            try:
                reminder = attempt > 0
                r = chat_relevance_multi_chunk(
                    args.url, args.model, group["query"], group["chunks"],
                    token=token, timeout=args.timeout, format_reminder=reminder,
                )
                return r, None, attempts
            except Exception as e:
                last_error = e
                if attempt < args.retries:
                    wait = 3 * (attempt + 1)
                    print(f"  ({group['ctx_id']}) 重试 {attempt + 1}/{args.retries}（format_reminder={reminder}），{wait}s… {e}", flush=True)
                    time.sleep(wait)
        return None, last_error, attempts

    def _emit_record(record: dict) -> bool:
        nonlocal done_n
        name = record.get("case") or record.get("multi_chunk_ctx")
        done_n += 1
        prog = f"[{done_n}/{len(cases)}]"
        all_results.append(record)
        if args.output:
            with write_lock:
                _append_record(args.output, record)
        if record.get("error"):
            print(f"{prog} ✗ {name}: {record['error']}", flush=True)
            return False
        gate = "PASS" if record["gate_passed"] else "fail"
        ctx_ref = record.get("multi_chunk_ctx")
        print(
            f"{prog} ✓ {name} ({ctx_ref}/[{record.get('multi_chunk_n')}]) "
            f"gate={gate} count={record['count']} degree={record['full_match_degree']}",
            flush=True,
        )
        completed.add(name)
        return True

    def _emit(ctx_id: str, case: dict, n: int, r: dict, attempts: int) -> None:
        count = r["counts"].get(n)
        if count is None:
            _emit_record({
                "case": case["name"],
                "query": case["query"],
                "error": f"multi-chunk: 目标块 [{n}] 未在模型输出 {r.get('content')!r}",
                "multi_chunk_ctx": ctx_id,
                "attempts": attempts,
            })
            return
        degree = 1.0 if count > 0 else 0.0
        record = {
            "case": case["name"],
            "chunk_index": case.get("chunk_index"),
            "query": case["query"],
            "expect_relevant": bool(case.get("expect_relevant")),
            "model": r.get("raw_model") or args.model,
            "source": case.get("source"),
            "clearly_zero": False,
            "multi_chunk_ctx": ctx_id,
            "multi_chunk_n": n,
            "attempts": attempts,
            "full_match_degree": degree,
            "gate_passed": degree >= SEMANTIC_MATCH_THRESHOLD,
            "count": count,
            "content": r.get("content"),
            "finish_reason": r.get("finish_reason"),
            "usage": r.get("usage"),
        }
        if case.get("disputed"):
            record["disputed"] = True
            if case.get("dispute_note"):
                record["dispute_note"] = case["dispute_note"]
        _emit_record(record)

    # 逐 Context 请求；同组任一目标块缺数只影响该条，不中断同组其它（不静默降级）。
    ctx_results: List[Tuple[dict, Optional[dict], Optional[BaseException], int]] = []
    for group in groups:
        if stop.is_set():
            break
        r, err, attempts = _run_group(group)
        ctx_results.append((group, r, err, attempts))
    for group, r, err, attempts in ctx_results:
        for i, case, n in group["targets"]:
            if r is None:
                _emit_record({
                    "case": case["name"],
                    "query": case["query"],
                    "error": f"multi-chunk: {err}",
                    "multi_chunk_ctx": group["ctx_id"],
                    "attempts": attempts,
                    "request_failed": True,
                })
            else:
                _emit(group["ctx_id"], case, n, r, attempts)
    case_s, req_s = format_fail_stats(all_results)
    print(f"失败统计：{case_s}" + (f"；{req_s}" if req_s else ""), flush=True)


def format_fail_stats(results: List[dict]) -> Tuple[str, Optional[str]]:
    """case 级 error 比例；有 multi_chunk_ctx 时再给请求（组）级首次/最终失败。"""
    n = len(results)
    err_n = sum(1 for r in results if r.get("error"))
    case_s = f"error={err_n}/{n}" + (f"（{err_n / n:.1%}）" if n else "")
    groups: Dict[Tuple[str, str], Dict[str, Any]] = {}
    for r in results:
        ctx = r.get("multi_chunk_ctx")
        if not ctx:
            continue
        key = (str(ctx), str(r.get("query") or ""))
        g = groups.setdefault(key, {"attempts": 1, "request_failed": False})
        att = r.get("attempts")
        if isinstance(att, int):
            g["attempts"] = max(g["attempts"], att)
        if r.get("request_failed"):
            g["request_failed"] = True
    if not groups:
        return case_s, None
    g_n = len(groups)
    term_fail = sum(1 for g in groups.values() if g["request_failed"])
    first_fail = sum(
        1 for g in groups.values() if g["attempts"] > 1 or g["request_failed"]
    )
    req_s = (
        f"请求 {g_n} 组，首次失败 {first_fail}（{first_fail / g_n:.1%}），"
        f"最终失败 {term_fail}"
    )
    return case_s, req_s


def write_review_markdown(
    results: List[dict], path: Path, clearly_zero: bool
) -> None:
    tn = tp = fp = fn = 0
    lines = [
        "# 远程 relevance 对照表",
        "",
        f"提示词变量 `clearly_zero` = **{clearly_zero}**",
        f"门控：解析 count 后 `full_match_degree = 1.0 if count > 0 else 0.0`，阈值 `{SEMANTIC_MATCH_THRESHOLD}`。",
        "",
        "| case | expect | disputed | gate | count | degree | verdict |",
        "|---|---|---|---|---:|---:|---|",
    ]
    for r in results:
        if r.get("error"):
            lines.append(
                f"| {r.get('case')} | {r.get('expect_relevant')} |  | — | — | — | error |"
            )
            continue
        expect = bool(r.get("expect_relevant"))
        passed = bool(r.get("gate_passed"))
        if expect and passed:
            tp += 1
            verdict = "OK"
        elif expect and not passed:
            fn += 1
            verdict = "**门控漏检**"
        elif (not expect) and not passed:
            tn += 1
            verdict = "**拒识OK**"
        else:
            fp += 1
            verdict = "**误放行**"
        note = "yes" if r.get("disputed") else ""
        lines.append(
            f"| {r.get('case')} | {expect} | {note} | "
            f"{'PASS' if passed else 'fail'} | {r.get('count', '—')} | "
            f"{r.get('full_match_degree', '—')} | {verdict} |"
        )
    total = tn + tp + fp + fn
    acc = (tn + tp) / total if total else 0.0
    case_s, req_s = format_fail_stats(results)
    summary = [
        f"汇总：TN（拒识对）={tn} TP（正检）={tp} "
        f"FP（误检）={fp} FN（漏检）={fn} acc={acc:.1%}（n={total}）；{case_s}",
    ]
    if req_s:
        summary.append(req_s)
    summary.append("")
    lines[4:4] = summary
    path.parent.mkdir(parents=True, exist_ok=True)
    path.write_text("\n".join(lines) + "\n", encoding="utf-8")
    print(f"✅ 对照表已写入 {path}")


def main() -> None:
    parser = argparse.ArgumentParser(
        description="远程 Chat API 相关性评测（默认 multi-chunk，与线上 v2 一致）"
    )
    parser.add_argument("-c", "--cases", type=Path, required=True, help="用例 JSON 数组")
    parser.add_argument("-o", "--output", type=Path, default=None, help="结果 JSONL（可续跑）")
    parser.add_argument("--review-md", type=Path, default=None, help="对照表 Markdown")
    parser.add_argument("--review-only", action="store_true", help="仅从 JSONL 生成对照表")
    parser.add_argument(
        "--model",
        default=DEFAULT_MODEL,
        help=f"模型 id（默认 {DEFAULT_MODEL}；OpenRouter / HF / 官方写法可能不同）",
    )
    parser.add_argument(
        "--clearly-zero",
        action="store_true",
        help=f'仅 --single-chunk：提示词追加 "{CLEARLY_ZERO_SENTENCE}"',
    )
    chunk_mode = parser.add_mutually_exclusive_group()
    chunk_mode.add_argument(
        "--multi-chunk",
        dest="chunk_mode",
        action="store_const",
        const="multi",
        help="多 chunk 模式（默认，可省略）：以文章切片 Context 为单位请求，与线上 v2 一致",
    )
    chunk_mode.add_argument(
        "--single-chunk",
        dest="chunk_mode",
        action="store_const",
        const="single",
        help="废弃：逐块单条请求（旧路径）",
    )
    parser.set_defaults(chunk_mode="multi")
    parser.add_argument(
        "--ctx-max",
        type=int,
        default=MULTI_CHUNK_MAX,
        metavar="N",
        help="全文章切片模式下每个 Context 最多容纳的 chunk 数（默认 %(default)s）。文章不足 N 块时整篇载入。"
        "结论（Hy3，现有全部 test case，配对 McNemar）：≤32 chunk 内 32 切片稳定、与单 chunk 基线无显著差异，"
        "小切片 16/24 会显著损伤；32 chunk 以上尚无明确数据。故默认 32。",
    )
    parser.add_argument(
        "--url",
        default=DEFAULT_API_BASE,
        help=f"OpenAI 兼容 API base（默认 {DEFAULT_API_BASE}）",
    )
    parser.add_argument("--hf-token", default=None, help="API token（兼容旧参数名）")
    parser.add_argument("--token", default=None, help="API token（优先于环境变量）")
    parser.add_argument("--retries", type=int, default=3)
    parser.add_argument("--timeout", type=int, default=180)
    parser.add_argument("--sleep", type=float, default=0.2, help="每条请求前额外等待秒（每 worker）")
    parser.add_argument(
        "-j",
        "--jobs",
        type=int,
        default=1,
        help="仅 --single-chunk：并发数（用例彼此独立；默认 1）",
    )
    args = parser.parse_args()

    _load_env_file(Path(__file__).resolve().parents[1] / ".env")

    if args.review_only:
        if not args.output or not args.review_md:
            print("错误: --review-only 需要 -o 与 --review-md")
            sys.exit(1)
        results = _load_jsonl(args.output)
        cz = bool(results[0].get("clearly_zero")) if results else args.clearly_zero
        write_review_markdown(results, args.review_md, cz)
        return

    cases = load_cases(args.cases)
    token = (
        args.token
        or args.hf_token
        or os.environ.get(OPENROUTER_TOKEN_ENV)
        or os.environ.get(HF_TOKEN_ENV)
    )
    if not token:
        print(f"错误: 需要 --token / {OPENROUTER_TOKEN_ENV} / {HF_TOKEN_ENV}")
        sys.exit(1)

    if args.clearly_zero and args.chunk_mode != "single":
        print("错误: --clearly-zero 仅用于已废弃的 --single-chunk")
        sys.exit(1)

    jobs = max(1, int(args.jobs))
    print(
        f"已加载 {len(cases)} 条用例；model={args.model}；"
        f"chunk_mode={args.chunk_mode}；"
        f"clearly_zero={args.clearly_zero}；jobs={jobs}"
    )

    completed = set()
    all_results: list = []
    if args.output and args.output.exists():
        all_results = _load_jsonl(args.output)
        completed = {r["case"] for r in all_results if "case" in r}
        print(f"已加载 {len(all_results)} 条历史，跳过 {len(completed)} 个 case")

    pending: List[Tuple[int, dict]] = [
        (i, case) for i, case in enumerate(cases) if case["name"] not in completed
    ]
    skipped = len(cases) - len(pending)
    if skipped:
        print(f"⏭ 跳过已完成 {skipped} 条", flush=True)

    if args.output:
        args.output.parent.mkdir(parents=True, exist_ok=True)
    write_lock = threading.Lock()
    stop = threading.Event()
    done_n = skipped

    def _commit(i: int, case: dict, record: dict) -> bool:
        """写入并打印；返回是否应继续（False=失败中断）。"""
        nonlocal done_n
        name = case["name"]
        done_n += 1
        prog = f"[{done_n}/{len(cases)}]"
        all_results.append(record)
        if args.output:
            with write_lock:
                _append_record(args.output, record)
        if record.get("error"):
            print(f"{prog} ✗ {name}: {record['error']}", flush=True)
            return False
        gate = "PASS" if record["gate_passed"] else "fail"
        print(
            f"{prog} ✓ {name} gate={gate} count={record['count']} "
            f"degree={record['full_match_degree']}",
            flush=True,
        )
        completed.add(name)
        return True

    if args.chunk_mode == "multi":
        _run_multi_chunk_main(
            args, pending, completed, all_results,
            write_lock=write_lock, stop=stop,
            done_n=done_n, cases=cases, token=token,
        )
    else:
        def _work(item: Tuple[int, dict]) -> Tuple[int, dict, dict]:
            i, case = item
            if stop.is_set():
                return i, case, {"case": case["name"], "error": "skipped after failure"}
            if args.sleep > 0:
                time.sleep(args.sleep)
            record = run_one(
                args.url,
                args.model,
                case,
                clearly_zero=args.clearly_zero,
                token=token,
                timeout=args.timeout,
                max_retries=args.retries,
            )
            return i, case, record

        if jobs == 1:
            for item in pending:
                if stop.is_set():
                    break
                i, case, record = _work(item)
                if not _commit(i, case, record):
                    print("⚠ 失败中断后续", flush=True)
                    stop.set()
                    break
        else:
            with ThreadPoolExecutor(max_workers=jobs) as ex:
                futures = {ex.submit(_work, item): item for item in pending}
                for fut in as_completed(futures):
                    i, case, record = fut.result()
                    if record.get("error") == "skipped after failure":
                        continue
                    if not _commit(i, case, record):
                        print("⚠ 失败中断后续（已提交的 in-flight 仍会跑完）", flush=True)
                        stop.set()
                        for f in futures:
                            f.cancel()
                        break

    if args.output:
        print(f"\n✅ 结果已写入 {args.output}（共 {len(all_results)} 条）")
    if args.review_md and all_results:
        order = {c["name"]: i for i, c in enumerate(cases)}
        ordered = sorted(
            all_results,
            key=lambda r: order.get(r.get("case") or "", 10**9),
        )
        write_review_markdown(ordered, args.review_md, args.clearly_zero)


if __name__ == "__main__":
    main()