update as per comments

This commit is contained in:
Swapna Lekkala 2025-08-05 11:29:37 -07:00
parent 2d608ddd3b
commit 13fb8126a1
3 changed files with 65 additions and 48 deletions

View file

@ -75,7 +75,7 @@ class SafetyRouter(Safety):
return matches[0]
shield_id = await get_shield_id(self, model)
logger.debug(f"SafetyRouter.create: {shield_id}")
logger.debug(f"SafetyRouter.run_moderation: {shield_id}")
provider = await self.routing_table.get_provider_impl(shield_id)
return await provider.run_moderation(