Skip to content

Commit 80305c1

Browse files
authored
[CI] Fix flaky test_streaming_response test (#20913)
Signed-off-by: NickLucche <nlucches@redhat.com>
1 parent 37e2eca commit 80305c1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/entrypoints/openai/test_transcription_validation.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,8 @@ async def post_with_stream(*args, **kwargs):
154154
file=winning_call,
155155
language="en",
156156
temperature=0.0,
157-
extra_body=dict(stream=True))
157+
extra_body=dict(stream=True),
158+
timeout=30)
158159
# Reconstruct from chunks and validate
159160
async for chunk in res:
160161
# just a chunk
@@ -184,7 +185,8 @@ async def post_with_stream(*args, **kwargs):
184185
temperature=0.0,
185186
extra_body=dict(stream=True,
186187
stream_include_usage=True,
187-
stream_continuous_usage_stats=True))
188+
stream_continuous_usage_stats=True),
189+
timeout=30)
188190
final = False
189191
continuous = True
190192
async for chunk in res:

0 commit comments

Comments
 (0)