phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(litellm_logging.py): fix 'use_custom_pricing_for_model' helper function

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-06 11:20:36 -07:00
parent ea9718a848
commit 1b14202582
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/litellm_core_utils/litellm_logging.py
View file

@ -2019,10 +2019,13 @@ def get_custom_logger_compatible_class(
return None return None
def use_custom_pricing_for_model(litellm_params: dict) -> bool: def use_custom_pricing_for_model(litellm_params: Optional[dict]) -> bool:
model_info: Optional[dict] = litellm_params.get("metadata", {}).get( if litellm_params is None:
"model_info", {} return False
) metadata: Optional[dict] = litellm_params.get("metadata", {})
if metadata is None:
return False
model_info: Optional[dict] = metadata.get("model_info", {})
if model_info is not None: if model_info is not None:
for k, v in model_info.items(): for k, v in model_info.items():
if k in SPECIAL_MODEL_INFO_PARAMS: if k in SPECIAL_MODEL_INFO_PARAMS: