phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 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'}]"
@pytest.mark.parametrize("test_case", ["chat_completion-01"])
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(
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?"},
],
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,
)
tool_invocation_content = extract_tool_invocation_content(response)