fix(router.py): handle edge case where user sets 'model_group' inside 'model_info'

This commit is contained in:
Krrish Dholakia 2025-04-21 13:48:19 -07:00
parent 10257426a2
commit 996154f2dc
2 changed files with 32 additions and 2 deletions

View file

@ -4983,8 +4983,12 @@ class Router:
)
if model_group_info is None:
model_group_info = ModelGroupInfo(
model_group=user_facing_model_group_name, providers=[llm_provider], **model_info # type: ignore
model_group_info = ModelGroupInfo( # type: ignore
**{
"model_group": user_facing_model_group_name,
"providers": [llm_provider],
**model_info,
}
)
else:
# if max_input_tokens > curr