diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index a498b970b..e1f4ecb92 100644 --- a/litellm/tests/test_custom_callback_input.py +++ b/litellm/tests/test_custom_callback_input.py @@ -1254,12 +1254,10 @@ def test_standard_logging_payload(model, turn_off_message_logging): assert "redacted-by-litellm" == slobject["messages"][0]["content"] -def test_standard_logging_payload_cache_hit(): +def test_aaastandard_logging_payload_cache_hit(): from litellm.types.utils import StandardLoggingPayload # sync completion - customHandler = CompletionCustomHandler() - litellm.callbacks = [customHandler] litellm.cache = Cache() @@ -1269,6 +1267,10 @@ def test_standard_logging_payload_cache_hit(): caching=True, ) + customHandler = CompletionCustomHandler() + litellm.callbacks = [customHandler] + litellm.success_callback = [] + with patch.object( customHandler, "log_success_event", new=MagicMock() ) as mock_client: