@marimo-team/islands 0.25.3-dev24 → 0.25.3-dev244

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 (757) hide show
  1. package/dist/{ConnectedDataExplorerComponent-DfG3PRaJ.js → ConnectedDataExplorerComponent-BBihxHrr.js} +120 -116
  2. package/dist/{ErrorBoundary-LHjDFqXG.js → ErrorBoundary-DKniGLW9.js} +16 -16
  3. package/dist/_baseSet-BfI5FjBM.js +328 -0
  4. package/dist/{abnfDiagram-VCTEODGH-B3w-XWX8.js → abnfDiagram-VCTEODGH-DBcJVxYP.js} +2 -2
  5. package/dist/any-language-editor-jGaR0EAo.js +75 -0
  6. package/dist/{architectureDiagram-5GKGNRK7-C18U9A6k.js → architectureDiagram-5GKGNRK7-BcHZXF4S.js} +1 -1
  7. package/dist/{arrays-CN1djC4k.js → arrays-DoQ-AILx.js} +15 -27
  8. package/dist/assets/__vite-browser-external-BKBpimif.js +1 -0
  9. package/dist/assets/worker-CgoMtD-_.js +113 -0
  10. package/dist/{blockDiagram-I7D4REHJ-Cg25UrxT.js → blockDiagram-I7D4REHJ-Dkhe6_PP.js} +1 -1
  11. package/dist/{button-CvjkwdSk.js → button-B6RjDKJx.js} +227 -217
  12. package/dist/{capabilities-CQ5IjhzL.js → capabilities-CkVzWJEX.js} +2 -2
  13. package/dist/{chat-ui-w8SVynkn.js → chat-ui-DgaVxFrc.js} +1309 -1307
  14. package/dist/{check-BMKBBrdc.js → check-Z_vtI3OL.js} +1 -1
  15. package/dist/{chunk-IMKFNOWR-CrS5bzqW.js → chunk-IMKFNOWR-CTZlwBAA.js} +1 -1
  16. package/dist/{chunk-SHT3W25Y-StpvPvUG.js → chunk-SHT3W25Y-53g_RZHm.js} +1 -1
  17. package/dist/{chunk-SVP7TREG-CBXjbYZC.js → chunk-SVP7TREG-P6abNGV_.js} +1 -1
  18. package/dist/{chunk-TICWLB2K-Dp6bIb-q.js → chunk-TICWLB2K-BTJJqkyF.js} +1 -1
  19. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  20. package/dist/{classDiagram-ZZMXUADV-BFYuxV5g.js → classDiagram-ZZMXUADV-BUGqOrTm.js} +2 -2
  21. package/dist/{classDiagram-v2-VYDZK3BY-D_tjXVMl.js → classDiagram-v2-VYDZK3BY-CXpQvO-s.js} +2 -2
  22. package/dist/code-visibility-CIUgREo9.js +134 -0
  23. package/dist/{common-1pRFQo1N.js → common-DzCPFBsv.js} +9917 -9686
  24. package/dist/config-DbjzjwxX.js +636 -0
  25. package/dist/{copy-icon-BgUO8jsi.js → copy-icon-CG57cxwb.js} +3 -3
  26. package/dist/createReducer-DePYJ0F3.js +63 -0
  27. package/dist/{cynefinDiagram-5FMLGOSQ-DSgYrHB2.js → cynefinDiagram-5FMLGOSQ-DOcnl8wW.js} +1 -1
  28. package/dist/{dagre-GXQ25YYZ-GYlruCBf.js → dagre-GXQ25YYZ-CuHR9QfG.js} +3 -3
  29. package/dist/{dagre-BGSfXNEb.js → dagre-mqiMAsFY.js} +531 -680
  30. package/dist/{data-grid-overlay-editor-UR0x91Ij.js → data-grid-overlay-editor-PstJqwhX.js} +5 -5
  31. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  32. package/dist/{debounce-C3jyATpc.js → debounce-DrhhziJv.js} +2 -2
  33. package/dist/{diagram-S7CK7UJ4-DRu5LX5_.js → diagram-S7CK7UJ4-BPiG4kso.js} +1 -1
  34. package/dist/{diagram-UQ7AKVKN-DD7uiejP.js → diagram-UQ7AKVKN-_piBOvOC.js} +1 -1
  35. package/dist/{diagram-VX7I27RA-ByH9KAJJ.js → diagram-VX7I27RA-DCzVeJ8M.js} +1 -1
  36. package/dist/{diagram-Z3DM3KII-BsQHPRke.js → diagram-Z3DM3KII-D64CF1jQ.js} +1 -1
  37. package/dist/{dialog-Cc479ZEH.js → dialog-DsZ49mmU.js} +26 -26
  38. package/dist/{dist-Bvs24R_E.js → dist-11T5wq3_.js} +31 -30
  39. package/dist/{dist-F8Ep36rk.js → dist-17ugwGZc.js} +59 -58
  40. package/dist/{dist-BMQf5zIS.js → dist-7cRjf--y.js} +123 -122
  41. package/dist/{dist-diIx5qJW.js → dist-B8pkfkzc.js} +31 -30
  42. package/dist/{dist-CN_BeG2I.js → dist-B9UoM_wq.js} +58 -57
  43. package/dist/{dist-CLdpGQVr.js → dist-BGuMTs1i.js} +118 -117
  44. package/dist/{dist-BK4Oj99o.js → dist-BHW11fni.js} +18 -17
  45. package/dist/dist-BLY4m1Wi.js +850 -0
  46. package/dist/{dist-PXYI_yuP.js → dist-BNlcVoxs.js} +24 -23
  47. package/dist/{dist-D6d86eNJ.js → dist-BQYe5Bju.js} +47 -46
  48. package/dist/{dist-CkTTIdBe.js → dist-BX1mn8na.js} +337 -337
  49. package/dist/{dist-DiDibOYr.js → dist-Bh0PTKUW.js} +151 -150
  50. package/dist/{dist-CSJ0MDRw.js → dist-CKaXK92p.js} +56 -55
  51. package/dist/{dist-C8fAKSH1.js → dist-CLhwfhpJ.js} +53 -52
  52. package/dist/{dist-DoE4XJBC.js → dist-CYeqK4Xw.js} +9801 -10649
  53. package/dist/{dist-BLfKmR-l.js → dist-Cxlp3WH6.js} +16 -15
  54. package/dist/{dist-D-wwqlzJ.js → dist-DSJPGj42.js} +158 -157
  55. package/dist/{dist-BorMD-uX.js → dist-DU9ICsWZ.js} +75 -74
  56. package/dist/{dist-DBtSl3sf.js → dist-DWv0ALBi.js} +47 -46
  57. package/dist/{dist-OR1OnN-y.js → dist-DiBpWEsB.js} +31 -30
  58. package/dist/{dist-CrSP_H1M.js → dist-DiLFTlnl.js} +242 -242
  59. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  60. package/dist/{dist-CSbh1PNP.js → dist-cVDV5IuI.js} +23 -22
  61. package/dist/{dist-BuaRHxOc.js → dist-fRfaOoq8.js} +44 -43
  62. package/dist/{dist-Cg7gzw-a.js → dist-gTTI-9Px.js} +78 -77
  63. package/dist/dropdown-menu-BRiE7Vnp.js +6083 -0
  64. package/dist/{ebnfDiagram-PWID7BFC-CQRK5_eT.js → ebnfDiagram-PWID7BFC-BapQPML-.js} +2 -2
  65. package/dist/{erDiagram-RLTQ6QDP-BKtWrDVh.js → erDiagram-RLTQ6QDP-CNQvC2_Q.js} +1 -1
  66. package/dist/{error-banner-CFPHFw4h.js → error-banner-IsonnvA0.js} +84 -83
  67. package/dist/error-entries-CRPfxMqm.js +2505 -0
  68. package/dist/{esm-D2gTsXEV.js → esm-BgKjMvWw.js} +530 -529
  69. package/dist/{esm-AMFssoNp.js → esm-CHjVSexr.js} +328 -329
  70. package/dist/feedback-button-C0iLkz-H.js +374 -0
  71. package/dist/{flowDiagram-HODETNUW-1IN-ih2E.js → flowDiagram-HODETNUW-BBt6S6oe.js} +2 -2
  72. package/dist/{formats-X4mMQWzX.js → formats-BK4VefeF.js} +6 -6
  73. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  74. package/dist/{glide-data-editor-DPjObn47.js → glide-data-editor-8CBjuK8C.js} +417 -415
  75. package/dist/{glide-utils-DM9zBO4W.js → glide-utils-CQddGpa-.js} +16 -16
  76. package/dist/{globals-DOHFyDcV.js → globals-M9WsksQz.js} +3 -3
  77. package/dist/{graphlib-BmY_CCV9.js → graphlib-nlwT3Wvy.js} +177 -220
  78. package/dist/hasIn-C20JsQ2m.js +28 -0
  79. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  80. package/dist/{error-entries-Damuuclv.js → html-to-image-Gx4q3ZoD.js} +26911 -26769
  81. package/dist/{infoDiagram-27XIBGKW-CkB-zr4p.js → infoDiagram-27XIBGKW-Hd8Ifz82.js} +1 -1
  82. package/dist/isArrayLikeObject-B-HXCsex.js +626 -0
  83. package/dist/isEmpty-CdX7MTE1.js +14 -0
  84. package/dist/{ishikawaDiagram-5VMMS53U-wlyLjBmM.js → ishikawaDiagram-5VMMS53U-CP7hnHWx.js} +1 -1
  85. package/dist/{kanban-definition-UXKFOSKX-C2usY0i-.js → kanban-definition-UXKFOSKX-CX91sa9b.js} +1 -1
  86. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  87. package/dist/{loader-BQOfjTIJ.js → loader-BisjOLiW.js} +12 -12
  88. package/dist/main.js +10234 -8104
  89. package/dist/{mermaid-BA-Z6wGL.js → mermaid-Djb3Ni9t.js} +54 -54
  90. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  91. package/dist/{mindmap-definition-YA3MSWOX-nKOWS8RR.js → mindmap-definition-YA3MSWOX-DgCNo07O.js} +1 -1
  92. package/dist/{now-jM8-M7IR.js → now-BKZCLE5P.js} +12 -13
  93. package/dist/{number-overlay-editor-B3cSnrr6.js → number-overlay-editor-uvNY9IUz.js} +3 -3
  94. package/dist/{dist-CN8zI-XC.js → objectWithoutPropertiesLoose-ai5NUqGa.js} +262 -240
  95. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  96. package/dist/{once-BtE-wKRL.js → once-NnpX_HnX.js} +1 -1
  97. package/dist/{pegDiagram-XKGWAZYB-Dac9EIKD.js → pegDiagram-XKGWAZYB-SGJhnIKo.js} +2 -2
  98. package/dist/pick-tCszMUbm.js +35 -0
  99. package/dist/{pieDiagram-E7YTZNPT-B7S-DsW_.js → pieDiagram-E7YTZNPT-1stXIWCQ.js} +1 -1
  100. package/dist/precisionRound-DhAzeRcm.js +49 -0
  101. package/dist/{process-output-CTPTtxXB.js → process-output-D2ISIyIs.js} +4 -4
  102. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  103. package/dist/{railroadDiagram-O6MQD6OU-Cfa5Q-hr.js → railroadDiagram-O6MQD6OU-DY1mojzB.js} +2 -2
  104. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  105. package/dist/{requirementDiagram-BXWQKSXE-d8zXCJIk.js → requirementDiagram-BXWQKSXE-gHzhUp0D.js} +1 -1
  106. package/dist/{reveal-component-CFFzSsPX.js → reveal-component-GMnfGiUU.js} +430 -407
  107. package/dist/schemas-CdscWNy_.js +4747 -0
  108. package/dist/select-BSxlITBo.js +1200 -0
  109. package/dist/{sequenceDiagram-WJ2MYXX4-b6pei3lJ.js → sequenceDiagram-WJ2MYXX4-rjPtD9fB.js} +1 -1
  110. package/dist/{spec-BJzgmvVa.js → spec-Dq8RNplp.js} +41 -11
  111. package/dist/{stateDiagram-D77RDMKH-Be-xSpaZ.js → stateDiagram-D77RDMKH-DMNxjUxV.js} +4 -4
  112. package/dist/{stateDiagram-v2-MP3YSRHH-5Y-FxesJ.js → stateDiagram-v2-MP3YSRHH-CumAaqGo.js} +2 -2
  113. package/dist/static-state-BldGd99T.js +72 -0
  114. package/dist/strings-BGW-fXVJ.js +507 -0
  115. package/dist/style.css +1 -1
  116. package/dist/{swimlanes-42K2YHIH-nRRl4NU0.js → swimlanes-42K2YHIH-BU14_Tej.js} +1 -1
  117. package/dist/{swimlanesDiagram-VR7AAH4N-BS79510V.js → swimlanesDiagram-VR7AAH4N-CieAnPqJ.js} +2 -2
  118. package/dist/{swiper-component-DKntyn8p.js → swiper-component-Dir1o8ml.js} +2 -2
  119. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  120. package/dist/{timeline-definition-24CTP7MA-Cfzf8WPf.js → timeline-definition-24CTP7MA-CrVj4E1K.js} +1 -1
  121. package/dist/toast-BWIPi3aF.js +491 -0
  122. package/dist/{tooltip-CtBZ5q8p.js → tooltip-C3UnvsRe.js} +329 -305
  123. package/dist/use-restore-focus-BWO0z1lo.js +1672 -0
  124. package/dist/{useAsyncData-BD534XmW.js → useAsyncData-DsZMvNXn.js} +15 -14
  125. package/dist/{useDateFormatter-DkajGwwd.js → useDateFormatter-ChB4eyim.js} +2 -2
  126. package/dist/{useDeepCompareMemoize-C72_RNsu.js → useDeepCompareMemoize-KMBBux8C.js} +1 -1
  127. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-cMNQ80CK.js} +1 -1
  128. package/dist/useLifecycle-BquH9ET-.js +112 -0
  129. package/dist/useTheme-DgOLVISn.js +337 -0
  130. package/dist/utils-DXkj82sK.js +813 -0
  131. package/dist/{vega-component-DfcYgj-Y.js → vega-component-Bwg_Kuh-.js} +171 -170
  132. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  133. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  134. package/dist/{vennDiagram-4TSXK5OY-CeY6Z1xZ.js → vennDiagram-4TSXK5OY-ZLxQ8yhm.js} +1 -1
  135. package/dist/{wardleyDiagram-VM6X3IG4-Cxs9NtVi.js → wardleyDiagram-VM6X3IG4-D88pcLY1.js} +1 -1
  136. package/dist/{xychartDiagram-S5SC5T6Z-DNWbvPK7.js → xychartDiagram-S5SC5T6Z-_13d4g8g.js} +2 -2
  137. package/package.json +16 -20
  138. package/src/__mocks__/document.tsx +63 -0
  139. package/src/__mocks__/notebook.ts +3 -4
  140. package/src/__mocks__/partial.ts +26 -0
  141. package/src/__mocks__/requests.ts +1 -4
  142. package/src/__tests__/main.test.tsx +24 -1
  143. package/src/__tests__/module-entry.test.ts +44 -0
  144. package/src/__tests__/mount.test.ts +1 -21
  145. package/src/__tests__/{host-imports.test.ts → replica-imports.test.ts} +12 -11
  146. package/src/components/ai/__tests__/ai-model-dropdown.test.tsx +63 -0
  147. package/src/components/ai/ai-model-dropdown.tsx +11 -6
  148. package/src/components/app-config/ai-config.tsx +155 -149
  149. package/src/components/app-config/app-config-button.tsx +6 -5
  150. package/src/components/app-config/app-config-form.tsx +5 -9
  151. package/src/components/app-config/common.tsx +7 -9
  152. package/src/components/app-config/data-form.tsx +112 -113
  153. package/src/components/app-config/mcp-config.tsx +30 -53
  154. package/src/components/app-config/user-config-form.css +91 -0
  155. package/src/components/app-config/user-config-form.tsx +266 -349
  156. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  157. package/src/components/chat/__tests__/notebook-access-selector.test.tsx +64 -0
  158. package/src/components/chat/acp/__tests__/session-tabs.test.tsx +43 -0
  159. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  160. package/src/components/chat/acp/blocks.tsx +3 -2
  161. package/src/components/chat/acp/session-tabs.tsx +8 -3
  162. package/src/components/chat/capabilities-popover.tsx +1 -5
  163. package/src/components/chat/chat-components.tsx +2 -0
  164. package/src/components/chat/chat-panel.tsx +23 -104
  165. package/src/components/chat/chat-utils.ts +4 -8
  166. package/src/components/chat/notebook-access-selector.stories.tsx +22 -0
  167. package/src/components/chat/notebook-access-selector.tsx +100 -0
  168. package/src/components/data-table/TableBottomBar.tsx +4 -1
  169. package/src/components/data-table/TableTopBar.tsx +2 -0
  170. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  171. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  172. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  173. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  174. package/src/components/data-table/__tests__/data-table.test.tsx +1 -0
  175. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  176. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  177. package/src/components/data-table/__tests__/keyboard-selection.test.tsx +188 -0
  178. package/src/components/data-table/__tests__/pagination.test.tsx +28 -1
  179. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  180. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  181. package/src/components/data-table/cell-selection/feature.ts +2 -2
  182. package/src/components/data-table/cell-selection/types.ts +2 -2
  183. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  184. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  185. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  186. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  187. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  188. package/src/components/data-table/charts/charts.tsx +40 -30
  189. package/src/components/data-table/charts/storage.ts +16 -5
  190. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  191. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  192. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  193. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  194. package/src/components/data-table/columns.tsx +45 -46
  195. package/src/components/data-table/context-menu.tsx +36 -20
  196. package/src/components/data-table/data-table.tsx +17 -1
  197. package/src/components/data-table/date-popover.tsx +6 -4
  198. package/src/components/data-table/export-actions.tsx +76 -48
  199. package/src/components/data-table/pagination.tsx +44 -33
  200. package/src/components/data-table/range-focus/atoms.ts +4 -6
  201. package/src/components/data-table/range-focus/use-cell-range-selection.ts +65 -38
  202. package/src/components/data-table/range-focus/use-scroll-into-view.ts +21 -2
  203. package/src/components/data-table/renderers.tsx +16 -7
  204. package/src/components/data-table/row-viewer-panel/__tests__/row-viewer.test.tsx +24 -1
  205. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +4 -4
  206. package/src/components/data-table/search-highlight.tsx +79 -0
  207. package/src/components/data-table/types.ts +1 -0
  208. package/src/components/data-table/url-detector.tsx +49 -9
  209. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  210. package/src/components/datasources/__tests__/column-preview.test.tsx +96 -5
  211. package/src/components/datasources/__tests__/datasources-interactions.test.tsx +181 -0
  212. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  213. package/src/components/datasources/__tests__/utils.test.ts +57 -6
  214. package/src/components/datasources/column-preview.tsx +17 -12
  215. package/src/components/datasources/datasources.tsx +56 -43
  216. package/src/components/datasources/utils.ts +32 -23
  217. package/src/components/debugger/debugger-code.tsx +5 -0
  218. package/src/components/dependency-graph/custom-node.tsx +1 -1
  219. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  220. package/src/components/dependency-graph/elements.ts +24 -21
  221. package/src/components/dependency-graph/minimap-content.tsx +13 -19
  222. package/src/components/dependency-graph/panels.tsx +4 -2
  223. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  224. package/src/components/editor/SortableCell.tsx +1 -3
  225. package/src/components/editor/__tests__/Output.test.tsx +50 -32
  226. package/src/{core/attachments/__tests__/toast.test.tsx → components/editor/__tests__/agent-toast.test.tsx} +1 -1
  227. package/src/components/editor/__tests__/data-attributes.test.tsx +28 -14
  228. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  229. package/src/components/editor/__tests__/output-persistence.test.tsx +6 -2
  230. package/src/components/editor/__tests__/package-alert.test.tsx +22 -1
  231. package/src/components/editor/actions/__tests__/name-cell-input.test.tsx +69 -0
  232. package/src/components/editor/actions/__tests__/pair-with-agent-commands.test.ts +3 -3
  233. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +40 -10
  234. package/src/components/editor/actions/__tests__/pair-with-agent.test.tsx +6 -10
  235. package/src/components/editor/actions/__tests__/useCellActionButton.test.tsx +113 -0
  236. package/src/components/editor/actions/name-cell-input.tsx +4 -3
  237. package/src/components/editor/actions/pair-with-agent-commands.ts +10 -6
  238. package/src/components/editor/actions/pair-with-agent-modal.tsx +24 -21
  239. package/src/components/editor/actions/pair-with-agent.tsx +15 -10
  240. package/src/components/editor/actions/useCellActionButton.tsx +36 -31
  241. package/src/components/editor/actions/useConfigActions.tsx +3 -4
  242. package/src/components/editor/actions/useNotebookActions.tsx +25 -29
  243. package/src/components/editor/actions/useRestartKernel.tsx +1 -1
  244. package/src/components/editor/actions/useSetCodeVisibility.ts +11 -12
  245. package/src/components/editor/actions/useSetOutputsExpanded.ts +14 -19
  246. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  247. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  248. package/src/components/editor/alerts/__tests__/connection-notice.test.tsx +1 -1
  249. package/src/components/editor/alerts/connection-notice.tsx +1 -1
  250. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  251. package/src/components/editor/alerts/stdin-blocking-alert.tsx +1 -1
  252. package/src/components/editor/app-container.tsx +1 -1
  253. package/src/components/editor/boundary/__tests__/ErrorBoundary.test.tsx +1 -1
  254. package/src/components/editor/cell/CreateCellButton.tsx +15 -21
  255. package/src/components/editor/cell/DeleteButton.tsx +3 -2
  256. package/src/components/editor/cell/RunButton.tsx +2 -2
  257. package/src/components/editor/cell/StagedAICell.tsx +10 -2
  258. package/src/components/editor/cell/StopButton.tsx +2 -2
  259. package/src/components/editor/cell/__tests__/StagedAICell.test.tsx +4 -2
  260. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  261. package/src/components/editor/cell/cell-actions.tsx +12 -1
  262. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  263. package/src/components/editor/cell/code/cell-editor.tsx +46 -165
  264. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  265. package/src/components/editor/cell/collapse.tsx +11 -4
  266. package/src/components/editor/cell/useAddCell.ts +4 -8
  267. package/src/components/editor/cell/useDeleteCell.tsx +10 -36
  268. package/src/components/editor/cell/useRunCells.ts +2 -5
  269. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  270. package/src/components/editor/cell/useSplitCell.tsx +10 -6
  271. package/src/components/editor/chrome/components/__tests__/feedback-button.test.tsx +1 -1
  272. package/src/components/editor/chrome/components/feedback-button.tsx +2 -8
  273. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +31 -2
  274. package/src/components/editor/chrome/panels/__tests__/runtime-details.test.tsx +1 -1
  275. package/src/components/editor/chrome/panels/__tests__/sandbox-recovery.test.tsx +1 -1
  276. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  277. package/src/components/editor/chrome/panels/dependency-graph-panel.tsx +1 -1
  278. package/src/components/editor/chrome/panels/logs-panel.tsx +2 -1
  279. package/src/components/editor/chrome/panels/outline/__tests__/outline-keyboard.test.tsx +84 -0
  280. package/src/components/editor/chrome/panels/outline/floating-outline.tsx +39 -14
  281. package/src/components/editor/chrome/panels/outline/useActiveOutline.tsx +3 -2
  282. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  283. package/src/components/editor/chrome/panels/secrets-panel.tsx +1 -0
  284. package/src/components/editor/chrome/panels/session-panel.tsx +1 -1
  285. package/src/components/editor/chrome/panels/snippets-panel.tsx +10 -12
  286. package/src/components/editor/chrome/panels/variable-panel.tsx +1 -5
  287. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +6 -9
  288. package/src/components/editor/chrome/wrapper/__tests__/document-sync-status.test.tsx +92 -0
  289. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +62 -3
  290. package/src/components/editor/chrome/wrapper/footer-items/attachment-status.tsx +4 -4
  291. package/src/components/editor/chrome/wrapper/footer-items/backend-status.tsx +1 -1
  292. package/src/components/editor/chrome/wrapper/footer-items/document-sync-status.tsx +54 -0
  293. package/src/components/editor/chrome/wrapper/footer-items/lsp-status.tsx +1 -1
  294. package/src/components/editor/chrome/wrapper/footer-items/machine-stats.tsx +1 -1
  295. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  296. package/src/components/editor/chrome/wrapper/footer.tsx +2 -2
  297. package/src/components/editor/chrome/wrapper/minimap-state.ts +1 -1
  298. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  299. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  300. package/src/components/editor/columns/__tests__/cell-column.test.tsx +1 -1
  301. package/src/components/editor/columns/cell-column.tsx +1 -1
  302. package/src/components/editor/columns/sortable-column.tsx +2 -2
  303. package/src/components/editor/connections/components.tsx +5 -10
  304. package/src/components/editor/controls/Controls.tsx +4 -4
  305. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  306. package/src/components/editor/controls/__tests__/keyboard-shortcuts.test.tsx +61 -0
  307. package/src/components/editor/controls/__tests__/notebook-menu-dropdown.test.tsx +10 -4
  308. package/src/components/editor/controls/command-palette.tsx +22 -10
  309. package/src/components/editor/controls/keyboard-shortcuts.tsx +14 -5
  310. package/src/components/editor/documentation.css +34 -11
  311. package/src/components/editor/dynamic-favicon.tsx +21 -40
  312. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +25 -20
  313. package/src/components/editor/errors/auto-fix.tsx +4 -8
  314. package/src/components/editor/errors/send-error-report-button.tsx +3 -6
  315. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  316. package/src/components/editor/file-tree/__tests__/file-explorer-navigation.test.tsx +47 -0
  317. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  318. package/src/components/editor/file-tree/file-explorer.tsx +48 -15
  319. package/src/components/editor/file-tree/file-name-input.tsx +1 -0
  320. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  321. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  322. package/src/components/editor/header/__tests__/filename-form.test.tsx +1 -4
  323. package/src/components/editor/header/__tests__/status.test.tsx +5 -3
  324. package/src/components/editor/header/app-header.tsx +1 -1
  325. package/src/components/editor/header/filename-form.tsx +1 -1
  326. package/src/components/editor/header/status.tsx +40 -20
  327. package/src/components/editor/links/__tests__/cell-link.test.tsx +148 -0
  328. package/src/components/editor/links/cell-link-list.tsx +1 -1
  329. package/src/components/editor/links/cell-link.tsx +17 -7
  330. package/src/components/editor/navigation/__tests__/clipboard.test.ts +60 -38
  331. package/src/components/editor/navigation/__tests__/navigation.test.ts +108 -85
  332. package/src/components/editor/navigation/__tests__/selection.test.ts +43 -21
  333. package/src/components/editor/navigation/clipboard.ts +17 -14
  334. package/src/components/editor/navigation/focus-utils.ts +23 -4
  335. package/src/components/editor/navigation/multi-cell-action-toolbar.tsx +17 -33
  336. package/src/components/editor/navigation/navigation.ts +71 -70
  337. package/src/components/editor/navigation/selection.ts +30 -29
  338. package/src/components/editor/notebook-banner.tsx +1 -0
  339. package/src/components/editor/notebook-cell.tsx +29 -32
  340. package/src/components/editor/output/MarimoErrorOutput.tsx +2 -1
  341. package/src/components/editor/output/MarimoTracebackOutput.tsx +19 -17
  342. package/src/components/editor/output/Outputs.css +8 -0
  343. package/src/components/editor/output/VideoOutput.tsx +6 -0
  344. package/src/components/editor/output/__tests__/VideoOutput.test.tsx +131 -0
  345. package/src/components/editor/output/__tests__/traceback.test.tsx +27 -1
  346. package/src/components/editor/output/ansi-reduce.ts +33 -20
  347. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  348. package/src/components/editor/output/console/__tests__/ConsoleOutput.test.tsx +125 -69
  349. package/src/components/editor/package-alert.tsx +14 -7
  350. package/src/components/editor/renderers/cell-array.tsx +22 -33
  351. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  352. package/src/components/editor/renderers/slides-layout/__tests__/plugin.test.ts +0 -1
  353. package/src/components/editor/renderers/slides-layout/slides-layout.tsx +1 -1
  354. package/src/components/editor/renderers/vertical-layout/__tests__/useFocusFirstEditor.test.ts +4 -4
  355. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  356. package/src/components/editor/renderers/vertical-layout/useFocusFirstEditor.ts +1 -1
  357. package/src/components/editor/renderers/vertical-layout/vertical-layout.tsx +1 -1
  358. package/src/components/editor/stdin-blocking-alert.tsx +1 -1
  359. package/src/components/home/components.tsx +17 -31
  360. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  361. package/src/components/icons/large-spinner.tsx +3 -14
  362. package/src/components/icons/spinner.tsx +1 -1
  363. package/src/components/markdown/markdown-renderer.tsx +4 -8
  364. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  365. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  366. package/src/components/pages/edit-page.tsx +19 -17
  367. package/src/components/pages/home-page.tsx +164 -91
  368. package/src/components/scratchpad/scratchpad.test.tsx +89 -0
  369. package/src/components/scratchpad/scratchpad.tsx +43 -32
  370. package/src/components/scratchpad/state.ts +7 -0
  371. package/src/components/slides/__tests__/minimap-actions.test.tsx +22 -29
  372. package/src/components/slides/__tests__/minimap.test.ts +38 -19
  373. package/src/components/slides/__tests__/slide-scroll-hint.test.tsx +37 -0
  374. package/src/components/slides/minimap.tsx +17 -12
  375. package/src/components/slides/reveal-component.tsx +58 -26
  376. package/src/components/slides/reveal-slides.css +5 -4
  377. package/src/components/slides/slide-cell-view.tsx +4 -5
  378. package/src/components/slides/slide-form.tsx +1 -0
  379. package/src/components/slides/slide-scroll-hint.tsx +3 -1
  380. package/src/components/sort/SortableCellsProvider.tsx +8 -25
  381. package/src/components/tracing/tracing-keyboard.test.tsx +85 -0
  382. package/src/components/tracing/tracing.tsx +46 -42
  383. package/src/components/ui/__tests__/draggable-popover.test.tsx +24 -0
  384. package/src/components/ui/__tests__/fullscreen.test.tsx +44 -2
  385. package/src/components/ui/__tests__/range-slider.test.tsx +165 -0
  386. package/src/components/ui/alert-dialog.tsx +1 -1
  387. package/src/components/ui/draggable-popover.tsx +24 -2
  388. package/src/components/ui/fullscreen.tsx +25 -1
  389. package/src/components/ui/range-slider.tsx +22 -9
  390. package/src/core/MarimoApp.tsx +3 -1
  391. package/src/core/ai/__tests__/staged-cells.test.ts +32 -33
  392. package/src/core/ai/context/providers/__tests__/cell-output.test.ts +8 -12
  393. package/src/core/ai/context/providers/cell-output.ts +2 -2
  394. package/src/core/ai/context/providers/datasource.ts +6 -2
  395. package/src/core/ai/staged-cells.ts +18 -19
  396. package/src/core/ai/tools/__tests__/edit-notebook-tool.test.ts +40 -88
  397. package/src/core/ai/tools/__tests__/registry.test.ts +17 -0
  398. package/src/core/ai/tools/base.ts +18 -15
  399. package/src/core/ai/tools/edit-notebook-tool.ts +86 -80
  400. package/src/core/ai/tools/registry.ts +13 -10
  401. package/src/core/ai/tools/sample-tool.ts +6 -7
  402. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -37
  403. package/src/core/cells/__tests__/actions.test.ts +624 -35
  404. package/src/core/cells/__tests__/add-missing-import.test.ts +7 -8
  405. package/src/core/cells/__tests__/arrange.test.ts +432 -0
  406. package/src/core/cells/__tests__/cell.test.ts +39 -0
  407. package/src/core/cells/__tests__/cells.test.ts +251 -2540
  408. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  409. package/src/core/cells/__tests__/document.test.ts +164 -0
  410. package/src/core/cells/__tests__/editor.test.ts +57 -0
  411. package/src/core/cells/__tests__/flush-document.test.ts +40 -0
  412. package/src/core/cells/__tests__/harness.ts +106 -0
  413. package/src/core/cells/__tests__/layout.test.ts +177 -0
  414. package/src/core/cells/__tests__/notebook-outline.test.ts +122 -0
  415. package/src/core/cells/__tests__/outputs.test.tsx +71 -53
  416. package/src/core/cells/__tests__/session.test.ts +445 -211
  417. package/src/core/cells/__tests__/utils.test.ts +9 -108
  418. package/src/core/cells/actions.ts +675 -6
  419. package/src/core/cells/add-missing-import.ts +47 -43
  420. package/src/core/cells/arrange.ts +529 -0
  421. package/src/core/cells/cell.ts +14 -3
  422. package/src/core/cells/cells.ts +126 -1363
  423. package/src/core/cells/columns.ts +38 -0
  424. package/src/core/cells/document.ts +178 -0
  425. package/src/core/cells/edit.ts +30 -0
  426. package/src/core/cells/editor.ts +82 -0
  427. package/src/core/cells/layout.ts +221 -0
  428. package/src/core/cells/outputs.ts +12 -28
  429. package/src/core/cells/pending-delete-service.ts +3 -7
  430. package/src/core/cells/session.ts +76 -53
  431. package/src/core/cells/types.ts +22 -6
  432. package/src/core/cells/utils.ts +10 -8
  433. package/src/core/codemirror/__tests__/format.test.ts +0 -34
  434. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  435. package/src/core/codemirror/ai/__tests__/resources.test.ts +117 -0
  436. package/src/core/codemirror/ai/resources.ts +1 -1
  437. package/src/core/codemirror/cells/__tests__/extensions.test.ts +2 -6
  438. package/src/core/codemirror/cells/extensions.ts +30 -28
  439. package/src/core/codemirror/cells/state.ts +1 -6
  440. package/src/core/codemirror/copilot/__tests__/getCodes.test.ts +6 -7
  441. package/src/core/codemirror/copilot/getCodes.ts +2 -2
  442. package/src/core/codemirror/format.ts +6 -23
  443. package/src/core/codemirror/keymaps/vim.ts +4 -8
  444. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  445. package/src/core/codemirror/language/__tests__/utils.test.ts +187 -15
  446. package/src/core/codemirror/language/cell-code.ts +114 -0
  447. package/src/core/codemirror/language/extension.ts +52 -35
  448. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  449. package/src/core/codemirror/language/languages/python.ts +16 -10
  450. package/src/core/codemirror/language/languages/sql/sql.ts +16 -24
  451. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  452. package/src/core/codemirror/language/types.ts +7 -7
  453. package/src/core/codemirror/language/utils.ts +9 -39
  454. package/src/core/codemirror/lsp/__tests__/notebook-lsp.test.ts +1 -2
  455. package/src/core/codemirror/lsp/notebook-lsp.ts +6 -5
  456. package/src/core/codemirror/lsp/utils.ts +1 -1
  457. package/src/core/codemirror/misc/__tests__/paste.test.ts +3 -1
  458. package/src/core/codemirror/misc/paste.ts +1 -1
  459. package/src/core/codemirror/readonly/__tests__/extension.test.ts +1 -1
  460. package/src/core/codemirror/readonly/extension.ts +2 -2
  461. package/src/core/codemirror/utils.ts +1 -0
  462. package/src/core/config/__tests__/app-config.test.ts +64 -0
  463. package/src/core/config/__tests__/config-schema.test.ts +56 -6
  464. package/src/core/config/config-schema.ts +17 -4
  465. package/src/core/config/config.ts +65 -20
  466. package/src/core/config/feature-flag.tsx +0 -2
  467. package/src/core/datasets/__tests__/data-source-discovery.test.ts +5 -5
  468. package/src/core/datasets/data-source-discovery.ts +4 -6
  469. package/src/core/documentation/doc-lookup.ts +1 -1
  470. package/src/core/edit-app.tsx +21 -28
  471. package/src/core/errors/error-entries.ts +4 -1
  472. package/src/core/export/hooks.ts +1 -1
  473. package/src/core/export/layout.ts +2 -2
  474. package/src/core/hotkeys/actions.ts +2 -4
  475. package/src/core/islands/__tests__/bootstrap.test.ts +69 -17
  476. package/src/core/islands/__tests__/bridge.test.ts +85 -27
  477. package/src/core/islands/bootstrap.ts +46 -53
  478. package/src/core/islands/bridge.ts +84 -25
  479. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  480. package/src/core/islands/components/web-components.tsx +1 -1
  481. package/src/core/islands/worker/__tests__/controller.test.ts +62 -4
  482. package/src/core/islands/worker/controller.ts +4 -10
  483. package/src/core/islands/worker/worker.tsx +33 -33
  484. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  485. package/src/core/kernel/__tests__/handlers.test.ts +7 -153
  486. package/src/core/kernel/handlers.ts +19 -72
  487. package/src/core/meta/__tests__/code-visibility.test.tsx +78 -70
  488. package/src/core/meta/code-visibility.ts +18 -9
  489. package/src/core/network/__tests__/connection.test.ts +1 -1
  490. package/src/core/network/__tests__/requests-lazy.test.ts +1 -3
  491. package/src/core/network/connection-notice.ts +1 -1
  492. package/src/core/network/connection.ts +1 -1
  493. package/src/core/network/requests-lazy.ts +2 -5
  494. package/src/core/network/requests-network.ts +5 -31
  495. package/src/core/network/requests-static.ts +1 -4
  496. package/src/core/network/requests-toasting.tsx +1 -4
  497. package/src/core/network/types.ts +6 -14
  498. package/src/core/network/useConnectionNotice.ts +8 -2
  499. package/src/core/notebook/__tests__/operations.test.ts +51 -0
  500. package/src/core/notebook/attachments.ts +38 -0
  501. package/src/core/notebook/document/__tests__/codemirror.test.ts +758 -0
  502. package/src/core/notebook/document/__tests__/draft.test.ts +363 -0
  503. package/src/core/notebook/document/__tests__/fixtures.ts +92 -0
  504. package/src/core/notebook/document/__tests__/hooks.test.tsx +391 -0
  505. package/src/core/notebook/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  506. package/src/core/notebook/document/__tests__/model.test.ts +247 -0
  507. package/src/core/notebook/document/__tests__/protocol-fixtures.ts +38 -0
  508. package/src/core/notebook/document/__tests__/recovery-scenarios.test.ts +230 -0
  509. package/src/core/notebook/document/__tests__/scenario-fixture.ts +43 -0
  510. package/src/core/notebook/document/__tests__/scenarios.test.ts +205 -0
  511. package/src/core/notebook/document/__tests__/state.test.ts +459 -0
  512. package/src/core/notebook/document/__tests__/submission.test.ts +203 -0
  513. package/src/core/notebook/document/__tests__/test-host.test.ts +92 -0
  514. package/src/core/notebook/document/__tests__/test-host.ts +247 -0
  515. package/src/core/notebook/document/__tests__/test-notebook.ts +149 -0
  516. package/src/core/notebook/document/__tests__/user.ts +61 -0
  517. package/src/core/notebook/document/codemirror.ts +216 -0
  518. package/src/core/notebook/document/draft.ts +502 -0
  519. package/src/core/notebook/document/hooks.tsx +118 -0
  520. package/src/core/notebook/document/index.ts +24 -0
  521. package/src/core/notebook/document/model.ts +102 -0
  522. package/src/core/notebook/document/select.ts +41 -0
  523. package/src/core/notebook/document/state.ts +378 -0
  524. package/src/core/notebook/document/sync.ts +369 -0
  525. package/src/core/notebook/index.ts +19 -0
  526. package/src/core/notebook/operations.ts +30 -0
  527. package/src/core/notebook/replica/AGENTS.md +11 -0
  528. package/src/core/{host → notebook/replica}/__tests__/attachments.test.ts +9 -10
  529. package/src/core/{host → notebook/replica}/__tests__/document.test.ts +18 -16
  530. package/src/core/notebook/replica/__tests__/executions.test.ts +201 -0
  531. package/src/core/notebook/replica/__tests__/fixtures.ts +79 -0
  532. package/src/core/notebook/replica/__tests__/replica.test.ts +559 -0
  533. package/src/core/notebook/replica/__tests__/results.test.ts +193 -0
  534. package/src/core/notebook/replica/__tests__/runtime.test.ts +78 -0
  535. package/src/core/notebook/replica/backend.ts +19 -0
  536. package/src/core/notebook/replica/connection.ts +42 -0
  537. package/src/core/notebook/replica/index.ts +20 -0
  538. package/src/core/notebook/replica/replica.ts +336 -0
  539. package/src/core/{host → notebook/replica}/resources/attachments.ts +6 -3
  540. package/src/core/{host → notebook/replica}/resources/document.ts +3 -3
  541. package/src/core/notebook/replica/resources/executions.ts +157 -0
  542. package/src/core/notebook/replica/resources/results.ts +48 -0
  543. package/src/core/notebook/replica/resources/runtime.ts +14 -0
  544. package/src/core/packages/__tests__/useSandboxController.test.tsx +1 -1
  545. package/src/core/packages/useSandboxController.ts +1 -1
  546. package/src/core/run-app.tsx +48 -51
  547. package/src/core/runtime/__tests__/adapter.test.ts +1 -1
  548. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  549. package/src/core/runtime/__tests__/runtime.test.ts +32 -74
  550. package/src/core/runtime/adapter.ts +1 -1
  551. package/src/core/runtime/config.ts +2 -2
  552. package/src/core/runtime/runtime.ts +33 -46
  553. package/src/core/runtime/types.ts +0 -2
  554. package/src/core/saving/__tests__/document.test.tsx +123 -0
  555. package/src/core/saving/__tests__/state.test.ts +111 -0
  556. package/src/core/saving/__tests__/useAutoSave.test.tsx +67 -0
  557. package/src/core/saving/document.ts +33 -0
  558. package/src/core/saving/filename.ts +1 -1
  559. package/src/core/saving/save-component.tsx +39 -60
  560. package/src/core/saving/state.ts +38 -39
  561. package/src/core/saving/useAutoSave.ts +17 -53
  562. package/src/core/session/__tests__/backend.test.ts +236 -0
  563. package/src/core/session/__tests__/useNotebookReplica.hook.test.tsx +184 -0
  564. package/src/core/{websocket → session}/__tests__/useNotebookSession.hook.test.tsx +213 -262
  565. package/src/core/session/active-notebook.ts +28 -0
  566. package/src/core/session/backend.ts +123 -0
  567. package/src/core/session/embedded.ts +7 -0
  568. package/src/core/{websocket → session}/types.ts +2 -2
  569. package/src/core/session/useNotebookReplica.ts +74 -0
  570. package/src/core/{websocket → session}/useNotebookSession.tsx +77 -64
  571. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  572. package/src/core/state/jotai.ts +1 -1
  573. package/src/core/static/__tests__/files.test.ts +38 -10
  574. package/src/core/static/files.ts +2 -0
  575. package/src/core/{websocket → transports}/__tests__/connection-state.test.ts +1 -13
  576. package/src/core/transports/__tests__/server.test.ts +408 -0
  577. package/src/core/transports/__tests__/static.test.ts +306 -0
  578. package/src/core/transports/__tests__/wasm.test.ts +166 -0
  579. package/src/core/{websocket → transports}/connection-state.ts +9 -16
  580. package/src/core/transports/server.ts +152 -0
  581. package/src/core/transports/static.ts +166 -0
  582. package/src/core/transports/wasm.ts +97 -0
  583. package/src/core/vscode/__tests__/vscode-bindings.test.ts +125 -0
  584. package/src/core/vscode/vscode-bindings.ts +14 -2
  585. package/src/core/wasm/__tests__/bridge.test.ts +105 -65
  586. package/src/core/wasm/__tests__/worker.test.ts +187 -0
  587. package/src/core/wasm/bridge.ts +69 -72
  588. package/src/core/wasm/rpc.ts +15 -3
  589. package/src/core/wasm/store.ts +3 -3
  590. package/src/core/wasm/worker/bootstrap.ts +55 -51
  591. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  592. package/src/core/wasm/worker/save-worker.ts +9 -10
  593. package/src/core/wasm/worker/types.ts +29 -18
  594. package/src/core/wasm/worker/worker.ts +89 -45
  595. package/src/css/app/codemirror.css +1 -1
  596. package/src/css/common.css +1 -1
  597. package/src/css/globals.css +66 -22
  598. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  599. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  600. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  601. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  602. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  603. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  604. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  605. package/src/hooks/useAsyncData.ts +3 -3
  606. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  607. package/src/hooks/useDebounce.ts +6 -10
  608. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  609. package/src/hooks/useDelayElapsed.ts +14 -10
  610. package/src/hooks/useElapsedTime.ts +3 -3
  611. package/src/hooks/useEventListener.ts +1 -2
  612. package/src/hooks/useInputHistory.ts +4 -7
  613. package/src/hooks/useInternalStateWithSync.ts +6 -6
  614. package/src/hooks/useResizeObserver.ts +4 -7
  615. package/src/hooks/useScript.ts +67 -70
  616. package/src/hooks/useStateForKey.ts +93 -0
  617. package/src/mount.tsx +20 -15
  618. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  619. package/src/plugins/core/registerReactComponent.tsx +1 -1
  620. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  621. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  622. package/src/plugins/impl/DataTablePlugin.tsx +69 -37
  623. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  624. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  625. package/src/plugins/impl/FormPlugin.tsx +1 -1
  626. package/src/plugins/impl/RangeSliderPlugin.tsx +16 -2
  627. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  628. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  629. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +387 -1
  630. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  631. package/src/plugins/impl/__tests__/RangeSliderPlugin.test.tsx +109 -0
  632. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  633. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +153 -0
  634. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  635. package/src/plugins/impl/anywidget/model.ts +3 -1
  636. package/src/plugins/impl/anywidget/registry.ts +18 -14
  637. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  638. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  639. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  640. package/src/plugins/impl/chat/chat-ui.tsx +14 -11
  641. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  642. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  643. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  644. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  645. package/src/plugins/impl/data-explorer/components/column-summary.tsx +4 -2
  646. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  647. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  648. package/src/plugins/impl/data-frames/__tests__/sidebar.test.tsx +160 -0
  649. package/src/plugins/impl/data-frames/forms/__tests__/__snapshots__/form.test.tsx.snap +66 -66
  650. package/src/plugins/impl/data-frames/panel.tsx +44 -16
  651. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  652. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  653. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  654. package/src/plugins/impl/panel/utils.ts +17 -0
  655. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  656. package/src/plugins/impl/plotly/selection.ts +25 -19
  657. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  658. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  659. package/src/plugins/layout/DownloadPlugin.tsx +7 -11
  660. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  661. package/src/plugins/layout/__test__/DownloadPlugin.test.tsx +196 -0
  662. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  663. package/src/stories/button.stories.tsx +1 -0
  664. package/src/stories/cell.stories.tsx +2 -5
  665. package/src/stories/data-table.stories.tsx +31 -0
  666. package/src/stories/editor.stories.tsx +24 -17
  667. package/src/stories/form-wrapper.stories.tsx +2 -2
  668. package/src/stories/layout/vertical/one-column.stories.tsx +28 -23
  669. package/src/stories/runtime-story.tsx +23 -17
  670. package/src/stories/select.stories.tsx +2 -2
  671. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  672. package/src/utils/__tests__/arrays.test.ts +35 -0
  673. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  674. package/src/utils/__tests__/{createReducer.test.ts → createReducer.test.tsx} +29 -0
  675. package/src/utils/__tests__/download.test.tsx +62 -9
  676. package/src/utils/__tests__/numbers.test.ts +105 -1
  677. package/src/utils/__tests__/objects.test.ts +21 -0
  678. package/src/utils/arrays.ts +11 -2
  679. package/src/utils/blob.ts +1 -1
  680. package/src/utils/createReducer.ts +53 -26
  681. package/src/utils/download.ts +55 -3
  682. package/src/utils/maps.ts +0 -3
  683. package/src/utils/numbers.ts +79 -18
  684. package/src/utils/objects.ts +4 -9
  685. package/src/utils/tracer.ts +9 -5
  686. package/src/utils/zod-utils.ts +1 -3
  687. package/dist/_baseMerge-BiGRwuqv.js +0 -165
  688. package/dist/any-language-editor-D4q-IFcM.js +0 -76
  689. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  690. package/dist/assets/worker-HPY9fbhz.js +0 -114
  691. package/dist/code-visibility-HTfE7PV1.js +0 -138
  692. package/dist/dropdown-menu-DHP9P4VE.js +0 -4678
  693. package/dist/feedback-button-BL7Ed1J3.js +0 -379
  694. package/dist/hasIn-o2Us0uVx.js +0 -132
  695. package/dist/html-to-image-BdA59IUD.js +0 -3251
  696. package/dist/isArrayLikeObject-BZl-15ir.js +0 -450
  697. package/dist/isEmpty-BSZ1LaUo.js +0 -35
  698. package/dist/isSymbol-kQcfYbNX.js +0 -8
  699. package/dist/label-DoysmLMW.js +0 -281
  700. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  701. package/dist/objectWithoutPropertiesLoose-BRDcTsfB.js +0 -23
  702. package/dist/pick-CJeoxYOS.js +0 -51
  703. package/dist/precisionRound-DMIVQrt2.js +0 -97
  704. package/dist/select-DemoKT4G.js +0 -1344
  705. package/dist/toDate-B5mIgxwZ.js +0 -1185
  706. package/dist/toast-PkdZraM8.js +0 -1830
  707. package/dist/use-restore-focus-CKjfFjLM.js +0 -6369
  708. package/dist/useLifecycle-BGWzTwHK.js +0 -169
  709. package/dist/useTheme-BhQvec43.js +0 -904
  710. package/src/components/editor/cell/code/__tests__/rtc-editor-config.test.ts +0 -50
  711. package/src/components/editor/chrome/wrapper/footer-items/rtc-status.tsx +0 -65
  712. package/src/components/editor/navigation/__tests__/state.test.ts +0 -81
  713. package/src/components/editor/navigation/state.ts +0 -65
  714. package/src/core/ai/tools/__tests__/run-cells-tool.test.ts +0 -656
  715. package/src/core/ai/tools/run-cells-tool.ts +0 -303
  716. package/src/core/attachments/state.ts +0 -18
  717. package/src/core/cells/__tests__/apply-transaction.test.ts +0 -991
  718. package/src/core/cells/__tests__/document-changes.test.ts +0 -1140
  719. package/src/core/cells/__tests__/document-roundtrip.test.ts +0 -392
  720. package/src/core/cells/document-changes.ts +0 -944
  721. package/src/core/codemirror/rtc/extension.ts +0 -453
  722. package/src/core/codemirror/rtc/loro/__tests__/sync.test.ts +0 -52
  723. package/src/core/codemirror/rtc/loro/awareness.ts +0 -531
  724. package/src/core/codemirror/rtc/loro/colors.ts +0 -38
  725. package/src/core/codemirror/rtc/loro/sync.ts +0 -168
  726. package/src/core/host/AGENTS.md +0 -11
  727. package/src/core/host/__tests__/events.test.ts +0 -46
  728. package/src/core/host/__tests__/fixtures.ts +0 -39
  729. package/src/core/host/__tests__/notebook-client.test.ts +0 -433
  730. package/src/core/host/__tests__/protocol.test.ts +0 -214
  731. package/src/core/host/__tests__/stream.test.ts +0 -209
  732. package/src/core/host/connection.ts +0 -81
  733. package/src/core/host/events.ts +0 -61
  734. package/src/core/host/notebook-client.ts +0 -290
  735. package/src/core/host/protocol.ts +0 -1943
  736. package/src/core/host/resources/resource.ts +0 -47
  737. package/src/core/host/stream.ts +0 -211
  738. package/src/core/rtc/state.ts +0 -34
  739. package/src/core/state/__tests__/jotai.test.ts +0 -25
  740. package/src/core/wasm/worker/message-buffer.ts +0 -33
  741. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  742. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +0 -170
  743. package/src/core/websocket/state.ts +0 -7
  744. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  745. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  746. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  747. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  748. package/src/core/websocket/transports/basic.ts +0 -102
  749. package/src/core/websocket/transports/create.ts +0 -31
  750. package/src/core/websocket/transports/sse.ts +0 -308
  751. package/src/core/websocket/transports/transport.ts +0 -50
  752. package/src/core/websocket/transports/ws.ts +0 -98
  753. package/src/core/websocket/useNotebookClient.ts +0 -88
  754. package/src/utils/__tests__/id-tree.test.ts +0 -2145
  755. package/src/utils/id-tree.tsx +0 -1134
  756. /package/src/{core/attachments/toast.ts → components/editor/agent-toast.ts} +0 -0
  757. /package/src/core/{websocket → session}/connection-utils.ts +0 -0
