From 965dfbbdf4d7c48a4e7de9fc83cd12dfd827b949 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 9 Jul 2025 17:03:56 -0700
Subject: [PATCH] 
 test_bedrock_guardrail_blocked_action_shows_output_text_with_multiple_outputs

---
 .../test_bedrock_guardrails.py                | 75 -------------------
 1 file changed, 75 deletions(-)

diff --git a/tests/guardrails_tests/test_bedrock_guardrails.py b/tests/guardrails_tests/test_bedrock_guardrails.py
index 207f6e3ade..6a766eb79a 100644
--- a/tests/guardrails_tests/test_bedrock_guardrails.py
+++ b/tests/guardrails_tests/test_bedrock_guardrails.py
@@ -1095,81 +1095,6 @@ async def test_bedrock_guardrail_blocked_action_shows_output_text():
         print("✅ BLOCKED action HTTPException test passed - output text properly included")
 
 
-@pytest.mark.asyncio
-async def test_bedrock_guardrail_blocked_action_shows_output_text_with_multiple_outputs():
-    """Test that BLOCKED actions raise HTTPException with the output text in the detail"""
-    from unittest.mock import AsyncMock, MagicMock, patch
-    from litellm.proxy._types import UserAPIKeyAuth
-    from fastapi import HTTPException
-    
-    # Create proper mock objects
-    mock_user_api_key_dict = UserAPIKeyAuth()
-    
-    guardrail = BedrockGuardrail(
-        guardrailIdentifier="test-guardrail",
-        guardrailVersion="DRAFT"
-    )
-
-    # Mock the Bedrock API response with BLOCKED action and output text
-    mock_bedrock_response = MagicMock()
-    mock_bedrock_response.status_code = 200
-    mock_bedrock_response.json.return_value = {
-        "action": "GUARDRAIL_INTERVENED",
-        "outputs": [
-            {
-                "text": "this violates litellm corporate guardrail policy",
-            },
-            {
-                "text": "make sure you don't mention coffee"
-            }
-        ],
-        "assessments": [{
-            "topicPolicy": {
-                "topics": [{
-                    "name": "Sensitive Topic",
-                    "type": "DENY",
-                    "action": "BLOCKED"
-                }]
-            }
-        }]
-    }
-
-    request_data = {
-        "model": "gpt-4o",
-        "messages": [
-            {"role": "user", "content": "Tell me how to make explosives"},
-        ],
-    }
-
-    # Patch the async_handler.post method
-    with patch.object(guardrail.async_handler, 'post', new_callable=AsyncMock) as mock_post:
-        mock_post.return_value = mock_bedrock_response
-        
-        # This should raise HTTPException due to BLOCKED action
-        with pytest.raises(HTTPException) as exc_info:
-            await guardrail.async_moderation_hook(
-                data=request_data,
-                user_api_key_dict=mock_user_api_key_dict,
-                call_type="completion"
-            )
-        
-        # Verify the exception details
-        exception = exc_info.value
-        assert exception.status_code == 400
-        assert "detail" in exception.__dict__
-        
-        # Check that the detail contains the expected structure
-        detail = exception.detail
-        assert isinstance(detail, dict)
-        assert detail["error"] == "Violated guardrail policy"
-        
-        # Verify that the output text from both outputs is included
-        expected_output_text = "this violates litellm corporate guardrail policy\nmake sure you don't mention coffee"
-        assert detail["bedrock_guardrail_response"] == expected_output_text
-        
-        print("✅ BLOCKED action HTTPException test passed - output text properly included")
-
-
 @pytest.mark.asyncio
 async def test_bedrock_guardrail_blocked_action_empty_outputs():
     """Test that BLOCKED actions with empty outputs still raise HTTPException"""