Extract Krea pair formatter route

2026-06-27 11:09:59 +02:00
parent 8398a97cdf
commit 176d4c9257
5 changed files with 287 additions and 162 deletions
@@ -43,6 +43,7 @@ import generation_profile_config  # noqa: E402
 import index_switch_policy  # noqa: E402
 import krea_cast  # noqa: E402
 import krea_formatter  # noqa: E402
+import krea_pair_formatter  # noqa: E402
 import location_config  # noqa: E402
 import loop_nodes  # noqa: E402
 import pair_builder  # noqa: E402
@@ -3196,6 +3197,15 @@ def smoke_krea_pair_clothing_state() -> None:
        hardcore_position_config=_action_filter("penetration_only"),
    )
    _expect_pair(pair, "krea_pair_clothing_state")
+    typed_route = krea_pair_formatter.format_insta_pair_result(
+        krea_pair_formatter.KreaPairFormatRequest(pair, "balanced", "preserve"),
+        krea_formatter._krea_pair_format_dependencies(),
+    )
+    legacy_route = krea_formatter._insta_pair_to_krea(pair, "balanced", "preserve")
+    _expect(
+        typed_route.as_tuple() == legacy_route,
+        "Typed Krea pair formatter route should match legacy wrapper output",
+    )
    krea = krea_formatter.format_krea2_prompt("", metadata_json=_json(pair), target="hardcore")
    prompt = _expect_text("krea_pair_clothing_state.krea_prompt", krea.get("krea_prompt"), 60)
    lower = prompt.lower()