add product context fallback

Author: Grant Harris

libraries/microsoft-agents-a365-observability-hosting/microsoft_agents_a365/observability/hosting/scope_helpers/utils.py

@@ -74,13 +74,36 @@ def get_channel_pairs(activity: Activity) -> Iterator[tuple[str, Any]]:
     sub_channel = None
 
     if channel_id is not None:
-        if isinstance(channel_id, str):
-            # Direct string value
-            channel_name = channel_id
-        elif hasattr(channel_id, "channel"):
+        # Check for ChannelId object first
+        if hasattr(channel_id, "channel"):
             # ChannelId object
             channel_name = channel_id.channel
             sub_channel = channel_id.sub_channel
+        elif isinstance(channel_id, str):
+            # Direct string value
+            channel_name = channel_id
+
+    # Try to get sub_channel from productContext in channel_data if sub_channel is not set
+    if not sub_channel and activity.channel_data:
+        try:
+            import json
+            # Convert channel_data to dict if it's a string
+            if isinstance(activity.channel_data, str):
+                channel_data_dict = json.loads(activity.channel_data)
+            elif isinstance(activity.channel_data, dict):
+                channel_data_dict = activity.channel_data
+            else:
+                # Try to convert to dict if it has __dict__
+                channel_data_dict = getattr(activity.channel_data, '__dict__', {})
+            
+            # Extract productContext if available
+            if isinstance(channel_data_dict, dict) and 'productContext' in channel_data_dict:
+                product_context = channel_data_dict['productContext']
+                if isinstance(product_context, str):
+                    sub_channel = product_context
+        except (json.JSONDecodeError, AttributeError, TypeError):
+            # Silently ignore any parsing errors
+            pass
 
     # Yield channel name as source name
     yield CHANNEL_NAME_KEY, channel_name

tests/observability/hosting/middleware/test_baggage_middleware.py

@@ -95,3 +95,84 @@ async def logic():
     assert logic_called is True
     # Baggage should NOT be set because the middleware skipped it
     assert captured_caller_id is None
+
+
+@pytest.mark.asyncio
+async def test_baggage_middleware_extracts_product_context_from_channel_data():
+    """BaggageMiddleware should extract productContext from channel_data when sub_channel is not set."""
+    from microsoft_agents.activity import ChannelId
+    from microsoft_agents_a365.observability.core.constants import CHANNEL_LINK_KEY
+
+    middleware = BaggageMiddleware()
+    
+    # Create activity with ChannelId (no sub_channel) and channel_data with productContext
+    activity = Activity(
+        type="message",
+        text="Hello",
+        from_property=ChannelAccount(
+            aad_object_id="caller-id",
+            name="Caller",
+        ),
+        recipient=ChannelAccount(
+            tenant_id="tenant-123",
+            name="Agent",
+        ),
+        conversation=ConversationAccount(id="conv-id"),
+        service_url="https://example.com",
+        channel_id=ChannelId(channel="msteams"),  # No sub_channel
+        channel_data={"productContext": "COPILOT"},
+    )
+    
+    adapter = MagicMock()
+    ctx = TurnContext(adapter, activity)
+
+    captured_channel_link = None
+
+    async def logic():
+        nonlocal captured_channel_link
+        captured_channel_link = baggage.get_baggage(CHANNEL_LINK_KEY)
+
+    await middleware.on_turn(ctx, logic)
+
+    assert captured_channel_link == "COPILOT"
+
+
+@pytest.mark.asyncio
+async def test_baggage_middleware_sub_channel_takes_precedence_over_product_context():
+    """BaggageMiddleware should use sub_channel when both sub_channel and productContext are present."""
+    from microsoft_agents.activity import ChannelId
+    from microsoft_agents_a365.observability.core.constants import CHANNEL_LINK_KEY
+
+    middleware = BaggageMiddleware()
+    
+    # Create activity with BOTH sub_channel and productContext in channel_data
+    activity = Activity(
+        type="message",
+        text="Hello",
+        from_property=ChannelAccount(
+            aad_object_id="caller-id",
+            name="Caller",
+        ),
+        recipient=ChannelAccount(
+            tenant_id="tenant-123",
+            name="Agent",
+        ),
+        conversation=ConversationAccount(id="conv-id"),
+        service_url="https://example.com",
+        channel_id=ChannelId(channel="msteams", sub_channel="teams-subchannel"),
+        channel_data={"productContext": "COPILOT"},  # Should be ignored
+    )
+    
+    adapter = MagicMock()
+    ctx = TurnContext(adapter, activity)
+
+    captured_channel_link = None
+
+    async def logic():
+        nonlocal captured_channel_link
+        captured_channel_link = baggage.get_baggage(CHANNEL_LINK_KEY)
+
+    await middleware.on_turn(ctx, logic)
+
+    # sub_channel should take precedence, productContext should be ignored
+    assert captured_channel_link == "teams-subchannel"