add a hard delete on memberships to fix stuck workflows

dust-tt · Dec 10, 2024 · 4535dde · 4535dde
1 parent e55b115
commit 4535dde
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/front/lib/resources/membership_resource.ts b/front/lib/resources/membership_resource.ts
@@ -309,6 +309,16 @@ export class MembershipResource extends BaseResource<MembershipModel> {
     });
   }
 
+  static async deleteAllForWorkspace(
+    workspace: LightWorkspaceType,
+    transaction?: Transaction
+  ) {
+    return this.model.destroy({
+      where: { workspaceId: workspace.id },
+      transaction,
+    });
+  }
+
   /**
    * Caller of this method should call `ServerSideTracking.trackCreateMembership`.
    */

diff --git a/front/poke/temporal/activities.ts b/front/poke/temporal/activities.ts
@@ -606,6 +606,7 @@ export async function deleteWorkspaceActivity({
     });
     await FileResource.deleteAllForWorkspace(workspace, t);
     await RunResource.deleteAllForWorkspace(workspace, t);
+    await MembershipResource.deleteAllForWorkspace(workspace, t);
     await AgentUserRelation.destroy({
       where: { workspaceId: workspace.id },
       transaction: t,