test: update test

This commit is contained in:
Krrish Dholakia 2025-07-16 09:13:16 -07:00 • committed by Ishaan Jaff
parent 88348b238d
commit 1ecb341034

View file

@ -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-05-20"]
test_models = ["gemini-2.5-flash"]
for model in test_models:
if model in VERTEX_MODELS_TO_NOT_TEST or (
"gecko" in model