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

Get agents tests working

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-04 10:14:55 -08:00 • committed by Ashwin Bharambe
parent 62dd3b376c
commit 66b658dcce
8 changed files with 352 additions and 269 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/tests/safety/test_safety.py
View file

@ -11,6 +11,11 @@ from llama_stack.apis.safety import * # noqa: F403
from llama_stack.distribution.datatypes import * # noqa: F403
# How to run this test:
#
# pytest -v -s llama_stack/providers/tests/safety/test_safety.py
# -m "ollama"
@pytest.mark.parametrize(
"inference_model", [pytest.param("Llama-Guard-3-1B", id="")], indirect=True