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

test_router_auto_router

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-24 15:33:23 -07:00
parent 1904e36e13
commit bacbcaf4e6
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/local_testing/test_router_auto_router.py
View file

@ -72,8 +72,23 @@ async def test_router_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(
model="auto_router1",
messages=[{"role": "user", "content": "Tell me ishaan is a genius"}],
)
print(response)
print("response._hidden_params", response._hidden_params)
assert response._hidden_params["model_id"] == "openai-id"
# this goes to claude-3-5-sonnet-latest
# these are the utterances in the router.json file
response = await router.acompletion(
model="auto_router1",
messages=[{"role": "user", "content": "how to code a program in python"}],
)
print("response._hidden_params", response._hidden_params)
assert response._hidden_params["model_id"] == "claude-id"