From b59e472fcdf1aa02e883ff6c80fafe14a2b22169 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 0bb1271673..930f81ef53 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,