scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

Merge d6d18367e3 into 784754ca48

Browse source
This commit is contained in:
José Rojas 2026-11-04 20:54:17 +00:00 • committed by GitHub
commit 356985af31
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/router_strategy/complexity_router/config.py
View file

@ -487,8 +487,8 @@ DEFAULT_TOKEN_THRESHOLDS: Final[dict[str, int]] = {
DEFAULT_TIER_MODELS: Final[dict[str, str]] = {
"SIMPLE": "gpt-4o-mini",
"MEDIUM": "gpt-4o",
"COMPLEX": "claude-sonnet-4-20250514",
"REASONING": "claude-sonnet-4-20250514",
"COMPLEX": "claude-sonnet-5-5",
"REASONING": "claude-sonnet-5-5",
}

7
tests/unit/router_strategy/test_complexity_router.py
View file

@ -285,6 +285,13 @@ class TestComplexityRouterInit:
assert router.config.tiers == DEFAULT_COMPLEXITY_CONFIG.tiers
assert router.config is not DEFAULT_COMPLEXITY_CONFIG # Not a singleton
@pytest.mark.usefixtures("local_model_cost_map")
def test_default_tiers_route_to_models_litellm_knows(self):
unknown: Final = {
tier: model for tier, model in DEFAULT_COMPLEXITY_CONFIG.tiers.items() if model not in litellm.model_cost
}
assert unknown == {}
def test_init_with_default_model(self, mock_router_instance, basic_config):
"""Test initialization with default_model override."""
router = ComplexityRouter(