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

[ez] move addoption to conftest in root test directory and fix broken imports

Browse source
This commit is contained in:
Sixian Yi 2025-01-06 21:34:07 -08:00
parent 965644ce68
commit 4d3f0ae79b
9 changed files with 28 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

24
llama_stack/providers/tests/conftest.py
View file

@ -73,6 +73,30 @@ def pytest_addoption(parser):
parser.addoption(
"--env", action="append", help="Set environment variables, e.g. --env KEY=value"
)
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",
)
parser.addoption(
"--embedding-model",
action="store",
default=None,
help="Specify the embedding model to use for testing",
)
parser.addoption(
"--judge-model",
action="store",
default="meta-llama/Llama-3.1-8B-Instruct",
help="Specify the judge model to use for testing",
)
def make_provider_id(providers: Dict[str, str]) -> str: