minor enhancement for test fixtures

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

View file

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

View file

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