Merge pull request #2478 from BerriAI/litellm_fix_dynamo_team_alias

(fix) patch dynamoDB team_model_alias bug
This commit is contained in:
Ishaan Jaff 2024-03-12 12:34:53 -07:00 • committed by GitHub
commit b2d5f775ca
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -417,7 +417,10 @@ async def user_api_key_auth(
# Check 1. If token can call model
_model_alias_map = {}
if valid_token.team_model_aliases is not None:
if (
hasattr(valid_token, "team_model_aliases")
and valid_token.team_model_aliases is not None
):
_model_alias_map = {
**valid_token.aliases,
**valid_token.team_model_aliases,