Better output

Author: antonpirker

sentry_sdk/consts.py

@@ -372,6 +372,12 @@ class SPANDATA:
     Example: "chat"
     """
 
+    GEN_AI_RESPONSE_TEXT = "gen_ai.response.text"
+    """
+    The model's response text messages.
+    Example: ["The weather in Paris is rainy and overcast, with temperatures around 57°F", "The weather in London is sunny and warm, with temperatures around 65°F"]
+    """
+
     GEN_AI_RESPONSE_TOOL_CALLS = "gen_ai.response.tool_calls"
     """
     The tool calls in the model's response.

sentry_sdk/integrations/openai_agents/utils.py

@@ -202,14 +202,14 @@ def _set_output_data(span, result):
             output_messages["tool"].append(output.dict())
         elif output.type == "message":
             for output_message in output.content:
-                output_messages["response"].append(output_message.to_json())
+                try:
+                    output_messages["response"].append(output_message.text)
+                except AttributeError:
+                    # Unknown output message type, just return the json
+                    output_messages["response"].append(output_message.to_json())
 
     if len(output_messages["tool"]) > 0:
         span.set_data(SPANDATA.GEN_AI_RESPONSE_TOOL_CALLS, output_messages["tool"])
 
     if len(output_messages["response"]) > 0:
-        span.set_data(SPANDATA.GEN_AI_CHOICE, output_messages["response"])
-
-        # Deprecated name just for first iteration.
-        # TODO-anton: define how to set tool response messages and document in sentry-conventions.
-        span.set_data("gen_ai.response.text", output_messages["response"])
+        span.set_data(SPANDATA.GEN_AI_RESPONSE_TEXT, output_messages["response"])