Fix import sorting in test_span_creation function

Co-Authored-By: Joe Moura <joao@crewai.com>

pyproject.toml

@@ -17,9 +17,11 @@ dependencies = [
     "pdfplumber>=0.11.4",
     "regex>=2024.9.11",
     # Telemetry and Monitoring
-    "opentelemetry-api>=1.22.0",
+    # Version constraints ensure compatibility with various OpenTelemetry versions
-    "opentelemetry-sdk>=1.22.0",
+    # while preventing installation of incompatible versions (issue #2372)
-    "opentelemetry-exporter-otlp-proto-http>=1.22.0",
+    "opentelemetry-api>=1.22.0,<1.32.0",
+    "opentelemetry-sdk>=1.22.0,<1.32.0",
+    "opentelemetry-exporter-otlp-proto-http>=1.22.0,<1.32.0",
     # Data Handling
     "chromadb>=0.5.23",
     "openpyxl>=3.1.5",

tests/telemetry/test_opentelemetry_compatibility.py

@@ -0,0 +1,116 @@
+from importlib import import_module
+
+import pytest
+
+
+def test_opentelemetry_imports():
+    """Test that all required OpenTelemetry modules can be imported.
+    
+    This test verifies that all necessary OpenTelemetry modules can be imported
+    correctly with the current version constraints, ensuring compatibility
+    between different OpenTelemetry packages.
+    """
+    try:
+        # Test basic imports
+        from opentelemetry import trace
+        from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
+            OTLPSpanExporter,
+        )
+        from opentelemetry.sdk.resources import SERVICE_NAME, Resource
+        from opentelemetry.sdk.trace import TracerProvider
+        from opentelemetry.sdk.trace.export import BatchSpanProcessor
+        from opentelemetry.trace import Span, Status, StatusCode
+        
+        # Verify that the imports are from the expected modules
+        assert trace.__name__ == 'opentelemetry.trace'
+        assert OTLPSpanExporter.__module__ == 'opentelemetry.exporter.otlp.proto.http.trace_exporter'
+        assert Resource.__module__ == 'opentelemetry.sdk.resources'
+        assert TracerProvider.__module__ == 'opentelemetry.sdk.trace'
+        assert BatchSpanProcessor.__module__ == 'opentelemetry.sdk.trace.export'
+        assert Span.__module__ == 'opentelemetry.trace.span'
+        assert Status.__module__ == 'opentelemetry.trace.status'
+        assert StatusCode.__module__ == 'opentelemetry.trace.status'
+    except ImportError as e:
+        pytest.fail(f"Failed to import OpenTelemetry modules: {e}")
+
+
+def test_telemetry_class_initialization():
+    """Test that the Telemetry class can be initialized with current OpenTelemetry versions.
+    
+    This test verifies that the Telemetry class can be properly initialized and
+    that its core attributes are set correctly. It also tests the set_tracer method
+    to ensure it works as expected with the current OpenTelemetry versions.
+    """
+    try:
+        from src.crewai.telemetry.telemetry import Telemetry
+        
+        # Create an instance of the Telemetry class
+        telemetry = Telemetry()
+        
+        # Check if the telemetry instance is initialized correctly
+        assert hasattr(telemetry, 'ready')
+        assert hasattr(telemetry, 'trace_set')
+        assert telemetry.trace_set is False  # Should be False initially
+        
+        # Try to set the tracer
+        telemetry.set_tracer()
+        
+        # After setting the tracer, trace_set should be True if ready is True
+        if telemetry.ready:
+            assert telemetry.trace_set is True
+    except ImportError as e:
+        pytest.fail(f"Failed to import Telemetry class: {e}")
+
+
+def test_telemetry_configuration():
+    """Test that the Telemetry class can be configured with different options.
+    
+    This test verifies that the Telemetry class respects environment variables
+    for disabling telemetry collection.
+    """
+    import os
+
+    from src.crewai.telemetry.telemetry import Telemetry
+    
+    # Test with telemetry disabled via environment variable
+    os.environ["OTEL_SDK_DISABLED"] = "true"
+    telemetry = Telemetry()
+    assert telemetry.ready is False
+    
+    # Reset environment variable
+    os.environ.pop("OTEL_SDK_DISABLED", None)
+
+
+def test_span_creation():
+    """Test that spans can be created with the current OpenTelemetry versions.
+    
+    This test verifies that the Telemetry class can create spans using the
+    OpenTelemetry tracer, which is a core functionality for telemetry.
+    """
+    import os
+
+    from opentelemetry import trace
+
+    from src.crewai.telemetry.telemetry import Telemetry
+    
+    # Ensure telemetry is enabled for this test
+    if "OTEL_SDK_DISABLED" in os.environ:
+        old_value = os.environ.pop("OTEL_SDK_DISABLED")
+    
+    try:
+        telemetry = Telemetry()
+        telemetry.set_tracer()
+        
+        # Only test span creation if telemetry is ready
+        if telemetry.ready and telemetry.trace_set:
+            tracer = trace.get_tracer("crewai.telemetry.test")
+            if tracer:
+                with tracer.start_as_current_span("test_operation") as span:
+                    assert span is not None
+                    assert span.is_recording()
+    except Exception as e:
+        pytest.fail(f"Failed to create span: {e}")
+    finally:
+        # Restore environment variable if it was set
+        if "old_value" in locals():
+            os.environ["OTEL_SDK_DISABLED"] = old_value