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

(test) do nothing if token==None in stream chunk

Browse source
This commit is contained in:
ishaan-jaff 2023-11-18 17:13:16 -08:00
parent e527a45ffc
commit 30fcc1f9a0
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_async_fn.py
View file

@ -142,7 +142,10 @@ def test_get_response_non_openai_streaming():
output = "" output = ""
i = 0 i = 0
async for chunk in response: async for chunk in response:
token = chunk["choices"][0]["delta"].get("content", "") token = chunk["choices"][0]["delta"].get("content", None)
if token == None:
continue
print(token)
output += token output += token
print(f"output: {output}") print(f"output: {output}")
assert output is not None, "output cannot be None." assert output is not None, "output cannot be None."