nuke safety/list_shields, we don't need it now

This commit is contained in:
Ashwin Bharambe 2024-09-22 19:49:08 -07:00
parent bc394882a0
commit 484dc2e4f5
3 changed files with 2 additions and 51 deletions

View file

@ -37,16 +37,8 @@ class RunShieldResponse(BaseModel):
violation: Optional[SafetyViolation] = None
@json_schema_type
class ListShieldsResponse(BaseModel):
shields: List[str] = None
class Safety(Protocol):
@webmethod(route="/safety/run_shield")
async def run_shield(
self, shield: str, messages: List[Message], params: Dict[str, Any] = None
) -> RunShieldResponse: ...
@webmethod(route="/safety/list_shields")
async def list_shields(self) -> ListShieldsResponse: ...