From c7eebf75a829ef039cb40eb5311f53ac8983efc9 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 8 Dec 2023 18:46:19 -0800 Subject: [PATCH] (ci/cd) test proxy custom logger --- ...oxy_custom_logger.py => test_amazing_proxy_custom_logger.py} | 2 ++ 1 file changed, 2 insertions(+) rename litellm/tests/{test_proxy_custom_logger.py => test_amazing_proxy_custom_logger.py} (99%) 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"]