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

fix(router): freeze retained forecast fields for lint

Browse source
This commit is contained in:
Tin Chi Lo 2026-09-21 16:40:55 -07:00
parent 7f997a420d
commit 7980a8f65f
1 changed files with 7 additions and 5 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

@ -1841,11 +1841,13 @@ class ComplexityRouter(CustomLogger):
decision["tier_litellm_params"] = masked_tier_litellm_params
if previous_decision is None:
return decision
forecast_fields: Final = {
field: value
for field, value in previous_decision.items()
if field.startswith("classifier_") or field == "heuristic_v2_forecast"
}
forecast_fields: Final = MappingProxyType(
{
field: value
for field, value in previous_decision.items()
if field.startswith("classifier_") or field == "heuristic_v2_forecast"
}
)
return cast( # cast-ok: retaining optional keys from a typed decision preserves their declared values
StandardLoggingRoutingDecision, {**forecast_fields, **decision}
)