@marimo-team/islands 0.23.10-dev2 → 0.23.10-dev20

package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap

@@ -80,7 +80,6 @@ exports[`cell reducer > can initialize stdin 3`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -110,7 +109,6 @@ exports[`cell reducer > can initialize stdin 4`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -147,7 +145,6 @@ exports[`cell reducer > can initialize stdin 5`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -184,7 +181,6 @@ exports[`cell reducer > can initialize stdin 6`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -227,7 +223,6 @@ exports[`cell reducer > can initialize stdin 7`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -271,7 +266,6 @@ exports[`cell reducer > can initialize stdin 8`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -321,7 +315,6 @@ exports[`cell reducer > can initialize stdin 9`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -387,7 +380,6 @@ exports[`cell reducer > can receive console when the cell is idle and will clear
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -424,7 +416,6 @@ exports[`cell reducer > can receive console when the cell is idle and will clear
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -454,7 +445,6 @@ exports[`cell reducer > can receive console when the cell is idle and will clear
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -491,7 +481,6 @@ exports[`cell reducer > can receive console when the cell is idle and will clear
   "output": null,
   "runElapsedTimeMs": 2000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -521,7 +510,6 @@ exports[`cell reducer > can run a stale cell 1`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -551,7 +539,6 @@ exports[`cell reducer > can run a stale cell 2`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": true,
   "status": "queued",
@@ -581,7 +568,6 @@ exports[`cell reducer > can run a stopped cell 1`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -611,7 +597,6 @@ exports[`cell reducer > can run a stopped cell 2`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -652,7 +637,6 @@ exports[`cell reducer > can run a stopped cell 3`] = `
   },
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -693,7 +677,6 @@ exports[`cell reducer > can run a stopped cell 4`] = `
   },
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -723,7 +706,6 @@ exports[`cell reducer > can run a stopped cell 5`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 40,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -840,7 +822,6 @@ exports[`cell reducer > can run cell and receive cell messages 4`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "queued",
@@ -870,7 +851,6 @@ exports[`cell reducer > can run cell and receive cell messages 5`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -907,7 +887,6 @@ exports[`cell reducer > can run cell and receive cell messages 6`] = `
   "output": null,
   "runElapsedTimeMs": null,
   "runStartTimestamp": 20,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "running",
@@ -949,7 +928,6 @@ exports[`cell reducer > can run cell and receive cell messages 7`] = `
   },
   "runElapsedTimeMs": 13000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -992,7 +970,6 @@ import numpy",
   },
   "runElapsedTimeMs": 13000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -1034,7 +1011,6 @@ exports[`cell reducer > can run cell and receive cell messages 9`] = `
   },
   "runElapsedTimeMs": 13000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -1077,7 +1053,6 @@ import numpy",
   },
   "runElapsedTimeMs": 13000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -1127,7 +1102,6 @@ import numpy",
   },
   "runElapsedTimeMs": 11000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -1175,7 +1149,6 @@ import numpy",
   },
   "runElapsedTimeMs": 11000,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",
@@ -1303,7 +1276,6 @@ exports[`cell reducer > errors reset status to idle 4`] = `
   },
   "runElapsedTimeMs": null,
   "runStartTimestamp": null,
-  "serialization": undefined,
   "serializedEditorState": null,
   "staleInputs": false,
   "status": "idle",

package/src/core/cells/__tests__/cell.test.ts

@@ -1,9 +1,10 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 import { describe, expect, it } from "vitest";
-import type { OutputMessage } from "@/core/kernel/messages";
+import type { CellMessage, OutputMessage } from "@/core/kernel/messages";
 import type { RuntimeState } from "@/core/network/types";
 import type { Seconds } from "@/utils/time";
-import { outputIsLoading, outputIsStale } from "../cell";
+import { outputIsLoading, outputIsStale, transitionCell } from "../cell";
+import { createCellRuntimeState } from "../types";
 
 const STATUSES: RuntimeState[] = [
   "queued",
@@ -191,3 +192,29 @@ describe("outputIsStale", () => {
     expect(outputIsStale(cell, edited)).toBe(false);
   });
 });
+
+describe("transitionCell serialization", () => {
+  function cellMessage(serialization?: string | null): CellMessage {
+    return {
+      cell_id: "cell-1",
+      ...(serialization === undefined ? {} : { serialization }),
+    } as CellMessage;
+  }
+
+  it("leaves the hint unchanged when serialization is omitted", () => {
+    const cell = createCellRuntimeState({ serialization: "Valid" });
+    expect(transitionCell(cell, cellMessage()).serialization).toBe("Valid");
+  });
+
+  it("clears the hint when serialization is null", () => {
+    const cell = createCellRuntimeState({ serialization: "Valid" });
+    expect(transitionCell(cell, cellMessage(null)).serialization).toBeNull();
+  });
+
+  it("sets the hint when serialization is a string", () => {
+    const cell = createCellRuntimeState();
+    expect(transitionCell(cell, cellMessage("Valid")).serialization).toBe(
+      "Valid",
+    );
+  });
+});

package/src/core/cells/cell.ts

@@ -73,7 +73,11 @@ export function transitionCell(
   nextCell.output = message.output ?? nextCell.output;
   nextCell.staleInputs = message.stale_inputs ?? nextCell.staleInputs;
   nextCell.status = message.status ?? nextCell.status;
-  nextCell.serialization = message.serialization;
+  // Tri-state partial update: an omitted field (undefined) leaves the hint
+  // unchanged; null explicitly clears it; a string sets it.
+  if (message.serialization !== undefined) {
+    nextCell.serialization = message.serialization;
+  }
 
   let didInterruptFromThisMessage = false;
 

package/src/core/codemirror/go-to-definition/__tests__/utils.test.ts

@@ -39,6 +39,43 @@ afterEach(() => {
 });
 
 describe("goToDefinitionAtCursorPosition", () => {
+  test("jumps to a reactive variable definition in another cell", async () => {
+    const definingCell = cellId("defining-cell");
+    const usageCell = cellId("usage-cell");
+    const definingCode = "a = 10";
+    const usageCode = "print(a)";
+
+    const definingView = createEditor(definingCode, definingCode.length);
+    const usageView = createEditor(usageCode, usageCode.indexOf("a"));
+    views.push(definingView, usageView);
+
+    const notebook = initialNotebookState();
+    notebook.cellHandles[definingCell] = {
+      current: { editorView: definingView, editorViewOrNull: definingView },
+    };
+    notebook.cellHandles[usageCell] = {
+      current: { editorView: usageView, editorViewOrNull: usageView },
+    };
+
+    store.set(notebookAtom, notebook);
+    store.set(variablesAtom, {
+      [variableName("a")]: {
+        dataType: "int",
+        declaredBy: [definingCell],
+        name: variableName("a"),
+        usedBy: [usageCell],
+        value: "10",
+      },
+    });
+
+    const result = goToDefinitionAtCursorPosition(usageView);
+
+    expect(result).toBe(true);
+    await tick();
+    expect(definingView.state.selection.main.head).toBe(0);
+    expect(usageView.state.selection.main.head).toBe(usageCode.indexOf("a"));
+  });
+
   test("prefers the current-cell local definition over a reactive global", async () => {
     const globalCell = cellId("global-cell");
     const localCell = cellId("local-cell");

package/src/core/codemirror/go-to-definition/commands.ts

@@ -28,10 +28,11 @@ interface VariableDeclaration {
 }
 
 function goToPosition(view: EditorView, from: number): void {
-  view.focus();
-  // Wait for the next frame, otherwise codemirror will
-  // add a cursor from a pointer click.
+  // Focus on the next frame: a synchronous focus is a no-op while a Radix
+  // context menu still owns focus, and codemirror would otherwise add a
+  // cursor from the pointer click.
   requestAnimationFrame(() => {
+    view.focus();
     view.dispatch({
       selection: {
         anchor: from,
@@ -403,22 +404,29 @@ function findScopedDefinitionPosition(
 }
 
 /**
- * This function will select the first occurrence of the given variable name,
- * for a given editor view.
+ * This function selects a scoped definition for the given variable name, when
+ * a usage position is available, or optionally falls back to the first matching
+ * variable name in the given editor view.
  * @param view The editor view which contains the variable name.
  * @param variableName The name of the variable to select, if found in the editor.
  * @param usagePosition The position of the variable usage, if available.
+ * @param fallbackToFirstMatch Whether to fall back to the first matching
+ * variable name when no scoped definition is found. Defaults to true.
  */
 export function goToVariableDefinition(
   view: EditorView,
   variableName: string,
   usagePosition?: number,
+  fallbackToFirstMatch = true,
 ): boolean {
   const { state } = view;
-  const from =
-    (usagePosition !== undefined
-      ? findScopedDefinitionPosition(state, variableName, usagePosition)
-      : null) ?? findFirstMatchingVariable(state, variableName);
+  let from: number | null = null;
+  if (usagePosition !== undefined) {
+    from = findScopedDefinitionPosition(state, variableName, usagePosition);
+  }
+  if (from === null && fallbackToFirstMatch) {
+    from = findFirstMatchingVariable(state, variableName);
+  }
 
   if (from === null) {
     return false;

package/src/core/codemirror/go-to-definition/utils.ts

@@ -82,6 +82,7 @@ export function goToDefinition(
       view,
       variableName,
       usagePosition,
+      false,
     );
     if (foundLocally) {
       return true;

package/src/core/codemirror/language/languages/sql/utils.ts

@@ -14,6 +14,7 @@ import {
 } from "@codemirror/lang-sql";
 import {
   BigQueryDialect,
+  DremioDialect,
   DuckDBDialect,
 } from "@marimo-team/codemirror-sql/dialects";
 import type { DataSourceConnection } from "@/core/kernel/messages";
@@ -97,6 +98,8 @@ export function guessDialect(
       return PLSQL;
     case "bigquery":
       return BigQueryDialect;
+    case "dremio":
+      return DremioDialect;
     case "timescaledb":
       return PostgreSQL; // TimescaleDB is a PostgreSQL dialect
     case "awsathena":
@@ -116,7 +119,6 @@ export function guessDialect(
     case "spark":
     case "databricks":
     case "datafusion":
-    case "dremio":
       Logger.debug("Unsupported dialect", { dialect });
       return ModifiedStandardSQL;
     default:

package/src/core/datasets/data-source-connections.ts

@@ -169,6 +169,7 @@ const {
         return {
           ...db,
           schemas: schemas,
+          schemas_resolved: true,
         };
       }),
     };
@@ -213,6 +214,7 @@ const {
             return {
               ...schema,
               tables: tables,
+              tables_resolved: true,
             };
           }),
         };

package/src/core/network/__tests__/requests-static.test.ts

@@ -0,0 +1,30 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { toast } from "@/components/ui/use-toast";
+import { createStaticRequests } from "../requests-static";
+
+vi.mock("@/components/ui/use-toast", () => ({ toast: vi.fn() }));
+
+describe("createStaticRequests static-notebook toast", () => {
+  beforeEach(() => {
+    vi.mocked(toast).mockClear();
+  });
+
+  it("requests a once-toast with a stable id on component value updates", async () => {
+    const requests = createStaticRequests();
+    await requests.sendComponentValues({} as never);
+
+    expect(toast).toHaveBeenCalledWith(
+      expect.objectContaining({ id: "static-notebook", once: true }),
+    );
+  });
+
+  it("uses the same once-toast for function requests", async () => {
+    const requests = createStaticRequests();
+    await requests.sendFunctionRequest({} as never);
+
+    expect(toast).toHaveBeenCalledWith(
+      expect.objectContaining({ id: "static-notebook", once: true }),
+    );
+  });
+});

package/src/core/network/requests-static.ts

@@ -3,6 +3,18 @@ import { toast } from "@/components/ui/use-toast";
 import { Logger } from "@/utils/Logger";
 import type { EditRequests, RunRequests } from "./types";
 
+const STATIC_NOTEBOOK_TOAST_ID = "static-notebook";
+
+function showStaticNotebookToast() {
+  toast({
+    id: STATIC_NOTEBOOK_TOAST_ID,
+    once: true,
+    title: "Static notebook",
+    description:
+      "This notebook is not connected to a kernel. Any interactive elements will not work.",
+  });
+}
+
 export function createStaticRequests(): EditRequests & RunRequests {
   const throwNotInEditMode = () => {
     throw new Error("Unreachable. Expected to be in run mode");
@@ -10,11 +22,7 @@ export function createStaticRequests(): EditRequests & RunRequests {
 
   return {
     sendComponentValues: async () => {
-      toast({
-        title: "Static notebook",
-        description:
-          "This notebook is not connected to a kernel. Any interactive elements will not work.",
-      });
+      showStaticNotebookToast();
       Logger.log("Updating UI elements is not supported in static mode");
       return null;
     },
@@ -27,11 +35,7 @@ export function createStaticRequests(): EditRequests & RunRequests {
       return null;
     },
     sendFunctionRequest: async () => {
-      toast({
-        title: "Static notebook",
-        description:
-          "This notebook is not connected to a kernel. Any interactive elements will not work.",
-      });
+      showStaticNotebookToast();
       Logger.log("Function requests are not supported in static mode");
       return null;
     },

package/src/plugins/layout/DownloadPlugin.tsx

@@ -98,7 +98,7 @@ const DownloadButton = ({
     try {
       const loadedData = await load({});
       downloadByURL(
-        loadedData.data,
+        loadedData.data || data.data,
         loadedData.filename ?? data.filename ?? undefined,
       );
     } catch (error) {