mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-16 23:03:49 +00:00
address feedback
This commit is contained in:
parent
0eaca98229
commit
fe072620c8
9 changed files with 16 additions and 33 deletions
|
@ -154,12 +154,12 @@ class SafetyRouter(Safety):
|
|||
self,
|
||||
shield_id: str,
|
||||
shield_type: ShieldType,
|
||||
provider_resource_identifier: Optional[str] = None,
|
||||
provider_shield_id: Optional[str] = None,
|
||||
provider_id: Optional[str] = None,
|
||||
params: Optional[Dict[str, Any]] = None,
|
||||
) -> Shield:
|
||||
return await self.routing_table.register_shield(
|
||||
shield_id, shield_type, provider_resource_identifier, provider_id, params
|
||||
shield_id, shield_type, provider_shield_id, provider_id, params
|
||||
)
|
||||
|
||||
async def run_shield(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue