phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-29 11:24:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix client-sdk inference text -- spurious parameterization of test_case

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-20 22:46:17 -08:00
parent 81ce39a607
commit 36b762303c
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/client-sdk/inference/test_text_inference.py
View file

@ -250,9 +250,12 @@ def test_text_chat_completion_with_tool_calling_and_streaming(
assert tool_invocation_content == "[get_weather, {'location': 'San Francisco, CA'}]" assert tool_invocation_content == "[get_weather, {'location': 'San Francisco, CA'}]"
@pytest.mark.parametrize("test_case", ["chat_completion-01"])
def test_text_chat_completion_with_tool_choice_required( def test_text_chat_completion_with_tool_choice_required(
llama_stack_client, text_model_id, get_weather_tool_definition, provider_tool_format, inference_provider_type llama_stack_client,
text_model_id,
get_weather_tool_definition,
provider_tool_format,
inference_provider_type,
): ):
response = llama_stack_client.inference.chat_completion( response = llama_stack_client.inference.chat_completion(
model_id=text_model_id, model_id=text_model_id,
@ -261,7 +264,10 @@ def test_text_chat_completion_with_tool_choice_required(
{"role": "user", "content": "What's the weather like in San Francisco?"}, {"role": "user", "content": "What's the weather like in San Francisco?"},
], ],
tools=[get_weather_tool_definition], tools=[get_weather_tool_definition],
tool_config={"tool_choice": "required", "tool_prompt_format": provider_tool_format}, tool_config={
"tool_choice": "required",
"tool_prompt_format": provider_tool_format,
},
stream=True, stream=True,
) )
tool_invocation_content = extract_tool_invocation_content(response) tool_invocation_content = extract_tool_invocation_content(response)