Skip to content

Commit 8e45aa3

Browse files
committed
wip
1 parent 84bfabb commit 8e45aa3

File tree

1 file changed

+17
-16
lines changed

1 file changed

+17
-16
lines changed

sentry_sdk/client.py

Lines changed: 17 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -599,22 +599,23 @@ def _prepare_event(
599599
event_scrubber.scrub_event(event)
600600

601601
if scope is not None and scope._gen_ai_messages_truncated:
602-
spans = event.get("spans", []) # type: List[Dict[str, Any]] | AnnotatedValue[List[Dict[str, Any]]]
603-
for span in spans:
604-
span_id = span.get("span_id", None)
605-
span_data = span.get("data", {})
606-
if (
607-
span_id
608-
and span_id in scope._gen_ai_messages_truncated
609-
and SPANDATA.GEN_AI_REQUEST_MESSAGES in span_data
610-
):
611-
span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES] = AnnotatedValue(
612-
span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES],
613-
{
614-
"len": scope._gen_ai_messages_truncated[span_id]
615-
+ len(span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES])
616-
},
617-
)
602+
spans = event.get("spans", []) # type: List[Dict[str, Any]] | AnnotatedValue
603+
if isinstance(spans, list):
604+
for span in spans:
605+
span_id = span.get("span_id", None)
606+
span_data = span.get("data", {})
607+
if (
608+
span_id
609+
and span_id in scope._gen_ai_messages_truncated
610+
and SPANDATA.GEN_AI_REQUEST_MESSAGES in span_data
611+
):
612+
span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES] = AnnotatedValue(
613+
span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES],
614+
{
615+
"len": scope._gen_ai_messages_truncated[span_id]
616+
+ len(span_data[SPANDATA.GEN_AI_REQUEST_MESSAGES])
617+
},
618+
)
618619
if previous_total_spans is not None:
619620
event["spans"] = AnnotatedValue(
620621
event.get("spans", []), {"len": previous_total_spans}

0 commit comments

Comments
 (0)