mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
"""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_unchanged() -> None:
|
||
|
|
# CEO preference: stay on Opus 4.6 (4.7/4.8 not preferred).
|
||
|
|
assert MODEL_MAP["opus"] == "claude-opus-4-6"
|
||
|
|
|
||
|
|
|
||
|
|
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_qa_role_routes_to_haiku() -> None:
|
||
|
|
# Phase 2: QA is mechanical gate work → cheapest tier.
|
||
|
|
assert ROLE_MODEL_MAP["qa"] == "haiku"
|
||
|
|
|
||
|
|
|
||
|
|
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"
|