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(