mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-06-28 02:53:30 +00:00
minor enhancement for test fixtures
This commit is contained in:
parent
fcc2132e6f
commit
ea52a3ee1c
2 changed files with 4 additions and 1 deletions
|
@ -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
|
||||
|
|
|
@ -47,6 +47,9 @@ def safety_shield(request):
|
|||
else:
|
||||
params = {}
|
||||
|
||||
if not shield_id:
|
||||
return None
|
||||
|
||||
return ShieldInput(
|
||||
shield_id=shield_id,
|
||||
params=params,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue