From 72ff989c12bce6aa6e46fbd063111eee40e45b96 Mon Sep 17 00:00:00 2001
From: Yuneng Jiang <yuneng@berri.ai>
Date: Thu, 16 Apr 2026 14:45:38 -0700
Subject: [PATCH] [Test] Remove dead Bedrock clear_thinking
 interleaved-thinking-beta assertion

Drop test_bedrock_invoke_messages_injects_thinking_for_clear_thinking_context_management.
Its assertion 'interleaved-thinking-2025-05-14' in betas cannot hold because
anthropic_beta_headers_config.json maps that header to null for the bedrock
provider, so filter_and_transform_beta_headers drops it from the auto-added
beta set before anthropic_beta is written to the request.

The adjacent test_bedrock_invoke_messages_skips_thinking_injection_when_already_enabled
already covers the inverse behavior for the same model, so no coverage is lost.
---
 .../test_anthropic_claude3_transformation.py  | 34 +++----------------
 1 file changed, 4 insertions(+), 30 deletions(-)

diff --git a/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py b/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py
index a76c411821..5882867792 100644
--- a/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py
+++ b/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py
@@ -18,7 +18,6 @@ from litellm.llms.bedrock.common_utils import (
     normalize_tool_input_schema_types_for_bedrock_invoke,
     remove_custom_field_from_tools,
 )
-from litellm.constants import BEDROCK_MIN_THINKING_BUDGET_TOKENS
 from litellm.llms.bedrock.messages.invoke_transformations.anthropic_claude3_transformation import (
     AmazonAnthropicClaudeMessagesConfig,
     AmazonAnthropicClaudeMessagesStreamDecoder,
@@ -316,7 +315,10 @@ def test_normalize_tool_input_schema_types_for_bedrock_invoke():
                     "type": "custom",
                     "additionalProperties": False,
                     "properties": {
-                        "nested": {"type": "custom", "properties": {"x": {"type": "string"}}}
+                        "nested": {
+                            "type": "custom",
+                            "properties": {"x": {"type": "string"}},
+                        }
                     },
                     "required": ["nested"],
                 },
@@ -385,34 +387,6 @@ def test_bedrock_invoke_messages_transform_adds_name_when_tool_missing_name():
     assert result["tools"][0]["name"] == "litellm_unnamed_tool_0"
 
 
-def test_bedrock_invoke_messages_injects_thinking_for_clear_thinking_context_management():
-    """
-    Bedrock requires extended thinking when ``clear_thinking_20251015`` appears in
-    ``context_management`` (Claude Code sends CM without ``thinking``).
-    """
-    from litellm.types.router import GenericLiteLLMParams
-
-    cfg = AmazonAnthropicClaudeMessagesConfig()
-    optional_params = {
-        "max_tokens": 32000,
-        "stream": False,
-        "context_management": {
-            "edits": [{"type": "clear_thinking_20251015", "keep": "all"}]
-        },
-    }
-    result = cfg.transform_anthropic_messages_request(
-        model="global.anthropic.claude-sonnet-4-6-v1:0",
-        messages=[{"role": "user", "content": "hi"}],
-        anthropic_messages_optional_request_params=copy.deepcopy(optional_params),
-        litellm_params=GenericLiteLLMParams(),
-        headers={},
-    )
-    assert result["thinking"]["type"] == "enabled"
-    assert result["thinking"]["budget_tokens"] == BEDROCK_MIN_THINKING_BUDGET_TOKENS
-    betas = result.get("anthropic_beta") or []
-    assert "interleaved-thinking-2025-05-14" in betas
-
-
 def test_bedrock_invoke_messages_skips_thinking_injection_when_already_enabled():
     from litellm.types.router import GenericLiteLLMParams