Correctly sets OTel trace provider.

Author: juliomenendez

tests/observability/core/test_execute_tool_scope.py

@@ -11,19 +11,17 @@
     AgentDetails,
     ExecutionType,
     ExecuteToolScope,
-    OpenTelemetryScope,
     Request,
     SourceMetadata,
     TenantDetails,
     ToolCallDetails,
     configure,
+    get_tracer_provider,
 )
 from microsoft_agents_a365.observability.core.constants import (
     GEN_AI_EXECUTION_SOURCE_DESCRIPTION_KEY,
     GEN_AI_EXECUTION_SOURCE_NAME_KEY,
 )
-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
 
@@ -37,12 +35,6 @@ def setUpClass(cls):
         # Configure Microsoft Agent 365 for testing
         os.environ["ENABLE_A365_OBSERVABILITY"] = "true"
 
-        # Set up tracer to capture spans
-        cls.span_exporter = InMemorySpanExporter()
-        tracer_provider = TracerProvider()
-        tracer_provider.add_span_processor(SimpleSpanProcessor(cls.span_exporter))
-        trace.set_tracer_provider(tracer_provider)
-
         configure(
             service_name="test-execute-tool-service",
             service_namespace="test-namespace",
@@ -61,6 +53,19 @@ def setUpClass(cls):
             description="Get current weather information for a location",
         )
 
+    def setUp(self):
+        super().setUp()
+
+        # Set up tracer to capture spans
+        self.span_exporter = InMemorySpanExporter()
+        tracer_provider = get_tracer_provider()
+        tracer_provider.add_span_processor(SimpleSpanProcessor(self.span_exporter))
+
+    def tearDown(self):
+        super().tearDown()
+
+        self.span_exporter.clear()
+
     def test_record_response_method_exists(self):
         """Test that record_response method exists on ExecuteToolScope."""
         scope = ExecuteToolScope.start(self.tool_details, self.agent_details, self.tenant_details)

tests/observability/core/test_inference_scope.py

@@ -12,19 +12,17 @@
     InferenceCallDetails,
     InferenceOperationType,
     InferenceScope,
-    OpenTelemetryScope,
     Request,
     SourceMetadata,
     TenantDetails,
     configure,
+    get_tracer_provider,
 )
 from microsoft_agents_a365.observability.core.agent_details import AgentDetails
 from microsoft_agents_a365.observability.core.constants import (
     GEN_AI_EXECUTION_SOURCE_DESCRIPTION_KEY,
     GEN_AI_EXECUTION_SOURCE_NAME_KEY,
 )
-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
 
@@ -38,12 +36,6 @@ def setUpClass(cls):
         # Configure Microsoft Agent 365 for testing
         os.environ["ENABLE_A365_OBSERVABILITY"] = "true"
 
-        # Set up tracer to capture spans
-        cls.span_exporter = InMemorySpanExporter()
-        tracer_provider = TracerProvider()
-        tracer_provider.add_span_processor(SimpleSpanProcessor(cls.span_exporter))
-        trace.set_tracer_provider(tracer_provider)
-
         configure(
             service_name="test-inference-service",
             service_namespace="test-namespace",
@@ -52,6 +44,20 @@ def setUpClass(cls):
         cls.agent_details = AgentDetails(agent_id="test-inference-agent")
         cls.tenant_details = TenantDetails(tenant_id="12345678-1234-5678-1234-567812345678")
 
+    def setUp(self):
+        super().setUp()
+
+        # Set up tracer to capture spans
+        self.span_exporter = InMemorySpanExporter()
+        tracer_provider = get_tracer_provider()
+        tracer_provider.add_span_processor(SimpleSpanProcessor(self.span_exporter))
+        # trace.set_tracer_provider(tracer_provider)
+
+    def tearDown(self):
+        super().tearDown()
+
+        self.span_exporter.clear()
+
     def test_inference_operation_type_enum(self):
         """Test InferenceOperationType enum values."""
         # Test enum values exist

tests/observability/core/test_invoke_agent_scope.py

@@ -17,6 +17,7 @@
     SourceMetadata,
     TenantDetails,
     configure,
+    get_tracer_provider,
 )
 from microsoft_agents_a365.observability.core.constants import (
     GEN_AI_CALLER_AGENT_USER_CLIENT_IP,
@@ -26,8 +27,6 @@
     GEN_AI_INPUT_MESSAGES_KEY,
 )
 from microsoft_agents_a365.observability.core.models.caller_details import CallerDetails
-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
 
@@ -41,12 +40,6 @@ def setUpClass(cls):
         # Configure Microsoft Agent 365 for testing
         os.environ["ENABLE_A365_OBSERVABILITY"] = "true"
 
-        # Set up tracer to capture spans
-        cls.span_exporter = InMemorySpanExporter()
-        tracer_provider = TracerProvider()
-        tracer_provider.add_span_processor(SimpleSpanProcessor(cls.span_exporter))
-        trace.set_tracer_provider(tracer_provider)
-
         configure(
             service_name="test-invoke-agent-service",
             service_namespace="test-namespace",
@@ -101,6 +94,19 @@ def setUpClass(cls):
             agent_client_ip="192.168.1.100",
         )
 
+    def setUp(self):
+        super().setUp()
+
+        # Set up tracer to capture spans
+        self.span_exporter = InMemorySpanExporter()
+        tracer_provider = get_tracer_provider()
+        tracer_provider.add_span_processor(SimpleSpanProcessor(self.span_exporter))
+
+    def tearDown(self):
+        super().tearDown()
+
+        self.span_exporter.clear()
+
     def test_record_response_method_exists(self):
         """Test that record_response method exists on InvokeAgentScope."""
         scope = InvokeAgentScope.start(self.invoke_details, self.tenant_details)
@@ -185,9 +191,8 @@ def test_caller_agent_client_ip_in_scope(self):
         """Test that caller agent client IP is properly handled when creating InvokeAgentScope."""
         # Set up tracer to capture spans
         span_exporter = InMemorySpanExporter()
-        tracer_provider = TracerProvider()
+        tracer_provider = get_tracer_provider()
         tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter))
-        trace.set_tracer_provider(tracer_provider)
 
         # Create scope with caller agent details that include client IP
         scope = InvokeAgentScope.start(