From 01d58754265e73d087708021c9c247ef2b282324 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 11 Dec 2023 23:35:50 -0800 Subject: [PATCH] (test) reset callbacks in completion() --- litellm/tests/test_completion.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index b6dc82e055..19473cf48e 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -21,6 +21,13 @@ messages = [{"content": user_message, "role": "user"}] def logger_fn(user_model_dict): print(f"user_model_dict: {user_model_dict}") +@pytest.fixture(autouse=True) +def reset_callbacks(): + print("\npytest fixture - resetting callbacks") + litellm.success_callback = [] + litellm._async_success_callback = [] + litellm.failure_callback = [] + litellm.callbacks = [] def test_completion_custom_provider_model_name(): try: