@marimo-team/islands 0.25.3-dev12 → 0.25.3-dev123

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 (391) hide show
  1. package/dist/{ConnectedDataExplorerComponent-CNMliVzw.js → ConnectedDataExplorerComponent-CFzqIsUg.js} +14 -14
  2. package/dist/{ErrorBoundary-CUd8VnYI.js → ErrorBoundary-BQ92dCB2.js} +8 -8
  3. package/dist/{abnfDiagram-VCTEODGH-D-Ez91gV.js → abnfDiagram-VCTEODGH-Di400kYu.js} +2 -2
  4. package/dist/any-language-editor-k2B8YI3Y.js +75 -0
  5. package/dist/{architectureDiagram-5GKGNRK7-CTpcNtO8.js → architectureDiagram-5GKGNRK7-BdTJcsNJ.js} +1 -1
  6. package/dist/{arrays-CN1djC4k.js → arrays-R6JsImiT.js} +8 -0
  7. package/dist/assets/__vite-browser-external-SaDYcg7x.js +1 -0
  8. package/dist/assets/{worker-HPY9fbhz.js → worker-CfME3uyi.js} +3 -2
  9. package/dist/{button-CvjkwdSk.js → button-BxB3RDjN.js} +1 -1
  10. package/dist/{capabilities-CQ5IjhzL.js → capabilities-BgR0osfN.js} +2 -2
  11. package/dist/{chat-ui-DaHynk8g.js → chat-ui-DA8zEZ1I.js} +1141 -1141
  12. package/dist/{check-EDwSwP89.js → check-P35dc2dH.js} +1 -1
  13. package/dist/{chunk-IMKFNOWR-DGxi-zeo.js → chunk-IMKFNOWR-mtD7aJBw.js} +1 -1
  14. package/dist/{chunk-SHT3W25Y-CV9J3rAB.js → chunk-SHT3W25Y-DybB1ShH.js} +1 -1
  15. package/dist/{chunk-SVP7TREG-BL-9Fqgg.js → chunk-SVP7TREG-DNhhdCex.js} +1 -1
  16. package/dist/{chunk-TICWLB2K-Cu_XMjDM.js → chunk-TICWLB2K-NhAQhlDf.js} +1 -1
  17. package/dist/{chunk-YOKDWASO-DPZk8vhS.js → chunk-YOKDWASO-DKplTrBY.js} +20 -20
  18. package/dist/{classDiagram-ZZMXUADV-BXXV13Hq.js → classDiagram-ZZMXUADV-GTL0l5xS.js} +2 -2
  19. package/dist/{classDiagram-v2-VYDZK3BY-B7FXJ1Cp.js → classDiagram-v2-VYDZK3BY-BWrM2ruI.js} +2 -2
  20. package/dist/{code-visibility-P0Ornave.js → code-visibility-PHelS0M0.js} +6 -6
  21. package/dist/{common-BtO98PuE.js → common-Bp8XWWmF.js} +5559 -5453
  22. package/dist/{copy-icon-Ft7t9bqz.js → copy-icon-CPoxUG7G.js} +3 -3
  23. package/dist/{cynefinDiagram-5FMLGOSQ-CH96rwba.js → cynefinDiagram-5FMLGOSQ-CX22wa7c.js} +1 -1
  24. package/dist/{dagre-GXQ25YYZ-BD3PmpFY.js → dagre-GXQ25YYZ-Df6Pp-cl.js} +1 -1
  25. package/dist/{data-grid-overlay-editor-B801aMeG.js → data-grid-overlay-editor-DKSlB5lk.js} +5 -5
  26. package/dist/{data-grid-types-BH_r7g7l.js → data-grid-types-0PJ-9N5n.js} +1 -1
  27. package/dist/{diagram-S7CK7UJ4-BsSJJldB.js → diagram-S7CK7UJ4-CctU9kVx.js} +1 -1
  28. package/dist/{diagram-UQ7AKVKN-muQtNZc4.js → diagram-UQ7AKVKN-FVHjCF9j.js} +1 -1
  29. package/dist/{diagram-VX7I27RA-CdhuZAqR.js → diagram-VX7I27RA-BSDbwFFv.js} +1 -1
  30. package/dist/{diagram-Z3DM3KII-BD8mXjOj.js → diagram-Z3DM3KII-BX8kW7Nc.js} +1 -1
  31. package/dist/{dialog-CEBFZNyc.js → dialog-C1PcpCqX.js} +3 -3
  32. package/dist/{dist-DpBpA_n1.js → dist-Lm-D-Vr6.js} +3 -3
  33. package/dist/{dropdown-menu-CP_qCYHC.js → dropdown-menu-B_7xD2X4.js} +177 -169
  34. package/dist/{ebnfDiagram-PWID7BFC-BIN1sweE.js → ebnfDiagram-PWID7BFC-DtapdLet.js} +2 -2
  35. package/dist/{erDiagram-RLTQ6QDP-Ctp5cfoN.js → erDiagram-RLTQ6QDP-BAru9jHt.js} +1 -1
  36. package/dist/{error-banner-D-WCIiMt.js → error-banner-Sl2FlSJ4.js} +2 -2
  37. package/dist/{error-entries-Bv6bZPF3.js → error-entries-g9iQiVIT.js} +844 -811
  38. package/dist/{feedback-button-Bpu-arCP.js → feedback-button-CVeAwpnu.js} +29 -29
  39. package/dist/{flowDiagram-HODETNUW-3EfLrlFb.js → flowDiagram-HODETNUW-CjaeFeea.js} +2 -2
  40. package/dist/{formats-27CW_UP4.js → formats-V8ifA0Bk.js} +3 -3
  41. package/dist/{ganttDiagram-EL5Y4UJY-BQT16JSc.js → ganttDiagram-EL5Y4UJY-Dh_4cBJZ.js} +2 -2
  42. package/dist/{glide-data-editor-XZB3u4tH.js → glide-data-editor-C3ejp-kx.js} +370 -368
  43. package/dist/{glide-utils-DJyqh0EM.js → glide-utils-Bt8_pBa2.js} +16 -16
  44. package/dist/{globals-BIQc0DK0.js → globals-CTjJ_AhA.js} +3 -3
  45. package/dist/{highlighted-body-KPVGNVTW-zwjq21tf.js → highlighted-body-KPVGNVTW-3Ziwfz-V.js} +2 -2
  46. package/dist/{html-to-image-Bf8QMGi4.js → html-to-image-Bg5ZoSO-.js} +29 -29
  47. package/dist/{infoDiagram-27XIBGKW-DipQoHxl.js → infoDiagram-27XIBGKW-B1hmUggC.js} +1 -1
  48. package/dist/{ishikawaDiagram-5VMMS53U-DgFMjSGz.js → ishikawaDiagram-5VMMS53U-DUeT3t-d.js} +1 -1
  49. package/dist/{kanban-definition-UXKFOSKX-w1ACVjgl.js → kanban-definition-UXKFOSKX-DEUnNNgc.js} +1 -1
  50. package/dist/{label-BDKF3Afc.js → label-BRnUC0NX.js} +2 -2
  51. package/dist/{linear-Cnaq3Tfb.js → linear-DWERh7eI.js} +121 -73
  52. package/dist/{loader-BQOfjTIJ.js → loader-Bm8cy3e8.js} +12 -12
  53. package/dist/main.js +3272 -3030
  54. package/dist/{mermaid-CHo49Yv-.js → mermaid-B0fXpsNP.js} +38 -38
  55. package/dist/mermaid-HWGCJPDP-Du3-HPQA.js +2 -0
  56. package/dist/{mindmap-definition-YA3MSWOX-CrT0KSfX.js → mindmap-definition-YA3MSWOX-CPexlhDQ.js} +1 -1
  57. package/dist/{number-overlay-editor-CEO_7tD7.js → number-overlay-editor-CYFUnKYE.js} +3 -3
  58. package/dist/{objects-CzUgoMIF.js → objects-DCmMvZtn.js} +1 -3
  59. package/dist/{once-BtE-wKRL.js → once-eG8Vmvjd.js} +1 -1
  60. package/dist/{pegDiagram-XKGWAZYB-BqvywNJA.js → pegDiagram-XKGWAZYB-CMpp1OOQ.js} +2 -2
  61. package/dist/{pieDiagram-E7YTZNPT-BfxJ_XCc.js → pieDiagram-E7YTZNPT-DlMWg5t1.js} +1 -1
  62. package/dist/precisionRound-DhAzeRcm.js +49 -0
  63. package/dist/{process-output-BZg56CCb.js → process-output-BtVPCGeI.js} +4 -4
  64. package/dist/{quadrantDiagram-AXDQQJYC-Ccw0xMjN.js → quadrantDiagram-AXDQQJYC-Blrm7V7_.js} +1 -1
  65. package/dist/{railroadDiagram-O6MQD6OU-8vFLVTPu.js → railroadDiagram-O6MQD6OU-DBJfseeO.js} +2 -2
  66. package/dist/{react-vega-BED0DjiB.js → react-vega-C4T9Rb_w.js} +12284 -12270
  67. package/dist/{requirementDiagram-BXWQKSXE-a_MxGvxe.js → requirementDiagram-BXWQKSXE-BC7V639r.js} +1 -1
  68. package/dist/{reveal-component-LPCUNcgp.js → reveal-component-e2LCo62o.js} +23 -23
  69. package/dist/{select-BButFi-d.js → select-B0LRDH0g.js} +4 -4
  70. package/dist/{sequenceDiagram-WJ2MYXX4-CXFMnxGh.js → sequenceDiagram-WJ2MYXX4-CgesU6F-.js} +1 -1
  71. package/dist/{spec-BwarBpUJ.js → spec-DTiOPKyk.js} +41 -11
  72. package/dist/{stateDiagram-D77RDMKH-DQeA_CwI.js → stateDiagram-D77RDMKH-lH_BUhs3.js} +2 -2
  73. package/dist/{stateDiagram-v2-MP3YSRHH-BxxQucm1.js → stateDiagram-v2-MP3YSRHH-8LL6FrBe.js} +2 -2
  74. package/dist/style.css +1 -1
  75. package/dist/{swimlanes-42K2YHIH-Do6L-cYg.js → swimlanes-42K2YHIH-Cuf9B9kE.js} +1 -1
  76. package/dist/{swimlanesDiagram-VR7AAH4N-DUzPvSVU.js → swimlanesDiagram-VR7AAH4N-BoMMtOPh.js} +2 -2
  77. package/dist/{swiper-component-DKntyn8p.js → swiper-component-DfhRvAFi.js} +2 -2
  78. package/dist/{time-BieKwdYS.js → time-BaMbJs4z.js} +18 -18
  79. package/dist/{timeline-definition-24CTP7MA-D3c46WM5.js → timeline-definition-24CTP7MA-yzzw4TF4.js} +1 -1
  80. package/dist/{toDate-UtiwmnH2.js → toDate-DKH0_kMo.js} +15 -18
  81. package/dist/{toast-B9UMEi8u.js → toast-CFWpihJU.js} +4 -4
  82. package/dist/{tooltip-BO1REbqs.js → tooltip-CKO1pHJo.js} +2 -2
  83. package/dist/{use-restore-focus-BEM7sFWl.js → use-restore-focus-DBCITLeF.js} +1 -1
  84. package/dist/{useAsyncData-Deh5kszT.js → useAsyncData-D0mLJstx.js} +15 -14
  85. package/dist/{useDateFormatter-D8XruNGG.js → useDateFormatter-c_yJtkAe.js} +2 -2
  86. package/dist/{useDeepCompareMemoize-DXwzRINI.js → useDeepCompareMemoize-BpPWi3G3.js} +1 -1
  87. package/dist/{useIframeCapabilities-DBETvvPl.js → useIframeCapabilities-BkdeVdXC.js} +1 -1
  88. package/dist/{useLifecycle-j7D9uD1o.js → useLifecycle-Dum8WSXq.js} +3 -3
  89. package/dist/{useTheme-CRnjkku0.js → useTheme-CfeaVw5D.js} +6 -6
  90. package/dist/{vega-component-BeF9dfyO.js → vega-component-DgytCKrL.js} +147 -147
  91. package/dist/{vega-embed-container-BNHE18G9.js → vega-embed-container--LYZqguO.js} +2 -2
  92. package/dist/{vega-loader.browser-DNST6uVf.js → vega-loader.browser-B9Pz0EGP.js} +298 -507
  93. package/dist/{vennDiagram-4TSXK5OY-BSZD8JcZ.js → vennDiagram-4TSXK5OY-eJOrTqqC.js} +1 -1
  94. package/dist/{wardleyDiagram-VM6X3IG4-48cUWkfA.js → wardleyDiagram-VM6X3IG4-TILPvee6.js} +1 -1
  95. package/dist/{xychartDiagram-S5SC5T6Z-CswYgYeu.js → xychartDiagram-S5SC5T6Z-DUPvRwwJ.js} +2 -2
  96. package/package.json +12 -12
  97. package/src/__tests__/host-imports.test.ts +1 -0
  98. package/src/__tests__/mount.test.ts +0 -20
  99. package/src/components/app-config/ai-config.tsx +7 -2
  100. package/src/components/chat/__tests__/message-queue.test.tsx +17 -0
  101. package/src/components/chat/acp/__tests__/state.test.ts +2 -2
  102. package/src/components/chat/acp/blocks.tsx +2 -2
  103. package/src/components/chat/chat-components.tsx +2 -0
  104. package/src/components/chat/chat-panel.tsx +9 -8
  105. package/src/components/chat/chat-utils.ts +4 -8
  106. package/src/components/data-table/__tests__/chart-spec-model.test.ts +195 -195
  107. package/src/components/data-table/__tests__/column-explorer.test.tsx +27 -1
  108. package/src/components/data-table/__tests__/columns.test.tsx +105 -0
  109. package/src/components/data-table/__tests__/context-menu-interaction.test.tsx +130 -0
  110. package/src/components/data-table/__tests__/date-popover.test.tsx +72 -0
  111. package/src/components/data-table/__tests__/export-actions.test.tsx +105 -0
  112. package/src/components/data-table/__tests__/search-highlight.test.tsx +109 -0
  113. package/src/components/data-table/__tests__/url-detector.test.tsx +74 -0
  114. package/src/components/data-table/cell-selection/feature.ts +2 -2
  115. package/src/components/data-table/cell-selection/types.ts +2 -2
  116. package/src/components/data-table/charts/__tests__/chart-panel.test.tsx +74 -1
  117. package/src/components/data-table/charts/__tests__/spec-snapshot.test.ts +20 -20
  118. package/src/components/data-table/charts/__tests__/spec.test.ts +78 -52
  119. package/src/components/data-table/charts/__tests__/storage.test.ts +4 -1
  120. package/src/components/data-table/charts/chart-spec/spec.ts +68 -43
  121. package/src/components/data-table/charts/charts.tsx +40 -30
  122. package/src/components/data-table/charts/storage.ts +16 -5
  123. package/src/components/data-table/column-explorer-panel/column-explorer.tsx +1 -1
  124. package/src/components/data-table/column-summary/chart-spec-model.tsx +23 -16
  125. package/src/components/data-table/column-summary/column-summary.tsx +2 -4
  126. package/src/components/data-table/column-visibility-dropdown.tsx +8 -13
  127. package/src/components/data-table/columns.tsx +45 -46
  128. package/src/components/data-table/context-menu.tsx +36 -20
  129. package/src/components/data-table/data-table.tsx +7 -1
  130. package/src/components/data-table/date-popover.tsx +6 -4
  131. package/src/components/data-table/export-actions.tsx +76 -48
  132. package/src/components/data-table/row-viewer-panel/row-viewer.tsx +2 -2
  133. package/src/components/data-table/search-highlight.tsx +79 -0
  134. package/src/components/data-table/types.ts +1 -0
  135. package/src/components/data-table/url-detector.tsx +49 -9
  136. package/src/components/datasources/__tests__/column-actions.test.tsx +97 -0
  137. package/src/components/datasources/__tests__/column-preview.test.tsx +63 -2
  138. package/src/components/datasources/__tests__/table-list.test.tsx +118 -0
  139. package/src/components/datasources/column-preview.tsx +8 -2
  140. package/src/components/datasources/datasources.tsx +47 -35
  141. package/src/components/debugger/debugger-code.tsx +5 -0
  142. package/src/components/dependency-graph/dependency-graph-tree.tsx +8 -8
  143. package/src/components/dependency-graph/elements.ts +24 -21
  144. package/src/components/dependency-graph/panels.tsx +2 -1
  145. package/src/components/dependency-graph/utils/__tests__/layout.test.ts +18 -11
  146. package/src/components/editor/SortableCell.tsx +1 -3
  147. package/src/components/editor/__tests__/dynamic-favicon.test.tsx +111 -18
  148. package/src/components/editor/actions/__tests__/pair-with-agent-modal.test.tsx +24 -0
  149. package/src/components/editor/actions/pair-with-agent-modal.tsx +18 -15
  150. package/src/components/editor/ai/add-cell-with-ai.tsx +3 -3
  151. package/src/components/editor/ai/ai-completion-editor.tsx +1 -0
  152. package/src/components/editor/alerts/startup-logs-alert.tsx +1 -0
  153. package/src/components/editor/cell/CreateCellButton.tsx +8 -10
  154. package/src/components/editor/cell/DeleteButton.tsx +1 -0
  155. package/src/components/editor/cell/__tests__/useShouldShowInterrupt.test.ts +39 -0
  156. package/src/components/editor/cell/code/__tests__/language-toggle.test.tsx +54 -0
  157. package/src/components/editor/cell/code/cell-editor.tsx +1 -1
  158. package/src/components/editor/cell/code/language-toggle.tsx +8 -7
  159. package/src/components/editor/cell/useShouldShowInterrupt.ts +2 -12
  160. package/src/components/editor/chrome/panels/__tests__/packages-panel.test.tsx +30 -1
  161. package/src/components/editor/chrome/panels/cache-panel.tsx +1 -0
  162. package/src/components/editor/chrome/panels/packages-panel.tsx +4 -7
  163. package/src/components/editor/chrome/wrapper/__tests__/attachment-status.test.tsx +12 -18
  164. package/src/components/editor/chrome/wrapper/__tests__/pending-ai-cells.test.tsx +58 -1
  165. package/src/components/editor/chrome/wrapper/footer-items/runtime-settings.tsx +1 -7
  166. package/src/components/editor/chrome/wrapper/pending-ai-cells.tsx +12 -2
  167. package/src/components/editor/chrome/wrapper/storage.ts +1 -1
  168. package/src/components/editor/connections/components.tsx +1 -1
  169. package/src/components/editor/controls/__tests__/command-palette.test.tsx +53 -0
  170. package/src/components/editor/controls/command-palette.tsx +22 -10
  171. package/src/components/editor/dynamic-favicon.tsx +21 -40
  172. package/src/components/editor/errors/__tests__/send-error-report-button.test.tsx +12 -23
  173. package/src/components/editor/file-tree/__tests__/file-explorer-interactions.test.tsx +6 -0
  174. package/src/components/editor/file-tree/__tests__/use-file-search.test.tsx +91 -0
  175. package/src/components/editor/file-tree/file-explorer.tsx +10 -5
  176. package/src/components/editor/file-tree/file-viewer.tsx +4 -2
  177. package/src/components/editor/file-tree/use-file-search.ts +35 -21
  178. package/src/components/editor/notebook-banner.tsx +1 -0
  179. package/src/components/editor/output/MarimoTracebackOutput.tsx +2 -2
  180. package/src/components/editor/output/ansi-reduce.ts +33 -20
  181. package/src/components/editor/output/console/ConsoleOutput.tsx +19 -25
  182. package/src/components/editor/package-alert.tsx +6 -4
  183. package/src/components/editor/renderers/grid-layout/grid-layout.tsx +12 -6
  184. package/src/components/editor/renderers/vertical-layout/sidebar/toggle.tsx +7 -1
  185. package/src/components/home/components.tsx +17 -31
  186. package/src/components/icons/__tests__/large-spinner.test.tsx +39 -0
  187. package/src/components/icons/large-spinner.tsx +3 -14
  188. package/src/components/icons/spinner.tsx +1 -1
  189. package/src/components/mcp/mcp-status-indicator.tsx +1 -1
  190. package/src/components/pages/__tests__/home-page.test.tsx +116 -0
  191. package/src/components/pages/home-page.tsx +164 -91
  192. package/src/components/scratchpad/scratchpad.tsx +2 -2
  193. package/src/components/slides/reveal-component.tsx +4 -2
  194. package/src/components/slides/slide-cell-view.tsx +1 -1
  195. package/src/components/slides/slide-form.tsx +1 -0
  196. package/src/components/tracing/tracing.tsx +2 -2
  197. package/src/components/ui/alert-dialog.tsx +1 -1
  198. package/src/core/ai/context/providers/datasource.ts +6 -2
  199. package/src/core/ai/tools/base.ts +16 -7
  200. package/src/core/ai/tools/edit-notebook-tool.ts +47 -45
  201. package/src/core/ai/tools/registry.ts +13 -8
  202. package/src/core/ai/tools/sample-tool.ts +6 -7
  203. package/src/core/attachments/state.ts +5 -8
  204. package/src/core/cells/__tests__/__snapshots__/cells.test.ts.snap +4 -0
  205. package/src/core/cells/__tests__/cell.test.ts +39 -0
  206. package/src/core/cells/__tests__/cells.test.ts +3 -1
  207. package/src/core/cells/__tests__/collapseConsoleOutputs.test.ts +54 -0
  208. package/src/core/cells/__tests__/notebook-outline.test.ts +113 -0
  209. package/src/core/cells/__tests__/session.test.ts +117 -36
  210. package/src/core/cells/cell.ts +14 -3
  211. package/src/core/cells/cells.ts +3 -1
  212. package/src/core/cells/pending-delete-service.ts +2 -6
  213. package/src/core/cells/session.ts +1 -1
  214. package/src/core/cells/types.ts +10 -1
  215. package/src/core/codemirror/__tests__/setup.test.ts +29 -29
  216. package/src/core/codemirror/cells/__tests__/extensions.test.ts +15 -0
  217. package/src/core/codemirror/cells/extensions.ts +7 -3
  218. package/src/core/codemirror/keymaps/vim.ts +2 -6
  219. package/src/core/codemirror/language/__tests__/sql.test.ts +22 -20
  220. package/src/core/codemirror/language/extension.ts +8 -8
  221. package/src/core/codemirror/language/languages/markdown.ts +8 -6
  222. package/src/core/codemirror/language/languages/python.ts +16 -10
  223. package/src/core/codemirror/language/languages/sql/sql.ts +9 -7
  224. package/src/core/codemirror/language/panel/panel.tsx +1 -0
  225. package/src/core/codemirror/language/types.ts +7 -7
  226. package/src/core/codemirror/rtc/extension.ts +24 -12
  227. package/src/core/codemirror/rtc/loro/awareness.ts +34 -19
  228. package/src/core/codemirror/rtc/loro/sync.ts +1 -0
  229. package/src/core/codemirror/utils.ts +1 -0
  230. package/src/core/datasets/__tests__/data-source-discovery.test.ts +4 -4
  231. package/src/core/datasets/data-source-discovery.ts +4 -6
  232. package/src/core/document/__tests__/codemirror.test.ts +483 -0
  233. package/src/core/document/__tests__/draft.test.ts +322 -0
  234. package/src/core/document/__tests__/fixtures.ts +92 -0
  235. package/src/core/document/__tests__/hooks.test.tsx +386 -0
  236. package/src/core/document/__tests__/lifecycle-scenarios.test.ts +61 -0
  237. package/src/core/document/__tests__/model.test.ts +158 -0
  238. package/src/core/document/__tests__/protocol-fixtures.ts +38 -0
  239. package/src/core/document/__tests__/recovery-scenarios.test.ts +230 -0
  240. package/src/core/document/__tests__/scenario-fixture.ts +42 -0
  241. package/src/core/document/__tests__/scenarios.test.ts +205 -0
  242. package/src/core/document/__tests__/state.test.ts +438 -0
  243. package/src/core/document/__tests__/submission.test.ts +203 -0
  244. package/src/core/document/__tests__/test-host.test.ts +92 -0
  245. package/src/core/document/__tests__/test-host.ts +247 -0
  246. package/src/core/document/__tests__/test-notebook.ts +139 -0
  247. package/src/core/document/__tests__/user.ts +61 -0
  248. package/src/core/document/codemirror.ts +311 -0
  249. package/src/core/document/draft.ts +472 -0
  250. package/src/core/document/hooks.tsx +118 -0
  251. package/src/core/document/index.ts +22 -0
  252. package/src/core/document/model.ts +52 -0
  253. package/src/core/document/select.ts +41 -0
  254. package/src/core/document/state.ts +374 -0
  255. package/src/core/document/sync.ts +363 -0
  256. package/src/core/host/AGENTS.md +12 -2
  257. package/src/core/host/__tests__/attachments.test.ts +56 -0
  258. package/src/core/host/__tests__/document.test.ts +126 -0
  259. package/src/core/host/__tests__/fixtures.ts +41 -0
  260. package/src/core/host/__tests__/notebook-client.test.ts +511 -132
  261. package/src/core/host/__tests__/static.test.ts +167 -0
  262. package/src/core/host/__tests__/wasm.test.ts +152 -0
  263. package/src/core/host/backend.ts +10 -0
  264. package/src/core/host/connection.ts +1 -40
  265. package/src/core/host/notebook-client.ts +204 -133
  266. package/src/core/host/resources/attachments.ts +22 -0
  267. package/src/core/host/resources/document.ts +101 -0
  268. package/src/core/host/static.ts +70 -0
  269. package/src/core/host/wasm.ts +92 -0
  270. package/src/core/hotkeys/actions.ts +2 -4
  271. package/src/core/islands/__tests__/bootstrap.test.ts +0 -26
  272. package/src/core/islands/bootstrap.ts +0 -1
  273. package/src/core/islands/components/__tests__/IslandControls.test.tsx +7 -5
  274. package/src/core/kernel/__tests__/RuntimeState.test.ts +1 -1
  275. package/src/core/network/connection.ts +4 -2
  276. package/src/core/network/useConnectionNotice.ts +8 -2
  277. package/src/core/runtime/__tests__/createWsUrl.test.ts +77 -7
  278. package/src/core/runtime/__tests__/runtime.test.ts +15 -58
  279. package/src/core/runtime/runtime.ts +25 -42
  280. package/src/core/runtime/types.ts +0 -2
  281. package/src/core/state/__tests__/jotai.test.tsx +108 -0
  282. package/src/core/state/jotai.ts +1 -1
  283. package/src/core/static/__tests__/files.test.ts +38 -10
  284. package/src/core/static/files.ts +2 -0
  285. package/src/core/wasm/bridge.ts +38 -10
  286. package/src/core/wasm/store.ts +3 -3
  287. package/src/core/wasm/worker/bootstrap.ts +2 -1
  288. package/src/core/wasm/worker/getMarimoWheel.ts +1 -1
  289. package/src/core/wasm/worker/types.ts +11 -0
  290. package/src/core/wasm/worker/worker.ts +48 -0
  291. package/src/core/websocket/__tests__/connection-state.test.ts +0 -12
  292. package/src/core/websocket/__tests__/useNotebookClient.hook.test.tsx +153 -70
  293. package/src/core/websocket/__tests__/useNotebookSession.hook.test.tsx +151 -428
  294. package/src/core/websocket/connection-state.ts +0 -10
  295. package/src/core/websocket/state.ts +7 -0
  296. package/src/core/websocket/transports/__tests__/notebook.test.ts +502 -0
  297. package/src/core/websocket/transports/notebook.ts +213 -0
  298. package/src/core/websocket/useNotebookClient.ts +38 -29
  299. package/src/core/websocket/useNotebookSession.tsx +1 -14
  300. package/src/css/globals.css +5 -0
  301. package/src/hooks/__tests__/useAsyncData.test.tsx +148 -0
  302. package/src/hooks/__tests__/useDebounce.test.tsx +51 -0
  303. package/src/hooks/__tests__/useDeepCompareMemoize.test.tsx +39 -0
  304. package/src/hooks/__tests__/useDelayElapsed.test.tsx +28 -0
  305. package/src/hooks/__tests__/useInternalStateWithSync.test.tsx +20 -0
  306. package/src/hooks/__tests__/useScript.test.tsx +77 -0
  307. package/src/hooks/__tests__/useStateForKey.test.tsx +257 -0
  308. package/src/hooks/useAsyncData.ts +3 -3
  309. package/src/hooks/useDataSourceDiscovery.ts +6 -4
  310. package/src/hooks/useDebounce.ts +6 -10
  311. package/src/hooks/useDeepCompareMemoize.ts +1 -1
  312. package/src/hooks/useDelayElapsed.ts +14 -10
  313. package/src/hooks/useElapsedTime.ts +3 -3
  314. package/src/hooks/useEventListener.ts +1 -2
  315. package/src/hooks/useInputHistory.ts +4 -7
  316. package/src/hooks/useInternalStateWithSync.ts +6 -6
  317. package/src/hooks/useResizeObserver.ts +4 -7
  318. package/src/hooks/useScript.ts +67 -70
  319. package/src/hooks/useStateForKey.ts +93 -0
  320. package/src/mount.tsx +0 -4
  321. package/src/plugins/__tests__/plugin-object-schemas.test.ts +104 -0
  322. package/src/plugins/core/registerReactComponent.tsx +1 -1
  323. package/src/plugins/impl/CodeEditorPlugin.tsx +1 -1
  324. package/src/plugins/impl/DataEditorPlugin.tsx +1 -1
  325. package/src/plugins/impl/DataTablePlugin.tsx +29 -33
  326. package/src/plugins/impl/DropdownPlugin.tsx +2 -6
  327. package/src/plugins/impl/FileBrowserPlugin.tsx +4 -16
  328. package/src/plugins/impl/FormPlugin.tsx +1 -1
  329. package/src/plugins/impl/RangeSliderPlugin.tsx +0 -1
  330. package/src/plugins/impl/RefreshPlugin.tsx +1 -0
  331. package/src/plugins/impl/SliderPlugin.tsx +6 -8
  332. package/src/plugins/impl/__tests__/DataTablePlugin.test.tsx +117 -1
  333. package/src/plugins/impl/__tests__/FileBrowserPlugin.test.tsx +45 -0
  334. package/src/plugins/impl/__tests__/SliderPlugin.test.tsx +24 -3
  335. package/src/plugins/impl/anywidget/__tests__/registry.test.ts +2 -0
  336. package/src/plugins/impl/anywidget/__tests__/widget-binding.test.ts +241 -2
  337. package/src/plugins/impl/anywidget/runtime.ts +21 -9
  338. package/src/plugins/impl/anywidget/widget-binding.ts +160 -82
  339. package/src/plugins/impl/chat/__tests__/chat-prompts.test.tsx +93 -0
  340. package/src/plugins/impl/chat/chat-ui.tsx +12 -10
  341. package/src/plugins/impl/code/any-language-editor.tsx +2 -2
  342. package/src/plugins/impl/data-editor/glide-data-editor.tsx +8 -5
  343. package/src/plugins/impl/data-editor/glide-utils.ts +3 -1
  344. package/src/plugins/impl/data-explorer/ConnectedDataExplorerComponent.tsx +1 -1
  345. package/src/plugins/impl/data-explorer/state/reducer.ts +1 -1
  346. package/src/plugins/impl/data-frames/DataFramePlugin.tsx +1 -2
  347. package/src/plugins/impl/mpl-interactive/MplInteractivePlugin.tsx +5 -3
  348. package/src/plugins/impl/panel/PanelPlugin.tsx +16 -17
  349. package/src/plugins/impl/panel/__tests__/utils.test.ts +39 -1
  350. package/src/plugins/impl/panel/utils.ts +17 -0
  351. package/src/plugins/impl/plotly/PlotlyPlugin.tsx +2 -5
  352. package/src/plugins/impl/plotly/selection.ts +25 -19
  353. package/src/plugins/impl/vega/VegaPlugin.tsx +2 -3
  354. package/src/plugins/impl/vega/make-selectable.ts +4 -7
  355. package/src/plugins/layout/RoutesPlugin.tsx +1 -1
  356. package/src/stories/__tests__/editor-story.test.tsx +33 -0
  357. package/src/stories/button.stories.tsx +1 -0
  358. package/src/stories/editor.stories.tsx +24 -17
  359. package/src/stories/form-wrapper.stories.tsx +2 -2
  360. package/src/stories/select.stories.tsx +2 -2
  361. package/src/stories/switchable-multi-select.stories.tsx +1 -1
  362. package/src/utils/__tests__/arrays.test.ts +35 -0
  363. package/src/utils/__tests__/batch-requests.test.ts +3 -3
  364. package/src/utils/__tests__/numbers.test.ts +48 -1
  365. package/src/utils/__tests__/objects.test.ts +21 -0
  366. package/src/utils/arrays.ts +10 -0
  367. package/src/utils/blob.ts +1 -1
  368. package/src/utils/numbers.ts +28 -17
  369. package/src/utils/objects.ts +3 -7
  370. package/src/utils/once.ts +19 -19
  371. package/src/utils/tracer.ts +9 -5
  372. package/src/utils/zod-utils.ts +1 -3
  373. package/dist/any-language-editor-DZ_YTVus.js +0 -76
  374. package/dist/assets/__vite-browser-external-BXYkiFsk.js +0 -1
  375. package/dist/mermaid-HWGCJPDP-ByCsZnYF.js +0 -2
  376. package/dist/precisionRound-DMIVQrt2.js +0 -97
  377. package/src/core/host/__tests__/events.test.ts +0 -46
  378. package/src/core/host/__tests__/protocol.test.ts +0 -214
  379. package/src/core/host/events.ts +0 -60
  380. package/src/core/host/protocol.ts +0 -1943
  381. package/src/core/state/__tests__/jotai.test.ts +0 -25
  382. package/src/core/websocket/__tests__/createConnectionTransport.test.ts +0 -49
  383. package/src/core/websocket/transports/__tests__/basic.test.ts +0 -275
  384. package/src/core/websocket/transports/__tests__/sse.test.ts +0 -335
  385. package/src/core/websocket/transports/__tests__/transport.test.ts +0 -308
  386. package/src/core/websocket/transports/__tests__/ws.test.ts +0 -125
  387. package/src/core/websocket/transports/basic.ts +0 -102
  388. package/src/core/websocket/transports/create.ts +0 -31
  389. package/src/core/websocket/transports/sse.ts +0 -308
  390. package/src/core/websocket/transports/transport.ts +0 -50
  391. package/src/core/websocket/transports/ws.ts +0 -98
