phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-12 18:17:57 -07:00
parent 2f85b5f6e1
commit 9328c6f790
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/router.py
View file

@ -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: