From 286a413d85a9b3051f2ae388db87a8f8bf3d15bc Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 11 Feb 2026 12:56:23 +0530 Subject: [PATCH] Fix failing tests --- .../anthropic_claude3_transformation.py | 38 +++++++++++++++++++ litellm/types/llms/anthropic.py | 1 + 2 files changed, 39 insertions(+) diff --git a/litellm/llms/bedrock/messages/invoke_transformations/anthropic_claude3_transformation.py b/litellm/llms/bedrock/messages/invoke_transformations/anthropic_claude3_transformation.py index 961158bf053..823a1c33c44 100644 --- a/litellm/llms/bedrock/messages/invoke_transformations/anthropic_claude3_transformation.py +++ b/litellm/llms/bedrock/messages/invoke_transformations/anthropic_claude3_transformation.py @@ -222,6 +222,44 @@ class AmazonAnthropicClaudeMessagesConfig( for beta in beta_headers_to_remove: beta_set.discard(beta) + def _supports_tool_search_on_bedrock(self, model: str) -> bool: + """ + Check if the model supports tool search on Bedrock. + + On Amazon Bedrock, server-side tool search is supported on Claude Opus 4.5 + and Claude Sonnet 4.5 with the tool-search-tool-2025-10-19 beta header. + + Ref: https://platform.claude.com/docs/en/agents-and-tools/tool-use/tool-search-tool + + Args: + model: The model name + + Returns: + True if the model supports tool search on Bedrock + """ + model_lower = model.lower() + + # Supported models for tool search on Bedrock + supported_patterns = [ + # Opus 4.5 + "opus-4.5", + "opus_4.5", + "opus-4-5", + "opus_4_5", + # Sonnet 4.5 + "sonnet-4.5", + "sonnet_4.5", + "sonnet-4-5", + "sonnet_4_5", + # Opus 4.6 + "opus-4.6", + "opus_4.6", + "opus-4-6", + "opus_4_6", + ] + + return any(pattern in model_lower for pattern in supported_patterns) + def _get_tool_search_beta_header_for_bedrock( self, model: str, diff --git a/litellm/types/llms/anthropic.py b/litellm/types/llms/anthropic.py index 94a7195a95b..20568e9b4b6 100644 --- a/litellm/types/llms/anthropic.py +++ b/litellm/types/llms/anthropic.py @@ -639,6 +639,7 @@ class ANTHROPIC_BETA_HEADER_VALUES(str, Enum): STRUCTURED_OUTPUT_2025_09_25 = "structured-outputs-2025-11-13" ADVANCED_TOOL_USE_2025_11_20 = "advanced-tool-use-2025-11-20" FAST_MODE_2026_02_01 = "fast-mode-2026-02-01" + COMPACT_2026_01_12 = "compact-2026-01-12" # Tool search beta header constant (for Anthropic direct API and Microsoft Foundry)