diff --git a/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py b/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py index d05645b6e9d..e5d805f04c7 100644 --- a/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py +++ b/tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py @@ -1211,6 +1211,12 @@ async def _wait_for_mock_call(mock, timeout=10, interval=0.1): class TestSpendLogsPayload: + def setup_method(self): + self._original_callbacks = litellm.callbacks[:] + + def teardown_method(self): + litellm.callbacks = self._original_callbacks + @pytest.mark.asyncio async def test_spend_logs_payload_e2e(self): litellm.callbacks = [_ProxyDBLogger(message_logging=False)]