From d066e0e152fd53c9ac9d82d3cfe7324d0905a113 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 8 Jun 2024 22:56:56 -0700 Subject: [PATCH] fix(proxy/_types.py): fix validation check --- litellm/proxy/_types.py | 2 ++ litellm/proxy/proxy_server.py | 6 ++---- litellm/tests/test_key_generate_prisma.py | 1 - 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 255ab43d6..f276a8562 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -719,6 +719,8 @@ class Member(LiteLLMBase): @model_validator(mode="before") @classmethod def check_user_info(cls, values): + if not isinstance(values, dict): + raise ValueError("input needs to be a dictionary") if values.get("user_id") is None and values.get("user_email") is None: raise ValueError("Either user id or user email must be provided") return values diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 73b4ac547..cce7861e7 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -10189,10 +10189,8 @@ async def team_member_add( complete_team_data.members_with_roles.extend(new_members) - # ADD MEMBER TO TEAM - _db_team_members = [ - m.model_dump() for m in complete_team_data.members_with_roles - ] + # ADD MEMBER TO TEAM + _db_team_members = [m.model_dump() for m in complete_team_data.members_with_roles] updated_team = await prisma_client.db.litellm_teamtable.update( where={"team_id": data.team_id}, data={"members_with_roles": json.dumps(_db_team_members)}, # type: ignore diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 083d84c2b..c52300759 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -1175,7 +1175,6 @@ def test_generate_and_update_key(prisma_client): asyncio.run(test()) except Exception as e: print("Got Exception", e) - print(e.message) pytest.fail(f"An exception occurred - {str(e)}")