From 36587c55a0c875051127dd94830cab0675e4edef Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 24 Jul 2025 14:46:10 -0700 Subject: [PATCH] fix auto router --- tests/local_testing/test_router_auto_router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_router_auto_router.py b/tests/local_testing/test_router_auto_router.py index 71d2e2f7378..f70978da884 100644 --- a/tests/local_testing/test_router_auto_router.py +++ b/tests/local_testing/test_router_auto_router.py @@ -48,7 +48,7 @@ router = Router( "litellm_params": { "model": "auto_router/auto_router_1", "auto_router_config_path": "auto_router/router.json", - "auto_router_default_model": "openai/gpt-4o-mini", + "auto_router_default_model": "gpt-4o-mini", "auto_router_embedding_model": "custom-text-embedding-model", }, }, @@ -57,7 +57,7 @@ router = Router( "litellm_params": { "model": "auto_router/auto_router_2", "auto_router_config_path": "auto_router/router_2.json", - "auto_router_default_model": "openai/gpt-4o-mini", + "auto_router_default_model": "gpt-4o-mini", "auto_router_embedding_model": "custom-text-embedding-model-2", }, },