From 08f71ade045a11479e66b59b7940d4b9e7277b2c Mon Sep 17 00:00:00 2001 From: joshua Date: Tue, 13 Oct 2026 19:17:05 +0000 Subject: [PATCH] style(router): format cost routing changes Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/router.py | 9 +++------ litellm/router_strategy/lowest_cost.py | 8 ++------ 2 files changed, 5 insertions(+), 12 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index f95cb60df94..3638dc0dd5f 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -1508,12 +1508,9 @@ class Router: routing_args=routing_strategy_args, ) case RoutingStrategy.COST_BASED.value: - def price_lookup( - model_id: str, model_name: str - ) -> tuple[float | None, float | None]: - model_info: Final = self.get_deployment_model_info( - model_id=model_id, model_name=model_name - ) + + def price_lookup(model_id: str, model_name: str) -> tuple[float | None, float | None]: + model_info: Final = self.get_deployment_model_info(model_id=model_id, model_name=model_name) if model_info is None: return None, None return ( diff --git a/litellm/router_strategy/lowest_cost.py b/litellm/router_strategy/lowest_cost.py index 00daa71fffa..de366489990 100644 --- a/litellm/router_strategy/lowest_cost.py +++ b/litellm/router_strategy/lowest_cost.py @@ -21,12 +21,8 @@ def _deployment_cost( price_lookup: Callable[[str, str], tuple[float | None, float | None]], ) -> float: lookup_input_price, lookup_output_price = price_lookup(model_id, model_name) - input_price: Final = ( - input_price_override if input_price_override is not None else lookup_input_price - ) - output_price: Final = ( - output_price_override if output_price_override is not None else lookup_output_price - ) + input_price: Final = input_price_override if input_price_override is not None else lookup_input_price + output_price: Final = output_price_override if output_price_override is not None else lookup_output_price input_cost: Final = input_price if input_price is not None else 5.0 output_cost: Final = output_price if output_price is not None else 5.0 return input_cost + output_cost