muse-agent-example/docs/write-chapter/step2a_retrofit_ch1_context.py

#!/usr/bin/env python3
"""真写一章 · 阶段二a 补建：第 1 章的开篇上下文与冻结记录。

原则：不存在没有上下文的写作。第 1 章之前也有设定、大纲、细纲——它的上下文是
asOf=0（开篇前冻结线）：证据只有设定包与细纲宣告的新事实，历史正文基线为空。
此前第 1 章按"基线章无上下文"的旧口径写成，看板运行详情因此缺一块上下文冻结；
本脚本按合同补建真实的第 1 章上下文并落冻结行（append-only，幂等），补齐记账。

只组装 + 落冻结，不改写第 1 章正文（正文早已接受为正式内容）。
"""
import json
import sys
import uuid
from datetime import datetime, timezone
from pathlib import Path

SCRIPT_DIR = Path(__file__).resolve().parent
AGENT_ROOT = SCRIPT_DIR.parents[1]
SKILLS = AGENT_ROOT / ".claude" / "skills"
for sub in (
    "assemble-context/scripts",
    "write-next-chapter/scripts",
    "record-run-evidence/scripts",
    "access-database/scripts",
):
    p = str(SKILLS / sub)
    if p not in sys.path:
        sys.path.insert(0, p)

from db import connect, DSN  # noqa: E402
from assemble_writer_context import assemble_context  # noqa: E402
from retrieve_writer_sources import (  # noqa: E402
    ProductionCardIndexRepository, FrozenProseRepository,
    build_retrieval_plan, retrieve_writer_sources,
)
from persist_context_freeze import persist_freeze  # noqa: E402
from run_writer import calculate_dynamic_output_contract  # noqa: E402

WORK_ID = 12
TARGET_CHAPTER = 1
AS_OF = 0
SETTING_SOURCE_REF = {"sourceId": "planning-section:3", "sourceVersion": "v1"}
OUTLINE_SOURCE_REF = {"sourceId": "fine-outline:work12-ch1", "sourceVersion": "v1"}


def main():
    run_id = f"retrofit-work12-ch1-context-{uuid.uuid4().hex[:8]}"
    generated_at = datetime.now(timezone.utc).isoformat()

    with connect(readonly=True) as conn:
        setting_row = conn.execute(
            "SELECT payload FROM example_planning_section WHERE id=3 AND work_id=%s "
            "AND section_type='setting' AND state='confirmed' AND deleted=false",
            (WORK_ID,)).fetchone()
        outline_row = conn.execute(
            "SELECT payload FROM example_planning_section WHERE work_id=%s AND "
            "section_type='fine_outline' AND target_chapter=1 AND state='confirmed' AND "
            "deleted=false ORDER BY version DESC LIMIT 1", (WORK_ID,)).fetchone()
    if not setting_row or not outline_row:
        raise SystemExit("设定包或第1章细纲缺失，无法补建上下文。")
    setting = setting_row[0] if isinstance(setting_row[0], dict) else json.loads(setting_row[0])
    fine_outline = outline_row[0] if isinstance(outline_row[0], dict) else json.loads(outline_row[0])

    # 第1章细纲的 declaredNewFacts 是旧形状（纯字符串）；合同要 {factId,text,sourceRef}。
    # 不重写已确认规划行，在读取侧按合同投影。
    normalized_declared = []
    for index, fact in enumerate(fine_outline.get("declaredNewFacts", []), 1):
        if isinstance(fact, dict):
            normalized_declared.append(fact)
        else:
            normalized_declared.append({
                "factId": f"fine-outline-ch1:{index}", "text": str(fact),
                "sourceRef": OUTLINE_SOURCE_REF,
            })
    fine_outline = {**fine_outline, "declaredNewFacts": normalized_declared}

    # 权威事实证据：设定包（正式设定）+ 细纲宣告的新事实
    authoritative_facts = []
    for key, label in (("world", "世界观"), ("power_system", "力量体系"),
                       ("core_premise", "核心设定")):
        if setting.get(key):
            authoritative_facts.append({
                "factId": f"setting:{key}", "fact": f"{label}：{setting[key]}",
                "sourceType": "formal_setting", "sourceRef": SETTING_SOURCE_REF,
                "riskLevel": "low",
            })
    for index, fact in enumerate(fine_outline.get("declaredNewFacts", []), 1):
        text = fact["text"] if isinstance(fact, dict) else str(fact)
        authoritative_facts.append({
            "factId": f"fine-outline-ch1:{index}", "fact": text,
            "sourceType": "fine_outline_declared_new", "sourceRef": OUTLINE_SOURCE_REF,
            "riskLevel": "low",
        })

    # 检索计划与执行（asOf=0；新书无卡，卡线诚实返空；事实线取权威事实）
    token_budget = {"maxContextChars": 200000}
    plan = build_retrieval_plan(
        run_id=run_id, work_id=WORK_ID, target_chapter=TARGET_CHAPTER, as_of=AS_OF,
        fine_outline=fine_outline, card_index_version="knowledge-index-v1",
        prose_index_version="content-block-v1", token_budget=token_budget)
    retrieval = retrieve_writer_sources(
        plan=plan, card_repository=ProductionCardIndexRepository(),
        prose_repository=FrozenProseRepository(dsn=DSN, tenant_id=0),
        authoritative_facts=authoritative_facts)
    print(f"检索: 卡={len(retrieval['cards'])} 事实={len(retrieval['factEvidence'])} "
          f"原文={len(retrieval['proseEvidence'])}（asOf=0，历史正文必为空）")

    # 组装第 1 章上下文（asOf=0，无历史正文基线）并落冻结
    output_contract = calculate_dynamic_output_contract(
        fine_outline=fine_outline, recent_chapter_bodies=[])
    narrative_state = {
        "time": "故事开篇之前",
        "location": "深渊星域",
        "characterPositions": {},
        "immediateSituation": "一切尚未发生；按设定与第1章细纲展开。",
    }
    authorization_snapshot = {"snapshotId": "auth-work12-production-v1",
                              "allowedPurpose": "production_generation",
                              "verifiedAt": generated_at, "sourceVersion": "v1"}
    assembled = assemble_context(
        run_id=run_id, attempt=1, mode="production", purpose="production",
        quality_policy_version="writer-production-v1", work_id=WORK_ID,
        target_chapter=TARGET_CHAPTER, as_of=AS_OF, source_version="outline@v1",
        authorization_snapshot=authorization_snapshot, source_status="active",
        retrieval_plan=plan, retrieval_result=retrieval, fine_outline=fine_outline,
        narrative_state=narrative_state, recent_chapters=[],
        output_contract=output_contract, token_budget=token_budget,
        pattern_references=[], generated_at=generated_at,
        evidence_strategy="production_dual_evidence")
    (SCRIPT_DIR / "artifacts").mkdir(exist_ok=True)
    (SCRIPT_DIR / "artifacts" / f"{run_id}-writer-context.json").write_text(
        assembled["contextJson"], encoding="utf-8")
    frozen = persist_freeze(assembled)
    print(f"第1章上下文冻结: {json.dumps(frozen, ensure_ascii=False)}")
    print(f"事实证据 {len(assembled['context']['factEvidence'])} 条，"
          f"正文证据 {len(assembled['context']['proseEvidence'])} 条（应为 0）")
    print(f"RUN_ID={run_id}")


if __name__ == "__main__":
    main()