phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-01 18:24:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[test automation] add support for test config

Browse source
This commit is contained in:
Sixian Yi 2025-01-07 00:14:43 -08:00
parent 4d3f0ae79b
commit 702cf2d563
3 changed files with 66 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

15
llama_stack/providers/tests/tools/conftest.py
View file

@ -34,21 +34,6 @@ def pytest_configure(config):
)
def pytest_addoption(parser):
parser.addoption(
"--inference-model",
action="store",
default="meta-llama/Llama-3.2-3B-Instruct",
help="Specify the inference model to use for testing",
)
parser.addoption(
"--safety-shield",
action="store",
default="meta-llama/Llama-Guard-3-1B",
help="Specify the safety shield to use for testing",
)
def pytest_generate_tests(metafunc):
if "tools_stack" in metafunc.fixturenames:
available_fixtures = {