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

fix test

Browse source
This commit is contained in:
Swapna Lekkala 2025-10-13 16:07:17 -07:00
parent f8861bc480
commit 74be622f57
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/providers/inline/agents/meta_reference/responses/test_streaming.py
View file

@ -117,7 +117,9 @@ async def test_apply_guardrails_with_violation(mock_safety_api, mock_inference_a
guardrail_ids = ["llama-guard"]
# Mock moderation to return flagged content
flagged_result = ModerationObjectResults(flagged=True, categories={"violence": True})
flagged_result = ModerationObjectResults(
flagged=True, categories={"violence": True}, user_message="Content flagged by moderation"
)
mock_moderation_object = ModerationObject(id="test-mod-id", model="llama-guard-model", results=[flagged_result])
mock_safety_api.run_moderation.return_value = mock_moderation_object
@ -136,7 +138,7 @@ async def test_apply_guardrails_with_violation(mock_safety_api, mock_inference_a
result = await orchestrator._apply_guardrails(text)
assert result == "Content flagged by moderation"
assert result == "Content flagged by moderation (flagged for: violence)"
async def test_apply_guardrails_empty_inputs(mock_safety_api, mock_inference_api, mock_context):