scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

chore(router): sync final tier routing edits

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
tin 2026-08-06 04:25:42 +00:00
parent 5bfdd07bf4
commit e7682eeab4
2 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/router_strategy/complexity_router/complexity_router.py
View file

@ -1156,18 +1156,12 @@ class ComplexityRouter(CustomLogger):
def _pick_from_tier_value(
model: str | list[str] | TierTarget, tier_key: str
) -> str: # mutable-ok: legacy pool inputs remain lists
if isinstance(model, TierTarget):
target_model: Final = model.model
if isinstance(target_model, str):
return target_model
if not target_model:
raise ValueError(f"Empty model pool for tier {tier_key}")
return random.choice(target_model)
if isinstance(model, str):
return model
if not model:
pool: Final = tier_pool(model)
if not pool:
raise ValueError(f"Empty model pool for tier {tier_key}")
return random.choice(model)
return random.choice(pool)
def _tier_pools(self) -> dict[str, list[str]]: # mutable-ok: adaptive router consumes mutable pools
return { # mutable-ok: router consumers require mutable tier pool mappings

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

@ -913,7 +913,10 @@ class TestSingletonMutation:
def test_default_config_not_mutated(self, mock_router_instance):
"""Test that creating routers without config doesn't mutate defaults."""
from litellm.router_strategy.complexity_router.config import ComplexityRouterConfig
from litellm.router_strategy.complexity_router.config import (
DEFAULT_CLASSIFIER_CONTEXT_WINDOW_SIZE,
ComplexityRouterConfig,
)
# Get original default
original_default = ComplexityRouterConfig().default_model