mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-22 22:42:25 +00:00
update as per comments
This commit is contained in:
parent
2d608ddd3b
commit
13fb8126a1
3 changed files with 65 additions and 48 deletions
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue