fix agent test w/ shields

This commit is contained in:
Xi Yan 2025-01-17 14:54:12 -08:00
parent 8cf7f72116
commit e5fcdf8aa5

View file

@ -182,6 +182,7 @@ 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()
if len(agent_config["output_shields"]) > 0:
assert "No Violation" in logs_str