From fc5e947a4dab4114d2daa4dfd55c994228434d4b Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Mon, 16 Feb 2026 15:57:14 +0530 Subject: [PATCH 1/3] Add azure-ai and vertex ai in test_anthropic_messages_with_all_beta_headers --- .../test_all_beta_headers.py | 2 ++ tests/proxy_e2e_anthropic_messages_tests/test_config.yaml | 6 ++++++ 2 files changed, 8 insertions(+) diff --git a/tests/proxy_e2e_anthropic_messages_tests/test_all_beta_headers.py b/tests/proxy_e2e_anthropic_messages_tests/test_all_beta_headers.py index cf688d7deae..14a301add57 100644 --- a/tests/proxy_e2e_anthropic_messages_tests/test_all_beta_headers.py +++ b/tests/proxy_e2e_anthropic_messages_tests/test_all_beta_headers.py @@ -32,6 +32,8 @@ def get_all_supported_anthropic_beta_headers(provider: str): "model_name,provider_name", [ ("claude-sonnet-4-5-20250929", "anthropic"), + ("azure-ai-claude-opus-4.5", "azure_ai"), + ("vertex-ai-claude-opus-4-6", "vertex_ai"), ], ) async def test_anthropic_messages_with_all_beta_headers(model_name, provider_name): diff --git a/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml b/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml index ae1514e4e42..e9f7342df8e 100644 --- a/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml +++ b/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml @@ -34,6 +34,12 @@ model_list: litellm_params: model: "bedrock/converse/us.anthropic.claude-sonnet-4-5-20250929-v1:0" aws_region_name: "us-east-1" + + # Azure AI models + - model_name: azure-ai-claude-opus-4.5 + litellm_params: + model: "azure_ai/claude-opus-4.5" + api_key: os.environ/AZURE_AI_API_KEY # Vertex AI models - model_name: vertex-ai-claude-opus-4-6 From a857337152f5c5b3383e2365967f1c381566da49 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Mon, 16 Feb 2026 15:55:23 +0530 Subject: [PATCH 2/3] Add databricks as anthropic model provider --- litellm/anthropic_beta_headers_config.json | 30 ++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/litellm/anthropic_beta_headers_config.json b/litellm/anthropic_beta_headers_config.json index 5dd8536f4c0..a06c7173ea3 100644 --- a/litellm/anthropic_beta_headers_config.json +++ b/litellm/anthropic_beta_headers_config.json @@ -148,5 +148,35 @@ "tool-search-tool-2025-10-19": "tool-search-tool-2025-10-19", "web-fetch-2025-09-10": null, "web-search-2025-03-05": "web-search-2025-03-05" + }, + "databricks": { + "advanced-tool-use-2025-11-20": "advanced-tool-use-2025-11-20", + "bash_20241022": null, + "bash_20250124": null, + "code-execution-2025-08-25": "code-execution-2025-08-25", + "compact-2026-01-12": "compact-2026-01-12", + "computer-use-2025-01-24": "computer-use-2025-01-24", + "computer-use-2025-11-24": "computer-use-2025-11-24", + "context-1m-2025-08-07": "context-1m-2025-08-07", + "context-management-2025-06-27": "context-management-2025-06-27", + "effort-2025-11-24": "effort-2025-11-24", + "fast-mode-2026-02-01": "fast-mode-2026-02-01", + "files-api-2025-04-14": "files-api-2025-04-14", + "structured-output-2024-03-01": null, + "fine-grained-tool-streaming-2025-05-14": "fine-grained-tool-streaming-2025-05-14", + "interleaved-thinking-2025-05-14": "interleaved-thinking-2025-05-14", + "mcp-client-2025-11-20": "mcp-client-2025-11-20", + "mcp-client-2025-04-04": "mcp-client-2025-04-04", + "mcp-servers-2025-12-04": null, + "oauth-2025-04-20": "oauth-2025-04-20", + "output-128k-2025-02-19": "output-128k-2025-02-19", + "prompt-caching-scope-2026-01-05": "prompt-caching-scope-2026-01-05", + "skills-2025-10-02": "skills-2025-10-02", + "structured-outputs-2025-11-13": "structured-outputs-2025-11-13", + "text_editor_20241022": null, + "text_editor_20250124": null, + "token-efficient-tools-2025-02-19": "token-efficient-tools-2025-02-19", + "web-fetch-2025-09-10": "web-fetch-2025-09-10", + "web-search-2025-03-05": "web-search-2025-03-05" } } \ No newline at end of file From 181a1c3a897ca44a561aca9219acdc11dee71d92 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Tue, 17 Feb 2026 21:09:01 +0530 Subject: [PATCH 3/3] Fix test conifg --- tests/proxy_e2e_anthropic_messages_tests/test_config.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml b/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml index e9f7342df8e..fbbb6d4114c 100644 --- a/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml +++ b/tests/proxy_e2e_anthropic_messages_tests/test_config.yaml @@ -40,6 +40,7 @@ model_list: litellm_params: model: "azure_ai/claude-opus-4.5" api_key: os.environ/AZURE_AI_API_KEY + api_base: os.environ/AZURE_AI_API_BASE # Vertex AI models - model_name: vertex-ai-claude-opus-4-6