forked from phoenix/litellm-mirror
fix(proxy/_types.py): fix validation check
This commit is contained in:
parent
def648ed3f
commit
d066e0e152
3 changed files with 4 additions and 5 deletions
|
@ -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
|
||||
|
|
|
@ -10190,9 +10190,7 @@ 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
|
||||
]
|
||||
_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
|
||||
|
|
|
@ -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)}")
|
||||
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue