diff --git a/tests/test_litellm_rust/test_integrations.py b/tests/test_litellm_rust/test_integrations.py index 5501d9e20d2..bb0be4dbd49 100644 --- a/tests/test_litellm_rust/test_integrations.py +++ b/tests/test_litellm_rust/test_integrations.py @@ -128,6 +128,7 @@ async def test_generic_guardrail_logging_only_verdict_is_exported_over_http( response: Final = await route.invoke(provider, callbacks=[guardrail, logger, recorder]) await recorder.wait_for_async("async_log_success_event") + await drain_logging() scan: Final = provider.requests[1] assert scan.path == "/beta/litellm_basic_guardrail_api"