[Fix] Tag Based Routing not work with wildcard routing (#5805)

* allow using tag routing for free

* only enforce tags for teams / keys
This commit is contained in:
Ishaan Jaff 2024-09-20 14:05:56 -07:00 committed by GitHub
parent 3933fba41f
commit 036fce8f18
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 21 additions and 9 deletions

View file

@ -207,6 +207,13 @@ async def generate_key_fn(
# Set tags on the new key
if "tags" in data_json:
from litellm.proxy.proxy_server import premium_user
if premium_user is not True:
raise ValueError(
f"Only premium users can add tags to teams. {CommonProxyErrors.not_premium_user.value}"
)
if data_json["metadata"] is None:
data_json["metadata"] = {"tags": data_json["tags"]}
else: