muse-agent-example/docs/write-chapter/step2a_retrofit_ch1_context.py
zizi 37d508ffb0 chore(workspace): 写章节脚本 step2 / step2a 接入人感五技能
step2_write_chapter.py 在 sys.path 加入 prevent-ai-flavor /
diagnose-ai-flavor,新增「1.5) 人感前置预防」环节——生成前调用
build_prevention_contract(负约束+声音账)冻结进 WriterContext,并把
diagnosis 列为 shadow 候选的伴随检测。step2a_retrofit_ch1_context.py
相应补回退补写。

脚本只做拼装与持久化触发,五技能合同与 hard gate 由各自 skill 把守。
人感体系资产层仍由 humanization/ 维护(example 自治);本脚本只消费。
2026-08-19 21:05:27 +08:00

152 lines
7.4 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/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",
"prevent-ai-flavor/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 prevent_ai_flavor import ( # noqa: E402
PreventionContractError, build_prevention_contract, persist_prevention, render_writer_constraints,
)
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"}
try:
humanization_contract = build_prevention_contract(f"work:{WORK_ID}", load_database=True)
humanization_contract["writer_constraints"] = render_writer_constraints(humanization_contract)
persist_prevention(humanization_contract)
except PreventionContractError as exc:
raise SystemExit(f"人感前置预防合同失败:{exc}") from exc
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=[], humanization_contract=humanization_contract, 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()