@marimo-team/islands 0.25.3-dev143 → 0.25.3-dev145

package/src/core/{document → notebook/document}/hooks.tsx

@@ -46,7 +46,7 @@ function useDocumentModel() {
 }
 
 /** Null while the document is unavailable; an empty array for an empty document. */
-export function useCellOrder() {
+export function useNotebookCellOrder() {
   const { documentAtom } = useDocumentModel();
   return useAtomValue(useMemo(() => selectOrder(documentAtom), [documentAtom]));
 }
@@ -62,17 +62,17 @@ function useCellField<K extends keyof EditableCell>(id: CellId, field: K) {
 }
 
 /** Undefined for an absent cell; null when the host withholds its source. */
-export function useCellCode(id: CellId) {
+export function useNotebookCellCode(id: CellId) {
   return useCellField(id, "code");
 }
 
 /** Undefined when the cell or document is unavailable. */
-export function useCellName(id: CellId) {
+export function useNotebookCellName(id: CellId) {
   return useCellField(id, "name");
 }
 
 /** Undefined when the cell or document is unavailable. */
-export function useCellConfig(id: CellId) {
+export function useNotebookCellConfig(id: CellId) {
   return useCellField(id, "config");
 }
 
@@ -81,10 +81,10 @@ export function useCellConfig(id: CellId) {
  * Stable until the cell id or document model changes.
  *
  * ```tsx
- * const extension = useCellCodeExtension(cellId);
+ * const extension = useNotebookCellCodeExtension(cellId);
  * ```
  */
-export function useCellCodeExtension(id: CellId) {
+export function useNotebookCellCodeExtension(id: CellId) {
   const model = useDocumentModel();
   const store = useStore();
   return useMemo(
@@ -97,11 +97,11 @@ export function useCellCodeExtension(id: CellId) {
  * A stable edit function. Synchronous callbacks publish atomically or roll back.
  *
  * ```ts
- * const edit = useDocumentEdit();
+ * const edit = useNotebookDocumentEdit();
  * edit((tx) => tx.setCode(cellId, code));
  * ```
  */
-export function useDocumentEdit() {
+export function useNotebookDocumentEdit() {
   return useSetAtom(useDocumentModel().editAtom);
 }
 
@@ -111,7 +111,7 @@ export function useDocumentEdit() {
  * Rejections retain local edits; retry explicitly resubmits them. Requests
  * with an unknown outcome reuse their original key within the retry window.
  */
-export function useDocumentSync() {
+export function useNotebookDocumentSync() {
   const model = useDocumentModel();
   const status = useAtomValue(model.syncStatusAtom);
   return { status, flush: model.flush, retry: model.retry };

package/src/core/{document → notebook/document}/index.ts

@@ -4,12 +4,12 @@
 
 export {
   DocumentProvider,
-  useCellCode,
-  useCellCodeExtension,
-  useCellConfig,
-  useCellName,
-  useCellOrder,
-  useDocumentEdit,
-  useDocumentSync,
+  useNotebookCellCode,
+  useNotebookCellCodeExtension,
+  useNotebookCellConfig,
+  useNotebookCellName,
+  useNotebookCellOrder,
+  useNotebookDocumentEdit,
+  useNotebookDocumentSync,
 } from "./hooks";
 export { createDocumentModel, type DocumentModel } from "./model";

package/src/core/{document → notebook/document}/model.ts

@@ -2,7 +2,7 @@
 
 import type * as Protocol from "@marimo-team/client/protocol";
 import type { Atom, createStore } from "jotai/vanilla";
-import type { Document, NotebookBackend } from "../host";
+import type { Document, NotebookBackend } from "../replica";
 import { createDocumentState, type DocumentState } from "./state";
 import { createDocumentSync, type DocumentSyncStatus } from "./sync";
 

package/src/core/{document → notebook/document}/state.ts

@@ -6,7 +6,7 @@ import {
   type createStore,
   type WritableAtom,
 } from "jotai/vanilla";
-import type { Document } from "../host";
+import type { Document } from "../replica";
 import * as Protocol from "@marimo-team/client/protocol";
 import {
   acknowledge,

package/src/core/{document → notebook/document}/sync.ts

@@ -3,7 +3,7 @@
 import { idempotencyKey, type Result } from "@marimo-team/client";
 import type * as Protocol from "@marimo-team/client/protocol";
 import { type Atom, atom, type createStore } from "jotai/vanilla";
-import type { Document, NotebookBackend } from "../host";
+import type { Document, NotebookBackend } from "../replica";
 import type { DocumentState, DocumentSubmission } from "./state";
 
 export type DocumentSyncStatus =

package/src/core/notebook/index.ts

@@ -0,0 +1,18 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+/** React hooks for the current notebook. */
+
+export {
+  useNotebookAttachments,
+  attachmentLabel,
+  type Attachment,
+} from "./attachments";
+export {
+  useNotebookCellCode,
+  useNotebookCellCodeExtension,
+  useNotebookCellConfig,
+  useNotebookCellName,
+  useNotebookCellOrder,
+  useNotebookDocumentEdit,
+  useNotebookDocumentSync,
+} from "./document";

package/src/core/notebook/replica/AGENTS.md

@@ -0,0 +1,11 @@
+# Notebook replica
+
+Production code may import replica modules, `@marimo-team/client`,
+`jotai/vanilla`, `zod`, and `@/utils/Logger`. Enforced by
+`marimo/replica-imports`. Tests are exempt.
+
+Code outside this directory imports it as `@/core/notebook/replica`.
+Enforced by `marimo/module-entry`.
+
+The protocol schemas and types live in `@marimo-team/client/protocol`.
+Define interfaces here and inject application behavior through adapters.

package/src/core/{host → notebook/replica}/__tests__/fixtures.ts

@@ -45,7 +45,7 @@ export function fakeBackend({
 /** Read the protocol's portable examples as transport messages. */
 export function fixtureMessages(name: string) {
   const path = new NodeUrl.URL(
-    `../../../../../protocols/host/fixtures/${name}`,
+    `../../../../../../protocols/host/fixtures/${name}`,
     import.meta.url,
   );
   return NodeFs.readFileSync(path, "utf8")

package/src/core/{host → notebook/replica}/__tests__/replica.test.ts

@@ -5,7 +5,7 @@ import { type NotebookTransport, StreamStopped } from "@marimo-team/client";
 import { createStore } from "jotai";
 import { afterEach, describe, expect, it, onTestFinished, vi } from "vitest";
 import { Deferred } from "@/utils/Deferred";
-import { ConnectionState } from "../../websocket/types";
+import { ConnectionState } from "../../../session/types";
 import type { ConnectionStatus } from "../connection";
 import { createNotebookReplica } from "../replica";
 import { fakeBackend, fixtureEvents, fixtureMessages } from "./fixtures";

package/src/core/{host → notebook/replica}/index.ts

@@ -1,6 +1,6 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-/** Notebook backends and the replica that follows them. */
+/** Backend-confirmed notebook state, independent of React and transports. */
 
 export type { NotebookBackend } from "./backend";
 export {
@@ -11,4 +11,3 @@ export {
 } from "./connection";
 export { createNotebookReplica, type NotebookReplica } from "./replica";
 export type { Document } from "./resources/document";
-export { createWasmNotebookBackend } from "./wasm";

package/src/core/packages/__tests__/useSandboxController.test.tsx

@@ -17,7 +17,7 @@ import type {
 } from "@/core/network/types";
 import { filenameAtom } from "@/core/saving/file-state";
 import { store } from "@/core/state/jotai";
-import { ConnectionState } from "@/core/websocket/types";
+import { ConnectionState } from "@/core/session/types";
 import { HTTPError } from "@/utils/errors";
 import { packageDataVersionAtom } from "../package-data";
 import {

package/src/core/packages/useSandboxController.ts

@@ -8,7 +8,7 @@ import { useRequestClient } from "@/core/network/requests";
 import { useFilename } from "@/core/saving/filename";
 import { waitFor } from "@/core/state/jotai";
 import { isWasm } from "@/core/wasm/utils";
-import { ConnectionState } from "@/core/websocket/types";
+import { ConnectionState } from "@/core/session/types";
 import { HTTPError, prettyError } from "@/utils/errors";
 import { invalidatePackageData } from "./package-data";
 import {

package/src/core/run-app.tsx

@@ -21,8 +21,8 @@ import { RuntimeState } from "./kernel/RuntimeState";
 import { getSessionId } from "./kernel/session";
 import { connectionAtom } from "./network/connection";
 import { useRequestClient } from "./network/requests";
-import { isAppConnecting } from "./websocket/connection-utils";
-import { useNotebookSession } from "./websocket/useNotebookSession";
+import { isAppConnecting } from "./session/connection-utils";
+import { useNotebookSession } from "./session/useNotebookSession";
 
 interface AppProps {
   appConfig: AppConfig;
@@ -49,7 +49,7 @@ export const RunApp: React.FC<AppProps> = ({ appConfig, hideHeader }) => {
     return () => {
       RuntimeState.INSTANCE.stop();
     };
-  }, []);
+  }, [sendComponentValues]);
 
   const { connection, reconnect } = useNotebookSession({
     autoInstantiate: true,

package/src/core/runtime/__tests__/adapter.test.ts

@@ -6,7 +6,7 @@ import { islandsInitializedAtom } from "../../islands/state";
 import { kernelStateAtom } from "../../kernel/state";
 import { connectionAtom } from "../../network/connection";
 import { wasmInitStateAtom } from "../../wasm/state";
-import { ConnectionState } from "../../websocket/types";
+import { ConnectionState } from "../../session/types";
 import {
   canRunCellsAtom,
   islandsAdapter,

package/src/core/runtime/adapter.ts

@@ -8,7 +8,7 @@ import { connectionAtom } from "../network/connection";
 import { isStaticNotebook } from "../static/static-state";
 import { wasmInitStateAtom } from "../wasm/state";
 import { isWasm } from "../wasm/utils";
-import { ConnectionState } from "../websocket/types";
+import { ConnectionState } from "../session/types";
 
 export type RuntimeKind = "wasm" | "remote" | "static" | "islands";
 

package/src/core/runtime/config.ts

@@ -4,8 +4,8 @@ import useEvent from "react-use-event-hook";
 import { Logger } from "@/utils/Logger";
 import { connectionAtom } from "../network/connection";
 import { store } from "../state/jotai";
-import { isAppNotStarted } from "../websocket/connection-utils";
-import { ConnectionState } from "../websocket/types";
+import { isAppNotStarted } from "../session/connection-utils";
+import { ConnectionState } from "../session/types";
 import { RuntimeManager } from "./runtime";
 import type { RuntimeConfig } from "./types";
 

package/src/core/saving/filename.ts

@@ -9,7 +9,7 @@ import { getAppConfig } from "../config/config";
 import { KnownQueryParams } from "../constants";
 import { connectionAtom } from "../network/connection";
 import { useRequestClient } from "../network/requests";
-import { ConnectionState } from "../websocket/types";
+import { ConnectionState } from "../session/types";
 import { filenameAtom } from "./file-state";
 
 export function useFilename() {

package/src/core/saving/save-component.tsx

@@ -34,7 +34,7 @@ import { layoutStateAtom } from "../layout/state";
 import { kioskModeAtom } from "../mode";
 import { connectionAtom } from "../network/connection";
 import { useRequestClient } from "../network/requests";
-import { ConnectionState } from "../websocket/types";
+import { ConnectionState } from "../session/types";
 import { isWasm } from "../wasm/utils";
 import { filenameAtom } from "./file-state";
 import { useFilename, useUpdateFilename } from "./filename";

package/src/core/saving/useAutoSave.ts

@@ -2,7 +2,7 @@
 import { useEffect, useRef } from "react";
 import type { UserConfig } from "../config/config-schema";
 import type { CellConfig } from "../network/types";
-import { type ConnectionStatus, ConnectionState } from "../websocket/types";
+import { type ConnectionStatus, ConnectionState } from "../session/types";
 
 export function useAutoSave(opts: {
   codes: string[];

package/src/core/{websocket → session}/__tests__/backend.test.ts

@@ -4,13 +4,13 @@ import { NotebookEvents, StreamStopped } from "@marimo-team/client";
 import { createStore } from "jotai/vanilla";
 import { afterEach, describe, expect, it, onTestFinished, vi } from "vitest";
 import { Deferred } from "@/utils/Deferred";
-import { ConnectionState } from "../../host/connection";
+import { ConnectionState } from "../../notebook/replica/connection";
 import {
   type FakeEdit,
   type FakeSave,
   fakeBackend,
-} from "../../host/__tests__/fixtures";
-import { createNotebookReplica } from "../../host/replica";
+} from "../../notebook/replica/__tests__/fixtures";
+import { createNotebookReplica } from "../../notebook/replica/replica";
 import { createEditorBackend } from "../backend";
 
 const wasm = vi.hoisted(() => ({ enabled: false, create: vi.fn() }));

package/src/core/{websocket → session}/__tests__/useNotebookReplica.hook.test.tsx

@@ -9,9 +9,12 @@ import type { NotebookTransport } from "@marimo-team/client";
 import { ConnectionState } from "../types";
 import { createEditorBackend } from "../backend";
 import { useNotebookReplica } from "../useNotebookReplica";
-import { fakeBackend, fixtureMessages } from "../../host/__tests__/fixtures";
-import { agentAttachmentsAtom } from "../../attachments/state";
-import { notebookReplicaAtom } from "../state";
+import {
+  fakeBackend,
+  fixtureMessages,
+} from "../../notebook/replica/__tests__/fixtures";
+import { useNotebookAttachments } from "@/core/notebook";
+import { activeNotebookReplicaAtom } from "../active-notebook";
 
 vi.mock("../backend", () => ({ createEditorBackend: vi.fn() }));
 
@@ -75,14 +78,25 @@ describe("useNotebookReplica", () => {
       fakeBackend({ transport: fake.transport }),
     );
     const { result, store, unmount } = setup();
+    const attachments = renderHook(
+      () => ({
+        all: useNotebookAttachments(),
+        agents: useNotebookAttachments("agent"),
+      }),
+      {
+        wrapper: ({ children }: PropsWithChildren) => (
+          <Provider store={store}>{children}</Provider>
+        ),
+      },
+    );
     await act(async () => Promise.resolve());
-    expect(store.get(notebookReplicaAtom)).toBe(result.current);
+    expect(store.get(activeNotebookReplicaAtom)).toBe(result.current);
     const [connection] = fake.opened;
     act(() => {
       connection.open();
       fixtureMessages("notebook-events.sse").forEach(connection.receive);
     });
-    expect(store.get(agentAttachmentsAtom)).toEqual([
+    expect(attachments.result.current.agents).toEqual([
       {
         id: "a2",
         notebook_id: "n1",
@@ -91,9 +105,37 @@ describe("useNotebookReplica", () => {
         since: "2026-10-01T09:20:03Z",
       },
     ]);
+    expect(attachments.result.current.all).toEqual(
+      store.get(result.current.attachmentsAtom),
+    );
+    expect(attachments.result.current.all.length).toBeGreaterThan(
+      attachments.result.current.agents.length,
+    );
+    const agents = attachments.result.current.agents;
+    const browser = attachments.result.current.all.find(
+      (attachment) => attachment.kind === "browser",
+    )!;
+    act(() =>
+      connection.receive({
+        event: "attachment",
+        data: { attachment: { ...browser, name: "Firefox" } },
+      }),
+    );
+    expect(attachments.result.current.agents).toBe(agents);
+
+    const renamedAgent = { ...agents[0], name: "Renamed agent" };
+    act(() =>
+      connection.receive({
+        event: "attachment",
+        data: { attachment: renamedAgent },
+      }),
+    );
+    expect(attachments.result.current.agents).toEqual([renamedAgent]);
+
     unmount();
-    expect(store.get(notebookReplicaAtom)).toBeNull();
-    expect(store.get(agentAttachmentsAtom)).toEqual([]);
+    expect(store.get(activeNotebookReplicaAtom)).toBeNull();
+    expect(attachments.result.current).toEqual({ all: [], agents: [] });
+    attachments.unmount();
   });
 
   it("uses current callbacks and URL providers without reconnecting on a rerender", async () => {

package/src/core/{websocket → session}/__tests__/useNotebookSession.hook.test.tsx

@@ -50,9 +50,9 @@ import { connectionAtom, startupProgressAtom } from "../../network/connection";
 import { generateSessionId } from "../../kernel/session";
 import { RuntimeManager } from "../../runtime/runtime";
 import { WebSocketClosedReason, ConnectionState } from "../types";
-import { classifyCloseEvent } from "../connection-state";
+import { classifyCloseEvent } from "../../transports/connection-state";
 import { createEditorBackend } from "../backend";
-import { fakeBackend } from "../../host/__tests__/fixtures";
+import { fakeBackend } from "../../notebook/replica/__tests__/fixtures";
 import { useNotebookSession } from "../useNotebookSession";
 import { type NotebookTransport, StreamStopped } from "@marimo-team/client";
 

package/src/core/{websocket/state.ts → session/active-notebook.ts}

@@ -1,7 +1,7 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
 import { atom } from "jotai";
-import type { NotebookReplica } from "../host";
+import type { NotebookReplica } from "../notebook/replica";
 
 /** The replica owned by the mounted notebook session. */
-export const notebookReplicaAtom = atom<NotebookReplica | null>(null);
+export const activeNotebookReplicaAtom = atom<NotebookReplica | null>(null);

package/src/core/{websocket → session}/backend.ts

@@ -4,15 +4,11 @@
 
 import { NotebookClient, StreamStopped } from "@marimo-team/client";
 import type { NotebookId } from "@marimo-team/client/protocol";
-import type { NotebookBackend } from "../host";
+import type { NotebookBackend } from "../notebook/replica";
 import { PyodideBridge } from "../wasm/bridge";
 import { isWasm } from "../wasm/utils";
-import {
-  idle,
-  sessionSse,
-  websocket,
-  withHeaders,
-} from "./transports/notebook";
+import { sessionSse, websocket, withHeaders } from "../transports/server";
+import { idle } from "../transports/static";
 
 /** How the editor follows its notebook. */
 export interface EditorBackendOptions {

package/src/core/{websocket → session}/types.ts

@@ -1,12 +1,12 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import type { ConnectionStatus as HostConnectionStatus } from "../host";
+import type { ConnectionStatus as HostConnectionStatus } from "../notebook/replica";
 import type { NotificationMessageData } from "../kernel/messages";
 
 export {
   ConnectionState,
   ConnectionClosedReason as WebSocketClosedReason,
-} from "../host";
+} from "../notebook/replica";
 
 export type ConnectionPhase =
   | NotificationMessageData<"startup-progress">["phase"]

package/src/core/{websocket → session}/useNotebookReplica.ts

@@ -9,8 +9,11 @@ import type {
   NotificationMessage,
   NotificationPayload,
 } from "../kernel/messages";
-import { createNotebookReplica, type NotebookReplica } from "../host";
-import { notebookReplicaAtom } from "./state";
+import {
+  createNotebookReplica,
+  type NotebookReplica,
+} from "../notebook/replica";
+import { activeNotebookReplicaAtom } from "./active-notebook";
 import { createEditorBackend, type EditorBackendOptions } from "./backend";
 
 type Options = EditorBackendOptions & {
@@ -51,7 +54,7 @@ export function useNotebookReplica(options: Options): NotebookReplica {
   );
 
   useEffect(() => {
-    store.set(notebookReplicaAtom, replica);
+    store.set(activeNotebookReplicaAtom, replica);
     const disconnect = replica.connect(backend, {
       // The host validates the envelope, not the native notification contents.
       onNotification: (notification) =>
@@ -59,8 +62,8 @@ export function useNotebookReplica(options: Options): NotebookReplica {
     });
     return () => {
       disconnect();
-      if (store.get(notebookReplicaAtom) === replica) {
-        store.set(notebookReplicaAtom, null);
+      if (store.get(activeNotebookReplicaAtom) === replica) {
+        store.set(activeNotebookReplicaAtom, null);
       }
     };
   }, [replica, backend, store, onNotification]);

package/src/core/{websocket → session}/useNotebookSession.tsx

@@ -12,7 +12,7 @@ import type {
   NotificationMessageData,
   NotificationPayload,
 } from "@/core/kernel/messages";
-import { useNotebookReplica } from "@/core/websocket/useNotebookReplica";
+import { useNotebookReplica } from "@/core/session/useNotebookReplica";
 import { renderHTML } from "@/plugins/core/RenderHTML";
 import {
   handleWidgetMessage,

package/src/core/{websocket → transports}/__tests__/connection-state.test.ts

@@ -2,7 +2,7 @@
 
 import { afterEach, describe, expect, it, vi } from "vitest";
 import { Logger } from "@/utils/Logger";
-import { WebSocketClosedReason, ConnectionState } from "../types";
+import { WebSocketClosedReason, ConnectionState } from "../../session/types";
 import { classifyCloseEvent } from "../connection-state";
 
 function classify(reason: string | undefined) {

package/src/core/{websocket/transports/__tests__/notebook.test.ts → transports/__tests__/server.test.ts}

@@ -7,11 +7,14 @@ import {
 } from "@marimo-team/client";
 import { createStore } from "jotai/vanilla";
 import { afterEach, describe, expect, it, onTestFinished, vi } from "vitest";
-import { createNotebookReplica } from "../../../host/replica";
-import { fakeBackend, fixtureMessages } from "../../../host/__tests__/fixtures";
-import { ConnectionState } from "../../../host/connection";
+import { createNotebookReplica } from "../../notebook/replica/replica";
+import {
+  fakeBackend,
+  fixtureMessages,
+} from "../../notebook/replica/__tests__/fixtures";
+import { ConnectionState } from "../../notebook/replica/connection";
 import { Deferred } from "@/utils/Deferred";
-import { sessionSse, websocket } from "../notebook";
+import { sessionSse, websocket } from "../server";
 
 const encoder = new TextEncoder();
 

package/src/core/{host → transports}/__tests__/static.test.ts

@@ -3,12 +3,12 @@
 import { createNotebookReducer, NotebookEvents } from "@marimo-team/client";
 import * as Protocol from "@marimo-team/client/protocol";
 import { afterEach, assert, describe, expect, it, vi } from "vitest";
-import { document as documentResource } from "../resources/document";
+import { document as documentResource } from "../../notebook/replica/resources/document";
 import {
   createStaticNotebookBackend,
   type StaticNotebookSnapshot,
 } from "../static";
-import { fixtureEvents } from "./fixtures";
+import { fixtureEvents } from "../../notebook/replica/__tests__/fixtures";
 
 function snapshot(): StaticNotebookSnapshot {
   const events = fixtureEvents("notebook-events.sse");

package/src/core/{host → transports}/__tests__/wasm.test.ts

@@ -4,7 +4,7 @@ import { NotebookEvents, StreamStopped } from "@marimo-team/client";
 import { afterEach, assert, describe, expect, it, vi } from "vitest";
 import { Deferred } from "@/utils/Deferred";
 import { createWasmNotebookBackend, type WasmNotebookBridge } from "../wasm";
-import { fixtureEvents } from "./fixtures";
+import { fixtureEvents } from "../../notebook/replica/__tests__/fixtures";
 
 const streams: NotebookEvents[] = [];
 afterEach(async () => {

package/src/core/{websocket → transports}/connection-state.ts

@@ -1,8 +1,11 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import type { CloseDecision } from "../host";
+import {
+  type CloseDecision,
+  ConnectionState,
+  ConnectionClosedReason,
+} from "../notebook/replica";
 import { Logger } from "@/utils/Logger";
-import { ConnectionState, WebSocketClosedReason } from "./types";
 
 // Keep server reasons in sync with ws_endpoint.py and ws/*.py.
 // SSE carries the same reasons in close events. Unclassified reasons retry.
@@ -16,7 +19,7 @@ export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
         kind: "terminal",
         status: {
           state: ConnectionState.CLOSED,
-          code: WebSocketClosedReason.KERNEL_DISCONNECTED,
+          code: ConnectionClosedReason.KERNEL_DISCONNECTED,
           reason: "kernel not found",
         },
       };
@@ -25,7 +28,7 @@ export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
         kind: "terminal",
         status: {
           state: ConnectionState.CLOSED,
-          code: WebSocketClosedReason.KERNEL_DISCONNECTED,
+          code: ConnectionClosedReason.KERNEL_DISCONNECTED,
           reason: "not authorized",
         },
       };
@@ -34,7 +37,7 @@ export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
         kind: "terminal",
         status: {
           state: ConnectionState.CLOSED,
-          code: WebSocketClosedReason.KERNEL_DISCONNECTED,
+          code: ConnectionClosedReason.KERNEL_DISCONNECTED,
           reason: "kiosk mode is not available for this session",
         },
       };
@@ -43,7 +46,7 @@ export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
         kind: "terminal",
         status: {
           state: ConnectionState.CLOSED,
-          code: WebSocketClosedReason.KERNEL_STARTUP_ERROR,
+          code: ConnectionClosedReason.KERNEL_STARTUP_ERROR,
           reason: "Failed to start kernel sandbox",
         },
       };

package/src/core/{websocket/transports/notebook.ts → transports/server.ts}

@@ -7,7 +7,7 @@ import {
   type StreamUrl,
 } from "@marimo-team/client";
 import { jsonParseWithSpecialChar } from "@/utils/json/json-parser";
-import { classifyCloseEvent } from "../connection-state";
+import { classifyCloseEvent } from "./connection-state";
 
 /** The server can be slow to accept when many notebooks are open. */
 const CONNECT_MS = 10_000;
@@ -150,10 +150,3 @@ export function sessionSse(
     }),
   );
 }
-
-/** A connection that opens and delivers nothing, for static notebooks. */
-export const idle: NotebookTransport = ({ signal }, { open }) =>
-  new Promise((settle) => {
-    open();
-    signal.addEventListener("abort", () => settle(), { once: true });
-  });