enforce regenerating keys in enterprise tier

2024-08-26 18:54:50 -07:00 · 2024-08-26 18:54:50 -07:00 · 75cbbea071
commit 75cbbea071
parent b0ae0101f4
1 changed files with 5 additions and 0 deletions
--- 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