phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: simplify true/false evaluation of input/output shields

Browse source 
This code was unnecessarily verbose when calculating has_input_shield
and has_output_shield values, and was simplified to take advantage of
Python's intrinsic treatment of empty lists and None values as false.

Signed-off-by: Ben Browning <bbrownin@redhat.com>
This commit is contained in:
Ben Browning 2025-05-14 16:41:21 -04:00
parent b3493ee94f
commit 18c2494259
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/agents/test_agents.py
View file

@ -698,8 +698,8 @@ def test_multi_tool_calls(llama_stack_client, agent_config):
)
steps = response.steps
has_input_shield = True if agent_config.get("input_shields", None) else False
has_output_shield = True if agent_config.get("output_shields", None) else False
has_input_shield = agent_config.get("input_shields")
has_output_shield = agent_config.get("output_shields")
assert len(steps) == 3 + (2 if has_input_shield else 0) + (2 if has_output_shield else 0)
if has_input_shield:
assert steps[0].step_type == "shield_call"