@marimo-team/islands 0.23.10 → 0.23.11-dev1

package/dist/{code-visibility-DrFozVlm.js → code-visibility-BDDI_K4x.js}

@@ -35981,7 +35981,7 @@ ${d}`,
       return Logger.warn("Failed to get version from mount config"), null;
     }
   }
-  marimoVersionAtom = atom(getVersionFromMountConfig() || "0.23.10");
+  marimoVersionAtom = atom(getVersionFromMountConfig() || "0.23.11-dev1");
   showCodeInRunModeAtom = atom(true);
   atom(null);
   var import_compiler_runtime = require_compiler_runtime();

package/dist/main.js

@@ -26,7 +26,7 @@ import { $ as reducer, B as safeExtractSetUIElementMessageBuffers, Bn as CircleA
 import { __tla as __tla_1 } from "./chunk-5FQGJX7Z-BbqSm5gU.js";
 import { o as useSize, s as Root$1, u as createLucideIcon } from "./dist-Dk6PV_d3.js";
 import { A as SquareFunction, C as DEFAULT_COLOR_SCHEME, D as SCALE_TYPE_DESCRIPTIONS, E as EMPTY_VALUE$1, O as TIME_UNIT_DESCRIPTIONS, S as DEFAULT_AGGREGATION, T as DEFAULT_TIME_UNIT, _ as AGGREGATION_TYPE_DESCRIPTIONS, a as AGGREGATION_FNS$1, b as COLOR_SCHEMES, c as COLOR_BY_FIELDS, d as NONE_VALUE, f as SELECTABLE_DATA_TYPES, g as TIME_UNITS, h as STRING_AGGREGATION_FNS, i as convertDataTypeToSelectable, j as ChartColumn, k as escapeFieldName, l as COMBINED_TIME_UNITS, m as SORT_TYPES, n as createSpecWithoutData, o as BIN_AGGREGATION, p as SINGLE_TIME_UNITS, r as isFieldSet, s as CHART_TYPES, t as augmentSpecWithData, u as ChartType, v as AGGREGATION_TYPE_ICON, w as DEFAULT_MAX_BINS_FACET, x as COUNT_FIELD, y as CHART_TYPE_ICON } from "./spec-X7FwLJni.js";
-import { $ as contextAwarePanelOpen, $t as $fae977aafc393c5c$export$6b862160d295c8e, A as prettifyRowCount, At as SELECT_COLUMN_ID, B as DatePicker, Bt as TabsContent, C as downloadSizeLimitAtom, Ct as DelayMount, D as ErrorState, Dt as loadTableAndRawData, E as EmptyState, Et as getPageIndexForRow, F as ContextMenuSeparator, Ft as Maps, G as CommandEmpty, Gt as ChartLoadingState, H as Combobox, Ht as TabsTrigger, I as ContextMenuTrigger, It as dateToLocalISODate, J as CommandList, Jt as RenderTextWithLinks, K as CommandInput, Kt as LazyVegaEmbed, L as useInternalStateWithSync, Lt as dateToLocalISODateTime, M as ContextMenu, Mt as toFieldTypes, N as ContextMenuContent, Nt as getMimeValues, O as LoadingState, Ot as loadTableData, P as ContextMenuItem, Pt as isNullishFilter, Q as PANEL_TYPES, Qt as $fae977aafc393c5c$export$588937bcd60ade55, R as useSelectList, Rt as dateToLocalISOTime, S as Filenames, St as ColumnChartSpecModel, T as ColumnPreviewContainer, Tt as usePrevious$1, U as ComboboxItem, Ut as ChartErrorState, V as DateRangePicker, Vt as TabsList, W as Command, Wt as ChartInfoState, X as smartMatch, Xt as HtmlOutput, Y as CommandSeparator, Yt as Kbd, Z as ContextAwarePanelItem, Zt as EmotionCacheProvider, _ as ADD_PRINTING_CLASS, _t as NAMELESS_COLUMN_PREFIX, an as Ellipsis, at as Toggle, b as downloadHTMLAsImage, bt as renderCellValue, c as Slide, cn as ChevronsUpDown, d as RadioGroupItem, dn as ChevronsDownUp, dt as Table, en as TextWrap, et as contextAwarePanelOwner, f as JsonOutput, fn as ChevronLeft, ft as TableBody, g as InstallPackageButton, gt as TableRow, h as DataTable, ht as TableHeader, it as slotsController, j as getColumnCountForDisplay, jt as TOO_MANY_ROWS, k as prettifyRowColumnCount, kt as INDEX_COLUMN_NAME, l as Switch, ln as ChevronsRight, lt as Fill, m as OutputRenderer, mt as TableHead, n as marimoVersionAtom, nn as Funnel, nt as isCellAwareAtom, o as SLIDE_TYPE_OPTIONS_BY_VALUE, on as Download, p as OutputArea, pn as ArrowDownWideNarrow, pt as TableCell, q as CommandItem, qt as useOverflowDetection, r as showCodeInRunModeAtom, rn as EyeOff, rt as SlotNames, sn as Code, t as useNotebookCodeAvailable, tn as GripHorizontal, tt as contextAwarePanelType, u as RadioGroup, un as ChevronsLeft, ut as Provider$1, v as downloadBlob, vt as generateColumns, w as ColumnName, wt as useIntersectionObserver, x as Progress, xt as ColumnChartContext, y as downloadByURL, yt as inferFieldTypes, z as CompactChipRow, zt as Tabs, __tla as __tla_2 } from "./code-visibility-DrFozVlm.js";
+import { $ as contextAwarePanelOpen, $t as $fae977aafc393c5c$export$6b862160d295c8e, A as prettifyRowCount, At as SELECT_COLUMN_ID, B as DatePicker, Bt as TabsContent, C as downloadSizeLimitAtom, Ct as DelayMount, D as ErrorState, Dt as loadTableAndRawData, E as EmptyState, Et as getPageIndexForRow, F as ContextMenuSeparator, Ft as Maps, G as CommandEmpty, Gt as ChartLoadingState, H as Combobox, Ht as TabsTrigger, I as ContextMenuTrigger, It as dateToLocalISODate, J as CommandList, Jt as RenderTextWithLinks, K as CommandInput, Kt as LazyVegaEmbed, L as useInternalStateWithSync, Lt as dateToLocalISODateTime, M as ContextMenu, Mt as toFieldTypes, N as ContextMenuContent, Nt as getMimeValues, O as LoadingState, Ot as loadTableData, P as ContextMenuItem, Pt as isNullishFilter, Q as PANEL_TYPES, Qt as $fae977aafc393c5c$export$588937bcd60ade55, R as useSelectList, Rt as dateToLocalISOTime, S as Filenames, St as ColumnChartSpecModel, T as ColumnPreviewContainer, Tt as usePrevious$1, U as ComboboxItem, Ut as ChartErrorState, V as DateRangePicker, Vt as TabsList, W as Command, Wt as ChartInfoState, X as smartMatch, Xt as HtmlOutput, Y as CommandSeparator, Yt as Kbd, Z as ContextAwarePanelItem, Zt as EmotionCacheProvider, _ as ADD_PRINTING_CLASS, _t as NAMELESS_COLUMN_PREFIX, an as Ellipsis, at as Toggle, b as downloadHTMLAsImage, bt as renderCellValue, c as Slide, cn as ChevronsUpDown, d as RadioGroupItem, dn as ChevronsDownUp, dt as Table, en as TextWrap, et as contextAwarePanelOwner, f as JsonOutput, fn as ChevronLeft, ft as TableBody, g as InstallPackageButton, gt as TableRow, h as DataTable, ht as TableHeader, it as slotsController, j as getColumnCountForDisplay, jt as TOO_MANY_ROWS, k as prettifyRowColumnCount, kt as INDEX_COLUMN_NAME, l as Switch, ln as ChevronsRight, lt as Fill, m as OutputRenderer, mt as TableHead, n as marimoVersionAtom, nn as Funnel, nt as isCellAwareAtom, o as SLIDE_TYPE_OPTIONS_BY_VALUE, on as Download, p as OutputArea, pn as ArrowDownWideNarrow, pt as TableCell, q as CommandItem, qt as useOverflowDetection, r as showCodeInRunModeAtom, rn as EyeOff, rt as SlotNames, sn as Code, t as useNotebookCodeAvailable, tn as GripHorizontal, tt as contextAwarePanelType, u as RadioGroup, un as ChevronsLeft, ut as Provider$1, v as downloadBlob, vt as generateColumns, w as ColumnName, wt as useIntersectionObserver, x as Progress, xt as ColumnChartContext, y as downloadByURL, yt as inferFieldTypes, z as CompactChipRow, zt as Tabs, __tla as __tla_2 } from "./code-visibility-BDDI_K4x.js";
 import { c as Calendar, i as createReducerAndAtoms, n as useOnUnmount, o as ToggleLeft, t as useOnMount } from "./useLifecycle-00mO3OSS.js";
 import { t as Check } from "./check-BCaJeT-J.js";
 import { A as Icon, C as logNever, D as $18f2051aff69b9bf$export$a54013f0d02a8f82, E as $18f2051aff69b9bf$export$43bb16f9c6d9e3f7, F as createCollection, I as X, M as clamp$2, N as usePrevious$2, P as useDirection, R as ChevronDown, S as assertNever, a as SelectGroup, c as SelectSeparator, d as NativeSelect, i as SelectContent, j as Trigger$1, l as SelectTrigger, n as capitalize, o as SelectItem, r as Select, s as SelectLabel, t as Strings, u as SelectValue, w as $a916eb452884faea$export$b7a616150fdb9f44 } from "./strings-J57tzLr3.js";
@@ -36187,7 +36187,7 @@ ${c}
   function _temp2$2(e) {
     e.target === e.currentTarget && e.key === "Enter" && (e.preventDefault(), e.stopPropagation(), e.currentTarget.click());
   }
-  var LazySlidesComponent = import_react.lazy(() => import("./reveal-component-3YQBHWuD.js"));
+  var LazySlidesComponent = import_react.lazy(() => import("./reveal-component-BOEk3vwx.js"));
   const SlidesLayoutRenderer = ({ layout: e, setLayout: r, cells: c, mode: l }) => {
     var _a3;
     let u = useAtomValue(kioskModeAtom), d = l === "read" || u, f = useAtomValue(numColumnsAtom) > 1, [p, m] = (0, import_react.useState)(null), { slideCells: h, skippedIds: g, noOutputIds: _, slideTypes: v, startCellIndex: y } = (0, import_react.useMemo)(() => computeSlideCellsInfo(c, e), [

package/dist/{reveal-component-3YQBHWuD.js → reveal-component-BOEk3vwx.js}

@@ -9,7 +9,7 @@ import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
 import { lt as kioskModeAtom } from "./html-to-image-BwfcJJxA.js";
 import "./chunk-5FQGJX7Z-BbqSm5gU.js";
 import { u as createLucideIcon } from "./dist-Dk6PV_d3.js";
-import { a as DEFAULT_SLIDE_TYPE, c as Slide, ct as PanelResizeHandle, i as DEFAULT_DECK_TRANSITION, in as Expand, ot as Panel, rn as EyeOff, s as SlideSidebar, sn as Code, st as PanelGroup, t as useNotebookCodeAvailable } from "./code-visibility-DrFozVlm.js";
+import { a as DEFAULT_SLIDE_TYPE, c as Slide, ct as PanelResizeHandle, i as DEFAULT_DECK_TRANSITION, in as Expand, ot as Panel, rn as EyeOff, s as SlideSidebar, sn as Code, st as PanelGroup, t as useNotebookCodeAvailable } from "./code-visibility-BDDI_K4x.js";
 import { X as useDebouncedCallback } from "./input-OdWHkobi.js";
 import "./toDate-d8RCRrRd.js";
 import "./react-dom-BTJzcVJ9.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@marimo-team/islands",
-  "version": "0.23.10",
+  "version": "0.23.11-dev1",
   "main": "dist/main.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/src/components/editor/ai/ai-completion-editor.tsx

@@ -27,7 +27,7 @@ import { Label } from "@/components/ui/label";
 import { Switch } from "@/components/ui/switch";
 import { Tooltip } from "@/components/ui/tooltip";
 import { toast } from "@/components/ui/use-toast";
-import { stagedAICellsAtom } from "@/core/ai/staged-cells";
+import { stripWrappingBackticks } from "@/core/ai/strip-wrapping-backticks";
 import { type AiCompletionCell, includeOtherCellsAtom } from "@/core/ai/state";
 import type { CellId } from "@/core/cells/ids";
 import { getCodes } from "@/core/codemirror/copilot/getCodes";
@@ -45,6 +45,7 @@ import {
   RejectCompletionButton,
 } from "./completion-handlers";
 import { addContextCompletion, getAICompletionBody } from "./completion-utils";
+import { stagedAICellsAtom } from "@/core/ai/staged-cells";
 
 const Original = CodeMirrorMerge.Original;
 const Modified = CodeMirrorMerge.Modified;
@@ -123,7 +124,6 @@ export const AiCompletionEditor: React.FC<Props> = ({
     api: runtimeManager.getAiURL("completion").toString(),
     headers: runtimeManager.headers(),
     initialInput: initialPrompt,
-    streamProtocol: "text",
     // Throttle the messages and data updates to 100ms
     experimental_throttle: 100,
     body: {
@@ -143,13 +143,14 @@ export const AiCompletionEditor: React.FC<Props> = ({
       });
     },
     onFinish: (_prompt, completion) => {
-      // Remove trailing new lines
-      setCompletion(completion.trimEnd());
+      setCompletion(stripWrappingBackticks(completion).trimEnd());
     },
   });
 
   const inputRef = React.useRef<ReactCodeMirrorRef>(null);
-  const completion = untrimmedCompletion.trimEnd();
+  const completion = stripWrappingBackticks(untrimmedCompletion, {
+    streaming: isLoading,
+  }).trimEnd();
 
   const initialSubmit = useCallback(() => {
     if (triggerImmediately && !isLoading && initialPrompt) {

package/src/core/ai/__tests__/strip-wrapping-backticks.test.ts

@@ -0,0 +1,133 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { describe, expect, it } from "vitest";
+import { stripWrappingBackticks } from "../strip-wrapping-backticks";
+
+// Cases aligned with `without_wrapping_backticks` on a complete string (single chunk).
+const CASES: [string[], string][] = [
+  [["Hello", " world", "!"], "Hello world!"],
+  [["```", "print('hello')", "print('world')"], "print('hello')print('world')"],
+  [
+    ["print('hello')", "print('world')", "```"],
+    "print('hello')print('world')```",
+  ],
+  [["```", "print('hello')", "```"], "print('hello')"],
+  [["Hello", " ``` ", "world"], "Hello ``` world"],
+  [["``", "`print('hello')", "``", "`"], "print('hello')"],
+  [["``", "`", "\n", "print('hello')", "\n", "``", "`"], "print('hello')"],
+  [
+    ["```\n", "print('hello')", "print('world')", "\n```"],
+    "print('hello')print('world')",
+  ],
+  [
+    ["```\nprint('hello')\n", "print('world')\n```"],
+    "print('hello')\nprint('world')",
+  ],
+  [
+    ["```\n", "def test():\n    ", "return True\n```"],
+    "def test():\n    return True",
+  ],
+  [[], ""],
+  [["```idk```"], "idk"],
+  [["Hello world"], "Hello world"],
+  [
+    ["```python\n", "def hello():\n    ", "print('world')\n```"],
+    "def hello():\n    print('world')",
+  ],
+  [
+    ["```python", "\ndef hello():\n    ", "print('world')\n```"],
+    "def hello():\n    print('world')",
+  ],
+  [
+    ["```sql", "SELECT * FROM table", " WHERE id = 1", "```"],
+    "SELECT * FROM table WHERE id = 1",
+  ],
+  [
+    ["```sql\n", "SELECT * FROM table\n", "WHERE id = 1\n```"],
+    "SELECT * FROM table\nWHERE id = 1",
+  ],
+  [["```", "print('hello')", "```  "], "print('hello')  "],
+  [["```python\n", "print('hello')", "\n``` "], "print('hello') "],
+  [["```", "code", "```\t\n"], "code\t\n"],
+  [["  ```", "code", "```"], "code"],
+  [[" ```python\n", "code", "```"], "code"],
+  [["\t```", "code", "```"], "code"],
+  [["\n", "\n", "```\n", "code", "\n```\n"], "code\n"],
+  [["\n", "\n", "```python\n", "code", "\n```\n"], "code\n"],
+  [["\n``", "`python\n", "code", "\n```\n"], "code\n"],
+  [["\n`", "`", "`python\n", "code", "\n```\n"], "code\n"],
+  [["```python ", "code", "```"], " code"],
+  [["```python\t", "code", "```"], "\tcode"],
+  [["```\n", "```"], ""],
+  [["```python\n", "```"], ""],
+  [["```\n", "code"], "code"],
+  [["```python\n", "code"], "code"],
+  [["code", "\n```"], "code\n```"],
+  [
+    ["```\n", "x = 1\n", "```\n", "```\n", "y = 2\n", "```"],
+    "x = 1\n```\n```\ny = 2",
+  ],
+  [["```python\n", "s = 'use `backticks`'\n", "```"], "s = 'use `backticks`'"],
+  [["``", "`\n", "code\n", "``", "`"], "code"],
+  [["```", "python\n", "code\n", "```"], "code"],
+  [["```", "code", "```"], "code"],
+  [["prefix ", "```\n", "code\n", "```"], "prefix ```\ncode\n```"],
+  [["```\n", "code\n", "```", " suffix"], "code\n``` suffix"],
+  [["```\n\n", "code\n\n", "```"], "\ncode\n"],
+  [["```\n", "  code  \n", "```"], "  code  "],
+  [["```\n", "\tcode\t\n", "```"], "\tcode\t"],
+  [
+    ["```\n", "x\n", "```\n", "```python\n", "y\n", "```"],
+    "x\n```\n```python\ny",
+  ],
+  [["```javascript\n", "console.log()\n", "```"], "javascript\nconsole.log()"],
+  [["```markdown\n", "# Title\n", "```"], "# Title"],
+];
+
+describe("stripWrappingBackticks", () => {
+  it.each(CASES)("strips fences for %j", (chunks, expected) => {
+    expect(stripWrappingBackticks(chunks.join(""))).toBe(expected);
+  });
+});
+
+// In streaming mode, the opening fence is stripped as soon as it is
+// unambiguous, but the closing fence is left untouched (it may not have
+// arrived yet, or a trailing "```" may be content).
+const STREAMING_CASES: [string, string][] = [
+  // No fence: passthrough.
+  ["import pandas as pd", "import pandas as pd"],
+  // Plain opening fence stripped once the first line is terminated.
+  ["```\n", ""],
+  ["```\ncode", "code"],
+  ["```\ncode\nmore", "code\nmore"],
+  // Opening fence stripped, closing fence intentionally kept while streaming.
+  ["```\ncode\n```", "code\n```"],
+  ["```python\ncode\n```", "code\n```"],
+  // Language fences stripped as soon as the full identifier is present.
+  ["```python", ""],
+  ["```python\n", ""],
+  ["```python\ncode", "code"],
+  ["```sql\nSELECT 1", "SELECT 1"],
+  ["```markdown\n# Title", "# Title"],
+  // Leading whitespace before the fence.
+  ["  ```python\ncode", "code"],
+  // Partial language identifiers are left intact until they resolve.
+  ["```", "```"],
+  ["```p", "```p"],
+  ["```py", "```py"],
+  ["```pyth", "```pyth"],
+  ["```s", "```s"],
+  ["```mark", "```mark"],
+  // First-line content that is not a known language prefix is stripped as a
+  // plain fence even before a newline arrives.
+  ["```x", "x"],
+  ["```x = 1", "x = 1"],
+  // Unsupported language identifiers are kept as content (matches final mode).
+  ["```json\ncode", "json\ncode"],
+];
+
+describe("stripWrappingBackticks (streaming)", () => {
+  it.each(STREAMING_CASES)("strips opening fence for %j", (text, expected) => {
+    expect(stripWrappingBackticks(text, { streaming: true })).toBe(expected);
+  });
+});

package/src/core/ai/stream-completion-text.ts

@@ -0,0 +1,48 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { consumeStream, parseJsonEventStream, uiMessageChunkSchema } from "ai";
+import { stripWrappingBackticks } from "./strip-wrapping-backticks";
+
+/**
+ * Read an AI SDK UI message stream response and return the assistant text.
+ */
+export async function streamCompletionText(
+  response: Response,
+): Promise<string> {
+  if (!response.ok) {
+    throw new Error(await response.text());
+  }
+
+  if (!response.body) {
+    throw new Error("Failed to get response body");
+  }
+
+  let result = "";
+
+  await consumeStream({
+    stream: parseJsonEventStream({
+      stream: response.body,
+      schema: uiMessageChunkSchema,
+    }).pipeThrough(
+      new TransformStream({
+        transform(part) {
+          if (!part.success) {
+            throw part.error;
+          }
+
+          const streamPart = part.value;
+          if (streamPart.type === "text-delta") {
+            result += streamPart.delta;
+          } else if (streamPart.type === "error") {
+            throw new Error(streamPart.errorText);
+          }
+        },
+      }),
+    ),
+    onError: (error) => {
+      throw error;
+    },
+  });
+
+  return stripWrappingBackticks(result);
+}

package/src/core/ai/strip-wrapping-backticks.ts

