adding more agent creation traces (#44263)

* adding more agent creation traces

* change based on review comments

Author: M-Hietala

sdk/ai/azure-ai-projects/azure/ai/projects/telemetry/_ai_project_instrumentor.py

@@ -40,6 +40,16 @@
     GEN_AI_RUN_STEP_END_TIMESTAMP,
     GEN_AI_RUN_STEP_STATUS,
     GEN_AI_AGENT_VERSION,
+    GEN_AI_AGENT_HOSTED_CPU,
+    GEN_AI_AGENT_HOSTED_MEMORY,
+    GEN_AI_AGENT_HOSTED_IMAGE,
+    GEN_AI_AGENT_HOSTED_PROTOCOL,
+    GEN_AI_AGENT_HOSTED_PROTOCOL_VERSION,
+    AGENT_TYPE_PROMPT,
+    AGENT_TYPE_WORKFLOW,
+    AGENT_TYPE_HOSTED,
+    AGENT_TYPE_UNKNOWN,
+    GEN_AI_AGENT_TYPE,
     ERROR_MESSAGE,
     OperationName,
     start_span,
@@ -530,6 +540,11 @@ def start_create_agent_span(  # pylint: disable=too-many-locals
         structured_inputs: Optional[Any] = None,
         agent_type: Optional[str] = None,
         workflow_yaml: Optional[str] = None,
+        hosted_cpu: Optional[str] = None,
+        hosted_memory: Optional[str] = None,
+        hosted_image: Optional[str] = None,
+        hosted_protocol: Optional[str] = None,
+        hosted_protocol_version: Optional[str] = None,
     ) -> "Optional[AbstractSpan]":
         span = start_span(
             OperationName.CREATE_AGENT,
@@ -551,7 +566,19 @@ def start_create_agent_span(  # pylint: disable=too-many-locals
             if description:
                 span.add_attribute(GEN_AI_AGENT_DESCRIPTION, description)
             if agent_type:
-                span.add_attribute("gen_ai.agent.type", agent_type)
+                span.add_attribute(GEN_AI_AGENT_TYPE, agent_type)
+
+            # Add hosted agent specific attributes
+            if hosted_cpu:
+                span.add_attribute(GEN_AI_AGENT_HOSTED_CPU, hosted_cpu)
+            if hosted_memory:
+                span.add_attribute(GEN_AI_AGENT_HOSTED_MEMORY, hosted_memory)
+            if hosted_image:
+                span.add_attribute(GEN_AI_AGENT_HOSTED_IMAGE, hosted_image)
+            if hosted_protocol:
+                span.add_attribute(GEN_AI_AGENT_HOSTED_PROTOCOL, hosted_protocol)
+            if hosted_protocol_version:
+                span.add_attribute(GEN_AI_AGENT_HOSTED_PROTOCOL_VERSION, hosted_protocol_version)
 
             # Extract response schema from text parameter if available
             response_schema = None
@@ -647,19 +674,41 @@ def _create_agent_span_from_parameters(
         workflow_yaml = definition.get("workflow")  # Extract workflow YAML for workflow agents
         # toolset = definition.get("toolset")
 
+        # Extract hosted agent specific attributes
+        hosted_cpu = definition.get("cpu")
+        hosted_memory = definition.get("memory")
+        hosted_image = definition.get("image")
+        hosted_protocol = None
+        hosted_protocol_version = None
+        container_protocol_versions = definition.get("container_protocol_versions")
+        if container_protocol_versions and len(container_protocol_versions) > 0:
+            # Extract protocol and version from first entry
+            protocol_record = container_protocol_versions[0]
+            if hasattr(protocol_record, "protocol"):
+                hosted_protocol = getattr(protocol_record, "protocol", None)
+            elif isinstance(protocol_record, dict):
+                hosted_protocol = protocol_record.get("protocol")
+
+            if hasattr(protocol_record, "version"):
+                hosted_protocol_version = getattr(protocol_record, "version", None)
+            elif isinstance(protocol_record, dict):
+                hosted_protocol_version = protocol_record.get("version")
+
         # Determine agent type from definition
-        # Check for workflow first (most specific)
+        # Check for hosted agent first (most specific - has container/image configuration)
         agent_type = None
-        if workflow_yaml:
-            agent_type = "workflow"
+        if hosted_image or hosted_cpu or hosted_memory:
+            agent_type = AGENT_TYPE_HOSTED
+        elif workflow_yaml:
+            agent_type = AGENT_TYPE_WORKFLOW
         elif instructions or model:
             # Prompt agent - identified by having instructions and/or a model.
             # Note: An agent with only a model (no instructions) is treated as a prompt agent,
             # though this is uncommon. Typically prompt agents have both model and instructions.
-            agent_type = "prompt"
+            agent_type = AGENT_TYPE_PROMPT
         else:
             # Unknown type - set to "unknown" to indicate we couldn't determine it
-            agent_type = "unknown"
+            agent_type = AGENT_TYPE_UNKNOWN
 
         # Extract reasoning effort and summary from reasoning if available
         reasoning_effort = None
@@ -739,6 +788,11 @@ def _create_agent_span_from_parameters(
             structured_inputs=structured_inputs,
             agent_type=agent_type,
             workflow_yaml=workflow_yaml,
+            hosted_cpu=hosted_cpu,
+            hosted_memory=hosted_memory,
+            hosted_image=hosted_image,
+            hosted_protocol=hosted_protocol,
+            hosted_protocol_version=hosted_protocol_version,
         )
 
     def trace_create_agent(self, function, *args, **kwargs):

sdk/ai/azure-ai-projects/azure/ai/projects/telemetry/_utils.py

@@ -74,6 +74,13 @@
 GEN_AI_RESPONSE_OBJECT = "gen_ai.response.object"
 GEN_AI_TOKEN_TYPE = "gen_ai.token.type"
 GEN_AI_MESSAGE_ROLE = "gen_ai.message.role"
+GEN_AI_AGENT_TYPE = "gen_ai.agent.type"
+GEN_AI_CONVERSATION_ITEM_TYPE = "gen_ai.conversation.item.type"
+GEN_AI_AGENT_HOSTED_CPU = "gen_ai.agent.hosted.cpu"
+GEN_AI_AGENT_HOSTED_MEMORY = "gen_ai.agent.hosted.memory"
+GEN_AI_AGENT_HOSTED_IMAGE = "gen_ai.agent.hosted.image"
+GEN_AI_AGENT_HOSTED_PROTOCOL = "gen_ai.agent.hosted.protocol"
+GEN_AI_AGENT_HOSTED_PROTOCOL_VERSION = "gen_ai.agent.hosted.protocol_version"
 
 # Event names
 GEN_AI_USER_MESSAGE_EVENT = "gen_ai.input.messages"
@@ -88,15 +95,13 @@
 GEN_AI_CLIENT_OPERATION_DURATION = "gen_ai.client.operation.duration"
 GEN_AI_CLIENT_TOKEN_USAGE = "gen_ai.client.token.usage"
 
-# Additional attribute names for agents
-GEN_AI_AGENT_TYPE = "gen_ai.agent.type"
-GEN_AI_CONVERSATION_ITEM_TYPE = "gen_ai.conversation.item.type"
-
 # Constant attribute values
 AZURE_AI_AGENTS_SYSTEM = "az.ai.agents"
 AZURE_AI_AGENTS_PROVIDER = "azure.ai.agents"
 AGENT_TYPE_PROMPT = "prompt"
 AGENT_TYPE_WORKFLOW = "workflow"
+AGENT_TYPE_HOSTED = "hosted"
+AGENT_TYPE_UNKNOWN = "unknown"
 
 
 class OperationName(Enum):