@marimo-team/frontend 0.20.5-dev4 → 0.20.5-dev41

package/src/components/data-table/pagination.tsx

@@ -37,6 +37,8 @@ import {
 } from "../ui/select";
 import type { DataTableSelection, PageRange } from "./types";
 
+const MAX_PAGES_BEFORE_CLAMPING = 100;
+
 interface DataTablePaginationProps<TData> {
   table: Table<TData>;
   selection?: DataTableSelection;
@@ -325,35 +327,39 @@ export const PageSelector = ({
             ),
           )}
         </div>
-        <DropdownMenuSeparator />
-        <div
-          className="px-2 pt-0.5 shrink-0"
-          onKeyDown={(e) => e.stopPropagation()}
-        >
-          <label
-            htmlFor={jumpInputId}
-            className="text-xs text-muted-foreground block mb-1"
-          >
-            Jump to page
-          </label>
-          <Input
-            id={jumpInputId}
-            type="number"
-            min={1}
-            max={totalPages}
-            placeholder={`1-${totalPages}`}
-            value={jumpValue}
-            onChange={(e) => setJumpValue(e.target.value)}
-            onKeyDown={(e) => {
-              if (e.key === "Enter") {
-                handleJump();
-              }
-              e.stopPropagation();
-            }}
-            className="h-6 text-xs"
-            data-testid="page-jump-input"
-          />
-        </div>
+        {totalPages > MAX_PAGES_BEFORE_CLAMPING && (
+          <>
+            <DropdownMenuSeparator />
+            <div
+              className="px-2 pt-0.5 shrink-0"
+              onKeyDown={(e) => e.stopPropagation()}
+            >
+              <label
+                htmlFor={jumpInputId}
+                className="text-xs text-muted-foreground block mb-1"
+              >
+                Jump to page
+              </label>
+              <Input
+                id={jumpInputId}
+                type="number"
+                min={1}
+                max={totalPages}
+                placeholder={`1-${totalPages}`}
+                value={jumpValue}
+                onChange={(e) => setJumpValue(e.target.value)}
+                onKeyDown={(e) => {
+                  if (e.key === "Enter") {
+                    handleJump();
+                  }
+                  e.stopPropagation();
+                }}
+                className="h-6 text-xs"
+                data-testid="page-jump-input"
+              />
+            </div>
+          </>
+        )}
       </DropdownMenuContent>
     </DropdownMenu>
   );
@@ -363,7 +369,7 @@ export function getPageRanges(
   currentPage: number,
   totalPages: number,
 ): PageRange[] {
-  if (totalPages <= 100) {
+  if (totalPages <= MAX_PAGES_BEFORE_CLAMPING) {
     return range(totalPages).map((i) => ({ type: "page", page: i + 1 }));
   }
 

package/src/components/datasources/components.tsx

@@ -1,18 +1,15 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import { ChevronRightIcon, LoaderCircle, XIcon } from "lucide-react";
+import { LoaderCircle, XIcon } from "lucide-react";
 import type { CSSProperties } from "react";
+import { TreeChevron } from "@/components/editor/file-tree/tree-actions";
 import type { DataType } from "@/core/kernel/messages";
 import { cn } from "@/utils/cn";
 import { DATA_TYPE_ICON, getDataTypeColor } from "../datasets/icons";
 
 export const RotatingChevron: React.FC<{ isExpanded: boolean }> = ({
   isExpanded,
-}) => (
-  <ChevronRightIcon
-    className={cn("h-3 w-3 transition-transform", isExpanded && "rotate-90")}
-  />
-);
+}) => <TreeChevron isExpanded={isExpanded} className="h-3 w-3" />;
 
 export const DatasourceLabel: React.FC<{
   children: React.ReactNode;

package/src/components/datasources/datasources.tsx

@@ -3,11 +3,12 @@
 import { CommandList } from "cmdk";
 import { atom, useAtomValue, useSetAtom } from "jotai";
 import { sortBy } from "lodash-es";
-import { PlusIcon, PlusSquareIcon, RefreshCwIcon, XIcon } from "lucide-react";
+import { PlusIcon, PlusSquareIcon, XIcon } from "lucide-react";
 import React from "react";
 import { dbDisplayName } from "@/components/databases/display";
 import { EngineVariable } from "@/components/databases/engine-variable";
 import { DatabaseLogo } from "@/components/databases/icon";
+import { RefreshIconButton } from "@/components/editor/file-tree/tree-actions";
 import { CopyClipboardIcon } from "@/components/icons/copy-icon";
 import { Button } from "@/components/ui/button";
 import { Command, CommandInput, CommandItem } from "@/components/ui/command";
@@ -265,15 +266,10 @@ const Engine: React.FC<{
   const engineName = internalEngine ? "In-Memory" : connection.name;
   const { previewDataSourceConnection } = useRequestClient();
 
-  const [isSpinning, setIsSpinning] = React.useState(false);
-
   const handleRefreshConnection = async () => {
-    setIsSpinning(true);
     await previewDataSourceConnection({
       engine: connection.name,
     });
-    // Artificially spin the icon if the request is really fast
-    setTimeout(() => setIsSpinning(false), 500);
   };
 
   return (
@@ -288,21 +284,12 @@ const Engine: React.FC<{
           (<EngineVariable variableName={engineName as VariableName} />)
         </span>
         {!internalEngine && (
-          <Tooltip content="Refresh connection">
-            <Button
-              variant="ghost"
-              size="icon"
-              className="ml-auto hover:bg-transparent hover:shadow-none"
-              onClick={handleRefreshConnection}
-            >
-              <RefreshCwIcon
-                className={cn(
-                  "h-4 w-4 text-muted-foreground hover:text-foreground",
-                  isSpinning && "animate-[spin_0.5s]",
-                )}
-              />
-            </Button>
-          </Tooltip>
+          <RefreshIconButton
+            onClick={handleRefreshConnection}
+            tooltip="Refresh connection"
+            className="ml-auto h-4 p-0"
+            iconClassName="h-3.5 w-3.5"
+          />
         )}
       </DatasourceLabel>
       {hasChildren ? (

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

@@ -27,6 +27,7 @@ export interface ActionButton {
   handleHeadless?: (event?: Event) => void;
   divider?: boolean;
   dropdown?: ActionButton[];
+  additionalKeywords?: string[];
 }
 
 export function isParentAction(
@@ -51,6 +52,10 @@ export function flattenActions(
     if (isParentAction(action)) {
       return flattenActions(action.dropdown, `${prevLabel + action.label} > `);
     }
-    return { ...action, label: prevLabel + action.label };
+    return {
+      ...action,
+      label: prevLabel + action.label,
+      additionalKeywords: action.additionalKeywords,
+    };
   });
 }

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

@@ -31,6 +31,7 @@ import {
   LayoutTemplateIcon,
   LinkIcon,
   MessagesSquareIcon,
+  NotebookIcon,
   PanelLeftIcon,
   PowerSquareIcon,
   PresentationIcon,
@@ -123,8 +124,13 @@ export function useNotebookActions() {
   const setCommandPaletteOpen = useSetAtom(commandPaletteAtom);
   const setSettingsDialogOpen = useSetAtom(settingDialogAtom);
   const setKeyboardShortcutsOpen = useSetAtom(keyboardShortcutsAtom);
-  const { exportAsMarkdown, readCode, saveCellConfig, updateCellOutputs } =
-    useRequestClient();
+  const {
+    exportAsIPYNB,
+    exportAsMarkdown,
+    readCode,
+    saveCellConfig,
+    updateCellOutputs,
+  } = useRequestClient();
   const takeScreenshots = useEnrichCellOutputs();
 
   const hasDisabledCells = useAtomValue(hasDisabledCellsAtom);
@@ -199,6 +205,27 @@ export function useNotebookActions() {
     setTimeout(() => window.dispatchEvent(afterprint), 0);
   };
 
+  const handleDownloadAsIPYNB = async () => {
+    if (!filename) {
+      toastNotebookMustBeNamed();
+      return;
+    }
+
+    const runDownload = async (progress: ProgressState) => {
+      await updateCellOutputsWithScreenshots({
+        takeScreenshots: () => takeScreenshots({ progress }),
+        updateCellOutputs,
+      });
+      const ipynb = await exportAsIPYNB({ download: false });
+      downloadBlob(
+        new Blob([ipynb], { type: "application/x-ipynb+json" }),
+        Filenames.toIPYNB(document.title),
+      );
+    };
+
+    await withLoadingToast("Downloading IPYNB...", runDownload);
+  };
+
   const actions: ActionButton[] = [
     {
       icon: <DownloadIcon size={14} strokeWidth={1.5} />,
@@ -240,6 +267,11 @@ export function useNotebookActions() {
             );
           },
         },
+        {
+          icon: <NotebookIcon size={14} strokeWidth={1.5} />,
+          label: "Download as ipynb",
+          handle: handleDownloadAsIPYNB,
+        },
         {
           icon: <CodeIcon size={14} strokeWidth={1.5} />,
           label: "Download Python code",
@@ -347,17 +379,20 @@ export function useNotebookActions() {
       label: "Helper panel",
       redundant: true,
       handle: NOOP_HANDLER,
-      dropdown: PANELS.flatMap(({ type: id, Icon, hidden }) => {
-        if (hidden) {
-          return [];
-        }
-        return {
-          label: startCase(id),
-          rightElement: renderCheckboxElement(selectedPanel === id),
-          icon: <Icon size={14} strokeWidth={1.5} />,
-          handle: () => toggleApplication(id),
-        };
-      }),
+      dropdown: PANELS.flatMap(
+        ({ type: id, Icon, hidden, additionalKeywords }) => {
+          if (hidden) {
+            return [];
+          }
+          return {
+            label: startCase(id),
+            rightElement: renderCheckboxElement(selectedPanel === id),
+            icon: <Icon size={14} strokeWidth={1.5} />,
+            handle: () => toggleApplication(id),
+            additionalKeywords,
+          };
+        },
+      ),
     },
 
     {
@@ -478,6 +513,7 @@ export function useNotebookActions() {
       label: "Restart kernel",
       variant: "danger",
       handle: restartKernel,
+      additionalKeywords: ["reset", "reload", "restart"],
     },
     {
       icon: <FastForwardIcon size={14} strokeWidth={1.5} />,
@@ -535,6 +571,7 @@ export function useNotebookActions() {
       label: "User settings",
       handle: () => setSettingsDialogOpen((open) => !open),
       redundant: true,
+      additionalKeywords: ["preferences", "options", "configuration"],
     },
     {
       icon: <ExternalLinkIcon size={14} strokeWidth={1.5} />,

package/src/components/editor/chrome/types.ts

@@ -59,6 +59,8 @@ export interface PanelDescriptor {
   defaultSection: PanelSection;
   /** Capability required for this panel to be visible. If the capability is false, the panel is hidden. */
   requiredCapability?: keyof Capabilities;
+  /** Additional search keywords for the command palette */
+  additionalKeywords?: string[];
 }
 
 /**
@@ -73,6 +75,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Files",
     tooltip: "View files",
     defaultSection: "sidebar",
+    additionalKeywords: ["explorer", "browser", "directory"],
   },
   {
     type: "variables",
@@ -80,6 +83,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Variables",
     tooltip: "Explore variables and data sources",
     defaultSection: "sidebar",
+    additionalKeywords: ["state", "scope", "inspector"],
   },
   {
     type: "packages",
@@ -87,6 +91,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Packages",
     tooltip: "Manage packages",
     defaultSection: "sidebar",
+    additionalKeywords: ["dependencies", "pip", "install"],
   },
   {
     type: "ai",
@@ -94,6 +99,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "AI",
     tooltip: "Chat & Agents",
     defaultSection: "sidebar",
+    additionalKeywords: ["chat", "copilot", "assistant"],
   },
   {
     type: "outline",
@@ -101,6 +107,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Outline",
     tooltip: "View outline",
     defaultSection: "sidebar",
+    additionalKeywords: ["toc", "structure", "headings"],
   },
   {
     type: "documentation",
@@ -108,6 +115,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Docs",
     tooltip: "View live docs",
     defaultSection: "sidebar",
+    additionalKeywords: ["reference", "api"],
   },
   {
     type: "dependencies",
@@ -115,6 +123,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Dependencies",
     tooltip: "Explore dependencies",
     defaultSection: "sidebar",
+    additionalKeywords: ["graph", "imports"],
   },
   // Developer panel defaults
   {
@@ -123,6 +132,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Errors",
     tooltip: "View errors",
     defaultSection: "developer-panel",
+    additionalKeywords: ["exceptions", "problems", "diagnostics"],
   },
   {
     type: "scratchpad",
@@ -130,6 +140,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Scratchpad",
     tooltip: "Scratchpad",
     defaultSection: "developer-panel",
+    additionalKeywords: ["scratch", "draft", "playground"],
   },
   {
     type: "tracing",
@@ -137,6 +148,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Tracing",
     tooltip: "View tracing",
     defaultSection: "developer-panel",
+    additionalKeywords: ["profiling", "performance"],
   },
   {
     type: "secrets",
@@ -145,6 +157,7 @@ export const PANELS: PanelDescriptor[] = [
     tooltip: "Manage secrets",
     defaultSection: "developer-panel",
     hidden: isWasm(),
+    additionalKeywords: ["env", "environment", "keys", "credentials"],
   },
   {
     type: "logs",
@@ -152,6 +165,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Logs",
     tooltip: "View logs",
     defaultSection: "developer-panel",
+    additionalKeywords: ["console", "stdout"],
   },
   {
     type: "terminal",
@@ -161,6 +175,7 @@ export const PANELS: PanelDescriptor[] = [
     hidden: isWasm(),
     defaultSection: "developer-panel",
     requiredCapability: "terminal",
+    additionalKeywords: ["shell", "console", "bash", "command"],
   },
   {
     type: "snippets",
@@ -168,6 +183,7 @@ export const PANELS: PanelDescriptor[] = [
     label: "Snippets",
     tooltip: "Snippets",
     defaultSection: "developer-panel",
+    additionalKeywords: ["templates", "examples"],
   },
   {
     type: "cache",
@@ -176,6 +192,7 @@ export const PANELS: PanelDescriptor[] = [
     tooltip: "View cache",
     defaultSection: "developer-panel",
     hidden: !getFeatureFlag("cache_panel"),
+    additionalKeywords: ["memory", "memoize"],
   },
 ];
 

package/src/components/editor/controls/command-palette.tsx

@@ -78,6 +78,7 @@ const CommandPalette = () => {
     return (
       <CommandItem
         disabled={props.disabled}
+        keywords={hotkey.additionalKeywords}
         onSelect={() => {
           addRecentCommand(shortcut);
           // Close first and then run the action, so the dialog doesn't steal focus
@@ -105,15 +106,18 @@ const CommandPalette = () => {
     handle,
     props = {},
     hotkey,
+    additionalKeywords,
   }: {
     label: string;
     handle: () => void;
     props?: { disabled?: boolean; tooltip?: React.ReactNode };
     hotkey?: HotkeyAction;
+    additionalKeywords?: string[];
   }) => {
     return (
       <CommandItem
         disabled={props.disabled}
+        keywords={additionalKeywords}
         onSelect={() => {
           addRecentCommand(label);
           setOpen(false);
@@ -163,6 +167,7 @@ const CommandPalette = () => {
                       disabled: action.disabled,
                       tooltip: action.tooltip,
                     },
+                    additionalKeywords: action.additionalKeywords,
                   });
                 }
                 return null;
@@ -190,6 +195,7 @@ const CommandPalette = () => {
               label: action.label,
               handle: action.handleHeadless || action.handle,
               props: { disabled: action.disabled, tooltip: action.tooltip },
+              additionalKeywords: action.additionalKeywords,
             });
           })}
           {cellActions.map((action) => {
@@ -200,6 +206,7 @@ const CommandPalette = () => {
               label: `Cell > ${action.label}`,
               handle: action.handleHeadless || action.handle,
               props: { disabled: action.disabled, tooltip: action.tooltip },
+              additionalKeywords: action.additionalKeywords,
             });
           })}
         </CommandGroup>

package/src/components/editor/controls/keyboard-shortcuts.tsx

@@ -174,7 +174,9 @@ export const KeyboardShortcuts: React.FC = () => {
                 return;
               }
 
-              let key = e.key.toLowerCase();
+              // Single character keys are always lowercase (e.g. "a", "b", "c")
+              // But we should preserve the original case for other keys (e.g. "Enter", "Escape")
+              let key = e.key.length === 1 ? e.key.toLowerCase() : e.key;
               // Handle edge cases
               if (e.key === " ") {
                 key = "Space";