scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 87

fix(tests): use CI_CD_DEFAULT_ANTHROPIC_MODEL env var in test_router

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-21 10:41:39 -08:00
parent 3b9af765dd
commit c20c42471e
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_router.py
View file

@ -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
@ -105,7 +107,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/{DEFAULT_ANTHROPIC_MODEL}",
messages=[{"role": "user", "content": "hello"}],
)
@ -126,7 +128,7 @@ async def test_router_provider_wildcard_routing():
print("response 3 = ", response3)
response4 = await router.acompletion(
model="claude-sonnet-4-5-20250929",
model=DEFAULT_ANTHROPIC_MODEL,
messages=[{"role": "user", "content": "hello"}],
)
@ -1650,7 +1652,7 @@ def test_router_anthropic_key_dynamic():
{
"model_name": "anthropic-claude",
"litellm_params": {
"model": "claude-3-5-haiku-20241022",
"model": DEFAULT_ANTHROPIC_MODEL,
"api_key": anthropic_api_key,
},
}