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

@ -154,12 +154,12 @@ class SafetyRouter(Safety):
async def run_shield(
self,
identifier: str,
shield_type: str,
messages: List[Message],
params: Dict[str, Any] = None,
) -> RunShieldResponse:
return await self.routing_table.get_provider_impl(identifier).run_shield(
identifier=identifier,
shield_type=shield_type,
messages=messages,
params=params,
)