Skip to content

Commit 99c9648

Browse files
committed
add audio & video
Signed-off-by: Roger Wang <[email protected]>
1 parent 334cb2a commit 99c9648

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tests/entrypoints/openai/test_audio.py

+2
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

tests/entrypoints/openai/test_video.py

+2
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

0 commit comments

Comments
 (0)