fix tracing for operation id to flow in exception table.

sdk/agentserver/azure-ai-agentserver-core/azure/ai/agentserver/core/_tracing.py

@@ -33,6 +33,7 @@
 OpenTelemetry is a required dependency — these functions always create
 real spans.  Azure Monitor export is optional (auto-configured by the distro).
 """
+import asyncio
 import logging
 import os
 from collections.abc import AsyncIterable, AsyncIterator  # pylint: disable=import-error
@@ -294,14 +295,25 @@ async def __call__(self, scope: Any, receive: Any, send: Any) -> None:
                 "x_request_id", x_request_id, context=ctx,
             )
 
+        # Create a NonRecordingSpan with the extracted trace context so that
+        # get_current_span() returns a span carrying the correct trace_id.
+        # Without this, the OTel LogRecord processor sees no active span and
+        # sets trace_id=0, causing zeroed operation_Id in Application Insights
+        # for logs emitted by Hypercorn's access logger.
+        span = trace.get_current_span(ctx)
+        span_ctx = span.get_span_context()
+        if span_ctx and span_ctx.is_valid:
+            non_recording = trace.NonRecordingSpan(span_ctx)
+            ctx = trace.set_span_in_context(non_recording, ctx)
+
+        # Attach request context for app execution and defer detach by one
+        # event-loop turn so server-side access logging that runs immediately
+        # after app return can still observe the request trace context.
         token = _otel_context.attach(ctx)
         try:
             await self.app(scope, receive, send)
         finally:
-            try:
-                _otel_context.detach(token)
-            except ValueError:
-                pass
+            asyncio.get_running_loop().call_soon(detach_context, token)
 
 
 def end_span(span: Any, exc: Optional[BaseException] = None) -> None:

sdk/agentserver/azure-ai-agentserver-core/tests/test_tracing.py

@@ -2,6 +2,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 """Tests for tracing configuration — not invocation spans (those live in the invocations package)."""
+import asyncio
 import os
 from unittest import mock
 
@@ -16,7 +17,7 @@
     resolve_agent_version,
     resolve_appinsights_connection_string,
 )
-from azure.ai.agentserver.core._tracing import _FoundryEnrichmentSpanProcessor
+from azure.ai.agentserver.core._tracing import TraceContextMiddleware, _FoundryEnrichmentSpanProcessor
 
 
 class _CollectorExporter(SpanExporter):
@@ -431,4 +432,69 @@ def test_agent_version_default_empty(self) -> None:
             assert resolve_agent_version() == ""
 
 
+class TestTraceContextMiddleware:
+    """Trace context middleware lifecycle behavior."""
+
+    def test_detach_is_deferred_until_after_app_returns(self) -> None:
+        events = []
+
+        async def app(scope, receive, send):
+            events.append("app-called")
+
+        async def run_test():
+            middleware = TraceContextMiddleware(app)
+            scope = {"type": "http", "headers": []}
+
+            async def receive():
+                return {}
+
+            async def send(_message):
+                return None
+
+            with mock.patch("azure.ai.agentserver.core._tracing._otel_context.attach", return_value="token") as attach_mock, \
+                    mock.patch("azure.ai.agentserver.core._tracing._otel_context.detach") as detach_mock:
+                await middleware(scope, receive, send)
+                assert events == ["app-called"]
+                attach_mock.assert_called_once()
+                detach_mock.assert_not_called()
+
+                await asyncio.sleep(0)
+                detach_mock.assert_called_once_with("token")
+
+        asyncio.run(run_test())
+
+    def test_detach_still_happens_when_app_raises(self) -> None:
+        class _AppError(RuntimeError):
+            pass
+
+        async def app(scope, receive, send):
+            raise _AppError("boom")
+
+        async def run_test():
+            middleware = TraceContextMiddleware(app)
+            scope = {"type": "http", "headers": []}
+
+            async def receive():
+                return {}
+
+            async def send(_message):
+                return None
+
+            with mock.patch("azure.ai.agentserver.core._tracing._otel_context.attach", return_value="token") as attach_mock, \
+                    mock.patch("azure.ai.agentserver.core._tracing._otel_context.detach") as detach_mock:
+                try:
+                    await middleware(scope, receive, send)
+                except _AppError:
+                    pass
+                else:
+                    raise AssertionError("expected middleware to propagate app exception")
+
+                attach_mock.assert_called_once()
+                detach_mock.assert_not_called()
+
+                await asyncio.sleep(0)
+                detach_mock.assert_called_once_with("token")
+
+        asyncio.run(run_test())
+