From cb6ddaf1f9e38653dec17a45c56ee9123b5862d8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 11 Jul 2024 08:01:18 -0700 Subject: [PATCH] test - otel spans --- litellm/tests/test_async_opentelemetry.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/litellm/tests/test_async_opentelemetry.py b/litellm/tests/test_async_opentelemetry.py index 4d174c050..fa9fc5110 100644 --- a/litellm/tests/test_async_opentelemetry.py +++ b/litellm/tests/test_async_opentelemetry.py @@ -1,21 +1,19 @@ import asyncio -import litellm - -from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig -from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter -from litellm._logging import verbose_logger import logging import time + import pytest +from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter + +import litellm +from litellm._logging import verbose_logger +from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig verbose_logger.setLevel(logging.DEBUG) -@pytest.mark.skip( - reason="new test. WIP. works locally but not on CI. Still figuring this out" -) @pytest.mark.asyncio -async def test_otel_callback(): +async def test_async_otel_callback(): exporter = InMemorySpanExporter() litellm.set_verbose = True litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))]