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

fix(router): reject unknown capability policy fields

Browse source
This commit is contained in:
Tin Chi Lo 2026-09-15 11:43:04 -07:00
parent 914ae9b248
commit e62f0d0376
2 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -616,7 +616,7 @@ class CapabilityCalibrationConfig(BaseModel):
class CapabilityClassifierConfig(BaseModel):
"""Switchyard-compatible probability threshold policy for two model tiers."""
model_config = ConfigDict(frozen=True)
model_config = ConfigDict(extra="forbid", frozen=True)
efficient_tier: str = Field(
description="Tier used when the efficient model's forecasted solve probability meets the adjusted threshold",

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

@ -2500,6 +2500,17 @@ class TestCapabilityClassifierConfig:
with pytest.raises(ValidationError, match="requires classifier_type 'capability'"):
ComplexityRouterConfig(**config)
def test_rejects_misspelled_optional_policy_instead_of_using_defaults(self) -> None:
with pytest.raises(ValidationError, match="threshold_steps"):
CapabilityClassifierConfig.model_validate(
{
"efficient_tier": "SIMPLE",
"capable_tier": "REASONING",
"base_threshold": 0.5,
"threshold_steps": 0.2,
}
)
def test_threshold_defaults_match_switchyard(self):
config = CapabilityClassifierConfig(efficient_tier=" SIMPLE ", capable_tier=" REASONING ", base_threshold=0.5)
assert config.efficient_tier == "SIMPLE"