Add Krea2 next test plans

This commit is contained in:
2026-06-29 03:55:17 +02:00
parent 333f4752f6
commit 3a09210f71
3 changed files with 65 additions and 1 deletions
+45 -1
View File
@@ -66,6 +66,34 @@ def coverage_summary() -> dict[str, Any]:
}
def next_test_plans() -> list[dict[str, Any]]:
rows_by_key = {str(row.get("key")): row for row in coverage_rows()}
plans: list[dict[str, Any]] = []
for key in coverage_summary()["next_test_candidates"]:
variant = krea2_pose_variant_catalog.get_variant(key)
if not variant:
continue
row = rows_by_key.get(key, {})
evidence = variant.get("evidence") or {}
plans.append(
{
"key": key,
"family": variant.get("family") or "",
"action_family": variant.get("action_family") or "",
"status": variant.get("status") or "",
"coverage_state": row.get("coverage_state") or "",
"canonical_geometry": variant.get("canonical_geometry") or "",
"prompt_cues": list(variant.get("prompt_cues") or []),
"avoid_cues": list(variant.get("avoid_cues") or []),
"reference_paths": [str(path) for path in krea2_pose_variant_catalog.reference_paths(key)],
"generator_hook": variant.get("generator_hook") or {},
"guide_section": evidence.get("guide_section") or "",
"notes": evidence.get("notes") or "",
}
)
return plans
def markdown_report() -> str:
lines = [
"# Krea2 Pose Variant Coverage",
@@ -87,5 +115,21 @@ def markdown_report() -> str:
*[f"- {key}" for key in summary["next_test_candidates"]],
]
)
plans = next_test_plans()
if plans:
lines.extend(["", "## Next Test Plans"])
for plan in plans:
lines.extend(
[
"",
f"### {plan['key']}",
"",
f"- Geometry: {plan['canonical_geometry']}",
f"- References: {', '.join(plan['reference_paths']) or 'none'}",
"- Prompt cues:",
*[f" - {cue}" for cue in plan["prompt_cues"]],
"- Avoid cues:",
*[f" - {cue}" for cue in plan["avoid_cues"]],
]
)
return "\n".join(lines)