This commit is contained in:
Ashwin Bharambe 2024-09-28 15:32:03 -07:00
parent 23028e26ff
commit e61c4954d5
3 changed files with 6 additions and 9 deletions

View file

@ -254,7 +254,6 @@ class LlamaGuardShield(ShieldBase):
for m in messages
]
)
return conversations_str
return PROMPT_TEMPLATE.substitute(
agent_type=messages[-1].role.capitalize(),
categories=categories_str,