From 9328c6f7901527921791adc128d34adc0d24fdfd Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 12 Apr 2025 18:17:57 -0700 Subject: [PATCH] fix: fix linting errors --- litellm/router.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/router.py b/litellm/router.py index e71452bc00..c934b1e9a8 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -4853,10 +4853,11 @@ class Router: from litellm.utils import _update_dictionary model_info: Optional[ModelInfo] = None + custom_model_info: Optional[dict] = None litellm_model_name_model_info: Optional[ModelInfo] = None try: - model_info = litellm.model_cost.get(model_id) + custom_model_info = litellm.model_cost.get(model_id) except Exception: pass @@ -4865,12 +4866,12 @@ class Router: except Exception: pass - if model_info is not None and litellm_model_name_model_info is not None: + if custom_model_info is not None and litellm_model_name_model_info is not None: model_info = cast( ModelInfo, _update_dictionary( cast(dict, litellm_model_name_model_info).copy(), - model_info, + custom_model_info, ), ) elif litellm_model_name_model_info is not None: