Fix streaming message capture order

cursoragent · claude · cursoragent · commit 6153a56a6a5d · 2026-04-07T10:54:49.000Z
Move input message extraction in the streaming ModelRequestNode wrapper to run after the original stream context exits, matching non-streaming behavior so post-execution instructions are captured.

Co-Authored-By: gpt-5.3-codex-high &lt;noreply@anthropic.com&gt;
diff --git a/sentry_sdk/integrations/pydantic_ai/patches/graph_nodes.py b/sentry_sdk/integrations/pydantic_ai/patches/graph_nodes.py
@@ -91,13 +91,13 @@ async def wrapped_model_request_stream(self: "Any", ctx: "Any") -> "Any":
 
             # Create chat span for streaming request
             with ai_client_span(None, model, model_settings) as span:
-                if messages:
-                    _set_input_messages(span, messages)
-
                 # Call the original stream method
                 async with original_stream_method(self, ctx) as stream:
                     yield stream
 
+                if messages:
+                    _set_input_messages(span, messages)
+
                 # After streaming completes, update span with response data
                 # The ModelRequestNode stores the final response in _result
                 model_response = None
