scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix(tests): read CI_CD_DEFAULT_ANTHROPIC_MODEL env var in test_router

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-21 10:44:52 -08:00
parent e9a7b694b5
commit 020c44a15b
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_router.py
View file

@ -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,
},
}