fix: test_multi_tool_calls with no shield

test_multi_tool_calls is intended to test if a tool was called twice,
the distro it was tested against may not have any shields.

Signed-off-by: Derek Higgins <[email protected]>

Author: derekhiggins

tests/integration/agents/test_agents.py

@@ -696,19 +696,11 @@ def test_multi_tool_calls(llama_stack_client, agent_config):
         stream=False,
     )
     steps = response.steps
-    assert len(steps) == 7
-    assert steps[0].step_type == "shield_call"
-    assert steps[1].step_type == "inference"
-    assert steps[2].step_type == "shield_call"
-    assert steps[3].step_type == "tool_execution"
-    assert steps[4].step_type == "shield_call"
-    assert steps[5].step_type == "inference"
-    assert steps[6].step_type == "shield_call"
-
-    tool_execution_step = steps[3]
-    assert len(tool_execution_step.tool_calls) == 2
-    assert tool_execution_step.tool_calls[0].tool_name.startswith("get_boiling_point")
-    assert tool_execution_step.tool_calls[1].tool_name.startswith("get_boiling_point")
+    for step in steps:
+        if step.step_type == "tool_execution":
+            assert len(step.tool_calls) == 2
+            assert step.tool_calls[0].tool_name.startswith("get_boiling_point")
+            assert step.tool_calls[1].tool_name.startswith("get_boiling_point")
 
     output = response.output_message.content.lower()
     assert "-100" in output and "-212" in output