@marimo-team/frontend 0.25.2-dev4 → 0.25.2-dev40

package/src/components/chat/tool-call/tool-result.tsx

@@ -21,9 +21,9 @@ function isUninformative(value: unknown): boolean {
 const SuccessResultSchema = z.looseObject({
   status: z.string().default("success"),
   auth_required: z.boolean().default(false),
-  action_url: z.any(),
-  next_steps: z.any(),
-  meta: z.any(),
+  action_url: z.any().optional(),
+  next_steps: z.any().optional(),
+  meta: z.any().optional(),
   message: z.string().nullish(),
 });
 

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

@@ -1,7 +1,91 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
+import { getCoreRowModel, useReactTable } from "@tanstack/react-table";
+import { fireEvent, render, screen } from "@testing-library/react";
 import { expect, test } from "vitest";
-import { matchingPageRanges } from "../pagination";
+import { TooltipProvider } from "@/components/ui/tooltip";
+import { DataTablePagination, matchingPageRanges } from "../pagination";
+
+function PaginationHarness({
+  totalPages = 3,
+  loading = false,
+}: {
+  totalPages?: number;
+  loading?: boolean;
+}) {
+  const table = useReactTable({
+    locale: "en-US",
+    data: [],
+    columns: [],
+    getCoreRowModel: getCoreRowModel(),
+    manualPagination: true,
+    pageCount: totalPages,
+  });
+  return (
+    <TooltipProvider>
+      <DataTablePagination table={table} tableLoading={loading} />
+    </TooltipProvider>
+  );
+}
+
+test("named navigation controls preserve page updates and boundaries", () => {
+  render(<PaginationHarness />);
+  expect(screen.getByRole("button", { name: "First page" })).toBeDisabled();
+  expect(screen.getByRole("button", { name: "Previous page" })).toBeDisabled();
+  fireEvent.click(screen.getByRole("button", { name: "Next page" }));
+  expect(
+    screen.getByRole("button", { name: "Choose page, current page 2 of 3" }),
+  ).toBeEnabled();
+  fireEvent.click(screen.getByRole("button", { name: "Last page" }));
+  expect(screen.getByRole("button", { name: "Next page" })).toBeDisabled();
+  expect(screen.getByRole("button", { name: "Last page" })).toBeDisabled();
+  fireEvent.click(screen.getByRole("button", { name: "Previous page" }));
+  fireEvent.click(screen.getByRole("button", { name: "First page" }));
+  expect(
+    screen.getByRole("button", { name: "Choose page, current page 1 of 3" }),
+  ).toBeEnabled();
+  expect(screen.getByRole("button", { name: "First page" })).toBeDisabled();
+  expect(screen.getByRole("button", { name: "Previous page" })).toBeDisabled();
+  expect(screen.getByRole("button", { name: "Next page" })).toBeEnabled();
+  expect(screen.getByRole("button", { name: "Last page" })).toBeEnabled();
+});
+
+test("loading prevents navigation and a single page disables controls", () => {
+  const { rerender } = render(<PaginationHarness loading={true} />);
+  fireEvent.click(screen.getByRole("button", { name: "Next page" }));
+  fireEvent.click(screen.getByRole("button", { name: "Last page" }));
+  expect(
+    screen.getByRole("button", { name: "Choose page, current page 1 of 3" }),
+  ).toBeEnabled();
+  rerender(<PaginationHarness />);
+  fireEvent.click(screen.getByRole("button", { name: "Next page" }));
+  expect(
+    screen.getByRole("button", { name: "Choose page, current page 2 of 3" }),
+  ).toBeEnabled();
+  rerender(<PaginationHarness key="single-page" totalPages={1} />);
+  for (const name of [
+    "First page",
+    "Previous page",
+    "Next page",
+    "Last page",
+    "Choose page, current page 1 of 1",
+  ]) {
+    expect(screen.getByRole("button", { name })).toBeDisabled();
+  }
+});
+
+test("page picker exposes named dialog and search", () => {
+  render(<PaginationHarness />);
+  fireEvent.click(
+    screen.getByRole("button", { name: "Choose page, current page 1 of 3" }),
+  );
+  expect(
+    screen.getByRole("dialog", { name: "Choose page" }),
+  ).toBeInTheDocument();
+  expect(
+    screen.getByRole("combobox", { name: "Search pages" }),
+  ).toBeInTheDocument();
+});
 
 test("empty prefix returns no ranges", () => {
   expect(matchingPageRanges("", 500)).toEqual([]);

package/src/components/data-table/charts/__tests__/schemas.test.ts

@@ -4,6 +4,31 @@ import { describe, expect, it } from "vitest";
 import { ChartSchema } from "../schemas";
 
 describe("ChartSchema", () => {
+  it.each(["hours", "yearmonthdatehours"] as const)(
+    "preserves the %s time unit in saved chart configurations",
+    (timeUnit) => {
+      const column = {
+        field: "timestamp",
+        selectedDataType: "temporal",
+        sort: "ascending",
+        timeUnit,
+      };
+      const config = {
+        general: {
+          xColumn: { ...column, aggregate: "none" },
+          yColumn: { ...column, aggregate: "none" },
+          facet: {
+            row: { ...column, linkYAxis: true, binned: true, maxbins: 6 },
+            column: { ...column, linkXAxis: true, binned: true, maxbins: 6 },
+          },
+        },
+        tooltips: { auto: true, fields: [] },
+      };
+
+      expect(ChartSchema.parse(config)).toEqual(config);
+    },
+  );
+
   it("keeps a known axis column type", () => {
     const result = ChartSchema.parse({
       general: { xColumn: { field: "count", type: "integer" } },

package/src/components/data-table/charts/__tests__/spec.test.ts

@@ -69,30 +69,33 @@ describe("getAxisEncoding", () => {
     });
   });
 
-  it("should return correct encoding for temporal field with timeUnit", () => {
-    const result = getAxisEncoding(
-      {
-        field: "date",
-        selectedDataType: "temporal",
-        aggregate: NONE_VALUE,
-        timeUnit: "yearmonth",
-      },
-      undefined,
-      "Date",
-      undefined,
-      ChartType.LINE,
-    );
+  it.each(["yearmonth", "hours", "yearmonthdatehours"] as const)(
+    "should preserve the %s time unit in temporal axis encodings",
+    (timeUnit) => {
+      const result = getAxisEncoding(
+        {
+          field: "date",
+          selectedDataType: "temporal",
+          aggregate: NONE_VALUE,
+          timeUnit,
+        },
+        undefined,
+        "Date",
+        undefined,
+        ChartType.LINE,
+      );
 
-    expect(result).toEqual({
-      field: "date",
-      type: "temporal",
-      bin: undefined,
-      title: "Date",
-      stack: undefined,
-      aggregate: undefined,
-      timeUnit: "yearmonth",
-    });
-  });
+      expect(result).toEqual({
+        field: "date",
+        type: "temporal",
+        bin: undefined,
+        title: "Date",
+        stack: undefined,
+        aggregate: undefined,
+        timeUnit,
+      });
+    },
+  );
 
   it("should return correct encoding for categorical field", () => {
     const result = getAxisEncoding(

package/src/components/data-table/charts/constants.ts

@@ -147,17 +147,18 @@ export const TIME_UNIT_DESCRIPTIONS: Record<
   [title: string, description: string]
 > = {
   year: ["Year", "2025"],
-  quarter: ["Quarter", "Q1 2025"],
-  month: ["Month", "Jan 2025"],
-  week: ["Week", "Jan 01, 2025"],
-  day: ["Day", "Jan 01, 2025"],
-  hours: ["Hour", "Jan 01, 2025 12:00"],
-  minutes: ["Minute", "Jan 01, 2025 12:34"],
-  seconds: ["Second", "Jan 01, 2025 12:34:56"],
-  milliseconds: ["Millisecond", "Jan 01, 2025 12:34:56.789"],
-  date: ["Date", "Jan 01, 2025"],
-  dayofyear: ["Day of Year", "Day 1 of 2025"],
+  quarter: ["Quarter of year", "Q1"],
+  month: ["Month of year", "Jan"],
+  week: ["Week of year", "Week 1"],
+  day: ["Day of week", "Sunday"],
+  hours: ["Hour of day", "12:00"],
+  minutes: ["Minute of hour", "34"],
+  seconds: ["Second of minute", "56"],
+  milliseconds: ["Millisecond of second", "789"],
+  date: ["Day of month", "1"],
+  dayofyear: ["Day of year", "Day 1"],
   yearmonth: ["Year Month", "Jan 2025"],
   yearmonthdate: ["Year Month Date", "Jan 01, 2025"],
+  yearmonthdatehours: ["Year Month Date Hours", "Jan 01, 2025 12:00"],
   monthdate: ["Month Date", "Jan 01"],
 };

package/src/components/data-table/charts/types.ts

@@ -40,6 +40,7 @@ export const SINGLE_TIME_UNITS = [
 export const COMBINED_TIME_UNITS = [
   "yearmonth",
   "yearmonthdate",
+  "yearmonthdatehours",
   "monthdate",
 ] as const;
 export const TIME_UNITS = [

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

@@ -117,6 +117,7 @@ export const DataTablePagination = <TData,>({
             size="xs"
             variant="text"
             data-testid="first-page-button"
+            aria-label="First page"
             className="hidden h-6 w-5 p-0 lg:flex"
             onClick={() => handlePageChange(() => table.setPageIndex(0))}
             onMouseDown={Events.preventFocus}
@@ -130,6 +131,7 @@ export const DataTablePagination = <TData,>({
             size="xs"
             variant="text"
             data-testid="previous-page-button"
+            aria-label="Previous page"
             className="h-6 w-5 p-0"
             onClick={() => handlePageChange(() => table.previousPage())}
             onMouseDown={Events.preventFocus}
@@ -156,6 +158,7 @@ export const DataTablePagination = <TData,>({
             size="xs"
             variant="text"
             data-testid="next-page-button"
+            aria-label="Next page"
             className="h-6 w-5 p-0"
             onClick={() => handlePageChange(() => table.nextPage())}
             onMouseDown={Events.preventFocus}
@@ -169,6 +172,7 @@ export const DataTablePagination = <TData,>({
             size="xs"
             variant="text"
             data-testid="last-page-button"
+            aria-label="Last page"
             className="hidden h-6 w-5 p-0 lg:flex"
             onClick={() =>
               handlePageChange(() =>
@@ -306,14 +310,24 @@ export const PageSelector = ({
               : "opacity-50 cursor-default",
           )}
           data-testid="page-select"
+          aria-label={`Choose page, current page ${currentPage} of ${totalPages}`}
           disabled={totalPages <= 1}
         >
           {currentPage}
           <ChevronDown className="h-3 w-3 opacity-50 mb-px" />
         </button>
       </PopoverTrigger>
-      <PopoverContent className="w-36 p-0" align="center" sideOffset={6}>
-        <Command shouldFilter={false} value={String(currentPage)}>
+      <PopoverContent
+        className="w-36 p-0"
+        align="center"
+        sideOffset={6}
+        aria-label="Choose page"
+      >
+        <Command
+          label="Search pages"
+          shouldFilter={false}
+          value={String(currentPage)}
+        >
           <CommandInput
             placeholder={`Page (1–${totalPages})`}
             rootClassName="px-2 h-8"

package/src/components/editor/actions/__tests__/pair-with-agent-commands.test.ts

@@ -154,63 +154,54 @@ describe("getTerminalCommand", () => {
       ).toBe(
         String.raw`${cli} "$(MARIMO_PAIR_NEXT=1 uvx marimo@latest pair prompt \
   --url http://localhost:8000 \
-  --file notebooks/example.py \
   --session s_ab12cd \
   --with-token)"`,
       );
     },
   );
 
-  it("includes the url and file for each agent", () => {
+  it("includes the url and agent flag for each agent", () => {
     expect(getTerminalCommand("claude", CONNECTION, false)).toBe(
-      `claude "$(uv run marimo pair prompt --url http://localhost:8000 --file notebooks/example.py --claude)"`,
+      `claude "$(uv run marimo pair prompt --url http://localhost:8000 --claude)"`,
     );
     expect(getTerminalCommand("codex", CONNECTION, false)).toBe(
-      `codex "$(uv run marimo pair prompt --url http://localhost:8000 --file notebooks/example.py --codex)"`,
+      `codex "$(uv run marimo pair prompt --url http://localhost:8000 --codex)"`,
     );
     expect(getTerminalCommand("opencode", CONNECTION, false)).toBe(
-      `opencode --prompt "$(uv run marimo pair prompt --url http://localhost:8000 --file notebooks/example.py --opencode)"`,
+      `opencode --prompt "$(uv run marimo pair prompt --url http://localhost:8000 --opencode)"`,
     );
   });
 
-  it("omits the file flag when the page URL has no file", () => {
-    const command = getTerminalCommand(
-      "claude",
-      CONNECTION_WITHOUT_FILE,
-      false,
-    );
+  it("never passes a file to marimo pair prompt", () => {
+    const command = getTerminalCommand("claude", CONNECTION, false);
     expect(command).not.toContain("--file");
-    expect(command).not.toContain("--session");
+    expect(command).not.toContain("notebooks/example.py");
   });
 
-  it("shell-escapes a url containing metacharacters", () => {
-    const command = getTerminalCommand(
-      "claude",
-      { url: "http://host:8000?auth=a&b", file: "notebook.py" },
-      false,
+  it("omits the session outside the preview", () => {
+    expect(getTerminalCommand("claude", CONNECTION, false)).not.toContain(
+      "--session",
     );
-    expect(command).toContain("--url 'http://host:8000?auth=a&b'");
   });
 
-  it.each([
-    ["relative/my notebook.py", "--file 'relative/my notebook.py'"],
-    ["/tmp/my notebook.py", "--file '/tmp/my notebook.py'"],
-    [
-      String.raw`C:\Users\Jane Doe\notebook.py`,
-      String.raw`--file 'C:\Users\Jane Doe\notebook.py'`,
-    ],
-    [
-      String.raw`\\server\share\my notebook.py`,
-      String.raw`--file '\\server\share\my notebook.py'`,
-    ],
-    ["notebooks/it's.py", `--file 'notebooks/it'"'"'s.py'`],
-  ])("shell-escapes file path %s", (file, expected) => {
+  it("omits the session when it is unknown", () => {
+    expect(
+      getTerminalCommand(
+        "claude",
+        CONNECTION_WITHOUT_FILE,
+        false,
+        PAIR_PREVIEW,
+      ),
+    ).not.toContain("--session");
+  });
+
+  it("shell-escapes a url containing metacharacters", () => {
     const command = getTerminalCommand(
       "claude",
-      { url: CONNECTION.url, file },
+      { url: "http://host:8000?auth=a&b" },
       false,
     );
-    expect(command).toContain(expected);
+    expect(command).toContain("--url 'http://host:8000?auth=a&b'");
   });
 
   it("adds --with-token before the agent flag when requested", () => {
@@ -230,23 +221,22 @@ describe("getRawPrompt", () => {
     expect(
       getRawPrompt(
         {
-          url: "http://host/{file}",
-          file: "{session}/it's.py",
-          session: "{command}",
+          url: "http://host/{session}",
+          session: "{command}/it's",
         },
-        "tok'en {file} $&",
+        "tok'en {session} $&",
         {
           ...PAIR_PREVIEW,
           command: "custom marimo",
           templates: {
             ...PAIR_PREVIEW.templates,
-            prompt: "{command}\n{url}\n{file}{session}{authentication}",
+            prompt: "{command}\n{url}\n{session}{authentication}",
           },
         },
       ),
     ).toBe(
-      "custom marimo\nhttp://host/{file}\nFile: {session}/it's.py\nSession: {command}\n" +
-        `\n\nFor authenticated Pair commands, set \`export MARIMO_TOKEN='tok'"'"'en {file} $&'\` in the shell that runs marimo.`,
+      "custom marimo\nhttp://host/{session}\nSession: {command}/it's\n" +
+        `\n\nFor authenticated Pair commands, set \`export MARIMO_TOKEN='tok'"'"'en {session} $&'\` in the shell that runs marimo.`,
     );
   });
 
@@ -256,10 +246,7 @@ describe("getRawPrompt", () => {
       { ...CONNECTION_WITHOUT_FILE, session: "s_ab12cd" },
       "Session: s_ab12cd\n",
     ],
-    [
-      { ...CONNECTION_WITHOUT_FILE, file: "notebook.py" },
-      "File: notebook.py\n",
-    ],
+    [{ ...CONNECTION_WITHOUT_FILE, file: "notebook.py" }, ""],
   ])("omits absent preview fields for %j", (connection, fields) => {
     expect(getRawPrompt(connection, null, PAIR_PREVIEW)).toBe(
       "Pair with me on this running marimo notebook.\n\n" +
@@ -270,13 +257,12 @@ describe("getRawPrompt", () => {
     );
   });
 
-  it("renders the shared preview template with optional session context", () => {
+  it("renders the shared preview template with the session and no file", () => {
     expect(
       getRawPrompt({ ...CONNECTION, session: "s_ab12cd" }, null, PAIR_PREVIEW),
     ).toBe(
       "Pair with me on this running marimo notebook.\n\n" +
         "URL: http://localhost:8000\n" +
-        "File: notebooks/example.py\n" +
         "Session: s_ab12cd\n\n" +
         "Run `uvx marimo@latest pair --help` first.\n" +
         "Use `uvx marimo@latest` for all marimo commands.\n\n" +

package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx

@@ -15,7 +15,10 @@ import { PAIR_PREVIEW } from "@/__tests__/fixtures/pair-preview";
 import { Dialog } from "@/components/ui/dialog";
 import { TooltipProvider } from "@/components/ui/tooltip";
 import { pairPreviewAtom } from "@/core/config/pair";
-import { getSessionId } from "@/core/kernel/session";
+import {
+  type StableSessionId,
+  stableSessionIdAtom,
+} from "@/core/kernel/session";
 import { filenameAtom } from "@/core/saving/file-state";
 import {
   DEFAULT_RUNTIME_CONFIG,
@@ -39,10 +42,13 @@ function wrapper({ children }: { children: React.ReactNode }) {
   );
 }
 
+const STABLE_ID = "sess-stable-1" as StableSessionId;
+
 describe("PairWithAgentModal", () => {
   beforeEach(() => {
     vi.clearAllMocks();
     store.set(pairPreviewAtom, PAIR_PREVIEW);
+    store.set(stableSessionIdAtom, STABLE_ID);
     store.set(filenameAtom, "/project/my notebook.py");
     store.set(runtimeConfigAtom, {
       ...DEFAULT_RUNTIME_CONFIG,
@@ -57,12 +63,13 @@ describe("PairWithAgentModal", () => {
 
   afterEach(() => {
     store.set(pairPreviewAtom, undefined);
+    store.set(stableSessionIdAtom, null);
     store.set(filenameAtom, null);
     vi.unstubAllGlobals();
   });
 
   it.each(["Claude", "Codex", "OpenCode", "Prompt"])(
-    "omits skill installation and includes the current session in %s preview",
+    "omits skill installation and targets the stable session in %s preview",
     async (tab) => {
       await act(async () => {
         render(<PairWithAgentModal onClose={vi.fn()} />, { wrapper });
@@ -74,59 +81,41 @@ describe("PairWithAgentModal", () => {
 
       const panel = screen.getByRole("tabpanel");
       expect(within(panel).queryAllByText(/skill/i)).toEqual([]);
-      expect(panel).toHaveTextContent(getSessionId());
-      expect(panel).toHaveTextContent(
-        tab === "Prompt" ? "File: notebook.py" : "--file notebook.py",
-      );
-      expect(panel).not.toHaveTextContent("/project/my notebook.py");
+      expect(panel).toHaveTextContent(STABLE_ID);
+      expect(panel).not.toHaveTextContent("--file");
+      expect(panel).not.toHaveTextContent("File:");
+      expect(panel).not.toHaveTextContent("notebook.py");
       expect(panel).toHaveTextContent(
         tab === "Prompt" ? "1. Copy this prompt" : "1. Run in your terminal",
       );
-    },
-  );
-
-  it.each(["Claude", "Codex", "OpenCode", "Prompt"])(
-    "includes the known notebook file in %s preview when the URL has no file",
-    async (tab) => {
-      window.history.replaceState({}, "", "/");
-      await act(async () => {
-        render(<PairWithAgentModal onClose={vi.fn()} />, { wrapper });
-      });
-      fireEvent.mouseDown(screen.getByRole("tab", { name: tab }), {
-        button: 0,
-        ctrlKey: false,
-      });
-
-      const panel = screen.getByRole("tabpanel");
-      const fileHint =
-        tab === "Prompt"
-          ? "File: /project/my notebook.py"
-          : "--file '/project/my notebook.py'";
-      expect(panel).toHaveTextContent(fileHint);
       fireEvent.click(
         within(panel).getByRole("button", {
           name: tab === "Prompt" ? "Copy prompt" : "Copy command",
         }),
       );
       await waitFor(() => expect(copyToClipboard).toHaveBeenCalledOnce());
-      expect(vi.mocked(copyToClipboard).mock.calls[0][0]).toContain(fileHint);
+      const copied = vi.mocked(copyToClipboard).mock.calls[0][0];
+      expect(copied).toContain(STABLE_ID);
+      expect(copied).not.toContain("notebook.py");
     },
   );
 
-  it.each([null, ""])(
-    "omits the file when the URL and notebook filename are absent (%s)",
-    async (filename) => {
-      window.history.replaceState({}, "", "/");
-      store.set(filenameAtom, filename);
+  it.each(["Claude", "Prompt"])(
+    "omits the session in %s preview before kernel-ready arrives",
+    async (tab) => {
+      store.set(stableSessionIdAtom, null);
       await act(async () => {
         render(<PairWithAgentModal onClose={vi.fn()} />, { wrapper });
       });
-      expect(screen.getByRole("tabpanel")).not.toHaveTextContent("--file");
-      fireEvent.mouseDown(screen.getByRole("tab", { name: "Prompt" }), {
+      fireEvent.mouseDown(screen.getByRole("tab", { name: tab }), {
         button: 0,
         ctrlKey: false,
       });
-      expect(screen.getByRole("tabpanel")).not.toHaveTextContent("File:");
+
+      const panel = screen.getByRole("tabpanel");
+      expect(panel).not.toHaveTextContent("--session");
+      expect(panel).not.toHaveTextContent("Session:");
+      expect(panel).toHaveTextContent("http://localhost:8000");
     },
   );
 
@@ -147,7 +136,8 @@ describe("PairWithAgentModal", () => {
       expect(
         within(panel).getByText("npx skills add marimo-team/marimo-pair"),
       ).toBeVisible();
-      expect(panel).not.toHaveTextContent(getSessionId());
+      expect(panel).not.toHaveTextContent(STABLE_ID);
+      expect(panel).not.toHaveTextContent("--file");
       expect(panel).not.toHaveTextContent("MARIMO_PAIR_NEXT");
       expect(panel).not.toHaveTextContent("/project/my notebook.py");
       expect(panel).toHaveTextContent(
@@ -205,8 +195,7 @@ describe("PairWithAgentModal", () => {
       expect(copyToClipboard).toHaveBeenCalledWith(
         String.raw`claude "$(MARIMO_PAIR_NEXT=1 uvx marimo@latest pair prompt \
   --url http://localhost:8000/ \
-  --file notebook.py \
-  --session ${getSessionId()} \
+  --session ${STABLE_ID} \
   --with-token)"`,
       ),
     );

package/src/components/editor/actions/__tests__/useSetOutputsExpanded.test.ts

@@ -0,0 +1,42 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { describe, expect, it } from "vitest";
+import { MockNotebook } from "@/__mocks__/notebook";
+import type { OutputMessage } from "@/core/kernel/messages";
+import { outputExpansionConfigs } from "../useSetOutputsExpanded";
+
+const output: OutputMessage = {
+  channel: "output",
+  mimetype: "text/plain",
+  data: "hello",
+  timestamp: 0,
+};
+
+describe("outputExpansionConfigs", () => {
+  const [withOutput, withoutOutput, expanded, expandedWithoutOutput] =
+    MockNotebook.cellIds();
+  const notebook = MockNotebook.notebookState({
+    cellData: {
+      [withOutput]: {},
+      [withoutOutput]: {},
+      [expanded]: { config: { expand_output: true } },
+      [expandedWithoutOutput]: { config: { expand_output: true } },
+    },
+    cellRuntime: {
+      [withOutput]: { output },
+      [expanded]: { output },
+    },
+  });
+
+  it("expands clamped cells with output and skips cells without output", () => {
+    expect(outputExpansionConfigs(notebook, true)).toEqual({
+      [withOutput]: { expand_output: true },
+    });
+  });
+
+  it("clamps expanded cells with output and skips cells without output", () => {
+    expect(outputExpansionConfigs(notebook, false)).toEqual({
+      [expanded]: { expand_output: false },
+    });
+  });
+});

package/src/components/editor/actions/pair-with-agent-commands.ts

@@ -38,8 +38,9 @@ function getFileFlag(file: string | undefined): string {
 /** Identifies the specific running notebook to pair on. */
 export interface ConnectionInfo {
   url: string;
-  /** The server's file key or notebook filename, when known. */
+  /** The server's file key, when known. */
   file?: string;
+  /** The stable session ID, when known. */
   session?: string;
 }
 
@@ -49,7 +50,7 @@ export interface ConnectionInfo {
  */
 export function getTerminalCommand(
   agent: Exclude<AgentTab, "prompt">,
-  { url, file, session }: ConnectionInfo,
+  { url, session }: ConnectionInfo,
   withToken: boolean,
   preview?: PairPreviewConfig,
 ): string {
@@ -58,7 +59,6 @@ export function getTerminalCommand(
   const base = [
     `${prefix}${command} pair prompt`,
     `--url ${shellQuote(url)}`,
-    file ? `--file ${shellQuote(file)}` : "",
     preview && session ? `--session ${shellQuote(session)}` : "",
     withToken ? "--with-token" : "",
     preview ? "" : `--${agent}`,
@@ -92,7 +92,6 @@ export function getRawPrompt(
     return formatPrompt(templates.prompt, {
       command,
       url,
-      file: file ? formatPrompt(templates.file, { file }) : "",
       session: session ? formatPrompt(templates.session, { session }) : "",
       authentication: token
         ? formatPrompt(templates.token, { token: shellQuote(token) })