fix safety sig

This commit is contained in:
Xi Yan 2024-11-06 16:11:36 -08:00
parent 994732e2e0
commit 67bfd946dc
2 changed files with 3 additions and 3 deletions

View file

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