phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 02:32:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix agent test w/ shields

Browse source
This commit is contained in:
Xi Yan 2025-01-17 14:54:12 -08:00
parent 8cf7f72116
commit e5fcdf8aa5
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/client-sdk/agents/test_agents.py
View file

@ -182,7 +182,8 @@ def test_builtin_tool_web_search(llama_stack_client, agent_config):
assert "tool_execution>" in logs_str
assert "Tool:brave_search Response:" in logs_str
assert "mark zuckerberg" in logs_str.lower()
assert "No Violation" in logs_str
if len(agent_config["output_shields"]) > 0:
assert "No Violation" in logs_str
def test_builtin_tool_code_execution(llama_stack_client, agent_config):