From 61e31ce6570c73d5b947aaa9acff6b7ebc2db002 Mon Sep 17 00:00:00 2001
From: venti <1308199824@qq.com>
Date: Sat, 30 May 2026 15:10:26 +0800
Subject: [PATCH] fix: include all usage detail fields in OTEL span attributes
 (fixes #5511)

---
 .../core/agent_framework/observability.py         | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/python/packages/core/agent_framework/observability.py b/python/packages/core/agent_framework/observability.py
index d7734f2457..898748802b 100644
--- a/python/packages/core/agent_framework/observability.py
+++ b/python/packages/core/agent_framework/observability.py
@@ -2294,12 +2294,15 @@ def _get_response_attributes(
     if model := getattr(response, "model", None):
         attributes[OtelAttr.RESPONSE_MODEL] = model
     if capture_usage and (usage := response.usage_details):
-        input_tokens = usage.get("input_token_count")
-        if input_tokens:
-            attributes[OtelAttr.INPUT_TOKENS] = input_tokens
-        output_tokens = usage.get("output_token_count")
-        if output_tokens:
-            attributes[OtelAttr.OUTPUT_TOKENS] = output_tokens
+        for key, value in usage.items():
+            if not isinstance(value, int):
+                continue
+            if key == "input_token_count":
+                attributes[OtelAttr.INPUT_TOKENS] = value
+            elif key == "output_token_count":
+                attributes[OtelAttr.OUTPUT_TOKENS] = value
+            else:
+                attributes[f"gen_ai.usage.{key}"] = value
     return attributes