phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

revert excluded cat defaults

Browse source
This commit is contained in:
Kate Plawiak 2024-07-22 22:09:44 -07:00
parent ab8a220faa
commit ab829b0557
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_toolchain/safety/shields/llama_guard.py
View file

@ -111,7 +111,7 @@ class LlamaGuardShield(ShieldBase):
def instance( def instance(
on_violation_action=OnViolationAction.RAISE, on_violation_action=OnViolationAction.RAISE,
model_dir: str = None, model_dir: str = None,
excluded_categories: List[str] = [], excluded_categories: List[str] = None,
disable_input_check: bool = False, disable_input_check: bool = False,
disable_output_check: bool = False, disable_output_check: bool = False,
) -> "LlamaGuardShield": ) -> "LlamaGuardShield":
@ -130,7 +130,7 @@ class LlamaGuardShield(ShieldBase):
self, self,
on_violation_action: OnViolationAction = OnViolationAction.RAISE, on_violation_action: OnViolationAction = OnViolationAction.RAISE,
model_dir: str = None, model_dir: str = None,
excluded_categories: List[str] = [], excluded_categories: List[str] = None,
disable_input_check: bool = False, disable_input_check: bool = False,
disable_output_check: bool = False, disable_output_check: bool = False,
): ):
@ -140,6 +140,9 @@ class LlamaGuardShield(ShieldBase):
assert model_dir is not None, "Llama Guard model_dir is None" assert model_dir is not None, "Llama Guard model_dir is None"
if excluded_categories is None:
excluded_categories = []
assert len(excluded_categories) == 0 or all( assert len(excluded_categories) == 0 or all(
x in SAFETY_CATEGORIES_TO_CODE_MAP.values() for x in excluded_categories x in SAFETY_CATEGORIES_TO_CODE_MAP.values() for x in excluded_categories
), "Invalid categories in excluded categories. Expected format is ['S1', 'S2', ..]" ), "Invalid categories in excluded categories. Expected format is ['S1', 'S2', ..]"