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

fix(key_management_endpoints.py): handle prepare metadata

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-30 15:59:39 -08:00
parent e93fc7c91a
commit 680701850f
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -459,7 +459,10 @@ def prepare_metadata_fields(
Check LiteLLM_ManagementEndpoint_MetadataFields (proxy/_types.py) for fields that are allowed to be updated Check LiteLLM_ManagementEndpoint_MetadataFields (proxy/_types.py) for fields that are allowed to be updated
""" """
non_default_values.setdefault("metadata", {}) non_default_values.setdefault("metadata", {})
non_default_values["metadata"].update(existing_metadata) if non_default_values["metadata"] is None:
non_default_values["metadata"] = existing_metadata
else:
non_default_values["metadata"].update(existing_metadata)
data_json = data.model_dump(exclude_unset=True) data_json = data.model_dump(exclude_unset=True)
try: try:
@ -474,7 +477,7 @@ def prepare_metadata_fields(
non_default_values["metadata"][k] = [] non_default_values["metadata"][k] = []
seen = set(non_default_values["metadata"][k]) seen = set(non_default_values["metadata"][k])
non_default_values["metadata"][k].extend( non_default_values["metadata"][k].extend(
x for x in v if x not in seen and not seen.add(x) x for x in v if x not in seen and not seen.add(x) # type: ignore
) # prevent duplicates from being added + maintain initial order ) # prevent duplicates from being added + maintain initial order
except Exception as e: except Exception as e: