From ae13dc6f6bba218ec13ebdf3fd08b8ec3b11b564 Mon Sep 17 00:00:00 2001 From: Yash Rathod <52967673+yash30389@users.noreply.github.com> Date: Wed, 28 Oct 2026 13:58:59 +0530 Subject: [PATCH] Refactor tag validation for premium users --- .../proxy/management_endpoints/key_management_endpoints.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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: