phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 14:08:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update test suite to moderations.create

Browse source
This commit is contained in:
Swapna Lekkala 2025-08-06 10:44:05 -07:00
parent 817cbc8a42
commit 215c8d91b8
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/safety/test_safety.py
View file

@ -70,7 +70,7 @@ def test_safe_examples_with_run_moderation(client_with_models, shield_id):
shield = [shield for shield in client_with_models.shields.list() if shield.identifier == shield_id][0] shield = [shield for shield in client_with_models.shields.list() if shield.identifier == shield_id][0]
model_id = shield.provider_resource_id model_id = shield.provider_resource_id
for example in examples: for example in examples:
moderation_object = client_with_models.safety.create( moderation_object = client_with_models.moderations.create(
input=[example], input=[example],
model=model_id, model=model_id,
) )
@ -90,7 +90,7 @@ def test_unsafe_examples_with_run_moderation(client_with_models, shield_id):
shield = [shield for shield in client_with_models.shields.list() if shield.identifier == shield_id][0] shield = [shield for shield in client_with_models.shields.list() if shield.identifier == shield_id][0]
model_id = shield.provider_resource_id model_id = shield.provider_resource_id
for example in examples: for example in examples:
moderation_object = client_with_models.safety.create( moderation_object = client_with_models.moderations.create(
input=[example], input=[example],
model=model_id, model=model_id,
) )