error handling

Author: antonpirker

sentry_sdk/tracing_utils.py

@@ -1092,43 +1092,44 @@ def _get_output_attributes(template, send_pii, result):
     attributes = {}  # type: dict[str, Any]
 
     if template in [SPANTEMPLATE.AI_AGENT, SPANTEMPLATE.AI_TOOL, SPANTEMPLATE.AI_CHAT]:
-        # Usage from result, result.usage, and result.metadata.usage
-        usage_candidates = [result]
-        if isinstance(result, dict):
-            usage_candidates.append(result.get("usage"))
-            meta = result.get("metadata")
-        else:
-            usage_candidates.append(getattr(result, "usage", None))
-            meta = getattr(result, "metadata", None)
-
-        if isinstance(meta, dict):
-            usage_candidates.append(meta.get("usage"))
-        elif meta is not None:
-            usage_candidates.append(getattr(meta, "usage", None))
-
-        for usage_candidate in usage_candidates:
-            if usage_candidate is not None:
-                attributes.update(_get_usage_attributes(usage_candidate))
-
-        # Response model
-        model_name = None
-        if isinstance(result, dict):
-            val = result.get("model")
-            if isinstance(val, str):
-                model_name = val
-            val = result.get("model_name")
-            if isinstance(val, str):
-                model_name = val
-        else:
-            val = getattr(result, "model", None)
-            if isinstance(val, str):
-                model_name = val
-            val = getattr(result, "model_name", None)
-            if isinstance(val, str):
-                model_name = val
-
-        if model_name is not None:
-            attributes[SPANDATA.GEN_AI_RESPONSE_MODEL] = model_name
+        with capture_internal_exceptions():
+            # Usage from result, result.usage, and result.metadata.usage
+            usage_candidates = [result]
+            if isinstance(result, dict):
+                usage_candidates.append(result.get("usage"))
+                meta = result.get("metadata")
+            else:
+                usage_candidates.append(getattr(result, "usage", None))
+                meta = getattr(result, "metadata", None)
+
+            if isinstance(meta, dict):
+                usage_candidates.append(meta.get("usage"))
+            elif meta is not None:
+                usage_candidates.append(getattr(meta, "usage", None))
+
+            for usage_candidate in usage_candidates:
+                if usage_candidate is not None:
+                    attributes.update(_get_usage_attributes(usage_candidate))
+
+            # Response model
+            model_name = None
+            if isinstance(result, dict):
+                val = result.get("model")
+                if isinstance(val, str):
+                    model_name = val
+                val = result.get("model_name")
+                if isinstance(val, str):
+                    model_name = val
+            else:
+                val = getattr(result, "model", None)
+                if isinstance(val, str):
+                    model_name = val
+                val = getattr(result, "model_name", None)
+                if isinstance(val, str):
+                    model_name = val
+
+            if model_name is not None:
+                attributes[SPANDATA.GEN_AI_RESPONSE_MODEL] = model_name
 
     # Tool output
     if template == SPANTEMPLATE.AI_TOOL and send_pii: