From 27a08b7266d0b1b27399e0966e4b8645d66caa7e Mon Sep 17 00:00:00 2001 From: Hardik Shah Date: Mon, 24 Feb 2025 13:16:40 -0800 Subject: [PATCH] test fix for sometimes tools get called more than once --- tests/client-sdk/agents/test_agents.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/client-sdk/agents/test_agents.py b/tests/client-sdk/agents/test_agents.py index 27a69c90a..876a9baf9 100644 --- a/tests/client-sdk/agents/test_agents.py +++ b/tests/client-sdk/agents/test_agents.py @@ -357,7 +357,7 @@ def test_tool_choice(llama_stack_client, agent_config): assert len(tool_execution_steps) == 0 tool_execution_steps = run_agent("get_boiling_point") - assert len(tool_execution_steps) == 1 and tool_execution_steps[0].tool_calls[0].tool_name == "get_boiling_point" + assert len(tool_execution_steps) >= 1 and tool_execution_steps[0].tool_calls[0].tool_name == "get_boiling_point" # TODO: fix this flaky test