phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-22 22:42:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update method name to run_moderation

Browse source
This commit is contained in:
Swapna Lekkala 2025-08-04 14:15:42 -07:00
parent 70ebc4b448
commit 2d608ddd3b
6 changed files with 295 additions and 295 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/core/routers/safety.py
View file

@ -62,7 +62,7 @@ class SafetyRouter(Safety):
params=params,
)
async def create(self, input: str | list[str], model: str) -> ModerationObject:
async def run_moderation(self, input: str | list[str], model: str) -> ModerationObject:
async def get_shield_id(self, model: str) -> str:
"""Get Shield id from model (provider_resource_id) of shield."""
list_shields_response = await self.routing_table.list_shields()
@ -78,7 +78,7 @@ class SafetyRouter(Safety):
logger.debug(f"SafetyRouter.create: {shield_id}")
provider = await self.routing_table.get_provider_impl(shield_id)
return await provider.create(
return await provider.run_moderation(
input=input,
model=model,
)