From aab4084ee30d8d85b41be92979c9b3485a2f8194 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 16 Dec 2024 21:12:18 -0800 Subject: [PATCH] fix shields step --- .../providers/inline/agents/meta_reference/agent_instance.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index 5f84f9b6a..95225b730 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -240,7 +240,6 @@ class ChatAgent(ShieldRunnerMixin): # final boolean (to see whether an exception happened) and then explicitly testing for it. if len(self.input_shields) > 0: - print("input shields", self.input_shields) async for res in self.run_multiple_shields_wrapper( turn_id, input_messages, self.input_shields, "user-input" ): @@ -282,7 +281,6 @@ class ChatAgent(ShieldRunnerMixin): shields: List[str], touchpoint: str, ) -> AsyncGenerator: - print("!!! HI run_multiple_shields_wrapper") with tracing.span("run_shields") as span: span.set_attribute("input", [m.model_dump_json() for m in messages]) if len(shields) == 0: