Antigravity エージェントの揺れる出力を CI で回帰検証する

import re
import json
 
# 環境由来の揺れを安定トークンに置換する
NORMALIZERS = [
    (re.compile(r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:\.\d+)?Z?"), "<TIMESTAMP>"),
    (re.compile(r"[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}"), "<UUID>"),
    (re.compile(r"/tmp/[^\s\"']+"), "<TMPPATH>"),
    (re.compile(r"run_id=\w+"), "run_id=<RUN_ID>"),
]
 
def normalize_text(text: str) -> str:
    for pattern, token in NORMALIZERS:
        text = pattern.sub(token, text)
    return text.strip()
 
def normalize_json(payload: dict) -> str:
    # キー順を固定し、配列内の dict も安定キーで並べ替える
    canonical = json.dumps(payload, ensure_ascii=False, sort_keys=True, indent=2)
    return normalize_text(canonical)

import os
from pathlib import Path
 
GOLDEN_DIR = Path(__file__).parent / "golden"
UPDATE = os.environ.get("UPDATE_GOLDEN") == "1"
 
def assert_against_golden(name: str, actual_raw: str):
    GOLDEN_DIR.mkdir(exist_ok=True)
    golden_path = GOLDEN_DIR / f"{name}.txt"
    actual = normalize_text(actual_raw)
 
    if UPDATE or not golden_path.exists():
        golden_path.write_text(actual, encoding="utf-8")
        return  # 記録モードでは照合しない
 
    expected = golden_path.read_text(encoding="utf-8")
    assert actual == expected, (
        f"スナップショット退行: {name}\n"
        f"--- 期待 ---\n{expected[:400]}\n"
        f"--- 実際 ---\n{actual[:400]}"
    )

def assert_structured(actual: dict, contract: dict):
    # 必須キーの存在と型だけを契約として検証する
    for key, expected_type in contract.items():
        assert key in actual, f"必須フィールド欠落: {key}"
        assert isinstance(actual[key], expected_type), (
            f"型不一致: {key} は {expected_type.__name__} を期待"
        )
    # 文面ではなく構造のスナップショットを取る
    shape = {k: type(v).__name__ for k, v in sorted(actual.items())}
    return shape
 
CONTRACT = {"title": str, "tags": list, "summary": str, "score": (int, float)}

from difflib import SequenceMatcher
 
def soft_match(actual_raw: str, expected: str, tolerance: float = 0.05) -> bool:
    actual = normalize_text(actual_raw)
    ratio = SequenceMatcher(None, actual, expected).ratio()
    drift = 1.0 - ratio
    return drift <= tolerance  # 5% 以内の揺れは許容