scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-06 07:48:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 112

Merge f752e2aa96 into d884167bf2

Browse source
This commit is contained in:
Yash Rathod 2026-10-29 13:13:00 -07:00 • committed by GitHub
commit 830a7f9b66
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -1668,16 +1668,23 @@ async def _common_key_generation_helper(
if "tags" in data_json:
from litellm.proxy.proxy_server import premium_user
if premium_user is not True and data_json["tags"] is not None:
raise ValueError(f"Only premium users can add tags to keys. {CommonProxyErrors.not_premium_user.value}")
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}"
)
_metadata: Final = data_json.get("metadata")
if not _metadata:
data_json["metadata"] = {"tags": data_json["tags"]}
if tags is None:
if isinstance(_metadata, dict):
_metadata.pop("tags", None)
data_json.pop("tags", None)
else:
data_json["metadata"]["tags"] = data_json["tags"]
data_json.pop("tags")
if not isinstance(_metadata, dict):
data_json["metadata"] = {"tags": tags}
else:
data_json["metadata"]["tags"] = tags
data_json.pop("tags", None)
# Validate MCP servers in object_permission are within team scope
_is_proxy_admin_caller: Final = user_api_key_dict.user_role == LitellmUserRoles.PROXY_ADMIN.value