From e7d19fa0cc6de9d2032598ad01b7172007c93609 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 28 Nov 2024 03:38:28 +0530
Subject: [PATCH] fix(key_management_endpoints.py): support 'tags' param on
 `/key/update`

---
 .../proxy/management_endpoints/key_management_endpoints.py  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py
index ba6df9a2e..4743db34d 100644
--- 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