From 73c6a70af222f5fef4d8e0721abfab84050d0e1e Mon Sep 17 00:00:00 2001 From: Praveen11558 <44603409+Praveen11558@users.noreply.github.com> Date: Mon, 23 Mar 2026 15:54:31 +0530 Subject: [PATCH] Update test_llm_cost_calc_utils.py --- .../llm_cost_calc/test_llm_cost_calc_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py index 33a166546ad..61679a6fa21 100644 --- a/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py +++ b/tests/test_litellm/litellm_core_utils/llm_cost_calc/test_llm_cost_calc_utils.py @@ -1464,7 +1464,7 @@ def test_character_count_billing_does_not_fill_prompt_token_gap(): model_info = litellm.model_cost["gemini-2.5-pro"] expected_prompt_cost = ( 100 * model_info["input_cost_per_token"] - + 1000 * model_info["input_cost_per_character"] + + 1000 * model_info.get("input_cost_per_character", 0) ) expected_completion_cost = 20 * model_info["output_cost_per_token"] @@ -1506,7 +1506,7 @@ def test_video_length_billing_does_not_fill_prompt_token_gap(): model_info = litellm.model_cost["gemini-2.5-pro"] expected_prompt_cost = ( 50 * model_info["input_cost_per_token"] - + 12.0 * model_info["input_cost_per_video_per_second"] + + 12.0 * model_info.get("input_cost_per_video_per_second", 0) ) expected_completion_cost = 10 * model_info["output_cost_per_token"]