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 78

fix(test): update stale gemini-1.5-flash-001 model name to gemini-2.0-flash-001 in batch cost test

Browse source
This commit is contained in:
joereyna 2026-03-11 19:20:55 -07:00
parent 50e68dc387
commit 0bd6cab7db
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/proxy/pass_through_endpoints/test_vertex_ai_batch_passthrough.py
View file

@ -243,7 +243,7 @@ class TestVertexAIBatchPassthroughHandler:
]
total_cost, usage = calculate_vertex_ai_batch_cost_and_usage(
vertex_ai_batch_responses, model_name="gemini-1.5-flash-001"
vertex_ai_batch_responses, model_name="gemini-2.0-flash-001"
)
assert usage.total_tokens == 15