From 174a1aa00705d19dd76665f0158449d160ed793f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 23 Apr 2025 10:51:07 -0700 Subject: [PATCH] test: update test --- tests/local_testing/test_amazing_vertex_completion.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 54982fe6d8..f3c8e32dc1 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -455,6 +455,7 @@ async def test_async_vertexai_response(): or "gemini-2.0-flash-thinking-exp" in model or "gemini-2.0-pro-exp-02-05" in model or "gemini-pro" in model + or "gemini-1.0-pro" in model ): # our account does not have access to this model continue @@ -465,6 +466,8 @@ async def test_async_vertexai_response(): model=model, messages=messages, temperature=0.7, timeout=5 ) print(f"response: {response}") + except litellm.NotFoundError as e: + pass except litellm.RateLimitError as e: pass except litellm.Timeout as e: @@ -503,6 +506,7 @@ async def test_async_vertexai_streaming_response(): or "gemini-2.0-flash-thinking-exp" in model or "gemini-2.0-pro-exp-02-05" in model or "gemini-pro" in model + or "gemini-1.0-pro" in model ): # our account does not have access to this model continue @@ -524,6 +528,8 @@ async def test_async_vertexai_streaming_response(): complete_response += chunk.choices[0].delta.content print(f"complete_response: {complete_response}") assert len(complete_response) > 0 + except litellm.NotFoundError as e: + pass except litellm.RateLimitError as e: pass except litellm.APIConnectionError: