phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix premium user check for tags on keys (#5826)

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-21 16:56:33 -07:00 • committed by GitHub
parent d100b32573
commit 39e872c7eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -209,9 +209,9 @@ async def generate_key_fn(
if "tags" in data_json: if "tags" in data_json:
from litellm.proxy.proxy_server import premium_user from litellm.proxy.proxy_server import premium_user
if premium_user is not True: if premium_user is not True and data_json["tags"] is not None:
raise ValueError( raise ValueError(
f"Only premium users can add tags to teams. {CommonProxyErrors.not_premium_user.value}" f"Only premium users can add tags to keys. {CommonProxyErrors.not_premium_user.value}"
) )
if data_json["metadata"] is None: if data_json["metadata"] is None: