From 647e867cf8ae30b3f65d8d2f806fba604ff5fe91 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 26 Feb 2024 22:07:33 -0800 Subject: [PATCH] test(test_amazing_vertex_completion.py): fix test to handle rate limit errors --- litellm/tests/test_amazing_vertex_completion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index d58d68507a..fb2ebc0d65 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -222,6 +222,8 @@ async def test_async_vertexai_response(): model=model, messages=messages, temperature=0.7, timeout=5 ) print(f"response: {response}") + except litellm.RateLimitError as e: + pass except litellm.Timeout as e: pass except Exception as e: