From f5996b2f6ba45ec3859a716e28f6e6eff0f7a0b3 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 23 Apr 2025 00:01:02 -0700 Subject: [PATCH] test: update test to skip 'gemini-pro' - model deprecated --- tests/local_testing/test_amazing_vertex_completion.py | 2 ++ 1 file changed, 2 insertions(+) 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