From 5019e0322f4e5596727f364200504e48ddd2b8e1 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 24 Aug 2024 17:50:40 -0700 Subject: [PATCH] fix(utils.py): fix linting errors --- litellm/proxy/management_helpers/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/management_helpers/utils.py b/litellm/proxy/management_helpers/utils.py index a361a442fcf..319569ec699 100644 --- a/litellm/proxy/management_helpers/utils.py +++ b/litellm/proxy/management_helpers/utils.py @@ -99,7 +99,9 @@ async def add_new_member( ): new_user_defaults["teams"] = [team_id] _returned_user = await prisma_client.insert_data(data=new_user_defaults, table_name="user") # type: ignore - returned_user = LiteLLM_UserTable(**_returned_user.model_dump()) + + if _returned_user is not None: + returned_user = LiteLLM_UserTable(**_returned_user.model_dump()) elif len(existing_user_row) == 1: user_info = existing_user_row[0] _returned_user = await prisma_client.db.litellm_usertable.update(