From 93d6aae4a36049010f668f4b91f0bcaf9e13d3f5 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Tue, 27 Jan 2026 14:06:54 -0800
Subject: [PATCH] mypy linting

---
 .../management_endpoints/key_management_endpoints.py      | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py
index f9fb0e5f49e..380e8bddc99 100644
--- a/litellm/proxy/management_endpoints/key_management_endpoints.py
+++ b/litellm/proxy/management_endpoints/key_management_endpoints.py
@@ -1568,7 +1568,7 @@ async def _process_single_key_update(
             check_db_only=True,
         )
         
-        if team_obj is not None:
+        if team_obj is not None and prisma_client is not None:
             await _check_team_key_limits(
                 team_table=team_obj,
                 data=update_key_request,
@@ -1606,6 +1606,12 @@ async def _process_single_key_update(
     )
     
     # Update key in database
+    if prisma_client is None:
+        raise HTTPException(
+            status_code=500,
+            detail={"error": "Database not connected"},
+        )
+    
     _data = {**non_default_values, "token": key_update_item.key}
     response = await prisma_client.update_data(
         token=key_update_item.key, data=_data