From 569f3ddda9f562dae4564f57f58e7a13416c4d8d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 9 Sep 2024 17:59:07 -0700 Subject: [PATCH] fix test_awesome_otel_with_message_logging_off --- litellm/tests/test_async_opentelemetry.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_async_opentelemetry.py b/litellm/tests/test_async_opentelemetry.py index 9d180306f..e94adb833 100644 --- a/litellm/tests/test_async_opentelemetry.py +++ b/litellm/tests/test_async_opentelemetry.py @@ -34,6 +34,8 @@ async def test_awesome_otel_with_message_logging_off(): ) litellm.callbacks = [otel_logger] + litellm.success_callback = [] + litellm.failure_callback = [] response = await litellm.acompletion( model="gpt-3.5-turbo", @@ -42,7 +44,7 @@ async def test_awesome_otel_with_message_logging_off(): ) print("response", response) - await asyncio.sleep(4) + await asyncio.sleep(5) assert otel_logger.kwargs["messages"] == [ {"role": "user", "content": "redacted-by-litellm"}