Add operation_source method to BaggageBuilder using SERVICE_NAME_KEY

Co-authored-by: nikhilNava <211831449+nikhilNava@users.noreply.github.com>

Author: Copilot

libraries/microsoft-agents-a365-observability-core/microsoft_agents_a365/observability/core/constants.py

@@ -94,6 +94,13 @@
 CHANNEL_NAME_KEY = "microsoft.channel.name"
 CHANNEL_LINK_KEY = "microsoft.channel.link"
 
+# Deprecated channel dimension keys - use CHANNEL_NAME_KEY and CHANNEL_LINK_KEY instead
+GEN_AI_EXECUTION_SOURCE_NAME_KEY = "microsoft.channel.name"
+GEN_AI_EXECUTION_SOURCE_DESCRIPTION_KEY = "microsoft.channel.link"
+
+# Event content key
+GEN_AI_EVENT_CONTENT = "gen_ai.event.content"
+
 # Custom parent id and parent name key
 CUSTOM_PARENT_SPAN_ID_KEY = "custom.parent.span.id"
 CUSTOM_SPAN_NAME_KEY = "custom.span.name"

libraries/microsoft-agents-a365-observability-core/microsoft_agents_a365/observability/core/middleware/baggage_builder.py

@@ -23,6 +23,7 @@
     GEN_AI_CALLER_UPN_KEY,
     GEN_AI_CONVERSATION_ID_KEY,
     GEN_AI_CONVERSATION_ITEM_LINK_KEY,
+    SERVICE_NAME_KEY,
     SESSION_DESCRIPTION_KEY,
     SESSION_ID_KEY,
     TENANT_ID_KEY,
@@ -55,6 +56,20 @@ def __init__(self):
         """Initialize the baggage builder."""
         self._pairs: dict[str, str] = {}
 
+    def operation_source(self, value: str | None) -> "BaggageBuilder":
+        """Set the operation source baggage value.
+
+        This captures the name of the service using the SDK.
+
+        Args:
+            value: The service name (e.g., "my-agent-service", "weather-bot")
+
+        Returns:
+            Self for method chaining
+        """
+        self._set(SERVICE_NAME_KEY, value)
+        return self
+
     def tenant_id(self, value: str | None) -> "BaggageBuilder":
         """Set the tenant ID baggage value.
 

tests/observability/core/test_baggage_builder.py

@@ -13,6 +13,7 @@
     GEN_AI_AGENT_UPN_KEY,
     GEN_AI_CALLER_CLIENT_IP_KEY,
     GEN_AI_CALLER_ID_KEY,
+    SERVICE_NAME_KEY,
     SESSION_DESCRIPTION_KEY,
     SESSION_ID_KEY,
     TENANT_ID_KEY,
@@ -319,6 +320,37 @@ def test_caller_client_ip_method(self):
             # Should be None due to proper exception handling
             self.assertIsNone(current_baggage.get(GEN_AI_CALLER_CLIENT_IP_KEY))
 
+    def test_operation_source_method(self):
+        """Test operation_source method sets service name baggage using string values."""
+        # Should exist and be callable
+        self.assertTrue(hasattr(self.builder, "operation_source"))
+        self.assertTrue(callable(self.builder.operation_source))
+
+        # Test with custom service name
+        with BaggageBuilder().operation_source("my-agent-service").build():
+            current_baggage = baggage.get_all()
+            self.assertEqual(current_baggage.get(SERVICE_NAME_KEY), "my-agent-service")
+
+        # Test with another service name
+        with BaggageBuilder().operation_source("weather-bot").build():
+            current_baggage = baggage.get_all()
+            self.assertEqual(current_baggage.get(SERVICE_NAME_KEY), "weather-bot")
+
+        # Test with SDK as string
+        with BaggageBuilder().operation_source("SDK").build():
+            current_baggage = baggage.get_all()
+            self.assertEqual(current_baggage.get(SERVICE_NAME_KEY), "SDK")
+
+        # Test with None value (should not set baggage)
+        with BaggageBuilder().operation_source(None).build():
+            current_baggage = baggage.get_all()
+            self.assertIsNone(current_baggage.get(SERVICE_NAME_KEY))
+
+        # Test with whitespace-only value (should not set baggage)
+        with BaggageBuilder().operation_source("   ").build():
+            current_baggage = baggage.get_all()
+            self.assertIsNone(current_baggage.get(SERVICE_NAME_KEY))
+
 
 if __name__ == "__main__":
     unittest.main()