191 lines
7.7 KiB
Python
191 lines
7.7 KiB
Python
# ============================================================
|
||
# 兜底验证器 v1（moduleId: core-fallback-verify, 可重生 ✅）
|
||
# 《Pi-Agent兜底能力详细方案》§4.1/§4.5 + GOAL-P2 交付 4 + P2-DESIGN §4：
|
||
# 兜底执行后的 world diff、验证规则（行数/数量对账）、报告生成。
|
||
# 铁律：报告里的每个数字都只许来自冻结快照（cp_before/cp_after 的
|
||
# world 深拷贝），绝不引用 Pi 报告文本或 Pi 自述。
|
||
# ============================================================
|
||
from __future__ import annotations
|
||
|
||
import hashlib
|
||
import json
|
||
from pathlib import Path
|
||
from typing import Any
|
||
|
||
# 对账覆盖的业务主数据表（分表 diff 的固定口径）
|
||
_DIFF_TABLES = (
|
||
"salesOrders", "flexOrders", "materials", "flexMaterials",
|
||
"flexEquipment", "flexMolds", "flexOperations", "flexRoutings",
|
||
"flexBom", "productionOrders", "workOrders",
|
||
)
|
||
|
||
# 数量对账类规则的数据源字段（存在即计入 quantityDelta）
|
||
_QUANTITY_FIELDS = ("quantity", "stock")
|
||
|
||
|
||
def _canonical(item: Any) -> str:
|
||
"""条目的规范 JSON（排序键 + 紧凑分隔符），modified 判定与指纹共用口径。"""
|
||
return json.dumps(item, ensure_ascii=True, sort_keys=True,
|
||
separators=(",", ":"), default=str)
|
||
|
||
|
||
def _entry_key(item: dict) -> tuple:
|
||
"""条目身份键：数值 id 优先，其次 orderNo/code，最后整条 canonical（防无键表）。"""
|
||
if isinstance(item, dict):
|
||
if isinstance(item.get("id"), int):
|
||
return ("id", item["id"])
|
||
if item.get("orderNo"):
|
||
return ("orderNo", str(item["orderNo"]))
|
||
if item.get("code"):
|
||
return ("code", str(item["code"]))
|
||
return ("json", _canonical(item))
|
||
|
||
|
||
def _quantity_of(item: dict) -> float:
|
||
total = 0.0
|
||
for field in _QUANTITY_FIELDS:
|
||
value = item.get(field) if isinstance(item, dict) else None
|
||
if isinstance(value, (int, float)):
|
||
total += float(value)
|
||
return total
|
||
|
||
|
||
def world_diff(before: dict, after: dict) -> dict:
|
||
"""分表 diff：{table: {"added","removed","modified","quantityDelta"}}。
|
||
|
||
modified 判定 = 同身份键条目内容变化（canonical JSON 不等）；
|
||
quantityDelta 对含 quantity/stock 字段的条目求和（added 减 removed）。
|
||
"""
|
||
diff: dict[str, dict] = {}
|
||
for table in _DIFF_TABLES:
|
||
before_rows = before.get(table) or []
|
||
after_rows = after.get(table) or []
|
||
before_map = {_entry_key(r): r for r in before_rows if isinstance(r, dict)}
|
||
after_map = {_entry_key(r): r for r in after_rows if isinstance(r, dict)}
|
||
added_keys = [k for k in after_map if k not in before_map]
|
||
removed_keys = [k for k in before_map if k not in after_map]
|
||
modified = sum(
|
||
1 for k in before_map.keys() & after_map.keys()
|
||
if _canonical(before_map[k]) != _canonical(after_map[k])
|
||
)
|
||
qty_delta = (
|
||
sum(_quantity_of(after_map[k]) for k in added_keys)
|
||
- sum(_quantity_of(before_map[k]) for k in removed_keys)
|
||
)
|
||
if added_keys or removed_keys or modified:
|
||
diff[table] = {
|
||
"added": len(added_keys),
|
||
"removed": len(removed_keys),
|
||
"modified": modified,
|
||
"quantityDelta": qty_delta,
|
||
}
|
||
return diff
|
||
|
||
|
||
def check_expectations(plan: dict, diff: dict) -> list[dict]:
|
||
"""逐条比对计划 expected（结构化字段）与实际 diff。
|
||
|
||
返回 [{"step","expect","actual","ok"} ...]；容差 = 0——任一声明字段不等即 ok=False，
|
||
调用方据此把报告 verdict 判为 MISMATCH（显式，不圆场）。
|
||
"""
|
||
checks: list[dict] = []
|
||
for step in plan.get("steps") or []:
|
||
seq = step.get("seq")
|
||
for expect in step.get("expected") or []:
|
||
if not isinstance(expect, dict) or not expect.get("table"):
|
||
continue
|
||
table = str(expect["table"])
|
||
actual = (diff.get(table) or {}).copy()
|
||
actual.setdefault("added", 0)
|
||
actual.setdefault("removed", 0)
|
||
actual.setdefault("modified", 0)
|
||
ok = True
|
||
for field in ("added", "removed", "modified"):
|
||
if field in expect and expect[field] is not None \
|
||
and int(expect[field]) != int(actual.get(field) or 0):
|
||
ok = False
|
||
checks.append({
|
||
"step": seq,
|
||
"expect": {k: expect[k] for k in ("table", "added", "removed", "modified")
|
||
if k in expect},
|
||
"actual": {"table": table,
|
||
**{k: actual.get(k, 0) for k in ("added", "removed", "modified")}},
|
||
"ok": ok,
|
||
})
|
||
return checks
|
||
|
||
|
||
def diff_summary_lines(diff: dict) -> list[str]:
|
||
"""分表 diff 的人类可读摘要行（回复文案与报告共用）。"""
|
||
lines = []
|
||
for table, d in diff.items():
|
||
parts = []
|
||
if d["added"]:
|
||
parts.append(f"+{d['added']}")
|
||
if d["removed"]:
|
||
parts.append(f"-{d['removed']}")
|
||
if d["modified"]:
|
||
parts.append(f"~{d['modified']}")
|
||
line = f"{table} {'/'.join(parts)}"
|
||
if d.get("quantityDelta"):
|
||
line += f"（数量净变化 {d['quantityDelta']:g}）"
|
||
lines.append(line)
|
||
return lines or ["（无业务主数据变化）"]
|
||
|
||
|
||
def build_report(run_dir: Path, plan: dict, *,
|
||
cp_before_id: str, cp_after_id: str,
|
||
before_world: dict, after_world: dict,
|
||
checks: list[dict]) -> Path:
|
||
"""生成 outbox/verify-report.md：计划摘要 / 分表 diff 表 / 对账结论 / 证据引用。
|
||
|
||
数字来源 = 两个冻结快照的 world（调用方保证传入的是 checkpoint 仓内深拷贝），
|
||
返回报告路径。verdict：全部对账通过 = PASS；任一不符 = MISMATCH（显式标注）。
|
||
"""
|
||
diff = world_diff(before_world, after_world)
|
||
verdict = "PASS" if all(c["ok"] for c in checks) else "MISMATCH"
|
||
|
||
lines = [
|
||
"# 兜底执行验证报告",
|
||
"",
|
||
f"- 运行：{plan.get('runId') or ''}",
|
||
f"- 场景：{plan.get('scenario') or ''} · 步骤数 {len(plan.get('steps') or [])}",
|
||
f"- 检查点：执行前 `{cp_before_id}` → 执行后 `{cp_after_id}`",
|
||
"",
|
||
"## 分表 diff（before → after）",
|
||
"",
|
||
"| 表 | 新增 | 移除 | 修改 | 数量净变化 |",
|
||
"|----|------|------|------|-----------|",
|
||
]
|
||
for table in _DIFF_TABLES:
|
||
d = diff.get(table)
|
||
if not d:
|
||
continue
|
||
lines.append(f"| {table} | {d['added']} | {d['removed']} | {d['modified']} "
|
||
f"| {d['quantityDelta']:g} |")
|
||
if not diff:
|
||
lines.append("| （无变化） | 0 | 0 | 0 | 0 |")
|
||
lines += ["", "## 对账结论", ""]
|
||
if checks:
|
||
for c in checks:
|
||
mark = "✅" if c["ok"] else "❌"
|
||
lines.append(f"- {mark} 步骤{c['step']} 期望 {json.dumps(c['expect'], ensure_ascii=False)}"
|
||
f" · 实际 {json.dumps(c['actual'], ensure_ascii=False)}")
|
||
else:
|
||
lines.append("- （计划未声明结构化预期，仅呈现实际 diff）")
|
||
lines += [
|
||
"",
|
||
f"**verdict: {verdict}**",
|
||
"",
|
||
f"本报告全部数字来自检查点 {cp_before_id} 与 {cp_after_id} 的冻结快照。",
|
||
]
|
||
out = Path(run_dir) / "outbox" / "verify-report.md"
|
||
out.parent.mkdir(parents=True, exist_ok=True)
|
||
out.write_text("\n".join(lines) + "\n", encoding="utf-8")
|
||
return out
|
||
|
||
|
||
def report_fingerprint(text: str) -> str:
|
||
"""报告文本 sha256（审计 rationale 引用用，不落全文）。"""
|
||
return hashlib.sha256(text.encode("utf-8")).hexdigest()[:16]
|