mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
(test) test accessing class attributes for streamed responses
This commit is contained in:
parent
1f1cf7a11c
commit
3548a6ffc3
1 changed files with 2 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue