diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index ec9b676772..54982fe6d8 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -454,6 +454,7 @@ async def test_async_vertexai_response(): or "002" in model or "gemini-2.0-flash-thinking-exp" in model or "gemini-2.0-pro-exp-02-05" in model + or "gemini-pro" in model ): # our account does not have access to this model continue @@ -501,6 +502,7 @@ async def test_async_vertexai_streaming_response(): or "002" in model or "gemini-2.0-flash-thinking-exp" in model or "gemini-2.0-pro-exp-02-05" in model + or "gemini-pro" in model ): # our account does not have access to this model continue