From 1278012a91b7c086e67bf5b6aafe8fd5e3a54408 Mon Sep 17 00:00:00 2001
From: Mateo Wang <277851410+mateo-berri@users.noreply.github.com>
Date: Wed, 24 Jun 2026 08:34:53 -0700
Subject: [PATCH] test(bedrock): update normalize_checks test to keep empty
 known checks

---
 .../test_bedrock_invoke_guardrail_checks.py              | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_bedrock_invoke_guardrail_checks.py b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_bedrock_invoke_guardrail_checks.py
index c89054be590..b31a9207fa1 100644
--- a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_bedrock_invoke_guardrail_checks.py
+++ b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_bedrock_invoke_guardrail_checks.py
@@ -63,11 +63,14 @@ def test_init_normalizes_checks_and_drops_unknown_keys():
     g = BedrockGuardrail(
         checks={
             "contentFilter": {"categories": [{"category": "VIOLENCE"}]},
-            "unknownCheck": {"foo": "bar"},
-            "promptAttack": {},  # empty -> dropped
+            "unknownCheck": {"foo": "bar"},  # unknown key -> dropped
+            "promptAttack": {},  # empty known check -> kept (enable with defaults)
         }
     )
-    assert g.checks == {"contentFilter": {"categories": [{"category": "VIOLENCE"}]}}
+    assert g.checks == {
+        "contentFilter": {"categories": [{"category": "VIOLENCE"}]},
+        "promptAttack": {},
+    }
 
 
 def test_init_empty_checks_falls_back_to_apply_mode():