From 70885bdba64a72acbfa8b1b397eeb51e32529dee Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 1 Nov 2023 08:38:18 -0700 Subject: [PATCH] (test) stream chunk builder --- litellm/tests/test_stream_chunk_builder.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_stream_chunk_builder.py b/litellm/tests/test_stream_chunk_builder.py index a4527521f2..3d04cbd912 100644 --- a/litellm/tests/test_stream_chunk_builder.py +++ b/litellm/tests/test_stream_chunk_builder.py @@ -35,6 +35,7 @@ def test_stream_chunk_builder(): chunks = [] for chunk in response: + print(chunk) chunks.append(chunk) try: @@ -51,8 +52,9 @@ def test_stream_chunk_builder(): message = choices["message"] role = message["role"] content = message["content"] - finnish_reason = choices["finish_reason"] - except: - raise Exception("stream_chunk_builder failed to rebuild response") + finish_reason = choices["finish_reason"] + print(role, content, finish_reason) + except Exception as e: + raise Exception("stream_chunk_builder failed to rebuild response", e) # test_stream_chunk_builder()