From 1acef7d88091ff0359c0867881deffb1fa37ac0a Mon Sep 17 00:00:00 2001
From: Venkat Donavalli <vdonavalli@live.com>
Date: Wed, 24 Jun 2026 09:10:01 -0500
Subject: [PATCH] style(bedrock/claude_platform): reformat with black + add
 coverage tests

Black and ruff format disagree on line wrapping; CI uses black. Reformat
common_utils.py with black. Add messages-path override test to cover
resolve_unsupported_override call in messages_transformation.py.
---
 .../bedrock/claude_platform/common_utils.py   | 24 +++++++++++----
 .../bedrock/test_claude_platform_provider.py  | 29 +++++++++++++++++++
 2 files changed, 47 insertions(+), 6 deletions(-)

diff --git a/litellm/llms/bedrock/claude_platform/common_utils.py b/litellm/llms/bedrock/claude_platform/common_utils.py
index 97f57b8efcc..40369861d4e 100644
--- a/litellm/llms/bedrock/claude_platform/common_utils.py
+++ b/litellm/llms/bedrock/claude_platform/common_utils.py
@@ -10,7 +10,9 @@ class _SupportsGet(Protocol):
     def get(self, key: str, default: object = None) -> object: ...
 
 
-CLAUDE_PLATFORM_SERVICE_NAME: Literal["aws-external-anthropic"] = "aws-external-anthropic"
+CLAUDE_PLATFORM_SERVICE_NAME: Literal["aws-external-anthropic"] = (
+    "aws-external-anthropic"
+)
 CLAUDE_PLATFORM_BEDROCK_ROUTE = "claude_platform/"
 
 # Auth/routing params consumed by validate_environment / sign_request that
@@ -55,7 +57,9 @@ def filter_claude_platform_request_body(
     Filters a copy so callers' ``sign_request`` still sees ``aws_region_name``.
     """
     unsupported = (
-        unsupported_override if unsupported_override is not None else CLAUDE_PLATFORM_ON_AWS_UNSUPPORTED_REQUEST_PARAMS
+        unsupported_override
+        if unsupported_override is not None
+        else CLAUDE_PLATFORM_ON_AWS_UNSUPPORTED_REQUEST_PARAMS
     )
     dropped_unsupported = [k for k in params if k in unsupported]
     if dropped_unsupported:
@@ -69,7 +73,9 @@ def filter_claude_platform_request_body(
     return {
         k: v
         for k, v in params.items()
-        if k not in CLAUDE_PLATFORM_ON_AWS_NON_REQUEST_PARAMS and k not in unsupported and not k.startswith("aws_")
+        if k not in CLAUDE_PLATFORM_ON_AWS_NON_REQUEST_PARAMS
+        and k not in unsupported
+        and not k.startswith("aws_")
     }
 
 
@@ -108,10 +114,14 @@ class BedrockClaudePlatformMixin(BaseAWSLLM):
             or litellm_params.get("anthropic-workspace-id")
         )
         if workspace_id is None:
-            workspace_id = optional_params.get("anthropic_workspace_id") or litellm_params.get("anthropic_workspace_id")
+            workspace_id = optional_params.get(
+                "anthropic_workspace_id"
+            ) or litellm_params.get("anthropic_workspace_id")
         if workspace_id is not None:
             return str(workspace_id)
-        return get_secret_str("ANTHROPIC_AWS_WORKSPACE_ID") or get_secret_str("ANTHROPIC_WORKSPACE_ID")
+        return get_secret_str("ANTHROPIC_AWS_WORKSPACE_ID") or get_secret_str(
+            "ANTHROPIC_WORKSPACE_ID"
+        )
 
     def _get_required_aws_region_name(self, optional_params: dict) -> str:
         aws_region_name = (
@@ -149,7 +159,9 @@ class BedrockClaudePlatformMixin(BaseAWSLLM):
         )
         if api_base is None:
             aws_region_name = self._get_required_aws_region_name(optional_params)
-            api_base = f"https://{CLAUDE_PLATFORM_SERVICE_NAME}.{aws_region_name}.api.aws"
+            api_base = (
+                f"https://{CLAUDE_PLATFORM_SERVICE_NAME}.{aws_region_name}.api.aws"
+            )
         if not api_base.endswith("/v1/messages"):
             api_base = f"{api_base.rstrip('/')}/v1/messages"
         return api_base
diff --git a/tests/test_litellm/llms/bedrock/test_claude_platform_provider.py b/tests/test_litellm/llms/bedrock/test_claude_platform_provider.py
index a3be1705940..51ee6fb3484 100644
--- a/tests/test_litellm/llms/bedrock/test_claude_platform_provider.py
+++ b/tests/test_litellm/llms/bedrock/test_claude_platform_provider.py
@@ -505,6 +505,35 @@ def test_claude_platform_unsupported_override_ignores_invalid_type():
     assert request_body["max_tokens"] == 10
 
 
+def test_claude_platform_messages_unsupported_override_allows_context_management():
+    """
+    Messages-path: operators can pass claude_platform_unsupported_params=[]
+    in litellm_params to allow context_management through.
+    """
+    import litellm
+    from litellm.types.utils import LlmProviders
+
+    config = litellm.ProviderConfigManager.get_provider_anthropic_messages_config(
+        model="claude_platform/claude-sonnet-4-6",
+        provider=LlmProviders.BEDROCK,
+    )
+    assert config is not None
+
+    request_body = config.transform_anthropic_messages_request(
+        model="claude_platform/claude-sonnet-4-6",
+        messages=[{"role": "user", "content": "hello"}],
+        anthropic_messages_optional_request_params={
+            "context_management": {"edits": [{"type": "clear_tool_uses_20250919"}]},
+            "max_tokens": 10,
+        },
+        litellm_params={"claude_platform_unsupported_params": []},
+        headers={},
+    )
+
+    assert "context_management" in request_body
+    assert request_body["max_tokens"] == 10
+
+
 def test_chat_completion_claude_platform_sigv4_body_has_no_auth_params():
     """
     End-to-end (mocked transport): a config-driven SigV4 call with