@@ -3,7 +3,7 @@ import { t } from "./src-CA_m4ONl.js";
3
3
  import { n } from "./chunk-Y2CYZVJY-DKKS1VyV.js";
4
4
  import { $ as r, H as i, K as a, U as o, a as s, b as c, c as l, et as u, f as d, nt as f, tt as p, v as m, w as h, y as g } from "./chunk-DU6HZSFF-BwiyaNpg.js";
5
5
  import { t as _ } from "./rough.esm-CtV9AnV1.js";
6
- import { p as v } from "./mermaid-CHo49Yv-.js";
6
+ import { p as v } from "./mermaid-B0fXpsNP.js";
7
7
  //#region ../node_modules/.pnpm/khroma@2.1.0/node_modules/khroma/dist/methods/transparentize.js
8
8
  var y = (e, t) => p(e, "a", -t), b = 1e-10;
9
9
  function x(e, t) {
@@ -2,7 +2,7 @@ import { i as e } from "./chunk-75Z2AOVW-D7gxIs2Q.js";
2
2
  import { n as t } from "./src-CA_m4ONl.js";
3
3
  import { n } from "./chunk-Y2CYZVJY-DKKS1VyV.js";
4
4
  import { D as r, H as i, K as a, U as o, a as s, b as c, c as l, v as u, w as d, x as f, y as p } from "./chunk-DU6HZSFF-BwiyaNpg.js";
5
- import { p as m } from "./mermaid-CHo49Yv-.js";
5
+ import { p as m } from "./mermaid-B0fXpsNP.js";
6
6
  import { n as h } from "./mermaid-parser.core-DxuSfYwJ.js";
7
7
  import { t as g } from "./chunk-JWPE2WC7-DJvv1t8U.js";
8
8
  //#region ../node_modules/.pnpm/mermaid@11.17.2/node_modules/mermaid/dist/chunks/mermaid.core/wardleyDiagram-VM6X3IG4.mjs
@@ -1,4 +1,4 @@
1
- import { t as e } from "./linear-Cnaq3Tfb.js";
1
+ import { t as e } from "./linear-DWERh7eI.js";
2
2
  import { n as t } from "./ordinal-c7efuCSS.js";
3
3
  import { t as n } from "./range-BAIvbVHp.js";
4
4
  import { q as r } from "./cubic-OaEBhQPK.js";
@@ -9,7 +9,7 @@ import { n as s } from "./chunk-Y2CYZVJY-DKKS1VyV.js";
9
9
  import { D as c, H as l, K as u, U as d, a as f, b as p, c as m, f as h, v as g, w as _, y as v, z as y } from "./chunk-DU6HZSFF-BwiyaNpg.js";
10
10
  import "./chunk-PWAF6VOD-CQw5qATZ.js";
11
11
  import { t as b } from "./chunk-GMAD6QVW-BhaJx0X3.js";
12
- import { p as x } from "./mermaid-CHo49Yv-.js";
12
+ import { p as x } from "./mermaid-B0fXpsNP.js";
13
13
  //#region ../node_modules/.pnpm/d3-scale@4.0.2/node_modules/d3-scale/src/band.js
14
14
  function S() {
15
15
  var e = t().unknown(void 0), r = e.domain, i = e.range, o = 0, s = 1, c, l, u = !1, d = 0, f = 0, p = .5;
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@marimo-team/islands",
3
- "version": "0.25.3-dev12",
3
+ "version": "0.25.3-dev123",
4
4
  "main": "dist/main.js",
5
5
  "types": "dist/index.d.ts",
6
6
  "type": "module",
@@ -57,6 +57,7 @@
57
57
  "@marimo-team/codemirror-mcp": "^0.1.7",
58
58
  "@marimo-team/codemirror-sql": "^0.3.0",
59
59
  "@marimo-team/llm-info": "workspace:*",
60
+ "@marimo-team/client": "workspace:*",
60
61
  "@marimo-team/marimo-api": "workspace:*",
61
62
  "@marimo-team/react-slotz": "^0.2.0",
62
63
  "@marimo-team/smart-cells": "workspace:*",
@@ -152,6 +153,7 @@
152
153
  "typescript-memoize": "^1.1.1",
153
154
  "use-acp": "0.2.6",
154
155
  "use-resize-observer": "^9.1.0",
156
+ "vega": "6.2.0",
155
157
  "vega-lite": "6.4.3",
156
158
  "vega-loader": "^5.1.3",
157
159
  "vega-parser": "^7.1.3",
@@ -184,8 +186,7 @@
184
186
  "storybook": "storybook dev -p 6006",
185
187
  "build-storybook": "storybook build",
186
188
  "sync-katex-css": "node --experimental-strip-types scripts/sync-katex-css.ts",
187
- "codegen": "node --experimental-strip-types scripts/generate-host-protocol.ts",
188
- "codegen:check": "node --experimental-strip-types scripts/generate-host-protocol.ts --check"
189
+ "doctor": "npx react-doctor@latest"
189
190
  },
190
191
  "peerDependencies": {
191
192
  "react": "^17 || ^18 || ^19",
@@ -194,10 +195,10 @@
194
195
  "devDependencies": {
195
196
  "@codecov/vite-plugin": "^1.9.1",
196
197
  "@csstools/postcss-light-dark-function": "^2.0.11",
197
- "@playwright/test": "^1.63.0",
198
- "@storybook/addon-docs": "^10.6.0",
199
- "@storybook/addon-links": "^10.6.0",
200
- "@storybook/react-vite": "^10.6.0",
198
+ "@playwright/test": "^1.64.0",
199
+ "@storybook/addon-docs": "^10.6.2",
200
+ "@storybook/addon-links": "^10.6.2",
201
+ "@storybook/react-vite": "^10.6.2",
201
202
  "@testing-library/jest-dom": "^7.0.1",
202
203
  "@testing-library/react": "^16.3.3",
203
204
  "@types/dompurify": "^3.2.0",
@@ -207,27 +208,26 @@
207
208
  "@types/react": "^19.3.0",
208
209
  "@types/react-dom": "^19.3.0",
209
210
  "@types/timestring": "^6.0.5",
210
- "@vitejs/plugin-react": "^6.1.1",
211
+ "@vitejs/plugin-react": "^6.1.2",
211
212
  "@vitest/coverage-v8": "4.1.11",
212
213
  "blob-polyfill": "^7.0.20220408",
213
214
  "cross-env": "^7.0.3",
214
215
  "jsdom": "^24.1.3",
215
216
  "msw": "^2.15.0",
216
217
  "npm-run-all2": "^6.2.6",
217
- "orval": "8.40.0",
218
218
  "oxc-transform-react": "0.145.0",
219
219
  "oxfmt": "^0.68.0",
220
220
  "oxlint": "^1.83.0",
221
- "postcss": "^8.5.28",
221
+ "postcss": "^8.5.29",
222
222
  "postcss-prefix-selector": "^2.2.1",
223
223
  "react": "^19.3.0",
224
224
  "react-dom": "^19.3.0",
225
- "storybook": "^10.6.0",
225
+ "storybook": "^10.6.2",
226
226
  "stylelint": "^16.26.1",
227
227
  "stylelint-config-standard": "^36.0.1",
228
228
  "tailwindcss": "^4.3.3",
229
229
  "vega-typings": "^2.3.0",
230
- "vite": "^8.3.0",
230
+ "vite": "^8.3.4",
231
231
  "vite-plugin-wasm": "^3.6.0",
232
232
  "vitest": "^4.1.11",
233
233
  "yaml": "^2.9.1"
@@ -19,6 +19,7 @@ it("enforces host dependencies for runtime and type imports", () => {
19
19
  "src/core/host/allowed.ts": `
20
20
  import { atom } from "jotai/vanilla";
21
21
  import { z } from "zod";
22
+ import * as Protocol from "@marimo-team/client/protocol";
22
23
  import { Logger } from "@/utils/Logger";
23
24
  import type { Document } from "./document";
24
25
  export * from "@/core/host/events";
@@ -6,7 +6,6 @@ import { initialLayoutState, layoutStateAtom } from "@/core/layout/state";
6
6
  import { kioskModeAtom } from "@/core/mode";
7
7
  import { connectionAtom } from "@/core/network/connection";
8
8
  import { store } from "@/core/state/jotai";
9
- import { runtimeConfigAtom } from "@/core/runtime/config";
10
9
  import { isStaticNotebook } from "@/core/static/static-state";
11
10
  import { ConnectionState } from "@/core/websocket/types";
12
11
  import { pairConfigAtom } from "@/core/config/pair";
@@ -102,25 +101,6 @@ describe("mount", () => {
102
101
  });
103
102
 
104
103
  describe("connection state initialization", () => {
105
- it("passes the host notebook identity to the local runtime", () => {
106
- mount({ ...baseOptions, notebookId: "nb_local" }, mockElement);
107
- expect(store.get(runtimeConfigAtom).notebookId).toBe("nb_local");
108
- });
109
-
110
- it("uses a remote runtime's own notebook identity", () => {
111
- mount(
112
- {
113
- ...baseOptions,
114
- notebookId: "nb_local",
115
- runtimeConfig: [
116
- { url: "https://remote.example", notebookId: "nb_remote" },
117
- ],
118
- },
119
- mockElement,
120
- );
121
- expect(store.get(runtimeConfigAtom).notebookId).toBe("nb_remote");
122
- });
123
-
124
104
  it("should set connection to CONNECTING when runtimeConfig has lazy=false", () => {
125
105
  mount(
126
106
  {
@@ -48,6 +48,7 @@ import { DEFAULT_AI_MODEL, type UserConfig } from "@/core/config/config-schema";
48
48
  import { isWasm } from "@/core/wasm/utils";
49
49
  import { cn } from "@/utils/cn";
50
50
  import { Events } from "@/utils/events";
51
+ import { Objects } from "@/utils/objects";
51
52
  import { Strings } from "@/utils/strings";
52
53
  import { AIModelDropdown, getProviderLabel } from "../ai/ai-model-dropdown";
53
54
  import {
@@ -501,6 +502,7 @@ const ModelListItem: React.FC<ModelListItemProps> = ({
501
502
  <ModelInfoCard model={model} />
502
503
  {model.custom && (
503
504
  <Button
505
+ aria-label={`Delete model ${model.name}`}
504
506
  variant="ghost"
505
507
  size="icon"
506
508
  onClick={handleDelete}
@@ -681,7 +683,10 @@ export const CustomProvidersConfig: React.FC<AiConfigProps> = ({
681
683
  };
682
684
 
683
685
  const removeProvider = (providerName: string) => {
684
- const { [providerName]: _, ...rest } = customProviders;
686
+ const rest = Objects.filter(
687
+ customProviders,
688
+ (_, name) => name !== providerName,
689
+ );
685
690
  // Reset to clear nested dirty state, then set new value
686
691
  form.resetField("ai.custom_providers");
687
692
  form.setValue("ai.custom_providers", rest, { shouldDirty: true });
@@ -1846,7 +1851,7 @@ export const AddModelForm: React.FC<{
1846
1851
  id={modelNameInputId}
1847
1852
  value={modelName}
1848
1853
  onChange={(e) => setModelName(e.target.value)}
1849
- placeholder="gpt-4"
1854
+ placeholder="gpt-5.6-sol"
1850
1855
  className="text-xs mb-0"
1851
1856
  />
1852
1857
  </div>
@@ -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();
@@ -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,9 +328,9 @@ 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
  `}
@@ -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()}
@@ -166,7 +166,12 @@ const ChatHeader: React.FC<ChatHeaderProps> = ({
166
166
  return (
167
167
  <div className="flex border-b px-2 py-1 justify-between shrink-0 items-center">
168
168
  <Tooltip content="New chat">
169
- <Button variant="text" size="icon" onClick={onNewChat}>
169
+ <Button
170
+ aria-label="New chat"
171
+ variant="text"
172
+ size="icon"
173
+ onClick={onNewChat}
174
+ >
170
175
  <PlusIcon className="h-4 w-4" />
171
176
  </Button>
172
177
  </Tooltip>
@@ -716,14 +721,10 @@ const ChatPanelBody = () => {
716
721
  );
717
722
 
718
723
  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
-
724
+ // useEvent reads the latest loading state, even when called after an await
725
+ // (e.g. resolving @-context) that outlives the original render's closure.
725
726
  const submitOrQueue = useEvent((parts: ChatMessagePart[]) => {
726
- if (isLoadingRef.current || hasQueuedRef.current) {
727
+ if (isLoading || hasQueuedRef.current) {
727
728
  enqueueUserMessage(parts);
728
729
  } else {
729
730
  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) => {