"""MODEL_MAP resolves the short aliases to current, priced model ids.""" from __future__ import annotations from roboco.billing.pricing import calculate_cost from roboco.models.runtime import MODEL_MAP, ROLE_MODEL_MAP _M = 1_000_000 def test_sonnet_alias_resolves_to_sonnet_5() -> None: assert MODEL_MAP["sonnet"] == "claude-sonnet-5" def test_opus_alias_resolves_to_opus_4_8() -> None: # CEO upgrade to the newest Opus tier ("Opus 5" does not exist — the # Claude 5 family is Fable/Mythos/Sonnet; Opus tops out at 4.8). assert MODEL_MAP["opus"] == "claude-opus-4-8" def test_sonnet_5_is_priced() -> None: # Guard against pointing an alias at an unpriced model (silent $0 cost-count). cost = calculate_cost(MODEL_MAP["sonnet"], tokens_input=_M, tokens_output=0) assert cost > 0.0 def test_opus_is_priced() -> None: # Same silent-$0 guard for the opus alias. cost = calculate_cost(MODEL_MAP["opus"], tokens_input=_M, tokens_output=0) assert cost > 0.0 def test_no_lifecycle_role_routes_below_the_structured_verb_floor() -> None: # Retired haiku from lifecycle roles (2026-07-24): a haiku QA/documenter # can't emit the structured review envelopes (criteria_verified, findings) # and loops without passing. No role's default may be a haiku-class tier. for role, model in ROLE_MODEL_MAP.items(): assert "haiku" not in model.lower(), f"{role} routes to a below-floor {model}" def test_main_pm_role_routes_to_sonnet() -> None: # Phase 2 experiment: main_pm off Opus (Sonnet 5 cache-write ~12x cheaper). assert ROLE_MODEL_MAP["main_pm"] == "sonnet"