This commit is contained in:
Xi Yan 2025-03-06 18:01:14 -08:00
parent 8234cdf1a5
commit 2949e6a9b4
3 changed files with 11640 additions and 13 deletions

View file

@ -276,7 +276,6 @@ def test_custom_tool(llama_stack_client_with_mocked_inference, agent_config):
agent_config = {
**agent_config,
"tools": ["builtin::websearch", client_tool],
"client_tools": [client_tool.get_tool_definition()],
}
agent = Agent(llama_stack_client_with_mocked_inference, **agent_config)
@ -571,7 +570,10 @@ def test_rag_and_code_agent(llama_stack_client_with_mocked_inference, agent_conf
assert expected_kw in response.output_message.content.lower()
@pytest.mark.parametrize("client_tools", [(get_boiling_point, False), (get_boiling_point_with_metadata, True)])
@pytest.mark.parametrize(
"client_tools",
[(get_boiling_point, False), (get_boiling_point_with_metadata, True)],
)
def test_create_turn_response(llama_stack_client_with_mocked_inference, agent_config, client_tools):
client_tool, expectes_metadata = client_tools
agent_config = {

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long