@marimo-team/islands 0.24.1-dev61 → 0.24.1-dev63

package/dist/{chat-ui-DYcwIX9k.js → chat-ui-DZQth8T9.js}

@@ -3,12 +3,12 @@ import { c as n, g as r, h as i, t as a } from "./button-DOCIubLd.js";
 import { t as o } from "./react-DMWcHin-.js";
 import { t as s } from "./compiler-runtime-CAH9RLw2.js";
 import { r as c } from "./copy-DgAphl4k.js";
-import { Dn as l, F as u, Fn as d, Ft as f, G as p, Gn as m, Hn as h, In as g, Nt as _, Qt as v, Rt as y, Wn as b, Wt as x, Xn as S, Xt as C, Zt as w, _ as ee, a as T, c as E, cn as te, d as D, dn as ne, dt as re, en as ie, et as ae, f as oe, ft as se, g as ce, gn as le, h as ue, ht as de, kn as fe, kt as pe, l as me, ln as he, lt as ge, m as _e, n as ve, on as ye, p as be, qn as xe, r as Se, sn as Ce, t as we, v as Te, x as Ee, y as De, zt as Oe } from "./html-to-image-DnwkphJg.js";
+import { Dn as l, F as u, Fn as d, Ft as f, G as p, Gn as m, Hn as h, In as g, Nt as _, Qt as v, Rt as y, Wn as b, Wt as x, Xn as S, Xt as C, Zt as w, _ as ee, a as T, c as E, cn as te, d as D, dn as ne, dt as re, en as ie, et as ae, f as oe, ft as se, g as ce, gn as le, h as ue, ht as de, kn as fe, kt as pe, l as me, ln as he, lt as ge, m as _e, n as ve, on as ye, p as be, qn as xe, r as Se, sn as Ce, t as we, v as Te, x as Ee, y as De, zt as Oe } from "./html-to-image-ChI8i0f-.js";
 import { u as ke } from "./dist-DTSuOdDc.js";
 import { C as Ae, I as je, R as Me, b as Ne } from "./select-Bakm9EUs.js";
 import { f as Pe, g as Fe, i as Ie, n as Le, t as Re, v as ze } from "./dropdown-menu-DrzAtluA.js";
 import { _ as Be, p as Ve } from "./toDate-Mzvy7JMZ.js";
-import { d as He, f as Ue, i as We, n as Ge, t as Ke, u as qe } from "./process-output-BkiZCdL9.js";
+import { d as He, f as Ue, i as We, n as Ge, t as Ke, u as qe } from "./process-output-e5z-_0AB.js";
 import { t as Je } from "./jsx-runtime-CpoKNRgu.js";
 import { Cn as O, Cr as Ye, In as k, It as Xe, Jn as A, Sn as Ze, Tt as Qe, Yn as $e, bt as et, cn as tt, jn as nt, kt as j, ln as M, mr as rt, un as N, vt as it, wt as at, xt as ot, zt as st } from "./schemas-DeBAkn_c.js";
 import { n as ct } from "./error-banner-BT4uIIQ2.js";

package/dist/{common-B_E9_TZZ.js → common-D0oV2dfP.js}

