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 e38a0a6c40
commit 1373debd61

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