(fix) patch dynamoDB tea_model_alias bug

This commit is contained in:
ishaan-jaff 2024-03-12 11:17:12 -07:00
parent 5172fb1de9
commit 6f069d26f3

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,