diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py
index 8365a1d4818..69e7c7c27d5 100644
--- a/litellm/proxy/management_endpoints/key_management_endpoints.py
+++ b/litellm/proxy/management_endpoints/key_management_endpoints.py
@@ -46,6 +46,7 @@ from litellm.proxy.auth.auth_checks import (
     can_team_access_model,
     get_org_object,
     get_project_object,
+    get_team_membership,
     get_team_object,
 )
 from litellm.proxy.auth.auth_utils import abbreviate_api_key
@@ -642,19 +643,17 @@ async def _common_key_generation_helper(  # noqa: PLR0915
         litellm.team_model_overrides_enabled
         or os.getenv("TEAM_MODEL_OVERRIDES", "").lower() == "true"
     ) and team_table is not None:
-        # Read member models from LiteLLM_TeamMembership table (authoritative source),
-        # NOT from members_with_roles JSON blob which can be stale after /team/member_update.
-        # Note: This is a management endpoint (/key/generate), not the hot auth path.
-        # The hot path (/chat/completions) uses the SQL view join with zero extra queries.
+        # Read member models from LiteLLM_TeamMembership via the standard cached helper
+        # (NOT from members_with_roles JSON blob which can be stale after /team/member_update).
         member_models: List[str] = []
         if data.user_id and prisma_client is not None:
-            _membership = await prisma_client.db.litellm_teammembership.find_unique(
-                where={
-                    "user_id_team_id": {
-                        "user_id": data.user_id,
-                        "team_id": team_table.team_id,
-                    }
-                }
+            from litellm.proxy.proxy_server import user_api_key_cache
+
+            _membership = await get_team_membership(
+                user_id=data.user_id,
+                team_id=team_table.team_id,
+                prisma_client=prisma_client,
+                user_api_key_cache=user_api_key_cache,
             )
             if _membership is not None:
                 member_models = _membership.models or []
diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py
index fcc8d1a0a91..479cfa24bcf 100644
--- a/litellm/proxy/management_endpoints/team_endpoints.py
+++ b/litellm/proxy/management_endpoints/team_endpoints.py
@@ -2530,17 +2530,19 @@ async def team_member_update(  # noqa: PLR0915
 
     ### update team member role
     # Resolve the effective models for this member (from the authoritative
-    # LiteLLM_TeamMembership table) so we can: (a) keep the members_with_roles
-    # JSON in sync, and (b) return the actual stored state in the response.
+    # LiteLLM_TeamMembership table via cached helper) so we can: (a) keep the
+    # members_with_roles JSON in sync, and (b) return the actual stored state
+    # in the response.
     stored_models = data.models
     if stored_models is None:
-        _tm_row = await prisma_client.db.litellm_teammembership.find_unique(
-            where={
-                "user_id_team_id": {
-                    "user_id": received_user_id,
-                    "team_id": data.team_id,
-                }
-            }
+        from litellm.proxy.auth.auth_checks import get_team_membership
+        from litellm.proxy.proxy_server import user_api_key_cache
+
+        _tm_row = await get_team_membership(
+            user_id=received_user_id,
+            team_id=data.team_id,
+            prisma_client=prisma_client,
+            user_api_key_cache=user_api_key_cache,
         )
         stored_models = (_tm_row.models or []) if _tm_row is not None else []