fix(proxy_server.py): pass model access groups to get_key/get_team mo… (#7281)

* fix(proxy_server.py): pass model access groups to get_key/get_team models

allows end user to see actual models they have access to, instead of default models

* fix(auth_checks.py): fix linting errors

* fix: fix linting errors
This commit is contained in:
Krish Dholakia 2024-12-18 09:33:33 -08:00 committed by GitHub
parent c7ff5a53d7
commit e7918f097b
9 changed files with 118 additions and 20 deletions

View file

@ -4653,7 +4653,9 @@ class Router:
return returned_models
return None
def get_model_access_groups(self, model_name: Optional[str] = None):
def get_model_access_groups(
self, model_name: Optional[str] = None
) -> Dict[str, List[str]]:
"""
If model_name is provided, only return access groups for that model.
"""