phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-03 21:02:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update client-sdk test config option handling

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-31 15:29:55 -08:00
parent 589a6911ba
commit f0ba367877
2 changed files with 6 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/client-sdk/conftest.py
View file

@ -52,6 +52,12 @@ def pytest_addoption(parser):
default=VISION_MODEL,
help="Specify the vision inference model to use for testing",
)
parser.addoption(
"--safety-shield",
action="store",
default="meta-llama/Llama-Guard-3-1B",
help="Specify the safety shield model to use for testing",
)
@pytest.fixture(scope="session")