scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix(ci): replace deprecated vertex_ai/gemini-pro with vertex_ai/gemini-3-pro-preview in cost discount test

Browse source 
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 <yuneng-jiang@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-03-12 04:52:39 +00:00
parent 73981d477c
commit 6918d79b3d
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/test_cost_calculator.py
View file

@ -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",
)