From 48f399bcc4bd3b868510038136dea172ad3888c3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 4 Jun 2024 15:01:56 -0700 Subject: [PATCH] fix - otel logging test --- .circleci/config.yml | 1 + litellm/tests/test_opentelemetry.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index a08ac8655..18bfeedb5 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -43,6 +43,7 @@ jobs: pip install "langfuse==2.27.1" pip install "logfire==0.29.0" pip install numpydoc + pip install traceloop-sdk==0.21.1 pip install opentelemetry-api==1.25.0 pip install opentelemetry-sdk==1.25.0 pip install opentelemetry-exporter-otlp==1.25.0 diff --git a/litellm/tests/test_opentelemetry.py b/litellm/tests/test_opentelemetry.py index 7caf2ead3..17a2a5674 100644 --- a/litellm/tests/test_opentelemetry.py +++ b/litellm/tests/test_opentelemetry.py @@ -5,6 +5,7 @@ from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfi from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter from litellm._logging import verbose_logger import logging +import time verbose_logger.setLevel(logging.DEBUG) @@ -26,5 +27,7 @@ def test_otel_callback(): ) ) + time.sleep(4) + spans = exporter.get_finished_spans() assert len(spans) == 1 + 1