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

update test auto_router_embedding_model

Browse source
This commit is contained in:
Ishaan Jaff 2025-07-24 14:15:32 -07:00
parent 328ab65888
commit 3091d73936
1 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/local_testing/test_router_auto_router.py
View file

@ -14,6 +14,20 @@ from litellm import Router
router = Router(
model_list=[
{
"model_name": "custom-text-embedding-model",
"litellm_params": {
"model": "text-embedding-3-large",
"api_key": os.getenv("OPENAI_API_KEY"),
},
},
{
"model_name": "custom-text-embedding-model-2",
"litellm_params": {
"model": "text-embedding-3-large",
"api_key": os.getenv("OPENAI_API_KEY"),
},
},
{
"model_name": "gpt-4.1",
"litellm_params": {
@ -33,14 +47,18 @@ router = Router(
"model_name": "auto_router1",
"litellm_params": {
"model": "auto_router/auto_router_1",
"router_config_path": "auto_router/router.json",
"auto_router_config_path": "auto_router/router.json",
"auto_router_default_model": "openai/gpt-4o-mini",
"auto_router_embedding_model": "custom-text-embedding-model",
},
},
{
"model_name": "auto_router_2",
"litellm_params": {
"model": "auto_router/auto_router_2",
"router_config_path": "auto_router/router_2.json",
"auto_router_config_path": "auto_router/router_2.json",
"auto_router_default_model": "openai/gpt-4o-mini",
"auto_router_embedding_model": "custom-text-embedding-model-2",
},
},
],