phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-26 22:19:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix unit test client calls

Browse source 
Signed-off-by: Derek Higgins <derekh@redhat.com>
This commit is contained in:
Derek Higgins 2025-07-11 09:35:56 +01:00
parent e33a50480d
commit 3e7ea4dd14
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/integration/inference/test_openai_completion.py
View file

@ -179,9 +179,7 @@ def test_openai_completion_prompt_logprobs(llama_stack_client, client_with_model
model=text_model_id, model=text_model_id,
prompt=prompt, prompt=prompt,
stream=False, stream=False,
extra_body={ prompt_logprobs=prompt_logprobs,
"prompt_logprobs": prompt_logprobs,
},
) )
assert len(response.choices) > 0 assert len(response.choices) > 0
choice = response.choices[0] choice = response.choices[0]
@ -196,9 +194,7 @@ def test_openai_completion_guided_choice(llama_stack_client, client_with_models,
model=text_model_id, model=text_model_id,
prompt=prompt, prompt=prompt,
stream=False, stream=False,
extra_body={ guided_choice=["joy", "sadness"],
"guided_choice": ["joy", "sadness"],
},
) )
assert len(response.choices) > 0 assert len(response.choices) > 0
choice = response.choices[0] choice = response.choices[0]