Update tests/test_litellm/litellm_core_utils/test_streaming_handler.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Emerson Gomes 2026-02-02 17:08:29 -03:00 • committed by Ishaan Jaffer
parent 16ad80fc90
commit 0a76a22f7e

View file

@ -1753,4 +1753,8 @@ async def test_custom_stream_wrapper_anext_does_not_block_event_loop_for_sync_it
finally:
if not anext_task.done():
anext_task.cancel()
try:
await anext_task
except asyncio.CancelledError:
pass
await bg_task