@@ -3,7 +3,7 @@ import { _ as n, f as r, g as i, h as a, m as o, n as s, p as c, r as l, t as u
 import { t as d } from "./react-DMWcHin-.js";
 import { t as f } from "./compiler-runtime-CAH9RLw2.js";
 import { n as p, r as m, t as h } from "./copy-DgAphl4k.js";
-import { $n as g, A as _, An as v, Bn as y, Bt as b, Cn as x, Ct as S, D as C, Dn as w, Dt as T, E, F as D, Fn as O, G as k, Gn as A, Hn as j, In as M, It as N, Jt as P, Kn as F, Kt as I, Lt as L, M as R, Mt as ee, N as z, Qn as te, Rn as ne, Rt as re, S as ie, T as ae, Tn as oe, Tt as se, V as ce, Vn as B, Wn as le, Wt as ue, X as de, Xn as fe, Y as pe, Z as me, _ as he, _n as ge, at as _e, dt as ve, en as ye, er as V, fn as be, ft as xe, g as Se, gn as Ce, ht as we, lt as Te, mt as Ee, n as De, nn as Oe, nt as ke, pt as Ae, qn as je, qt as Me, r as Ne, s as Pe, t as Fe, tt as Ie, u as Le, ut as Re, v as ze, w as Be, wn as Ve, wt as He, y as Ue, yt as We } from "./html-to-image-DnwkphJg.js";
+import { $n as g, A as _, An as v, Bn as y, Bt as b, Cn as x, Ct as S, D as C, Dn as w, Dt as T, E, F as D, Fn as O, G as k, Gn as A, Hn as j, In as M, It as N, Jt as P, Kn as F, Kt as I, Lt as L, M as R, Mt as ee, N as z, Qn as te, Rn as ne, Rt as re, S as ie, T as ae, Tn as oe, Tt as se, V as ce, Vn as B, Wn as le, Wt as ue, X as de, Xn as fe, Y as pe, Z as me, _ as he, _n as ge, at as _e, dt as ve, en as ye, er as V, fn as be, ft as xe, g as Se, gn as Ce, ht as we, lt as Te, mt as Ee, n as De, nn as Oe, nt as ke, pt as Ae, qn as je, qt as Me, r as Ne, s as Pe, t as Fe, tt as Ie, u as Le, ut as Re, v as ze, w as Be, wn as Ve, wt as He, y as Ue, yt as We } from "./html-to-image-ChI8i0f-.js";
 import { o as Ge, u as H } from "./dist-DTSuOdDc.js";
 import { c as Ke, i as qe, r as Je } from "./useLifecycle-ClOphI_o.js";
 import { a as Ye, i as Xe, n as Ze, o as Qe, r as $e, t as et } from "./useDateFormatter-DjRKOAkR.js";
@@ -31786,7 +31786,7 @@ function ZG() {
 		return i.warn("Failed to get version from mount config"), null;
 	}
 }
-var QG = ZG() || "0.24.1-dev61", $G = xi(QG), eK = xi(!0);
+var QG = ZG() || "0.24.1-dev63", $G = xi(QG), eK = xi(!0);
 xi(null);
 //#endregion
 //#region src/core/meta/code-visibility.ts

package/dist/{html-to-image-DnwkphJg.js → html-to-image-ChI8i0f-.js}

