o nib@sHdZddlZddlmZddlmZGdddeZGdddeZdS) zPublic API: teams.N)gql)Attrscs4eZdZedZfddZddZddZZS)Memberz mutation DeleteInvite($id: String, $entityName: String) { deleteInvite(input: {id: $id, entityName: $entityName}) { success } } cst|||_||_dSN)super__init___clientteam)selfclientr attrs __class__U/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/apis/public/teams.pyrs  zMember.__init__cCs>z|j|j|j|jdddWStjjyYdSw)z^Remove a member from a team. Returns: Boolean indicating success )id entityName deleteInvitesuccessF)rexecuteDELETE_MEMBER_MUTATIONrr requests exceptions HTTPErrorr rrrdeletesz Member.deletecCsd|jd|jdS)Nz)name account_typerrrr__repr__&szMember.__repr__) __name__ __module__ __qualname__rrrrr __classcell__rrr rr s  rcspeZdZedZedZedZedZdfdd Ze ddd Z dd d Z d dZ dddZ ddZZS)TeamaN mutation CreateTeam($teamName: String!, $teamAdminUserName: String) { createTeam(input: {teamName: $teamName, teamAdminUserName: $teamAdminUserName}) { entity { id name available photoUrl limits } } } a mutation CreateInvite($entityName: String!, $email: String, $username: String, $admin: Boolean) { createInvite(input: {entityName: $entityName, email: $email, username: $username, admin: $admin}) { invite { id name email createdAt toUser { name } } } } aT query Entity($name: String!) { entity(name: $name) { id name available photoUrl readOnly readOnlyAdmin isTeam privateOnly storageBytes codeSavingEnabled defaultAccess isPaid members { id admin pending email username name photoUrl accountType apiKey } } } a  mutation CreateServiceAccount($entityName: String!, $description: String!) { createServiceAccount( input: {description: $description, entityName: $entityName} ) { user { id } } } Ncs(t|pi||_||_|dSr)rrrrload)r r rr r rrrxs z Team.__init__cCs>z |j|j||dWn tjjyYnwt|j|S)a'Create a new team. Args: api: (`Api`) The api instance to use team: (str) The name of the team admin_username: (str) optional username of the admin user of the team, defaults to the current user. Returns: A `Team` object )teamNameteamAdminUserName)r rCREATE_TEAM_MUTATIONrrrr#)clsapir admin_usernamerrrcreate~s  z Team.createFcCsV|j|d}d|vr||d<n||d<z |j|j|WdStjjy*YdSw)aGInvite a user to a team. Args: username_or_email: (str) The username or email address of the user you want to invite admin: (bool) Whether to make this user a team admin, defaults to False Returns: True on success, False if user was already invited or didn't exist )radmin@emailusernameFT)rrrCREATE_INVITE_MUTATIONrrr)r username_or_emailr, variablesrrrinvites  z Team.invitecCsHz|j|j||jd|d|jdWStjjy#YdSw)zCreate a service account for the team. Args: description: (str) A description for this service account Returns: The service account `Member` object, or None on failure ) descriptionrTN) rrCREATE_SERVICE_ACCOUNT_MUTATIONrr$membersrrr)r r4rrrcreate_service_accounts    zTeam.create_service_accountcsN|sjs$jjdji}|d_fddjdDjd<jS)Nrentitycsg|] }tjj|qSr)rrr).0memberrrr szTeam.load..r7)_attrsrr TEAM_QUERYr)r forceresponserrrr$s    z Team.loadcCsd|jdS)Nz)rrrrrrsz Team.__repr__r)F)rr r!rr'r0r>r6r classmethodr+r3r8r$rr"rrr rr#*s(    r#)__doc__r wandb_gqlrwandb.apis.attrsrrr#rrrrs   !