From 11ff05e99dfffdaa8e99a30180f44cc8aa3fdd18 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Thu, 18 Dec 2025 20:34:33 -0800 Subject: [PATCH] Adding deleted key capture to team member delete --- .../management_endpoints/team_endpoints.py | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/litellm/proxy/management_endpoints/team_endpoints.py b/litellm/proxy/management_endpoints/team_endpoints.py index 3a9af9d54f3..239586b5740 100644 --- a/litellm/proxy/management_endpoints/team_endpoints.py +++ b/litellm/proxy/management_endpoints/team_endpoints.py @@ -1974,6 +1974,28 @@ async def team_member_delete( ## DELETE KEYS CREATED BY USER FOR THIS TEAM if user_ids_to_delete: + from litellm.proxy.management_endpoints.key_management_endpoints import ( + _persist_deleted_verification_tokens, + ) + + # Fetch keys before deletion to persist them + keys_to_delete: List[LiteLLM_VerificationToken] = ( + await prisma_client.db.litellm_verificationtoken.find_many( + where={ + "user_id": {"in": list(user_ids_to_delete)}, + "team_id": data.team_id, + } + ) + ) + + if keys_to_delete: + await _persist_deleted_verification_tokens( + keys=keys_to_delete, + prisma_client=prisma_client, + user_api_key_dict=user_api_key_dict, + litellm_changed_by=None, + ) + await prisma_client.db.litellm_verificationtoken.delete_many( where={ "user_id": {"in": list(user_ids_to_delete)},