add azure monitor exporter trace ai field

Author: Karlie-777

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/export/trace/_exporter.py

@@ -407,7 +407,7 @@ def _convert_span_to_envelope(span: ReadableSpan) -> TelemetryItem:
                     span.attributes,
                 )
             elif gen_ai_attributes.GEN_AI_SYSTEM in span.attributes:  # GenAI
-                data.type = span.attributes[gen_ai_attributes.GEN_AI_SYSTEM]
+                data.type = "GenAI | {}".format(span.attributes[gen_ai_attributes.GEN_AI_SYSTEM])
             else:
                 data.type = "N/A"
         elif span.kind is SpanKind.PRODUCER:  # Messaging
@@ -428,6 +428,8 @@ def _convert_span_to_envelope(span: ReadableSpan) -> TelemetryItem:
             data.type = "InProc"
             if _AZURE_SDK_NAMESPACE_NAME in span.attributes:
                 data.type += " | {}".format(span.attributes[_AZURE_SDK_NAMESPACE_NAME])
+            elif gen_ai_attributes.GEN_AI_SYSTEM in span.attributes:  # GenAI
+                data.type = "GenAI | {}".format(span.attributes[gen_ai_attributes.GEN_AI_SYSTEM])
         # Apply truncation
         # See https://github.com/MohanGsk/ApplicationInsights-Home/tree/master/EndpointSpecs/Schemas/Bond
         if data.name:

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/export/trace/_utils.py

@@ -11,6 +11,7 @@
     user_agent_attributes,
 )
 from opentelemetry.semconv.trace import DbSystemValues, SpanAttributes
+from opentelemetry.semconv._incubating.attributes import gen_ai_attributes
 from opentelemetry.util.types import Attributes
 
 
@@ -202,6 +203,11 @@ def _get_target_and_path_for_http_dependency(
                     target = parsed_url.hostname
             elif parsed_url.netloc:
                 target = parsed_url.netloc
+        elif gen_ai_attributes.GEN_AI_SYSTEM in attributes:
+         # If no fields are available to set target using standard rules, set Dependency Target to gen_ai.system if present
+            gen_ai_system = attributes.get(gen_ai_attributes.GEN_AI_SYSTEM)
+            if gen_ai_system:
+                target = gen_ai_system
         if not target:
             # Get target from peer.* attributes that are NOT peer.service
             target = _get_target_for_dependency_from_peer(attributes)

sdk/monitor/azure-monitor-opentelemetry-exporter/tests/trace/test_trace.py

@@ -762,8 +762,32 @@ def test_span_to_envelope_client_gen_ai(self):
         self.assertEqual(envelope.data.base_data.result_code, "0")
 
         self.assertEqual(envelope.data.base_type, "RemoteDependencyData")
-        self.assertEqual(envelope.data.base_data.type, "az.ai.inference")
+        self.assertEqual(envelope.data.base_data.type, "GenAI | az.ai.inference")
         self.assertEqual(len(envelope.data.base_data.properties), 1)
+        
+    def test_span_to_envelope_client_internal_genai_type(self):
+        exporter = self._exporter
+        start_time = 1575494316027613500
+        end_time = start_time + 1001000000
+
+        span = trace._Span(
+            name="genai-test",
+            context=SpanContext(
+                trace_id=36873507687745823477771305566750195431,
+                span_id=12030755672171557337,
+                is_remote=False,
+            ),
+            attributes={
+                "gen_ai.system": "az.ai.inference",
+            },
+            kind=SpanKind.INTERNAL,
+        )
+        span.start(start_time=start_time)
+        span.end(end_time=end_time)
+        span._status = Status(status_code=StatusCode.UNSET)
+        envelope = exporter._span_to_envelope(span)
+
+        self.assertEqual(envelope.data.base_data.type, "GenAI | az.ai.inference")
 
     def test_span_to_envelope_client_azure(self):
         exporter = self._exporter