fix - otel logging test

This commit is contained in:
Ishaan Jaff 2024-06-04 15:01:56 -07:00
parent 8369988579
commit 48f399bcc4
2 changed files with 4 additions and 0 deletions

View file

@ -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

View file

@ -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