Skip to content

GitLab

Unverified Commit c77eb8a3 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[Bugfix] Set temperature=0.7 in test_guided_choice_chat (#11264)

parent 2d1b9baa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/entrypoints/openai/test_chat.py
View file @ c77eb8a3
...@@ -482,6 +482,7 @@ async def test_guided_choice_chat(client: openai.AsyncOpenAI, ...@@ -482,6 +482,7 @@ async def test_guided_choice_chat(client: openai.AsyncOpenAI,
model=MODEL_NAME, model=MODEL_NAME,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
temperature=0.7,
extra_body=dict(guided_choice=sample_guided_choice, extra_body=dict(guided_choice=sample_guided_choice,
guided_decoding_backend=guided_decoding_backend)) guided_decoding_backend=guided_decoding_backend))
choice1 = chat_completion.choices[0].message.content choice1 = chat_completion.choices[0].message.content
...@@ -496,6 +497,7 @@ async def test_guided_choice_chat(client: openai.AsyncOpenAI, ...@@ -496,6 +497,7 @@ async def test_guided_choice_chat(client: openai.AsyncOpenAI,
model=MODEL_NAME, model=MODEL_NAME,
messages=messages, messages=messages,
max_completion_tokens=10, max_completion_tokens=10,
temperature=0.7,
extra_body=dict(guided_choice=sample_guided_choice, extra_body=dict(guided_choice=sample_guided_choice,
guided_decoding_backend=guided_decoding_backend)) guided_decoding_backend=guided_decoding_backend))
choice2 = chat_completion.choices[0].message.content choice2 = chat_completion.choices[0].message.content
......
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