From 9cac629ca60ecfdd3bf9b37cedf01bfe241d6d52 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 16 Jul 2025 09:13:16 -0700 Subject: [PATCH] test: update test --- tests/local_testing/test_amazing_vertex_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 064f95db39..4219c8f1a7 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -501,7 +501,7 @@ async def test_async_vertexai_streaming_response(): ) test_models = random.sample(test_models, 1) test_models += litellm.vertex_language_models # always test gemini-pro - test_models = ["gemini-2.5-flash-preview"] + test_models = ["gemini-2.5-flash"] for model in test_models: if model in VERTEX_MODELS_TO_NOT_TEST or ( "gecko" in model