safety api

This commit is contained in:
Xi Yan 2024-09-11 13:41:15 -07:00
parent 959c499cac
commit 4b34f741d0
3 changed files with 9 additions and 11 deletions

View file

@ -86,5 +86,6 @@ class Safety(Protocol):
@webmethod(route="/safety/run_shields")
async def run_shields(
self,
request: RunShieldRequest,
messages: List[Message],
shields: List[ShieldDefinition],
) -> RunShieldResponse: ...