Merge pull request #4610 from andresrguzman/litellm_fix_add_new_member

fix(utils.py): change update to upsert
2026-08-05 12:23:28 +00:00 · 2024-07-08 14:57:33 -07:00
parent cb52c59481 a650160390
commit 1aba2c5bc0
1 changed files with 10 additions and 2 deletions
@@ -29,9 +29,17 @@ async def add_new_member(
    """
    ## ADD TEAM ID, to USER TABLE IF NEW ##
    if new_member.user_id is not None:
-        await prisma_client.db.litellm_usertable.update(
+        await prisma_client.db.litellm_usertable.upsert(
            where={"user_id": new_member.user_id},
-            data={"teams": {"push": [team_id]}},
+            data={
+                "update": {
+                    "teams": {"push": [team_id]}
+                },
+                "create": {
+                    "user_id": new_member.user_id, 
+                    "teams": [team_id]
+                }
+            },
        )
    elif new_member.user_email is not None:
        user_data = {"user_id": str(uuid.uuid4()), "user_email": new_member.user_email}