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

fix(litellm_pre_call_utils.py): don't override k-v pair sent in spend_logs_metadata by user

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-23 07:10:03 -07:00
parent 4ac78a0765
commit afb00a27cb
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/litellm_pre_call_utils.py
View file

@ -248,9 +248,11 @@ async def add_litellm_data_to_request(
if "spend_logs_metadata" in data[_metadata_variable_name] and isinstance(
data[_metadata_variable_name]["spend_logs_metadata"], dict
):
data[_metadata_variable_name]["spend_logs_metadata"].update(
key_metadata["spend_logs_metadata"]
)
for key, value in key_metadata["spend_logs_metadata"].items():
if (
key not in data[_metadata_variable_name]["spend_logs_metadata"]
): # don't override k-v pair sent by request (user request)
data[_metadata_variable_name]["spend_logs_metadata"][key] = value
else:
data[_metadata_variable_name]["spend_logs_metadata"] = key_metadata[
"spend_logs_metadata"
@ -271,9 +273,11 @@ async def add_litellm_data_to_request(
if "spend_logs_metadata" in data[_metadata_variable_name] and isinstance(
data[_metadata_variable_name]["spend_logs_metadata"], dict
):
data[_metadata_variable_name]["spend_logs_metadata"].update(
team_metadata["spend_logs_metadata"]
)
for key, value in team_metadata["spend_logs_metadata"].items():
if (
key not in data[_metadata_variable_name]["spend_logs_metadata"]
): # don't override k-v pair sent by request (user request)
data[_metadata_variable_name]["spend_logs_metadata"][key] = value
else:
data[_metadata_variable_name]["spend_logs_metadata"] = team_metadata[
"spend_logs_metadata"