@@ -77,6 +77,23 @@ describe("useMessageQueue", () => {
77
77
  expect(result.current.messages).toEqual([]);
78
78
  });
79
79
 
80
+ it("preserves multiple enqueues and flushes before React renders", () => {
81
+ const { result } = renderHook(() => useMessageQueue());
82
+ const send = vi.fn();
83
+
84
+ act(() => {
85
+ result.current.enqueue([textPart("first")]);
86
+ result.current.enqueue([textPart("second")]);
87
+ result.current.flushNext(send);
88
+ result.current.flushNext(send);
89
+ });
90
+
91
+ expect(send).toHaveBeenNthCalledWith(1, [textPart("first")]);
92
+ expect(send).toHaveBeenNthCalledWith(2, [textPart("second")]);
93
+ expect(result.current.messages).toEqual([]);
94
+ expect(result.current.hasQueuedRef.current).toBe(false);
95
+ });
96
+
80
97
  it("flushNext on an empty queue is a no-op", () => {
81
98
  const { result } = renderHook(() => useMessageQueue());
82
99
  const send = vi.fn();
@@ -0,0 +1,64 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { fireEvent, render, screen } from "@testing-library/react";
4
+ import { describe, expect, it, vi } from "vitest";
5
+ import type { CopilotMode } from "@/core/config/config-schema";
6
+ import { NotebookAccessSelector } from "../notebook-access-selector";
7
+
8
+ Element.prototype.scrollIntoView = vi.fn();
9
+
10
+ describe("NotebookAccessSelector", () => {
11
+ it("shows legacy Manual as Read only and offers exactly three levels", async () => {
12
+ const onValueChange = vi.fn();
13
+ render(
14
+ <NotebookAccessSelector value="manual" onValueChange={onValueChange} />,
15
+ );
16
+
17
+ fireEvent.keyDown(
18
+ screen.getByRole("combobox", { name: "Notebook access: Read only" }),
19
+ { key: "ArrowDown" },
20
+ );
21
+ const options = await screen.findAllByRole("option");
22
+ expect(options.map((option) => option.getAttribute("data-state"))).toEqual([
23
+ "checked",
24
+ "unchecked",
25
+ "unchecked",
26
+ ]);
27
+ expect(options.map((option) => option.textContent)).toEqual([
28
+ "Read onlyRead notebook code, outputs, and variable previews.",
29
+ "Suggest editsRead the notebook and propose cell changes for review.",
30
+ "Full accessRead, edit, and run cells, with Python scratchpad access.",
31
+ ]);
32
+ expect(onValueChange).not.toHaveBeenCalled();
33
+ });
34
+
35
+ it.each<{ mode: CopilotMode; label: string }>([
36
+ { mode: "ask", label: "Read only" },
37
+ { mode: "agent", label: "Suggest edits" },
38
+ { mode: "code_mode", label: "Full access" },
39
+ ])("maps $label to the existing $mode setting", async ({ mode, label }) => {
40
+ const onValueChange = vi.fn();
41
+ const { rerender } = render(
42
+ <NotebookAccessSelector value={mode} onValueChange={onValueChange} />,
43
+ );
44
+
45
+ expect(
46
+ screen.getByRole("combobox", { name: `Notebook access: ${label}` }),
47
+ ).toHaveTextContent(label);
48
+ expect(onValueChange).not.toHaveBeenCalled();
49
+
50
+ rerender(
51
+ <NotebookAccessSelector
52
+ value={mode === "ask" ? "agent" : "ask"}
53
+ onValueChange={onValueChange}
54
+ />,
55
+ );
56
+ fireEvent.keyDown(screen.getByRole("combobox"), { key: "ArrowDown" });
57
+ const option = await screen.findByRole("option", {
58
+ name: new RegExp(`^${label} `),
59
+ });
60
+ fireEvent.keyDown(option, { key: "Enter" });
61
+
62
+ expect(onValueChange).toHaveBeenCalledExactlyOnceWith(mode);
63
+ });
64
+ });
@@ -0,0 +1,43 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import { fireEvent, render, screen } from "@testing-library/react";
4
+ import { createStore, Provider } from "jotai";
5
+ import { expect, it, vi } from "vitest";
6
+ import { SessionTabs } from "../session-tabs";
7
+ import { addSession, agentSessionStateAtom } from "../state";
8
+
9
+ vi.mock("../agent-selector", () => ({ AgentSelector: () => null }));
10
+
11
+ it("keeps session selection and close as separate named buttons", () => {
12
+ const first = addSession(
13
+ { sessions: [], activeTabId: null },
14
+ { agentId: "claude", firstMessage: "First" },
15
+ ).sessions[0];
16
+ const second = addSession(
17
+ { sessions: [], activeTabId: null },
18
+ { agentId: "codex", firstMessage: "Second" },
19
+ ).sessions[0];
20
+ const store = createStore();
21
+ store.set(agentSessionStateAtom, {
22
+ sessions: [first, second],
23
+ activeTabId: first.tabId,
24
+ });
25
+ render(
26
+ <Provider store={store}>
27
+ <SessionTabs />
28
+ </Provider>,
29
+ );
30
+ const select = screen.getByRole("button", {
31
+ name: /Second$/,
32
+ pressed: false,
33
+ });
34
+ select.focus();
35
+ expect(select).toHaveFocus();
36
+ fireEvent.click(select);
37
+ expect(select).toHaveAttribute("aria-pressed", "true");
38
+ fireEvent.click(screen.getByRole("button", { name: "Close First" }));
39
+ expect(store.get(agentSessionStateAtom).activeTabId).toBe(second.tabId);
40
+ expect(
41
+ store.get(agentSessionStateAtom).sessions.map((session) => session.title),
42
+ ).toEqual(["Second"]);
43
+ });
@@ -74,7 +74,7 @@ describe("state utility functions", () => {
74
74
  const newState = addSession(initialState, session);
75
75
 
76
76
  // Remove the dynamic tabId for snapshot comparison
77
- const { tabId, ...sessionWithoutId } = newState.sessions[0];
77
+ const { tabId: _tabId, ...sessionWithoutId } = newState.sessions[0];
78
78
  expect({
79
79
  ...newState,
80
80
  sessions: [sessionWithoutId],
@@ -116,7 +116,7 @@ describe("state utility functions", () => {
116
116
  const newState = addSession(initialState, newSession);
117
117
 
118
118
  // Remove dynamic tabId for snapshot
119
- const { tabId, ...sessionWithoutId } = newState.sessions[0];
119
+ const { tabId: _tabId, ...sessionWithoutId } = newState.sessions[0];
120
120
  expect({
121
121
  ...newState,
122
122
  sessions: [sessionWithoutId],
@@ -328,15 +328,16 @@ export const PlansBlock = (props: { data: PlanNotificationEvent[] }) => {
328
328
  </span>
329
329
  </div>
330
330
  <ul className="flex flex-col gap-1">
331
- {plans.map((item, index) => (
331
+ {plans.map((item) => (
332
332
  <li
333
- key={`${item.status}-${index}`}
333
+ key={item.content}
334
334
  className={`
335
335
  flex items-center gap-2 px-2 py-1 rounded
336
336
  `}
337
337
  >
338
338
  <input
339
339
  type="checkbox"
340
+ aria-label={item.content}
340
341
  checked={item.status === "completed"}
341
342
  readOnly={true}
342
343
  className="accent-primary h-4 w-4 rounded border border-muted-foreground/30"
@@ -32,17 +32,22 @@ const SessionTab: React.FC<SessionTabProps> = memo(
32
32
  "flex items-center gap-1 px-2 py-1 text-xs border-r border-border bg-muted/30 hover:bg-muted/50 cursor-pointer min-w-0",
33
33
  isActive && "bg-background border-b-0 relative z-1",
34
34
  )}
35
- onClick={() => onSelect(session.tabId)}
36
35
  >
37
- <div className="flex items-center gap-1 min-w-0 flex-1">
36
+ <button
37
+ type="button"
38
+ aria-pressed={isActive}
39
+ onClick={() => onSelect(session.tabId)}
40
+ className="flex items-center gap-1 min-w-0 flex-1 text-left focus-visible:outline focus-visible:outline-ring"
41
+ >
38
42
  <span className="text-muted-foreground text-[10px] font-medium">
39
43
  <AiProviderIcon provider={session.agentId} className="h-3 w-3" />
40
44
  </span>
41
45
  <span className="truncate" title={session.title}>
42
46
  {session.title}
43
47
  </span>
44
- </div>
48
+ </button>
45
49
  <Button
50
+ aria-label={`Close ${session.title}`}
46
51
  variant="ghost"
47
52
  size="sm"
48
53
  className="h-4 w-4 p-0 hover:bg-destructive/20 hover:text-destructive shrink-0"
@@ -9,7 +9,6 @@ import {
9
9
  PopoverContent,
10
10
  PopoverTrigger,
11
11
  } from "@/components/ui/popover";
12
- import { ExternalLink } from "@/components/ui/links";
13
12
  import { Switch } from "@/components/ui/switch";
14
13
  import { chatOptionsAtom } from "@/core/ai/state";
15
14
  import { cn } from "@/utils/cn";
@@ -46,10 +45,7 @@ export const CapabilitiesPopover: React.FC = () => {
46
45
  <div className="flex flex-col gap-0.5">
47
46
  <span className="text-sm font-semibold">Web search</span>
48
47
  <span className="text-xs text-muted-foreground">
49
- Search the web when the model supports it.{" "}
50
- <ExternalLink href="https://docs.marimo.io/guides/editor_features/tools#web-search-and-fetch">
51
- Learn more
52
- </ExternalLink>
48
+ Search the web when the model supports it.
53
49
  </span>
54
50
  </div>
55
51
  <Switch
@@ -124,6 +124,7 @@ export const AddContextButton = ({
124
124
  return (
125
125
  <Tooltip content="Add context">
126
126
  <Button
127
+ aria-label="Add context"
127
128
  variant="text"
128
129
  size="icon"
129
130
  onClick={handleAddContext}
@@ -148,6 +149,7 @@ export const AttachFileButton = ({
148
149
  <>
149
150
  <Tooltip content="Attach a file">
150
151
  <Button
152
+ aria-label="Attach a file"
151
153
  variant="text"
152
154
  size="icon"
153
155
  onClick={() => fileInputRef.current?.click()}
@@ -13,28 +13,15 @@ import {
13
13
  import { useAtom, useAtomValue, useSetAtom, useStore } from "jotai";
14
14
  import {
15
15
  BotMessageSquareIcon,
16
- HatGlasses,
17
16
  ArrowRightIcon,
18
17
  Loader2,
19
- type LucideIcon,
20
- MessageCircleIcon,
21
- NotebookText,
22
18
  PlusIcon,
23
19
  SparklesIcon,
24
20
  SettingsIcon,
25
- CodeIcon,
26
21
  } from "lucide-react";
27
22
  import { memo, useEffect, useRef, useState } from "react";
28
23
  import useEvent from "react-use-event-hook";
29
24
  import { Button } from "@/components/ui/button";
30
- import {
31
- Select,
32
- SelectContent,
33
- SelectGroup,
34
- SelectItem,
35
- SelectLabel,
36
- SelectTrigger,
37
- } from "@/components/ui/select";
38
25
  import { replaceMessagesInChat } from "@/core/ai/chat-utils";
39
26
  import { useAIConfigActions } from "@/core/ai/config";
40
27
  import { AI_SDK_UI_THROTTLE_MS } from "@/core/ai/constants";
@@ -49,11 +36,8 @@ import {
49
36
  pendingAiPromptAtom,
50
37
  } from "@/core/ai/state";
51
38
  import type { ToolNotebookContext } from "@/core/ai/tools/base";
52
- import {
53
- type CopilotMode,
54
- FRONTEND_TOOL_REGISTRY,
55
- } from "@/core/ai/tools/registry";
56
- import { useCellActions } from "@/core/cells/cells";
39
+ import { FRONTEND_TOOL_REGISTRY } from "@/core/ai/tools/registry";
40
+ import { useCellActions } from "@/core/cells/actions";
57
41
  import { aiAtom, aiModelConfiguredAtom } from "@/core/config/config";
58
42
  import { DEFAULT_AI_MODEL } from "@/core/config/config-schema";
59
43
  import { useRequestClient } from "@/core/network/requests";
@@ -97,6 +81,7 @@ import {
97
81
  import { renderUIMessage } from "./chat-display";
98
82
  import { ChatHistoryPopover } from "./chat-history-popover";
99
83
  import { CapabilitiesPopover } from "./capabilities-popover";
84
+ import { NotebookAccessSelector } from "./notebook-access-selector";
100
85
  import {
101
86
  type ChatMessagePart,
102
87
  convertToFileUIPart,
@@ -117,7 +102,7 @@ import ScrollToBottomButton from "./acp/scroll-to-bottom-button";
117
102
  import { useChatControllerId } from "./use-chat-controller-id";
118
103
 
119
104
  // Default mode for the AI
120
- const DEFAULT_MODE = "manual";
105
+ const DEFAULT_MODE = "ask";
121
106
 
122
107
  const QueuedMessageDisplay = memo(
123
108
  ({ message }: { message: QueuedUserMessage }) => {
@@ -166,13 +151,18 @@ const ChatHeader: React.FC<ChatHeaderProps> = ({
166
151
  return (
167
152
  <div className="flex border-b px-2 py-1 justify-between shrink-0 items-center">
168
153
  <Tooltip content="New chat">
169
- <Button variant="text" size="icon" onClick={onNewChat}>
154
+ <Button
155
+ aria-label="New chat"
156
+ variant="text"
157
+ size="icon"
158
+ onClick={onNewChat}
159
+ >
170
160
  <PlusIcon className="h-4 w-4" />
171
161
  </Button>
172
162
  </Tooltip>
173
163
  <div className="flex items-center gap-2">
174
- <Tooltip content="Connect your agent">
175
- <PairWithAgentButton className="mr-1" label="Connect" />
164
+ <Tooltip content="Pair your agent">
165
+ <PairWithAgentButton className="mr-1" label="Pair" />
176
166
  </Tooltip>
177
167
  <MCPStatusIndicator />
178
168
  <Tooltip content="AI Settings">
@@ -316,82 +306,17 @@ const ChatInputFooter: React.FC<ChatInputFooterProps> = memo(
316
306
 
317
307
  const { saveModeChange } = useAIConfigActions();
318
308
 
319
- const modeOptions: {
320
- value: CopilotMode;
321
- label: string;
322
- subtitle: string;
323
- Icon: LucideIcon;
324
- }[] = [
325
- {
326
- value: "manual",
327
- label: "Manual",
328
- subtitle: "Pure chat, no tool usage",
329
- Icon: MessageCircleIcon,
330
- },
331
- {
332
- value: "ask",
333
- label: "Ask",
334
- subtitle: "AI with access to read-only tools like documentation search",
335
- Icon: NotebookText,
336
- },
337
- {
338
- value: "agent",
339
- label: "Agent",
340
- subtitle: "AI with access to read and write tools",
341
- Icon: HatGlasses,
342
- },
343
- {
344
- value: "code_mode",
345
- label: "Code Mode (beta)",
346
- subtitle:
347
- "AI with access to the notebook's kernel. Potentially destructive.",
348
- Icon: CodeIcon,
349
- },
350
- ];
351
-
352
309
  const isAttachmentSupported =
353
310
  PROVIDERS_THAT_SUPPORT_ATTACHMENTS.has(currentProvider);
354
311
 
355
- const currentModeOption = modeOptions.find((o) => o.value === currentMode);
356
- const CurrentModeIcon = currentModeOption?.Icon;
357
- const CurrentModeLabel = currentModeOption?.label;
358
-
359
312
  return (
360
313
  <TooltipProvider>
361
314
  <div className="px-3 py-2 border-t border-border/20 flex flex-wrap items-center gap-1">
362
315
  <div className="flex flex-wrap items-center gap-1">
363
- <Select value={currentMode} onValueChange={saveModeChange}>
364
- <SelectTrigger className="h-6 text-xs border-border shadow-none! ring-0! bg-muted hover:bg-muted/30 py-0 px-2 gap-1.5 shrink-0">
365
- {CurrentModeIcon && <CurrentModeIcon className="h-3 w-3" />}
366
- <span className="text-nowrap">{CurrentModeLabel}</span>
367
- </SelectTrigger>
368
- <SelectContent>
369
- <SelectGroup>
370
- <SelectLabel className="text-xs uppercase tracking-wider text-muted-foreground/70 font-medium">
371
- AI Mode
372
- </SelectLabel>
373
- {modeOptions.map((option) => (
374
- <SelectItem
375
- key={option.value}
376
- value={option.value}
377
- className="text-xs py-1"
378
- >
379
- <div className="flex items-start gap-2.5">
380
- <span className="mt-1 text-muted-foreground">
381
- <option.Icon className="h-3 w-3" />
382
- </span>
383
- <div className="flex flex-col gap-0.5">
384
- <span className="font-semibold">{option.label}</span>
385
- <span className="text-muted-foreground">
386
- {option.subtitle}
387
- </span>
388
- </div>
389
- </div>
390
- </SelectItem>
391
- ))}
392
- </SelectGroup>
393
- </SelectContent>
394
- </Select>
316
+ <NotebookAccessSelector
317
+ value={currentMode}
318
+ onValueChange={saveModeChange}
319
+ />
395
320
  <AIModelDropdown
396
321
  placeholder="Model"
397
322
  triggerClassName="h-6 text-xs shadow-none! ring-0! bg-muted hover:bg-muted/30 rounded-sm max-w-[200px]"
@@ -506,7 +431,7 @@ const PairWithAgentCallout: React.FC<{
506
431
  onClick={onPairWithAgent}
507
432
  >
508
433
  <SparklesIcon className="h-3.5 w-3.5 shrink-0" />
509
- <span>Connect your own agent to this notebook</span>
434
+ <span>Pair your own agent with this notebook</span>
510
435
  <ArrowRightIcon className="h-3 w-3 shrink-0" />
511
436
  </Button>
512
437
  );
@@ -568,7 +493,7 @@ const ChatPanelBody = () => {
568
493
  setAbortNotice(null);
569
494
  });
570
495
  const runtimeManager = useRuntimeManager();
571
- const { invokeAiTool, sendRun } = useRequestClient();
496
+ const { invokeAiTool } = useRequestClient();
572
497
  const { openModal, closeModal } = useImperativeModal();
573
498
 
574
499
  const activeChatId = activeChat?.id;
@@ -577,12 +502,10 @@ const ChatPanelBody = () => {
577
502
  const store = useStore();
578
503
 
579
504
  const { addStagedCell } = useStagedAICellsActions();
580
- const { createNewCell, prepareForRun } = useCellActions();
505
+ const { createCell } = useCellActions();
581
506
  const toolContext: ToolNotebookContext = {
582
507
  addStagedCell,
583
- createNewCell,
584
- prepareForRun,
585
- sendRun,
508
+ createCell,
586
509
  store,
587
510
  };
588
511
 
@@ -716,14 +639,10 @@ const ChatPanelBody = () => {
716
639
  );
717
640
 
718
641
  const isLoading = status === "submitted" || status === "streaming";
719
- // Read via a ref so the queue-vs-send decision stays correct even when it is
720
- // made after an `await` (e.g. resolving @-context), by which point the render
721
- // closure's `isLoading` may be stale.
722
- const isLoadingRef = useRef(isLoading);
723
- isLoadingRef.current = isLoading;
724
-
642
+ // useEvent reads the latest loading state, even when called after an await
643
+ // (e.g. resolving @-context) that outlives the original render's closure.
725
644
  const submitOrQueue = useEvent((parts: ChatMessagePart[]) => {
726
- if (isLoadingRef.current || hasQueuedRef.current) {
645
+ if (isLoading || hasQueuedRef.current) {
727
646
  enqueueUserMessage(parts);
728
647
  } else {
729
648
  sendUserMessage(parts);
@@ -294,16 +294,12 @@ export function useMessageQueue() {
294
294
  // invoked from a callback (e.g. `onFinish`) in the same tick as `enqueue`.
295
295
  const messagesRef = useRef<QueuedUserMessage[]>([]);
296
296
  const hasQueuedRef = useRef(false);
297
- messagesRef.current = messages;
298
- hasQueuedRef.current = messages.length > 0;
299
297
 
300
298
  const enqueue = useEvent((parts: ChatMessagePart[]) => {
301
- setMessages((prev) => {
302
- const next = [...prev, { id: generateUUID(), parts }];
303
- messagesRef.current = next;
304
- hasQueuedRef.current = true;
305
- return next;
306
- });
299
+ const next = [...messagesRef.current, { id: generateUUID(), parts }];
300
+ messagesRef.current = next;
301
+ hasQueuedRef.current = true;
302
+ setMessages(next);
307
303
  });
308
304
 
309
305
  const flushNext = useEvent((send: (parts: ChatMessagePart[]) => void) => {
@@ -0,0 +1,22 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import type { Meta, StoryObj } from "@storybook/react-vite";
4
+ import { useState } from "react";
5
+ import { NotebookAccessSelector } from "./notebook-access-selector";
6
+
7
+ const meta = {
8
+ title: "Chat/Notebook access",
9
+ component: NotebookAccessSelector,
10
+ args: { value: "ask", onValueChange: () => undefined },
11
+ render: function Render(args) {
12
+ const [value, setValue] = useState(args.value);
13
+ return <NotebookAccessSelector value={value} onValueChange={setValue} />;
14
+ },
15
+ } satisfies Meta<typeof NotebookAccessSelector>;
16
+
17
+ export default meta;
18
+ type Story = StoryObj<typeof meta>;
19
+
20
+ export const ReadOnly: Story = { args: { value: "ask" } };
21
+ export const SuggestEdits: Story = { args: { value: "agent" } };
22
+ export const FullAccess: Story = { args: { value: "code_mode" } };
@@ -0,0 +1,100 @@
1
+ /* Copyright 2026 Marimo. All rights reserved. */
2
+
3
+ import {
4
+ EyeIcon,
5
+ type LucideIcon,
6
+ PencilIcon,
7
+ TerminalIcon,
8
+ } from "lucide-react";
9
+ import {
10
+ Select,
11
+ SelectContent,
12
+ SelectGroup,
13
+ SelectItem,
14
+ SelectLabel,
15
+ SelectTrigger,
16
+ } from "@/components/ui/select";
17
+ import type { CopilotMode } from "@/core/config/config-schema";
18
+
19
+ const ACCESS_LEVELS = {
20
+ ask: {
21
+ label: "Read only",
22
+ description: "Read notebook code, outputs, and variable previews.",
23
+ Icon: EyeIcon,
24
+ },
25
+ agent: {
26
+ label: "Suggest edits",
27
+ description: "Read the notebook and propose cell changes for review.",
28
+ Icon: PencilIcon,
29
+ },
30
+ code_mode: {
31
+ label: "Full access",
32
+ description: "Read, edit, and run cells, with Python scratchpad access.",
33
+ Icon: TerminalIcon,
34
+ },
35
+ } satisfies Record<
36
+ Exclude<CopilotMode, "manual">,
37
+ { label: string; description: string; Icon: LucideIcon }
38
+ >;
39
+
40
+ const ACCESS_MODES = ["ask", "agent", "code_mode"] as const;
41
+
42
+ interface NotebookAccessSelectorProps {
43
+ value: CopilotMode;
44
+ onValueChange: (value: CopilotMode) => void;
45
+ }
46
+
47
+ export const NotebookAccessSelector = ({
48
+ value,
49
+ onValueChange,
50
+ }: NotebookAccessSelectorProps) => {
51
+ const mode = value === "manual" ? "ask" : value;
52
+ const selected = ACCESS_LEVELS[mode];
53
+
54
+ return (
55
+ <Select
56
+ value={mode}
57
+ onValueChange={(value) => {
58
+ const mode = ACCESS_MODES.find((mode) => mode === value);
59
+ if (mode) {
60
+ onValueChange(mode);
61
+ }
62
+ }}
63
+ >
64
+ <SelectTrigger
65
+ aria-label={`Notebook access: ${selected.label}`}
66
+ title={`Notebook access: ${selected.description}`}
67
+ className="h-6 text-xs border-border shadow-none! ring-0! bg-muted hover:bg-muted/30 py-0 px-2 gap-1.5 shrink-0"
68
+ >
69
+ <selected.Icon className="h-3 w-3" />
70
+ <span className="text-nowrap">{selected.label}</span>
71
+ </SelectTrigger>
72
+ <SelectContent className="max-w-[calc(100vw-2rem)]">
73
+ <SelectGroup>
74
+ <SelectLabel className="text-xs uppercase tracking-wider text-muted-foreground/70 font-medium">
75
+ Notebook access
76
+ </SelectLabel>
77
+ {ACCESS_MODES.map((mode) => {
78
+ const { label, description, Icon } = ACCESS_LEVELS[mode];
79
+ return (
80
+ <SelectItem
81
+ key={mode}
82
+ value={mode}
83
+ textValue={label}
84
+ className="text-xs py-1"
85
+ >
86
+ <div className="flex items-start gap-2.5">
87
+ <Icon className="h-3 w-3 mt-1 shrink-0 text-muted-foreground" />
88
+ <div className="flex flex-col gap-0.5">
89
+ <span className="font-semibold">{label}</span>
90
+ <span className="text-muted-foreground">{description}</span>
91
+ </div>
92
+ </div>
93
+ </SelectItem>
94
+ );
95
+ })}
96
+ </SelectGroup>
97
+ </SelectContent>
98
+ </Select>
99
+ );
100
+ };
@@ -17,6 +17,7 @@ import type { DataTableSelection } from "./types";
17
17
 
18
18
  interface TableBottomBarProps<TData> {
19
19
  pagination: boolean;
20
+ hasNextPage?: boolean;
20
21
  totalColumns: number;
21
22
  selection?: DataTableSelection;
22
23
  onRowSelectionChange?: (value: RowSelectionState) => void;
@@ -30,6 +31,7 @@ interface TableBottomBarProps<TData> {
30
31
 
31
32
  export const TableBottomBar = <TData,>({
32
33
  pagination,
34
+ hasNextPage,
33
35
  totalColumns,
34
36
  selection,
35
37
  onRowSelectionChange,
@@ -147,7 +149,7 @@ export const TableBottomBar = <TData,>({
147
149
  const { totalColumns: effectiveTotalColumns, hiddenColumns } =
148
150
  getColumnCountForDisplay(table, totalColumns);
149
151
  const { rowsAndColumns, hiddenSuffix } = prettifyRowColumnCount({
150
- numRows: table.getRowCount(),
152
+ numRows: hasNextPage === undefined ? table.getRowCount() : "too_many",
151
153
  totalColumns: effectiveTotalColumns,
152
154
  hiddenColumns,
153
155
  locale,
@@ -178,6 +180,7 @@ export const TableBottomBar = <TData,>({
178
180
  <DataTablePagination
179
181
  table={table}
180
182
  tableLoading={tableLoading}
183
+ hasNextPage={hasNextPage}
181
184
  showPageSizeSelector={showPageSizeSelector}
182
185
  />
183
186
  )}
@@ -76,6 +76,7 @@ export const TableTopBar = <TData,>({
76
76
  <input
77
77
  ref={inputRef}
78
78
  type="text"
79
+ aria-label="Search table"
79
80
  className="h-6 border-none bg-transparent focus:outline-hidden text-sm w-full min-w-0"
80
81
  value={internalValue}
81
82
  onKeyDown={(e) => {
@@ -90,6 +91,7 @@ export const TableTopBar = <TData,>({
90
91
  {reloading && <Spinner size="small" />}
91
92
  {internalValue && (
92
93
  <Button
94
+ aria-label="Clear table search"
93
95
  variant="text"
94
96
  size="xs"
95
97
  className="h-5 w-5 p-0 shrink-0"