Skip to content

Commit 29c7489

Browse files
authored
[CI] Fix flaky entrypoint tests (vllm-project#11403)
Signed-off-by: Roger Wang <ywang@roblox.com>
1 parent c2d1b07 commit 29c7489

File tree

3 files changed

+9
-0
lines changed

3 files changed

+9
-0
lines changed

tests/entrypoints/openai/test_audio.py

+3
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ async def test_single_chat_session_audio(client: openai.AsyncOpenAI,
7474
messages=messages,
7575
max_completion_tokens=10,
7676
logprobs=True,
77+
temperature=0.0,
7778
top_logprobs=5)
7879
assert len(chat_completion.choices) == 1
7980

@@ -130,6 +131,7 @@ async def test_single_chat_session_audio_base64encoded(
130131
messages=messages,
131132
max_completion_tokens=10,
132133
logprobs=True,
134+
temperature=0.0,
133135
top_logprobs=5)
134136
assert len(chat_completion.choices) == 1
135137

@@ -150,6 +152,7 @@ async def test_single_chat_session_audio_base64encoded(
150152
model=model_name,
151153
messages=messages,
152154
max_completion_tokens=10,
155+
temperature=0.0,
153156
)
154157
message = chat_completion.choices[0].message
155158
assert message.content is not None and len(message.content) >= 0

tests/entrypoints/openai/test_video.py

+3
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ async def test_single_chat_session_video(client: openai.AsyncOpenAI,
8282
messages=messages,
8383
max_completion_tokens=10,
8484
logprobs=True,
85+
temperature=0.0,
8586
top_logprobs=5)
8687
assert len(chat_completion.choices) == 1
8788

@@ -174,6 +175,7 @@ async def test_single_chat_session_video_base64encoded(
174175
messages=messages,
175176
max_completion_tokens=10,
176177
logprobs=True,
178+
temperature=0.0,
177179
top_logprobs=5)
178180
assert len(chat_completion.choices) == 1
179181

@@ -194,6 +196,7 @@ async def test_single_chat_session_video_base64encoded(
194196
model=model_name,
195197
messages=messages,
196198
max_completion_tokens=10,
199+
temperature=0.0,
197200
)
198201
message = chat_completion.choices[0].message
199202
assert message.content is not None and len(message.content) >= 0

tests/entrypoints/openai/test_vision.py

+3
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ async def test_single_chat_session_image(client: openai.AsyncOpenAI,
8383
messages=messages,
8484
max_completion_tokens=10,
8585
logprobs=True,
86+
temperature=0.0,
8687
top_logprobs=5)
8788
assert len(chat_completion.choices) == 1
8889

@@ -175,6 +176,7 @@ async def test_single_chat_session_image_base64encoded(
175176
messages=messages,
176177
max_completion_tokens=10,
177178
logprobs=True,
179+
temperature=0.0,
178180
top_logprobs=5)
179181
assert len(chat_completion.choices) == 1
180182

@@ -195,6 +197,7 @@ async def test_single_chat_session_image_base64encoded(
195197
model=model_name,
196198
messages=messages,
197199
max_completion_tokens=10,
200+
temperature=0.0,
198201
)
199202
message = chat_completion.choices[0].message
200203
assert message.content is not None and len(message.content) >= 0

0 commit comments

Comments
 (0)