phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add_team_based_tags_to_metadata

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-29 13:06:03 -07:00
parent 3bed17fc95
commit bc06920f04
1 changed files with 24 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/litellm_pre_call_utils.py
View file

@ -379,6 +379,12 @@ async def add_litellm_data_to_request(
# unpack callback_vars in data # unpack callback_vars in data
for k, v in callback_settings_obj.callback_vars.items(): for k, v in callback_settings_obj.callback_vars.items():
data[k] = v data[k] = v
# Team based tags
add_team_based_tags_to_metadata(
data=data,
_metadata_variable_name=_metadata_variable_name,
user_api_key_dict=user_api_key_dict,
)
# Guardrails # Guardrails
move_guardrails_to_metadata( move_guardrails_to_metadata(
@ -390,6 +396,24 @@ async def add_litellm_data_to_request(
return data return data
def add_team_based_tags_to_metadata(
data: dict,
_metadata_variable_name: str,
user_api_key_dict: UserAPIKeyAuth,
):
from litellm.proxy.proxy_server import premium_user
if premium_user is True:
if (
user_api_key_dict.team_metadata is not None
and "tags" in user_api_key_dict.team_metadata
):
_team_tags = user_api_key_dict.team_metadata["tags"]
_tags_in_metadata = data[_metadata_variable_name].get("tags", [])
_tags_in_metadata.extend(_team_tags)
data[_metadata_variable_name]["tags"] = _tags_in_metadata
def move_guardrails_to_metadata( def move_guardrails_to_metadata(
data: dict, data: dict,
_metadata_variable_name: str, _metadata_variable_name: str,