update integration

Author: alexander-alderman-webb

sentry_sdk/integrations/openai_agents/spans/invoke_agent.py

@@ -3,7 +3,6 @@
     get_start_span_function,
     set_data_normalized,
     normalize_message_roles,
-    truncate_and_annotate_messages,
 )
 from sentry_sdk.consts import OP, SPANDATA
 from sentry_sdk.scope import should_send_default_pii
@@ -63,17 +62,12 @@ def invoke_agent_span(
 
         if len(messages) > 0:
             normalized_messages = normalize_message_roles(messages)
-            scope = sentry_sdk.get_current_scope()
-            messages_data = truncate_and_annotate_messages(
-                normalized_messages, span, scope
+            set_data_normalized(
+                span,
+                SPANDATA.GEN_AI_REQUEST_MESSAGES,
+                normalized_messages,
+                unpack=False,
             )
-            if messages_data is not None:
-                set_data_normalized(
-                    span,
-                    SPANDATA.GEN_AI_REQUEST_MESSAGES,
-                    messages_data,
-                    unpack=False,
-                )
 
     _set_agent_data(span, agent)
 

sentry_sdk/integrations/openai_agents/utils.py

@@ -6,7 +6,6 @@
     normalize_message_roles,
     set_data_normalized,
     normalize_message_role,
-    truncate_and_annotate_messages,
 )
 from sentry_sdk.consts import SPANDATA, SPANSTATUS, OP
 from sentry_sdk.integrations import DidNotEnable
@@ -172,16 +171,12 @@ def _set_input_data(
                     }
                 )
 
-    normalized_messages = normalize_message_roles(request_messages)
-    scope = sentry_sdk.get_current_scope()
-    messages_data = truncate_and_annotate_messages(normalized_messages, span, scope)
-    if messages_data is not None:
-        set_data_normalized(
-            span,
-            SPANDATA.GEN_AI_REQUEST_MESSAGES,
-            messages_data,
-            unpack=False,
-        )
+    set_data_normalized(
+        span,
+        SPANDATA.GEN_AI_REQUEST_MESSAGES,
+        normalize_message_roles(request_messages),
+        unpack=False,
+    )
 
 
 def _set_output_data(span: "sentry_sdk.tracing.Span", result: "Any") -> None: