From 875b7d1c7f5e024e5d06f1fa1b33704561fb23f6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 9 Dec 2023 10:42:41 -0800 Subject: [PATCH] (test) custom logger --- litellm/tests/test_custom_logger.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_custom_logger.py b/litellm/tests/test_custom_logger.py index bc4d34567..4cdaaa084 100644 --- a/litellm/tests/test_custom_logger.py +++ b/litellm/tests/test_custom_logger.py @@ -177,7 +177,7 @@ def test_azure_completion_stream(): # checks if the model response available in the async + stream callbacks is equal to the received response customHandler2 = MyCustomHandler() litellm.callbacks = [customHandler2] - litellm.set_verbose = False + litellm.set_verbose = True messages = [ {"role": "system", "content": "You are a helpful assistant."}, { @@ -196,6 +196,7 @@ def test_azure_completion_stream(): complete_streaming_response += chunk["choices"][0]["delta"]["content"] or "" print(complete_streaming_response) + time.sleep(0.5) # wait 1/2 second before checking callbacks response_in_success_handler = customHandler2.sync_stream_collected_response response_in_success_handler = response_in_success_handler["choices"][0]["message"]["content"] print("\n\n")