From 227e6abe3e056bb73643bd4a1709275a6dc094e8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 15 Jan 2026 15:32:13 -0800 Subject: [PATCH] fix _update_headers_with_anthropic_beta --- litellm/types/llms/anthropic_tool_search.py | 36 +++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 litellm/types/llms/anthropic_tool_search.py diff --git a/litellm/types/llms/anthropic_tool_search.py b/litellm/types/llms/anthropic_tool_search.py new file mode 100644 index 00000000000..d8656ce8bb3 --- /dev/null +++ b/litellm/types/llms/anthropic_tool_search.py @@ -0,0 +1,36 @@ +""" +Tool Search Beta Header Configuration + +Reference: https://platform.claude.com/docs/en/agents-and-tools/tool-use/tool-search-tool +""" + +from typing import Dict + +from litellm.types.utils import LlmProviders + +# Tool search beta header values +TOOL_SEARCH_BETA_HEADER_ANTHROPIC = "advanced-tool-use-2025-11-20" +TOOL_SEARCH_BETA_HEADER_VERTEX = "tool-search-tool-2025-10-19" +TOOL_SEARCH_BETA_HEADER_BEDROCK = "tool-search-tool-2025-10-19" + + +# Mapping of custom_llm_provider -> tool search beta header +TOOL_SEARCH_BETA_HEADER_BY_PROVIDER: Dict[str, str] = { + LlmProviders.ANTHROPIC.value: TOOL_SEARCH_BETA_HEADER_ANTHROPIC, + LlmProviders.AZURE.value: TOOL_SEARCH_BETA_HEADER_ANTHROPIC, + LlmProviders.AZURE_AI.value: TOOL_SEARCH_BETA_HEADER_ANTHROPIC, + LlmProviders.VERTEX_AI.value: TOOL_SEARCH_BETA_HEADER_VERTEX, + LlmProviders.VERTEX_AI_BETA.value: TOOL_SEARCH_BETA_HEADER_VERTEX, + LlmProviders.BEDROCK.value: TOOL_SEARCH_BETA_HEADER_BEDROCK, +} + + +def get_tool_search_beta_header(custom_llm_provider: str) -> str: + """ + Get the tool search beta header for a given provider. + """ + return TOOL_SEARCH_BETA_HEADER_BY_PROVIDER.get( + custom_llm_provider, + TOOL_SEARCH_BETA_HEADER_ANTHROPIC + ) +