microsoft-agents-a365-observability-core 0.2.1.dev45__tar.gz → 0.2.1.dev46__tar.gz

{microsoft_agents_a365_observability_core-0.2.1.dev45 → microsoft_agents_a365_observability_core-0.2.1.dev46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-a365-observability-core
-Version: 0.2.1.dev45
+Version: 0.2.1.dev46
 Summary: Telemetry, tracing, and monitoring components for AI agents
 Author-email: Microsoft <support@microsoft.com>
 License: MIT

{microsoft_agents_a365_observability_core-0.2.1.dev45 → microsoft_agents_a365_observability_core-0.2.1.dev46}/microsoft_agents_a365/observability/core/__init__.py

@@ -11,7 +11,6 @@ from .config import (
     is_configured,
 )
 from .execute_tool_scope import ExecuteToolScope
-from .execution_type import ExecutionType
 from .exporters.agent365_exporter_options import Agent365ExporterOptions
 from .exporters.enriched_span import EnrichedReadableSpan
 from .exporters.enriching_span_processor import (
@@ -23,13 +22,16 @@ from .exporters.spectra_exporter_options import SpectraExporterOptions
 from .inference_call_details import InferenceCallDetails, ServiceEndpoint
 from .inference_operation_type import InferenceOperationType
 from .inference_scope import InferenceScope
-from .invoke_agent_details import InvokeAgentDetails
+from .invoke_agent_details import InvokeAgentScopeDetails
 from .invoke_agent_scope import InvokeAgentScope
 from .middleware.baggage_builder import BaggageBuilder
+from .models.caller_details import CallerDetails
+from .models.user_details import UserDetails
 from .opentelemetry_scope import OpenTelemetryScope
 from .request import Request
 from .channel import Channel
-from .tenant_details import TenantDetails
+from .span_details import SpanDetails
+from .spans_scopes.output_scope import OutputScope
 from .tool_call_details import ToolCallDetails
 from .tool_type import ToolType
 from .trace_processor.span_processor import SpanProcessor
@@ -57,26 +59,26 @@ __all__ = [
     "ExecuteToolScope",
     "InvokeAgentScope",
     "InferenceScope",
+    "OutputScope",
     # Middleware
     "BaggageBuilder",
     # Data classes
-    "InvokeAgentDetails",
+    "InvokeAgentScopeDetails",
     "AgentDetails",
-    "TenantDetails",
+    "CallerDetails",
+    "UserDetails",
     "ToolCallDetails",
     "Channel",
     "Request",
+    "SpanDetails",
     "InferenceCallDetails",
     "ServiceEndpoint",
     # Enums
-    "ExecutionType",
     "InferenceOperationType",
     "ToolType",
     # Utility functions
     "extract_context_from_headers",
     "get_traceparent",
-    # Constants
-    # all constants from constants.py are exported via *
 ]
 
 # This is a namespace package

{microsoft_agents_a365_observability_core-0.2.1.dev45 → microsoft_agents_a365_observability_core-0.2.1.dev46}/microsoft_agents_a365/observability/core/agent_details.py

@@ -18,11 +18,11 @@ class AgentDetails:
     agent_description: Optional[str] = None
     """A description of the AI agent's purpose or capabilities."""
 
-    agent_auid: Optional[str] = None
+    agentic_user_id: Optional[str] = None
     """Agentic User ID for the agent."""
 
-    agent_upn: Optional[str] = None
-    """User Principal Name (UPN) for the agentic user."""
+    agentic_user_email: Optional[str] = None
+    """Email address for the agentic user."""
 
     agent_blueprint_id: Optional[str] = None
     """Blueprint/Application ID for the agent."""
@@ -33,9 +33,6 @@ class AgentDetails:
     tenant_id: Optional[str] = None
     """Tenant ID for the agent."""
 
-    conversation_id: Optional[str] = None
-    """Optional conversation ID for compatibility."""
-
     icon_uri: Optional[str] = None
     """Optional icon URI for the agent."""
 

{microsoft_agents_a365_observability_core-0.2.1.dev45 → microsoft_agents_a365_observability_core-0.2.1.dev46}/microsoft_agents_a365/observability/core/constants.py

@@ -87,8 +87,10 @@ GEN_AI_AGENT_AUID_KEY = "microsoft.agent.user.id"
 GEN_AI_AGENT_EMAIL_KEY = "microsoft.agent.user.email"
 GEN_AI_AGENT_BLUEPRINT_ID_KEY = "microsoft.a365.agent.blueprint.id"
 
+# Error type constants
+ERROR_TYPE_CANCELLED = "TaskCanceledException"
+
 # Execution context dimensions
-GEN_AI_EXECUTION_TYPE_KEY = "gen_ai.execution.type"
 GEN_AI_EXECUTION_PAYLOAD_KEY = "gen_ai.execution.payload"
 
 # Channel dimensions

{microsoft_agents_a365_observability_core-0.2.1.dev45 → microsoft_agents_a365_observability_core-0.2.1.dev46}/microsoft_agents_a365/observability/core/execute_tool_scope.py

@@ -1,9 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from datetime import datetime
-
-from opentelemetry.context import Context
 from opentelemetry.trace import SpanKind
 
 from .agent_details import AgentDetails
@@ -11,6 +8,8 @@ from .constants import (
     CHANNEL_LINK_KEY,
     CHANNEL_NAME_KEY,
     EXECUTE_TOOL_OPERATION_NAME,
+    GEN_AI_CALLER_CLIENT_IP_KEY,
+    GEN_AI_CONVERSATION_ID_KEY,
     GEN_AI_TOOL_ARGS_KEY,
     GEN_AI_TOOL_CALL_ID_KEY,
     GEN_AI_TOOL_DESCRIPTION_KEY,
@@ -18,11 +17,16 @@ from .constants import (
     GEN_AI_TOOL_TYPE_KEY,
     SERVER_ADDRESS_KEY,
     SERVER_PORT_KEY,
+    USER_EMAIL_KEY,
+    USER_ID_KEY,
+    USER_NAME_KEY,
 )
+from .models.user_details import UserDetails
 from .opentelemetry_scope import OpenTelemetryScope
 from .request import Request
-from .tenant_details import TenantDetails
+from .span_details import SpanDetails
 from .tool_call_details import ToolCallDetails
+from .utils import validate_and_normalize_ip
 
 
 class ExecuteToolScope(OpenTelemetryScope):
@@ -30,88 +34,71 @@ class ExecuteToolScope(OpenTelemetryScope):
 
     @staticmethod
     def start(
+        request: Request,
         details: ToolCallDetails,
         agent_details: AgentDetails,
-        tenant_details: TenantDetails,
-        request: Request | None = None,
-        parent_context: Context | None = None,
-        start_time: datetime | None = None,
-        end_time: datetime | None = None,
-        span_kind: SpanKind | None = None,
+        user_details: UserDetails | None = None,
+        span_details: SpanDetails | None = None,
     ) -> "ExecuteToolScope":
         """Creates and starts a new scope for tool execution tracing.
 
         Args:
+            request: Request details for the tool execution
             details: The details of the tool call
             agent_details: The details of the agent making the call
-            tenant_details: The details of the tenant
-            request: Optional request details for additional context
-            parent_context: Optional OpenTelemetry Context used to link this span to an
-                upstream operation. Use ``extract_context_from_headers()`` to convert a
-                Context from HTTP headers containing W3C traceparent.
-            start_time: Optional explicit start time as a datetime object. Useful when
-                recording a tool call after execution has already completed.
-            end_time: Optional explicit end time as a datetime object. When provided,
-                the span will use this timestamp when disposed instead of the
-                current wall-clock time.
-            span_kind: Optional span kind override. Defaults to ``SpanKind.INTERNAL``.
-                Use ``SpanKind.CLIENT`` when the tool calls an external service.
+            user_details: Optional human user details
+            span_details: Optional span configuration (parent context, timing, kind)
 
         Returns:
             A new ExecuteToolScope instance
         """
         return ExecuteToolScope(
+            request,
             details,
             agent_details,
-            tenant_details,
-            request,
-            parent_context,
-            start_time,
-            end_time,
-            span_kind,
+            user_details,
+            span_details,
         )
 
     def __init__(
         self,
+        request: Request,
         details: ToolCallDetails,
         agent_details: AgentDetails,
-        tenant_details: TenantDetails,
-        request: Request | None = None,
-        parent_context: Context | None = None,
-        start_time: datetime | None = None,
-        end_time: datetime | None = None,
-        span_kind: SpanKind | None = None,
+        user_details: UserDetails | None = None,
+        span_details: SpanDetails | None = None,
     ):
         """Initialize the tool execution scope.
 
         Args:
+            request: Request details for the tool execution
             details: The details of the tool call
             agent_details: The details of the agent making the call
-            tenant_details: The details of the tenant
-            request: Optional request details for additional context
-            parent_context: Optional OpenTelemetry Context used to link this span to an
-                upstream operation. Use ``extract_context_from_headers()`` to convert a
-                Context from HTTP headers containing W3C traceparent.
-            start_time: Optional explicit start time as a datetime object. Useful when
-                recording a tool call after execution has already completed.
-            end_time: Optional explicit end time as a datetime object. When provided,
-                the span will use this timestamp when disposed instead of the
-                current wall-clock time.
-            span_kind: Optional span kind override. Defaults to ``SpanKind.INTERNAL``.
-                Use ``SpanKind.CLIENT`` when the tool calls an external service.
+            user_details: Optional human user details
+            span_details: Optional span configuration (parent context, timing, kind)
         """
+        kind = SpanKind.INTERNAL
+        parent_context = None
+        start_time = None
+        end_time = None
+        if span_details is not None:
+            if span_details.span_kind is not None:
+                kind = span_details.span_kind
+            parent_context = span_details.parent_context
+            start_time = span_details.start_time
+            end_time = span_details.end_time
+
         super().__init__(
-            kind=span_kind if span_kind is not None else SpanKind.INTERNAL,
+            kind=kind,
             operation_name=EXECUTE_TOOL_OPERATION_NAME,
             activity_name=f"{EXECUTE_TOOL_OPERATION_NAME} {details.tool_name}",
             agent_details=agent_details,
-            tenant_details=tenant_details,
             parent_context=parent_context,
             start_time=start_time,
             end_time=end_time,
         )
 
-        # Extract details using deconstruction-like approach
+        # Extract details
         tool_name = details.tool_name
         arguments = details.arguments
         tool_call_id = details.tool_call_id
@@ -124,6 +111,7 @@ class ExecuteToolScope(OpenTelemetryScope):
         self.set_tag_maybe(GEN_AI_TOOL_TYPE_KEY, tool_type)
         self.set_tag_maybe(GEN_AI_TOOL_CALL_ID_KEY, tool_call_id)
         self.set_tag_maybe(GEN_AI_TOOL_DESCRIPTION_KEY, description)
+        self.set_tag_maybe(GEN_AI_CONVERSATION_ID_KEY, request.conversation_id)
 
         if endpoint:
             self.set_tag_maybe(SERVER_ADDRESS_KEY, endpoint.hostname)
@@ -131,10 +119,20 @@ class ExecuteToolScope(OpenTelemetryScope):
                 self.set_tag_maybe(SERVER_PORT_KEY, endpoint.port)
 
         # Set request metadata if provided
-        if request and request.channel:
+        if request.channel:
             self.set_tag_maybe(CHANNEL_NAME_KEY, request.channel.name)
             self.set_tag_maybe(CHANNEL_LINK_KEY, request.channel.link)
 
+        # Set user details if provided
+        if user_details:
+            self.set_tag_maybe(USER_ID_KEY, user_details.user_id)
+            self.set_tag_maybe(USER_EMAIL_KEY, user_details.user_email)
+            self.set_tag_maybe(USER_NAME_KEY, user_details.user_name)
+            self.set_tag_maybe(
+                GEN_AI_CALLER_CLIENT_IP_KEY,
+                validate_and_normalize_ip(user_details.user_client_ip),
+            )
+
     def record_response(self, response: str) -> None:
         """Records response information for telemetry tracking.
 

{microsoft_agents_a365_observability_core-0.2.1.dev45 → microsoft_agents_a365_observability_core-0.2.1.dev46}/microsoft_agents_a365/observability/core/inference_scope.py

@@ -1,16 +1,14 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from datetime import datetime
 from typing import List
 
-from opentelemetry.context import Context
-
 from .agent_details import AgentDetails
 from .constants import (
     CHANNEL_LINK_KEY,
     CHANNEL_NAME_KEY,
     GEN_AI_AGENT_THOUGHT_PROCESS_KEY,
+    GEN_AI_CONVERSATION_ID_KEY,
     GEN_AI_INPUT_MESSAGES_KEY,
     GEN_AI_OPERATION_NAME_KEY,
     GEN_AI_OUTPUT_MESSAGES_KEY,
@@ -21,12 +19,17 @@ from .constants import (
     GEN_AI_USAGE_OUTPUT_TOKENS_KEY,
     SERVER_ADDRESS_KEY,
     SERVER_PORT_KEY,
+    USER_EMAIL_KEY,
+    USER_ID_KEY,
+    USER_NAME_KEY,
+    GEN_AI_CALLER_CLIENT_IP_KEY,
 )
 from .inference_call_details import InferenceCallDetails
+from .models.user_details import UserDetails
 from .opentelemetry_scope import OpenTelemetryScope
 from .request import Request
-from .tenant_details import TenantDetails
-from .utils import safe_json_dumps
+from .span_details import SpanDetails
+from .utils import safe_json_dumps, validate_and_normalize_ip
 
 
 class InferenceScope(OpenTelemetryScope):
@@ -34,71 +37,64 @@ class InferenceScope(OpenTelemetryScope):
 
     @staticmethod
     def start(
+        request: Request,
         details: InferenceCallDetails,
         agent_details: AgentDetails,
-        tenant_details: TenantDetails,
-        request: Request | None = None,
-        parent_context: Context | None = None,
-        start_time: datetime | None = None,
-        end_time: datetime | None = None,
+        user_details: UserDetails | None = None,
+        span_details: SpanDetails | None = None,
     ) -> "InferenceScope":
         """Creates and starts a new scope for inference tracing.
 
         Args:
+            request: Request details for the inference
             details: The details of the inference call
             agent_details: The details of the agent making the call
-            tenant_details: The details of the tenant
-            request: Optional request details for additional context
-            parent_context: Optional OpenTelemetry Context used to link this span to an
-                upstream operation. Use ``extract_context_from_headers()`` to convert a
-                Context from HTTP headers containing W3C traceparent.
-            start_time: Optional explicit start time as a datetime object.
-            end_time: Optional explicit end time as a datetime object.
+            user_details: Optional human user details
+            span_details: Optional span configuration (parent context, timing)
 
         Returns:
             A new InferenceScope instance
         """
-        return InferenceScope(
-            details, agent_details, tenant_details, request, parent_context, start_time, end_time
-        )
+        return InferenceScope(request, details, agent_details, user_details, span_details)
 
     def __init__(
         self,
+        request: Request,
         details: InferenceCallDetails,
         agent_details: AgentDetails,
-        tenant_details: TenantDetails,
-        request: Request | None = None,
-        parent_context: Context | None = None,
-        start_time: datetime | None = None,
-        end_time: datetime | None = None,
+        user_details: UserDetails | None = None,
+        span_details: SpanDetails | None = None,
     ):
         """Initialize the inference scope.
 
         Args:
+            request: Request details for the inference
             details: The details of the inference call
             agent_details: The details of the agent making the call
-            tenant_details: The details of the tenant
-            request: Optional request details for additional context
-            parent_context: Optional OpenTelemetry Context used to link this span to an
-                upstream operation. Use ``extract_context_from_headers()`` to convert a
-                Context from HTTP headers containing W3C traceparent.
-            start_time: Optional explicit start time as a datetime object.
-            end_time: Optional explicit end time as a datetime object.
+            user_details: Optional human user details
+            span_details: Optional span configuration (parent context, timing)
         """
+        parent_context = None
+        start_time = None
+        end_time = None
+        if span_details is not None:
+            parent_context = span_details.parent_context
+            start_time = span_details.start_time
+            end_time = span_details.end_time
 
         super().__init__(
             kind="Client",
             operation_name=details.operationName.value,
             activity_name=f"{details.operationName.value} {details.model}",
             agent_details=agent_details,
-            tenant_details=tenant_details,
             parent_context=parent_context,
             start_time=start_time,
             end_time=end_time,
         )
 
-        if request:
+        if request.content:
             self.set_tag_maybe(GEN_AI_INPUT_MESSAGES_KEY, request.content)
+        self.set_tag_maybe(GEN_AI_CONVERSATION_ID_KEY, request.conversation_id)
 
         self.set_tag_maybe(GEN_AI_OPERATION_NAME_KEY, details.operationName.value)
         self.set_tag_maybe(GEN_AI_REQUEST_MODEL_KEY, details.model)
@@ -124,10 +120,20 @@ class InferenceScope(OpenTelemetryScope):
                 self.set_tag_maybe(SERVER_PORT_KEY, str(details.endpoint.port))
 
         # Set request metadata if provided
-        if request and request.channel:
+        if request.channel:
             self.set_tag_maybe(CHANNEL_NAME_KEY, request.channel.name)
             self.set_tag_maybe(CHANNEL_LINK_KEY, request.channel.link)
 
+        # Set user details if provided
+        if user_details:
+            self.set_tag_maybe(USER_ID_KEY, user_details.user_id)
+            self.set_tag_maybe(USER_EMAIL_KEY, user_details.user_email)
+            self.set_tag_maybe(USER_NAME_KEY, user_details.user_name)
+            self.set_tag_maybe(
+                GEN_AI_CALLER_CLIENT_IP_KEY,
+                validate_and_normalize_ip(user_details.user_client_ip),
+            )
+
     def record_input_messages(self, messages: List[str]) -> None:
         """Records the input messages for telemetry tracking.
 

microsoft_agents_a365_observability_core-0.2.1.dev46/microsoft_agents_a365/observability/core/invoke_agent_details.py

@@ -0,0 +1,14 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+# Data class for invoke agent scope details.
+
+from dataclasses import dataclass
+from urllib.parse import ParseResult
+
+
+@dataclass
+class InvokeAgentScopeDetails:
+    """Scope-level configuration for agent invocation tracing."""
+
+    endpoint: ParseResult | None = None

microsoft_agents_a365_observability_core-0.2.1.dev46/microsoft_agents_a365/observability/core/invoke_agent_scope.py

@@ -0,0 +1,187 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+# Invoke agent scope for tracing agent invocation.
+
+import logging
+
+from opentelemetry.trace import SpanKind
+
+from .agent_details import AgentDetails
+from .constants import (
+    CHANNEL_LINK_KEY,
+    CHANNEL_NAME_KEY,
+    GEN_AI_CALLER_AGENT_APPLICATION_ID_KEY,
+    GEN_AI_CALLER_AGENT_EMAIL_KEY,
+    GEN_AI_CALLER_AGENT_ID_KEY,
+    GEN_AI_CALLER_AGENT_NAME_KEY,
+    GEN_AI_CALLER_AGENT_PLATFORM_ID_KEY,
+    GEN_AI_CALLER_AGENT_USER_ID_KEY,
+    GEN_AI_CALLER_CLIENT_IP_KEY,
+    GEN_AI_CONVERSATION_ID_KEY,
+    GEN_AI_INPUT_MESSAGES_KEY,
+    GEN_AI_OUTPUT_MESSAGES_KEY,
+    INVOKE_AGENT_OPERATION_NAME,
+    SERVER_ADDRESS_KEY,
+    SERVER_PORT_KEY,
+    SESSION_ID_KEY,
+    USER_EMAIL_KEY,
+    USER_ID_KEY,
+    USER_NAME_KEY,
+)
+from .invoke_agent_details import InvokeAgentScopeDetails
+from .models.caller_details import CallerDetails
+from .opentelemetry_scope import OpenTelemetryScope
+from .request import Request
+from .span_details import SpanDetails
+from .utils import safe_json_dumps, validate_and_normalize_ip
+
+logger = logging.getLogger(__name__)
+
+
+class InvokeAgentScope(OpenTelemetryScope):
+    """Provides OpenTelemetry tracing scope for AI agent invocation operations."""
+
+    @staticmethod
+    def start(
+        request: Request,
+        scope_details: InvokeAgentScopeDetails,
+        agent_details: AgentDetails,
+        caller_details: CallerDetails | None = None,
+        span_details: SpanDetails | None = None,
+    ) -> "InvokeAgentScope":
+        """Create and start a new scope for agent invocation tracing.
+
+        Args:
+            request: Request details for the invocation
+            scope_details: Scope-level configuration (endpoint)
+            agent_details: The details of the agent being invoked
+            caller_details: Optional composite caller details (human user and/or
+                calling agent for A2A scenarios)
+            span_details: Optional span configuration (parent context, timing, kind)
+
+        Returns:
+            A new InvokeAgentScope instance
+        """
+        return InvokeAgentScope(
+            request,
+            scope_details,
+            agent_details,
+            caller_details,
+            span_details,
+        )
+
+    def __init__(
+        self,
+        request: Request,
+        scope_details: InvokeAgentScopeDetails,
+        agent_details: AgentDetails,
+        caller_details: CallerDetails | None = None,
+        span_details: SpanDetails | None = None,
+    ):
+        """Initialize the agent invocation scope.
+
+        Args:
+            request: Request details for the invocation
+            scope_details: Scope-level configuration (endpoint)
+            agent_details: The details of the agent being invoked
+            caller_details: Optional composite caller details (human user and/or
+                calling agent for A2A scenarios)
+            span_details: Optional span configuration (parent context, timing, kind)
+        """
+        activity_name = INVOKE_AGENT_OPERATION_NAME
+        if agent_details.agent_name:
+            activity_name = f"{INVOKE_AGENT_OPERATION_NAME} {agent_details.agent_name}"
+
+        kind = SpanKind.CLIENT
+        parent_context = None
+        start_time = None
+        end_time = None
+        if span_details is not None:
+            if span_details.span_kind is not None:
+                kind = span_details.span_kind
+            parent_context = span_details.parent_context
+            start_time = span_details.start_time
+            end_time = span_details.end_time
+
+        super().__init__(
+            kind=kind,
+            operation_name=INVOKE_AGENT_OPERATION_NAME,
+            activity_name=activity_name,
+            agent_details=agent_details,
+            parent_context=parent_context,
+            start_time=start_time,
+            end_time=end_time,
+        )
+
+        self.set_tag_maybe(SESSION_ID_KEY, request.session_id)
+        self.set_tag_maybe(GEN_AI_CONVERSATION_ID_KEY, request.conversation_id)
+
+        endpoint = scope_details.endpoint
+        if endpoint:
+            self.set_tag_maybe(SERVER_ADDRESS_KEY, endpoint.hostname)
+            if endpoint.port and endpoint.port != 443:
+                self.set_tag_maybe(SERVER_PORT_KEY, endpoint.port)
+
+        # Set request metadata
+        if request.channel:
+            self.set_tag_maybe(CHANNEL_NAME_KEY, request.channel.name)
+            self.set_tag_maybe(CHANNEL_LINK_KEY, request.channel.link)
+        if request.content:
+            self.set_tag_maybe(GEN_AI_INPUT_MESSAGES_KEY, safe_json_dumps([request.content]))
+
+        # Set caller details tags
+        if caller_details:
+            user_details = caller_details.user_details
+            if user_details:
+                self.set_tag_maybe(USER_ID_KEY, user_details.user_id)
+                self.set_tag_maybe(USER_EMAIL_KEY, user_details.user_email)
+                self.set_tag_maybe(USER_NAME_KEY, user_details.user_name)
+                self.set_tag_maybe(
+                    GEN_AI_CALLER_CLIENT_IP_KEY,
+                    validate_and_normalize_ip(user_details.user_client_ip),
+                )
+
+            # Set caller agent details tags
+            caller_agent_details = caller_details.caller_agent_details
+            if caller_agent_details:
+                self.set_tag_maybe(GEN_AI_CALLER_AGENT_NAME_KEY, caller_agent_details.agent_name)
+                self.set_tag_maybe(GEN_AI_CALLER_AGENT_ID_KEY, caller_agent_details.agent_id)
+                self.set_tag_maybe(
+                    GEN_AI_CALLER_AGENT_APPLICATION_ID_KEY,
+                    caller_agent_details.agent_blueprint_id,
+                )
+                self.set_tag_maybe(
+                    GEN_AI_CALLER_AGENT_USER_ID_KEY, caller_agent_details.agentic_user_id
+                )
+                self.set_tag_maybe(
+                    GEN_AI_CALLER_AGENT_EMAIL_KEY, caller_agent_details.agentic_user_email
+                )
+                self.set_tag_maybe(
+                    GEN_AI_CALLER_AGENT_PLATFORM_ID_KEY,
+                    caller_agent_details.agent_platform_id,
+                )
+
+    def record_response(self, response: str) -> None:
+        """Record response information for telemetry tracking.
+
+        Args:
+            response: The response string to record
+        """
+        self.record_output_messages([response])
+
+    def record_input_messages(self, messages: list[str]) -> None:
+        """Record the input messages for telemetry tracking.
+
+        Args:
+            messages: List of input messages to record
+        """
+        self.set_tag_maybe(GEN_AI_INPUT_MESSAGES_KEY, safe_json_dumps(messages))
+
+    def record_output_messages(self, messages: list[str]) -> None:
+        """Record the output messages for telemetry tracking.
+
+        Args:
+            messages: List of output messages to record
+        """
+        self.set_tag_maybe(GEN_AI_OUTPUT_MESSAGES_KEY, safe_json_dumps(messages))