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 auto router

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-24 16:16:14 -07:00
parent 0aaee5e1ec
commit cff82cce6a
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/local_testing/test_router_auto_router.py
View file

@ -12,6 +12,9 @@ sys.path.insert(
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(
model_list=[
{
@ -47,7 +50,7 @@ router = Router(
"model_name": "auto_router1",
"litellm_params": {
"model": "auto_router/auto_router_1",
"auto_router_config_path": "auto_router/router.json",
"auto_router_config_path": router_json_path,
"auto_router_default_model": "gpt-4o-mini",
"auto_router_embedding_model": "custom-text-embedding-model",
},
@ -56,7 +59,7 @@ router = Router(
"model_name": "auto_router_2",
"litellm_params": {
"model": "auto_router/auto_router_2",
"auto_router_config_path": "auto_router/router.json",
"auto_router_config_path": router_json_path,
"auto_router_default_model": "gpt-4o-mini",
"auto_router_embedding_model": "custom-text-embedding-model-2",
},