ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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