From 3dbb547af0555abfbd5a10a2a69894a5313f9e25 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 5 Apr 2024 14:02:01 -0700
Subject: [PATCH] catch gemini 429 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 d2901de99cb..7641badfbe7 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -521,6 +521,8 @@ def test_gemini_pro_function_calling():
         print(f"completion: {completion}")
         assert completion.choices[0].message.content is None
         assert len(completion.choices[0].message.tool_calls) == 1
+    except litellm.RateLimitError as e:
+        pass
     except Exception as e:
         if "429 Quota exceeded" in str(e):
             pass