Remove InferenceOperationType.TEXT_COMPLETION, InferenceOperationType.GENERATE_CONTENT from the list

Author: AlexLu2016

libraries/microsoft-agents-a365-observability-core/microsoft_agents_a365/observability/core/exporters/agent365_exporter.py

@@ -18,7 +18,6 @@
 from opentelemetry.trace import StatusCode
 
 from .utils import (
-    INFERENCE_OPERATION_TYPE_NAMES,
     build_export_url,
     get_validated_domain_override,
     hex_span_id,
@@ -29,7 +28,6 @@
     status_name,
     truncate_span,
 )
-from ..constants import CHAT_OPERATION_NAME, GEN_AI_OPERATION_NAME_KEY
 
 # ---- Exporter ---------------------------------------------------------------
 
@@ -279,14 +277,6 @@ def _map_span(self, sp: ReadableSpan) -> dict[str, Any]:
         # attributes
         attrs = dict(sp.attributes or {})
 
-        # Normalize gen_ai.operation.name from any InferenceOperationType enum
-        # value (Chat, TextCompletion, GenerateContent) to the canonical "chat"
-        # value the ingest service accepts. This is applied only on the export
-        # payload; the underlying span attribute is left untouched.
-        op_name = attrs.get(GEN_AI_OPERATION_NAME_KEY)
-        if isinstance(op_name, str) and op_name in INFERENCE_OPERATION_TYPE_NAMES:
-            attrs[GEN_AI_OPERATION_NAME_KEY] = CHAT_OPERATION_NAME
-
         # events
         events = []
         for ev in sp.events:

libraries/microsoft-agents-a365-observability-core/microsoft_agents_a365/observability/core/exporters/utils.py

@@ -40,18 +40,6 @@
         OUTPUT_MESSAGES_OPERATION_NAME,
         CHAT_OPERATION_NAME,
         InferenceOperationType.CHAT.value,
-        InferenceOperationType.TEXT_COMPLETION.value,
-        InferenceOperationType.GENERATE_CONTENT.value,
-    }
-)
-
-# Inference operation type values that the ingest service expects to be
-# normalized to the canonical "chat" gen_ai.operation.name.
-INFERENCE_OPERATION_TYPE_NAMES: frozenset[str] = frozenset(
-    {
-        InferenceOperationType.CHAT.value,
-        InferenceOperationType.TEXT_COMPLETION.value,
-        InferenceOperationType.GENERATE_CONTENT.value,
     }
 )
 

tests/observability/core/test_agent365_exporter.py

@@ -703,12 +703,27 @@ def test_export_filters_out_only_non_genai_spans_returns_success(self):
             self.assertEqual(result, SpanExportResult.SUCCESS)
             mock_post.assert_not_called()
 
-    def test_export_includes_inference_operation_type_spans(self):
-        """Spans with InferenceOperationType enum values are kept and normalized."""
-        # Arrange
+    def test_export_includes_inference_operation_type_chat_spans(self):
+        """Spans with InferenceOperationType.CHAT value ('Chat') are kept without normalization."""
+        # Arrange — server accepts 'Chat' via case-insensitive matching
         chat_span = self._create_mock_span(
             "chat_span", trace_id=1, span_id=2, operation_name="Chat"
         )
+
+        with patch.object(self.exporter, "_post_with_retries", return_value=True) as mock_post:
+            result = self.exporter.export([chat_span])
+
+            self.assertEqual(result, SpanExportResult.SUCCESS)
+            mock_post.assert_called_once()
+            _, body, _ = mock_post.call_args[0]
+            request_data = json.loads(body)
+            spans_out = request_data["resourceSpans"][0]["scopeSpans"][0]["spans"]
+            self.assertEqual(len(spans_out), 1)
+            # Value is preserved as-is; no normalization
+            self.assertEqual(spans_out[0]["attributes"]["gen_ai.operation.name"], "Chat")
+
+    def test_export_filters_out_unsupported_inference_operation_types(self):
+        """Spans with TextCompletion / GenerateContent are filtered out."""
         text_completion_span = self._create_mock_span(
             "text_completion_span", trace_id=3, span_id=4, operation_name="TextCompletion"
         )
@@ -717,18 +732,11 @@ def test_export_includes_inference_operation_type_spans(self):
         )
 
         with patch.object(self.exporter, "_post_with_retries", return_value=True) as mock_post:
-            # Act
-            result = self.exporter.export([chat_span, text_completion_span, generate_content_span])
+            result = self.exporter.export([text_completion_span, generate_content_span])
 
-            # Assert: all three are exported and normalized to "chat"
+            # Both are filtered out — nothing to export
             self.assertEqual(result, SpanExportResult.SUCCESS)
-            mock_post.assert_called_once()
-            _, body, _ = mock_post.call_args[0]
-            request_data = json.loads(body)
-            spans_out = request_data["resourceSpans"][0]["scopeSpans"][0]["spans"]
-            self.assertEqual(len(spans_out), 3)
-            for span in spans_out:
-                self.assertEqual(span["attributes"]["gen_ai.operation.name"], "chat")
+            mock_post.assert_not_called()
 
     def test_export_does_not_normalize_canonical_operation_names(self):
         """invoke_agent / execute_tool / output_messages / chat are not rewritten."""