From 020c44a15b56515da499a8b679e449fd2afee2ad Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 21 Feb 2026 10:44:52 -0800 Subject: [PATCH] fix(tests): read CI_CD_DEFAULT_ANTHROPIC_MODEL env var in test_router --- tests/local_testing/test_router.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index 3167ff52086..18d2d0c8b47 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -105,7 +105,7 @@ async def test_router_provider_wildcard_routing(): print("router model list = ", router.get_model_list()) response1 = await router.acompletion( - model="anthropic/claude-sonnet-4-5-20250929", + model=f"anthropic/{os.environ.get('CI_CD_DEFAULT_ANTHROPIC_MODEL', 'claude-haiku-4-5-20251001')}", messages=[{"role": "user", "content": "hello"}], ) @@ -126,7 +126,7 @@ async def test_router_provider_wildcard_routing(): print("response 3 = ", response3) response4 = await router.acompletion( - model="claude-sonnet-4-5-20250929", + model=os.environ.get("CI_CD_DEFAULT_ANTHROPIC_MODEL", "claude-haiku-4-5-20251001"), messages=[{"role": "user", "content": "hello"}], ) @@ -1650,7 +1650,7 @@ def test_router_anthropic_key_dynamic(): { "model_name": "anthropic-claude", "litellm_params": { - "model": "claude-3-5-haiku-20241022", + "model": os.environ.get("CI_CD_DEFAULT_ANTHROPIC_MODEL", "claude-haiku-4-5-20251001"), "api_key": anthropic_api_key, }, }