From 4b110b3fa41ec663230d663ba4b7961b35ce61b1 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 18 Nov 2023 17:32:15 -0800 Subject: [PATCH] (test) palm streaming test --- litellm/tests/test_streaming.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 6c8e25f8126..78c441dc68b 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -260,6 +260,7 @@ def test_completion_claude_stream(): def test_completion_palm_stream(): try: + litellm.set_verbose=False print("Streaming palm response") messages = [ {"role": "system", "content": "You are a helpful assistant."}, @@ -276,7 +277,8 @@ 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) + print(chunk) + # print(chunk.choices[0].delta) chunk, finished = streaming_format_tests(idx, chunk) if finished: break