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