@marimo-team/islands 0.23.7-dev55 → 0.23.7-dev56

package/dist/{chat-ui-DCyW3OUK.js → chat-ui-D3XBept8.js}

@@ -6,13 +6,13 @@ import { _ as Logger, c as Objects, g as cn, l as useEventListener, t as Button
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
 import { r as toast } from "./copy-DLf4aN7I.js";
-import { C as AccordionContent, D as ChatBubbleIcon, En as Trash2, Ft as jotaiJsonStorage, G as cellErrorsAtom, Ht as allTablesAtom, In as ExternalLink, It as variablesAtom, Kt as getRequestClient, Lt as PluralWord, Nn as Info, Ot as moveToEndOfEditor, P as base64ToDataURL, Pn as FileText, S as Accordion, T as AccordionTrigger, Tn as Wrench, Tt as createVariableInfoElement, Ut as dataSourceConnectionsAtom, V as renderHTML, Wt as getTableType, Xt as singleFacet, Y as notebookAtom, _ as Boosts, _n as atomWithStorage, b as AIContextProvider, c as Popover, d as PopoverTrigger, f as isOutputEmpty, h as DatasourceContextProvider, jn as LoaderCircle, n as MarkdownRenderer, on as CellOutputId, s as Spinner, t as toPng, tn as ZodLocalStorage, u as PopoverContent, v as Sections, vt as displayCellName, w as AccordionItem, x as AIContextRegistry, y as contextToXml, yn as Anchor2, zn as CircleX } from "./html-to-image-40ZXSWP-.js";
+import { C as AccordionContent, D as ChatBubbleIcon, En as Trash2, Ft as jotaiJsonStorage, G as cellErrorsAtom, Ht as allTablesAtom, In as ExternalLink, It as variablesAtom, Kt as getRequestClient, Lt as PluralWord, Nn as Info, Ot as moveToEndOfEditor, P as base64ToDataURL, Pn as FileText, S as Accordion, T as AccordionTrigger, Tn as Wrench, Tt as createVariableInfoElement, Ut as dataSourceConnectionsAtom, V as renderHTML, Wt as getTableType, Xt as singleFacet, Y as notebookAtom, _ as Boosts, _n as atomWithStorage, b as AIContextProvider, c as Popover, d as PopoverTrigger, f as isOutputEmpty, h as DatasourceContextProvider, jn as LoaderCircle, n as MarkdownRenderer, on as CellOutputId, s as Spinner, t as toPng, tn as ZodLocalStorage, u as PopoverContent, v as Sections, vt as displayCellName, w as AccordionItem, x as AIContextRegistry, y as contextToXml, yn as Anchor2, zn as CircleX } from "./html-to-image-CpggM7u1.js";
 import "./chunk-5FQGJX7Z-BOg95xG5.js";
 import { u as createLucideIcon } from "./dist-D3ZI9nhS.js";
 import { F as X, L as ChevronDown, S as logNever, t as Strings } from "./strings-BiIhGaI8.js";
 import { a as NumberField, b as DropdownMenuTrigger, d as DropdownMenuContent, p as DropdownMenuItem, r as Input, u as DropdownMenu } from "./input-D4kjoQUB.js";
-import { p as isUrl, v as CircleQuestionMark } from "./toDate-CJWlVNGD.js";
-import { a as MarimoIncomingMessageEvent, d as File$1, n as blobToString, t as processOutput, u as deserializeBlob } from "./process-output-CCeeXIBd.js";
+import { p as isUrl, v as CircleQuestionMark } from "./toDate-CIpC_34u.js";
+import { a as MarimoIncomingMessageEvent, d as File$1, n as blobToString, t as processOutput, u as deserializeBlob } from "./process-output-X8TR20AK.js";
 import "./react-dom-BWRJ_g_k.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
 import { A as looseObject, B as union, C as any, D as discriminatedUnion, E as custom, H as safeParseAsync, I as record, L as strictObject, M as never, N as number, O as lazy, P as object$1, R as string, S as _null, T as boolean, V as unknown, W as toJSONSchema, b as _enum, k as literal, w as array$1, x as _instanceof } from "./zod-BWkcDORu.js";

package/dist/{code-visibility-CJ7U5FE0.js → code-visibility-nPfbiA_L.js}

@@ -6,14 +6,14 @@ import { _ as Logger, c as Objects, g as cn, h as Events, m as useComposedRefs,
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
 import { n as Copy, r as toast, t as copyToClipboard } from "./copy-DLf4aN7I.js";
-import { $ as useCellActions, $t as getTracebackInfo, A as ChevronRightIcon, An as Minus, C as AccordionContent, En as Trash2, Ft as jotaiJsonStorage, Hn as esm_default, I as base64ToUint8Array, In as ExternalLink, Jt as useRequestClient, Kt as getRequestClient, Lt as PluralWord, M as PinLeftIcon, Mt as useChromeActions, N as PinRightIcon, O as CheckIcon, Pn as FileText, Pt as adaptForLocalStorage, Qt as extractAllTracebackInfo, R as extractBase64FromDataURL, S as Accordion, St as Checkbox, T as AccordionTrigger, Tn as Wrench, Un as import_lib, V as renderHTML, Vn as Braces, Y as notebookAtom, Zt as elementContainsMarimoCellFile, _n as atomWithStorage, _t as getCellDomProps, at as AnsiUp, bn as Close$1, c as Popover, cn as SCRATCH_CELL_ID, ct as kioskModeAtom, d as PopoverTrigger, dt as outputIsLoading, et as useCellIds, g as getDatasourceContext, gt as DATA_CELL_ID, hn as jsonToTSV, ht as sanitizeHtml, j as DotFilledIcon, jn as LoaderCircle, kn as NotebookPen, kt as goToCellLine, l as PopoverClose$1, lt as useInstallAllowed, m as useExpandedOutput, mn as jsonToMarkdown, n as MarkdownRenderer, nn as filenameAtom, on as CellOutputId, pn as jsonParseWithSpecialChar, q as getCellEditorView, s as Spinner, sn as HTMLCellId, t as toPng, tt as useCellNames, u as PopoverContent, ut as viewStateAtom, vt as displayCellName, w as AccordionItem, z as isDataURLString, zn as CircleX, zt as DATA_TYPE_ICON, __tla as __tla_0 } from "./html-to-image-40ZXSWP-.js";
+import { $ as useCellActions, $t as getTracebackInfo, A as ChevronRightIcon, An as Minus, C as AccordionContent, En as Trash2, Ft as jotaiJsonStorage, Hn as esm_default, I as base64ToUint8Array, In as ExternalLink, Jt as useRequestClient, Kt as getRequestClient, Lt as PluralWord, M as PinLeftIcon, Mt as useChromeActions, N as PinRightIcon, O as CheckIcon, Pn as FileText, Pt as adaptForLocalStorage, Qt as extractAllTracebackInfo, R as extractBase64FromDataURL, S as Accordion, St as Checkbox, T as AccordionTrigger, Tn as Wrench, Un as import_lib, V as renderHTML, Vn as Braces, Y as notebookAtom, Zt as elementContainsMarimoCellFile, _n as atomWithStorage, _t as getCellDomProps, at as AnsiUp, bn as Close$1, c as Popover, cn as SCRATCH_CELL_ID, ct as kioskModeAtom, d as PopoverTrigger, dt as outputIsLoading, et as useCellIds, g as getDatasourceContext, gt as DATA_CELL_ID, hn as jsonToTSV, ht as sanitizeHtml, j as DotFilledIcon, jn as LoaderCircle, kn as NotebookPen, kt as goToCellLine, l as PopoverClose$1, lt as useInstallAllowed, m as useExpandedOutput, mn as jsonToMarkdown, n as MarkdownRenderer, nn as filenameAtom, on as CellOutputId, pn as jsonParseWithSpecialChar, q as getCellEditorView, s as Spinner, sn as HTMLCellId, t as toPng, tt as useCellNames, u as PopoverContent, ut as viewStateAtom, vt as displayCellName, w as AccordionItem, z as isDataURLString, zn as CircleX, zt as DATA_TYPE_ICON, __tla as __tla_0 } from "./html-to-image-CpggM7u1.js";
 import { o as useSize, u as createLucideIcon } from "./dist-D3ZI9nhS.js";
 import { i as createReducerAndAtoms, r as Badge } from "./useLifecycle-BF6-z62y.js";
 import { a as ListFilter, i as Table$1, o as ChartPie, t as $896ba0a80a8f4d36$export$85fd5fdf27bacc79 } from "./useDateFormatter-B_9k85Ex.js";
 import { t as Check } from "./check-BcUIXnUT.js";
 import { C as $a916eb452884faea$export$b7a616150fdb9f44, F as X, L as ChevronDown, M as usePrevious$1, N as useDirection, S as logNever, T as $18f2051aff69b9bf$export$43bb16f9c6d9e3f7, _ as menuItemVariants, b as menuSubTriggerVariants, g as menuControlVariants, h as menuControlCheckVariants, i as SelectContent, l as SelectTrigger, m as menuContentCommon, o as SelectItem, p as MENU_ITEM_DISABLED, r as Select, t as Strings, u as SelectValue, v as menuLabelVariants, x as assertNever, y as menuSeparatorVariants } from "./strings-BiIhGaI8.js";
 import { $t as Portal, G as marked, Gt as Anchor2, Jt as Content2$1, Kt as Arrow2, Qt as Label, Xt as Item2$2, Yt as Group, Zt as ItemIndicator, _ as DropdownMenuSub, a as NumberField, an as SubContent, b as DropdownMenuTrigger, c as prettyNumber, cn as Item, d as DropdownMenuContent, dn as Search, en as RadioGroup$2, fn as Circle, g as DropdownMenuSeparator, h as DropdownMenuPortal, in as Sub, l as prettyScientificNumber, ln as Root$2, m as DropdownMenuLabel, nn as Root3, o as maxFractionalDigits, on as SubTrigger, p as DropdownMenuItem, pn as ChevronRight, qt as CheckboxItem, r as Input, rn as Separator$1, s as prettyEngineeringNumber, sn as createMenuScope, tn as RadioItem, u as DropdownMenu, un as createRovingFocusGroupScope, v as DropdownMenuSubContent, y as DropdownMenuSubTrigger } from "./input-D4kjoQUB.js";
-import { _ as isWasm, a as millisecondsInMinute, c as asRemoteURL, d as isStaticNotebook, i as millisecondsInHour, n as constructFrom, o as millisecondsInSecond, p as isUrl, r as millisecondsInDay, s as millisecondsInWeek, t as toDate } from "./toDate-CJWlVNGD.js";
+import { _ as isWasm, a as millisecondsInMinute, c as asRemoteURL, d as isStaticNotebook, i as millisecondsInHour, n as constructFrom, o as millisecondsInSecond, p as isUrl, r as millisecondsInDay, s as millisecondsInWeek, t as toDate } from "./toDate-CIpC_34u.js";
 import { n as clsx_default } from "./clsx-CyyyQ8Ue.js";
 import { t as require_react_dom } from "./react-dom-BWRJ_g_k.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
@@ -28,7 +28,7 @@ import { t as invariant } from "./invariant-UcGKQEhF.js";
 import { c as uniqueBy, t as Arrays } from "./arrays-CldYf7p7.js";
 import { n as once, t as memoizeLastValue } from "./once-rJImu7SE.js";
 import { a as parser } from "./dist-nuW5EDYT.js";
-import { a as isValid, i as AlertTitle, n as Alert, t as arrow } from "./formats-CpgZM9BM.js";
+import { a as isValid, i as AlertTitle, n as Alert, t as arrow } from "./formats-Dsy9kkZu.js";
 import { n as memoize$1, t as isPropValid } from "./emotion-is-prop-valid.esm-DtW2o230.js";
 import { t as ErrorBoundary } from "./ErrorBoundary-D3wrPNma.js";
 import { n as formats } from "./vega-loader.browser-3_z8GoFC.js";
@@ -26003,7 +26003,7 @@ ${_}`,
       return Logger.warn("Failed to get version from mount config"), null;
     }
   }
-  marimoVersionAtom = atom(getVersionFromMountConfig() || "0.23.7-dev55");
+  marimoVersionAtom = atom(getVersionFromMountConfig() || "0.23.7-dev56");
   showCodeInRunModeAtom = atom(true);
   atom(null);
   var import_compiler_runtime = require_compiler_runtime();

package/dist/{formats-CpgZM9BM.js → formats-Dsy9kkZu.js}

@@ -2,7 +2,7 @@ import { s as __toESM } from "./chunk-BNovOVIE.js";
 import { g as cn, r as cva } from "./button-Dj4BTre0.js";
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
-import { t as toDate } from "./toDate-CJWlVNGD.js";
+import { t as toDate } from "./toDate-CIpC_34u.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
 import { i as tableFromIPC } from "./loader-Dr8Qem8p.js";
 function isDate(e) {

package/dist/{html-to-image-40ZXSWP-.js → html-to-image-CpggM7u1.js}

@@ -12,7 +12,7 @@ import { a as Type, c as Calendar, i as createReducerAndAtoms, o as ToggleLeft,
 import { t as Check } from "./check-BcUIXnUT.js";
 import { F as X, L as ChevronDown, M as usePrevious, N as useDirection, P as createCollection, S as logNever, a as SelectGroup, c as SelectSeparator, i as SelectContent, l as SelectTrigger, o as SelectItem, r as Select, s as SelectLabel, t as Strings, u as SelectValue, x as assertNever } from "./strings-BiIhGaI8.js";
 import { G as marked, W as useNonce, mt as $f6c31cce2adf654f$export$45712eceda6fad21, q as useDebouncedCallback } from "./input-D4kjoQUB.js";
-import { _ as isWasm, g as Deferred, h as waitForConnectionOpen, l as getRuntimeManager, t as toDate, v as CircleQuestionMark } from "./toDate-CJWlVNGD.js";
+import { _ as isWasm, g as Deferred, h as waitForConnectionOpen, l as getRuntimeManager, t as toDate, v as CircleQuestionMark } from "./toDate-CIpC_34u.js";
 import { t as require_react_dom } from "./react-dom-BWRJ_g_k.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
 import { $ as StyleNamespace, B as union, N as number, P as object, R as string, T as boolean, X as withFullScreenAsRoot, Y as MAX_HEIGHT_OFFSET, Z as withSmartCollisionBoundary, _t as Primitive, at as FocusScope, ct as Root$1, dt as Presence, et as hideOthers, ft as useControllableState, gt as createContextScope, it as Portal, k as literal, lt as useCallbackRef, mt as composeEventHandlers, ot as Branch, pt as useLayoutEffect2, rt as useFocusGuards, st as DismissableLayer, tt as Combination_default, ut as useId, vt as dispatchDiscreteCustomEvent, w as array } from "./zod-BWkcDORu.js";

package/dist/main.js

@@ -22,18 +22,18 @@ import { _ as Logger, c as Objects, g as cn, h as Events, i as NOT_SET, l as use
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
 import { n as Copy, r as toast, t as copyToClipboard } from "./copy-DLf4aN7I.js";
-import { $ as useCellActions, At as DeferredRequestRegistry, B as safeExtractSetUIElementMessageBuffers, Bn as CircleAlert, Bt as getDataTypeColor, C as AccordionContent, Cn as Root2$1, Ct as customPythonLanguageSupport, Dn as Table2, Dt as Paths, E as BorderAllIcon, En as Trash2, Et as PathBuilder, F as base64ToDataView, Fn as Eye, Ft as jotaiJsonStorage, Gt as convertStatsName, H as getMarimoExportContext, J as getCellNames, Jt as useRequestClient, K as createActions, Kt as getRequestClient, L as dataViewToBase64, Ln as Database, Mn as Layers, Nn as Info, Nt as repl, On as PaintRoller, Pn as FileText, Q as reducer, Rn as Columns2, Rt as PluralWords, S as Accordion, Sn as Item$1, St as Checkbox, T as AccordionTrigger, U as hasTrustedExportContext, V as renderHTML, Vn as Braces, Vt as require_client, W as hasRunAnyCellAtom, X as notebookOutline, Y as notebookAtom, Yt as isUninstantiated, Z as numColumnsAtom, _n as atomWithStorage, a as useLastFocusedCellId, an as parseInitialValue, bt as isInternalCellName, ct as kioskModeAtom, dn as OBJECT_ID_ATTR, dt as outputIsLoading, en as NotebookScopedLocalStorage, et as useCellIds, f as isOutputEmpty, fn as RANDOM_ID_ATTR, ft as outputIsStale, gn as atomWithReducer, i as useCellFocusActions, in as parseDataset, jn as LoaderCircle, jt as generateUUID, k as ChevronDownIcon, ln as UIElementId, mt as headingToIdentifier, nt as createCell, o as maybeAddAltairImport, ot as getInitialAppMode, p as useExpandedConsoleOutput, pn as jsonParseWithSpecialChar, pt as isErrorMime, qt as requestClientAtom, r as LazyAnyLanguageCodeMirror, rn as parseAttrValue, s as Spinner, sn as HTMLCellId, st as initialModeAtom, un as findCellId, vn as selectAtom, w as AccordionItem, wn as Trigger2, wt as MarkdownLanguageAdapter, xn as Content2, xt as normalizeName, yt as getValidName, zt as DATA_TYPE_ICON, __tla as __tla_0 } from "./html-to-image-40ZXSWP-.js";
+import { $ as useCellActions, At as DeferredRequestRegistry, B as safeExtractSetUIElementMessageBuffers, Bn as CircleAlert, Bt as getDataTypeColor, C as AccordionContent, Cn as Root2$1, Ct as customPythonLanguageSupport, Dn as Table2, Dt as Paths, E as BorderAllIcon, En as Trash2, Et as PathBuilder, F as base64ToDataView, Fn as Eye, Ft as jotaiJsonStorage, Gt as convertStatsName, H as getMarimoExportContext, J as getCellNames, Jt as useRequestClient, K as createActions, Kt as getRequestClient, L as dataViewToBase64, Ln as Database, Mn as Layers, Nn as Info, Nt as repl, On as PaintRoller, Pn as FileText, Q as reducer, Rn as Columns2, Rt as PluralWords, S as Accordion, Sn as Item$1, St as Checkbox, T as AccordionTrigger, U as hasTrustedExportContext, V as renderHTML, Vn as Braces, Vt as require_client, W as hasRunAnyCellAtom, X as notebookOutline, Y as notebookAtom, Yt as isUninstantiated, Z as numColumnsAtom, _n as atomWithStorage, a as useLastFocusedCellId, an as parseInitialValue, bt as isInternalCellName, ct as kioskModeAtom, dn as OBJECT_ID_ATTR, dt as outputIsLoading, en as NotebookScopedLocalStorage, et as useCellIds, f as isOutputEmpty, fn as RANDOM_ID_ATTR, ft as outputIsStale, gn as atomWithReducer, i as useCellFocusActions, in as parseDataset, jn as LoaderCircle, jt as generateUUID, k as ChevronDownIcon, ln as UIElementId, mt as headingToIdentifier, nt as createCell, o as maybeAddAltairImport, ot as getInitialAppMode, p as useExpandedConsoleOutput, pn as jsonParseWithSpecialChar, pt as isErrorMime, qt as requestClientAtom, r as LazyAnyLanguageCodeMirror, rn as parseAttrValue, s as Spinner, sn as HTMLCellId, st as initialModeAtom, un as findCellId, vn as selectAtom, w as AccordionItem, wn as Trigger2, wt as MarkdownLanguageAdapter, xn as Content2, xt as normalizeName, yt as getValidName, zt as DATA_TYPE_ICON, __tla as __tla_0 } from "./html-to-image-CpggM7u1.js";
 import { __tla as __tla_1 } from "./chunk-5FQGJX7Z-BOg95xG5.js";
 import { o as useSize, s as Root$2, u as createLucideIcon } from "./dist-D3ZI9nhS.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-hVaaZsY5.js";
-import { $ as filtersToFilterGroup, A as contextAwarePanelOwner, At as Funnel, B as TableCell, Bt as ChevronLeft, C as prettifyRowCount, Ct as useOverflowDetection, D as ContextAwarePanelItem, Dt as EmotionCacheProvider, E as ComboboxItem, Et as HtmlOutput, F as Toggle, Ft as Code, G as generateColumns, H as TableHeader, I as Fill, It as ChevronsUpDown, J as ColumnChartContext, K as inferFieldTypes, L as Provider$1, Lt as ChevronsRight, M as isCellAwareAtom, N as SlotNames, Nt as Ellipsis, O as PANEL_TYPES, Ot as TextWrap, P as slotsController, Pt as Download, Q as usePrevious$1, R as Table, Rt as ChevronsLeft, S as prettifyRowColumnCount, St as LazyVegaEmbed, T as Combobox, Tt as Kbd, U as TableRow, V as TableHead, Vt as ArrowDownWideNarrow, W as NAMELESS_COLUMN_PREFIX, X as DelayMount, Y as ColumnChartSpecModel, Z as useIntersectionObserver, _ as downloadBlob, _t as TabsList, at as TOO_MANY_ROWS, b as Progress, bt as ChartInfoState, c as Slide, ct as Command, d as JsonOutput, dt as CommandItem, et as getPageIndexForRow, f as OutputArea, ft as CommandList, g as ADD_PRINTING_CLASS, gt as TabsContent, h as InstallPackageButton, ht as Tabs, it as SELECT_COLUMN_ID, j as contextAwarePanelType, jt as EyeOff, k as contextAwarePanelOpen, kt as GripHorizontal, l as RadioGroup, lt as CommandEmpty, m as DataTable, mt as Maps, n as marimoVersionAtom, nt as loadTableData, o as SLIDE_TYPE_OPTIONS_BY_VALUE, ot as toFieldTypes, p as OutputRenderer, pt as CommandSeparator, q as renderCellValue, r as showCodeInRunModeAtom, rt as INDEX_COLUMN_NAME, st as getMimeValues, t as useNotebookCodeAvailable, tt as loadTableAndRawData, u as RadioGroupItem, ut as CommandInput, v as downloadByURL, vt as TabsTrigger, w as useInternalStateWithSync, wt as RenderTextWithLinks, x as Filenames, xt as ChartLoadingState, y as downloadHTMLAsImage, yt as ChartErrorState, z as TableBody, zt as ChevronsDownUp, __tla as __tla_2 } from "./code-visibility-CJ7U5FE0.js";
+import { $ as filtersToFilterGroup, A as contextAwarePanelOwner, At as Funnel, B as TableCell, Bt as ChevronLeft, C as prettifyRowCount, Ct as useOverflowDetection, D as ContextAwarePanelItem, Dt as EmotionCacheProvider, E as ComboboxItem, Et as HtmlOutput, F as Toggle, Ft as Code, G as generateColumns, H as TableHeader, I as Fill, It as ChevronsUpDown, J as ColumnChartContext, K as inferFieldTypes, L as Provider$1, Lt as ChevronsRight, M as isCellAwareAtom, N as SlotNames, Nt as Ellipsis, O as PANEL_TYPES, Ot as TextWrap, P as slotsController, Pt as Download, Q as usePrevious$1, R as Table, Rt as ChevronsLeft, S as prettifyRowColumnCount, St as LazyVegaEmbed, T as Combobox, Tt as Kbd, U as TableRow, V as TableHead, Vt as ArrowDownWideNarrow, W as NAMELESS_COLUMN_PREFIX, X as DelayMount, Y as ColumnChartSpecModel, Z as useIntersectionObserver, _ as downloadBlob, _t as TabsList, at as TOO_MANY_ROWS, b as Progress, bt as ChartInfoState, c as Slide, ct as Command, d as JsonOutput, dt as CommandItem, et as getPageIndexForRow, f as OutputArea, ft as CommandList, g as ADD_PRINTING_CLASS, gt as TabsContent, h as InstallPackageButton, ht as Tabs, it as SELECT_COLUMN_ID, j as contextAwarePanelType, jt as EyeOff, k as contextAwarePanelOpen, kt as GripHorizontal, l as RadioGroup, lt as CommandEmpty, m as DataTable, mt as Maps, n as marimoVersionAtom, nt as loadTableData, o as SLIDE_TYPE_OPTIONS_BY_VALUE, ot as toFieldTypes, p as OutputRenderer, pt as CommandSeparator, q as renderCellValue, r as showCodeInRunModeAtom, rt as INDEX_COLUMN_NAME, st as getMimeValues, t as useNotebookCodeAvailable, tt as loadTableAndRawData, u as RadioGroupItem, ut as CommandInput, v as downloadByURL, vt as TabsTrigger, w as useInternalStateWithSync, wt as RenderTextWithLinks, x as Filenames, xt as ChartLoadingState, y as downloadHTMLAsImage, yt as ChartErrorState, z as TableBody, zt as ChevronsDownUp, __tla as __tla_2 } from "./code-visibility-nPfbiA_L.js";
 import { c as Calendar, i as createReducerAndAtoms, n as useOnUnmount, o as ToggleLeft, t as useOnMount } from "./useLifecycle-BF6-z62y.js";
 import { n as $fb18d541ea1ad717$export$ad991b66133851cf, r as $5a387cc49350e6db$export$722debc0e56fea39, t as $896ba0a80a8f4d36$export$85fd5fdf27bacc79 } from "./useDateFormatter-B_9k85Ex.js";
 import { t as Check } from "./check-BcUIXnUT.js";
 import { A as Trigger$1, C as $a916eb452884faea$export$b7a616150fdb9f44, D as $b5e257d569688ac6$export$535bd6ca7f90a273, E as $18f2051aff69b9bf$export$a54013f0d02a8f82, F as X, L as ChevronDown, M as usePrevious$2, N as useDirection, P as createCollection, S as logNever, T as $18f2051aff69b9bf$export$43bb16f9c6d9e3f7, a as SelectGroup, c as SelectSeparator, d as NativeSelect, f as selectStyles, i as SelectContent, j as clamp$2, k as Icon, l as SelectTrigger, n as capitalize, o as SelectItem, r as Select, s as SelectLabel, t as Strings, u as SelectValue, w as $488c6ddbf4ef74c2$export$cc77c4ff7e8673c5, x as assertNever } from "./strings-BiIhGaI8.js";
 import { $ as $e5be200c675c3b3a$export$aca958c65c314e6c, A as $d2b4bc8c273e7be6$export$24d547caef80ccd1, At as $c87311424ea30a05$export$fedb369cb70207f1, B as $64fa3d84918910a7$export$c62b8e45d58ddad9, Bt as $431fbd86ca7dc216$export$f21a1ffae260145a, C as $a049562f99e7db0e$export$eb2fcfdbd7ba97d4, Ct as $8ae05eaa5c114e9c$export$7f54fc3180508a52, D as $ee014567cb39d3f0$export$ff05c3ac10437e03, Dt as $c87311424ea30a05$export$78551043582a6a98, E as $ee014567cb39d3f0$export$f551688fc98f2e09, Et as $c87311424ea30a05$export$6446a186d09e379e, F as $64fa3d84918910a7$export$2881499e37b75b9a, Ft as $d4ee10de306f2510$export$b4f377a2b6254582, H as $64fa3d84918910a7$export$ef03459518577ad4, Ht as $bdb11010cef70236$export$b4cc09c592e8fdb8, I as $64fa3d84918910a7$export$29f1550f4b0d4415, It as $d4ee10de306f2510$export$cd4e5573fbe2b576, J as $d2e8511e6f209edf$export$e908e06f4b8e3402, K as useDebounceControlledState, L as $64fa3d84918910a7$export$4d86445c2cf5e3, Lt as $d4ee10de306f2510$export$e58f029f0fbfdb29, M as $01b77f81d0f07f68$export$75b6ee27786ba447, Mt as $65484d02dcb7eb3e$export$457c3d6518dd4c6f, N as $01b77f81d0f07f68$export$b04be29aa201d4f5, Nt as $3ef42575df84b30b$export$9d1611c77c2fe928, O as $514c0188e459b4c0$export$5f1af8db9871e1d6, Ot as $c87311424ea30a05$export$9ac100e40613ea10, P as $f39a9eba43920ace$export$b5d7cc18bb8d2b59, Pt as $d4ee10de306f2510$export$4282f70798064fe0, Q as $e5be200c675c3b3a$export$a763b9476acd3eb, R as $64fa3d84918910a7$export$9d4c57ee4c6ffdd8, Rt as $f4e2df6bd15f8569$export$98658e8c59125e6a, S as $3985021b0ad6602f$export$f5b8910cec6cf069, St as $e9faafb641e167db$export$90fc3a17d93f704c, T as $d3e0e05bdfcf66bd$export$c24727297075ec6a, Tt as $313b98861ee5dd6c$export$d6875122194c7b44, U as $64fa3d84918910a7$export$fabf2dc03a41866e, Ut as $bdb11010cef70236$export$f680877a34711e37, V as $64fa3d84918910a7$export$df3a06d6289f983e, Vt as $ff5963eb1fccf552$export$e08e3b67e392101e, Wt as $f0a04ccd8dbdd83b$export$e5c5a5f917a5871c, X as $e93e671b31057976$export$b8473d3665f3a75a, Y as $2baaea4c71418dea$export$294aa081a6c6f55d, Z as $e5be200c675c3b3a$export$75ee7c75d68f5b0e, _t as $9446cca9a3875146$export$7d15b64cf5a3a4c4, a as NumberField, at as $6c7bd7858deea686$export$cd11ab140839f11d, b as DropdownMenuTrigger, bt as $b4b717babfbb907b$export$bebd5a1431fec25d, c as prettyNumber, ct as $6db58dc88e78b024$export$2f817fcdc4b89ae0, d as DropdownMenuContent, dt as $9ab94262bd0047c7$export$420e68273165f4ec, et as $e5be200c675c3b3a$export$dad6ae84456c676a, f as DropdownMenuGroup, fn as Circle, ft as $3ad3f6e1647bc98d$export$80f3e147d781571c, g as DropdownMenuSeparator, gt as $ae1eeba8b9eafd08$export$5165eccb35aaadb5, ht as _class_private_field_init, i as OnBlurredInput, it as $701a24aa0da5b062$export$ea18c227d4417cc3, j as $d2b4bc8c273e7be6$export$353f5b6fc5456de1, jt as $7215afc6de606d6b$export$de79e2c695e052f3, k as $514c0188e459b4c0$export$9afb8bc826b033ea, kt as $c87311424ea30a05$export$a11b0059900ceec8, l as prettyScientificNumber, lt as $5b160d28a433310d$export$c17fa47878dc55b6, m as DropdownMenuLabel, mt as $f6c31cce2adf654f$export$45712eceda6fad21, n as DebouncedNumberInput, nt as $319e236875307eab$export$a9b970dcc4ae71a9, ot as $fca6afa0e843324b$export$87b761675e8eaa10, p as DropdownMenuItem, pn as ChevronRight, pt as $507fabe10e71c6fb$export$630ff653c5ada6a9, q as useDebouncedCallback, r as Input, rt as $f7dceffc5ad7768b$export$4e328f61c538687f, st as $fca6afa0e843324b$export$f12b703ca79dfbb1, t as DebouncedInput, tt as $e5be200c675c3b3a$export$fc1a364ae1f3ff10, u as DropdownMenu, ut as $6179b936705e76d3$export$ae780daf29e6d456, vt as $458b0a5536c1a7cf$export$40bfa8c7b0832715, w as $a049562f99e7db0e$export$f9c6924e160136d1, wt as $df56164dff5785e2$export$4338b53315abf666, x as $3985021b0ad6602f$export$37fb8590cf2c088c, xt as $99facab73266f662$export$5add1d006293d136, yt as $b4b717babfbb907b$export$4c063cf1350e6fed, z as $64fa3d84918910a7$export$c245e6201fed2f75, zt as $431fbd86ca7dc216$export$b204af158042fbac } from "./input-D4kjoQUB.js";
-import { _ as isWasm, c as asRemoteURL, d as isStaticNotebook, f as appendQueryParams, g as Deferred, m as require_cuid2, u as getStaticVirtualFiles, v as CircleQuestionMark } from "./toDate-CJWlVNGD.js";
-import { a as MarimoIncomingMessageEvent, c as MarimoValueUpdateEvent, d as File, i as PythonIcon, l as createInputEvent, n as blobToString, o as MarimoValueInputEvent, r as filesToBase64, s as MarimoValueReadyEvent, t as processOutput, u as deserializeBlob } from "./process-output-CCeeXIBd.js";
+import { _ as isWasm, c as asRemoteURL, d as isStaticNotebook, f as appendQueryParams, g as Deferred, m as require_cuid2, u as getStaticVirtualFiles, v as CircleQuestionMark } from "./toDate-CIpC_34u.js";
+import { a as MarimoIncomingMessageEvent, c as MarimoValueUpdateEvent, d as File, i as PythonIcon, l as createInputEvent, n as blobToString, o as MarimoValueInputEvent, r as filesToBase64, s as MarimoValueReadyEvent, t as processOutput, u as deserializeBlob } from "./process-output-X8TR20AK.js";
 import { i as Pencil, n as Trash, r as Plus, t as BulkEdit } from "./types-Dzuoc3LN.js";
 import { t as require_react_dom } from "./react-dom-BWRJ_g_k.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
@@ -59,7 +59,7 @@ import "./dist-DGAfI2rB.js";
 import { r as python } from "./dist-nuW5EDYT.js";
 import { n as minimalSetup, t as esm_default } from "./esm-CWp0KQeK.js";
 import "./purify.es-DT70lfR0.js";
-import { i as AlertTitle, n as Alert, r as AlertDescription } from "./formats-CpgZM9BM.js";
+import { i as AlertTitle, n as Alert, r as AlertDescription } from "./formats-Dsy9kkZu.js";
 import { n as require_prop_types, t as ErrorBoundary } from "./ErrorBoundary-D3wrPNma.js";
 import "./vega-loader.browser-3_z8GoFC.js";
 import { a as getContainerWidth, n as vegaLoadData, o as getVegaFieldTypes, s as tooltipHandler } from "./loader-Dr8Qem8p.js";
@@ -13569,7 +13569,7 @@ let __tla = Promise.all([
       };
     }
   };
-  var LazyChatbot = import_react.lazy(() => import("./chat-ui-DCyW3OUK.js").then((e) => ({
+  var LazyChatbot = import_react.lazy(() => import("./chat-ui-D3XBept8.js").then((e) => ({
     default: e.Chatbot
   }))), messageSchema = array(object({
     id: string(),
@@ -35036,7 +35036,7 @@ ${c}
         return true;
     }
   }
-  var LazyVegaComponent = import_react.lazy(() => import("./vega-component-BtvQ-Kc4.js")), VegaPlugin = class {
+  var LazyVegaComponent = import_react.lazy(() => import("./vega-component-cSdqoAxe.js")), VegaPlugin = class {
     constructor() {
       __publicField(this, "tagName", "marimo-vega");
       __publicField(this, "validator", object({
@@ -44606,7 +44606,7 @@ ${c}
     if (l && l !== "slide") return l;
     if (c == null ? void 0 : c.has(e)) return "skip";
   }
-  var LazySlidesComponent = import_react.lazy(() => import("./reveal-component-Bopa1DsA.js"));
+  var LazySlidesComponent = import_react.lazy(() => import("./reveal-component-DuRlSS4j.js"));
   const SlidesLayoutPlugin = {
     type: "slides",
     name: "Slides",

package/dist/{process-output-CCeeXIBd.js → process-output-X8TR20AK.js}

@@ -1,6 +1,6 @@
 import { s as __toESM } from "./chunk-BNovOVIE.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
-import { it as parseHtmlContent, rt as ansiToPlainText } from "./html-to-image-40ZXSWP-.js";
+import { it as parseHtmlContent, rt as ansiToPlainText } from "./html-to-image-CpggM7u1.js";
 import { u as createLucideIcon } from "./dist-D3ZI9nhS.js";
 import { t as Strings } from "./strings-BiIhGaI8.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";

package/dist/{reveal-component-Bopa1DsA.js → reveal-component-DuRlSS4j.js}

@@ -6,11 +6,11 @@ import { s as __toESM } from "./chunk-BNovOVIE.js";
 import { _ as Logger, g as cn, h as Events, l as useEventListener, t as Button } from "./button-Dj4BTre0.js";
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
-import "./html-to-image-40ZXSWP-.js";
+import "./html-to-image-CpggM7u1.js";
 import "./chunk-5FQGJX7Z-BOg95xG5.js";
-import { Ft as Code, Mt as Expand, a as DEFAULT_SLIDE_TYPE, c as Slide, i as DEFAULT_DECK_TRANSITION, jt as EyeOff, s as SlideSidebar, t as useNotebookCodeAvailable } from "./code-visibility-CJ7U5FE0.js";
+import { Ft as Code, Mt as Expand, a as DEFAULT_SLIDE_TYPE, c as Slide, i as DEFAULT_DECK_TRANSITION, jt as EyeOff, s as SlideSidebar, t as useNotebookCodeAvailable } from "./code-visibility-nPfbiA_L.js";
 import "./input-D4kjoQUB.js";
-import "./toDate-CJWlVNGD.js";
+import "./toDate-CIpC_34u.js";
 import "./react-dom-BWRJ_g_k.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
 import "./zod-BWkcDORu.js";

package/dist/{toDate-CJWlVNGD.js → toDate-CIpC_34u.js}

@@ -512,15 +512,21 @@ var RuntimeManager = class {
   get isSameOrigin() {
     return this.httpURL.origin === window.location.origin;
   }
-  formatHttpURL(e, b, x = true) {
-    e || (e = "");
+  get isServerless() {
+    return isWasm() || isIslands() || isStaticNotebook();
+  }
+  formatHttpURL({ path: e = "", searchParams: b, restrictToKnownQueryParams: x = true }) {
     let S = this.httpURL, w = new URLSearchParams(window.location.search);
     if (b) for (let [e2, x2] of b.entries()) S.searchParams.set(e2, x2);
     for (let [e2, b2] of w.entries()) x && !Object.values(KnownQueryParams).includes(e2) || S.searchParams.set(e2, b2);
     return S.pathname = `${S.pathname.replace(/\/$/, "")}/${e.replace(/^\//, "")}`, S.hash = "", S;
   }
   formatWsURL(e, b) {
-    let x = this.formatHttpURL(e, b, false);
+    let x = this.formatHttpURL({
+      path: e,
+      searchParams: b,
+      restrictToKnownQueryParams: false
+    });
     return !this.isSameOrigin && this.config.authToken && x.searchParams.set(KnownQueryParams.accessToken, this.config.authToken), asWsUrl(x.toString());
   }
   getWsURL(e) {
@@ -546,26 +552,33 @@ var RuntimeManager = class {
     return this.formatWsURL(`/lsp/${e}`);
   }
   getAiURL(e) {
-    return this.formatHttpURL(`/api/ai/${e}`);
+    return this.formatHttpURL({ path: `/api/ai/${e}` });
   }
   healthURL() {
-    return this.formatHttpURL("/health");
+    return this.formatHttpURL({ path: "/health" });
   }
-  async isHealthy() {
-    if (isWasm() || isIslands() || isStaticNotebook()) return true;
+  async fetchHealth() {
     try {
-      let e = await fetch(this.healthURL().toString());
-      if (e.redirected) {
-        Logger.debug(`Runtime redirected to ${e.url}`);
-        let x2 = new URL(e.url);
-        x2.pathname = x2.pathname.replace(/\/health$/, ""), this.config.url = x2.toString();
-      }
-      let x = e.ok;
-      return x && this.setDOMBaseUri(this.config.url), x;
+      return await fetch(this.healthURL().toString());
     } catch (e) {
-      return Logger.error(`Failed to check health: ${e instanceof Error ? e.message : "Unknown error"}`, { cause: e }), false;
+      return Logger.error(`Failed to check health: ${e instanceof Error ? e.message : "Unknown error"}`, { cause: e }), null;
     }
   }
+  async reconcileFromHealth() {
+    if (this.isServerless) return true;
+    let e = await this.fetchHealth();
+    if (!e) return false;
+    if (e.redirected) {
+      Logger.debug(`Runtime redirected to ${e.url}`);
+      let x = new URL(e.url);
+      x.pathname = x.pathname.replace(/\/health$/, ""), this.config.url = x.toString();
+    }
+    return e.ok && this.setDOMBaseUri(this.config.url), e.ok;
+  }
+  async probeHealth() {
+    var _a;
+    return this.isServerless ? true : ((_a = await this.fetchHealth()) == null ? void 0 : _a.ok) ?? false;
+  }
   setDOMBaseUri(e) {
     e = e.split("?", 1)[0], e.endsWith("/") || (e += "/");
     let b = document.querySelector("base");
@@ -574,7 +587,7 @@ var RuntimeManager = class {
   async init(e) {
     Logger.debug("Initializing runtime...");
     let x = 0;
-    for (; !await this.isHealthy(); ) {
+    for (; !await this.reconcileFromHealth(); ) {
       if (x >= 25) {
         Logger.error("Failed to connect after 25 retries"), this.initialHealthyCheck.reject(/* @__PURE__ */ Error("Failed to connect after 25 retries"));
         return;

package/dist/{vega-component-BtvQ-Kc4.js → vega-component-cSdqoAxe.js}

@@ -2,7 +2,7 @@ import { s as __toESM } from "./chunk-BNovOVIE.js";
 import { _ as Logger, c as Objects, g as cn, h as Events } from "./button-Dj4BTre0.js";
 import { t as require_react } from "./react-DA-nE2FX.js";
 import { t as require_compiler_runtime } from "./compiler-runtime-CEbnTgxf.js";
-import { c as asRemoteURL, v as CircleQuestionMark } from "./toDate-CJWlVNGD.js";
+import { c as asRemoteURL, v as CircleQuestionMark } from "./toDate-CIpC_34u.js";
 import "./react-dom-BWRJ_g_k.js";
 import { t as require_jsx_runtime } from "./jsx-runtime-COBk7ree.js";
 import "./zod-BWkcDORu.js";
@@ -11,7 +11,7 @@ import { t as Tooltip } from "./tooltip-DRaMBu06.js";
 import { i as debounce_default } from "./constants-D0gkYoE2.js";
 import { n as useTheme, w as useEvent_default } from "./useTheme-DykuNHR2.js";
 import { s as uniq } from "./arrays-CldYf7p7.js";
-import { a as isValid, i as AlertTitle, n as Alert, t as arrow } from "./formats-CpgZM9BM.js";
+import { a as isValid, i as AlertTitle, n as Alert, t as arrow } from "./formats-Dsy9kkZu.js";
 import { n as formats } from "./vega-loader.browser-3_z8GoFC.js";
 import { a as getContainerWidth, n as vegaLoadData, s as tooltipHandler } from "./loader-Dr8Qem8p.js";
 import { t as useAsyncData } from "./useAsyncData-C56Khv_R.js";

package/package.json

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

package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx

@@ -57,7 +57,7 @@ export const BackendConnectionStatus: React.FC = () => {
     }
 
     try {
-      const isHealthy = await runtime.isHealthy();
+      const isHealthy = await runtime.probeHealth();
       setConnectionStatus(isHealthy ? "healthy" : "unhealthy");
       return {
         isHealthy,

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

@@ -275,14 +275,14 @@ describe("RuntimeManager", () => {
     });
   });
 
-  describe("isHealthy", () => {
+  describe("probeHealth", () => {
     it("should return true for successful health check", async () => {
       global.fetch = vi.fn().mockResolvedValue({
         ok: true,
       });
 
       const runtime = new RuntimeManager(mockConfig);
-      const result = await runtime.isHealthy();
+      const result = await runtime.probeHealth();
 
       expect(result).toBe(true);
       expect(fetch).toHaveBeenCalledWith("https://example.com/health");
@@ -294,7 +294,7 @@ describe("RuntimeManager", () => {
       });
 
       const runtime = new RuntimeManager(mockConfig);
-      const result = await runtime.isHealthy();
+      const result = await runtime.probeHealth();
 
       expect(result).toBe(false);
     });
@@ -304,11 +304,32 @@ describe("RuntimeManager", () => {
       global.fetch = vi.fn().mockRejectedValue(error);
 
       const runtime = new RuntimeManager(mockConfig);
-      const result = await runtime.isHealthy();
+      const result = await runtime.probeHealth();
 
       expect(result).toBe(false);
     });
 
+    it("should not mutate config.url on redirect", async () => {
+      global.fetch = vi.fn().mockResolvedValue({
+        ok: true,
+        redirected: true,
+        url: "https://sandbox.example.com/health?some_value=abc123",
+      });
+
+      const runtime = new RuntimeManager(
+        {
+          ...mockConfig,
+          url: "https://backend.example.com/lazy?some_value=abc123",
+        },
+        true,
+      );
+      const result = await runtime.probeHealth();
+      expect(result).toBe(true);
+      expect(runtime.httpURL.hostname).toBe("backend.example.com");
+    });
+  });
+
+  describe("reconcileFromHealth", () => {
     it("should update config.url on redirect, stripping /health from pathname", async () => {
       global.fetch = vi.fn().mockResolvedValue({
         ok: true,
@@ -323,7 +344,7 @@ describe("RuntimeManager", () => {
         },
         true, // lazy — don't call init() in constructor
       );
-      const result = await runtime.isHealthy();
+      const result = await runtime.reconcileFromHealth();
 
       expect(result).toBe(true);
       // Should strip /health from pathname but preserve query params
@@ -342,7 +363,7 @@ describe("RuntimeManager", () => {
     it("should resolve immediately if healthy", async () => {
       const runtime = new RuntimeManager(mockConfig, true);
 
-      vi.spyOn(runtime, "isHealthy").mockResolvedValue(true);
+      vi.spyOn(runtime, "reconcileFromHealth").mockResolvedValue(true);
       runtime.init();
 
       await expect(runtime.waitForHealthy()).resolves.toBeUndefined();
@@ -351,7 +372,7 @@ describe("RuntimeManager", () => {
     it("should retry and eventually succeed", async () => {
       const runtime = new RuntimeManager(mockConfig, true);
       const healthySpy = vi
-        .spyOn(runtime, "isHealthy")
+        .spyOn(runtime, "reconcileFromHealth")
         .mockResolvedValueOnce(false)
         .mockResolvedValueOnce(false)
         .mockResolvedValueOnce(true);
@@ -364,7 +385,7 @@ describe("RuntimeManager", () => {
 
     it("should throw after max retries", async () => {
       const runtime = new RuntimeManager(mockConfig, true);
-      vi.spyOn(runtime, "isHealthy").mockResolvedValue(false);
+      vi.spyOn(runtime, "reconcileFromHealth").mockResolvedValue(false);
       runtime.init({ disableRetryDelay: true });
 
       await expect(runtime.waitForHealthy()).rejects.toThrow(
@@ -431,7 +452,7 @@ describe("RuntimeManager", () => {
       // Mock failed health check
       global.fetch = vi.fn().mockResolvedValue({ ok: false });
 
-      await runtime.isHealthy();
+      await runtime.reconcileFromHealth();
 
       baseElement = document.querySelector("base");
       expect(baseElement).toBeNull();
@@ -443,7 +464,7 @@ describe("RuntimeManager", () => {
       // Mock successful health check
       global.fetch = vi.fn().mockResolvedValue({ ok: true });
 
-      await runtime.isHealthy();
+      await runtime.reconcileFromHealth();
 
       const baseElement = document.querySelector("base");
       expect(baseElement).toBeTruthy();
@@ -461,7 +482,7 @@ describe("RuntimeManager", () => {
       // Mock successful health check
       global.fetch = vi.fn().mockResolvedValue({ ok: true });
 
-      await runtime.isHealthy();
+      await runtime.reconcileFromHealth();
 
       const baseElement = document.querySelector("base");
       expect(baseElement).toBe(existingBase); // Should be the same element
@@ -483,7 +504,7 @@ describe("RuntimeManager", () => {
       // Mock successful health check
       global.fetch = vi.fn().mockResolvedValue({ ok: true });
 
-      await runtime.isHealthy();
+      await runtime.reconcileFromHealth();
 
       const baseElement = document.querySelector("base");
       expect(baseElement).toBeTruthy();
@@ -524,11 +545,11 @@ describe("RuntimeManager", () => {
       });
 
       const wsUrl = runtime.getWsURL("test" as SessionId);
-      const httpUrl = runtime.formatHttpURL(
-        "api/test",
-        new URLSearchParams(),
-        false,
-      );
+      const httpUrl = runtime.formatHttpURL({
+        path: "api/test",
+        searchParams: new URLSearchParams(),
+        restrictToKnownQueryParams: false,
+      });
 
       // Should preserve base URL query params
       expect(wsUrl.searchParams.get("base_param")).toBe("existing");

package/src/core/runtime/runtime.ts

@@ -44,17 +44,22 @@ export class RuntimeManager {
     return this.httpURL.origin === window.location.origin;
   }
 
+  private get isServerless(): boolean {
+    return isWasm() || isIslands() || isStaticNotebook();
+  }
+
   /**
    * The base URL of the runtime.
    */
-  formatHttpURL(
-    path?: string,
-    searchParams?: URLSearchParams,
+  formatHttpURL({
+    path = "",
+    searchParams,
     restrictToKnownQueryParams = true,
-  ): URL {
-    if (!path) {
-      path = "";
-    }
+  }: {
+    path?: string;
+    searchParams?: URLSearchParams;
+    restrictToKnownQueryParams?: boolean;
+  }): URL {
     // URL may be something like "http://localhost:8000?auth=123"
     const baseUrl = this.httpURL;
     const currentParams = new URLSearchParams(window.location.search);
@@ -84,11 +89,11 @@ export class RuntimeManager {
   formatWsURL(path: string, searchParams?: URLSearchParams): URL {
     // We don't restrict to known query parameters, since mo.query_params()
     // can accept arbitrary parameters.
-    const url = this.formatHttpURL(
+    const url = this.formatHttpURL({
       path,
       searchParams,
-      /* restrictToKnownQueryParams =*/ false,
-    );
+      restrictToKnownQueryParams: false,
+    });
 
     // For cross-origin runtimes, pass the auth token as a query parameter.
     // WebSocket connections cannot send custom headers (no Authorization
@@ -168,45 +173,63 @@ export class RuntimeManager {
   }
 
   getAiURL(path: "completion" | "chat"): URL {
-    return this.formatHttpURL(`/api/ai/${path}`);
+    return this.formatHttpURL({ path: `/api/ai/${path}` });
   }
 
   /**
    * The URL of the health check endpoint.
    */
   healthURL(): URL {
-    return this.formatHttpURL("/health");
+    return this.formatHttpURL({ path: "/health" });
   }
 
-  async isHealthy(): Promise<boolean> {
-    // Always healthy if WASM, Islands, or a static notebook (no server)
-    if (isWasm() || isIslands() || isStaticNotebook()) {
-      return true;
-    }
-
+  private async fetchHealth(): Promise<Response | null> {
     try {
-      const response = await fetch(this.healthURL().toString());
-      // If there is a redirect, update the URL in the config
-      if (response.redirected) {
-        Logger.debug(`Runtime redirected to ${response.url}`);
-        // strip /health from the URL, using URL parsing to handle query params
-        const redirected = new URL(response.url);
-        redirected.pathname = redirected.pathname.replace(/\/health$/, "");
-        this.config.url = redirected.toString();
-      }
-
-      const success = response.ok;
-      if (success) {
-        this.setDOMBaseUri(this.config.url);
-      }
-      return success;
+      return await fetch(this.healthURL().toString());
     } catch (error) {
       Logger.error(
         `Failed to check health: ${error instanceof Error ? error.message : "Unknown error"}`,
         { cause: error },
       );
+      return null;
+    }
+  }
+
+  async reconcileFromHealth(): Promise<boolean> {
+    // Always healthy if WASM, Islands, or a static notebook (no server)
+    if (this.isServerless) {
+      return true;
+    }
+
+    const response = await this.fetchHealth();
+
+    if (!response) {
       return false;
     }
+
+    if (response.redirected) {
+      Logger.debug(`Runtime redirected to ${response.url}`);
+      // strip /health from the URL, using URL parsing to handle query params
+      const redirected = new URL(response.url);
+      redirected.pathname = redirected.pathname.replace(/\/health$/, "");
+      this.config.url = redirected.toString();
+    }
+
+    if (response.ok) {
+      this.setDOMBaseUri(this.config.url);
+    }
+
+    return response.ok;
+  }
+
+  async probeHealth(): Promise<boolean> {
+    // Always healthy if WASM, Islands, or a static notebook (no server)
+    if (this.isServerless) {
+      return true;
+    }
+
+    const response = await this.fetchHealth();
+    return response?.ok ?? false;
   }
 
   /**
@@ -251,7 +274,7 @@ export class RuntimeManager {
     const growthFactor = 1.2;
     const maxDelay = 2000;
 
-    while (!(await this.isHealthy())) {
+    while (!(await this.reconcileFromHealth())) {
       if (retries >= maxRetries) {
         Logger.error(`Failed to connect after ${maxRetries} retries`);
         this.initialHealthyCheck.reject(

package/src/core/websocket/__tests__/useMarimoKernelConnection.hook.test.tsx

@@ -35,7 +35,6 @@ import { useConnectionTransport } from "../useWebSocket";
 
 interface MockTransport {
   readyState: 0 | 1 | 2 | 3;
-  retryCount: number;
   reconnect: ReturnType<typeof vi.fn>;
   close: ReturnType<typeof vi.fn>;
   send: ReturnType<typeof vi.fn>;
@@ -48,7 +47,6 @@ function makeTransport(
 ): MockTransport {
   return {
     readyState,
-    retryCount: 0,
     reconnect: vi.fn(),
     close: vi.fn(),
     send: vi.fn(),
@@ -57,9 +55,12 @@ function makeTransport(
   };
 }
 
-function makeRuntimeManager(isHealthy = vi.fn().mockResolvedValue(true)) {
+function makeRuntimeManager(
+  reconcileFromHealth = vi.fn().mockResolvedValue(true),
+) {
   return {
-    isHealthy,
+    reconcileFromHealth,
+    probeHealth: vi.fn().mockResolvedValue(true),
     getWsURL: () => new URL("ws://localhost/ws"),
     waitForHealthy: vi.fn().mockResolvedValue(undefined),
     isSameOrigin: true,

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

@@ -4,50 +4,22 @@ import { afterEach, describe, expect, it, vi } from "vitest";
 import { Logger } from "@/utils/Logger";
 import { WebSocketClosedReason, WebSocketState } from "../types";
 import { classifyCloseEvent } from "../useMarimoKernelConnection";
-import { MAX_RETRIES } from "../useWebSocket";
 
-function classify(
-  reason: string | undefined,
-  retryCount = 0,
-  maxRetries = MAX_RETRIES,
-) {
-  return classifyCloseEvent({ reason }, { retryCount, maxRetries });
+function classify(reason: string | undefined) {
+  return classifyCloseEvent({ reason });
 }
 
 describe("classifyCloseEvent", () => {
   describe("transient closes (default branch)", () => {
-    it("retries when retryCount < maxRetries", () => {
-      const decision = classify(undefined, 0);
+    it("retries on empty/undefined reason", () => {
+      const decision = classify(undefined);
       expect(decision.kind).toBe("retry");
       expect(decision.status).toEqual({ state: WebSocketState.CONNECTING });
     });
 
-    it("retries on each intermediate close event during a retry storm", () => {
-      for (let n = 0; n < MAX_RETRIES; n++) {
-        const decision = classify(undefined, n);
-        expect(decision.kind).toBe("retry");
-        expect(decision.status).toEqual({ state: WebSocketState.CONNECTING });
-      }
-    });
-
-    it("transitions to CLOSED when retryCount reaches maxRetries", () => {
-      const decision = classify(undefined, MAX_RETRIES);
-      expect(decision.kind).toBe("gave-up");
-      expect(decision.status).toEqual({
-        state: WebSocketState.CLOSED,
-        code: WebSocketClosedReason.KERNEL_DISCONNECTED,
-        reason: "kernel not found",
-      });
-    });
-
-    it("transitions to CLOSED when retryCount exceeds maxRetries", () => {
-      const decision = classify(undefined, MAX_RETRIES + 5);
-      expect(decision.kind).toBe("gave-up");
-    });
-
     it("treats unknown reason strings as transient and logs a warning", () => {
       const logger = vi.spyOn(Logger, "warn").mockImplementation(() => {});
-      const decision = classify("something-else", 3);
+      const decision = classify("something-else");
       expect(decision.kind).toBe("retry");
       expect(logger).toHaveBeenCalled();
       logger.mockRestore();
@@ -60,7 +32,7 @@ describe("classifyCloseEvent", () => {
 
   describe("terminal closes (server-initiated)", () => {
     it("MARIMO_ALREADY_CONNECTED → terminal + closeTransport, with takeover", () => {
-      const decision = classify("MARIMO_ALREADY_CONNECTED", 0);
+      const decision = classify("MARIMO_ALREADY_CONNECTED");
       expect(decision.kind).toBe("terminal");
       expect(decision.status).toMatchObject({
         state: WebSocketState.CLOSED,
@@ -79,7 +51,7 @@ describe("classifyCloseEvent", () => {
       "MARIMO_NO_SESSION",
       "MARIMO_SHUTDOWN",
     ])("%s → terminal with KERNEL_DISCONNECTED, closes transport", (reason) => {
-      const decision = classify(reason, 0);
+      const decision = classify(reason);
       expect(decision.kind).toBe("terminal");
       expect(decision.status).toMatchObject({
         state: WebSocketState.CLOSED,
@@ -91,7 +63,7 @@ describe("classifyCloseEvent", () => {
     });
 
     it("MARIMO_MALFORMED_QUERY → terminal but does NOT close transport", () => {
-      const decision = classify("MARIMO_MALFORMED_QUERY", 0);
+      const decision = classify("MARIMO_MALFORMED_QUERY");
       expect(decision.kind).toBe("terminal");
       expect(decision.status).toMatchObject({
         state: WebSocketState.CLOSED,
@@ -103,7 +75,7 @@ describe("classifyCloseEvent", () => {
     });
 
     it("MARIMO_KERNEL_STARTUP_ERROR → terminal + closeTransport", () => {
-      const decision = classify("MARIMO_KERNEL_STARTUP_ERROR", 0);
+      const decision = classify("MARIMO_KERNEL_STARTUP_ERROR");
       expect(decision.kind).toBe("terminal");
       expect(decision.status).toMatchObject({
         state: WebSocketState.CLOSED,
@@ -113,25 +85,17 @@ describe("classifyCloseEvent", () => {
         expect(decision.closeTransport).toBe(true);
       }
     });
-
-    it("terminal closes ignore retryCount entirely", () => {
-      const decision = classify("MARIMO_SHUTDOWN", 99);
-      expect(decision.kind).toBe("terminal");
-    });
   });
 
-  describe("retry budget exhaustion", () => {
-    it("yields retry on attempts 1..maxRetries-1 and gave-up on the final close", () => {
-      const states: string[] = [];
-      for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
-        states.push(classify(undefined, attempt - 1).kind);
-      }
-      states.push(classify(undefined, MAX_RETRIES).kind);
-
-      expect(states).toEqual([
-        ...Array.from({ length: MAX_RETRIES }, () => "retry"),
-        "gave-up",
-      ]);
+  describe("transport exhaustion", () => {
+    it("MARIMO_TRANSPORT_EXHAUSTED → gave-up with KERNEL_DISCONNECTED", () => {
+      const decision = classify("MARIMO_TRANSPORT_EXHAUSTED");
+      expect(decision.kind).toBe("gave-up");
+      expect(decision.status).toEqual({
+        state: WebSocketState.CLOSED,
+        code: WebSocketClosedReason.KERNEL_DISCONNECTED,
+        reason: "kernel not found",
+      });
     });
   });
 });

package/src/core/websocket/transports/__tests__/ws.test.ts

@@ -0,0 +1,125 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { MAX_RETRIES, WsTransport, TRANSPORT_EXHAUSTED_REASON } from "../ws";
+
+let innerListeners: Record<string, ((e: unknown) => void)[]>;
+
+vi.mock("partysocket/ws", () => {
+  class FakeReconnectingWebSocket {
+    retryCount = 0;
+    readyState = WebSocket.CONNECTING;
+    constructor() {
+      innerListeners = { open: [], close: [], message: [], error: [] };
+    }
+    addEventListener(event: string, cb: (e: unknown) => void) {
+      innerListeners[event].push(cb);
+    }
+    removeEventListener(event: string, cb: (e: unknown) => void) {
+      innerListeners[event] = innerListeners[event].filter((c) => c !== cb);
+    }
+    reconnect() {}
+    close() {}
+    send() {}
+  }
+  return { default: FakeReconnectingWebSocket };
+});
+
+interface FakeReconnectingWebSocket {
+  retryCount: number;
+  readyState: number;
+}
+
+function dispatchClose(reason = "") {
+  const evt = new CloseEvent("close", { reason, code: 1006 });
+  for (const cb of innerListeners.close) {
+    cb(evt);
+  }
+}
+
+describe("WsTransport", () => {
+  let transport: WsTransport;
+  let inner: FakeReconnectingWebSocket;
+
+  beforeEach(() => {
+    transport = new WsTransport(() => "ws://example.invalid/ws");
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    inner = (transport as any).inner;
+  });
+
+  describe("close event reason", () => {
+    it("forwards the original reason before retry budget is exhausted", () => {
+      const seen: CloseEvent[] = [];
+      transport.addEventListener("close", (e) => seen.push(e));
+
+      inner.retryCount = MAX_RETRIES - 1;
+      dispatchClose("");
+
+      expect(seen).toHaveLength(1);
+      expect(seen[0].reason).toBe("");
+    });
+
+    it("rewrites reason to MARIMO_TRANSPORT_EXHAUSTED on exhaustion", () => {
+      const seen: CloseEvent[] = [];
+      transport.addEventListener("close", (e) => seen.push(e));
+
+      inner.retryCount = MAX_RETRIES;
+      dispatchClose("");
+
+      expect(seen).toHaveLength(1);
+      expect(seen[0].reason).toBe(TRANSPORT_EXHAUSTED_REASON);
+    });
+
+    it("does not rewrite a server-sent reason on exhaustion", () => {
+      // If partysocket happens to deliver a real MARIMO_* reason at the same
+      // moment the retry budget exhausts, the exhausted-state rewrite still
+      // wins because the wrapper keys off retryCount, not the original reason.
+      // Document the behavior so a future change is deliberate.
+      const seen: CloseEvent[] = [];
+      transport.addEventListener("close", (e) => seen.push(e));
+
+      inner.retryCount = MAX_RETRIES;
+      dispatchClose("MARIMO_SHUTDOWN");
+
+      expect(seen[0].reason).toBe(TRANSPORT_EXHAUSTED_REASON);
+    });
+  });
+
+  describe("addEventListener dedupe", () => {
+    it("does not double-fire when the same close listener is added twice", () => {
+      const cb = vi.fn();
+      transport.addEventListener("close", cb);
+      transport.addEventListener("close", cb);
+
+      dispatchClose("");
+
+      expect(cb).toHaveBeenCalledTimes(1);
+    });
+
+    it("a single removeEventListener fully unregisters a duplicated add", () => {
+      const cb = vi.fn();
+      transport.addEventListener("close", cb);
+      transport.addEventListener("close", cb);
+      transport.removeEventListener("close", cb);
+
+      dispatchClose("");
+
+      expect(cb).not.toHaveBeenCalled();
+      // Inner socket has no orphaned wrappers left.
+      expect(innerListeners.close).toHaveLength(0);
+    });
+  });
+
+  describe("removeEventListener", () => {
+    it("unregisters the right wrapper for close listeners", () => {
+      const cb = vi.fn();
+      transport.addEventListener("close", cb);
+      transport.removeEventListener("close", cb);
+
+      inner.retryCount = MAX_RETRIES;
+      dispatchClose("");
+
+      expect(cb).not.toHaveBeenCalled();
+    });
+  });
+});

package/src/core/websocket/transports/basic.ts

@@ -43,9 +43,7 @@ export class BasicTransport implements IConnectionTransport {
     return WebSocket.OPEN;
   }
 
-  readonly retryCount = 0;
-
-  reconnect(code?: number | undefined, reason?: string | undefined): void {
+  reconnect(_code?: number | undefined, _reason?: string | undefined): void {
     this.close();
     this.connect();
     return;

package/src/core/websocket/transports/transport.ts

@@ -25,7 +25,6 @@ export interface IConnectionTransport {
     callback: ConnectionTransportCallback<T>,
   ): void;
   readyState: WebSocket["readyState"];
-  readonly retryCount: number;
 }
 
 export class ConnectionSubscriptions {

package/src/core/websocket/transports/ws.ts

@@ -0,0 +1,94 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import ReconnectingWebSocket from "partysocket/ws";
+import type {
+  ConnectionEvent,
+  ConnectionTransportCallback,
+  IConnectionTransport,
+} from "./transport";
+
+// Per-`reconnect()` retry budget. After exhaustion, partysocket stops silently;
+// the wrapper rewrites the close-event reason to surface the give-up.
+export const MAX_RETRIES = 10;
+
+export const TRANSPORT_EXHAUSTED_REASON = "MARIMO_TRANSPORT_EXHAUSTED";
+
+export class WsTransport implements IConnectionTransport {
+  private inner: ReconnectingWebSocket;
+  private closeWrappers = new WeakMap<
+    ConnectionTransportCallback<"close">,
+    ConnectionTransportCallback<"close">
+  >();
+
+  constructor(urlProvider: () => string) {
+    this.inner = new ReconnectingWebSocket(urlProvider, undefined, {
+      maxRetries: MAX_RETRIES,
+      debug: false,
+      startClosed: true,
+      // long timeout — the server can become slow when many notebooks are open.
+      connectionTimeout: 10_000,
+    });
+  }
+
+  get readyState(): WebSocket["readyState"] {
+    return this.inner.readyState as WebSocket["readyState"];
+  }
+
+  reconnect(code?: number, reason?: string): void {
+    this.inner.reconnect(code, reason);
+  }
+
+  close(): void {
+    this.inner.close();
+  }
+
+  send(data: string | ArrayBuffer | Blob | ArrayBufferView): void {
+    this.inner.send(data);
+  }
+
+  addEventListener<T extends ConnectionEvent>(
+    event: T,
+    callback: ConnectionTransportCallback<T>,
+  ): void {
+    if (event === "close") {
+      const userCb = callback as ConnectionTransportCallback<"close">;
+      // Match native EventTarget dedupe: a second addEventListener with the
+      // same listener is a no-op. Without this, repeated adds leak wrappers
+      // on the inner socket and double-fire on close.
+      if (this.closeWrappers.has(userCb)) return;
+      const wrapper: ConnectionTransportCallback<"close"> = (e) => {
+        if (this.inner.retryCount >= MAX_RETRIES) {
+          userCb(
+            new CloseEvent("close", {
+              code: e.code,
+              reason: TRANSPORT_EXHAUSTED_REASON,
+              wasClean: e.wasClean,
+            }),
+          );
+        } else {
+          userCb(e);
+        }
+      };
+      this.closeWrappers.set(userCb, wrapper);
+      this.inner.addEventListener("close", wrapper);
+      return;
+    }
+    this.inner.addEventListener(event, callback as never);
+  }
+
+  removeEventListener<T extends ConnectionEvent>(
+    event: T,
+    callback: ConnectionTransportCallback<T>,
+  ): void {
+    if (event === "close") {
+      const userCb = callback as ConnectionTransportCallback<"close">;
+      const wrapper = this.closeWrappers.get(userCb);
+      if (wrapper) {
+        this.closeWrappers.delete(userCb);
+        this.inner.removeEventListener("close", wrapper);
+      }
+      return;
+    }
+    this.inner.removeEventListener(event, callback as never);
+  }
+}

package/src/core/websocket/useMarimoKernelConnection.tsx

@@ -11,10 +11,8 @@ import type {
   NotificationMessageData,
   NotificationPayload,
 } from "@/core/kernel/messages";
-import {
-  MAX_RETRIES,
-  useConnectionTransport,
-} from "@/core/websocket/useWebSocket";
+import { TRANSPORT_EXHAUSTED_REASON } from "@/core/websocket/transports/ws";
+import { useConnectionTransport } from "@/core/websocket/useWebSocket";
 import { renderHTML } from "@/plugins/core/RenderHTML";
 import {
   handleWidgetMessage,
@@ -80,16 +78,27 @@ import {
 
 const SUPPORTS_LAZY_KERNELS = true;
 
+// All MARIMO_* reasons except TRANSPORT_EXHAUSTED are emitted by the backend
+// (marimo/_server/api/endpoints/ws_endpoint.py and ws/*.py). Keep in sync with
+// the backend literals.
+export type CloseReason =
+  | "MARIMO_ALREADY_CONNECTED"
+  | "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;
+
 export type CloseDecision =
   | { kind: "terminal"; status: ConnectionStatus; closeTransport: boolean }
   | { kind: "gave-up"; status: ConnectionStatus }
   | { kind: "retry"; status: ConnectionStatus };
 
-export function classifyCloseEvent(
-  event: { reason?: string },
-  context: { retryCount: number; maxRetries: number },
-): CloseDecision {
-  switch (event.reason) {
+export function classifyCloseEvent(event: { reason?: string }): CloseDecision {
+  switch (event.reason as CloseReason | undefined) {
     case "MARIMO_ALREADY_CONNECTED":
       return {
         kind: "terminal",
@@ -101,6 +110,15 @@ export function classifyCloseEvent(
         },
         closeTransport: true,
       };
+    case TRANSPORT_EXHAUSTED_REASON:
+      return {
+        kind: "gave-up",
+        status: {
+          state: WebSocketState.CLOSED,
+          code: WebSocketClosedReason.KERNEL_DISCONNECTED,
+          reason: "kernel not found",
+        },
+      };
     case "MARIMO_WRONG_KERNEL_ID":
     case "MARIMO_NO_FILE_KEY":
     case "MARIMO_NO_SESSION_ID":
@@ -144,18 +162,7 @@ export function classifyCloseEvent(
         logNever(event.reason as never);
       }
   }
-  // partysocket stops retrying silently once `maxRetries` is hit; surface
-  // CLOSED so callers can detect the give-up.
-  if (context.retryCount >= context.maxRetries) {
-    return {
-      kind: "gave-up",
-      status: {
-        state: WebSocketState.CLOSED,
-        code: WebSocketClosedReason.KERNEL_DISCONNECTED,
-        reason: "kernel not found",
-      },
-    };
-  }
+
   return {
     kind: "retry",
     status: { state: WebSocketState.CONNECTING },
@@ -440,7 +447,7 @@ export function useMarimoKernelConnection(opts: {
     }
     shouldTryReconnecting.current = true;
     setConnection({ state: WebSocketState.CONNECTING });
-    const healthy = await runtimeManager.isHealthy();
+    const healthy = await runtimeManager.reconcileFromHealth();
     if (!healthy) {
       shouldTryReconnecting.current = false;
       setConnection({
@@ -512,10 +519,7 @@ export function useMarimoKernelConnection(opts: {
      */
     onClose: (e) => {
       Logger.warn("WebSocket closed", e.code, e.reason);
-      const decision = classifyCloseEvent(e, {
-        retryCount: ws.retryCount,
-        maxRetries: MAX_RETRIES,
-      });
+      const decision = classifyCloseEvent(e);
       setConnection(decision.status);
       if (decision.kind === "terminal" && decision.closeTransport) {
         ws.close(); // close to prevent reconnecting

package/src/core/websocket/useWebSocket.tsx

@@ -1,11 +1,11 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import ReconnectingWebSocket from "partysocket/ws";
 import { useEffect, useState } from "react";
 import { Logger } from "@/utils/Logger";
 import { createPyodideConnection } from "../wasm/bridge";
 import { isWasm } from "../wasm/utils";
 import { BasicTransport } from "./transports/basic";
+import { WsTransport } from "./transports/ws";
 import type { IConnectionTransport } from "./transports/transport";
 
 interface UseConnectionTransportOptions {
@@ -18,10 +18,6 @@ interface UseConnectionTransportOptions {
   onError: (event: WebSocketEventMap["error"]) => void;
 }
 
-// Per-`reconnect()` retry budget for partysocket. After exhaustion, partysocket
-// stops silently; treat `retryCount >= MAX_RETRIES` as the give-up signal.
-export const MAX_RETRIES = 10;
-
 function createConnectionTransport(
   options: Pick<UseConnectionTransportOptions, "url" | "static">,
 ): IConnectionTransport {
@@ -31,19 +27,8 @@ function createConnectionTransport(
   if (isWasm()) {
     return createPyodideConnection();
   }
-  // Create a connection transport using the ReconnectingWebSocket from partysocket
-  // This handles reconnecting when the connection is lost.
-  const urlProvider = options.url; // We don't call the URL provider now since it may change (i.e. if the runtime redirects)
-  // Cast needed: ReconnectingWebSocket types readyState as `number`
-  // but IConnectionTransport expects `0 | 1 | 2 | 3`
-  return new ReconnectingWebSocket(urlProvider, undefined, {
-    maxRetries: MAX_RETRIES,
-    debug: false,
-    startClosed: true,
-    // long timeout -- the server can become slow when many notebooks
-    // are open.
-    connectionTimeout: 10_000,
-  }) as unknown as IConnectionTransport;
+  // urlProvider is passed lazily; it may change after a runtime redirect.
+  return new WsTransport(options.url);
 }
 
 /**