From 90afdd265be0a8bea4ac44deecc45d7023527a33 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 3 Jun 2024 22:43:00 -0700
Subject: [PATCH] test(test_completion.py): fix testing to retry

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

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 42623a25cc..0f0f76437f 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -97,6 +97,8 @@ def test_null_role_response():
     """
     import openai
 
+    litellm.success_callback = ["lunary"]
+
     openai_client = openai.OpenAI()
     with patch.object(
         openai_client.chat.completions, "create", side_effect=_openai_mock_response
@@ -112,6 +114,8 @@ def test_null_role_response():
 
         assert response.choices[0].message.role == "assistant"
 
+    assert False
+
 
 def test_completion_azure_command_r():
     try:
@@ -409,6 +413,7 @@ async def test_anthropic_no_content_error():
 def test_gemini_completion_call_error():
     try:
         print("test completion + streaming")
+        litellm.num_retries = 3
         litellm.set_verbose = True
         messages = [{"role": "user", "content": "what is the capital of congo?"}]
         response = completion(