Update safety implementation inside agents

This commit is contained in:
Ashwin Bharambe 2024-09-20 14:27:54 -07:00 committed by Xi Yan
parent 82ddd851c8
commit d6a41d98d2
8 changed files with 17 additions and 67 deletions

View file

@ -25,7 +25,7 @@ class BedrockSafetyAdapter(Safety):
async def run_shield(
self,
shield: ShieldType,
shield: str,
messages: List[Message],
) -> RunShieldResponse:
# clients will set api_keys by doing something like: