Unverified Commit 29c74893 authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[CI] Fix flaky entrypoint tests (#11403)


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent c2d1b075
...@@ -74,6 +74,7 @@ async def test_single_chat_session_audio(client: openai.AsyncOpenAI, ...@@ -74,6 +74,7 @@ async def test_single_chat_session_audio(client: openai.AsyncOpenAI,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
logprobs=True, logprobs=True,
temperature=0.0,
top_logprobs=5) top_logprobs=5)
assert len(chat_completion.choices) == 1 assert len(chat_completion.choices) == 1
...@@ -130,6 +131,7 @@ async def test_single_chat_session_audio_base64encoded( ...@@ -130,6 +131,7 @@ async def test_single_chat_session_audio_base64encoded(
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
logprobs=True, logprobs=True,
temperature=0.0,
top_logprobs=5) top_logprobs=5)
assert len(chat_completion.choices) == 1 assert len(chat_completion.choices) == 1
...@@ -150,6 +152,7 @@ async def test_single_chat_session_audio_base64encoded( ...@@ -150,6 +152,7 @@ async def test_single_chat_session_audio_base64encoded(
model=model_name, model=model_name,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
temperature=0.0,
) )
message = chat_completion.choices[0].message message = chat_completion.choices[0].message
assert message.content is not None and len(message.content) >= 0 assert message.content is not None and len(message.content) >= 0
......
...@@ -82,6 +82,7 @@ async def test_single_chat_session_video(client: openai.AsyncOpenAI, ...@@ -82,6 +82,7 @@ async def test_single_chat_session_video(client: openai.AsyncOpenAI,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
logprobs=True, logprobs=True,
temperature=0.0,
top_logprobs=5) top_logprobs=5)
assert len(chat_completion.choices) == 1 assert len(chat_completion.choices) == 1
...@@ -174,6 +175,7 @@ async def test_single_chat_session_video_base64encoded( ...@@ -174,6 +175,7 @@ async def test_single_chat_session_video_base64encoded(
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
logprobs=True, logprobs=True,
temperature=0.0,
top_logprobs=5) top_logprobs=5)
assert len(chat_completion.choices) == 1 assert len(chat_completion.choices) == 1
...@@ -194,6 +196,7 @@ async def test_single_chat_session_video_base64encoded( ...@@ -194,6 +196,7 @@ async def test_single_chat_session_video_base64encoded(
model=model_name, model=model_name,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
temperature=0.0,
) )
message = chat_completion.choices[0].message message = chat_completion.choices[0].message
assert message.content is not None and len(message.content) >= 0 assert message.content is not None and len(message.content) >= 0
......
...@@ -83,6 +83,7 @@ async def test_single_chat_session_image(client: openai.AsyncOpenAI, ...@@ -83,6 +83,7 @@ async def test_single_chat_session_image(client: openai.AsyncOpenAI,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
logprobs=True, logprobs=True,
temperature=0.0,
top_logprobs=5) top_logprobs=5)
assert len(chat_completion.choices) == 1 assert len(chat_completion.choices) == 1
...@@ -175,6 +176,7 @@ async def test_single_chat_session_image_base64encoded( ...@@ -175,6 +176,7 @@ async def test_single_chat_session_image_base64encoded(
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
logprobs=True, logprobs=True,
temperature=0.0,
top_logprobs=5) top_logprobs=5)
assert len(chat_completion.choices) == 1 assert len(chat_completion.choices) == 1
...@@ -195,6 +197,7 @@ async def test_single_chat_session_image_base64encoded( ...@@ -195,6 +197,7 @@ async def test_single_chat_session_image_base64encoded(
model=model_name, model=model_name,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
temperature=0.0,
) )
message = chat_completion.choices[0].message message = chat_completion.choices[0].message
assert message.content is not None and len(message.content) >= 0 assert message.content is not None and len(message.content) >= 0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment