diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 03bf54486..4660ea914 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -14,6 +14,7 @@ from litellm import embedding, completion, completion_cost, Timeout from litellm import RateLimitError litellm.num_retries = 3 litellm.cache = None +litellm.success_callback = [] user_message = "Write a short poem about the sky" messages = [{"content": user_message, "role": "user"}]