From 0bd6cab7db8928787d62535da9ff33c2c6509459 Mon Sep 17 00:00:00 2001 From: joereyna Date: Wed, 11 Mar 2026 19:20:55 -0700 Subject: [PATCH] fix(test): update stale gemini-1.5-flash-001 model name to gemini-2.0-flash-001 in batch cost test --- .../pass_through_endpoints/test_vertex_ai_batch_passthrough.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/pass_through_endpoints/test_vertex_ai_batch_passthrough.py b/tests/test_litellm/proxy/pass_through_endpoints/test_vertex_ai_batch_passthrough.py index c2f6d3fd539..41a573689f2 100644 --- a/tests/test_litellm/proxy/pass_through_endpoints/test_vertex_ai_batch_passthrough.py +++ b/tests/test_litellm/proxy/pass_through_endpoints/test_vertex_ai_batch_passthrough.py @@ -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