@@ -0,0 +1,88 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+const LANGS = ["python", "sql", "markdown"] as const;
+
+interface StripOptions {
+  /**
+   * When true, the text is treated as a partial stream:
+   * - the opening fence is only stripped once we can be sure of its language
+   *   identifier (e.g. "```py" is left untouched since it may still become
+   *   "```python"), and
+   * - the closing fence is left in place, since it may not have arrived yet (or
+   *   a trailing "```" may be part of the content).
+   */
+  streaming?: boolean;
+}
+
+/**
+ * Removes wrapping markdown code fences from a completion string.
+ */
+export function stripWrappingBackticks(
+  text: string,
+  opts: StripOptions = {},
+): string {
+  const { streaming = false } = opts;
+  const leadingWhitespace = text.match(/^\s*/)?.[0] ?? "";
+  const rest = text.slice(leadingWhitespace.length);
+
+  let body: string | null = null;
+  let strippedOpening = false;
+
+  for (const lang of LANGS) {
+    if (rest.startsWith(`\`\`\`${lang}`)) {
+      strippedOpening = true;
+      body = rest.slice(3 + lang.length);
+      if (body.startsWith("\n")) {
+        body = body.slice(1);
+      }
+      break;
+    }
+  }
+
+  if (!strippedOpening && rest.startsWith("```")) {
+    const afterFence = rest.slice(3);
+    if (streaming && isPartialLanguageFence(afterFence)) {
+      return text;
+    }
+    strippedOpening = true;
+    body = afterFence;
+    if (body.startsWith("\n")) {
+      body = body.slice(1);
+    }
+  }
+
+  if (!strippedOpening || body === null) {
+    return text;
+  }
+
+  // While streaming, the closing fence may not have arrived yet, so leave the
+  // body as-is once the opening fence is removed.
+  if (streaming) {
+    return body;
+  }
+
+  const strippedEnd = body.trimEnd();
+  const trailingSpace = body.slice(strippedEnd.length);
+
+  if (strippedEnd.endsWith("\n```")) {
+    return strippedEnd.slice(0, -4) + trailingSpace;
+  }
+  if (strippedEnd.endsWith("```")) {
+    return strippedEnd.slice(0, -3) + trailingSpace;
+  }
+
+  return body;
+}
+
+/**
+ * Returns true if the text after an opening "```" has no terminating newline
+ * yet and could still become a known language identifier.
+ */
+function isPartialLanguageFence(afterFence: string): boolean {
+  // Once the first line is terminated, we already know it is not a known
+  // language fence (those are handled above), so treat it as a plain fence.
+  if (afterFence.includes("\n")) {
+    return false;
+  }
+  return LANGS.some((lang) => lang.startsWith(afterFence));
+}

package/src/core/codemirror/ai/request.ts

@@ -2,6 +2,7 @@
 
 import { waitForConnectionOpen } from "@/core/network/connection";
 import type { AiCompletionRequest } from "@/core/network/types";
+import { streamCompletionText } from "@/core/ai/stream-completion-text";
 import { getRuntimeManager } from "@/core/runtime/config";
 import type { LanguageAdapterType } from "../language/types";
 
@@ -47,20 +48,7 @@ ${opts.codeAfter}
 
   const firstLineIndent = opts.selection.match(/^\s*/)?.[0] || "";
 
-  const reader = response.body?.getReader();
-  if (!reader) {
-    throw new Error("Failed to get response reader");
-  }
-
-  let result = "";
-  // oxlint-disable-next-line no-constant-condition
-  while (true) {
-    const { done, value } = await reader.read();
-    if (done) {
-      break;
-    }
-    result += new TextDecoder().decode(value);
-  }
+  let result = await streamCompletionText(response);
 
   // Add back the indent if it was stripped, which can happen with
   // LLM responses