From f92ff86b967f18f83199a7db7f5a42987b0f765b Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 23 Sep 2024 21:22:22 -0700 Subject: [PATCH] fix shields in agents safety --- llama_stack/providers/impls/meta_reference/agents/safety.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/impls/meta_reference/agents/safety.py b/llama_stack/providers/impls/meta_reference/agents/safety.py index b3aa53728..fb5821f6a 100644 --- a/llama_stack/providers/impls/meta_reference/agents/safety.py +++ b/llama_stack/providers/impls/meta_reference/agents/safety.py @@ -43,7 +43,7 @@ class ShieldRunnerMixin: for shield_type in shield_types ] ) - for shield_type, response in zip(shields, responses): + for shield_type, response in zip(shield_types, responses): if not response.violation: continue