Skip to content

fix(openai): Attach response model with streamed Responses API #5554

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
alexander-alderman-webb wants to merge 9 commits into
base: webb/openai/streaming-response
Choose a base branch
from
+9 −0
Open

fix(openai): Attach response model with streamed Responses API #5554

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
ece696a
fix(openai): Attach response model with streamed Responses API
alexander-alderman-webb Feb 26, 2026
9683ef6
Merge branch 'webb/openai/streaming-response' into webb/openai/add-re…
alexander-alderman-webb Feb 26, 2026
0ce3879
Merge branch 'webb/openai/streaming-response' into webb/openai/add-re…
alexander-alderman-webb Feb 26, 2026
731754f
add assertion
alexander-alderman-webb Feb 26, 2026
f93c208
Merge branch 'webb/openai/streaming-response' into webb/openai/add-re…
alexander-alderman-webb Feb 26, 2026
63a9c71
.
alexander-alderman-webb Feb 26, 2026
1cdc590
Merge branch 'webb/openai/streaming-response' into webb/openai/add-re…
alexander-alderman-webb Feb 26, 2026
0bd6503
Merge branch 'webb/openai/streaming-response' into webb/openai/add-re…
alexander-alderman-webb Feb 26, 2026
071d1f0
Merge branch 'webb/openai/streaming-response' into webb/openai/add-re…
alexander-alderman-webb Feb 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions sentry_sdk/integrations/openai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -754,6 +754,8 @@ def new_iterator() -> "Iterator[ChatCompletionChunk]":
data_buf[0].append(x.delta or "")

if isinstance(x, ResponseCompletedEvent):
span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, x.response.model)

_calculate_token_usage(
input,
x.response,
Expand Down Expand Up @@ -801,6 +803,8 @@ async def new_iterator_async() -> "AsyncIterator[ChatCompletionChunk]":
data_buf[0].append(x.delta or "")

if isinstance(x, ResponseCompletedEvent):
span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, x.response.model)

_calculate_token_usage(
input,
x.response,
Expand Down
5 changes: 5 additions & 0 deletions tests/integrations/openai/test_openai.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2445,6 +2445,7 @@ async def test_ai_client_span_streaming_responses_async_api(

expected_data = {
"gen_ai.operation.name": "responses",
"gen_ai.response.model": "response-model-id",
"gen_ai.response.streaming": True,
"gen_ai.system": "openai",
"gen_ai.response.time_to_first_token": mock.ANY,
Expand Down Expand Up @@ -2769,6 +2770,8 @@ def test_streaming_responses_api(
(span,) = transaction["spans"]
assert span["op"] == "gen_ai.responses"

assert span["data"][SPANDATA.GEN_AI_RESPONSE_MODEL] == "response-model-id"

if send_default_pii and include_prompts:
assert span["data"][SPANDATA.GEN_AI_REQUEST_MESSAGES] == '["hello"]'
assert span["data"][SPANDATA.GEN_AI_RESPONSE_TEXT] == "hello world"
Expand Down Expand Up @@ -2824,6 +2827,8 @@ async def test_streaming_responses_api_async(
(span,) = transaction["spans"]
assert span["op"] == "gen_ai.responses"

assert span["data"][SPANDATA.GEN_AI_RESPONSE_MODEL] == "response-model-id"

if send_default_pii and include_prompts:
assert span["data"][SPANDATA.GEN_AI_REQUEST_MESSAGES] == '["hello"]'
assert span["data"][SPANDATA.GEN_AI_RESPONSE_TEXT] == "hello world"
Expand Down
Loading