Update safety implementation inside agents

This commit is contained in:
Ashwin Bharambe 2024-09-20 14:27:54 -07:00
parent 51245a417b
commit 6e0f283f52
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: