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

This commit is contained in:
Krish Dholakia 2024-11-29 15:32:54 +05:30 committed by GitHub
parent 05f810922c
commit bd59f18809
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -497,6 +497,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