From 75cbbea0719bf646a9c042d89af88f224b08fdbe Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 26 Aug 2024 18:54:50 -0700 Subject: [PATCH] enforce regenerating keys in enterprise tier --- .../proxy/management_endpoints/key_management_endpoints.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index 4688445fee..9bb07cfee3 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -993,6 +993,11 @@ async def regenerate_key_fn( Endpoint for regenerating a key """ + if premium_user is not True: + raise ValueError( + f"Regenerating Virtual Keys is an Enterprise feature, {CommonProxyErrors.not_premium_user.value}" + ) + # Check if key exists, raise exception if key is not in the DB ### 1. Create New copy that is duplicate of existing key