From cf4574dd65fc1d86bb2151ca74907604c6410470 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 22 Aug 2024 08:54:59 -0700 Subject: [PATCH] test(test_custom_callback_input.py): fix test --- litellm/tests/test_custom_callback_input.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_custom_callback_input.py b/litellm/tests/test_custom_callback_input.py index a498b970bd2..e1f4ecb929b 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: