From 1aeccf3f0afa582899ccac1c294e1b21ff5a62c9 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 2 Apr 2024 20:50:47 -0700
Subject: [PATCH] proxy test all-tea-models

---
 litellm/proxy/proxy_server.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 24a0f5fab1e..0923cfa7e75 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -639,9 +639,10 @@ async def user_api_key_auth(
                 pass
             elif (
                 isinstance(valid_token.models, list)
-                and "all-proxy-models" in valid_token.models
+                and "all-team-models" in valid_token.models
             ):
-                # Admin UI - Special alias to allow `all_models`
+                # Do not do any validation at this step
+                # the validation will occur when checking the team has access to this model
                 pass
             else:
                 try: