test(test_completion.py): fix test

This commit is contained in:
Krrish Dholakia 2024-06-03 21:36:25 -07:00
parent 7311e82f47
commit 7c1e2bf01c

View file

@ -97,8 +97,6 @@ def test_null_role_response():
""" """
import openai import openai
litellm.success_callback = ["lunary"]
openai_client = openai.OpenAI() openai_client = openai.OpenAI()
with patch.object( with patch.object(
openai_client.chat.completions, "create", side_effect=_openai_mock_response openai_client.chat.completions, "create", side_effect=_openai_mock_response
@ -114,8 +112,6 @@ def test_null_role_response():
assert response.choices[0].message.role == "assistant" assert response.choices[0].message.role == "assistant"
assert False
def test_completion_azure_command_r(): def test_completion_azure_command_r():
try: try: