scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 154

test(test_streaming.py): add test to make sure role is not returned in middle chunks during streaming

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-01 09:55:46 -08:00
parent d46df34ff5
commit 874ebad31d
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_streaming.py
View file

@ -92,6 +92,7 @@ def validate_second_format(chunk):
for choice in chunk["choices"]:
assert isinstance(choice["index"], int), "'index' should be an integer."
assert "role" not in choice["delta"], "'role' should be a string."
# openai v1.0.0 returns content as None
assert (choice["finish_reason"] is None) or isinstance(
choice["finish_reason"], str