phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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