tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-08 22:25:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_logging_opentelemetry_context_propagation

Browse Source
This commit is contained in:
Ishaan Jaff
2025-09-13 09:44:36 -07:00
parent 38efd358eb
commit 69e451458c
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/litellm_core_utils/test_litellm_logging.py
+7 -4
View File
@@ -473,13 +473,16 @@ async def test_logging_opentelemetry_context_propagation():
Test that OpenTelemtry context propagation works with async completion.
"""
import asyncio
import litellm
from litellm.integrations.custom_logger import CustomLogger
from opentelemetry import trace
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import SimpleSpanProcessor
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
from opentelemetry.sdk.trace.export.in_memory_span_exporter import (
InMemorySpanExporter,
)
import litellm
from litellm.integrations.custom_logger import CustomLogger
provider = TracerProvider()
exporter = InMemorySpanExporter()
@@ -490,7 +493,7 @@ async def test_logging_opentelemetry_context_propagation():
class MockOpenTelemetryLogger(CustomLogger):
async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
span = tracer.start_span(start_time=start_time.timestamp() * 1e9, name="async_log_success_event")
span.end(end_time=end_time)
span.end(end_time=end_time.timestamp() * 1e9)
mock_logging_obj = MockOpenTelemetryLogger()