llama-stack-mirror/tests/integration/safety
Matthew Farrellee ef26259209
feat: add llama guard 4 model (#2579)
add support for Llama Guard 4 model to the llama_guard safety provider

test with -

0. NVIDIA_API_KEY=... llama stack build --image-type conda --image-name
env-nvidia --providers
inference=remote::nvidia,safety=inline::llama-guard --run
1. llama-stack-client models register meta-llama/Llama-Guard-4-12B
--provider-model-id meta/llama-guard-4-12b
2. pytest tests/integration/safety/test_llama_guard.py

Co-authored-by: raghotham <rsm@meta.com>
2025-07-03 22:29:04 -07:00
..
resources refactor: tests/unittests -> tests/unit; tests/api -> tests/integration 2025-03-04 09:57:00 -08:00
__init__.py fix: remove ruff N999 (#1388) 2025-03-07 11:14:04 -08:00
test_llama_guard.py feat: add llama guard 4 model (#2579) 2025-07-03 22:29:04 -07:00
test_safety.py fix: misc fixes for tests kill horrible warnings 2025-04-12 17:12:11 -07:00
test_vision_safety.py refactor(test): introduce --stack-config and simplify options (#1404) 2025-03-05 17:02:02 -08:00