From bcfe62b80194e61c75a3fc9276caba65707d08ef Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 8 Jun 2024 18:24:00 -0700 Subject: [PATCH] fix test otel logger --- .../{test_opentelemetry.py => test_async_opentelemetry.py} | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) rename litellm/tests/{test_opentelemetry.py => test_async_opentelemetry.py} (95%) diff --git a/litellm/tests/test_opentelemetry.py b/litellm/tests/test_async_opentelemetry.py similarity index 95% rename from litellm/tests/test_opentelemetry.py rename to litellm/tests/test_async_opentelemetry.py index 92d7b85153..26af43a1d6 100644 --- a/litellm/tests/test_opentelemetry.py +++ b/litellm/tests/test_async_opentelemetry.py @@ -12,19 +12,20 @@ verbose_logger.setLevel(logging.DEBUG) # @pytest.mark.skip(reason="new test") -def test_otel_callback(): +@pytest.mark.asyncio +async def test_otel_callback(): exporter = InMemorySpanExporter() litellm.set_verbose = True litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))] - litellm.completion( + await litellm.acompletion( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "hi"}], temperature=0.1, user="OTEL_USER", ) - time.sleep(4) + await asyncio.sleep(4) spans = exporter.get_finished_spans() print("spans", spans)