phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 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, default=VISION_MODEL,
help="Specify the vision inference model to use for testing", 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") @pytest.fixture(scope="session")

9
tests/client-sdk/safety/conftest.py
View file

@ -5,15 +5,6 @@
# the root directory of this source tree. # the root directory of this source tree.
def pytest_addoption(parser):
parser.addoption(
"--safety-shield",
action="store",
default="meta-llama/Llama-Guard-3-1B",
help="Specify the safety shield model to use for testing",
)
def pytest_generate_tests(metafunc): def pytest_generate_tests(metafunc):
if "llama_guard_text_shield_id" in metafunc.fixturenames: if "llama_guard_text_shield_id" in metafunc.fixturenames:
metafunc.parametrize( metafunc.parametrize(