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

minor fix

Browse source
This commit is contained in:
Swapna Lekkala 2025-08-01 16:21:40 -07:00
parent 0d07ba1a65
commit 70ebc4b448
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/safety/llama_guard/llama_guard.py
View file

@ -423,7 +423,7 @@ class LlamaGuardShield:
def create_safe_moderation_object(self, model: str) -> ModerationObject: def create_safe_moderation_object(self, model: str) -> ModerationObject:
"""Create a ModerationObject for safe content.""" """Create a ModerationObject for safe content."""
categories = dict.fromkeys(OPENAI_TO_LLAMA_CATEGORIES_MAP.keys(), False) categories = dict.fromkeys(OPENAI_TO_LLAMA_CATEGORIES_MAP.keys(), False)
category_scores = dict.fromkeys(OPENAI_TO_LLAMA_CATEGORIES_MAP.keys(), 0.0) category_scores = dict.fromkeys(OPENAI_TO_LLAMA_CATEGORIES_MAP.keys(), 1.0)
category_applied_input_types = {key: [] for key in OPENAI_TO_LLAMA_CATEGORIES_MAP.keys()} category_applied_input_types = {key: [] for key in OPENAI_TO_LLAMA_CATEGORIES_MAP.keys()}
return ModerationObject( return ModerationObject(