scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

fix: test_levo_logger_instantiation

Browse source
This commit is contained in:
Sameer Kankute 2026-01-07 16:00:58 +05:30
parent adf8fa1507
commit 8fc7d78ec1
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/integrations/levo/test_levo.py
View file

@ -4,15 +4,15 @@ from unittest.mock import patch
import pytest
from litellm.integrations.levo.levo import LevoConfig, LevoLogger
from litellm.integrations.opentelemetry import OpenTelemetryConfig
from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
# Try to import OpenTelemetry packages, skip tests if not available
try:
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 import SimpleSpanProcessor
OPENTELEMETRY_AVAILABLE = True
except ImportError:
@ -190,8 +190,6 @@ class TestLevoIntegration(unittest.TestCase):
# Verify it's an instance of OpenTelemetry
self.assertIsInstance(levo_logger, LevoLogger)
# Check it extends OpenTelemetry by checking base classes
from litellm.integrations.opentelemetry import OpenTelemetry
self.assertIsInstance(levo_logger, OpenTelemetry)
# Verify callback_name is set