From 36b0aff118047718e2b8451bb1c710a9b8bfe4a7 Mon Sep 17 00:00:00 2001 From: Praveen11558 <44603409+Praveen11558@users.noreply.github.com> Date: Mon, 23 Mar 2026 15:44:42 +0530 Subject: [PATCH] Update model version in cost calculation tests --- .../llm_cost_calc/test_llm_cost_calc_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 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 be16ed62179..33a166546ad 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 @@ -1456,12 +1456,12 @@ def test_character_count_billing_does_not_fill_prompt_token_gap(): ) prompt_cost, completion_cost = generic_cost_per_token( - model="gemini-1.0-pro", + model="gemini-2.5-pro", usage=usage, custom_llm_provider="vertex_ai", ) - model_info = litellm.model_cost["gemini-1.0-pro"] + 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"] @@ -1498,12 +1498,12 @@ def test_video_length_billing_does_not_fill_prompt_token_gap(): ) prompt_cost, completion_cost = generic_cost_per_token( - model="gemini-1.0-pro", + model="gemini-2.5-pro", usage=usage, custom_llm_provider="vertex_ai", ) - model_info = litellm.model_cost["gemini-1.0-pro"] + 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"]