.

Author: alexander-alderman-webb

sentry_sdk/integrations/openai.py

@@ -473,9 +473,6 @@ def _common_set_output_data(
     if hasattr(response, "model"):
         set_data_normalized(span, SPANDATA.GEN_AI_RESPONSE_MODEL, response.model)
 
-    if input is not None and isinstance(input, str):
-        input = [input]
-
     ttft: "Optional[float]" = None
 
     if hasattr(response, "choices"):
@@ -723,6 +720,9 @@ def _set_completions_api_output_data(
 ) -> None:
     messages = kwargs.get("messages")
 
+    if messages is not None and isinstance(messages, str):
+        messages = [messages]
+
     _common_set_output_data(
         span,
         response,
@@ -743,6 +743,9 @@ def _set_streaming_completions_api_output_data(
 ) -> None:
     messages = kwargs.get("messages")
 
+    if messages is not None and isinstance(messages, str):
+        messages = [messages]
+
     _common_set_output_data(
         span,
         response,
@@ -763,6 +766,9 @@ def _set_responses_api_output_data(
 ) -> None:
     input = kwargs.get("input")
 
+    if input is not None and isinstance(input, str):
+        input = [input]
+
     _common_set_output_data(
         span,
         response,
@@ -783,6 +789,9 @@ def _set_streaming_responses_api_output_data(
 ) -> None:
     input = kwargs.get("input")
 
+    if input is not None and isinstance(input, str):
+        input = [input]
+
     _common_set_output_data(
         span,
         response,
@@ -803,6 +812,9 @@ def _set_embeddings_output_data(
 ) -> None:
     input = kwargs.get("input")
 
+    if input is not None and isinstance(input, str):
+        input = [input]
+
     _common_set_output_data(
         span,
         response,