minor updates

This commit is contained in:
Dinesh Yeduguru 2024-11-07 22:40:15 -08:00
parent 04a2965967
commit 39f0c5f544
2 changed files with 2 additions and 2 deletions

View file

@ -37,7 +37,7 @@ class ShieldRunnerMixin:
responses = await asyncio.gather(
*[
self.safety_api.run_shield(
identifier=identifier,
shield_id=identifier,
messages=messages,
)
for identifier in identifiers