scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-20 05:04:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 155

test(proxy): cover direct completed response streaming

Browse source
This commit is contained in:
Devin AI 2026-07-14 03:37:25 +00:00
parent 779a4e0324
commit 2ff3eeabfc
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/proxy/proxy_server/test_streaming_helpers.py
View file

@ -674,8 +674,11 @@ async def test_async_data_generator_yields_sse_chunks_and_done(monkeypatch):
@pytest.mark.asyncio
async def test_async_data_generator_adapts_completed_model_response(monkeypatch):
_patch_logging_flags(monkeypatch, needs_wrap=True)
@pytest.mark.parametrize("needs_wrap", [False, True])
async def test_async_data_generator_adapts_completed_model_response(
monkeypatch, needs_wrap
):
_patch_logging_flags(monkeypatch, needs_wrap=needs_wrap)
completed_response = ModelResponse(
id="chatcmpl-completed",
@ -722,9 +725,12 @@ async def test_async_data_generator_adapts_completed_model_response(monkeypatch)
)
]
assert len(iterator_hook_responses) == 1
assert iterator_hook_responses[0] is not completed_response
assert hasattr(iterator_hook_responses[0], "__aiter__")
if needs_wrap:
assert len(iterator_hook_responses) == 1
assert iterator_hook_responses[0] is not completed_response
assert hasattr(iterator_hook_responses[0], "__aiter__")
else:
assert iterator_hook_responses == []
assert output[-1] == "data: [DONE]\n\n"
first_chunk = output[0]
assert isinstance(first_chunk, str)