revoke_invitation

redbrick-ai · Jan 30, 2025 · f77868d · f77868d
1 parent 39eb085
commit f77868d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/docs/sdk.rst b/docs/sdk.rst
@@ -21,7 +21,7 @@ Organization
 Team
 ----------------------
 .. autoclass:: redbrick.member.Team
-   :members: get_member, list_members, remove_member, list_invites, invite_user, delete_invitation
+   :members: get_member, list_members, remove_member, list_invites, invite_user, revoke_invitation
    :show-inheritance:
 
 Workspace

diff --git a/redbrick/member/team.py b/redbrick/member/team.py
@@ -129,13 +129,13 @@ def invite_user(self, invitation: OrgInvite) -> OrgInvite:
         )
         return OrgInvite.from_entity(invite)
 
-    def delete_invitation(self, invitation: OrgInvite) -> None:
-        """Invite a user to the organization.
+    def revoke_invitation(self, invitation: OrgInvite) -> None:
+        """Revoke org user invitation.
 
         .. code:: python
 
             org = redbrick.get_org(org_id, api_key)
-            org.team.delete_invitation(OrgInvite(email="..."))
+            org.team.revoke_invitation(OrgInvite(email="..."))
 
         Parameters
         --------------

diff --git a/redbrick/repo/member.py b/redbrick/repo/member.py
@@ -108,7 +108,7 @@ def delete_org_invitation(self, org_id: str, invitation: Dict) -> None:
         mutation removeInviteSDK(
             $orgId: UUID!
             $email: String!
-            $idProvider: IDProvider!
+            $idProvider: IDProvider
         ) {
             removeInvite(
                 orgId: $orgId
@@ -122,7 +122,7 @@ def delete_org_invitation(self, org_id: str, invitation: Dict) -> None:
         query_variables = {
             "orgId": org_id,
             "email": invitation["email"],
-            "idProvider": invitation["idProvider"] or "COGNITO",
+            "idProvider": invitation["idProvider"],
         }
         self.client.execute_query(query_string, query_variables)