test: update test to skip 'gemini-pro' - model deprecated
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 19s
Helm unit test / unit-test (push) Successful in 32s

This commit is contained in:
Krrish Dholakia 2025-04-23 00:01:02 -07:00
parent 217681eb5e
commit f5996b2f6b

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