From 1b1420258243523d8b0f16ff4f0f3e5a8474c34c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 6 Jul 2024 11:20:36 -0700 Subject: [PATCH] fix(litellm_logging.py): fix 'use_custom_pricing_for_model' helper function --- litellm/litellm_core_utils/litellm_logging.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/litellm/litellm_core_utils/litellm_logging.py b/litellm/litellm_core_utils/litellm_logging.py index 2bac73b98..f9f32552d 100644 --- a/litellm/litellm_core_utils/litellm_logging.py +++ b/litellm/litellm_core_utils/litellm_logging.py @@ -2019,10 +2019,13 @@ def get_custom_logger_compatible_class( return None -def use_custom_pricing_for_model(litellm_params: dict) -> bool: - model_info: Optional[dict] = litellm_params.get("metadata", {}).get( - "model_info", {} - ) +def use_custom_pricing_for_model(litellm_params: Optional[dict]) -> bool: + if litellm_params is None: + 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: for k, v in model_info.items(): if k in SPECIAL_MODEL_INFO_PARAMS: