From 8fc7d78ec14db42428ebfecab59f527b0e859621 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 7 Jan 2026 16:00:58 +0530 Subject: [PATCH] fix: test_levo_logger_instantiation --- tests/test_litellm/integrations/levo/test_levo.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/integrations/levo/test_levo.py b/tests/test_litellm/integrations/levo/test_levo.py index 5d042cbc060..f03b8452d0d 100644 --- a/tests/test_litellm/integrations/levo/test_levo.py +++ b/tests/test_litellm/integrations/levo/test_levo.py @@ -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