From 6918d79b3db69fc038b10bdbc7d6ecfd38ecfb54 Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Thu, 12 Mar 2026 04:52:39 +0000 Subject: [PATCH] fix(ci): replace deprecated vertex_ai/gemini-pro with vertex_ai/gemini-3-pro-preview in cost discount test The model gemini-pro was removed from the model_prices_and_context_window.json cost map, causing test_cost_discount_vertex_ai to fail with 'model isn't mapped yet'. Co-authored-by: yuneng-jiang --- tests/test_litellm/test_cost_calculator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_litellm/test_cost_calculator.py b/tests/test_litellm/test_cost_calculator.py index b991dcaf4ed..d370ab7976a 100644 --- a/tests/test_litellm/test_cost_calculator.py +++ b/tests/test_litellm/test_cost_calculator.py @@ -975,7 +975,7 @@ def test_cost_discount_vertex_ai(): id="test-id", choices=[], created=1234567890, - model="gemini-pro", + model="gemini-3-pro-preview", object="chat.completion", usage=Usage(prompt_tokens=100, completion_tokens=50, total_tokens=150), ) @@ -984,7 +984,7 @@ def test_cost_discount_vertex_ai(): litellm.cost_discount_config = {} cost_without_discount = completion_cost( completion_response=response, - model="vertex_ai/gemini-pro", + model="vertex_ai/gemini-3-pro-preview", custom_llm_provider="vertex_ai", ) @@ -994,7 +994,7 @@ def test_cost_discount_vertex_ai(): # Calculate cost with discount cost_with_discount = completion_cost( completion_response=response, - model="vertex_ai/gemini-pro", + model="vertex_ai/gemini-3-pro-preview", custom_llm_provider="vertex_ai", )