jupyterlab-ipyflow 0.0.178 → 0.0.179

package/lib/index.js

@@ -70,20 +70,13 @@ class IpyflowSessionState {
         let numFinished = 0;
         for (const cell of cells) {
             // if any of them fail, change the [*] to [ ] on subsequent cells
-            CodeCell.execute(cell, this.session).then((msg) => {
-                var _a, _b, _c;
-                if (((_b = (_a = msg) === null || _a === void 0 ? void 0 : _a.content) === null || _b === void 0 ? void 0 : _b.status) === 'error') {
-                    numFinished = cells.length;
-                    for (const cell of cells) {
-                        if ((_c = cell.promptNode.textContent) === null || _c === void 0 ? void 0 : _c.includes('[*]')) {
-                            cell.setPrompt('');
-                        }
-                    }
+            CodeCell.execute(cell, this.session).then(() => {
+                var _a;
+                if ((_a = cell.promptNode.textContent) === null || _a === void 0 ? void 0 : _a.includes('[*]')) {
+                    // can happen if a preceding cell errored
+                    cell.setPrompt('');
                 }
-                else {
-                    numFinished++;
-                }
-                if (numFinished === cells.length) {
+                if (++numFinished === cells.length) {
                     this.requestComputeExecSchedule();
                 }
             });
@@ -507,6 +500,7 @@ const addUnsafeCellInteraction = (elem, linkedElems, cellsById, collapserFun, ev
     attachCleanupListener(elem, evt, listener);
 };
 const connectToComm = (session, notebooks, notebook) => {
+    var _a, _b;
     initSessionState(session.session.id);
     const state = ipyflowState[session.session.id];
     state.activeCell = notebook.activeCell;
@@ -762,6 +756,11 @@ const connectToComm = (session, notebooks, notebook) => {
             state.settings = payload.settings;
             state.cellParents = payload.cell_parents;
             state.cellChildren = payload.cell_children;
+            notebook.model.setMetadata('ipyflow', {
+                cell_parents: state.cellParents,
+                cell_children: state.cellChildren,
+            });
+            void notebooks.currentWidget.context.save();
             state.waitingCells = new Set(payload.waiting_cells);
             state.readyCells = new Set(payload.ready_cells);
             if (state.numPendingForcedReactiveCounterBumps === 0) {
@@ -910,8 +909,12 @@ const connectToComm = (session, notebooks, notebook) => {
             }
         }
     };
+    const ipyflow_metadata = notebook.model.getMetadata('ipyflow');
     comm.open({
         interface: 'jupyterlab',
+        cell_metadata_by_id: state.gatherCellMetadataAndContent(),
+        cell_parents: (_a = ipyflow_metadata === null || ipyflow_metadata === void 0 ? void 0 : ipyflow_metadata.cell_parents) !== null && _a !== void 0 ? _a : {},
+        cell_children: (_b = ipyflow_metadata === null || ipyflow_metadata === void 0 ? void 0 : ipyflow_metadata.cell_children) !== null && _b !== void 0 ? _b : {},
     });
     // return a disconnection handle
     return () => {

package/package.json

@@ -67,5 +67,5 @@
     "extension": true,
     "outputDir": "../../core/ipyflow/resources/labextension/"
   },
-  "version": "0.0.178"
+  "version": "0.0.179"
 }