From bd59f18809e1aac7c913e38a3e0411bd130336f7 Mon Sep 17 00:00:00 2001
From: Krish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 29 Nov 2024 15:32:54 +0530
Subject: [PATCH] fix(key_management_endpoints.py): support 'tags' param on
 `/key/update` (#6945)

---
 .../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 d95cfad24..27d1ec0a4 100644
--- a/litellm/proxy/management_endpoints/key_management_endpoints.py
+++ b/litellm/proxy/management_endpoints/key_management_endpoints.py
@@ -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