File size: 940 Bytes
d3cadd5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import pytest
def test_map_model_name_downgrades_opus():
from kiro_proxy.config import map_model_name
assert map_model_name("claude-opus-4.5") == "claude-sonnet-4.5"
assert map_model_name("claude-3-opus-20240229") == "claude-sonnet-4.5"
assert map_model_name("claude-3-opus-latest") == "claude-sonnet-4.5"
assert map_model_name("claude-4-opus") == "claude-sonnet-4.5"
assert map_model_name("o1") == "claude-sonnet-4.5"
assert map_model_name("o1-preview") == "claude-sonnet-4.5"
assert map_model_name("opus") == "claude-sonnet-4.5"
@pytest.mark.asyncio
async def test_models_fallback_does_not_advertise_opus(monkeypatch):
from kiro_proxy.routers import protocols
monkeypatch.setattr(protocols.state, "get_available_account", lambda *args, **kwargs: None)
resp = await protocols.models()
model_ids = {m["id"] for m in resp.get("data", [])}
assert "claude-opus-4.5" not in model_ids
|