From 5a53909a0dc6c7aa5511046a9cf488f6e2eef2a1 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 26 Aug 2023 11:56:52 -0700 Subject: [PATCH] fix linting tests --- litellm/tests/test_streaming.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index d0da5e4d7..25fa5c047 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -28,6 +28,7 @@ try: model="gpt-3.5-turbo", messages=messages, stream=True, logger_fn=logger_fn ) response = "" + start_time = time.time() for chunk in response: chunk_time = time.time() print(f"time since initial request: {chunk_time - start_time:.2f}") @@ -46,6 +47,7 @@ try: model="azure/chatgpt-test", messages=messages, stream=True, logger_fn=logger_fn ) response = "" + start_time = time.time() for chunk in response: chunk_time = time.time() print(f"time since initial request: {chunk_time - start_time:.2f}") @@ -64,6 +66,7 @@ try: model="claude-instant-1", messages=messages, stream=True, logger_fn=logger_fn ) response = "" + start_time = time.time() for chunk in response: chunk_time = time.time() print(f"time since initial request: {chunk_time - start_time:.2f}")