minor enhancement for test fixtures

meta-llama · Nov 19, 2024 · ea52a3e · ea52a3e
1 parent fcc2132
commit ea52a3e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/llama_stack/providers/tests/agents/fixtures.py b/llama_stack/providers/tests/agents/fixtures.py
@@ -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
diff --git a/llama_stack/providers/tests/safety/fixtures.py b/llama_stack/providers/tests/safety/fixtures.py
@@ -47,6 +47,9 @@ def safety_shield(request):
     else:
         params = {}
 
+    if not shield_id:
+        return None
+
     return ShieldInput(
         shield_id=shield_id,
         params=params,