@marimo-team/frontend 0.25.3-dev10 → 0.25.3-dev100

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (452) hide show
  1. package/dist/assets/CellStatus-fK4n2NPc.js +1 -0
  2. package/dist/assets/{ConnectedDataExplorerComponent-t8fH_TNc.js → ConnectedDataExplorerComponent-Dft8R6l5.js} +1 -1
  3. package/dist/assets/{ErrorBoundary-B4n9EnHB.js → ErrorBoundary-CAEGdoFW.js} +2 -2
  4. package/dist/assets/JsonOutput-CyAKmZDD.js +11 -0
  5. package/dist/assets/{LazyAnyLanguageCodeMirror-CKcCbG2G.js → LazyAnyLanguageCodeMirror-BPYE76jR.js} +2 -2
  6. package/dist/assets/MarimoErrorOutput-CgZvP0p0.js +10 -0
  7. package/dist/assets/{RenderHTML-9bMgbYmo.js → RenderHTML-DQvxDDNS.js} +1 -1
  8. package/dist/assets/RunButton-BBDnDL6G.js +1 -0
  9. package/dist/assets/__vite-browser-external-8vDMR0Kk.js +1 -0
  10. package/dist/assets/__vite-browser-external-mKcnhO20.js +1 -0
  11. package/dist/assets/{add-cell-with-ai-f_MnLCwI.js → add-cell-with-ai-Clg5FELx.js} +7 -7
  12. package/dist/assets/{add-connection-dialog-CoE_PhuD.js → add-connection-dialog-BqdWmD-7.js} +2 -2
  13. package/dist/assets/{agent-panel-BWZz24jK.js → agent-panel-5RxG5Y82.js} +3 -3
  14. package/dist/assets/ai-model-dropdown-Bvl1ktQq.js +13 -0
  15. package/dist/assets/{alert-dialog-BaPGzeOK.js → alert-dialog-C-sWmKva.js} +1 -1
  16. package/dist/assets/any-language-editor-DUMe1UCk.js +1 -0
  17. package/dist/assets/{app-config-button-BhU0ulYs.js → app-config-button-0QTeQ2Mv.js} +1 -1
  18. package/dist/assets/arrays-0R2reF2y.js +1 -0
  19. package/dist/assets/{blob-iOBwGaP9.js → blob-5VcB8K_t.js} +1 -1
  20. package/dist/assets/{button-COI-NGrx.js → button-D8gVKwe-.js} +1 -1
  21. package/dist/assets/{cache-panel-DTY1dqvi.js → cache-panel-BV-c7ed0.js} +1 -1
  22. package/dist/assets/{capabilities-BJvf-MD4.js → capabilities-Pq1YNPlK.js} +1 -1
  23. package/dist/assets/cell-editor-Byw5y-oI.js +42 -0
  24. package/dist/assets/cell-link-C5yYyzuF.js +1 -0
  25. package/dist/assets/{cells-rA_jsZCj.js → cells-D_11MjVB.js} +60 -60
  26. package/dist/assets/{chat-display-BmEPmhDP.js → chat-display-BCR_ySiV.js} +1 -1
  27. package/dist/assets/chat-panel-RPRvBsaU.js +4 -0
  28. package/dist/assets/chat-ui-CGviVaRA.js +4 -0
  29. package/dist/assets/{circle-DXf-VWZG.js → circle-B6D_6D8w.js} +1 -1
  30. package/dist/assets/code-visibility-DYDagVy7.js +1 -0
  31. package/dist/assets/coerce-DWjFeXOY.js +1 -0
  32. package/dist/assets/column-preview-xe533-Cn.js +1 -0
  33. package/dist/assets/command-palette-BB9bWdWp.js +1 -0
  34. package/dist/assets/{common-lisZ0l6a.js → common-CaeItbcI.js} +1 -1
  35. package/dist/assets/components-CBE8cOLK.js +1 -0
  36. package/dist/assets/{components-CP_UoFtT.js → components-DCo7mAWO.js} +1 -1
  37. package/dist/assets/{components-Dvd1GnD6.js → components-G6frbw0c.js} +1 -1
  38. package/dist/assets/config-CD1IuZFJ.js +1 -0
  39. package/dist/assets/{config-DJLAC21v.js → config-DMz7oN0r.js} +1 -1
  40. package/dist/assets/config-DpY86_Yq.js +1 -0
  41. package/dist/assets/{copy-D8O_CYCG.js → copy-KlhazlhB.js} +1 -1
  42. package/dist/assets/{copy-icon-BIiRXh_3.js → copy-icon-8-a5vovc.js} +1 -1
  43. package/dist/assets/{createReducer-DgRODTW6.js → createReducer-CaI_Oa0v.js} +1 -1
  44. package/dist/assets/{datasource-Dr1xNKUn.js → datasource-deuXnlU-.js} +1 -1
  45. package/dist/assets/{dates-CfmdTL6B.js → dates-CQMwFko2.js} +1 -1
  46. package/dist/assets/{dependency-graph-panel-B0uP-_id.js → dependency-graph-panel-CIykLnJe.js} +2 -2
  47. package/dist/assets/{dist-iaThVDkA.js → dist-jlhMTKo_.js} +1 -1
  48. package/dist/assets/{documentation-panel-DVl_ebWy.js → documentation-panel-2PQ8kQLa.js} +1 -1
  49. package/dist/assets/{download-Bx55LMki.js → download-dK52bLxZ.js} +1 -1
  50. package/dist/assets/{dropdown-menu-C1rnl-0o.js → dropdown-menu-DS6umAnH.js} +1 -1
  51. package/dist/assets/edit-page-DrElzw10.js +9 -0
  52. package/dist/assets/{error-banner-Cd69vvyX.js → error-banner-B5kzVf_0.js} +1 -1
  53. package/dist/assets/{error-entries-DOWpdGvI.js → error-entries-3GVk2KjS.js} +1 -1
  54. package/dist/assets/{error-panel-DpbNKSLs.js → error-panel-CzDRGPeC.js} +1 -1
  55. package/dist/assets/{errors-DCrsa1nH.js → errors-CNsgAw6m.js} +1 -1
  56. package/dist/assets/{es-Cqst59WX.js → es-N9SHpmG7.js} +1 -1
  57. package/dist/assets/{feedback-button-B99ThNJW.js → feedback-button-C5NgWpFg.js} +2 -2
  58. package/dist/assets/feedback-button-CONwLFpw.js +1 -0
  59. package/dist/assets/file-explorer-panel-B50xDnTZ.js +1 -0
  60. package/dist/assets/{file-name-input-ChnBsDE_.js → file-name-input-DqwMSgGU.js} +1 -1
  61. package/dist/assets/{floating-outline-MYqQW4Vp.js → floating-outline-ChmAuvxz.js} +1 -1
  62. package/dist/assets/{formats-yZ06h1Pl.js → formats-zylfwejq.js} +1 -1
  63. package/dist/assets/{formatting-BqrP-n49.js → formatting-uom1es4b.js} +1 -1
  64. package/dist/assets/{fullscreen-B9tHcfpV.js → fullscreen-BWTs8pYZ.js} +1 -1
  65. package/dist/assets/{gallery-page-DSL0J4md.js → gallery-page-CiL8aMS1.js} +1 -1
  66. package/dist/assets/{ganttDiagram-EL5Y4UJY-2n7Ig5Jm.js → ganttDiagram-EL5Y4UJY-CYqkCfFG.js} +1 -1
  67. package/dist/assets/{glide-data-editor-B1MWlKeM.js → glide-data-editor-CqRclBkp.js} +2 -2
  68. package/dist/assets/{glide-utils-CIrAxdjv.js → glide-utils-HFqhvWHI.js} +2 -2
  69. package/dist/assets/home-page-BgZVl4dP.js +1 -0
  70. package/dist/assets/{hooks-BIogTGCr.js → hooks-DM5n7Osw.js} +1 -1
  71. package/dist/assets/{html-to-image-DK9Zl1lU.js → html-to-image-D_0rd3_B.js} +1 -1
  72. package/dist/assets/index-BnhSnwvB.css +2 -0
  73. package/dist/assets/index-EG2n8Oht.js +30 -0
  74. package/dist/assets/{input-BoBxR50a.js → input-DmHMEZGh.js} +1 -1
  75. package/dist/assets/kiosk-mode-DFS6HopF.js +1 -0
  76. package/dist/assets/layout-BpP8nTjb.js +9 -0
  77. package/dist/assets/linear-Bma8J3iM.js +1 -0
  78. package/dist/assets/links-DX2w97N9.js +1 -0
  79. package/dist/assets/{logs-panel-y0XnfulJ.js → logs-panel-B5eVAZtH.js} +1 -1
  80. package/dist/assets/{markdown-renderer-Q5AhbRU3.js → markdown-renderer-BWUzqkJM.js} +1 -1
  81. package/dist/assets/{menu-items-BrwnTgyK.js → menu-items-BgUyNUpu.js} +1 -1
  82. package/dist/assets/{mermaid-a3xYymTy.js → mermaid-BKXlWk4F.js} +2 -2
  83. package/dist/assets/{name-cell-input-BNOnF6-0.js → name-cell-input-DXW68gRW.js} +1 -1
  84. package/dist/assets/numbers-CsZVw3Dc.js +1 -0
  85. package/dist/assets/{objects-snlMn4yk.js → objects-rLPlQ3X_.js} +1 -1
  86. package/dist/assets/{once-Haz_uxEL.js → once-6KNTRW0u.js} +1 -1
  87. package/dist/assets/{outline-panel-Be7stWwv.js → outline-panel-DPxJhJ_A.js} +1 -1
  88. package/dist/assets/{package-data-dv4FBHxR.js → package-data-CKPmBDtg.js} +1 -1
  89. package/dist/assets/packages-panel-gBE2FWQr.js +1 -0
  90. package/dist/assets/{panel-accordion-state-GeMEdCPw.js → panel-accordion-state-CCF5Gxnc.js} +1 -1
  91. package/dist/assets/panels-Dr6qMCvU.js +105 -0
  92. package/dist/assets/precisionRound-BxlUs8G4.js +1 -0
  93. package/dist/assets/process-output-Cw9IDcKQ.js +1 -0
  94. package/dist/assets/{quadrantDiagram-AXDQQJYC-yBhCDI7E.js → quadrantDiagram-AXDQQJYC-D9KH5Zge.js} +1 -1
  95. package/dist/assets/react-vega-BTNeboR2.js +189 -0
  96. package/dist/assets/{readonly-code-display-YULOXf4h.js → readonly-code-display-DYcZQ28T.js} +1 -1
  97. package/dist/assets/{readonly-python-code-CadgXgRu.js → readonly-python-code-RFuQP50v.js} +1 -1
  98. package/dist/assets/{renderShortcut-BkVJQ_Zh.js → renderShortcut-rUwGqKgC.js} +1 -1
  99. package/dist/assets/{requests-lt0JK8Jt.js → requests-BIFR-h4M.js} +1 -1
  100. package/dist/assets/requests-DuLJN9Tf.js +1 -0
  101. package/dist/assets/{reveal-component-BVAR7ayZ.js → reveal-component-yRrDmCQz.js} +37 -37
  102. package/dist/assets/run-page-B69M4-zE.js +1 -0
  103. package/dist/assets/{runs-BLp5jezv.js → runs-C49hleBy.js} +1 -1
  104. package/dist/assets/{runtime-details-state-DuChzsA6.js → runtime-details-state-pobtu6TO.js} +1 -1
  105. package/dist/assets/{save-worker-BbUSwZRG.js → save-worker-pCkSHhuz.js} +1 -1
  106. package/dist/assets/{schemas-Dv42wLFq.js → schemas-ChRqMOTL.js} +1 -1
  107. package/dist/assets/scratchpad-panel-H44_pvb_.js +1 -0
  108. package/dist/assets/{scroll-to-bottom-button-1FyowpWV.js → scroll-to-bottom-button-CIkON2WJ.js} +1 -1
  109. package/dist/assets/{secrets-panel-Bctbtkrp.js → secrets-panel-bcfEKxRx.js} +1 -1
  110. package/dist/assets/{select-DTBn5eTb.js → select-3CVGlYrC.js} +1 -1
  111. package/dist/assets/{session-DqoWmb7X.js → session-CMLIzZKw.js} +1 -1
  112. package/dist/assets/session-panel-Bnd1iXkO.js +1 -0
  113. package/dist/assets/{snippets-panel-CQ1dn3lA.js → snippets-panel-DKApxKx9.js} +1 -1
  114. package/dist/assets/{spec-BfSBcO5B.js → spec-uCpVvaAR.js} +1 -1
  115. package/dist/assets/{state-uwRvJLb8.js → state-3mcyTS9q.js} +1 -1
  116. package/dist/assets/state-B_MCKzBL.js +1 -0
  117. package/dist/assets/state-Bl9BiY_s.js +3 -0
  118. package/dist/assets/{state-C3oDoyFe.js → state-C5WAeuTu.js} +1 -1
  119. package/dist/assets/{state-DxQnC_1I.js → state-Cv2JW3l9.js} +1 -1
  120. package/dist/assets/{state-Bfh_2SLl.js → state-Dv059B3A.js} +1 -1
  121. package/dist/assets/{strings-B2vv-ENv.js → strings-C25fhMR_.js} +1 -1
  122. package/dist/assets/{swiper-component-a07domrZ.js → swiper-component-CRGRy5_O.js} +1 -1
  123. package/dist/assets/{switch-CTaiAjrz.js → switch-DYjBfpNJ.js} +1 -1
  124. package/dist/assets/{tabs-BuQlZFbt.js → tabs-CY8E4r9U.js} +1 -1
  125. package/dist/assets/{terminal-Ck_vK-34.js → terminal-Ca-kubKb.js} +1 -1
  126. package/dist/assets/{time-MgToa8VM.js → time-DswSXBDk.js} +1 -1
  127. package/dist/assets/{toast-BGzLGnvw.js → toast-Gzxi-le_.js} +1 -1
  128. package/dist/assets/tooltip-Brya4Ghn.js +61 -0
  129. package/dist/assets/{tooltip-C9TuY18o.js → tooltip-CkGm8X7m.js} +1 -1
  130. package/dist/assets/tracing-Clwy4SlW.js +1 -0
  131. package/dist/assets/tracing-panel-BwjLG_aC.js +2 -0
  132. package/dist/assets/{tree-actions-DAd80MbP.js → tree-actions-BZNTQ12N.js} +1 -1
  133. package/dist/assets/{use-restore-focus-BxEGwIbk.js → use-restore-focus-DdS7Tl0F.js} +1 -1
  134. package/dist/assets/{useAddCell-DGstJK0S.js → useAddCell-Cjuqk2YR.js} +1 -1
  135. package/dist/assets/useAsyncData-qvoPYHtx.js +1 -0
  136. package/dist/assets/useBoolean-Dd2nWXFJ.js +1 -0
  137. package/dist/assets/useCellActionButton-MoWLvuGe.js +1 -0
  138. package/dist/assets/useDebounce-DwTQ_1Pu.js +1 -0
  139. package/dist/assets/{useDeepCompareMemoize-DFIVN9dn.js → useDeepCompareMemoize-B2DVK2De.js} +1 -1
  140. package/dist/assets/useDelayElapsed-BlsDT4TV.js +1 -0
  141. package/dist/assets/{useDeleteCell-DFBGlSQk.js → useDeleteCell-BC-MW7d1.js} +1 -1
  142. package/dist/assets/{useDependencyPanelTab-DydxsE5d.js → useDependencyPanelTab-CZU7i-B3.js} +1 -1
  143. package/dist/assets/{useEvent-Dfcbgjhg.js → useEvent-BDjgaBtz.js} +1 -1
  144. package/dist/assets/{useEventListener-D9cftcHj.js → useEventListener-CjDeLqFC.js} +1 -1
  145. package/dist/assets/{useIframeCapabilities-CwTkczVN.js → useIframeCapabilities-Czr9e4uG.js} +1 -1
  146. package/dist/assets/{useInstallPackage-iX1eufdl.js → useInstallPackage-wcDq9hYT.js} +1 -1
  147. package/dist/assets/{useInterval--x3s1VpI.js → useInterval-DJdQmXnq.js} +1 -1
  148. package/dist/assets/useNotebookActions-wFxVdRXQ.js +1 -0
  149. package/dist/assets/{useRunCells-BFjFX3hz.js → useRunCells-ByaWxoIt.js} +1 -1
  150. package/dist/assets/{useSplitCell-Tqv5pPtL.js → useSplitCell-YJIystwv.js} +1 -1
  151. package/dist/assets/useStateForKey-CGMJQT8x.js +1 -0
  152. package/dist/assets/{useTheme-COhSHkCj.js → useTheme-BAA27lh7.js} +1 -1
  153. package/dist/assets/{utils-CaQYvazl.js → utils-DfElfBpz.js} +1 -1
  154. package/dist/assets/vega-component-BeyxAJad.js +1 -0
  155. package/dist/assets/{vega-embed-container-DJPNkHPb.js → vega-embed-container-DtArSJz1.js} +1 -1
  156. package/dist/assets/vega-loader.browser-Bu25TKax.js +6 -0
  157. package/dist/assets/{worker-DTQ7oH2A.js → worker-bSCjCvwz.js} +1 -1
  158. package/dist/assets/{write-secret-modal-DzKAInL2.js → write-secret-modal-BEx-EUlh.js} +1 -1
  159. package/dist/assets/{xychartDiagram-S5SC5T6Z-KbGzG6-u.js → xychartDiagram-S5SC5T6Z-CeRnBS5e.js} +1 -1
  160. package/dist/index.html +80 -78
  161. package/package.json +7 -7
  162. package/src/__tests__/host-imports.test.ts +1 -0
  163. package/src/components/app-config/ai-config.tsx +7 -2
  164. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  165. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  166. package/src/components/chat/acp/blocks.tsx +2 -2
  167. package/src/components/chat/chat-components.tsx +2 -0
  168. package/src/components/chat/chat-panel.tsx +9 -8
  169. package/src/components/chat/chat-utils.ts +4 -8
  170. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  171. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  172. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  173. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  174. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  175. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  176. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  177. package/src/components/data-table/cell-selection/feature.ts +2 -2
  178. package/src/components/data-table/cell-selection/types.ts +2 -2
  179. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  180. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  181. package/src/components/data-table/charts/charts.tsx +35 -25
  182. package/src/components/data-table/charts/storage.ts +16 -5
  183. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  184. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  185. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  186. package/src/components/data-table/columns.tsx +45 -46
  187. package/src/components/data-table/context-menu.tsx +36 -20
  188. package/src/components/data-table/data-table.tsx +7 -1
  189. package/src/components/data-table/date-popover.tsx +6 -4
  190. package/src/components/data-table/export-actions.tsx +76 -48
  191. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +2 -2
  192. package/src/components/data-table/search-highlight.tsx +79 -0
  193. package/src/components/data-table/types.ts +1 -0
  194. package/src/components/data-table/url-detector.tsx +49 -9
  195. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  196. package/src/components/datasources/__tests__/column-preview.test.tsx +63 -2
  197. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  198. package/src/components/datasources/column-preview.tsx +8 -2
  199. package/src/components/datasources/datasources.tsx +47 -35
  200. package/src/components/debugger/debugger-code.tsx +5 -0
  201. package/src/components/dependency-graph/panels.tsx +2 -1
  202. package/src/components/editor/SortableCell.tsx +1 -3
  203. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  204. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +24 -0
  205. package/src/components/editor/actions/pair-with-agent-modal.tsx +18 -15
  206. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  207. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  208. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  209. package/src/components/editor/cell/CreateCellButton.tsx +8 -10
  210. package/src/components/editor/cell/DeleteButton.tsx +1 -0
  211. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  212. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  213. package/src/components/editor/cell/code/cell-editor.tsx +1 -1
  214. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  215. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  216. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +30 -1
  217. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  218. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  219. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +12 -18
  220. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +58 -1
  221. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  222. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  223. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  224. package/src/components/editor/connections/components.tsx +1 -1
  225. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  226. package/src/components/editor/controls/command-palette.tsx +22 -10
  227. package/src/components/editor/dynamic-favicon.tsx +21 -40
  228. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +12 -23
  229. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  230. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  231. package/src/components/editor/file-tree/file-explorer.tsx +10 -5
  232. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  233. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  234. package/src/components/editor/notebook-banner.tsx +1 -0
  235. package/src/components/editor/output/MarimoTracebackOutput.tsx +2 -2
  236. package/src/components/editor/output/ansi-reduce.ts +6 -0
  237. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  238. package/src/components/editor/package-alert.tsx +6 -4
  239. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  240. package/src/components/home/components.tsx +17 -31
  241. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  242. package/src/components/icons/large-spinner.tsx +3 -14
  243. package/src/components/icons/spinner.tsx +1 -1
  244. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  245. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  246. package/src/components/pages/home-page.tsx +164 -91
  247. package/src/components/scratchpad/scratchpad.tsx +2 -2
  248. package/src/components/slides/reveal-component.tsx +4 -2
  249. package/src/components/slides/slide-cell-view.tsx +1 -1
  250. package/src/components/slides/slide-form.tsx +1 -0
  251. package/src/components/tracing/tracing.tsx +2 -2
  252. package/src/components/ui/alert-dialog.tsx +1 -1
  253. package/src/core/ai/context/providers/datasource.ts +6 -2
  254. package/src/core/attachments/state.ts +5 -8
  255. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -0
  256. package/src/core/cells/__tests__/cell.test.ts +39 -0
  257. package/src/core/cells/__tests__/cells.test.ts +3 -1
  258. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  259. package/src/core/cells/__tests__/notebook-outline.test.ts +113 -0
  260. package/src/core/cells/cell.ts +14 -3
  261. package/src/core/cells/cells.ts +3 -1
  262. package/src/core/cells/pending-delete-service.ts +2 -6
  263. package/src/core/cells/session.ts +1 -1
  264. package/src/core/cells/types.ts +10 -1
  265. package/src/core/codemirror/__tests__/setup.test.ts +1 -1
  266. package/src/core/codemirror/keymaps/vim.ts +2 -6
  267. package/src/core/codemirror/language/languages/markdown.ts +1 -1
  268. package/src/core/codemirror/language/languages/python.ts +3 -3
  269. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  270. package/src/core/datasets/__tests__/data-source-discovery.test.ts +4 -4
  271. package/src/core/datasets/data-source-discovery.ts +4 -6
  272. package/src/core/document/__tests__/draft.test.ts +322 -0
  273. package/src/core/document/__tests__/fixtures.ts +92 -0
  274. package/src/core/document/__tests__/model.test.ts +483 -0
  275. package/src/core/document/__tests__/state.test.ts +446 -0
  276. package/src/core/document/__tests__/submission.test.ts +346 -0
  277. package/src/core/document/draft.ts +472 -0
  278. package/src/core/document/index.ts +10 -0
  279. package/src/core/document/model.ts +53 -0
  280. package/src/core/document/state.ts +401 -0
  281. package/src/core/document/sync.ts +359 -0
  282. package/src/core/edit-app.tsx +2 -2
  283. package/src/core/host/AGENTS.md +12 -2
  284. package/src/core/host/__tests__/attachments.test.ts +56 -0
  285. package/src/core/host/__tests__/document.test.ts +126 -0
  286. package/src/core/host/__tests__/fixtures.ts +41 -0
  287. package/src/core/host/__tests__/notebook-client.test.ts +570 -0
  288. package/src/core/host/connection.ts +42 -0
  289. package/src/core/host/notebook-client.ts +297 -0
  290. package/src/core/host/resources/attachments.ts +22 -0
  291. package/src/core/host/resources/document.ts +101 -0
  292. package/src/core/hotkeys/actions.ts +2 -4
  293. package/src/core/islands/__tests__/bootstrap.test.ts +0 -26
  294. package/src/core/islands/bootstrap.ts +0 -1
  295. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  296. package/src/core/network/connection.ts +4 -2
  297. package/src/core/network/useConnectionNotice.ts +8 -2
  298. package/src/core/run-app.tsx +2 -2
  299. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  300. package/src/core/runtime/__tests__/runtime.test.ts +18 -17
  301. package/src/core/runtime/runtime.ts +26 -30
  302. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  303. package/src/core/state/jotai.ts +1 -1
  304. package/src/core/static/files.ts +2 -0
  305. package/src/core/wasm/bridge.ts +2 -10
  306. package/src/core/wasm/store.ts +3 -3
  307. package/src/core/wasm/worker/bootstrap.ts +1 -1
  308. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  309. package/src/core/websocket/__tests__/{useMarimoKernelConnection.test.ts → connection-state.test.ts} +15 -29
  310. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +223 -0
  311. package/src/core/websocket/__tests__/useNotebookSession.hook.test.tsx +606 -0
  312. package/src/core/websocket/connection-state.ts +60 -0
  313. package/src/core/websocket/state.ts +7 -0
  314. package/src/core/websocket/transports/__tests__/notebook.test.ts +502 -0
  315. package/src/core/websocket/transports/notebook.ts +213 -0
  316. package/src/core/websocket/types.ts +9 -39
  317. package/src/core/websocket/useNotebookClient.ts +89 -0
  318. package/src/core/websocket/{useMarimoKernelConnection.tsx → useNotebookSession.tsx} +53 -233
  319. package/src/css/globals.css +5 -0
  320. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  321. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  322. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  323. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  324. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  325. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  326. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  327. package/src/hooks/useAsyncData.ts +3 -3
  328. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  329. package/src/hooks/useDebounce.ts +5 -8
  330. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  331. package/src/hooks/useDelayElapsed.ts +14 -10
  332. package/src/hooks/useElapsedTime.ts +3 -3
  333. package/src/hooks/useInputHistory.ts +4 -7
  334. package/src/hooks/useInternalStateWithSync.ts +6 -6
  335. package/src/hooks/useResizeObserver.ts +4 -7
  336. package/src/hooks/useScript.ts +67 -70
  337. package/src/hooks/useStateForKey.ts +93 -0
  338. package/src/plugins/core/registerReactComponent.tsx +1 -1
  339. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  340. package/src/plugins/impl/DataTablePlugin.tsx +14 -15
  341. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  342. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  343. package/src/plugins/impl/FormPlugin.tsx +1 -1
  344. package/src/plugins/impl/RangeSliderPlugin.tsx +0 -1
  345. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  346. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  347. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +117 -1
  348. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  349. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  350. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +2 -0
  351. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  352. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  353. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  354. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  355. package/src/plugins/impl/chat/chat-ui.tsx +12 -10
  356. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  357. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  358. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  359. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  360. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  361. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +0 -1
  362. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  363. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  364. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  365. package/src/plugins/impl/panel/utils.ts +17 -0
  366. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  367. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  368. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  369. package/src/stories/button.stories.tsx +1 -0
  370. package/src/stories/editor.stories.tsx +24 -17
  371. package/src/stories/form-wrapper.stories.tsx +2 -2
  372. package/src/stories/select.stories.tsx +2 -2
  373. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  374. package/src/utils/__tests__/arrays.test.ts +35 -0
  375. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  376. package/src/utils/__tests__/numbers.test.ts +48 -1
  377. package/src/utils/__tests__/objects.test.ts +21 -0
  378. package/src/utils/arrays.ts +10 -0
  379. package/src/utils/blob.ts +1 -1
  380. package/src/utils/numbers.ts +28 -17
  381. package/src/utils/objects.ts +3 -7
  382. package/src/utils/once.ts +19 -19
  383. package/src/utils/zod-utils.ts +1 -3
  384. package/dist/assets/CellStatus-nZpuDkm7.js +0 -1
  385. package/dist/assets/JsonOutput-BxB0NL8w.js +0 -11
  386. package/dist/assets/MarimoErrorOutput-BTy4MOzX.js +0 -10
  387. package/dist/assets/RunButton-CFQUNN5c.js +0 -1
  388. package/dist/assets/__vite-browser-external-CYO_P3S8.js +0 -1
  389. package/dist/assets/__vite-browser-external-FeDVGZU8.js +0 -1
  390. package/dist/assets/ai-model-dropdown-zm-2fXzf.js +0 -13
  391. package/dist/assets/any-language-editor-BLCAD3yx.js +0 -1
  392. package/dist/assets/arrays-BqUx5USx.js +0 -1
  393. package/dist/assets/cell-editor-CAj1UKxx.js +0 -21
  394. package/dist/assets/cell-link-1CynTVGQ.js +0 -1
  395. package/dist/assets/chat-panel-Ce2Xh7dM.js +0 -4
  396. package/dist/assets/chat-ui-DTy98rP3.js +0 -4
  397. package/dist/assets/circle-play-CYIb0qT6.js +0 -1
  398. package/dist/assets/code-visibility-B-_GvwR5.js +0 -1
  399. package/dist/assets/coerce-CpE6kUdX.js +0 -1
  400. package/dist/assets/column-preview-DU6dT80y.js +0 -1
  401. package/dist/assets/command-palette-BMqFKXVr.js +0 -1
  402. package/dist/assets/components-CXfO-F47.js +0 -1
  403. package/dist/assets/config-DX8Y2WbJ.js +0 -1
  404. package/dist/assets/config-VPSq73n8.js +0 -1
  405. package/dist/assets/edit-page-DlfCLUaq.js +0 -9
  406. package/dist/assets/feedback-button-DlqZkJwz.js +0 -1
  407. package/dist/assets/file-explorer-panel-Bo9eN3rW.js +0 -1
  408. package/dist/assets/home-page-KrhQ55Vg.js +0 -4
  409. package/dist/assets/index-6b4mf8Ai.css +0 -2
  410. package/dist/assets/index-B3fcrmuf.js +0 -30
  411. package/dist/assets/kiosk-mode-BZECfg9k.js +0 -1
  412. package/dist/assets/layout-DOpdUwym.js +0 -9
  413. package/dist/assets/linear-0eij_aAG.js +0 -1
  414. package/dist/assets/link-NuFVR7G9.js +0 -1
  415. package/dist/assets/links-DkkL9DVa.js +0 -1
  416. package/dist/assets/numbers-Mk3-CcoU.js +0 -1
  417. package/dist/assets/packages-panel-ClxOAEGy.js +0 -1
  418. package/dist/assets/panels-DfCmzLtT.js +0 -4
  419. package/dist/assets/precisionRound-Bq_nmf5u.js +0 -1
  420. package/dist/assets/process-output-DNFKMODr.js +0 -1
  421. package/dist/assets/react-vega-C3rYGtP6.js +0 -189
  422. package/dist/assets/requests-CgSY1X4a.js +0 -1
  423. package/dist/assets/run-page-Cxu1yV8Q.js +0 -1
  424. package/dist/assets/scratchpad-panel-By-ybhjg.js +0 -1
  425. package/dist/assets/session-panel-BmW6PH4z.js +0 -1
  426. package/dist/assets/state-BAJYRhmO.js +0 -1
  427. package/dist/assets/state-z3dlofZB.js +0 -3
  428. package/dist/assets/tooltip-mT3CGest.js +0 -61
  429. package/dist/assets/tracing-BARlJ_Dk.js +0 -1
  430. package/dist/assets/tracing-panel-dgnr2fIN.js +0 -2
  431. package/dist/assets/useAsyncData-C7aJJz0Y.js +0 -1
  432. package/dist/assets/useBoolean-BDD6_xQQ.js +0 -1
  433. package/dist/assets/useCellActionButton-DY9RlJTb.js +0 -1
  434. package/dist/assets/useDebounce-B_ccBZTp.js +0 -1
  435. package/dist/assets/useNotebookActions-DfvB7RD_.js +0 -1
  436. package/dist/assets/vega-component-CnadTgoE.js +0 -1
  437. package/dist/assets/vega-loader.browser-CLRkNgm7.js +0 -6
  438. package/dist/assets/ws-BXWY541V.js +0 -22
  439. package/src/core/host/__tests__/protocol.test.ts +0 -214
  440. package/src/core/host/protocol.ts +0 -1943
  441. package/src/core/state/__tests__/jotai.test.ts +0 -25
  442. package/src/core/websocket/__tests__/useMarimoKernelConnection.hook.test.tsx +0 -789
  443. package/src/core/websocket/__tests__/useWebSocket.test.ts +0 -49
  444. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  445. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -338
  446. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  447. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  448. package/src/core/websocket/transports/basic.ts +0 -102
  449. package/src/core/websocket/transports/sse.ts +0 -308
  450. package/src/core/websocket/transports/transport.ts +0 -61
  451. package/src/core/websocket/transports/ws.ts +0 -98
  452. package/src/core/websocket/useWebSocket.tsx +0 -88
@@ -1,7 +1,7 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
3
  import { CheckIcon, CopyIcon } from "lucide-react";
4
- import React, { useEffect, useState } from "react";
4
+ import React, { useEffect, useMemo, useState } from "react";
5
5
  import { useAtomValue } from "jotai";
6
6
  import { Button } from "@/components/ui/button";
7
7
  import {
@@ -53,21 +53,20 @@ export const PairWithAgentModal: React.FC<{
53
53
  const config = useAtomValue(pairConfigAtom);
54
54
  const hostEnabled = config?.hostEnabled ?? false;
55
55
  const stableSessionId = useAtomValue(stableSessionIdAtom);
56
+ const lookupIdentity = useMemo(
57
+ () => ({ hostEnabled, stableSessionId }),
58
+ [hostEnabled, stableSessionId],
59
+ );
56
60
  const [lookup, setLookup] = useState<
57
61
  {
58
- sessionId: string;
59
- } & (
60
- | { status: "ready"; notebookId: string | null }
61
- | { status: "loading" | "error" }
62
- )
62
+ identity: typeof lookupIdentity;
63
+ } & ({ status: "ready"; notebookId: string | null } | { status: "error" })
63
64
  >();
64
65
  useEffect(() => {
65
- if (!hostEnabled || !stableSessionId) {
66
- setLookup(undefined);
66
+ const { hostEnabled, stableSessionId: sessionId } = lookupIdentity;
67
+ if (!hostEnabled || !sessionId) {
67
68
  return;
68
69
  }
69
- const sessionId = stableSessionId;
70
- setLookup({ sessionId, status: "loading" });
71
70
  let cancelled = false;
72
71
  API.get<Record<string, { session_id: string; notebook_id: string | null }>>(
73
72
  "/sessions",
@@ -79,22 +78,26 @@ export const PairWithAgentModal: React.FC<{
79
78
  );
80
79
  setLookup(
81
80
  session
82
- ? { sessionId, status: "ready", notebookId: session.notebook_id }
83
- : { sessionId, status: "error" },
81
+ ? {
82
+ identity: lookupIdentity,
83
+ status: "ready",
84
+ notebookId: session.notebook_id,
85
+ }
86
+ : { identity: lookupIdentity, status: "error" },
84
87
  );
85
88
  }
86
89
  })
87
90
  .catch(() => {
88
91
  if (!cancelled) {
89
- setLookup({ sessionId, status: "error" });
92
+ setLookup({ identity: lookupIdentity, status: "error" });
90
93
  }
91
94
  });
92
95
  return () => {
93
96
  cancelled = true;
94
97
  };
95
- }, [hostEnabled, stableSessionId]);
98
+ }, [lookupIdentity]);
96
99
  const currentLookup =
