From 9d7aaa91e05dff452c29a629823ce8733fe840e9 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 5 Apr 2024 14:04:29 -0700 Subject: [PATCH] test(test_amazing_vertex_completion.py): handle vertex ai rate limit errors --- litellm/tests/test_amazing_vertex_completion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 05782b51db..112939c77b 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -428,7 +428,8 @@ def test_gemini_pro_vision(): # DO Not DELETE this ASSERT # Google counts the prompt tokens for us, we should ensure we use the tokens from the orignal response assert prompt_tokens == 263 # the gemini api returns 263 to us - + except litellm.RateLimitError as e: + pass except Exception as e: if "500 Internal error encountered.'" in str(e): pass