scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

test(router): cover complexity-router tag selection helpers directly

Browse source
This commit is contained in:
Devin AI 2026-07-17 05:28:44 +00:00
parent bcf3c61611
commit a8412e540d
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/router_strategy/test_complexity_router.py
View file

@ -1609,6 +1609,24 @@ class TestComplexityRouterTagBasedRouting:
assert result is not None
assert result.model == "cn-simple"
def test_select_complexity_router_picks_by_tag(self):
router = self._make_router(enable_tag_filtering=True)
assert router._select_complexity_router(model="unknown", request_kwargs={}) is None
selected = router._select_complexity_router(
model="smart-router", request_kwargs={"metadata": {"tags": ["row"]}}
)
assert selected is not None
assert selected.model_id == "row-router"
def test_resolve_alias_index_matches_selected_model_id(self):
router = self._make_router(enable_tag_filtering=True)
assert router._resolve_alias_index(model="smart-router", selected_model_id="row-router") == 1
assert router._resolve_alias_index(model="smart-router", selected_model_id=None) == 0
assert router._resolve_alias_index(model="unknown", selected_model_id=None) is None
class TestAdaptiveSoftFloors:
def test_adaptive_defaults_use_cost_weighted_cold_policy(self):