From 5b7826d8a02703689d2eef2da6ef1d37e5c066d9 Mon Sep 17 00:00:00 2001 From: seva Date: Mon, 23 Feb 2026 11:12:59 +0100 Subject: [PATCH] feat(bedrock): add context_management support for Anthropic models on Bedrock Converse Fixes #21912 Add context_management to supported params and pass-through in map_openai_params for Anthropic models on Bedrock Converse API. The parameter flows into additionalModelRequestFields automatically. The beta header (context-management-2025-06-27) is already supported in anthropic_beta_headers_config.json for bedrock_converse. --- litellm/llms/bedrock/chat/converse_transformation.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index d4fd0606302..39095566398 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -577,6 +577,11 @@ class AmazonConverseConfig(BaseConfig): ): supported_params.append("thinking") supported_params.append("reasoning_effort") + + # Anthropic models on Bedrock support context_management + if base_model.startswith("anthropic"): + supported_params.append("context_management") + return supported_params def map_tool_choice_values( @@ -942,6 +947,9 @@ class AmazonConverseConfig(BaseConfig): optional_params = self._add_tools_to_optional_params( optional_params=optional_params, tools=[grounding_tool] ) + if param == "context_management" and isinstance(value, dict): + # Pass through Anthropic-specific context_management parameter + optional_params["context_management"] = value # Only update thinking tokens for non-GPT-OSS models and non-Nova-Lite-2 models # Nova 2 handles token budgeting differently through reasoningConfig