From 3548a6ffc319afc5bd16e8a1efb285cd3c869e31 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 20 Oct 2023 13:06:24 -0700 Subject: [PATCH] (test) test accessing class attributes for streamed responses --- litellm/tests/test_streaming.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index cf2bec7308..9bf202929f 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -432,6 +432,7 @@ def test_completion_claude_stream(): def test_completion_palm_stream(): try: + print("Streaming palm response") messages = [ {"role": "system", "content": "You are a helpful assistant."}, { @@ -447,6 +448,7 @@ def test_completion_palm_stream(): complete_response = "" # Add any assertions here to check the response for idx, chunk in enumerate(response): + print(chunk.choices[0].delta) chunk, finished = streaming_format_tests(idx, chunk) if finished: break