.

Author: alexander-alderman-webb

sentry_sdk/integrations/openai.py

@@ -706,9 +706,15 @@ def _new_chat_completion_common(f: "Any", *args: "Any", **kwargs: "Any") -> "Any
     start_time = time.perf_counter()
     response = yield f, args, kwargs
 
-    _set_completions_api_output_data(
-        span, response, kwargs, integration, start_time, finish_span=True
-    )
+    is_streaming_response = kwargs.get("stream", False)
+    if is_streaming_response:
+        _set_streaming_completions_api_output_data(
+            span, response, kwargs, integration, start_time, finish_span=True
+        )
+    else:
+        _set_completions_api_output_data(
+            span, response, kwargs, integration, start_time, finish_span=True
+        )
 
     return response
 
@@ -731,6 +737,24 @@ def _set_completions_api_output_data(
     )
 
 
+def _set_streaming_completions_api_output_data(
+    span: "Span",
+    response: "Any",
+    kwargs: "dict[str, Any]",
+    integration: "OpenAIIntegration",
+    start_time: "Optional[float]" = None,
+    finish_span: bool = True,
+) -> None:
+    _common_set_output_data(
+        span,
+        response,
+        kwargs,
+        integration,
+        start_time,
+        finish_span,
+    )
+
+
 def _set_responses_api_output_data(
     span: "Span",
     response: "Any",
@@ -749,6 +773,24 @@ def _set_responses_api_output_data(
     )
 
 
+def _set_streaming_responses_api_output_data(
+    span: "Span",
+    response: "Any",
+    kwargs: "dict[str, Any]",
+    integration: "OpenAIIntegration",
+    start_time: "Optional[float]" = None,
+    finish_span: bool = True,
+) -> None:
+    _common_set_output_data(
+        span,
+        response,
+        kwargs,
+        integration,
+        start_time,
+        finish_span,
+    )
+
+
 def _set_embeddings_output_data(
     span: "Span",
     response: "Any",
@@ -943,9 +985,15 @@ def _new_responses_create_common(f: "Any", *args: "Any", **kwargs: "Any") -> "An
     start_time = time.perf_counter()
     response = yield f, args, kwargs
 
-    _set_responses_api_output_data(
-        span, response, kwargs, integration, start_time, finish_span=True
-    )
+    is_streaming_response = kwargs.get("stream", False)
+    if is_streaming_response:
+        _set_responses_api_output_data(
+            span, response, kwargs, integration, start_time, finish_span=True
+        )
+    else:
+        _set_streaming_responses_api_output_data(
+            span, response, kwargs, integration, start_time, finish_span=True
+        )
 
     return response