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

working bedrock tests

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-07 14:57:12 -08:00
parent d960f9b60f
commit e0f227f23c
4 changed files with 64 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/providers/tests/safety/conftest.py
View file

@ -57,7 +57,7 @@ DEFAULT_PROVIDER_COMBINATIONS = [
def pytest_configure(config):
for mark in ["meta_reference", "ollama", "together", "remote"]:
for mark in ["meta_reference", "ollama", "together", "remote", "bedrock"]:
config.addinivalue_line(
"markers",
f"{mark}: marks tests as {mark} specific",
@ -71,6 +71,18 @@ def pytest_addoption(parser):
default=None,
help="Specify the safety model to use for testing",
)
parser.addoption(
"--bedrock-guardrail-id",
action="store",
default=None,
help="Specify the guard rail ID to use for testing bedrock",
)
parser.addoption(
"--bedrock-guardrail-version",
action="store",
default=None,
help="Specify the guard rail version to use for testing bedrock",
)
SAFETY_MODEL_PARAMS = [