feat(ai) adding unpack parameter to set_data_normalized to control whether to automatically unpack single elements in lists

Author: constantinius

sentry_sdk/ai/utils.py

@@ -7,8 +7,8 @@
 from sentry_sdk.utils import logger
 
 
-def _normalize_data(data):
-    # type: (Any) -> Any
+def _normalize_data(data, unpack=True):
+    # type: (Any, bool) -> Any
 
     # convert pydantic data (e.g. OpenAI v1+) to json compatible format
     if hasattr(data, "model_dump"):
@@ -18,7 +18,7 @@ def _normalize_data(data):
             logger.warning("Could not convert pydantic data to JSON: %s", e)
             return data
     if isinstance(data, list):
-        if len(data) == 1:
+        if unpack and len(data) == 1:
             return _normalize_data(data[0])  # remove empty dimensions
         return list(_normalize_data(x) for x in data)
     if isinstance(data, dict):
@@ -27,9 +27,9 @@ def _normalize_data(data):
     return data
 
 
-def set_data_normalized(span, key, value):
-    # type: (Span, str, Any) -> None
-    normalized = _normalize_data(value)
+def set_data_normalized(span, key, value, unpack=True):
+    # type: (Span, str, Any, bool) -> None
+    normalized = _normalize_data(value, unpack=unpack)
     if isinstance(normalized, (int, float, bool, str)):
         span.set_data(key, normalized)
     else: