Merge branch 'main' into actionlint

Author: mikeharder

sdk/agentserver/azure-ai-agentserver-core/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Release History
 
+## 2.0.0b6 (2026-06-12)
+
+### Bugs Fixed
+
+- Populated agent metadata when operation IDs are zeroed so agent metadata remains available for telemetry and downstream processing.
+- Suppressed noisy observability/exporter INFO logs by default in tracing setup while preserving DEBUG visibility when explicitly enabled.
+
 ## 2.0.0b5 (2026-05-25)
 
 ### Bugs Fixed

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

@@ -345,6 +345,15 @@ def resolve_project_id() -> str:
     return os.environ.get(_ENV_FOUNDRY_PROJECT_ARM_ID, "")
 
 
+def resolve_session_id() -> str:
+    """Resolve the default session ID from the ``FOUNDRY_AGENT_SESSION_ID`` environment variable.
+
+    :return: The default session ID, or an empty string if not set.
+    :rtype: str
+    """
+    return os.environ.get(_ENV_FOUNDRY_AGENT_SESSION_ID, "")
+
+
 def resolve_sse_keepalive_interval(interval: Optional[int] = None) -> int:
     """Resolve the SSE keep-alive interval from argument, env var, or default.
 

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

@@ -146,10 +146,16 @@ def configure_observability(
             logging.getLogger(_noisy).setLevel(logging.WARNING)
 
     # Tracing and OTel export
-    _configure_tracing(connection_string=connection_string, enable_sensitive_data=enable_sensitive_data)
+    _configure_tracing(
+        connection_string=connection_string,
+        enable_sensitive_data=enable_sensitive_data,
+    )
 
 
-def _configure_tracing(connection_string: Optional[str] = None, enable_sensitive_data: bool = False) -> None:
+def _configure_tracing(
+    connection_string: Optional[str] = None,
+    enable_sensitive_data: bool = False,
+) -> None:
     """Configure OpenTelemetry exporters via the microsoft-opentelemetry distro.
 
     Internal helper called by :func:`configure_observability`.
@@ -182,7 +188,13 @@ def _configure_tracing(connection_string: Optional[str] = None, enable_sensitive
             agent_tenant_id=agent_tenant_id,
         ),
     ]
-    log_record_processors = [_BaggageLogRecordProcessor()]  # type: ignore[list-item]
+    log_record_processors = [  # type: ignore[list-item]
+        _BaggageLogRecordProcessor(
+            agent_name=agent_name,
+            agent_version=agent_version,
+            session_id=_config.resolve_session_id() or None,
+        ),
+    ]
 
     try:
         _setup_distro_export(
@@ -521,18 +533,44 @@ class _BaggageLogRecordProcessor:
     for end-to-end correlation.
     """
 
+    def __init__(
+        self,
+        *,
+        agent_name: Optional[str] = None,
+        agent_version: Optional[str] = None,
+        session_id: Optional[str] = None,
+    ) -> None:
+        self.agent_name = agent_name
+        self.agent_version = agent_version
+        self.session_id = session_id
+
     def on_emit(self, log_data: Any) -> None:  # pylint: disable=unused-argument
         """Copy baggage entries into the log record's attributes.
 
         :param log_data: The log data being emitted.
         :type log_data: any
         """
         try:
+            if not hasattr(log_data, "log_record") or not log_data.log_record:
+                return
+
+            attrs = log_data.log_record.attributes  # type: ignore[assignment]
+
             ctx = _otel_context.get_current()
             entries = _otel_baggage.get_all(context=ctx)
-            if entries and hasattr(log_data, 'log_record') and log_data.log_record:
+            if entries:
                 for key, value in entries.items():
-                    log_data.log_record.attributes[key] = value  # type: ignore[index]
+                    attrs[key] = value  # type: ignore[index]
+
+            if self.agent_name and _ATTR_GEN_AI_AGENT_NAME not in attrs:
+                attrs[_ATTR_GEN_AI_AGENT_NAME] = self.agent_name
+            if self.agent_version and _ATTR_GEN_AI_AGENT_VERSION not in attrs:
+                attrs[_ATTR_GEN_AI_AGENT_VERSION] = self.agent_version
+
+            bag_session = _otel_baggage.get_baggage(_BAGGAGE_SESSION_ID, context=ctx)
+            resolved_session = bag_session or self.session_id
+            if resolved_session and _ATTR_SESSION_ID not in attrs:
+                attrs[_ATTR_SESSION_ID] = resolved_session
         except Exception:  # pylint: disable=broad-except
             pass
 

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

@@ -2,4 +2,4 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 
-VERSION = "2.0.0b5"
+VERSION = "2.0.0b6"

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

@@ -15,8 +15,12 @@
     resolve_agent_name,
     resolve_agent_version,
     resolve_appinsights_connection_string,
+    resolve_session_id,
+)
+from azure.ai.agentserver.core._tracing import (
+    _BaggageLogRecordProcessor,
+    _FoundryEnrichmentSpanProcessor,
 )
-from azure.ai.agentserver.core._tracing import _FoundryEnrichmentSpanProcessor
 
 
 class _CollectorExporter(SpanExporter):
@@ -408,7 +412,7 @@ def test_invocation_id_propagates_to_child_spans(self) -> None:
 
 
 class TestAgentIdentityResolution:
-    """Tests for resolve_agent_name() and resolve_agent_version()."""
+    """Tests for agent identity/session resolution helpers."""
 
     def test_agent_name_from_env(self) -> None:
         with mock.patch.dict(os.environ, {"FOUNDRY_AGENT_NAME": "my-agent"}):
@@ -430,5 +434,78 @@ def test_agent_version_default_empty(self) -> None:
         with mock.patch.dict(os.environ, env, clear=True):
             assert resolve_agent_version() == ""
 
+    def test_session_id_from_env(self) -> None:
+        with mock.patch.dict(os.environ, {"FOUNDRY_AGENT_SESSION_ID": "session-1"}):
+            assert resolve_session_id() == "session-1"
+
+    def test_session_id_default_empty(self) -> None:
+        env = os.environ.copy()
+        env.pop("FOUNDRY_AGENT_SESSION_ID", None)
+        with mock.patch.dict(os.environ, env, clear=True):
+            assert resolve_session_id() == ""
+
+
+class _FakeLogRecord:
+    def __init__(self, attributes):
+        self.attributes = attributes
+
+
+class _FakeLogData:
+    def __init__(self, attributes):
+        self.log_record = _FakeLogRecord(attributes)
+
+
+class TestBaggageLogRecordProcessor:
+    def test_adds_agent_and_fallback_session_attributes(self) -> None:
+        proc = _BaggageLogRecordProcessor(
+            agent_name="agent-a",
+            agent_version="1.2.3",
+            session_id="session-fallback-1",
+        )
+        log_data = _FakeLogData({})
+
+        proc.on_emit(log_data)
 
+        attrs = log_data.log_record.attributes
+        assert attrs["gen_ai.agent.name"] == "agent-a"
+        assert attrs["gen_ai.agent.version"] == "1.2.3"
+        assert attrs["microsoft.session.id"] == "session-fallback-1"
 
+    def test_prefers_baggage_session_id_over_fallback(self) -> None:
+        proc = _BaggageLogRecordProcessor(
+            agent_name="agent-a",
+            agent_version="1.2.3",
+            session_id="session-fallback-1",
+        )
+        log_data = _FakeLogData({})
+
+        ctx = _otel_baggage.set_baggage(
+            "azure.ai.agentserver.session_id", "session-from-baggage",
+        )
+        token = _otel_context.attach(ctx)
+        try:
+            proc.on_emit(log_data)
+        finally:
+            _otel_context.detach(token)
+
+        attrs = log_data.log_record.attributes
+        assert attrs["microsoft.session.id"] == "session-from-baggage"
+
+    def test_does_not_overwrite_existing_log_attributes(self) -> None:
+        proc = _BaggageLogRecordProcessor(
+            agent_name="agent-a",
+            agent_version="1.2.3",
+            session_id="session-fallback-1",
+        )
+        attrs = {
+            "gen_ai.agent.name": "existing-name",
+            "gen_ai.agent.version": "0.0.1",
+            "microsoft.session.id": "existing-session",
+        }
+        log_data = _FakeLogData(attrs)
+
+        proc.on_emit(log_data)
+
+        assert attrs["gen_ai.agent.name"] == "existing-name"
+        assert attrs["gen_ai.agent.version"] == "0.0.1"
+        assert attrs["microsoft.session.id"] == "existing-session"

sdk/agentserver/azure-ai-agentserver-invocations/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Release History
 
+## 1.0.0b5 (2026-06-12)
+
+### Bugs Fixed
+
+- Fixed exception tracing for streaming responses so errors raised while iterating streaming results are captured correctly and invocation/session logging context is reset after streaming completes.
+
 ## 1.0.0b4 (2026-05-21)
 
 ### Features Added

sdk/agentserver/azure-ai-agentserver-invocations/azure/ai/agentserver/invocations/_invocation.py

@@ -12,12 +12,12 @@
 import re
 import threading
 import uuid
-from collections.abc import Awaitable, Callable  # pylint: disable=import-error
+from collections.abc import AsyncIterator, Awaitable, Callable  # pylint: disable=import-error
 from typing import Any, Optional
 
-from opentelemetry import baggage as _otel_baggage, context as _otel_context
+from opentelemetry import baggage as _otel_baggage, context as _otel_context, trace
 from starlette.requests import Request
-from starlette.responses import JSONResponse, Response
+from starlette.responses import JSONResponse, Response, StreamingResponse
 from starlette.routing import Route
 
 from azure.ai.agentserver.core import (  # pylint: disable=no-name-in-module
@@ -357,6 +357,51 @@ async def _get_openapi_spec_endpoint(self, request: Request) -> Response:  # pyl
             )
         return JSONResponse(spec)
 
+    def _wrap_streaming_response(
+        self,
+        response: StreamingResponse,
+        invocation_id: str,
+        session_id: str,
+    ) -> StreamingResponse:
+        """Wrap streaming body iteration with invocation logging/tracing context.
+
+        :param response: Streaming response to wrap.
+        :type response: StreamingResponse
+        :param invocation_id: Invocation identifier to stamp in context/logging.
+        :type invocation_id: str
+        :param session_id: Session identifier to stamp in context/logging.
+        :type session_id: str
+        :return: The response with a wrapped body_iterator.
+        :rtype: StreamingResponse
+        """
+        original_iterator = response.body_iterator
+
+        async def _wrapped_body() -> AsyncIterator[Any]:
+            # Re-establish the invocation context for the streaming task.
+            stream_inv_token = _invocation_id_var.set(invocation_id)
+            stream_session_token = _session_id_var.set(session_id)
+            try:
+                async for chunk in original_iterator:
+                    yield chunk
+            except Exception as exc:  # pylint: disable=broad-exception-caught
+                logger.error(
+                    "Error processing invocation %s: %s",
+                    invocation_id, exc, exc_info=True,
+                )
+                # Record the exception on the current span.
+                span = trace.get_current_span()
+                if span and span.is_recording():
+                    span.set_status(trace.StatusCode.ERROR, str(exc))
+                    span.set_attribute("error.type", type(exc).__name__)
+                    span.record_exception(exc)
+                raise
+            finally:
+                _invocation_id_var.reset(stream_inv_token)
+                _session_id_var.reset(stream_session_token)
+
+        response.body_iterator = _wrapped_body()
+        return response
+
     async def _create_invocation_endpoint(self, request: Request) -> Response:
         generated_id = str(uuid.uuid4())
         raw_invocation_id = request.headers.get(InvocationConstants.INVOCATION_ID_HEADER) or ""
@@ -427,13 +472,21 @@ async def _create_invocation_endpoint(self, request: Request) -> Response:
                 ),
             )
         finally:
+            # Always reset the request-scope tokens and detach baggage from the
+            # calling context here. The streaming wrapper separately resets the
+            # tokens it sets for stream iteration.
             _invocation_id_var.reset(inv_token)
             _session_id_var.reset(session_token)
             try:
                 _otel_context.detach(baggage_token)
             except ValueError:
                 pass
 
+        # Wrap streaming response body so exceptions during iteration are
+        # recorded on the current trace span and logged as invocation errors.
+        if isinstance(response, StreamingResponse):
+            response = self._wrap_streaming_response(response, invocation_id, session_id)
+
         return response
 
     async def _traced_invocation_endpoint(

sdk/agentserver/azure-ai-agentserver-invocations/azure/ai/agentserver/invocations/_version.py

@@ -2,4 +2,4 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 
-VERSION = "1.0.0b4"
+VERSION = "1.0.0b5"