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

fix(key_management_endpoints.py): support 'tags' param on /key/update

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-28 03:38:28 +05:30
parent 21156ff5d0
commit e7d19fa0cc
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -491,6 +491,12 @@ def prepare_key_update_data(
_metadata["model_rpm_limit"].update(data.model_rpm_limit)
non_default_values["metadata"] = _metadata
if data.tags:
if "tags" not in _metadata:
_metadata["tags"] = []
_metadata["tags"].extend(data.tags)
non_default_values["metadata"] = _metadata
if data.guardrails:
_metadata["guardrails"] = data.guardrails
non_default_values["metadata"] = _metadata