Skip to content

Commit fa7db04

Browse files
Merge branch 'webb/pydantic-ai/streaming-attribute' into webb/pydantic-ai/remove-contextvar-stack
2 parents c650c5b + 57268a9 commit fa7db04

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sentry_sdk/integrations/pydantic_ai/patches/graph_nodes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ async def wrapped_model_request_run(self: "Any", ctx: "Any") -> "Any":
6161
messages, model, model_settings = _extract_span_data(self, ctx)
6262

6363
with ai_client_span(messages, None, model, model_settings) as span:
64-
span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
64+
span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, False)
6565

6666
result = await original_model_request_run(self, ctx)
6767

@@ -90,7 +90,7 @@ async def wrapped_model_request_stream(self: "Any", ctx: "Any") -> "Any":
9090

9191
# Create chat span for streaming request
9292
with ai_client_span(messages, None, model, model_settings) as span:
93-
span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, False)
93+
span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
9494

9595
# Call the original stream method
9696
async with original_stream_method(self, ctx) as stream:

0 commit comments

Comments
 (0)