From 5d655507321f765112b6b2ef68e4a9fd057f01b6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 9 Dec 2023 10:04:10 -0800 Subject: [PATCH] (test) custom logger print success handler --- litellm/tests/test_custom_logger.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_custom_logger.py b/litellm/tests/test_custom_logger.py index 44d0119325c..dcfd0e0e85f 100644 --- a/litellm/tests/test_custom_logger.py +++ b/litellm/tests/test_custom_logger.py @@ -157,10 +157,13 @@ def test_async_custom_handler_stream(): response_in_success_handler = customHandler2.stream_collected_response response_in_success_handler = response_in_success_handler["choices"][0]["message"]["content"] + print("\n\n") + print("response_in_success_handler: ", response_in_success_handler) + print("complete_streaming_response: ", complete_streaming_response) assert response_in_success_handler == complete_streaming_response except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_async_custom_handler_stream() +test_async_custom_handler_stream() def test_async_custom_handler(): try: