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

fix shields step

Browse source
This commit is contained in:
Xi Yan 2024-12-16 21:12:18 -08:00
parent bf961f8aa5
commit aab4084ee3
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -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: