extract tokens from apm span metrics instead of meta

DataDog · Feb 4, 2025 · 5b770cf · 5b770cf
1 parent 2bf6bad
commit 5b770cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ddtrace/llmobs/_integrations/bedrock.py b/ddtrace/llmobs/_integrations/bedrock.py
@@ -66,8 +66,8 @@ def _llmobs_set_tags(
     def _llmobs_metrics(span: Span, response: Optional[Dict[str, Any]]) -> Dict[str, Any]:
         metrics = {}
         if response and response.get("text"):
-            prompt_tokens = int(span.get_tag("bedrock.usage.prompt_tokens") or 0)
-            completion_tokens = int(span.get_tag("bedrock.usage.completion_tokens") or 0)
+            prompt_tokens = int(span.get_metric("bedrock.usage.prompt_tokens") or 0)
+            completion_tokens = int(span.get_metric("bedrock.usage.completion_tokens") or 0)
             metrics[INPUT_TOKENS_METRIC_KEY] = prompt_tokens
             metrics[OUTPUT_TOKENS_METRIC_KEY] = completion_tokens
             metrics[TOTAL_TOKENS_METRIC_KEY] = prompt_tokens + completion_tokens