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

2024-11-28 03:38:28 +05:30
1 changed files with 6 additions and 0 deletions
--- a/litellm/proxy/management_endpoints/key_management_endpoints.py
+++ b/litellm/proxy/management_endpoints/key_management_endpoints.py
@ -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