diff --git a/tests/local_testing/test_router_budget_limiter.py b/tests/local_testing/test_router_budget_limiter.py index bda1f648076..00e1a97a234 100644 --- a/tests/local_testing/test_router_budget_limiter.py +++ b/tests/local_testing/test_router_budget_limiter.py @@ -150,7 +150,7 @@ async def test_provider_budgets_e2e_test_expect_to_fail(): response = await router.acompletion( messages=[{"role": "user", "content": "Hello, how are you?"}], - model="anthropic/claude-sonnet-4-5-20250929", + model=f"anthropic/{os.environ.get('CI_CD_DEFAULT_ANTHROPIC_MODEL', 'claude-haiku-4-5-20251001')}", ) print(response) @@ -160,7 +160,7 @@ async def test_provider_budgets_e2e_test_expect_to_fail(): with pytest.raises(Exception, match="Exceeded budget for provider") as exc_info: await router.acompletion( messages=[{"role": "user", "content": "Hello, how are you?"}], - model="anthropic/claude-sonnet-4-5-20250929", + model=f"anthropic/{os.environ.get('CI_CD_DEFAULT_ANTHROPIC_MODEL', 'claude-haiku-4-5-20251001')}", ) await asyncio.sleep(0.5) diff --git a/tests/openai_endpoints_tests/test_e2e_openai_responses_api.py b/tests/openai_endpoints_tests/test_e2e_openai_responses_api.py index be565972b94..81ce75d0ffd 100644 --- a/tests/openai_endpoints_tests/test_e2e_openai_responses_api.py +++ b/tests/openai_endpoints_tests/test_e2e_openai_responses_api.py @@ -1,3 +1,5 @@ +import os + import httpx from openai import OpenAI, BadRequestError, APIStatusError import pytest @@ -124,7 +126,7 @@ def test_bad_request_bad_param_error(): def test_anthropic_with_responses_api(): client = get_test_client() response = client.responses.create( - model="anthropic/claude-sonnet-4-5-20250929", + model=f"anthropic/{os.environ.get('CI_CD_DEFAULT_ANTHROPIC_MODEL', 'claude-haiku-4-5-20251001')}", input="just respond with the word 'ping'", previous_response_id="hi", ) diff --git a/tests/pass_through_tests/test_anthropic_passthrough_basic.py b/tests/pass_through_tests/test_anthropic_passthrough_basic.py index 21e53994dcc..ba66ba83027 100644 --- a/tests/pass_through_tests/test_anthropic_passthrough_basic.py +++ b/tests/pass_through_tests/test_anthropic_passthrough_basic.py @@ -1,3 +1,5 @@ +import os + from base_anthropic_messages_test import BaseAnthropicMessagesTest import anthropic @@ -21,7 +23,7 @@ class TestAnthropicMessagesEndpoint(BaseAnthropicMessagesTest): def test_anthropic_messages_to_wildcard_model(self): client = self.get_client() response = client.messages.create( - model="anthropic/claude-haiku-4-5-20251001", + model=f"anthropic/{os.environ.get('CI_CD_DEFAULT_ANTHROPIC_MODEL', 'claude-haiku-4-5-20251001')}", messages=[{"role": "user", "content": "Hello, world!"}], max_tokens=100, )