@@ -22865,8 +22865,8 @@ function Yg(e, t) {
 var Xg = P((e) => {
 	let t = e($);
 	return pn.fromEntries(t.cellIds.inOrderIds.map((e) => {
-		let n = t.cellHandles[e];
-		return [e, n?.current?.editorView ? cC(n.current.editorView) : ""];
+		let n = t.cellHandles[e]?.current?.editorViewOrNull;
+		return [e, n ? cC(n) : t.cellData[e]?.code ?? ""];
 	}));
 }), Zg = P((e) => e($).cellIds.inOrderIds), Qg = P((e) => e_(e(Zg), e(Pm))), $g = P((e) => ({
 	cellIds: e(Qg),

package/dist/main.js

@@ -3,16 +3,16 @@ import { _ as a, c as o, f as s, g as c, h as l, i as u, l as d, m as f, n as p,
 import { t as y } from "./react-DMWcHin-.js";
 import { t as b } from "./compiler-runtime-CAH9RLw2.js";
 import { n as x, r as S, t as C } from "./copy-DgAphl4k.js";
-import { $t as w, At as T, B as E, C as D, Cn as O, Ct as k, Dn as A, Dt as j, En as M, Et as N, F as P, Ft as F, G as I, Gn as L, Gt as ee, H as te, Hn as R, Ht as ne, I as z, In as re, J as ie, Jn as ae, K as oe, Kn as se, Kt as ce, L as le, Ln as ue, Mn as de, Nn as fe, On as pe, Ot as me, P as he, Pn as ge, Pt as _e, Q as ve, Qn as ye, R as be, Rt as xe, Sn as Se, St as Ce, Tt as we, U as Te, Un as Ee, Ut as De, Vt as Oe, W as ke, Wn as Ae, X as je, Y as Me, Yn as Ne, Yt as Pe, Zn as Fe, _ as Ie, _n as Le, _t as Re, a as ze, an as Be, b as Ve, bn as He, bt as Ue, ct as We, en as Ge, g as Ke, gt as qe, hn as Je, i as Ye, in as Xe, it as Ze, j as Qe, jn as $e, jt as et, k as tt, mn as nt, n as rt, nn as it, nt as at, o as ot, ot as st, pn as ct, q as lt, rn as ut, rt as dt, st as ft, tn as pt, un as mt, v as ht, vt as gt, xn as _t, xt as vt, y as yt, yn as bt, yt as xt, z as St, zn as Ct } from "./html-to-image-DnwkphJg.js";
+import { $t as w, At as T, B as E, C as D, Cn as O, Ct as k, Dn as A, Dt as j, En as M, Et as N, F as P, Ft as F, G as I, Gn as L, Gt as ee, H as te, Hn as R, Ht as ne, I as z, In as re, J as ie, Jn as ae, K as oe, Kn as se, Kt as ce, L as le, Ln as ue, Mn as de, Nn as fe, On as pe, Ot as me, P as he, Pn as ge, Pt as _e, Q as ve, Qn as ye, R as be, Rt as xe, Sn as Se, St as Ce, Tt as we, U as Te, Un as Ee, Ut as De, Vt as Oe, W as ke, Wn as Ae, X as je, Y as Me, Yn as Ne, Yt as Pe, Zn as Fe, _ as Ie, _n as Le, _t as Re, a as ze, an as Be, b as Ve, bn as He, bt as Ue, ct as We, en as Ge, g as Ke, gt as qe, hn as Je, i as Ye, in as Xe, it as Ze, j as Qe, jn as $e, jt as et, k as tt, mn as nt, n as rt, nn as it, nt as at, o as ot, ot as st, pn as ct, q as lt, rn as ut, rt as dt, st as ft, tn as pt, un as mt, v as ht, vt as gt, xn as _t, xt as vt, y as yt, yn as bt, yt as xt, z as St, zn as Ct } from "./html-to-image-ChI8i0f-.js";
 import { o as wt, s as Tt, u as Et } from "./dist-DTSuOdDc.js";
 import { D as Dt, E as Ot, O as kt, S as At, T as jt, _ as Mt, a as Nt, c as Pt, d as Ft, f as It, g as Lt, h as Rt, i as zt, l as Bt, m as Vt, n as Ht, o as Ut, p as Wt, r as Gt, s as Kt, t as qt, u as Jt, v as Yt, w as Xt, x as Zt, y as Qt } from "./spec-iNjNWU_x.js";
-import { $ as $t, $t as en, A as tn, At as nn, B as rn, Bt as an, C as on, Ct as sn, D as cn, Dt as ln, E as un, Et as dn, F as fn, Ft as pn, G as mn, Gt as hn, H as gn, Ht as _n, I as vn, It as yn, Jt as bn, K as xn, Kt as Sn, L as Cn, Lt as wn, M as Tn, Mt as En, N as Dn, Nt as On, O as kn, Ot as An, P as jn, Pt as Mn, Q as Nn, Qt as Pn, R as Fn, Rt as In, S as Ln, St as Rn, T as zn, Tt as Bn, U as Vn, Ut as Hn, V as Un, Vt as Wn, W as Gn, Wt as Kn, Xt as qn, Yt as Jn, Z as Yn, Zt as Xn, _ as Zn, an as Qn, at as $n, b as er, bt as tr, c as nr, cn as rr, ct as ir, d as ar, dn as or, dt as sr, en as cr, et as lr, f as ur, fn as dr, ft as fr, g as pr, gt as mr, h as hr, ht as gr, i as _r, it as vr, j as yr, jt as br, k as xr, kt as Sr, ln as Cr, lt as wr, m as Tr, mt as Er, n as Dr, nn as Or, nt as kr, on as Ar, ot as jr, p as Mr, pn as Nr, pt as Pr, q as Fr, qt as Ir, r as Lr, rn as Rr, rt as zr, sn as Br, st as Vr, t as Hr, tn as Ur, tt as Wr, u as Gr, un as Kr, ut as qr, v as Jr, w as Yr, wt as Xr, x as Zr, xt as Qr, y as $r, yt as ei, z as ti, zt as ni } from "./common-B_E9_TZZ.js";
+import { $ as $t, $t as en, A as tn, At as nn, B as rn, Bt as an, C as on, Ct as sn, D as cn, Dt as ln, E as un, Et as dn, F as fn, Ft as pn, G as mn, Gt as hn, H as gn, Ht as _n, I as vn, It as yn, Jt as bn, K as xn, Kt as Sn, L as Cn, Lt as wn, M as Tn, Mt as En, N as Dn, Nt as On, O as kn, Ot as An, P as jn, Pt as Mn, Q as Nn, Qt as Pn, R as Fn, Rt as In, S as Ln, St as Rn, T as zn, Tt as Bn, U as Vn, Ut as Hn, V as Un, Vt as Wn, W as Gn, Wt as Kn, Xt as qn, Yt as Jn, Z as Yn, Zt as Xn, _ as Zn, an as Qn, at as $n, b as er, bt as tr, c as nr, cn as rr, ct as ir, d as ar, dn as or, dt as sr, en as cr, et as lr, f as ur, fn as dr, ft as fr, g as pr, gt as mr, h as hr, ht as gr, i as _r, it as vr, j as yr, jt as br, k as xr, kt as Sr, ln as Cr, lt as wr, m as Tr, mt as Er, n as Dr, nn as Or, nt as kr, on as Ar, ot as jr, p as Mr, pn as Nr, pt as Pr, q as Fr, qt as Ir, r as Lr, rn as Rr, rt as zr, sn as Br, st as Vr, t as Hr, tn as Ur, tt as Wr, u as Gr, un as Kr, ut as qr, v as Jr, w as Yr, wt as Xr, x as Zr, xt as Qr, y as $r, yt as ei, z as ti, zt as ni } from "./common-D0oV2dfP.js";
 import { c as ri, i as ii, n as ai, o as oi, t as si } from "./useLifecycle-ClOphI_o.js";
 import { t as ci } from "./check-DIJJHY-E.js";
 import { A as li, C as ui, D as di, E as fi, F as pi, I as mi, M as hi, N as gi, P as _i, R as vi, S as yi, a as bi, b as xi, c as Si, i as Ci, j as wi, l as Ti, n as Ei, o as Di, r as Oi, s as ki, t as Ai, w as ji, x as Mi } from "./select-Bakm9EUs.js";
 import { B as Ni, C as Pi, Ft as Fi, G as Ii, It as Li, S as Ri, V as zi, Wt as Bi, X as Vi, Y as Hi, _ as Ui, a as Wi, c as Gi, f as Ki, g as qi, gn as Ji, h as Yi, hn as Xi, i as Zi, m as Qi, n as $i, ot as ea, pt as ta, r as na, s as ra, st as ia, t as aa, v as oa, w as sa, xt as ca, y as la } from "./dropdown-menu-DrzAtluA.js";
 import { _ as ua, c as da, d as fa, f as pa, g as ma, h as ha, m as ga, u as _a } from "./toDate-Mzvy7JMZ.js";
-import { a as va, c as ya, d as ba, f as xa, i as Sa, l as Ca, n as wa, o as Ta, r as Ea, s as Da, t as Oa } from "./process-output-BkiZCdL9.js";
+import { a as va, c as ya, d as ba, f as xa, i as Sa, l as Ca, n as wa, o as Ta, r as Ea, s as Da, t as Oa } from "./process-output-e5z-_0AB.js";
 import { i as ka, n as Aa, r as ja, t as Ma } from "./types-D8U7wKuc.js";
 import { n as Na, t as Pa } from "./ErrorBoundary-BF3QLrI_.js";
 import { t as Fa } from "./react-dom-C94wmq5w.js";
@@ -15397,7 +15397,7 @@ var uM = (e) => {
 			output: t
 		};
 	} };
-} }, gM = q.lazy(() => import("./chat-ui-DYcwIX9k.js").then((e) => ({ default: e.Chatbot }))), _M = U(B({
+} }, gM = q.lazy(() => import("./chat-ui-DZQth8T9.js").then((e) => ({ default: e.Chatbot }))), _M = U(B({
 	id: V(),
 	role: Pp([
 		"system",
@@ -35256,7 +35256,7 @@ function Gq(e) {
 }
 //#endregion
 //#region src/components/editor/renderers/slides-layout/slides-layout.tsx
-var Kq = q.lazy(() => import("./reveal-component-iJ-KAo0l.js")), qq = (e) => {
+var Kq = q.lazy(() => import("./reveal-component-DZmM1R5M.js")), qq = (e) => {
 	let t = (0, Y.c)(33), { layout: n, setLayout: r, cells: i, mode: a } = e, o = Nm(k), s = a === "read" || o, c = Nm(lt) > 1, [l, u] = (0, q.useState)(null), d;
 	t[0] !== i || t[1] !== n ? (d = UU(i, n), t[0] = i, t[1] = n, t[2] = d) : d = t[2];
 	let { slideCells: f, skippedIds: p, noOutputIds: m, slideTypes: h, startCellIndex: g } = d, _;

package/dist/{process-output-BkiZCdL9.js → process-output-e5z-_0AB.js}

@@ -1,4 +1,4 @@
-import { $ as e, O as t, et as n } from "./html-to-image-DnwkphJg.js";
+import { $ as e, O as t, et as n } from "./html-to-image-ChI8i0f-.js";
 import { u as r } from "./dist-DTSuOdDc.js";
 import { b as i } from "./select-Bakm9EUs.js";
 import { n as a } from "./semaphore-89jsoHtS.js";

package/dist/{reveal-component-iJ-KAo0l.js → reveal-component-DZmM1R5M.js}

@@ -2,9 +2,9 @@ import { a as e } from "./rolldown-runtime-Cb76FRpK.js";
 import { c as t, g as n, h as r, m as i, t as a } from "./button-DOCIubLd.js";
 import { t as o } from "./react-DMWcHin-.js";
 import { t as s } from "./compiler-runtime-CAH9RLw2.js";
-import { Ct as c, rt as l } from "./html-to-image-DnwkphJg.js";
+import { Ct as c, rt as l } from "./html-to-image-ChI8i0f-.js";
 import { u } from "./dist-DTSuOdDc.js";
-import { J as d, X as f, Y as p, in as m, l as h, n as g, rn as _, sn as v, t as y, u as b } from "./common-B_E9_TZZ.js";
+import { J as d, X as f, Y as p, in as m, l as h, n as g, rn as _, sn as v, t as y, u as b } from "./common-D0oV2dfP.js";
 import { R as x } from "./select-Bakm9EUs.js";
 import { X as S } from "./dropdown-menu-DrzAtluA.js";
 import { t as C } from "./jsx-runtime-CpoKNRgu.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.24.1-dev61",
+  "version": "0.24.1-dev63",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/core/codemirror/copilot/__tests__/getCodes.test.ts

@@ -5,13 +5,18 @@ import { EditorView } from "@codemirror/view";
 import { describe, expect, it } from "vitest";
 import { cellId, variableName } from "@/__tests__/branded";
 import { initialNotebookState, notebookAtom } from "@/core/cells/cells";
+import { createCell } from "@/core/cells/types";
 import { OverridingHotkeyProvider } from "@/core/hotkeys/hotkeys";
 import { store } from "@/core/state/jotai";
 import { variablesAtom } from "@/core/variables/state";
 import { MultiColumn } from "@/utils/id-tree";
 import { cellConfigExtension } from "../../config/extension";
 import { adaptiveLanguageConfiguration } from "../../language/extension";
-import { getCodes, getTopologicalCellIds } from "../getCodes";
+import {
+  getCodes,
+  getTopologicalCellIds,
+  topologicalCodesAtom,
+} from "../getCodes";
 
 const Cells = {
   cell1: cellId("cell1"),
@@ -208,4 +213,64 @@ describe("getCodes", () => {
     const result = getCodes(otherCode);
     expect(result).toEqual("import os\nx = 1\ny = x + 1\nprint('Hello World')");
   });
+
+  it("should read code from the store when no editor is mounted", () => {
+    const otherCode = "print('Hello World')";
+    store.set(notebookAtom, {
+      ...initialNotebookState(),
+      cellIds: MultiColumn.from([[Cells.cell1, Cells.cell2]]),
+      cellData: {
+        [Cells.cell1]: createCell({ id: Cells.cell1, code: "import os" }),
+        [Cells.cell2]: createCell({ id: Cells.cell2, code: "x = 1" }),
+      },
+    });
+    const result = getCodes(otherCode);
+    expect(result).toEqual("import os\nx = 1\nprint('Hello World')");
+  });
+
+  it("should prefer the mounted editor code over the store code", () => {
+    const otherCode = "print('Hello World')";
+    store.set(notebookAtom, {
+      ...initialNotebookState(),
+      cellIds: MultiColumn.from([[Cells.cell1, Cells.cell2]]),
+      cellData: {
+        [Cells.cell1]: createCell({ id: Cells.cell1, code: "import os" }),
+        [Cells.cell2]: createCell({ id: Cells.cell2, code: "x = 1" }),
+      },
+      cellHandles: {
+        [Cells.cell2]: { current: createMockEditorView("x = 2") },
+      },
+    });
+    const result = getCodes(otherCode);
+    expect(result).toEqual("import os\nx = 2\nprint('Hello World')");
+  });
+});
+
+describe("topologicalCodesAtom", () => {
+  it("should include every cell when no editor is mounted", () => {
+    store.set(notebookAtom, {
+      ...initialNotebookState(),
+      cellIds: MultiColumn.from([[Cells.cell1, Cells.cell2, Cells.cell3]]),
+      cellData: {
+        [Cells.cell1]: createCell({ id: Cells.cell1, code: "import os" }),
+        [Cells.cell2]: createCell({ id: Cells.cell2, code: "x = 1" }),
+        [Cells.cell3]: createCell({ id: Cells.cell3, code: "y = x + 1" }),
+      },
+    });
+    store.set(variablesAtom, {
+      [Variables.var1]: {
+        name: Variables.var1,
+        declaredBy: [Cells.cell1],
+        usedBy: [Cells.cell2, Cells.cell3],
+      },
+    });
+
+    const { cellIds, codes } = store.get(topologicalCodesAtom);
+    expect(cellIds).toEqual([Cells.cell1, Cells.cell2, Cells.cell3]);
+    expect(codes).toEqual({
+      [Cells.cell1]: "import os",
+      [Cells.cell2]: "x = 1",
+      [Cells.cell3]: "y = x + 1",
+    });
+  });
 });

package/src/core/codemirror/copilot/getCodes.ts

@@ -1,13 +1,32 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
 import { atom } from "jotai";
-import { getAllEditorViews, notebookAtom } from "@/core/cells/cells";
+import { notebookAtom } from "@/core/cells/cells";
 import type { CellId } from "@/core/cells/ids";
+import { store } from "@/core/state/jotai";
 import { variablesAtom } from "@/core/variables/state";
 import type { Variables } from "@/core/variables/types";
 import { Objects } from "@/utils/objects";
 import { getEditorCodeAsPython } from "../language/utils";
 
+const notebookCellCodes = atom((get) => {
+  const notebook = get(notebookAtom);
+  const codes = Objects.fromEntries(
+    notebook.cellIds.inOrderIds.map((id) => {
+      // A cell editor mounts progressively, so the notebook store is the only
+      // complete source of code until every editor is built.
+      const editorView = notebook.cellHandles[id]?.current?.editorViewOrNull;
+      return [
+        id,
+        editorView
+          ? getEditorCodeAsPython(editorView)
+          : (notebook.cellData[id]?.code ?? ""),
+      ];
+    }),
+  );
+  return codes;
+});
+
 export function getCodes(otherCode: string) {
   const codes = getOtherCellsCode(otherCode);
 
@@ -18,15 +37,8 @@ export function getOtherCellsCode(otherCode: string) {
   // Get all other cells' code
   // Put `import` statements at the top, as it can help copilot give better suggestions
   // TODO: we should sort this topologically
-  const codes = getAllEditorViews()
-    .map((editorView) => {
-      const code = getEditorCodeAsPython(editorView);
-      if (code === otherCode) {
-        return null;
-      }
-      return code;
-    })
-    .filter(Boolean)
+  const codes = Object.values(store.get(notebookCellCodes))
+    .filter((code) => code !== "" && code !== otherCode)
     .toSorted((a, b) => {
       if (a.startsWith("import") && !b.startsWith("import")) {
         return -1;
@@ -40,21 +52,6 @@ export function getOtherCellsCode(otherCode: string) {
   return codes;
 }
 
-const notebookCellCodes = atom((get) => {
-  const notebook = get(notebookAtom);
-  const codes = Objects.fromEntries(
-    notebook.cellIds.inOrderIds.map((id) => {
-      const handle = notebook.cellHandles[id];
-      return [
-        id,
-        handle?.current?.editorView
-          ? getEditorCodeAsPython(handle.current.editorView)
-          : "",
-      ];
-    }),
-  );
-  return codes;
-});
 const inOrderCellIdsAtom = atom((get) => {
   const notebook = get(notebookAtom);
   return notebook.cellIds.inOrderIds;

package/src/core/codemirror/lsp/__tests__/notebook-lsp.test.ts

@@ -8,7 +8,10 @@ import { beforeEach, describe, expect, it, type Mocked, vi } from "vitest";
 import * as LSP from "vscode-languageserver-protocol";
 import { cellId } from "@/__tests__/branded";
 import type { CellId } from "@/core/cells/ids";
+import { initialNotebookState, notebookAtom } from "@/core/cells/cells";
+import { createCell } from "@/core/cells/types";
 import { store } from "@/core/state/jotai";
+import { MultiColumn } from "@/utils/id-tree";
 import { topologicalCodesAtom } from "../../copilot/getCodes";
 import { lspWorkspaceAtom } from "@/core/saving/file-state";
 import { languageAdapterState } from "../../language/extension";
@@ -62,6 +65,28 @@ function isPublishDiagnosticsNotification(
   );
 }
 
+describe("merged document with unmounted editors", () => {
+  it("should include the code of cells whose editors are not mounted", () => {
+    store.set(notebookAtom, {
+      ...initialNotebookState(),
+      cellIds: MultiColumn.from([[Cells.cell1, Cells.cell2]]),
+      cellData: {
+        [Cells.cell1]: createCell({ id: Cells.cell1, code: "import math" }),
+        [Cells.cell2]: createCell({
+          id: Cells.cell2,
+          code: "print(math.sqrt(4))",
+        }),
+      },
+    });
+
+    const { cellIds, codes } = store.get(topologicalCodesAtom);
+    const lens = createNotebookLens(cellIds, codes);
+
+    expect(lens.mergedText).toContain("import math");
+    expect(lens.mergedText).toContain("print(math.sqrt(4))");
+  });
+});
+
 describe("createNotebookLens", () => {
   it("should produce correct lens for same inputs", () => {
     // Use unique content for this test