phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix safety signature mismatch (#388)

Browse source 
* fix safety sig

* shield_type->identifier
This commit is contained in:
Xi Yan 2024-11-06 16:30:47 -08:00 • committed by GitHub
parent 7c340f0236
commit 8fc2d212a2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/meta_reference/safety/safety.py
View file

@ -57,13 +57,13 @@ class MetaReferenceSafetyImpl(Safety, ShieldsProtocolPrivate):
async def run_shield( async def run_shield(
self, self,
shield_type: str, identifier: str,
messages: List[Message], messages: List[Message],
params: Dict[str, Any] = None, params: Dict[str, Any] = None,
) -> RunShieldResponse: ) -> RunShieldResponse:
shield_def = await self.shield_store.get_shield(shield_type) shield_def = await self.shield_store.get_shield(identifier)
if not shield_def: if not shield_def:
raise ValueError(f"Unknown shield {shield_type}") raise ValueError(f"Unknown shield {identifier}")
shield = self.get_shield_impl(shield_def) shield = self.get_shield_impl(shield_def)