diff --git a/litellm/tests/test_proxy_custom_logger.py b/litellm/tests/test_amazing_proxy_custom_logger.py similarity index 99% rename from litellm/tests/test_proxy_custom_logger.py rename to litellm/tests/test_amazing_proxy_custom_logger.py index 06e0ab10fa..3e3e7468f9 100644 --- a/litellm/tests/test_proxy_custom_logger.py +++ b/litellm/tests/test_amazing_proxy_custom_logger.py @@ -141,6 +141,8 @@ def test_chat_completion_stream(client): print("\n\nHERE is the complete streaming response string", complete_response) print("\n\nHERE IS the streaming Response from callback\n\n") print(my_custom_logger.streaming_response_obj) + import time + time.sleep(0.5) streamed_response = my_custom_logger.streaming_response_obj assert complete_response == streamed_response["choices"][0]["message"]["content"]