diff --git a/litellm/litellm_core_utils/llm_cost_calc/utils.py b/litellm/litellm_core_utils/llm_cost_calc/utils.py index 5c93a91806c..640ccdc803a 100644 --- a/litellm/litellm_core_utils/llm_cost_calc/utils.py +++ b/litellm/litellm_core_utils/llm_cost_calc/utils.py @@ -99,10 +99,8 @@ def _generic_cost_per_character( assert ( "input_cost_per_character" in model_info and model_info["input_cost_per_character"] is not None - ), ( - "model info for model={} does not have 'input_cost_per_character'-pricing\nmodel_info={}".format( - model, model_info - ) + ), "model info for model={} does not have 'input_cost_per_character'-pricing\nmodel_info={}".format( + model, model_info ) custom_prompt_cost = model_info["input_cost_per_character"] @@ -122,10 +120,8 @@ def _generic_cost_per_character( assert ( "output_cost_per_character" in model_info and model_info["output_cost_per_character"] is not None - ), ( - "model info for model={} does not have 'output_cost_per_character'-pricing\nmodel_info={}".format( - model, model_info - ) + ), "model info for model={} does not have 'output_cost_per_character'-pricing\nmodel_info={}".format( + model, model_info ) custom_completion_cost = model_info["output_cost_per_character"] completion_cost = completion_characters * custom_completion_cost