From c9413dfded0efcb135f55645abc2a9fe8cf5c8ce Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sun, 25 Feb 2024 00:49:31 -0800
Subject: [PATCH] fix: bug fixes

---
 litellm/proxy/proxy_server.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 6dea62763..7dec12912 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -4526,7 +4526,7 @@ async def update_team(
                         "teams": [team_row["team_id"]],
                         "models": team_row["data"].models,
                     },
-                    update_key_values={
+                    update_key_values_custom_query={
                         "teams": {
                             "push": [team_row["team_id"]],
                         }
@@ -4632,7 +4632,7 @@ async def team_member_add(
         await prisma_client.update_data(
             user_id=new_member.user_id,
             data=user_data,
-            update_key_values={
+            update_key_values_custom_query={
                 "teams": {
                     "push": [team_row["team_id"]],
                 }