From c8e7dc1f61d3c3176850e920acf93b5c39fc55b6 Mon Sep 17 00:00:00 2001 From: Yulin Li Date: Sun, 25 May 2025 12:26:17 +0800 Subject: [PATCH] [ai-inference] fix tracing for empty choice --- sdk/ai/azure-ai-inference/azure/ai/inference/tracing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/ai/azure-ai-inference/azure/ai/inference/tracing.py b/sdk/ai/azure-ai-inference/azure/ai/inference/tracing.py index e9dbaf329829..ed53d038aa8f 100644 --- a/sdk/ai/azure-ai-inference/azure/ai/inference/tracing.py +++ b/sdk/ai/azure-ai-inference/azure/ai/inference/tracing.py @@ -386,7 +386,7 @@ def _accumulate_response(self, item, accumulate: Dict[str, Any]) -> None: accumulate["message"]["tool_calls"][-1]["function"]["arguments"] += tool_call.function.arguments def _accumulate_async_streaming_response(self, item, accumulate: Dict[str, Any]) -> None: - if not "choices" in item: + if not "choices" in item or not item["choices"]: return if "finish_reason" in item["choices"][0] and item["choices"][0]["finish_reason"]: accumulate["finish_reason"] = item["choices"][0]["finish_reason"]