fix type issues

Author: shellmayr

sentry_sdk/ai/utils.py

@@ -14,9 +14,6 @@
 if TYPE_CHECKING:
     from typing import Any, Dict, List, Optional
 
-from sentry_sdk._types import AnnotatedValue
-from sentry_sdk.serializer import serialize
-
 MAX_GEN_AI_MESSAGE_BYTES = 20_000  # 20KB
 
 

sentry_sdk/client.py

@@ -607,30 +607,27 @@ def _prepare_event(
                 event.get("breadcrumbs", []), {"len": previous_total_breadcrumbs}
             )
 
-        # Annotate truncated gen_ai messages in spans
         if scope is not None and scope._gen_ai_messages_truncated:
-            spans = event.get("spans", [])
-            if isinstance(spans, AnnotatedValue):
-                spans = spans.value
-
+            spans = []  # type: List[Dict[str, Any]]
             for span in spans:
                 if isinstance(span, dict):
                     span_id = span.get("span_id")
                     if span_id and span_id in scope._gen_ai_messages_truncated:
                         span_data = span.get("data", {})
-                        original_count = span_data.pop(
-                            "_gen_ai_messages_original_count", None
-                        )
-                        if (
-                            original_count is not None
-                            and SPANDATA.GEN_AI_REQUEST_MESSAGES in span_data
-                        ):
-                            span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES] = (
-                                AnnotatedValue(
-                                    span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES],
-                                    {"len": original_count},
-                                )
+                        if isinstance(span_data, dict):
+                            original_count = span_data.pop(
+                                "_gen_ai_messages_original_count", None
                             )
+                            if (
+                                original_count is not None
+                                and SPANDATA.GEN_AI_REQUEST_MESSAGES in span_data
+                            ):
+                                span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES] = (
+                                    AnnotatedValue(
+                                        span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES],
+                                        {"len": original_count},
+                                    )
+                                )
 
         # Postprocess the event here so that annotated types do
         # generally not surface in before_send