fix(router.py): support adding models across multiple orgs, with 1 model definition

This commit is contained in:
Krrish Dholakia 2024-06-18 19:57:10 -07:00
parent 14b66c3daa
commit 164dccbbd4
3 changed files with 28 additions and 1 deletions

View file

@ -1970,7 +1970,7 @@ class ProxyConfig:
router = litellm.Router(
**router_params, assistants_config=assistants_config
) # type:ignore
return router, model_list, general_settings
return router, router.get_model_list(), general_settings
def get_model_info_with_id(self, model, db_model=False) -> RouterModelInfo:
"""