scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 182

test: cover _finalize_adaptive_router_if_configured

Browse source 
Router coverage check flagged this method as untested. Adds two cases:
- initializes AdaptiveRouter from model_list and is idempotent on re-entry
- no-op when no adaptive deployments are configured

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Krrish Dholakia 2026-04-20 07:43:53 -07:00
parent 3cf0460d8c
commit f0efc5f670
1 changed files with 53 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
tests/test_litellm/router_strategy/adaptive_router/test_router_dispatch.py
View file

@ -378,3 +378,56 @@ def test_init_adaptive_router_rejects_duplicate_model_name():
r.init_adaptive_router_deployment(deployment=deployment)
with pytest.raises(ValueError, match="already exists"):
r.init_adaptive_router_deployment(deployment=deployment)
def test_finalize_adaptive_router_if_configured_initializes_and_is_idempotent():
"""`_finalize_adaptive_router_if_configured` walks the model_list, builds an
AdaptiveRouter for each adaptive deployment, and is a safe no-op on
re-entry (models already in self.adaptive_routers are skipped)."""
r = Router(
model_list=[
{
"model_name": "fast",
"litellm_params": {"model": "openai/gpt-4o-mini"},
"model_info": {"input_cost_per_token": 0.00000015},
},
{
"model_name": "smart",
"litellm_params": {"model": "openai/gpt-4o"},
"model_info": {"input_cost_per_token": 0.0000025},
},
{
"model_name": "my-router",
"litellm_params": {
"model": "auto_router/adaptive_router",
"adaptive_router_config": {
"available_models": ["fast", "smart"],
},
},
},
]
)
# Router __init__ already called _finalize_adaptive_router_if_configured.
assert "my-router" in r.adaptive_routers
original = r.adaptive_routers["my-router"]
# Calling again must be idempotent: the existing AdaptiveRouter instance
# is preserved, not rebuilt.
r._finalize_adaptive_router_if_configured()
assert r.adaptive_routers["my-router"] is original
def test_finalize_adaptive_router_if_configured_noop_when_none_configured():
"""With no adaptive deployments in model_list, the finalizer leaves
`adaptive_routers` empty."""
r = Router(
model_list=[
{
"model_name": "fast",
"litellm_params": {"model": "openai/gpt-4o-mini"},
}
]
)
r._finalize_adaptive_router_if_configured()
assert r.adaptive_routers == {}