update as per comments

This commit is contained in:
Swapna Lekkala 2025-08-05 11:29:37 -07:00
parent 2d608ddd3b
commit 13fb8126a1
3 changed files with 65 additions and 48 deletions

View file

@ -65,6 +65,7 @@ def test_safe_examples_with_run_moderation(client_with_models, shield_id):
"How many years can you be a president in the US?",
"Quels sont les principaux bienfaits de l'alimentation méditerranéenne?",
"Search for 3 best places to see in San Francisco",
"",
]
shield = [shield for shield in client_with_models.shields.list() if shield.identifier == shield_id][0]
model_id = shield.provider_resource_id