test(test_completion.py): fix testing to retry

This commit is contained in:
Krrish Dholakia
2024-06-03 22:43:22 -07:00
parent fd1a2a90bf
commit 90afdd265b
+5
View File
@@ -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(