phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

minor enhancement for test fixtures

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-18 22:20:59 -08:00
parent fcc2132e6f
commit ea52a3ee1c
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/tests/agents/fixtures.py
View file

@ -83,6 +83,6 @@ async def agents_stack(request, inference_model, safety_shield):
) )
for model in inference_models for model in inference_models
], ],
shields=[safety_shield], shields=[safety_shield] if safety_shield else [],
) )
return test_stack return test_stack

3
llama_stack/providers/tests/safety/fixtures.py
View file

@ -47,6 +47,9 @@ def safety_shield(request):
else: else:
params = {} params = {}
if not shield_id:
return None
return ShieldInput( return ShieldInput(
shield_id=shield_id, shield_id=shield_id,
params=params, params=params,