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

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

Browse source 
* 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
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/litellm_pre_call_utils.py
View file

@ -351,13 +351,6 @@ async def add_litellm_data_to_request(
# Enterprise Only - Check if using tag based routing # Enterprise Only - Check if using tag based routing
if llm_router and llm_router.enable_tag_filtering is True: if llm_router and llm_router.enable_tag_filtering is True:
if premium_user is not True:
verbose_proxy_logger.warning(
"router.enable_tag_filtering is on %s \n switched off router.enable_tag_filtering",
CommonProxyErrors.not_premium_user.value,
)
llm_router.enable_tag_filtering = False
else:
if "tags" in data: if "tags" in data:
data[_metadata_variable_name]["tags"] = data["tags"] data[_metadata_variable_name]["tags"] = data["tags"]

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

@ -207,6 +207,13 @@ async def generate_key_fn(
# Set tags on the new key # Set tags on the new key
if "tags" in data_json: 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: if data_json["metadata"] is None:
data_json["metadata"] = {"tags": data_json["tags"]} data_json["metadata"] = {"tags": data_json["tags"]}
else: else:

12
litellm/proxy/management_endpoints/team_endpoints.py
View file

@ -232,6 +232,12 @@ async def new_team(
# Set tags on the new team # Set tags on the new team
if data.tags is not None: if data.tags is not None:
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 complete_team_data.metadata is None: if complete_team_data.metadata is None:
complete_team_data.metadata = {"tags": data.tags} complete_team_data.metadata = {"tags": data.tags}
else: else:
@ -381,6 +387,12 @@ async def update_team(
# check if user is trying to update tags for team # check if user is trying to update tags for team
if "tags" in updated_kv and updated_kv["tags"] is not None: if "tags" in updated_kv and updated_kv["tags"] is not None:
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}"
)
# remove tags from updated_kv # remove tags from updated_kv
_tags = updated_kv.pop("tags") _tags = updated_kv.pop("tags")
if "metadata" in updated_kv and updated_kv["metadata"] is not None: if "metadata" in updated_kv and updated_kv["metadata"] is not None: