diff --git a/tests/local_testing/test_router_retries.py b/tests/local_testing/test_router_retries.py index 1d72d5914c5..302ab50bf85 100644 --- a/tests/local_testing/test_router_retries.py +++ b/tests/local_testing/test_router_retries.py @@ -3,6 +3,8 @@ import asyncio import os + +DEFAULT_ANTHROPIC_MODEL = os.getenv("CI_CD_DEFAULT_ANTHROPIC_MODEL", "claude-haiku-4-5-20251001") import sys import time import traceback @@ -859,7 +861,7 @@ async def test_router_timeout_model_specific_and_global(): { "model_name": "anthropic-claude", "litellm_params": { - "model": "anthropic/claude-sonnet-4-5-20250929", + "model": f"anthropic/{DEFAULT_ANTHROPIC_MODEL}", "timeout": 1, }, }