phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test - otel spans

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-11 08:01:18 -07:00
parent 19d993e120
commit cb6ddaf1f9
1 changed files with 7 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_async_opentelemetry.py
View file

@ -1,21 +1,19 @@
import asyncio 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 logging
import time import time
import pytest 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) 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 @pytest.mark.asyncio
async def test_otel_callback(): async def test_async_otel_callback():
exporter = InMemorySpanExporter() exporter = InMemorySpanExporter()
litellm.set_verbose = True litellm.set_verbose = True
litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))] litellm.callbacks = [OpenTelemetry(OpenTelemetryConfig(exporter=exporter))]