@@ -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