fix failing test

Author: radofuchs

tests/e2e/features/steps/info.py

@@ -84,14 +84,16 @@ def check_shield_structure(context: Context) -> None:
 
     assert found_shield is not None, "No shield found in response"
 
+    expected_model = context.default_model
+
     # Validate structure and values
     assert found_shield["type"] == "shield", "type should be 'shield'"
     assert (
         found_shield["provider_id"] == "llama-guard"
     ), "provider_id should be 'llama-guard'"
     assert (
-        found_shield["provider_resource_id"] == "llama-guard-shield"
-    ), "provider_resource_id should be 'llama-guard-shield'"
+        found_shield["provider_resource_id"] == expected_model
+    ), f"provider_resource_id should be '{expected_model}'"
     assert (
         found_shield["identifier"] == "llama-guard-shield"
     ), "identifier should be 'llama-guard-shield'"