diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index 25db059e4b7..1dad3717cc4 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -1666,12 +1666,10 @@ async def _common_key_generation_helper( # Set tags on the new key if "tags" in data_json: from litellm.proxy.proxy_server import premium_user - + tags = data_json.get("tags") if premium_user is not True and isinstance(tags, list) and len(tags) > 0: - raise ValueError( - f"Only premium users can add tags to keys. {CommonProxyErrors.not_premium_user.value}" - ) + raise ValueError(f"Only premium users can add tags to keys. {CommonProxyErrors.not_premium_user.value}") _metadata: Final = data_json.get("metadata") if tags is None: