This commit is contained in:
Ashwin Bharambe 2025-07-25 10:58:36 -07:00
parent f6ba8a123d
commit 145da06fdf
4 changed files with 7 additions and 7 deletions

View file

@ -161,7 +161,7 @@ def get_distribution_template() -> DistributionTemplate:
# if the
ShieldInput(
shield_id="llama-guard",
provider_id="${env.SAFETY_MODEL:+inline::llama-guard}",
provider_id="${env.SAFETY_MODEL:+llama-guard}",
provider_shield_id="${env.SAFETY_MODEL:=}",
),
]