scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-23 05:34:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 25

test(proxy): cover deferred callback failures

Browse source
This commit is contained in:
Ivana Dunisijevic 2026-10-08 17:47:49 +02:00
parent 8bee832c4e
commit aedc0e3aec
No known key found for this signature in database
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/unit/proxy/guardrails/test_deferred_guardrail_logging.py
View file

@ -1331,6 +1331,25 @@ class TestFireDeferredStreamLogging:
assert mock_logging_obj._on_deferred_stream_complete is None
assert mock_logging_obj._deferred_stream_complete_args is None
@pytest.mark.asyncio
async def test_callback_failure_is_logged_without_escaping(self):
"""A failed background callback must not surface as an unhandled task error."""
callback_error = RuntimeError("deferred callback failed")
async def failing_callback():
raise callback_error
mock_logging_obj = MagicMock()
mock_logging_obj._on_deferred_stream_complete = failing_callback
mock_logging_obj._deferred_stream_complete_args = ()
with patch("litellm.proxy.utils.verbose_proxy_logger.exception") as log_exception:
deferred_task = ProxyLogging._fire_deferred_stream_logging({"litellm_logging_obj": mock_logging_obj})
assert deferred_task is not None
await deferred_task
log_exception.assert_called_once_with("Error in deferred stream callback: %s", callback_error)
@pytest.mark.asyncio
async def test_deferred_guardrails_wait_for_success_logging(self):
logging_started = asyncio.Event()