From 84060160961be934fc24946e6885c79653126c10 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 21 Aug 2024 07:17:27 -0700
Subject: [PATCH] test(test_completion.py): fix gemini rate limit error

---
 litellm/tests/test_completion.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 4b275a342e1..2ef6cf674ad 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -707,6 +707,8 @@ def test_gemini_completion_call_error():
         print(f"response: {response}")
         for chunk in response:
             print(chunk)
+    except litellm.RateLimitError:
+        pass
     except litellm.InternalServerError:
         pass
     except Exception as e: