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 7e70d61d2d
commit 49eb647d4d

View file

@ -497,6 +497,12 @@ def prepare_key_update_data(
_metadata["model_rpm_limit"].update(data.model_rpm_limit) _metadata["model_rpm_limit"].update(data.model_rpm_limit)
non_default_values["metadata"] = _metadata 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: if data.guardrails:
_metadata["guardrails"] = data.guardrails _metadata["guardrails"] = data.guardrails
non_default_values["metadata"] = _metadata non_default_values["metadata"] = _metadata