From 6f069d26f3aeba72b583aea27046f660a33cb630 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 12 Mar 2024 11:17:12 -0700 Subject: [PATCH] (fix) patch dynamoDB tea_model_alias bug --- litellm/proxy/proxy_server.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 0bb12716730..930f81ef530 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -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,