ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(test) stream chunk builder

Browse Source
This commit is contained in:
ishaan-jaff 2023-11-01 08:38:18 -07:00
parent d898b1c412
commit 70885bdba6
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
litellm/tests/test_stream_chunk_builder.py
View File

@ -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()