mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-11 17:05:43 +00:00
(test) test accessing class attributes for streamed responses
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user