Skip to content

Commit 9683ef6

Browse files
Merge branch 'webb/openai/streaming-response' into webb/openai/add-response-model
2 parents ece696a + 6c5f254 commit 9683ef6

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/integrations/openai/test_openai.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3019,7 +3019,9 @@ def test_streaming_chat_completion_ttft(sentry_init, capture_events):
30193019

30203020
with start_transaction(name="openai tx"):
30213021
response_stream = client.chat.completions.create(
3022-
model="some-model", messages=[{"role": "user", "content": "Say hello"}]
3022+
model="some-model",
3023+
messages=[{"role": "user", "content": "Say hello"}],
3024+
stream=True,
30233025
)
30243026
# Consume the stream
30253027
for _ in response_stream:
@@ -3083,7 +3085,9 @@ async def test_streaming_chat_completion_ttft_async(sentry_init, capture_events)
30833085

30843086
with start_transaction(name="openai tx"):
30853087
response_stream = await client.chat.completions.create(
3086-
model="some-model", messages=[{"role": "user", "content": "Say hello"}]
3088+
model="some-model",
3089+
messages=[{"role": "user", "content": "Say hello"}],
3090+
stream=True,
30873091
)
30883092
# Consume the stream
30893093
async for _ in response_stream:

0 commit comments

Comments
 (0)