shield_type->identifier

This commit is contained in:
Xi Yan 2024-11-06 16:25:35 -08:00
parent 67bfd946dc
commit ea9f5368cb
3 changed files with 6 additions and 6 deletions

View file

@ -48,5 +48,5 @@ class Safety(Protocol):
@webmethod(route="/safety/run_shield")
async def run_shield(
self, shield_type: str, messages: List[Message], params: Dict[str, Any] = None
self, identifier: str, messages: List[Message], params: Dict[str, Any] = None
) -> RunShieldResponse: ...