scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

Merge 103e14e23a into e768ad55ce

Browse source
This commit is contained in:
Kartik Chilkoti 2026-10-03 08:12:29 -04:00 • committed by GitHub
commit 97157cc895
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 26 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/test_router_fallback_chain.py Normal file
View file

@ -0,0 +1,26 @@
import pytest
from litellm import Router
def test_router_fallbacks_initialization():
fallbacks = [
{"gpt-4": ["azure/gpt-4-east", "bedrock/claude-3-5-sonnet"]}
]
router = Router(
model_list=[
{"model_name": "gpt-4", "litellm_params": {"model": "azure/gpt-4-east"}},
{"model_name": "azure/gpt-4-east", "litellm_params": {"model": "azure/gpt-4-east"}},
{"model_name": "bedrock/claude-3-5-sonnet", "litellm_params": {"model": "bedrock/claude-3-5-sonnet"}},
],
fallbacks=fallbacks,
)
assert router.fallbacks == fallbacks
assert len(router.fallbacks) == 1
assert "gpt-4" in router.fallbacks[0]
def test_router_empty_fallbacks():
router = Router(
model_list=[
{"model_name": "gpt-3.5-turbo", "litellm_params": {"model": "gpt-3.5-turbo"}}
]
)
assert router.fallbacks is None or router.fallbacks == []