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

precommit

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-03 11:11:32 -07:00
parent 2a54a2433f
commit cbe7391574
6 changed files with 35 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/integration/responses/test_extra_body_shields.py
View file

@ -25,9 +25,7 @@ def test_shields_via_extra_body(compat_client, text_model_id):
model=text_model_id,
input="What is the capital of France?",
stream=False,
extra_body={
"shields": ["test-shield-1", "test-shield-2"]
}
extra_body={"shields": ["test-shield-1", "test-shield-2"]},
)
# Verify the error message indicates shields are not implemented
@ -35,8 +33,6 @@ def test_shields_via_extra_body(compat_client, text_model_id):
assert "not yet implemented" in error_message.lower() or "not implemented" in error_message.lower()
def test_response_without_shields_still_works(compat_client, text_model_id):
"""Test that responses still work without shields parameter (backwards compatibility)."""
@ -70,9 +66,7 @@ def test_shields_parameter_received_end_to_end(compat_client, text_model_id):
model=text_model_id,
input="Test message for shields verification",
stream=False,
extra_body={
"shields": ["shield-1", "shield-2"]
}
extra_body={"shields": ["shield-1", "shield-2"]},
)
# The NotImplementedError proves that: