@marimo-team/frontend 0.23.11-dev9 → 0.23.12-dev0

package/src/components/datasources/datasources.tsx

@@ -83,6 +83,8 @@ import {
   areSchemasResolved,
   areTablesResolved,
   isSchemaless,
+  schemaSubtreeMatchesSearch,
+  shouldExpandDatabaseForSearch,
   sqlCode,
   tableUniqueId,
 } from "./utils";
@@ -360,7 +362,6 @@ export const DataSources: React.FC = () => {
                 key={database.name}
                 connection={connection}
                 database={database}
-                hasSearch={hasSearch}
                 searchValue={searchValue}
               />
             ))}
@@ -435,7 +436,6 @@ interface DataSourceTree {
   dialect: string;
   engineName: string;
   databaseName: string;
-  hasSearch: boolean;
   searchValue?: string;
 }
 
@@ -470,9 +470,8 @@ function buildSqlTableContext(
 const DatabaseTree: React.FC<{
   connection: DataSourceConnection;
   database: Database;
-  hasSearch: boolean;
   searchValue?: string;
-}> = ({ connection, database, hasSearch, searchValue }) => {
+}> = ({ connection, database, searchValue }) => {
   const tree = React.useMemo<DataSourceTree>(
     () => ({
       engineName: connection.name,
@@ -480,7 +479,6 @@ const DatabaseTree: React.FC<{
       dialect: connection.dialect,
       defaultSchema: connection.default_schema,
       defaultDatabase: connection.default_database,
-      hasSearch,
       searchValue,
     }),
     [
@@ -489,7 +487,6 @@ const DatabaseTree: React.FC<{
       connection.default_schema,
       connection.default_database,
       database.name,
-      hasSearch,
       searchValue,
     ],
   );
@@ -498,7 +495,7 @@ const DatabaseTree: React.FC<{
     <DatabaseItem
       engineName={connection.name}
       database={database}
-      hasSearch={hasSearch}
+      searchValue={searchValue}
     >
       <DataSourceTreeContext.Provider value={tree}>
         <SchemaList
@@ -513,18 +510,30 @@ const DatabaseTree: React.FC<{
 };
 
 const DatabaseItem: React.FC<{
-  hasSearch: boolean;
+  searchValue?: string;
   engineName: string;
   database: Database;
   children: React.ReactNode;
-}> = ({ hasSearch, engineName, database, children }) => {
-  const [isExpanded, setIsExpanded] = React.useState(false);
+}> = ({ searchValue, engineName, database, children }) => {
+  // Auto-expand during search only when a loaded schema under this database
+  // contains a matching table.
+  const expandForSearch = shouldExpandDatabaseForSearch(database, searchValue);
+  const [isExpanded, setIsExpanded] = React.useState(expandForSearch);
   const [isSelected, setIsSelected] = React.useState(false);
-  const [prevHasSearch, setPrevHasSearch] = React.useState(hasSearch);
-
-  if (prevHasSearch !== hasSearch) {
-    setPrevHasSearch(hasSearch);
-    setIsExpanded(hasSearch);
+  const [prevSearchValue, setPrevSearchValue] = React.useState(searchValue);
+  const [prevExpandForSearch, setPrevExpandForSearch] =
+    React.useState(expandForSearch);
+
+  // Re-sync when the query changes or when newly resolved data turns this
+  // branch into a match under the same query (deferred subtrees resolve
+  // asynchronously, so the decision can flip without the query changing).
+  if (
+    prevSearchValue !== searchValue ||
+    prevExpandForSearch !== expandForSearch
+  ) {
+    setPrevSearchValue(searchValue);
+    setPrevExpandForSearch(expandForSearch);
+    setIsExpanded(expandForSearch);
   }
 
   return (
@@ -657,12 +666,30 @@ interface SchemaNodeProps {
 const SchemaNode: React.FC<SchemaNodeProps> = (props) => {
   const { schema, schemaPath, depth } = props;
   const tree = useDataSourceTree();
-  const { databaseName, hasSearch, searchValue } = tree;
-  const [isExpanded, setIsExpanded] = React.useState(hasSearch);
+  const { databaseName, searchValue } = tree;
+  // Auto-expand during search only when this schema's loaded subtree contains a
+  // matching table.
+  const expandForSearch = schemaSubtreeMatchesSearch(schema, searchValue);
+  const [isExpanded, setIsExpanded] = React.useState(expandForSearch);
   const [isSelected, setIsSelected] = React.useState(false);
+  const [prevSearchValue, setPrevSearchValue] = React.useState(searchValue);
+  const [prevExpandForSearch, setPrevExpandForSearch] =
+    React.useState(expandForSearch);
   const uniqueValue = `${databaseName}:${schemaPath.join(".")}`;
   const childSchemas = schema.child_schemas ?? [];
 
+  // Re-sync when the query changes or when newly resolved data turns this
+  // subtree into a match under the same query (deferred tables/child schemas
+  // resolve asynchronously, so the decision can flip without the query changing).
+  if (
+    prevSearchValue !== searchValue ||
+    prevExpandForSearch !== expandForSearch
+  ) {
+    setPrevSearchValue(searchValue);
+    setPrevExpandForSearch(expandForSearch);
+    setIsExpanded(expandForSearch);
+  }
+
   return (
     <>
       <CommandItem

package/src/components/datasources/install-package-button.tsx

@@ -2,6 +2,7 @@
 
 import React from "react";
 import { Button } from "@/components/ui/button";
+import { useInstallAllowed } from "@/core/mode";
 import { useInstallPackages } from "@/core/packages/useInstallPackage";
 import { cn } from "@/utils/cn";
 
@@ -23,8 +24,9 @@ export const InstallPackageButton: React.FC<InstallPackageButtonProps> = ({
   onInstall,
 }) => {
   const { handleInstallPackages } = useInstallPackages();
+  const installAllowed = useInstallAllowed();
 
-  if (!packages || packages.length === 0) {
+  if (!packages || packages.length === 0 || !installAllowed) {
     return null;
   }
 

package/src/components/datasources/utils.ts

@@ -52,6 +52,51 @@ export function areChildSchemasResolved(schema: DatabaseSchema): boolean {
   return schema.child_schemas_resolved !== false;
 }
 
+/**
+ * Whether a loaded schema subtree contains a table whose name matches
+ * `searchValue`. Deferred (not-yet-fetched) tables and child schemas are
+ * treated as non-matching, so searching never triggers a lazy catalog fetch on
+ * large/remote connections.
+ */
+export function schemaSubtreeMatchesSearch(
+  schema: DatabaseSchema,
+  searchValue: string | undefined,
+): boolean {
+  const query = searchValue?.trim().toLowerCase();
+  if (!query) {
+    return false;
+  }
+  if (
+    areTablesResolved(schema) &&
+    schema.tables.some((table) => table.name.toLowerCase().includes(query))
+  ) {
+    return true;
+  }
+  if (areChildSchemasResolved(schema)) {
+    return (schema.child_schemas ?? []).some((child) =>
+      schemaSubtreeMatchesSearch(child, query),
+    );
+  }
+  return false;
+}
+
+/**
+ * Auto-expand a database row during search only when one of its loaded schemas
+ * contains a matching table. Returns false when the schema list itself is
+ * deferred.
+ */
+export function shouldExpandDatabaseForSearch(
+  database: Database,
+  searchValue: string | undefined,
+): boolean {
+  if (!searchValue?.trim() || !areSchemasResolved(database)) {
+    return false;
+  }
+  return database.schemas.some((schema) =>
+    schemaSubtreeMatchesSearch(schema, searchValue),
+  );
+}
+
 interface SqlCodeFormatter {
   /**
    * Format the table path based on dialect-specific rules

package/src/components/editor/actions/useNotebookActions.tsx

@@ -14,6 +14,7 @@ import {
   DownloadIcon,
   EditIcon,
   ExternalLinkIcon,
+  EyeIcon,
   EyeOffIcon,
   FastForwardIcon,
   FileIcon,
@@ -41,7 +42,10 @@ import {
   YoutubeIcon,
   ZapIcon,
 } from "lucide-react";
-import { settingDialogAtom } from "@/components/app-config/state";
+import {
+  settingDialogAtom,
+  useOpenSettingsToTab,
+} from "@/components/app-config/state";
 import { MarkdownIcon } from "@/components/editor/cell/code/icons";
 import { MarimoPlusIcon } from "@/components/icons/marimo-icons";
 import { useImperativeModal } from "@/components/modal/ImperativeModal";
@@ -57,7 +61,8 @@ import {
   useCellActions,
 } from "@/core/cells/cells";
 import { disabledCellIds } from "@/core/cells/utils";
-import { useResolvedMarimoConfig } from "@/core/config/config";
+import { capabilitiesAtom } from "@/core/config/capabilities";
+import { aiEnabledAtom, useResolvedMarimoConfig } from "@/core/config/config";
 import { Constants } from "@/core/constants";
 import {
   updateCellOutputsWithScreenshots,
@@ -86,7 +91,7 @@ import { Strings } from "@/utils/strings";
 import { newNotebookURL } from "@/utils/urls";
 import { useRunAllCells } from "../cell/useRunCells";
 import { useChromeActions, useChromeState } from "../chrome/state";
-import { PANELS } from "../chrome/types";
+import { isPanelHidden, PANELS } from "../chrome/types";
 import { AddConnectionDialogContent } from "../connections/add-connection-dialog";
 import { keyboardShortcutsAtom } from "../controls/keyboard-shortcuts";
 import { commandPaletteAtom } from "../controls/state";
@@ -95,8 +100,8 @@ import { LAYOUT_TYPES } from "../renderers/types";
 import { runServerSidePDFDownload } from "./pdf-export";
 import type { ActionButton } from "./types";
 import { useCopyNotebook } from "./useCopyNotebook";
-import { useHideAllMarkdownCode } from "./useHideAllMarkdownCode";
 import { useRestartKernel } from "./useRestartKernel";
+import { useSetCodeVisibility } from "./useSetCodeVisibility";
 
 const NOOP_HANDLER = (event?: Event) => {
   event?.preventDefault();
@@ -110,8 +115,10 @@ export function useNotebookActions() {
   const { selectedPanel } = useChromeState();
   const [viewState] = useAtom(viewStateAtom);
   const kioskMode = useAtomValue(kioskModeAtom);
-  const hideAllMarkdownCode = useHideAllMarkdownCode();
+  const setCodeVisibility = useSetCodeVisibility();
   const [resolvedConfig] = useResolvedMarimoConfig();
+  const capabilities = useAtomValue(capabilitiesAtom);
+  const aiEnabled = useAtomValue(aiEnabledAtom);
 
   const {
     updateCellConfig,
@@ -126,6 +133,7 @@ export function useNotebookActions() {
   const copyNotebook = useCopyNotebook(filename);
   const setCommandPaletteOpen = useSetAtom(commandPaletteAtom);
   const setSettingsDialogOpen = useSetAtom(settingDialogAtom);
+  const { handleClick: openSettings } = useOpenSettingsToTab();
   const setKeyboardShortcutsOpen = useSetAtom(keyboardShortcutsAtom);
   const {
     exportAsIPYNB,
@@ -407,20 +415,31 @@ export function useNotebookActions() {
       label: "Helper panel",
       redundant: true,
       handle: NOOP_HANDLER,
-      dropdown: PANELS.flatMap(
-        ({ type: id, Icon, hidden, additionalKeywords }) => {
-          if (hidden) {
-            return [];
-          }
-          return {
-            label: Strings.startCase(id),
-            rightElement: renderCheckboxElement(selectedPanel === id),
-            icon: <Icon size={14} strokeWidth={1.5} />,
-            handle: () => toggleApplication(id),
-            additionalKeywords,
-          };
-        },
-      ),
+      dropdown: PANELS.flatMap((panel) => {
+        // Still show the AI panel in the command palette so users can try AI
+        // features. When AI is disabled, open settings instead of the panel.
+        const openAiSettingsWhenDisabled = panel.type === "ai" && !aiEnabled;
+        if (
+          isPanelHidden({ panel, capabilities, aiEnabled }) &&
+          !openAiSettingsWhenDisabled
+        ) {
+          return [];
+        }
+        const { type: id, Icon, additionalKeywords } = panel;
+        return {
+          label: Strings.startCase(id),
+          rightElement: renderCheckboxElement(selectedPanel === id),
+          icon: <Icon size={14} strokeWidth={1.5} />,
+          handle: () => {
+            if (openAiSettingsWhenDisabled) {
+              openSettings("ai", "ai-features");
+              return;
+            }
+            toggleApplication(id);
+          },
+          additionalKeywords,
+        };
+      }),
     },
 
     {
@@ -560,11 +579,33 @@ export function useNotebookActions() {
         clearAllCellOutputs();
       },
     },
+    {
+      icon: <EyeIcon size={14} strokeWidth={1.5} />,
+      label: "Show all code",
+      hotkey: "global.showAllCode",
+      handle: () => setCodeVisibility(false, "code"),
+      redundant: true,
+    },
+    {
+      icon: <EyeOffIcon size={14} strokeWidth={1.5} />,
+      label: "Hide all code",
+      hotkey: "global.hideAllCode",
+      handle: () => setCodeVisibility(true, "code"),
+      redundant: true,
+    },
+    {
+      icon: <EyeIcon size={14} strokeWidth={1.5} />,
+      label: "Show all markdown code",
+      hotkey: "global.showAllMarkdownCode",
+      handle: () => setCodeVisibility(false, "markdown"),
+      redundant: true,
+    },
     {
       icon: <EyeOffIcon size={14} strokeWidth={1.5} />,
       label: "Hide all markdown code",
-      handle: hideAllMarkdownCode,
-      redundant: true, // hidden by default
+      hotkey: "global.hideAllMarkdownCode",
+      handle: () => setCodeVisibility(true, "markdown"),
+      redundant: true,
     },
     {
       icon: <ChevronRightCircleIcon size={14} strokeWidth={1.5} />,

package/src/components/editor/actions/useSetCodeVisibility.ts

@@ -0,0 +1,57 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { useCallback } from "react";
+import { getNotebook, useCellActions } from "@/core/cells/cells";
+import type { CellId } from "@/core/cells/ids";
+import { MarkdownLanguageAdapter } from "@/core/codemirror/language/languages/markdown";
+import { useRequestClient } from "@/core/network/requests";
+import type { CellConfig } from "@/core/network/types";
+import { Objects } from "@/utils/objects";
+
+const markdownAdapter = new MarkdownLanguageAdapter();
+
+/** The kind of cell a visibility change applies to. */
+export type CellCodeKind = "code" | "markdown";
+
+const cellCodeKind = (code: string): CellCodeKind =>
+  markdownAdapter.isSupported(code) ? "markdown" : "code";
+
+/**
+ * Hook returning a callback that hides or shows the code editor for every
+ * cell of the given kind.
+ */
+export const useSetCodeVisibility = () => {
+  const { updateCellConfig } = useCellActions();
+  const { saveCellConfig } = useRequestClient();
+
+  return useCallback(
+    async (hidden: boolean, kind: CellCodeKind) => {
+      const notebook = getNotebook();
+
+      const newConfigs: Record<CellId, Partial<CellConfig>> = {};
+
+      for (const cellId of notebook.cellIds.inOrderIds) {
+        const cell = notebook.cellData[cellId];
+        if (cell === undefined || cell.config.hide_code === hidden) {
+          continue;
+        }
+        if (cellCodeKind(cell.code) !== kind) {
+          continue;
+        }
+        newConfigs[cellId] = { hide_code: hidden };
+      }
+
+      const entries = Objects.entries(newConfigs);
+      if (entries.length === 0) {
+        return;
+      }
+
+      await saveCellConfig({ configs: newConfigs });
+
+      for (const [cellId, config] of entries) {
+        updateCellConfig({ cellId, config });
+      }
+    },
+    [updateCellConfig, saveCellConfig],
+  );
+};

package/src/components/editor/header/__tests__/status.test.tsx

@@ -67,10 +67,6 @@ describe("StatusOverlay disconnect indicator", () => {
   });
 
   it.each([
-    [
-      WebSocketClosedReason.MALFORMED_QUERY,
-      "the kernel did not recognize a request; please file a bug with marimo",
-    ],
     [
       WebSocketClosedReason.KERNEL_STARTUP_ERROR,
       "Failed to start kernel sandbox",

package/src/components/editor/header/status.tsx

@@ -22,9 +22,8 @@ export const StatusOverlay: React.FC<{
   const { mode } = useAtomValue(viewStateAtom);
   const isClosed = connection.state === WebSocketState.CLOSED;
   const isOpen = connection.state === WebSocketState.OPEN;
-  // Only KERNEL_DISCONNECTED is recoverable by a retry. Other terminal
-  // reasons (MALFORMED_QUERY, KERNEL_STARTUP_ERROR) would deterministically
-  // fail the same way.
+  // Only KERNEL_DISCONNECTED is recoverable by a retry. KERNEL_STARTUP_ERROR
+  // would deterministically fail the same way.
   const canReconnect =
     isClosed && connection.code === WebSocketClosedReason.KERNEL_DISCONNECTED;
 

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

@@ -2920,6 +2920,39 @@ describe("cell reducer", () => {
       expect(state.untouchedNewCells.has(newCellId)).toBe(true);
     });
 
+    it("adds interactively-created hidden cell with boilerplate code to untouchedNewCells", () => {
+      // Markdown cells are created with hideCode and non-empty default code
+      // (e.g. `mo.md(r"""\n""")`). They are user-initiated, so their editor
+      // should be shown until first blur.
+      actions.createNewCell({
+        cellId: "__end__",
+        before: false,
+        code: 'mo.md(r"""\n""")',
+        hideCode: true,
+        autoFocus: true,
+      });
+
+      const newCellId =
+        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
+      expect(state.untouchedNewCells.has(newCellId)).toBe(true);
+    });
+
+    it("does not add programmatically-created hidden cell with code to untouchedNewCells", () => {
+      // Cells created by the kernel (e.g. via code_mode) carry code and
+      // autoFocus=false; their hide_code must take effect immediately.
+      actions.createNewCell({
+        cellId: "__end__",
+        before: false,
+        code: "x = 1",
+        hideCode: true,
+        autoFocus: false,
+      });
+
+      const newCellId =
+        state.cellIds.inOrderIds[state.cellIds.inOrderIds.length - 1];
+      expect(state.untouchedNewCells.has(newCellId)).toBe(false);
+    });
+
     it("does not add cell to untouchedNewCells when hideCode is false", () => {
       actions.createNewCell({
         cellId: "__end__",

package/src/core/cells/cells.ts

@@ -276,7 +276,7 @@ const {
       },
       scrollKey: autoFocus ? newCellId : null,
       untouchedNewCells:
-        hideCode && !code
+        hideCode && autoFocus
           ? new Set([...state.untouchedNewCells, newCellId])
           : state.untouchedNewCells,
     };

package/src/core/config/config-schema.ts

@@ -45,7 +45,12 @@ export const DEFAULT_AI_MODEL = "openai/gpt-4o";
 const AUTO_DOWNLOAD_FORMATS = ["html", "markdown", "ipynb"] as const;
 
 export type CopilotMode = NonNullable<schemas["AiConfig"]["mode"]>;
-export const COPILOT_MODES: CopilotMode[] = ["manual", "ask", "agent"];
+export const COPILOT_MODES: CopilotMode[] = [
+  "manual",
+  "ask",
+  "agent",
+  "code_mode",
+];
 
 const AiConfigSchema = z
   .object({

package/src/core/edit-app.tsx

@@ -20,6 +20,7 @@ import {
   useRunAllCells,
   useRunStaleCells,
 } from "../components/editor/cell/useRunCells";
+import { useSetCodeVisibility } from "../components/editor/actions/useSetCodeVisibility";
 import { CellArray } from "../components/editor/renderers/cell-array";
 import { CellsRenderer } from "../components/editor/renderers/cells-renderer";
 import { useHotkey } from "../hooks/useHotkey";
@@ -112,6 +113,7 @@ export const EditApp: React.FC<AppProps> = ({
   const runStaleCells = useRunStaleCells();
   const runAllCells = useRunAllCells();
   const togglePresenting = useTogglePresenting();
+  const setCodeVisibility = useSetCodeVisibility();
 
   // HOTKEYS
   useHotkey("global.runStale", () => {
@@ -126,6 +128,18 @@ export const EditApp: React.FC<AppProps> = ({
   useHotkey("global.runAll", () => {
     runAllCells();
   });
+  useHotkey("global.showAllCode", () => {
+    setCodeVisibility(false, "code");
+  });
+  useHotkey("global.hideAllCode", () => {
+    setCodeVisibility(true, "code");
+  });
+  useHotkey("global.showAllMarkdownCode", () => {
+    setCodeVisibility(false, "markdown");
+  });
+  useHotkey("global.hideAllMarkdownCode", () => {
+    setCodeVisibility(true, "markdown");
+  });
   useHotkey("global.collapseAllSections", () => {
     collapseAllCells();
   });

package/src/core/hotkeys/hotkeys.ts

@@ -370,6 +370,26 @@ const DEFAULT_HOT_KEY = {
     group: "Other",
     key: "Mod-j",
   },
+  "global.showAllCode": {
+    name: "Show all code",
+    group: "Editing",
+    key: NOT_SET,
+  },
+  "global.hideAllCode": {
+    name: "Hide all code",
+    group: "Editing",
+    key: NOT_SET,
+  },
+  "global.showAllMarkdownCode": {
+    name: "Show all markdown code",
+    group: "Editing",
+    key: NOT_SET,
+  },
+  "global.hideAllMarkdownCode": {
+    name: "Hide all markdown code",
+    group: "Editing",
+    key: NOT_SET,
+  },
   "global.collapseAllSections": {
     name: "Collapse all sections",
     group: "Editing",

package/src/core/websocket/__tests__/useMarimoKernelConnection.test.ts

@@ -32,7 +32,6 @@ describe("classifyCloseEvent", () => {
 
   describe("terminal closes (server-initiated)", () => {
     it.each([
-      "MARIMO_WRONG_KERNEL_ID",
       "MARIMO_NO_FILE_KEY",
       "MARIMO_NO_SESSION_ID",
       "MARIMO_NO_SESSION",
@@ -49,18 +48,6 @@ describe("classifyCloseEvent", () => {
       }
     });
 
-    it("MARIMO_MALFORMED_QUERY → terminal but does NOT close transport", () => {
-      const decision = classify("MARIMO_MALFORMED_QUERY");
-      expect(decision.kind).toBe("terminal");
-      expect(decision.status).toMatchObject({
-        state: WebSocketState.CLOSED,
-        code: WebSocketClosedReason.MALFORMED_QUERY,
-      });
-      if (decision.kind === "terminal") {
-        expect(decision.closeTransport).toBe(false);
-      }
-    });
-
     it("MARIMO_KERNEL_STARTUP_ERROR → terminal + closeTransport", () => {
       const decision = classify("MARIMO_KERNEL_STARTUP_ERROR");
       expect(decision.kind).toBe("terminal");

package/src/core/websocket/types.ts

@@ -14,7 +14,6 @@ export type WebSocketState =
 
 export const WebSocketClosedReason = {
   KERNEL_DISCONNECTED: "KERNEL_DISCONNECTED",
-  MALFORMED_QUERY: "MALFORMED_QUERY",
   KERNEL_STARTUP_ERROR: "KERNEL_STARTUP_ERROR",
 } as const;
 

package/src/core/websocket/useMarimoKernelConnection.tsx

@@ -82,12 +82,10 @@ const SUPPORTS_LAZY_KERNELS = true;
 // (marimo/_server/api/endpoints/ws_endpoint.py and ws/*.py). Keep in sync with
 // the backend literals.
 export type CloseReason =
-  | "MARIMO_WRONG_KERNEL_ID"
   | "MARIMO_NO_FILE_KEY"
   | "MARIMO_NO_SESSION_ID"
   | "MARIMO_NO_SESSION"
   | "MARIMO_SHUTDOWN"
-  | "MARIMO_MALFORMED_QUERY"
   | "MARIMO_KERNEL_STARTUP_ERROR"
   | typeof TRANSPORT_EXHAUSTED_REASON;
 
@@ -107,7 +105,6 @@ export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
           reason: "kernel not found",
         },
       };
-    case "MARIMO_WRONG_KERNEL_ID":
     case "MARIMO_NO_FILE_KEY":
     case "MARIMO_NO_SESSION_ID":
     case "MARIMO_NO_SESSION":
@@ -121,17 +118,6 @@ export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
         },
         closeTransport: true,
       };
-    case "MARIMO_MALFORMED_QUERY":
-      return {
-        kind: "terminal",
-        status: {
-          state: WebSocketState.CLOSED,
-          code: WebSocketClosedReason.MALFORMED_QUERY,
-          reason:
-            "the kernel did not recognize a request; please file a bug with marimo",
-        },
-        closeTransport: false,
-      };
     case "MARIMO_KERNEL_STARTUP_ERROR":
       return {
         kind: "terminal",

package/src/css/app/print.css

@@ -16,6 +16,10 @@ body.printing #App {
     height: fit-content !important;
     overflow-y: visible !important;
     overflow-x: hidden !important;
+
+    /* With @page margin:0 the themed background reaches the paper edge;
+       this padding insets printed content from all four edges. */
+    padding: 2rem !important;
   }
 
   h1,
@@ -69,4 +73,8 @@ body.printing #App {
     padding-right: 0 !important;
     max-height: none !important;
   }
+
+  @page {
+    margin: 0;
+  }
 }

package/src/css/md.css

@@ -49,6 +49,18 @@
 .markdown h6 {
   font-weight: inherit;
   font-family: var(--heading-font);
+
+  /*
+  Anchor heading spacing to GitHub's markdown rhythm (Primer): a uniform
+  24px top / 16px bottom margin and 1.25 line-height across every level,
+  letting font-size — not margin — carry the hierarchy. Tailwind Typography's
+  prose defaults scale heading margins per-level up to ~2em (≈48px on h2),
+  which is tuned for long-form articles and reads as too airy in a notebook.
+  Ref: https://github.com/sindresorhus/github-markdown-css
+  */
+  margin-top: 1.5rem;
+  margin-bottom: 1rem;
+  line-height: 1.25;
 }
 
 .markdown hr {