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

test_router_auto_router

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-24 16:20:00 -07:00
parent cff82cce6a
commit b8386ec01e
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/local_testing/test_router_auto_router.py
View file

@ -15,7 +15,16 @@ from litellm import Router
current_path = os.path.dirname(os.path.abspath(__file__))
router_json_path = os.path.join(current_path, "auto_router", "router.json")
router = Router(
@pytest.mark.asyncio
async def test_router_auto_router():
"""
Simple e2e test to validate we get an llm response from the auto router
"""
import litellm
litellm._turn_on_debug()
router = Router(
model_list=[
{
"model_name": "custom-text-embedding-model",
@ -68,14 +77,6 @@ router = Router(
)
@pytest.mark.asyncio
async def test_router_auto_router():
"""
Simple e2e test to validate we get an llm response from the auto router
"""
import litellm
litellm._turn_on_debug()
# this goes to gpt-4.1
# these are the utterances in the router.json file
response = await router.acompletion(