test: update test to skip 'gemini-pro' - model deprecated

This commit is contained in:
Krrish Dholakia 2025-04-23 00:01:02 -07:00 committed by Christian Owusu
parent d4ef43f34c
commit 1430c98c78

View file

@ -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