From fc5e947a4dab4114d2daa4dfd55c994228434d4b Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Mon, 16 Feb 2026 15:57:14 +0530 Subject: [PATCH] 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 cf688d7dea..14a301add5 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 ae1514e4e4..e9f7342df8 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