97
- lookup?.sessionId === stableSessionId ? lookup : undefined;
100
+ lookup?.identity === lookupIdentity ? lookup : undefined;
98
101
  const verified = !hostEnabled || currentLookup?.status === "ready";
99
102
  const notebookId =
100
103
  hostEnabled && currentLookup?.status === "ready"
@@ -106,9 +106,9 @@ export const AddCellWithAI: React.FC<{
106
106
  const runtimeManager = useRuntimeManager();
107
107
 
108
108
  const inputRef = useRef<ReactCodeMirrorRef>(null);
109
- const submissionController = useRef(
110
- new StagedCellSubmissionController(),
111
- ).current;
109
+ const [submissionController] = useState(
110
+ () => new StagedCellSubmissionController(),
111
+ );
112
112
 
113
113
  const fileInputRef = useRef<HTMLInputElement>(null);
114
114
  const { files, addFiles, removeFile } = useFileState();
@@ -408,6 +408,7 @@ export const AiCompletionEditor: React.FC<Props> = ({
408
408
  </div>
409
409
  </Tooltip>
410
410
  <Button
411
+ aria-label="Discard AI completion"
411
412
  data-testid="decline-completion-button"
412
413
  variant="text"
413
414
  size="icon"
@@ -51,6 +51,7 @@ export const StartupLogsAlert: React.FC = () => {
51
51
  {isRunning ? "Initializing..." : "Initialized"}
52
52
  </span>
53
53
  <Button
54
+ aria-label="Dismiss startup logs"
54
55
  variant="text"
55
56
  data-testid="remove-startup-logs-button"
56
57
  size="icon"
@@ -1,6 +1,6 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
  import { DatabaseIcon, DiamondPlusIcon, PlusIcon } from "lucide-react";
3
- import { useState } from "react";
3
+ import { useRef, useState } from "react";
4
4
  import { Button } from "@/components/editor/inputs/Inputs";
5
5
  import { MinimalHotkeys } from "@/components/shortcuts/renderShortcut";
6
6
  import {
@@ -36,7 +36,8 @@ export const CreateCellButton = ({
36
36
  const { createNewCell, addSetupCellIfDoesntExist } = useCellActions();
37
37
  const shortcut = `${oneClickShortcut}-Click`;
38
38
  const [open, setOpen] = useState(false);
39
- const [justOpened, setJustOpened] = useState(false);
39
+ // Time the dropdown last opened, to ignore the click that opened it
40
+ const openedAt = useRef(0);
40
41
 
41
42
  const baseTooltipContent =
42
43
  getConnectionTooltip(connectionState) || tooltipContent;
@@ -83,9 +84,7 @@ export const CreateCellButton = ({
83
84
 
84
85
  const openDropdown = () => {
85
86
  setOpen(true);
86
- setJustOpened(true);
87
- // Allow interactions after a brief delay to prevent the dropdown items immediately being clicked
88
- setTimeout(() => setJustOpened(false), 200);
87
+ openedAt.current = Date.now();
89
88
  };
90
89
 
91
90
  // We use onPointerDownCapture (not onPointerDown) to intercept events in
@@ -121,15 +120,14 @@ export const CreateCellButton = ({
121
120
  const handleOpenChange = (isOpen: boolean) => {
122
121
  setOpen(isOpen);
123
122
  if (isOpen) {
124
- setJustOpened(true);
125
- // Allow interactions after a brief delay
126
- setTimeout(() => setJustOpened(false), 200);
123
+ openedAt.current = Date.now();
127
124
  }
128
125
  };
129
126
 
130
127
  const handleFirstItemClick = (e: React.MouseEvent) => {
131
- // Hack to prevent the first item from being clicked when the dropdown is opened
132
- if (justOpened) {
128
+ // Hack to prevent the first item from being clicked when the dropdown is
129
+ // opened: allow interactions only after a brief delay
130
+ if (Date.now() - openedAt.current < 200) {
133
131
  e.preventDefault();
134
132
  e.stopPropagation();
135
133
  return;
@@ -37,6 +37,7 @@ export const DeleteButton = (props: {
37
37
  return (
38
38
  <Tooltip content={tooltipMsg} usePortal={false}>
39
39
  <Button
40
+ aria-label="Delete cell"
40
41
  variant="ghost"
41
42
  size="icon"
42
43
  onClick={onClick}
@@ -0,0 +1,39 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { act, renderHook } from "@testing-library/react";
4
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
5
+ import { useShouldShowInterrupt } from "../useShouldShowInterrupt";
6
+
7
+ describe("useShouldShowInterrupt", () => {
8
+ beforeEach(() => vi.useFakeTimers());
9
+ afterEach(() => vi.useRealTimers());
10
+
11
+ it("waits for each run and hides immediately when it stops", () => {
12
+ const { result, rerender } = renderHook(useShouldShowInterrupt, {
13
+ initialProps: false,
14
+ });
15
+ expect(result.current).toBe(false);
16
+ rerender(true);
17
+ expect(result.current).toBe(false);
18
+ act(() => vi.advanceTimersByTime(200));
19
+ expect(result.current).toBe(true);
20
+ rerender(false);
21
+ expect(result.current).toBe(false);
22
+ rerender(true);
23
+ expect(result.current).toBe(false);
24
+ act(() => vi.advanceTimersByTime(199));
25
+ expect(result.current).toBe(false);
26
+ act(() => vi.advanceTimersByTime(1));
27
+ expect(result.current).toBe(true);
28
+ });
29
+
30
+ it("cancels the delay for a short run", () => {
31
+ const { result, rerender } = renderHook(useShouldShowInterrupt, {
32
+ initialProps: true,
33
+ });
34
+ act(() => vi.advanceTimersByTime(100));
35
+ rerender(false);
36
+ act(() => vi.advanceTimersByTime(200));
37
+ expect(result.current).toBe(false);
38
+ });
39
+ });
@@ -0,0 +1,54 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { EditorState } from "@codemirror/state";
4
+ import { EditorView } from "@codemirror/view";
5
+ import { fireEvent, render, screen } from "@testing-library/react";
6
+ import { createRef } from "react";
7
+ import { afterEach, describe, expect, it, vi } from "vitest";
8
+ import { TooltipProvider } from "@/components/ui/tooltip";
9
+ import { switchLanguage } from "@/core/codemirror/language/extension";
10
+ import { LanguageToggle } from "../language-toggle";
11
+
12
+ vi.mock("@/core/codemirror/language/extension", () => ({
13
+ switchLanguage: vi.fn(),
14
+ }));
15
+
16
+ describe("LanguageToggle", () => {
17
+ afterEach(() => {
18
+ vi.mocked(switchLanguage).mockClear();
19
+ });
20
+
21
+ it("uses the editor mounted after the button rendered", () => {
22
+ const editorViewRef = createRef<EditorView>();
23
+ const onAfterToggle = vi.fn();
24
+ render(
25
+ <TooltipProvider>
26
+ <LanguageToggle
27
+ editorViewRef={editorViewRef}
28
+ currentLanguageAdapter="python"
29
+ canSwitchToLanguage={true}
30
+ toType="markdown"
31
+ displayName="Markdown"
32
+ icon="Markdown"
33
+ onAfterToggle={onAfterToggle}
34
+ />
35
+ </TooltipProvider>,
36
+ );
37
+
38
+ fireEvent.click(screen.getByRole("button", { name: "Markdown" }));
39
+ expect(switchLanguage).not.toHaveBeenCalled();
40
+ expect(onAfterToggle).not.toHaveBeenCalled();
41
+
42
+ const view = new EditorView({ state: EditorState.create() });
43
+ editorViewRef.current = view;
44
+ try {
45
+ fireEvent.click(screen.getByRole("button", { name: "Markdown" }));
46
+ expect(switchLanguage).toHaveBeenCalledWith(view, {
47
+ language: "markdown",
48
+ });
49
+ expect(onAfterToggle).toHaveBeenCalledOnce();
50
+ } finally {
51
+ view.destroy();
52
+ }
53
+ });
54
+ });
@@ -572,7 +572,7 @@ const CellEditorInternal = ({
572
572
  <div className="absolute top-1 right-5">
573
573
  <LanguageToggles
574
574
  code={code}
575
- editorView={editorViewRef.current}
575
+ editorViewRef={editorViewRef}
576
576
  currentLanguageAdapter={languageAdapter}
577
577
  onAfterToggle={afterToggleMarkdown}
578
578
  />
@@ -13,7 +13,7 @@ import { Functions } from "@/utils/functions";
13
13
  import { MarkdownIcon, PythonIcon } from "./icons";
14
14
 
15
15
  interface LanguageTogglesProps {
16
- editorView: EditorView | null;
16
+ editorViewRef: React.RefObject<EditorView | null>;
17
17
  code: string;
18
18
  currentLanguageAdapter: LanguageAdapter["type"] | undefined;
19
19
  onAfterToggle: () => void;
@@ -25,7 +25,7 @@ interface LanguageTogglesProps {
25
25
  }
26
26
 
27
27
  export const LanguageToggles: React.FC<LanguageTogglesProps> = ({
28
- editorView,
28
+ editorViewRef,
29
29
  code,
30
30
  currentLanguageAdapter,
31
31
  onAfterToggle,
@@ -43,7 +43,7 @@ export const LanguageToggles: React.FC<LanguageTogglesProps> = ({
43
43
  return (
44
44
  <div className={className}>
45
45
  <LanguageToggle
46
- editorView={editorView}
46
+ editorViewRef={editorViewRef}
47
47
  currentLanguageAdapter={currentLanguageAdapter}
48
48
  canSwitchToLanguage={canUseSQL && currentLanguageAdapter === "python"}
49
49
  icon={
@@ -58,7 +58,7 @@ export const LanguageToggles: React.FC<LanguageTogglesProps> = ({
58
58
  onAfterToggle={onAfterToggle}
59
59
  />
60
60
  <LanguageToggle
61
- editorView={editorView}
61
+ editorViewRef={editorViewRef}
62
62
  currentLanguageAdapter={currentLanguageAdapter}
63
63
  canSwitchToLanguage={
64
64
  canUseMarkdown && currentLanguageAdapter === "python"
@@ -75,7 +75,7 @@ export const LanguageToggles: React.FC<LanguageTogglesProps> = ({
75
75
  onAfterToggle={onAfterToggle}
76
76
  />
77
77
  <LanguageToggle
78
- editorView={editorView}
78
+ editorViewRef={editorViewRef}
79
79
  currentLanguageAdapter={currentLanguageAdapter}
80
80
  canSwitchToLanguage={true}
81
81
  icon={
@@ -95,7 +95,7 @@ export const LanguageToggles: React.FC<LanguageTogglesProps> = ({
95
95
 
96
96
  interface Props {
97
97
  className?: string;
98
- editorView: EditorView | null;
98
+ editorViewRef: React.RefObject<EditorView | null>;
99
99
  canSwitchToLanguage: boolean;
100
100
  currentLanguageAdapter: LanguageAdapter["type"] | undefined;
101
101
  toType: LanguageAdapter["type"];
@@ -105,7 +105,7 @@ interface Props {
105
105
  }
106
106
 
107
107
  export const LanguageToggle: React.FC<Props> = ({
108
- editorView,
108
+ editorViewRef,
109
109
  currentLanguageAdapter,
110
110
  canSwitchToLanguage,
111
111
  icon,
@@ -114,6 +114,7 @@ export const LanguageToggle: React.FC<Props> = ({
114
114
  onAfterToggle,
115
115
  }) => {
116
116
  const handleClick = () => {
117
+ const editorView = editorViewRef.current;
117
118
  if (!editorView) {
118
119
  return;
119
120
  }
@@ -1,5 +1,5 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
- import { useEffect, useState } from "react";
2
+ import { useDelayElapsed } from "@/hooks/useDelayElapsed";
3
3
 
4
4
  const DELAY_MS = 200;
5
5
 
@@ -10,17 +10,7 @@ const DELAY_MS = 200;
10
10
  export function useShouldShowInterrupt(running: boolean) {
11
11
  // Start a timer when the run starts.
12
12
  // After 200ms, show the interrupt button to avoid flickering.
13
- const [hasRunLongEnough, setHasRunLongEnough] = useState(false);
14
- useEffect(() => {
15
- if (!running) {
16
- return;
17
- }
18
- setHasRunLongEnough(false);
19
- const timeout = setTimeout(() => {
20
- setHasRunLongEnough(true);
21
- }, DELAY_MS);
22
- return () => clearTimeout(timeout);
23
- }, [running]);
13
+ const hasRunLongEnough = useDelayElapsed(running ? DELAY_MS : 0);
24
14
 
25
15
  return running && hasRunLongEnough;
26
16
  }
@@ -18,7 +18,10 @@ import type {
18
18
  DependencyTreeNode,
19
19
  DependencyTreeResponse,
20
20
  } from "@/core/network/types";
21
- import { withPackageInvalidation } from "@/core/packages/package-data";
21
+ import {
22
+ invalidatePackageData,
23
+ withPackageInvalidation,
24
+ } from "@/core/packages/package-data";
22
25
  import { useInstallPackages } from "@/core/packages/useInstallPackage";
23
26
  import { sandboxAtom, sandboxSyncAtom } from "@/core/packages/sandbox-state";
24
27
  import { store } from "@/core/state/jotai";
@@ -143,6 +146,32 @@ describe("PackagesPanel", () => {
143
146
  expect(screen.queryByText("cycle")).not.toBeInTheDocument();
144
147
  });
145
148
 
149
+ it("collapses expanded nodes when the dependency tree is refreshed", async () => {
150
+ const { client } = renderPanel(
151
+ { kind: "sandbox", backend: "pixi" },
152
+ populatedTree,
153
+ );
154
+ fireEvent.click(await screen.findByRole("treeitem", { name: /polars/ }));
155
+ expect(screen.getByText("already in tree")).toBeInTheDocument();
156
+
157
+ vi.mocked(client.getDependencyTree).mockResolvedValue({
158
+ tree: structuredClone(populatedTree),
159
+ context: { kind: "sandbox", backend: "pixi" },
160
+ });
161
+ act(() => invalidatePackageData());
162
+
163
+ await waitFor(() =>
164
+ expect(client.getDependencyTree).toHaveBeenCalledTimes(2),
165
+ );
166
+ await waitFor(() =>
167
+ expect(screen.queryByText("already in tree")).not.toBeInTheDocument(),
168
+ );
169
+ expect(screen.getByRole("treeitem", { name: /polars/ })).toHaveAttribute(
170
+ "aria-expanded",
171
+ "false",
172
+ );
173
+ });
174
+
146
175
  it("keeps the configured package manager changeable outside a sandbox", async () => {
147
176
  const { getPackageList } = renderPanel({
148
177
  kind: "package-manager",
@@ -104,6 +104,7 @@ const CachePanel = () => {
104
104
  {/* Header with Refresh Button */}
105
105
  <div className="flex items-center justify-end">
106
106
  <Button
107
+ aria-label="Refresh cache"
107
108
  variant="ghost"
108
109
  size="icon"
109
110
  className="h-6 w-6"
@@ -31,6 +31,7 @@ import { usePackageAction } from "@/core/packages/usePackageAction";
31
31
  import { usePackageDependencies } from "@/core/packages/usePackageDependencies";
32
32
  import { useInstallPackages } from "@/core/packages/useInstallPackage";
33
33
  import { isWasm } from "@/core/wasm/utils";
34
+ import { useStateForKey } from "@/hooks/useStateForKey";
34
35
  import { ErrorBanner } from "@/plugins/impl/common/error-banner";
35
36
  import { cn } from "@/utils/cn";
36
37
  import { copyToClipboard } from "@/utils/copy";
@@ -440,15 +441,11 @@ const DependencyTree: React.FC<{
440
441
  error?: Error | null;
441
442
  sandboxBackend: "uv" | "pixi" | null;
442
443
  }> = ({ tree, error, sandboxBackend }) => {
443
- const [expandedNodes, setExpandedNodes] = React.useState<Set<string>>(
444
- new Set(),
444
+ const [expandedNodes, setExpandedNodes] = useStateForKey(
445
+ tree,
446
+ () => new Set<string>(),
445
447
  );
446
448
 
447
- // Reset tree to collapsed state when tree data changes (including refetches)
448
- React.useEffect(() => {
449
- setExpandedNodes(new Set());
450
- }, [tree]);
451
-
452
449
  if (error) {
453
450
  return <ErrorBanner error={error} />;
454
451
  }
@@ -8,27 +8,31 @@ import {
8
8
  screen,
9
9
  waitFor,
10
10
  } from "@testing-library/react";
11
- import { createStore, Provider } from "jotai";
11
+ import { atom, createStore, Provider } from "jotai";
12
12
  import { userEvent } from "storybook/test";
13
13
  import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
14
14
  import { PAIR_CONFIG } from "@/__tests__/fixtures/pair-config";
15
15
  import { TooltipProvider } from "@/components/ui/tooltip";
16
16
  import { toast } from "@/components/ui/use-toast";
17
- import { type Attachment, attachmentsAtom } from "@/core/attachments/state";
17
+ import type { Attachment } from "@/core/attachments/state";
18
18
  import { pairConfigAtom } from "@/core/config/pair";
19
+ import { createNotebookClient } from "@/core/host/notebook-client";
19
20
  import {
20
21
  type StableSessionId,
21
22
  stableSessionIdAtom,
22
23
  } from "@/core/kernel/session";
23
24
  import { API } from "@/core/network/api";
24
25
  import { connectionAtom } from "@/core/network/connection";
25
- import { WebSocketClosedReason, ConnectionState } from "@/core/websocket/types";
26
+ import { notebookClientAtom } from "@/core/websocket/state";
27
+ import { ConnectionState } from "@/core/websocket/types";
26
28
  import { Deferred } from "@/utils/Deferred";
27
29
  import { AttachmentStatus } from "../footer-items/attachment-status";
28
30
 
29
31
  vi.mock("@/components/ui/use-toast", () => ({ toast: vi.fn() }));
30
32
  vi.mock("@/core/network/api", () => ({ API: { post: vi.fn() } }));
31
33
 
34
+ const attachmentsAtom = atom<Attachment[]>([]);
35
+
32
36
  const AGENT: Attachment = {
33
37
  id: "a1",
34
38
  kind: "agent",
@@ -41,6 +45,11 @@ function renderStatus(attachments: Attachment[] = [AGENT], preview = true) {
41
45
  const store = createStore();
42
46
  store.set(connectionAtom, { state: ConnectionState.OPEN });
43
47
  store.set(pairConfigAtom, { ...PAIR_CONFIG, hostEnabled: preview });
48
+ store.set(notebookClientAtom, {
49
+ ...createNotebookClient(store).client,
50
+ connectionAtom: atom({ state: ConnectionState.OPEN }),
51
+ attachmentsAtom,
52
+ });
44
53
  store.set(attachmentsAtom, attachments);
45
54
  store.set(stableSessionIdAtom, "stable/test" as StableSessionId);
46
55
  return {
@@ -342,21 +351,6 @@ describe("AttachmentStatus", () => {
342
351
  ).toBeDisabled();
343
352
  });
344
353
 
345
- it.each([
346
- { state: ConnectionState.CONNECTING },
347
- { state: ConnectionState.CLOSING },
348
- { state: ConnectionState.NOT_STARTED },
349
- {
350
- state: ConnectionState.CLOSED,
351
- code: WebSocketClosedReason.KERNEL_DISCONNECTED,
352
- reason: "offline",
353
- },
354
- ])("hides connected claims while the browser is %j", (connection) => {
355
- const { store, container } = renderStatus();
356
- act(() => store.set(connectionAtom, connection));
357
- expect(container).toBeEmptyDOMElement();
358
- });
359
-
360
354
  it("removes detached agents when the list is replaced", () => {
361
355
  const { store, container } = renderStatus([
362
356
  AGENT,
@@ -1,11 +1,12 @@
1
1
  /* Copyright 2026 Marimo. All rights reserved. */
2
2
 
3
- import { act, render, screen, within } from "@testing-library/react";
3
+ import { act, fireEvent, render, screen, within } from "@testing-library/react";
4
4
  import { createStore, Provider } from "jotai";
5
5
  import { describe, expect, it, vi } from "vitest";
6
6
  import { cellId } from "@/__tests__/branded";
7
7
  import { TooltipProvider } from "@/components/ui/tooltip";
8
8
  import { stagedAICellsAtom, visibleForTesting } from "@/core/ai/staged-cells";
9
+ import { scrollAndHighlightCell } from "../../../links/cell-link";
9
10
  import { PendingAICells } from "../pending-ai-cells";
10
11
 
11
12
  vi.mock("@/components/editor/cell/useDeleteCell", () => ({
@@ -29,6 +30,62 @@ vi.mock("@/components/editor/links/cell-link", () => ({
29
30
  }));
30
31
 
31
32
  describe("PendingAICells", () => {
33
+ it("navigates between pending cells through named controls", () => {
34
+ const store = createStore();
35
+ const first = cellId("first-pending-cell");
36
+ const second = cellId("second-pending-cell");
37
+ const third = cellId("third-pending-cell");
38
+ store.set(
39
+ stagedAICellsAtom,
40
+ new Map([
41
+ [first, { type: "add_cell" }],
42
+ [second, { type: "add_cell" }],
43
+ [third, { type: "add_cell" }],
44
+ ]),
45
+ );
46
+
47
+ render(
48
+ <Provider store={store}>
49
+ <TooltipProvider>
50
+ <PendingAICells />
51
+ </TooltipProvider>
52
+ </Provider>,
53
+ );
54
+
55
+ const next = screen.getByRole("button", { name: "Next pending AI cell" });
56
+ const previous = screen.getByRole("button", {
57
+ name: "Previous pending AI cell",
58
+ });
59
+
60
+ fireEvent.click(next);
61
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
62
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
63
+
64
+ fireEvent.click(next);
65
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(second, "focus");
66
+ expect(screen.getByText("2 / 3")).toBeInTheDocument();
67
+
68
+ fireEvent.click(previous);
69
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
70
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
71
+
72
+ fireEvent.click(next);
73
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(second, "focus");
74
+ expect(screen.getByText("2 / 3")).toBeInTheDocument();
75
+
76
+ fireEvent.click(next);
77
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(third, "focus");
78
+ expect(screen.getByText("3 / 3")).toBeInTheDocument();
79
+
80
+ fireEvent.click(next);
81
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(first, "focus");
82
+ expect(screen.getByText("1 / 3")).toBeInTheDocument();
83
+
84
+ fireEvent.click(previous);
85
+ expect(scrollAndHighlightCell).toHaveBeenLastCalledWith(third, "focus");
86
+ expect(screen.getByText("3 / 3")).toBeInTheDocument();
87
+ });
88
+
32
89
  it("disables bulk actions while cells are still generating", () => {
33
90
  const store = createStore();
34
91
  const generationId = Symbol("staged-cell-generation");
@@ -26,13 +26,7 @@ import { isWasm } from "@/core/wasm/utils";
26
26
  import { cn } from "@/utils/cn";
27
27
  import { FooterItem } from "../footer-item";
28
28
 
29
- interface RuntimeSettingsProps {
30
- className?: string;
31
- }
32
-
33
- export const RuntimeSettings: React.FC<RuntimeSettingsProps> = ({
34
- className,
35
- }) => {
29
+ export const RuntimeSettings: React.FC = () => {
36
30
  const { saveUserConfig } = useRequestClient();
37
31
  const [config, setUserConfig] = useResolvedMarimoConfig();
38
32
 
@@ -79,7 +79,12 @@ export const PendingAICells: React.FC = () => {
79
79
  <SparklesIcon className="h-4 w-4 text-primary" />
80
80
 
81
81
  <div className="flex items-center">
82
- <Button variant="ghost" size="icon" onClick={() => clickNext("up")}>
82
+ <Button
83
+ aria-label="Previous pending AI cell"
84
+ variant="ghost"
85
+ size="icon"
86
+ onClick={() => clickNext("down")}
87
+ >
83
88
  <ChevronUp className="h-3.5 w-3.5" />
84
89
  </Button>
85
90
  <span className="text-xs font-mono min-w-14 text-center">
@@ -87,7 +92,12 @@ export const PendingAICells: React.FC = () => {
87
92
  ? `${listStagedCells.length} pending`
88
93
  : `${currentIndex + 1} / ${listStagedCells.length}`}
89
94
  </span>
90
- <Button variant="ghost" size="icon" onClick={() => clickNext("down")}>
95
+ <Button
96
+ aria-label="Next pending AI cell"
97
+ variant="ghost"
98
+ size="icon"
99
+ onClick={() => clickNext("up")}
100
+ >
91
101
  <ChevronDown className="h-3.5 w-3.5" />
92
102
  </Button>
93
103
  </div>
@@ -15,7 +15,7 @@ let storedValue: string | null = null;
15
15
  */
16
16
  export function createStorage(location: "left" | "bottom"): PanelGroupStorage {
17
17
  return {
18
- getItem(name) {
18
+ getItem(_name) {
19
19
  if (!storedValue) {
20
20
  return storedValue ?? null;
21
21
  }
@@ -48,7 +48,7 @@ export const SelectorButton: React.FC<{
48
48
  }> = ({ name, color, icon, onSelect }) => (
49
49
  <button
50
50
  type="button"
51
- className="py-3 flex flex-col items-center justify-center gap-1 transition-all hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid"
51
+ className="py-3 flex flex-col items-center justify-center gap-1 transition-[scale,filter,box-shadow] hover:scale-105 hover:brightness-110 rounded shadow-sm-solid hover:shadow-md-solid"
52
52
  style={{ backgroundColor: color }}
53
53
  onClick={onSelect}
54
54
  >