From b4942942b911481d37d099dece58f638c3635f64 Mon Sep 17 00:00:00 2001 From: seva Date: Mon, 23 Feb 2026 11:21:59 +0100 Subject: [PATCH] Handle both OpenAI list and Anthropic dict formats for context_management Use AnthropicConfig.map_openai_context_management_to_anthropic() to transform OpenAI list format to Anthropic dict format, consistent with the direct Anthropic provider implementation. --- litellm/llms/bedrock/chat/converse_transformation.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index 39095566398..6bf1fee1eb7 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -947,9 +947,15 @@ 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 + if param == "context_management" and isinstance(value, (list, dict)): + # Supports both OpenAI list format and Anthropic dict format + anthropic_context_management = ( + AnthropicConfig.map_openai_context_management_to_anthropic(value) + ) + if anthropic_context_management is not None: + optional_params["context_management"] = ( + anthropic_context_management + ) # Only update thinking tokens for non-GPT-OSS models and non-Nova-Lite-2 models # Nova 2 handles token budgeting differently through reasoningConfig