aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.py')
-rw-r--r--.venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.py32
1 files changed, 32 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.py b/.venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.py
new file mode 100644
index 00000000..ed3494f6
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.py
@@ -0,0 +1,32 @@
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+ from typing import Any
+
+from sentry_sdk.tracing import Span
+from sentry_sdk.utils import logger
+
+
+def _normalize_data(data):
+ # type: (Any) -> Any
+
+ # convert pydantic data (e.g. OpenAI v1+) to json compatible format
+ if hasattr(data, "model_dump"):
+ try:
+ return data.model_dump()
+ except Exception as e:
+ logger.warning("Could not convert pydantic data to JSON: %s", e)
+ return data
+ if isinstance(data, list):
+ if len(data) == 1:
+ return _normalize_data(data[0]) # remove empty dimensions
+ return list(_normalize_data(x) for x in data)
+ if isinstance(data, dict):
+ return {k: _normalize_data(v) for (k, v) in data.items()}
+ return data
+
+
+def set_data_normalized(span, key, value):
+ # type: (Span, str, Any) -> None
+ normalized = _normalize_data(value)
+ span.set_data(key, normalized)