simpler function tool trace (#45083)

* simpler function tool call trace format

* test fix

* updating test recordings

* fix based on review comment]

* updating test recordings

* updated tests

Author: M-Hietala

sdk/ai/azure-ai-projects/CHANGELOG.md

@@ -28,6 +28,7 @@
 * Tracing: response generation uses gen_ai.input.messages and gen_ai.output.messages attributes directly under the span instead of events.
 * Tracing: agent creation uses gen_ai.system.instructions attribute directly under the span instead of an event.
 * Tracing: "gen_ai.provider.name" attribute value changed to "microsoft.foundry".
+* Tracing: the format of the function tool call related traces in input and output messages changed to {"type": "tool_call", "id": "...", "name": "...", "arguments": {...}} and {"type": "tool_call_response", "id": "...", "result": "..."}
 
 ## 2.0.0b3 (2026-01-06)
 

sdk/ai/azure-ai-projects/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ai/azure-ai-projects",
-  "Tag": "python/ai/azure-ai-projects_8d150306ae"
+  "Tag": "python/ai/azure-ai-projects_6896cca62c"
 }

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

@@ -58,6 +58,7 @@
     SPAN_NAME_CHAT,
     SPAN_NAME_INVOKE_AGENT,
     _get_use_message_events,
+    _get_use_simple_tool_format,
     start_span,
     RESPONSES_PROVIDER,
 )
@@ -171,6 +172,18 @@ def is_binary_data_enabled(self) -> bool:
         """
         return self._impl.is_binary_data_enabled()
 
+    def is_simple_tool_format_enabled(self) -> bool:
+        """This function gets the simple tool format value.
+
+        When enabled, function tool calls use a simplified OTEL-compliant format:
+          tool_call: {"type": "tool_call", "id": "...", "name": "...", "arguments": {...}}
+          tool_call_response: {"type": "tool_call_response", "id": "...", "result": "..."}
+
+        :return: A bool value indicating whether simple tool format is enabled.
+        :rtype: bool
+        """
+        return _get_use_simple_tool_format()
+
 
 class _ResponsesInstrumentorPreview:  # pylint: disable=too-many-instance-attributes,too-many-statements,too-many-public-methods
     """
@@ -703,9 +716,21 @@ def _add_tool_message_events(  # pylint: disable=too-many-branches
                                             output_value = {k: v for k, v in output_value.items() if k != "image_url"}
                                 tool_output["output"] = output_value
 
-                    # Add to parts array (type "tool_call_output" wraps the tool output)
-                    # Always include type and id, even when content recording is disabled
-                    parts.append({"type": "tool_call_output", "content": tool_output})
+                    # Add to parts array
+                    # Check if simple tool format is enabled for function_call_output types
+                    if _get_use_simple_tool_format() and item_type == "function_call_output":
+                        # Build simplified OTEL-compliant format:
+                        # {"type": "tool_call_response", "id": "...", "result": "..."}
+                        simple_response: Dict[str, Any] = {"type": "tool_call_response"}
+                        if "id" in tool_output:
+                            simple_response["id"] = tool_output["id"]
+                        if _trace_responses_content and "output" in tool_output:
+                            simple_response["result"] = tool_output["output"]
+                        parts.append(simple_response)
+                    else:
+                        # Original nested format (type "tool_call_output" wraps the tool output)
+                        # Always include type and id, even when content recording is disabled
+                        parts.append({"type": "tool_call_output", "content": tool_output})
                 except Exception:  # pylint: disable=broad-exception-caught
                     # Skip items that can't be processed
                     logger.debug(
@@ -1044,8 +1069,25 @@ def _emit_tool_call_event(
         conversation_id: Optional[str] = None,
     ) -> None:
         """Helper to emit a single tool call event or attribute."""
-        # Wrap tool call in parts array
-        parts = [{"type": "tool_call", "content": tool_call}]
+        # Check if simple tool format is enabled for function_call types
+        if _get_use_simple_tool_format() and tool_call.get("type") == "function_call":
+            # Build simplified OTEL-compliant format:
+            # {"type": "tool_call", "id": "...", "name": "...", "arguments": {...}}
+            simple_tool_call: Dict[str, Any] = {"type": "tool_call"}
+            if "id" in tool_call:
+                simple_tool_call["id"] = tool_call["id"]
+            # Extract name and arguments from nested function object if content recording enabled
+            if _trace_responses_content and "function" in tool_call:
+                func = tool_call["function"]
+                if "name" in func:
+                    simple_tool_call["name"] = func["name"]
+                if "arguments" in func:
+                    simple_tool_call["arguments"] = func["arguments"]
+            parts = [simple_tool_call]
+        else:
+            # Original nested format
+            parts = [{"type": "tool_call", "content": tool_call}]
+
         content_array = [{"role": "assistant", "parts": parts}]
 
         if _get_use_message_events():
@@ -1069,9 +1111,21 @@ def _emit_tool_output_event(
         conversation_id: Optional[str] = None,
     ) -> None:
         """Helper to emit a single tool output event or attribute."""
-        # Wrap tool output in parts array
-        # Tool outputs are inputs TO the model (from tool execution), so use role "tool"
-        parts = [{"type": "tool_call_output", "content": tool_output}]
+        # Check if simple tool format is enabled for function_call_output types
+        if _get_use_simple_tool_format() and tool_output.get("type") == "function_call_output":
+            # Build simplified OTEL-compliant format:
+            # {"type": "tool_call_response", "id": "...", "result": "..."}
+            simple_tool_response: Dict[str, Any] = {"type": "tool_call_response"}
+            if "id" in tool_output:
+                simple_tool_response["id"] = tool_output["id"]
+            # Add result only if content recording is enabled
+            if _trace_responses_content and "output" in tool_output:
+                simple_tool_response["result"] = tool_output["output"]
+            parts = [simple_tool_response]
+        else:
+            # Original nested format
+            parts = [{"type": "tool_call_output", "content": tool_output}]
+
         content_array = [{"role": "tool", "parts": parts}]
 
         if _get_use_message_events():

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

@@ -121,6 +121,33 @@
 # Set to True for event-based, False for attribute-based (default)
 _use_message_events = False
 
+# Configuration: Controls whether function tool calls use simplified OTEL-compliant format
+# When True (default), function_call and function_call_output use a simpler structure:
+#   tool_call: {"type": "tool_call", "id": "...", "name": "...", "arguments": {...}}
+#   tool_call_response: {"type": "tool_call_response", "id": "...", "result": "..."}
+# When False, the full nested structure is used
+_use_simple_tool_format = True
+
+
+def _get_use_simple_tool_format() -> bool:
+    """Get the current tool format mode (simple vs nested). Internal use only.
+
+    :return: True if using simple OTEL-compliant format, False if using nested format
+    :rtype: bool
+    """
+    return _use_simple_tool_format
+
+
+def _set_use_simple_tool_format(use_simple: bool) -> None:
+    """
+    Set the tool format mode at runtime. Internal use only.
+
+    :param use_simple: True to use simple OTEL-compliant format, False for nested format
+    :type use_simple: bool
+    """
+    global _use_simple_tool_format  # pylint: disable=global-statement
+    _use_simple_tool_format = use_simple
+
 
 def _get_use_message_events() -> bool:
     """Get the current message tracing mode (events vs attributes). Internal use only.