fix(router.py): fix get_router_model_info for azure models

This commit is contained in:
Krrish Dholakia 2024-06-28 22:13:29 -07:00
parent 9556bfda81
commit c9a424d28d
3 changed files with 5 additions and 4 deletions

View file

@ -812,6 +812,7 @@ def test_router_context_window_check_pre_call_check():
"base_model": "azure/gpt-35-turbo",
"mock_response": "Hello world 1!",
},
"model_info": {"base_model": "azure/gpt-35-turbo"},
},
{
"model_name": "gpt-3.5-turbo", # openai model name