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

fix(main.py): safely fail stream_chunk_builder calls

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-10 10:22:26 -07:00
parent 6ff21433da
commit 3fd02a1587
3 changed files with 259 additions and 231 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_stream_chunk_builder.py
View file

@ -16,9 +16,8 @@ import pytest
from openai import OpenAI
import litellm
from litellm import completion, stream_chunk_builder
import litellm.tests.stream_chunk_testdata
from litellm import completion, stream_chunk_builder
dotenv.load_dotenv()
@ -219,3 +218,11 @@ def test_stream_chunk_builder_litellm_mixed_calls():
"id": "toolu_01H3AjkLpRtGQrof13CBnWfK",
"type": "function",
}
def test_stream_chunk_builder_litellm_empty_chunks():
with pytest.raises(litellm.APIError):
response = stream_chunk_builder(chunks=None)
response = stream_chunk_builder(chunks=[])
assert response is None