From aa2a60b24c8188ff52a2bfdee63d6d6b4d06d998 Mon Sep 17 00:00:00 2001 From: mateo Date: Fri, 7 Aug 2026 20:20:13 +0000 Subject: [PATCH] fix(proxy): broadcast the real cache key when deleting keys Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../management_endpoints/key_management_endpoints.py | 5 +---- .../management_endpoints/test_key_management_endpoints.py | 8 ++------ 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index 0aa1157c2df..a7b2b2a997d 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -4145,11 +4145,8 @@ async def delete_verification_tokens( raise e for key in tokens: - user_api_key_cache.delete_cache(key) - # remove hash token from cache - hashed_token = hash_token(cast(str, key)) await _delete_cache_key_object( - hashed_token=hashed_token, + hashed_token=cast(str, key), user_api_key_cache=user_api_key_cache, proxy_logging_obj=proxy_logging_obj, ) diff --git a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py index 940a333779c..bfa691bed31 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py @@ -15464,11 +15464,7 @@ async def test_delete_verification_tokens_broadcasts_key_eviction(monkeypatch): monkeypatch.setattr( "litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed", - lambda token: token, - ) - monkeypatch.setattr( - "litellm.proxy.management_endpoints.key_management_endpoints.hash_token", - lambda token: token, + lambda token: "hashed-token-1", ) monkeypatch.setattr("litellm.proxy.proxy_server.prisma_client", mock_prisma_client) monkeypatch.setattr( @@ -15477,7 +15473,7 @@ async def test_delete_verification_tokens_broadcasts_key_eviction(monkeypatch): ) await delete_verification_tokens( - tokens=["hashed-token-1"], + tokens=["sk-token-1"], user_api_key_cache=MagicMock(), user_api_key_dict=UserAPIKeyAuth( user_id="admin-user",