From 1b9fb346795457d0012fd8c72ad79694fe1e0caf Mon Sep 17 00:00:00 2001 From: Tin Chi Lo Date: Mon, 21 Sep 2026 16:47:21 -0700 Subject: [PATCH] refactor(router): simplify optional forecast retention --- .../router_strategy/complexity_router/complexity_router.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/router_strategy/complexity_router/complexity_router.py b/litellm/router_strategy/complexity_router/complexity_router.py index c42c5435c6e..f0fef3974f4 100644 --- a/litellm/router_strategy/complexity_router/complexity_router.py +++ b/litellm/router_strategy/complexity_router/complexity_router.py @@ -1839,12 +1839,10 @@ class ComplexityRouter(CustomLogger): masked_tier_litellm_params: Final = mask_credentials_in_payload(tier_litellm_params) if isinstance(masked_tier_litellm_params, Mapping): decision["tier_litellm_params"] = masked_tier_litellm_params - if previous_decision is None: - return decision forecast_fields: Final = MappingProxyType( { field: value - for field, value in previous_decision.items() + for field, value in (previous_decision.items() if previous_decision is not None else ()) if field.startswith("classifier_") or field == "heuristic_v2_forecast" } )