phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_custom_callback_input.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-22 08:54:59 -07:00
parent 25820680eb
commit cf4574dd65
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_custom_callback_input.py
View file

@ -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: