Fix telemetry issues. (#42698)

* Fix telemetry issues.

* Fix

* Add support for OpenAPI step

* Fix OpenAPI tool logging

* Fix comment

* Fix

Author: nick863

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

@@ -10,6 +10,9 @@
 
 ### Bugs Fixed
 
+* Fix the issue with logging Agent message, when the message has "in progress" status (related to [issue](https://github.com/Azure/azure-sdk-for-python/issues/42645)).
+* Fix the issue with `RunStepOpenAPIToolCall` logging [issue](https://github.com/Azure/azure-sdk-for-python/issues/42645).
+
 ### Sample updates
 
 ## 1.2.0b3 (2025-08-22)

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

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ai/azure-ai-agents",
-  "Tag": "python/ai/azure-ai-agents_cf22042479"
+  "Tag": "python/ai/azure-ai-agents_ac2e010ef5"
 }

sdk/ai/azure-ai-agents/azure/ai/agents/telemetry/_ai_agents_instrumentor.py

@@ -18,7 +18,9 @@
 from azure.ai.agents.models._enums import (
     AgentsResponseFormatMode,
     MessageRole,
+    MessageStatus,
     RunStepStatus,
+    RunStepType,
 )
 from azure.ai.agents.models import (
     MessageAttachment,
@@ -28,6 +30,7 @@
     RunStepDeltaChunk,
     RunStepError,
     RunStepFunctionToolCall,
+    RunStepOpenAPIToolCall,
     RunStepToolCallDetails,
     RunStepCodeInterpreterToolCall,
     RunStepBingGroundingToolCall,
@@ -439,6 +442,12 @@ def _process_tool_calls(self, step: RunStep) -> List[Dict[str, Any]]:
                     "type": t.type,
                     t.type: t.bing_grounding,
                 }
+            elif isinstance(t, RunStepOpenAPIToolCall):
+                tool_call = {
+                    "id": t.id,
+                    "type": t.type,
+                    'function': t.as_dict().get('function', {})
+                }
             else:
                 tool_details = t.as_dict()[t.type]
 
@@ -2057,7 +2066,8 @@ def on_thread_message(self, message: "ThreadMessage") -> None:  # type: ignore[f
         else:
             retval = super().on_thread_message(message)  # pylint: disable=assignment-from-none # type: ignore
 
-        if message.status in {"completed", "incomplete"}:
+        # Message status may be in progress, even if the thread.message.completed event has arrived.
+        if message.status in {MessageStatus.COMPLETED, MessageStatus.INCOMPLETE} or (message.status == MessageStatus.IN_PROGRESS and message.content):
             self.last_message = message
 
         return retval  # type: ignore
@@ -2081,14 +2091,14 @@ def on_run_step(self, step: "RunStep") -> None:  # type: ignore[func-returns-val
             retval = super().on_run_step(step)  # pylint: disable=assignment-from-none # type: ignore
 
         if (
-            step.type == "tool_calls"
+            step.type == RunStepType.TOOL_CALLS
             and isinstance(step.step_details, RunStepToolCallDetails)
             and step.status == RunStepStatus.COMPLETED
         ):
             self.instrumentor._add_tool_assistant_message_event(  # pylint: disable=protected-access # pyright: ignore [reportFunctionMemberAccess]
                 self.span, step
             )
-        elif step.type == "message_creation" and step.status == RunStepStatus.COMPLETED:
+        elif step.type == RunStepType.MESSAGE_CREATION and step.status == RunStepStatus.COMPLETED:
             self.instrumentor.add_thread_message_event(self.span, cast(ThreadMessage, self.last_message), step.usage)
             if (
                 self.span
@@ -2194,7 +2204,8 @@ async def on_thread_message(self, message: "ThreadMessage") -> None:  # type: ig
         else:
             retval = await super().on_thread_message(message)  # type: ignore
 
-        if message.status in {"completed", "incomplete"}:
+        # Message status may be in progress, even if the thread.message.completed event has arrived.
+        if message.status in {MessageStatus.COMPLETED, MessageStatus.INCOMPLETE} or (message.status == MessageStatus.IN_PROGRESS and message.content):
             self.last_message = message
 
         return retval  # type: ignore
@@ -2218,14 +2229,14 @@ async def on_run_step(self, step: "RunStep") -> None:  # type: ignore[func-retur
             retval = await super().on_run_step(step)  # type: ignore
 
         if (
-            step.type == "tool_calls"
+            step.type == RunStepType.TOOL_CALLS
             and isinstance(step.step_details, RunStepToolCallDetails)
             and step.status == RunStepStatus.COMPLETED
         ):
             self.instrumentor._add_tool_assistant_message_event(  # pylint: disable=protected-access # pyright: ignore [reportFunctionMemberAccess]
                 self.span, step
             )
-        elif step.type == "message_creation" and step.status == RunStepStatus.COMPLETED:
+        elif step.type == RunStepType.MESSAGE_CREATION and step.status == RunStepStatus.COMPLETED:
             self.instrumentor.add_thread_message_event(self.span, cast(ThreadMessage, self.last_message), step.usage)
             if (
                 self.span