fix(utils.py): fix linting errors

2024-08-24 17:50:40 -07:00 · 2024-08-24 17:50:40 -07:00 · 5019e0322f
commit 5019e0322f
parent b2388e3409
1 changed files with 3 additions and 1 deletions
--- 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(