From 71e087fa78bd8d119fa30b6362437652cf805683 Mon Sep 17 00:00:00 2001 From: yucheng Date: Wed, 30 Sep 2026 23:38:34 +0000 Subject: [PATCH] test(guardrails): reuse the existing text moderation response helper Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../azure/test_azure_text_moderation.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/tests/test_litellm/proxy/guardrails/guardrail_hooks/azure/test_azure_text_moderation.py b/tests/test_litellm/proxy/guardrails/guardrail_hooks/azure/test_azure_text_moderation.py index faa66fffb4f..accc72b4b39 100644 --- a/tests/test_litellm/proxy/guardrails/guardrail_hooks/azure/test_azure_text_moderation.py +++ b/tests/test_litellm/proxy/guardrails/guardrail_hooks/azure/test_azure_text_moderation.py @@ -80,20 +80,6 @@ async def test_azure_text_moderation_scans_responses_input() -> None: assert mock_post.call_args.kwargs["json"]["text"] == "Review this response input" -def _moderation_response(severity: int) -> Mock: - response = Mock() - response.json.return_value = { - "blocklistsMatch": [], - "categoriesAnalysis": [ - {"category": "Hate", "severity": severity}, - {"category": "Sexual", "severity": 0}, - {"category": "SelfHarm", "severity": 0}, - {"category": "Violence", "severity": 0}, - ], - } - return response - - def _moderation_flagging(flagged: str): def azure_by_text(*args: object, **kwargs: object) -> Mock: body = kwargs["json"]