@assistant-ui/core 0.3.15 → 0.3.17

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 (455) hide show
  1. package/dist/index.d.ts +2 -2
  2. package/dist/index.js +2 -2
  3. package/dist/index.js.map +1 -1
  4. package/dist/internal.d.ts +8 -2
  5. package/dist/internal.js +9 -3
  6. package/dist/model-context/frame/host.d.ts.map +1 -1
  7. package/dist/model-context/frame/host.js +3 -2
  8. package/dist/model-context/frame/host.js.map +1 -1
  9. package/dist/model-context/frame/provider.d.ts +5 -0
  10. package/dist/model-context/frame/provider.d.ts.map +1 -1
  11. package/dist/model-context/frame/provider.js +141 -28
  12. package/dist/model-context/frame/provider.js.map +1 -1
  13. package/dist/model-context/interactable-composer-metadata.d.ts +1 -1
  14. package/dist/model-context/interactable-composer-metadata.js +1 -1
  15. package/dist/model-context/interactable-composer-metadata.js.map +1 -1
  16. package/dist/model-context/registry.d.ts +2 -2
  17. package/dist/model-context/registry.d.ts.map +1 -1
  18. package/dist/model-context/registry.js +9 -28
  19. package/dist/model-context/registry.js.map +1 -1
  20. package/dist/model-context/tool-call-text.d.ts +26 -0
  21. package/dist/model-context/tool-call-text.d.ts.map +1 -0
  22. package/dist/model-context/tool-call-text.js +18 -0
  23. package/dist/model-context/tool-call-text.js.map +1 -0
  24. package/dist/react/RuntimeAdapter.js +1 -1
  25. package/dist/react/adapters/LocalStorageThreadListAdapter.d.ts.map +1 -1
  26. package/dist/react/adapters/LocalStorageThreadListAdapter.js +1 -1
  27. package/dist/react/adapters/LocalStorageThreadListAdapter.js.map +1 -1
  28. package/dist/react/client/AdaptedRemoteThread.js +1 -1
  29. package/dist/react/client/InMemoryThreadList.d.ts.map +1 -1
  30. package/dist/react/client/InMemoryThreadList.js +25 -21
  31. package/dist/react/client/InMemoryThreadList.js.map +1 -1
  32. package/dist/react/client/Interactables.d.ts.map +1 -1
  33. package/dist/react/client/Interactables.js +60 -178
  34. package/dist/react/client/Interactables.js.map +1 -1
  35. package/dist/react/client/RemoteThreadList.d.ts +15 -0
  36. package/dist/react/client/RemoteThreadList.d.ts.map +1 -1
  37. package/dist/react/client/RemoteThreadList.js +263 -194
  38. package/dist/react/client/RemoteThreadList.js.map +1 -1
  39. package/dist/react/client/Tools.d.ts.map +1 -1
  40. package/dist/react/client/Tools.js +6 -2
  41. package/dist/react/client/Tools.js.map +1 -1
  42. package/dist/react/index.d.ts +4 -2
  43. package/dist/react/index.js +2 -1
  44. package/dist/react/interactables-legacy/Interactables.d.ts.map +1 -1
  45. package/dist/react/interactables-legacy/Interactables.js +60 -178
  46. package/dist/react/interactables-legacy/Interactables.js.map +1 -1
  47. package/dist/react/interactables-shared/useInteractablePersistenceQueue.d.ts +24 -0
  48. package/dist/react/interactables-shared/useInteractablePersistenceQueue.d.ts.map +1 -0
  49. package/dist/react/interactables-shared/useInteractablePersistenceQueue.js +138 -0
  50. package/dist/react/interactables-shared/useInteractablePersistenceQueue.js.map +1 -0
  51. package/dist/react/model-context/toolbox.d.ts +2 -14
  52. package/dist/react/model-context/toolbox.d.ts.map +1 -1
  53. package/dist/react/model-context/toolbox.js +1 -13
  54. package/dist/react/model-context/toolbox.js.map +1 -1
  55. package/dist/react/model-context/useAssistantContext.js +37 -7
  56. package/dist/react/model-context/useAssistantContext.js.map +1 -1
  57. package/dist/react/model-context/useAuiToolOverrides.js +31 -5
  58. package/dist/react/model-context/useAuiToolOverrides.js.map +1 -1
  59. package/dist/react/model-context/useInlineRender.d.ts.map +1 -1
  60. package/dist/react/model-context/useInlineRender.js +10 -36
  61. package/dist/react/model-context/useInlineRender.js.map +1 -1
  62. package/dist/react/primitive-hooks/useMessageError.d.ts.map +1 -1
  63. package/dist/react/primitive-hooks/useMessageError.js +2 -8
  64. package/dist/react/primitive-hooks/useMessageError.js.map +1 -1
  65. package/dist/react/primitive-hooks/useThreadListLoadMore.d.ts.map +1 -1
  66. package/dist/react/primitive-hooks/useThreadListLoadMore.js +2 -4
  67. package/dist/react/primitive-hooks/useThreadListLoadMore.js.map +1 -1
  68. package/dist/react/primitives/composer/ComposerAttachments.d.ts.map +1 -1
  69. package/dist/react/primitives/composer/ComposerAttachments.js +4 -3
  70. package/dist/react/primitives/composer/ComposerAttachments.js.map +1 -1
  71. package/dist/react/primitives/composer/ComposerIf.js.map +1 -1
  72. package/dist/react/primitives/composer/ComposerQueue.d.ts +2 -1
  73. package/dist/react/primitives/composer/ComposerQueue.d.ts.map +1 -1
  74. package/dist/react/primitives/composer/ComposerQueue.js +4 -3
  75. package/dist/react/primitives/composer/ComposerQueue.js.map +1 -1
  76. package/dist/react/primitives/message/MessageAttachments.d.ts.map +1 -1
  77. package/dist/react/primitives/message/MessageAttachments.js +7 -6
  78. package/dist/react/primitives/message/MessageAttachments.js.map +1 -1
  79. package/dist/react/primitives/message/MessageGroupedParts.js +2 -2
  80. package/dist/react/primitives/message/MessageGroupedParts.js.map +1 -1
  81. package/dist/react/primitives/message/MessageParts.js +3 -3
  82. package/dist/react/primitives/message/MessageParts.js.map +1 -1
  83. package/dist/react/primitives/threadList/ThreadListItems.d.ts.map +1 -1
  84. package/dist/react/primitives/threadList/ThreadListItems.js +5 -4
  85. package/dist/react/primitives/threadList/ThreadListItems.js.map +1 -1
  86. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.d.ts +7 -5
  87. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.d.ts.map +1 -1
  88. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.js +44 -22
  89. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.js.map +1 -1
  90. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.d.ts +7 -6
  91. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.d.ts.map +1 -1
  92. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.js +102 -117
  93. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.js.map +1 -1
  94. package/dist/react/runtimes/RemoteThreadResource.d.ts +1 -6
  95. package/dist/react/runtimes/RemoteThreadResource.d.ts.map +1 -1
  96. package/dist/react/runtimes/RemoteThreadResource.js +3 -3
  97. package/dist/react/runtimes/RemoteThreadResource.js.map +1 -1
  98. package/dist/react/runtimes/assistant-transport/runManager.d.ts.map +1 -1
  99. package/dist/react/runtimes/assistant-transport/runManager.js +2 -12
  100. package/dist/react/runtimes/assistant-transport/runManager.js.map +1 -1
  101. package/dist/react/runtimes/assistant-transport/useLatestRef.js +2 -2
  102. package/dist/react/runtimes/assistant-transport/useLatestRef.js.map +1 -1
  103. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.d.ts +3 -10
  104. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.d.ts.map +1 -1
  105. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.js +43 -85
  106. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.js.map +1 -1
  107. package/dist/react/runtimes/cloud/auiV0.d.ts +5 -1
  108. package/dist/react/runtimes/cloud/auiV0.d.ts.map +1 -1
  109. package/dist/react/runtimes/cloud/auiV0.js.map +1 -1
  110. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.d.ts +29 -0
  111. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.d.ts.map +1 -0
  112. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.js +159 -0
  113. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.js.map +1 -0
  114. package/dist/react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.d.ts +12 -0
  115. package/dist/react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.d.ts.map +1 -0
  116. package/dist/react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.js +10 -0
  117. package/dist/react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.js.map +1 -0
  118. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.d.ts +1 -9
  119. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.d.ts.map +1 -1
  120. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.js +16 -153
  121. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.js.map +1 -1
  122. package/dist/react/runtimes/cloud/useCloudThreadListRuntime.js.map +1 -1
  123. package/dist/react/runtimes/createMessageConverter.d.ts +2 -1
  124. package/dist/react/runtimes/createMessageConverter.d.ts.map +1 -1
  125. package/dist/react/runtimes/external-message-converter.d.ts +6 -25
  126. package/dist/react/runtimes/external-message-converter.d.ts.map +1 -1
  127. package/dist/react/runtimes/external-message-converter.js +33 -233
  128. package/dist/react/runtimes/external-message-converter.js.map +1 -1
  129. package/dist/react/runtimes/useRemoteThreadListRuntime.d.ts.map +1 -1
  130. package/dist/react/runtimes/useRemoteThreadListRuntime.js +14 -6
  131. package/dist/react/runtimes/useRemoteThreadListRuntime.js.map +1 -1
  132. package/dist/react/types/MessagePartComponentTypes.d.ts +7 -3
  133. package/dist/react/types/MessagePartComponentTypes.d.ts.map +1 -1
  134. package/dist/react/types/scopes/tools.d.ts +2 -0
  135. package/dist/react/types/scopes/tools.d.ts.map +1 -1
  136. package/dist/runtime/api/composer-runtime.d.ts +1 -1
  137. package/dist/runtime/api/composer-runtime.js.map +1 -1
  138. package/dist/runtime/api/message-part-runtime.d.ts +2 -2
  139. package/dist/runtime/api/message-part-runtime.d.ts.map +1 -1
  140. package/dist/runtime/api/message-part-runtime.js +1 -1
  141. package/dist/runtime/api/message-part-runtime.js.map +1 -1
  142. package/dist/runtime/api/thread-list-runtime.d.ts +9 -1
  143. package/dist/runtime/api/thread-list-runtime.d.ts.map +1 -1
  144. package/dist/runtime/api/thread-list-runtime.js +5 -0
  145. package/dist/runtime/api/thread-list-runtime.js.map +1 -1
  146. package/dist/runtime/api/thread-runtime.d.ts +2 -3
  147. package/dist/runtime/api/thread-runtime.d.ts.map +1 -1
  148. package/dist/runtime/base/base-composer-runtime-core.d.ts +1 -1
  149. package/dist/runtime/base/base-composer-runtime-core.d.ts.map +1 -1
  150. package/dist/runtime/base/base-composer-runtime-core.js +14 -29
  151. package/dist/runtime/base/base-composer-runtime-core.js.map +1 -1
  152. package/dist/runtime/base/base-thread-runtime-core.d.ts +4 -6
  153. package/dist/runtime/base/base-thread-runtime-core.d.ts.map +1 -1
  154. package/dist/runtime/base/base-thread-runtime-core.js +5 -11
  155. package/dist/runtime/base/base-thread-runtime-core.js.map +1 -1
  156. package/dist/runtime/base/default-thread-composer-runtime-core.d.ts +1 -1
  157. package/dist/runtime/base/default-thread-composer-runtime-core.d.ts.map +1 -1
  158. package/dist/runtime/base/default-thread-composer-runtime-core.js +1 -1
  159. package/dist/runtime/base/default-thread-composer-runtime-core.js.map +1 -1
  160. package/dist/runtime/interfaces/composer-runtime-core.d.ts +1 -1
  161. package/dist/runtime/interfaces/thread-list-runtime-core.d.ts +14 -2
  162. package/dist/runtime/interfaces/thread-list-runtime-core.d.ts.map +1 -1
  163. package/dist/runtime/interfaces/thread-runtime-core.d.ts +16 -2
  164. package/dist/runtime/interfaces/thread-runtime-core.d.ts.map +1 -1
  165. package/dist/runtime/internal.d.ts +2 -1
  166. package/dist/runtime/internal.js +2 -1
  167. package/dist/runtime/queue/external-thread-queue-adapter.d.ts +1 -1
  168. package/dist/runtime/queue/message-queue.d.ts.map +1 -1
  169. package/dist/runtime/queue/message-queue.js +1 -1
  170. package/dist/runtime/queue/message-queue.js.map +1 -1
  171. package/dist/runtime/queue/queue-item.d.ts +12 -0
  172. package/dist/runtime/queue/queue-item.d.ts.map +1 -0
  173. package/dist/runtime/queue/queue-item.js +6 -0
  174. package/dist/runtime/queue/queue-item.js.map +1 -0
  175. package/dist/runtime/utils/abortable-thread-load.d.ts +15 -0
  176. package/dist/runtime/utils/abortable-thread-load.d.ts.map +1 -0
  177. package/dist/runtime/utils/abortable-thread-load.js +35 -0
  178. package/dist/runtime/utils/abortable-thread-load.js.map +1 -0
  179. package/dist/runtime/utils/attachment-add-operations.d.ts +18 -0
  180. package/dist/runtime/utils/attachment-add-operations.d.ts.map +1 -0
  181. package/dist/runtime/utils/attachment-add-operations.js +43 -0
  182. package/dist/runtime/utils/attachment-add-operations.js.map +1 -0
  183. package/dist/runtime/utils/auto-status.d.ts +7 -2
  184. package/dist/runtime/utils/auto-status.d.ts.map +1 -1
  185. package/dist/runtime/utils/auto-status.js +13 -3
  186. package/dist/runtime/utils/auto-status.js.map +1 -1
  187. package/dist/runtime/utils/external-message-conversion.d.ts +48 -0
  188. package/dist/runtime/utils/external-message-conversion.d.ts.map +1 -0
  189. package/dist/runtime/utils/external-message-conversion.js +222 -0
  190. package/dist/runtime/utils/external-message-conversion.js.map +1 -0
  191. package/dist/runtime/utils/external-store-message.d.ts +2 -1
  192. package/dist/runtime/utils/external-store-message.d.ts.map +1 -1
  193. package/dist/runtime/utils/external-store-message.js +2 -1
  194. package/dist/runtime/utils/external-store-message.js.map +1 -1
  195. package/dist/runtime/utils/message-repository-session.d.ts +30 -0
  196. package/dist/runtime/utils/message-repository-session.d.ts.map +1 -0
  197. package/dist/runtime/utils/message-repository-session.js +130 -0
  198. package/dist/runtime/utils/message-repository-session.js.map +1 -0
  199. package/dist/runtime/utils/message-repository.d.ts.map +1 -1
  200. package/dist/runtime/utils/message-repository.js +3 -4
  201. package/dist/runtime/utils/message-repository.js.map +1 -1
  202. package/dist/runtime/utils/pending-tool-calls.d.ts +20 -0
  203. package/dist/runtime/utils/pending-tool-calls.d.ts.map +1 -0
  204. package/dist/runtime/utils/pending-tool-calls.js +22 -0
  205. package/dist/runtime/utils/pending-tool-calls.js.map +1 -0
  206. package/dist/runtime/utils/resolveToolApprovalResponse.d.ts +6 -3
  207. package/dist/runtime/utils/resolveToolApprovalResponse.d.ts.map +1 -1
  208. package/dist/runtime/utils/resolveToolApprovalResponse.js +12 -3
  209. package/dist/runtime/utils/resolveToolApprovalResponse.js.map +1 -1
  210. package/dist/runtime/utils/thread-message-like.d.ts +3 -10
  211. package/dist/runtime/utils/thread-message-like.d.ts.map +1 -1
  212. package/dist/runtime/utils/thread-message-like.js.map +1 -1
  213. package/dist/runtime/utils/tool-call-tree.d.ts +39 -0
  214. package/dist/runtime/utils/tool-call-tree.d.ts.map +1 -0
  215. package/dist/runtime/utils/tool-call-tree.js +85 -0
  216. package/dist/runtime/utils/tool-call-tree.js.map +1 -0
  217. package/dist/runtimes/external-store/external-store-adapter.d.ts +29 -2
  218. package/dist/runtimes/external-store/external-store-adapter.d.ts.map +1 -1
  219. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.d.ts +2 -5
  220. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.d.ts.map +1 -1
  221. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.js +3 -10
  222. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.js.map +1 -1
  223. package/dist/runtimes/external-store/external-store-thread-runtime-core.d.ts +21 -2
  224. package/dist/runtimes/external-store/external-store-thread-runtime-core.d.ts.map +1 -1
  225. package/dist/runtimes/external-store/external-store-thread-runtime-core.js +154 -46
  226. package/dist/runtimes/external-store/external-store-thread-runtime-core.js.map +1 -1
  227. package/dist/runtimes/inert/InertThreadRuntimeCore.d.ts +1 -1
  228. package/dist/runtimes/inert/InertThreadRuntimeCore.d.ts.map +1 -1
  229. package/dist/runtimes/inert/InertThreadRuntimeCore.js.map +1 -1
  230. package/dist/runtimes/local/local-thread-runtime-core.d.ts +4 -2
  231. package/dist/runtimes/local/local-thread-runtime-core.d.ts.map +1 -1
  232. package/dist/runtimes/local/local-thread-runtime-core.js +23 -3
  233. package/dist/runtimes/local/local-thread-runtime-core.js.map +1 -1
  234. package/dist/runtimes/remote-thread-list/remote-thread-state.d.ts +12 -1
  235. package/dist/runtimes/remote-thread-list/remote-thread-state.d.ts.map +1 -1
  236. package/dist/runtimes/remote-thread-list/remote-thread-state.js +107 -14
  237. package/dist/runtimes/remote-thread-list/remote-thread-state.js.map +1 -1
  238. package/dist/runtimes/remote-thread-list/title.d.ts +9 -0
  239. package/dist/runtimes/remote-thread-list/title.d.ts.map +1 -0
  240. package/dist/runtimes/remote-thread-list/title.js +6 -0
  241. package/dist/runtimes/remote-thread-list/title.js.map +1 -0
  242. package/dist/runtimes/remote-thread-list/types.d.ts +6 -5
  243. package/dist/runtimes/remote-thread-list/types.d.ts.map +1 -1
  244. package/dist/runtimes/tool-invocations/ToolInvocationTracker.d.ts +31 -8
  245. package/dist/runtimes/tool-invocations/ToolInvocationTracker.d.ts.map +1 -1
  246. package/dist/runtimes/tool-invocations/ToolInvocationTracker.js +98 -59
  247. package/dist/runtimes/tool-invocations/ToolInvocationTracker.js.map +1 -1
  248. package/dist/store/clients/external-thread.d.ts +1 -1
  249. package/dist/store/clients/external-thread.d.ts.map +1 -1
  250. package/dist/store/clients/external-thread.js +85 -103
  251. package/dist/store/clients/external-thread.js.map +1 -1
  252. package/dist/store/clients/single-thread-list.d.ts.map +1 -1
  253. package/dist/store/clients/single-thread-list.js +2 -2
  254. package/dist/store/clients/single-thread-list.js.map +1 -1
  255. package/dist/store/clients/suggestions.d.ts.map +1 -1
  256. package/dist/store/clients/suggestions.js +33 -12
  257. package/dist/store/clients/suggestions.js.map +1 -1
  258. package/dist/store/clients/thread-message-client.js +1 -1
  259. package/dist/store/clients/thread-selection-events.js +1 -1
  260. package/dist/store/index.d.ts +6 -3
  261. package/dist/store/index.js +2 -1
  262. package/dist/store/internal.d.ts +2 -2
  263. package/dist/store/internal.js +2 -2
  264. package/dist/store/primitive-predicates.d.ts +4 -1
  265. package/dist/store/primitive-predicates.d.ts.map +1 -1
  266. package/dist/store/primitive-predicates.js +9 -1
  267. package/dist/store/primitive-predicates.js.map +1 -1
  268. package/dist/store/runtime-clients/composer-runtime-client.js +1 -1
  269. package/dist/store/runtime-clients/message-runtime-client.js +1 -1
  270. package/dist/store/runtime-clients/thread-list-item-runtime-client.js +1 -1
  271. package/dist/store/runtime-clients/thread-list-runtime-client.d.ts.map +1 -1
  272. package/dist/store/runtime-clients/thread-list-runtime-client.js +124 -105
  273. package/dist/store/runtime-clients/thread-list-runtime-client.js.map +1 -1
  274. package/dist/store/runtime-clients/thread-runtime-client.js +1 -1
  275. package/dist/store/runtime-clients/useSubscribable.d.ts +3 -1
  276. package/dist/store/runtime-clients/useSubscribable.d.ts.map +1 -1
  277. package/dist/store/runtime-clients/useSubscribable.js +1 -1
  278. package/dist/store/runtime-clients/useSubscribable.js.map +1 -1
  279. package/dist/store/scope-registration.d.ts +1 -1
  280. package/dist/store/scopes/composer.d.ts +2 -1
  281. package/dist/store/scopes/composer.d.ts.map +1 -1
  282. package/dist/store/scopes/message.d.ts +1 -1
  283. package/dist/store/scopes/part.d.ts +6 -3
  284. package/dist/store/scopes/part.d.ts.map +1 -1
  285. package/dist/store/scopes/queue-item.d.ts +2 -9
  286. package/dist/store/scopes/queue-item.d.ts.map +1 -1
  287. package/dist/store/scopes/queue-item.js +0 -6
  288. package/dist/subscribable/subscribable.d.ts +8 -1
  289. package/dist/subscribable/subscribable.d.ts.map +1 -1
  290. package/dist/subscribable/subscribable.js +24 -16
  291. package/dist/subscribable/subscribable.js.map +1 -1
  292. package/dist/types/message.d.ts +47 -4
  293. package/dist/types/message.d.ts.map +1 -1
  294. package/dist/types/message.js +7 -1
  295. package/dist/types/message.js.map +1 -1
  296. package/dist/utils/data-url.d.ts +14 -1
  297. package/dist/utils/data-url.d.ts.map +1 -1
  298. package/dist/utils/data-url.js +13 -1
  299. package/dist/utils/data-url.js.map +1 -1
  300. package/dist/utils/invoke-user-callback.d.ts +5 -0
  301. package/dist/utils/invoke-user-callback.d.ts.map +1 -0
  302. package/dist/utils/invoke-user-callback.js +19 -0
  303. package/dist/utils/invoke-user-callback.js.map +1 -0
  304. package/dist/utils/normalizePartStatus.js.map +1 -1
  305. package/package.json +12 -17
  306. package/src/index.ts +6 -1
  307. package/src/internal.ts +20 -0
  308. package/src/model-context/frame/host.test.ts +91 -7
  309. package/src/model-context/frame/host.ts +7 -2
  310. package/src/model-context/frame/provider.test.ts +484 -13
  311. package/src/model-context/frame/provider.ts +205 -35
  312. package/src/model-context/interactable-composer-metadata.ts +1 -1
  313. package/src/model-context/registry.test.ts +44 -0
  314. package/src/model-context/registry.ts +13 -35
  315. package/src/model-context/tool-call-text.test.ts +67 -0
  316. package/src/model-context/tool-call-text.ts +67 -0
  317. package/src/react/AssistantRuntimeProvider.test.tsx +66 -2
  318. package/src/react/adapters/LocalStorageThreadListAdapter.tsx +1 -3
  319. package/src/react/client/InMemoryThreadList.test.tsx +49 -0
  320. package/src/react/client/InMemoryThreadList.ts +7 -1
  321. package/src/react/client/Interactables.ts +19 -178
  322. package/src/react/client/RemoteThreadList.background.test.ts +384 -0
  323. package/src/react/client/RemoteThreadList.concurrent.test.tsx +274 -0
  324. package/src/react/client/RemoteThreadList.test.ts +369 -1
  325. package/src/react/client/RemoteThreadList.ts +320 -142
  326. package/src/react/client/Tools.ts +7 -1
  327. package/src/react/index.ts +4 -0
  328. package/src/react/interactables-legacy/Interactables.ts +20 -177
  329. package/src/react/interactables-shared/useInteractablePersistenceQueue.test.tsx +161 -0
  330. package/src/react/interactables-shared/useInteractablePersistenceQueue.ts +206 -0
  331. package/src/react/model-context/modelContextHooks.test.tsx +131 -0
  332. package/src/react/model-context/toolbox.test.ts +9 -9
  333. package/src/react/model-context/toolbox.ts +9 -36
  334. package/src/react/model-context/useAssistantContext.ts +4 -2
  335. package/src/react/model-context/useAuiToolOverrides.ts +4 -2
  336. package/src/react/model-context/useInlineRender.test.tsx +48 -0
  337. package/src/react/model-context/useInlineRender.ts +14 -10
  338. package/src/react/primitive-hooks/useMessageError.ts +2 -19
  339. package/src/react/primitive-hooks/useThreadListLoadMore.ts +2 -3
  340. package/src/react/primitives/composer/ComposerAttachments.test.tsx +76 -0
  341. package/src/react/primitives/composer/ComposerAttachments.tsx +9 -4
  342. package/src/react/primitives/composer/ComposerQueue.test.tsx +72 -0
  343. package/src/react/primitives/composer/ComposerQueue.tsx +7 -4
  344. package/src/react/primitives/message/MessageAttachments.test.tsx +107 -37
  345. package/src/react/primitives/message/MessageAttachments.tsx +10 -7
  346. package/src/react/primitives/message/MessageGroupedParts.tsx +2 -2
  347. package/src/react/primitives/message/MessageParts.tsx +3 -3
  348. package/src/react/primitives/threadList/ThreadListItems.test.tsx +99 -0
  349. package/src/react/primitives/threadList/ThreadListItems.tsx +8 -5
  350. package/src/react/runtimes/RemoteThreadListHookInstanceManager.running.test.tsx +94 -1
  351. package/src/react/runtimes/RemoteThreadListHookInstanceManager.tsx +72 -31
  352. package/src/react/runtimes/RemoteThreadListThreadListRuntimeCore.switch-delete.test.ts +169 -0
  353. package/src/react/runtimes/RemoteThreadListThreadListRuntimeCore.tsx +140 -124
  354. package/src/react/runtimes/RemoteThreadResource.ts +1 -4
  355. package/src/react/runtimes/assistant-transport/runManager.ts +4 -17
  356. package/src/react/runtimes/assistant-transport/transport-scheduling.test.ts +51 -2
  357. package/src/react/runtimes/assistant-transport/useLatestRef.ts +2 -2
  358. package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.test.ts +45 -0
  359. package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.test.tsx +157 -1
  360. package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.ts +82 -176
  361. package/src/react/runtimes/cloud/auiV0.ts +5 -0
  362. package/src/react/runtimes/cloud/createCloudThreadListAdapter.test.ts +88 -0
  363. package/src/react/runtimes/cloud/createCloudThreadListAdapter.ts +236 -0
  364. package/src/react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.test.ts +37 -0
  365. package/src/react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.ts +18 -0
  366. package/src/react/runtimes/cloud/useCloudThreadListAdapter.test.tsx +90 -1
  367. package/src/react/runtimes/cloud/useCloudThreadListAdapter.tsx +33 -238
  368. package/src/react/runtimes/external-message-converter.test.ts +0 -14
  369. package/src/react/runtimes/external-message-converter.test.tsx +77 -0
  370. package/src/react/runtimes/external-message-converter.ts +59 -481
  371. package/src/react/runtimes/useRemoteThreadListRuntime.concurrent.test.tsx +216 -0
  372. package/src/react/runtimes/useRemoteThreadListRuntime.ts +24 -6
  373. package/src/react/types/MessagePartComponentTypes.ts +7 -3
  374. package/src/react/types/scopes/tools.ts +2 -0
  375. package/src/runtime/api/composer-runtime.ts +1 -1
  376. package/src/runtime/api/message-part-runtime.ts +3 -3
  377. package/src/runtime/api/message-runtime.test.ts +28 -1
  378. package/src/runtime/api/thread-list-runtime.ts +25 -1
  379. package/src/runtime/base/base-composer-runtime-core.ts +20 -40
  380. package/src/runtime/base/base-thread-runtime-core.test.ts +110 -3
  381. package/src/runtime/base/base-thread-runtime-core.ts +16 -15
  382. package/src/runtime/base/default-thread-composer-runtime-core.ts +1 -4
  383. package/src/runtime/interfaces/composer-runtime-core.ts +1 -1
  384. package/src/runtime/interfaces/thread-list-runtime-core.ts +20 -1
  385. package/src/runtime/interfaces/thread-runtime-core.ts +18 -2
  386. package/src/runtime/internal.ts +5 -0
  387. package/src/runtime/queue/external-thread-queue-adapter.ts +1 -1
  388. package/src/runtime/queue/message-queue.ts +1 -4
  389. package/src/runtime/queue/queue-item.ts +10 -0
  390. package/src/runtime/utils/abortable-thread-load.test.ts +87 -0
  391. package/src/runtime/utils/abortable-thread-load.ts +53 -0
  392. package/src/runtime/utils/attachment-add-operations.ts +59 -0
  393. package/src/runtime/utils/auto-status.test.ts +57 -0
  394. package/src/runtime/utils/auto-status.ts +46 -1
  395. package/src/runtime/utils/external-message-conversion.test.ts +210 -0
  396. package/src/runtime/utils/external-message-conversion.ts +510 -0
  397. package/src/runtime/utils/external-store-message.ts +2 -0
  398. package/src/runtime/utils/message-repository-session.test.ts +195 -0
  399. package/src/runtime/utils/message-repository-session.ts +176 -0
  400. package/src/runtime/utils/message-repository.ts +7 -4
  401. package/src/runtime/utils/pending-tool-calls.test.ts +57 -0
  402. package/src/runtime/utils/pending-tool-calls.ts +35 -0
  403. package/src/runtime/utils/resolveToolApprovalResponse.test.ts +73 -0
  404. package/src/runtime/utils/resolveToolApprovalResponse.ts +27 -6
  405. package/src/runtime/utils/thread-message-like.ts +4 -10
  406. package/src/runtime/utils/tool-call-tree.test.ts +191 -0
  407. package/src/runtime/utils/tool-call-tree.ts +97 -0
  408. package/src/runtimes/external-store/external-store-adapter.ts +38 -2
  409. package/src/runtimes/external-store/external-store-thread-list-runtime-core.ts +6 -14
  410. package/src/runtimes/external-store/external-store-thread-runtime-core.ts +231 -61
  411. package/src/runtimes/inert/InertThreadRuntimeCore.ts +1 -1
  412. package/src/runtimes/local/local-thread-runtime-core.test.ts +201 -4
  413. package/src/runtimes/local/local-thread-runtime-core.ts +53 -4
  414. package/src/runtimes/remote-thread-list/remote-thread-state.test.ts +149 -0
  415. package/src/runtimes/remote-thread-list/remote-thread-state.ts +178 -16
  416. package/src/runtimes/remote-thread-list/title.ts +3 -0
  417. package/src/runtimes/remote-thread-list/types.ts +6 -5
  418. package/src/runtimes/tool-invocations/EDGE_CASES.md +125 -3
  419. package/src/runtimes/tool-invocations/ToolInvocationTracker.test.ts +856 -102
  420. package/src/runtimes/tool-invocations/ToolInvocationTracker.ts +145 -89
  421. package/src/store/clients/external-thread.ts +30 -69
  422. package/src/store/clients/single-thread-list.ts +4 -1
  423. package/src/store/clients/suggestions.test.ts +185 -0
  424. package/src/store/clients/suggestions.ts +42 -35
  425. package/src/store/index.ts +4 -0
  426. package/src/store/internal.ts +2 -0
  427. package/src/store/primitive-predicates.ts +35 -0
  428. package/src/store/runtime-clients/thread-list-runtime-client.ts +16 -2
  429. package/src/store/runtime-clients/useSubscribable.ts +8 -2
  430. package/src/store/scopes/message.ts +1 -1
  431. package/src/store/scopes/part.ts +6 -3
  432. package/src/store/scopes/queue-item.ts +2 -9
  433. package/src/subscribable/subscribable.test.ts +50 -0
  434. package/src/subscribable/subscribable.ts +33 -21
  435. package/src/tests/MessageRepository.test.ts +77 -0
  436. package/src/tests/RemoteThreadListThreadListRuntimeCore-load-race.test.ts +305 -0
  437. package/src/tests/RemoteThreadListThreadListRuntimeCore-loadMore.test.ts +78 -0
  438. package/src/tests/RemoteThreadListThreadListRuntimeCore-reload.test.ts +52 -0
  439. package/src/tests/RemoteThreadListThreadListRuntimeCore-title-race.test.ts +158 -0
  440. package/src/tests/external-store-optimistic-id.test.ts +79 -0
  441. package/src/tests/external-store-repository-instance.test.ts +166 -0
  442. package/src/tests/external-store-thread-runtime-core-adapter.test.ts +186 -16
  443. package/src/tests/external-store-thread-runtime-core.test.ts +642 -0
  444. package/src/tests/external-thread-attachments.test.tsx +258 -1
  445. package/src/tests/remote-thread-list-adapter-switch.test.ts +66 -1
  446. package/src/tests/single-thread-list-item.test.tsx +47 -0
  447. package/src/tests/thread-message-like.test.ts +39 -1
  448. package/src/tests/thread-switch-events.test.tsx +397 -2
  449. package/src/types/message.ts +58 -5
  450. package/src/utils/data-url.test.ts +52 -1
  451. package/src/utils/data-url.ts +21 -0
  452. package/src/utils/invoke-user-callback.test.ts +82 -0
  453. package/src/utils/invoke-user-callback.ts +31 -0
  454. package/src/utils/normalizePartStatus.ts +1 -1
  455. package/dist/store/scopes/queue-item.js.map +0 -1
@@ -21,6 +21,7 @@ import type {
21
21
  ExportedMessageRepository,
22
22
  ExportedMessageRepositoryItem,
23
23
  } from "../../internal";
24
+ import { isRecord } from "../../utils/json/is-json";
24
25
  import {
25
26
  RuntimeAdapterProvider,
26
27
  type RuntimeAdapters,
@@ -82,9 +83,6 @@ type StoredSystemMessage = Extract<ThreadMessage, { role: "system" }>;
82
83
  type StoredUserMessage = Extract<ThreadMessage, { role: "user" }>;
83
84
  type StoredAssistantMessage = Extract<ThreadMessage, { role: "assistant" }>;
84
85
 
85
- const isRecord = (value: unknown): value is Record<string, unknown> =>
86
- typeof value === "object" && value !== null && !Array.isArray(value);
87
-
88
86
  const parseJSON = (raw: string | null): unknown => {
89
87
  if (!raw) return undefined;
90
88
  try {
@@ -146,3 +146,52 @@ describe("InMemoryThreadList delete", () => {
146
146
  expect(state.mainThreadId).not.toBe("main");
147
147
  });
148
148
  });
149
+
150
+ describe("InMemoryThreadList item index selectors", () => {
151
+ it("resolves index selectors within the archived and regular subsets", async () => {
152
+ const { getAui } = setup();
153
+ await act(async () => {});
154
+
155
+ await act(async () => {
156
+ getAui().threads.switchToNewThread();
157
+ });
158
+ await act(async () => {});
159
+ const b = getAui().threads.getState().mainThreadId;
160
+
161
+ await act(async () => {
162
+ getAui().threads.switchToNewThread();
163
+ });
164
+ await act(async () => {});
165
+
166
+ await act(async () => {
167
+ getAui().threads.item({ id: b }).archive();
168
+ });
169
+ await act(async () => {});
170
+
171
+ const state = getAui().threads.getState();
172
+ expect(state.archivedThreadIds).toEqual([b]);
173
+ expect(state.threadIds).toHaveLength(2);
174
+
175
+ for (const [index, id] of state.archivedThreadIds.entries()) {
176
+ expect(
177
+ getAui().threads.item({ index, archived: true }).getState().id,
178
+ ).toBe(id);
179
+ }
180
+ for (const [index, id] of state.threadIds.entries()) {
181
+ expect(getAui().threads.item({ index }).getState().id).toBe(id);
182
+ expect(
183
+ getAui().threads.item({ index, archived: false }).getState().id,
184
+ ).toBe(id);
185
+ }
186
+
187
+ expect(() =>
188
+ getAui().threads.item({ index: state.threadIds.length }),
189
+ ).toThrow("out of bounds");
190
+ expect(() =>
191
+ getAui().threads.item({
192
+ index: state.archivedThreadIds.length,
193
+ archived: true,
194
+ }),
195
+ ).toThrow("out of bounds");
196
+ });
197
+ });
@@ -236,7 +236,13 @@ const useInMemoryThreadList = (
236
236
  const index = threads.findIndex((t) => t.id === selector.id);
237
237
  return threadListItems.get({ index });
238
238
  }
239
- return threadListItems.get(selector);
239
+ // The lookup is keyed over the combined thread array, while index
240
+ // selectors address the regular/archived subset the primitives render.
241
+ const ids = selector.archived ? state.archivedThreadIds : state.threadIds;
242
+ const id = ids[selector.index];
243
+ if (id === undefined) return threadListItems.get({ index: -1 });
244
+ const index = threads.findIndex((t) => t.id === id);
245
+ return threadListItems.get({ index });
240
246
  },
241
247
  thread: () => mainThreadClient.methods,
242
248
  };
@@ -24,8 +24,7 @@ import {
24
24
  interactableToolName,
25
25
  } from "../../model-context/interactable-composer-metadata";
26
26
  import { notifySubscribers as notifyStateSubscribers } from "../../subscribable/subscribable";
27
-
28
- const PERSISTENCE_DEBOUNCE_MS = 500;
27
+ import { useInteractablePersistenceQueue } from "../interactables-shared/useInteractablePersistenceQueue";
29
28
 
30
29
  type RestorePersistedStateOptions = {
31
30
  stash: Map<string, unknown>;
@@ -109,26 +108,6 @@ const useInteractablesResource = ({
109
108
  const adapterRef = useRef<
110
109
  Unstable_InteractablePersistenceAdapter | undefined
111
110
  >(undefined);
112
- const debounceTimerRef = useRef<ReturnType<typeof setTimeout> | undefined>(
113
- undefined,
114
- );
115
- const syncSeqRef = useRef(0);
116
- const latestSyncSeqByIdRef = useRef(new Map<string, number>());
117
- const inFlightPersistenceRef = useRef(0);
118
- const flushResolversRef = useRef<Array<() => void>>([]);
119
- const dirtyIdsRef = useRef(new Set<string>());
120
-
121
- type PersistenceBatch = {
122
- adapter: Unstable_InteractablePersistenceAdapter;
123
- payload: Unstable_InteractablePersistedState;
124
- dirtyIds: Set<string>;
125
- seq: number;
126
- };
127
-
128
- const outgoingQueueRef = useRef<PersistenceBatch[]>([]);
129
- const runPersistenceRef = useRef<(batch?: PersistenceBatch) => void>(
130
- () => {},
131
- );
132
111
 
133
112
  const setStateAndRef = useCallback(
134
113
  (
@@ -152,149 +131,28 @@ const useInteractablesResource = ({
152
131
  return result;
153
132
  }, []);
154
133
 
155
- const takeDirtyBatch = useCallback(
134
+ const updatePersistenceStatus = useCallback(
156
135
  (
157
- adapter: Unstable_InteractablePersistenceAdapter,
158
- ): PersistenceBatch | undefined => {
159
- if (dirtyIdsRef.current.size === 0) return;
160
- const dirtyIds = new Set(dirtyIdsRef.current);
161
- dirtyIdsRef.current.clear();
162
- const seq = ++syncSeqRef.current;
163
- for (const id of dirtyIds) latestSyncSeqByIdRef.current.set(id, seq);
164
- return { adapter, payload: exportState(), dirtyIds, seq };
165
- },
166
- [exportState],
167
- );
168
-
169
- const enqueuePersistence = useCallback(
170
- (adapter: Unstable_InteractablePersistenceAdapter) => {
171
- const batch = takeDirtyBatch(adapter);
172
- if (!batch) return;
173
- if (inFlightPersistenceRef.current === 0) {
174
- runPersistenceRef.current(batch);
175
- } else {
176
- outgoingQueueRef.current.push(batch);
177
- }
178
- },
179
- [takeDirtyBatch],
180
- );
181
-
182
- const runPersistence = useCallback(
183
- async (batch?: PersistenceBatch) => {
184
- const resolved =
185
- batch ??
186
- (adapterRef.current ? takeDirtyBatch(adapterRef.current) : undefined);
187
- if (!resolved) {
188
- if (inFlightPersistenceRef.current === 0) {
189
- for (const resolve of flushResolversRef.current) resolve();
190
- flushResolversRef.current = [];
191
- }
192
- return;
193
- }
194
-
195
- const { adapter, payload, dirtyIds, seq } = resolved;
196
- inFlightPersistenceRef.current += 1;
197
-
198
- setStateAndRef((prev) => ({
199
- ...prev,
200
- persistence: {
201
- ...prev.persistence,
202
- ...Object.fromEntries(
203
- [...dirtyIds].map((id) => [
204
- id,
205
- { isPending: true, error: undefined },
206
- ]),
207
- ),
208
- },
209
- }));
210
-
211
- try {
212
- await adapter.save(payload);
213
- setStateAndRef((prev) => {
214
- let changed = false;
215
- const persistence = { ...prev.persistence };
216
- for (const id of dirtyIds) {
217
- if (
218
- latestSyncSeqByIdRef.current.get(id) !== seq ||
219
- dirtyIdsRef.current.has(id)
220
- )
221
- continue;
222
- latestSyncSeqByIdRef.current.delete(id);
223
- delete persistence[id];
224
- changed = true;
225
- }
226
- return changed ? { ...prev, persistence } : prev;
227
- });
228
- } catch (e) {
229
- setStateAndRef((prev) => {
230
- let changed = false;
231
- const persistence = { ...prev.persistence };
232
- for (const id of dirtyIds) {
233
- if (
234
- latestSyncSeqByIdRef.current.get(id) !== seq ||
235
- dirtyIdsRef.current.has(id)
236
- )
237
- continue;
238
- latestSyncSeqByIdRef.current.delete(id);
239
- persistence[id] = { isPending: false, error: e };
240
- changed = true;
241
- }
242
- return changed ? { ...prev, persistence } : prev;
243
- });
244
- } finally {
245
- inFlightPersistenceRef.current -= 1;
246
- const next =
247
- outgoingQueueRef.current.shift() ??
248
- (adapterRef.current && dirtyIdsRef.current.size > 0
249
- ? takeDirtyBatch(adapterRef.current)
250
- : undefined);
251
- if (next) {
252
- if (debounceTimerRef.current !== undefined) {
253
- clearTimeout(debounceTimerRef.current);
254
- debounceTimerRef.current = undefined;
255
- }
256
- runPersistenceRef.current(next);
257
- } else if (inFlightPersistenceRef.current === 0) {
258
- for (const resolve of flushResolversRef.current) resolve();
259
- flushResolversRef.current = [];
260
- }
261
- }
136
+ updater: (
137
+ prev: Unstable_InteractablesState["persistence"],
138
+ ) => Unstable_InteractablesState["persistence"],
139
+ ) => {
140
+ setStateAndRef((prev) => {
141
+ const persistence = updater(prev.persistence);
142
+ return persistence === prev.persistence
143
+ ? prev
144
+ : { ...prev, persistence };
145
+ });
262
146
  },
263
- [setStateAndRef, takeDirtyBatch],
147
+ [setStateAndRef],
264
148
  );
265
- runPersistenceRef.current = (nextBatch) => {
266
- void runPersistence(nextBatch);
267
- };
268
149
 
269
- const flushIfPending = useCallback(() => {
270
- if (debounceTimerRef.current !== undefined) {
271
- clearTimeout(debounceTimerRef.current);
272
- debounceTimerRef.current = undefined;
273
- }
274
- if (adapterRef.current) enqueuePersistence(adapterRef.current);
275
- }, [enqueuePersistence]);
276
-
277
- const schedulePersistence = useCallback(
278
- (id: string) => {
279
- if (!adapterRef.current) return;
280
- dirtyIdsRef.current.add(id);
281
- if (debounceTimerRef.current !== undefined) {
282
- clearTimeout(debounceTimerRef.current);
283
- }
284
- debounceTimerRef.current = setTimeout(() => {
285
- debounceTimerRef.current = undefined;
286
- if (inFlightPersistenceRef.current === 0 && adapterRef.current) {
287
- enqueuePersistence(adapterRef.current);
288
- } else {
289
- debounceTimerRef.current = setTimeout(() => {
290
- debounceTimerRef.current = undefined;
291
- if (adapterRef.current) enqueuePersistence(adapterRef.current);
292
- }, PERSISTENCE_DEBOUNCE_MS);
293
- }
294
- }, PERSISTENCE_DEBOUNCE_MS);
295
- },
296
- [enqueuePersistence],
297
- );
150
+ const { flushIfPending, schedulePersistence, flush } =
151
+ useInteractablePersistenceQueue({
152
+ adapterRef,
153
+ snapshot: exportState,
154
+ updatePersistenceStatus,
155
+ });
298
156
 
299
157
  const restorePersistedState = useCallback(
300
158
  (
@@ -394,23 +252,6 @@ const useInteractablesResource = ({
394
252
  };
395
253
  }, [persistence, setPersistenceAdapter]);
396
254
 
397
- const flush = useCallback(async () => {
398
- if (debounceTimerRef.current !== undefined) {
399
- clearTimeout(debounceTimerRef.current);
400
- debounceTimerRef.current = undefined;
401
- }
402
- const hasWork =
403
- inFlightPersistenceRef.current > 0 ||
404
- dirtyIdsRef.current.size > 0 ||
405
- outgoingQueueRef.current.length > 0;
406
- if (!hasWork) return;
407
- const p = new Promise<void>((resolve) => {
408
- flushResolversRef.current.push(resolve);
409
- });
410
- if (adapterRef.current) enqueuePersistence(adapterRef.current);
411
- return p;
412
- }, [enqueuePersistence]);
413
-
414
255
  const setDefState = useCallback(
415
256
  (id: string, updater: (prev: unknown) => unknown) => {
416
257
  touchedIdsRef.current.add(id);
@@ -0,0 +1,384 @@
1
+ import { useEffect, useState } from "react";
2
+ import { useResource } from "@assistant-ui/tap";
3
+ import { describe, expect, it, vi } from "vitest";
4
+ import { flushTapSync, resource, withKey } from "@assistant-ui/tap";
5
+ import { AuiConfig, createAssistantClient } from "@assistant-ui/store/client";
6
+ import type { ThreadHistoryAdapter } from "../../adapters/thread-history";
7
+ import type { RemoteThreadListAdapter } from "../../runtimes/remote-thread-list/types";
8
+ import { useRuntimeAdapters } from "../runtimes/RuntimeAdapterProvider";
9
+ import { RemoteThreadList } from "./RemoteThreadList";
10
+
11
+ const stubComposer = { getState: () => ({}) };
12
+ const stubSuggestions = { getState: () => ({ suggestions: [] }) };
13
+
14
+ type Tracker = {
15
+ mounts: string[];
16
+ alive: Set<string>;
17
+ running: Set<string>;
18
+ messagesOf?: (id: string) => readonly { status?: { type: string } }[];
19
+ };
20
+
21
+ const useTrackedThread = (props: { threadId: string; tracker: Tracker }) => {
22
+ useState(() => {
23
+ props.tracker.mounts.push(props.threadId);
24
+ return true;
25
+ });
26
+ useEffect(() => {
27
+ const { tracker, threadId } = props;
28
+ tracker.alive.add(threadId);
29
+ return () => {
30
+ tracker.alive.delete(threadId);
31
+ };
32
+ // oxlint-disable-next-line react-hooks/exhaustive-deps -- lifecycle probe tracks the mounted identity
33
+ }, []);
34
+ return {
35
+ getState: () => ({
36
+ isRunning: props.tracker.running.has(props.threadId),
37
+ messages: props.tracker.messagesOf?.(props.threadId) ?? [],
38
+ }),
39
+ composer: () => stubComposer,
40
+ suggestions: () => stubSuggestions,
41
+ };
42
+ };
43
+ const TrackedThread = resource(useTrackedThread);
44
+
45
+ const useHistoryLoadingThread = (props: {
46
+ threadId: string;
47
+ tracker: Tracker;
48
+ loads: string[];
49
+ }) => {
50
+ const adapters = useRuntimeAdapters();
51
+ useState(() => {
52
+ if (adapters?.history) {
53
+ props.loads.push(props.threadId);
54
+ void adapters.history.load();
55
+ }
56
+ return true;
57
+ });
58
+ return useTrackedThread(props);
59
+ };
60
+ const HistoryLoadingThread = resource(useHistoryLoadingThread);
61
+
62
+ const makeAdapter = (
63
+ overrides: Partial<RemoteThreadListAdapter> = {},
64
+ ): RemoteThreadListAdapter => ({
65
+ list: vi.fn(async () => ({ threads: [] })),
66
+ initialize: vi.fn(async (threadId: string) => ({
67
+ remoteId: `remote-${threadId}`,
68
+ externalId: undefined,
69
+ })),
70
+ rename: vi.fn(async () => {}),
71
+ archive: vi.fn(async () => {}),
72
+ unarchive: vi.fn(async () => {}),
73
+ delete: vi.fn(async () => {}),
74
+ generateTitle: vi.fn(
75
+ async () =>
76
+ new ReadableStream({
77
+ start(controller) {
78
+ controller.close();
79
+ },
80
+ }) as never,
81
+ ),
82
+ fetch: vi.fn(async (id: string) => ({
83
+ status: "regular" as const,
84
+ remoteId: id,
85
+ externalId: undefined,
86
+ title: id,
87
+ })),
88
+ ...overrides,
89
+ });
90
+
91
+ const mountBackgroundList = (
92
+ adapter: RemoteThreadListAdapter,
93
+ tracker: Tracker,
94
+ ) => {
95
+ const handle = createAssistantClient(
96
+ AuiConfig({
97
+ threads: RemoteThreadList({
98
+ adapter,
99
+ backgroundThreads: true,
100
+ thread: (id) =>
101
+ withKey(id, TrackedThread({ threadId: id, tracker }) as never),
102
+ }),
103
+ }),
104
+ );
105
+ handle.subscribe(() => {});
106
+ return handle;
107
+ };
108
+
109
+ const createTracker = (): Tracker => ({
110
+ mounts: [],
111
+ alive: new Set(),
112
+ running: new Set(),
113
+ });
114
+
115
+ describe("RemoteThreadList backgroundThreads", () => {
116
+ it("keeps a switched-away thread mounted with live isRunning", async () => {
117
+ const adapter = makeAdapter({
118
+ list: vi.fn(async () => ({
119
+ threads: [{ status: "regular" as const, remoteId: "t1", title: "One" }],
120
+ })),
121
+ });
122
+ const tracker = createTracker();
123
+ const handle = mountBackgroundList(adapter, tracker);
124
+ const aui = handle.getClient();
125
+ await aui.threads.getLoadThreadsPromise();
126
+ await vi.waitFor(() => {
127
+ expect(aui.threads.getState().threadIds).toEqual(["t1"]);
128
+ });
129
+ const newThreadId = aui.threads.getState().mainThreadId;
130
+
131
+ flushTapSync(() => aui.threads.switchToThread("t1"));
132
+ await vi.waitFor(() => {
133
+ expect(aui.threads.getState().mainThreadId).toBe("t1");
134
+ expect(tracker.mounts).toContain("t1");
135
+ });
136
+ const t1MountsAfterFirstVisit = tracker.mounts.filter(
137
+ (id) => id === "t1",
138
+ ).length;
139
+
140
+ tracker.running.add("t1");
141
+ flushTapSync(() => aui.threads.switchToThread(newThreadId));
142
+ await vi.waitFor(() => {
143
+ expect(aui.threads.getState().mainThreadId).toBe(newThreadId);
144
+ });
145
+
146
+ expect(tracker.alive.has("t1")).toBe(true);
147
+ await vi.waitFor(() => {
148
+ expect(aui.threads.item({ id: "t1" }).getState().isRunning).toBe(true);
149
+ });
150
+ expect(aui.threads.item("main").getState().isRunning).toBe(false);
151
+ expect(tracker.mounts.filter((id) => id === "t1")).toHaveLength(
152
+ t1MountsAfterFirstVisit,
153
+ );
154
+ handle.destroy();
155
+ });
156
+
157
+ it("keeps archived bodies mounted and unmounts on delete and detach", async () => {
158
+ const adapter = makeAdapter({
159
+ list: vi.fn(async () => ({
160
+ threads: [
161
+ { status: "regular" as const, remoteId: "t1", title: "One" },
162
+ { status: "regular" as const, remoteId: "t2", title: "Two" },
163
+ ],
164
+ })),
165
+ });
166
+ const tracker = createTracker();
167
+ const handle = mountBackgroundList(adapter, tracker);
168
+ const aui = handle.getClient();
169
+ await aui.threads.getLoadThreadsPromise();
170
+ await vi.waitFor(() => {
171
+ expect(aui.threads.getState().threadIds).toEqual(["t1", "t2"]);
172
+ });
173
+
174
+ flushTapSync(() => aui.threads.switchToThread("t1"));
175
+ await vi.waitFor(() => {
176
+ expect(aui.threads.getState().mainThreadId).toBe("t1");
177
+ });
178
+ flushTapSync(() => aui.threads.switchToThread("t2"));
179
+ await vi.waitFor(() => {
180
+ expect(aui.threads.getState().mainThreadId).toBe("t2");
181
+ });
182
+
183
+ flushTapSync(() => aui.threads.item({ id: "t1" }).archive());
184
+ await vi.waitFor(() => {
185
+ expect(aui.threads.getState().archivedThreadIds).toEqual(["t1"]);
186
+ });
187
+ expect(tracker.alive.has("t1")).toBe(true);
188
+
189
+ flushTapSync(() => aui.threads.item({ id: "t1" }).delete());
190
+ await vi.waitFor(() => {
191
+ expect(tracker.alive.has("t1")).toBe(false);
192
+ });
193
+
194
+ flushTapSync(() => aui.threads.item({ id: "t2" }).detach());
195
+ await vi.waitFor(() => {
196
+ expect(tracker.alive.has("t2")).toBe(false);
197
+ });
198
+ expect(aui.threads.getState().mainThreadId).not.toBe("t2");
199
+ handle.destroy();
200
+ });
201
+
202
+ it("loads history once per thread body instead of once per switch", async () => {
203
+ const history: ThreadHistoryAdapter = {
204
+ load: async () => ({ messages: [] }),
205
+ append: async () => {},
206
+ };
207
+ const useAdapters = () => ({ history });
208
+ const adapter = makeAdapter({
209
+ list: vi.fn(async () => ({
210
+ threads: [{ status: "regular" as const, remoteId: "t1", title: "One" }],
211
+ })),
212
+ unstable_useAdapters: useAdapters,
213
+ });
214
+ const tracker = createTracker();
215
+ const loads: string[] = [];
216
+ const handle = createAssistantClient(
217
+ AuiConfig({
218
+ threads: RemoteThreadList({
219
+ adapter,
220
+ backgroundThreads: true,
221
+ thread: (id) =>
222
+ HistoryLoadingThread({ threadId: id, tracker, loads }) as never,
223
+ }),
224
+ }),
225
+ );
226
+ handle.subscribe(() => {});
227
+ const aui = handle.getClient();
228
+ await aui.threads.getLoadThreadsPromise();
229
+ await vi.waitFor(() => {
230
+ expect(aui.threads.getState().threadIds).toEqual(["t1"]);
231
+ });
232
+ const newThreadId = aui.threads.getState().mainThreadId;
233
+
234
+ flushTapSync(() => aui.threads.switchToThread("t1"));
235
+ await vi.waitFor(() => {
236
+ expect(loads).toContain("t1");
237
+ });
238
+ const t1LoadsAfterFirstVisit = loads.filter((id) => id === "t1").length;
239
+ flushTapSync(() => aui.threads.switchToThread(newThreadId));
240
+ await vi.waitFor(() => {
241
+ expect(aui.threads.getState().mainThreadId).toBe(newThreadId);
242
+ });
243
+ flushTapSync(() => aui.threads.switchToThread("t1"));
244
+ await vi.waitFor(() => {
245
+ expect(aui.threads.getState().mainThreadId).toBe("t1");
246
+ });
247
+
248
+ expect(loads.filter((id) => id === "t1")).toHaveLength(
249
+ t1LoadsAfterFirstVisit,
250
+ );
251
+ expect(tracker.alive.has("t1")).toBe(true);
252
+ handle.destroy();
253
+ });
254
+
255
+ it("generates a title once after a new thread initializes, never for loaded threads", async () => {
256
+ const adapter = makeAdapter({
257
+ list: vi.fn(async () => ({
258
+ threads: [{ status: "regular" as const, remoteId: "t1", title: "One" }],
259
+ })),
260
+ });
261
+ const tracker = createTracker();
262
+ tracker.messagesOf = () => [{ status: { type: "complete" } }];
263
+ const handle = mountBackgroundList(adapter, tracker);
264
+ const aui = handle.getClient();
265
+ await aui.threads.getLoadThreadsPromise();
266
+ await vi.waitFor(() => {
267
+ expect(aui.threads.getState().threadIds).toEqual(["t1"]);
268
+ });
269
+ const newThreadId = aui.threads.getState().mainThreadId;
270
+
271
+ flushTapSync(() => aui.threads.switchToThread("t1"));
272
+ await vi.waitFor(() => {
273
+ expect(aui.threads.getState().mainThreadId).toBe("t1");
274
+ });
275
+ flushTapSync(() => aui.threads.switchToThread(newThreadId));
276
+ await vi.waitFor(() => {
277
+ expect(aui.threads.getState().mainThreadId).toBe(newThreadId);
278
+ });
279
+ expect(adapter.generateTitle).not.toHaveBeenCalled();
280
+
281
+ await aui.threads.item("main").initialize();
282
+ await vi.waitFor(() => {
283
+ expect(adapter.generateTitle).toHaveBeenCalledOnce();
284
+ });
285
+ expect(adapter.generateTitle).toHaveBeenCalledWith(
286
+ `remote-${newThreadId}`,
287
+ expect.any(Array),
288
+ );
289
+
290
+ flushTapSync(() => aui.threads.switchToThread("t1"));
291
+ await vi.waitFor(() => {
292
+ expect(aui.threads.getState().mainThreadId).toBe("t1");
293
+ });
294
+ expect(adapter.generateTitle).toHaveBeenCalledOnce();
295
+ handle.destroy();
296
+ });
297
+
298
+ it("keeps the mode fixed when the prop flips after mount", async () => {
299
+ const adapter = makeAdapter({
300
+ list: vi.fn(async () => ({
301
+ threads: [{ status: "regular" as const, remoteId: "t1", title: "One" }],
302
+ })),
303
+ });
304
+ const tracker = createTracker();
305
+ let flip!: () => void;
306
+ const useFlippingThreads = () => {
307
+ const [background, setBackground] = useState(true);
308
+ flip = () => setBackground(false);
309
+ return useResource(
310
+ RemoteThreadList({
311
+ adapter,
312
+ backgroundThreads: background,
313
+ thread: (id) =>
314
+ withKey(id, TrackedThread({ threadId: id, tracker }) as never),
315
+ }),
316
+ );
317
+ };
318
+ const FlippingThreads = resource(useFlippingThreads);
319
+ const handle = createAssistantClient(
320
+ AuiConfig({ threads: FlippingThreads() as never }),
321
+ );
322
+ handle.subscribe(() => {});
323
+ const aui = handle.getClient();
324
+ await aui.threads.getLoadThreadsPromise();
325
+ await vi.waitFor(() => {
326
+ expect(aui.threads.getState().threadIds).toEqual(["t1"]);
327
+ });
328
+ const newThreadId = aui.threads.getState().mainThreadId;
329
+ flushTapSync(() => aui.threads.switchToThread("t1"));
330
+ await vi.waitFor(() => {
331
+ expect(aui.threads.getState().mainThreadId).toBe("t1");
332
+ });
333
+ flushTapSync(() => aui.threads.switchToThread(newThreadId));
334
+ await vi.waitFor(() => {
335
+ expect(aui.threads.getState().mainThreadId).toBe(newThreadId);
336
+ });
337
+ expect(tracker.alive.has("t1")).toBe(true);
338
+
339
+ flushTapSync(() => flip());
340
+ await vi.waitFor(() => {
341
+ expect(aui.threads.getState().mainThreadId).toBe(newThreadId);
342
+ });
343
+ expect(tracker.alive.has("t1")).toBe(true);
344
+ handle.destroy();
345
+ });
346
+
347
+ it("keeps one body when a reload remaps a local thread to its remote id", async () => {
348
+ let listed: { status: "regular"; remoteId: string; title: string }[] = [];
349
+ const adapter = makeAdapter({
350
+ list: vi.fn(async () => ({ threads: listed })),
351
+ });
352
+ const tracker = createTracker();
353
+ const handle = mountBackgroundList(adapter, tracker);
354
+ const aui = handle.getClient();
355
+ await aui.threads.getLoadThreadsPromise();
356
+ const localId = aui.threads.getState().mainThreadId;
357
+
358
+ await aui.threads.item("main").initialize();
359
+ const remoteId = `remote-${localId}`;
360
+ listed = [{ status: "regular" as const, remoteId, title: "One" }];
361
+ await aui.threads.reload();
362
+ await vi.waitFor(() => {
363
+ expect(aui.threads.getState().threadIds).toContain(localId);
364
+ });
365
+ expect(aui.threads.getState().threadIds).not.toContain(remoteId);
366
+
367
+ flushTapSync(() => aui.threads.switchToNewThread());
368
+ await vi.waitFor(() => {
369
+ expect(aui.threads.getState().mainThreadId).not.toBe(localId);
370
+ });
371
+ const localMounts = tracker.mounts.filter((id) => id === localId).length;
372
+ flushTapSync(() => aui.threads.switchToThread(remoteId));
373
+ await vi.waitFor(() => {
374
+ expect(aui.threads.item("main").getState().remoteId).toBe(remoteId);
375
+ });
376
+
377
+ expect(tracker.alive.has(localId)).toBe(true);
378
+ expect(tracker.alive.has(remoteId)).toBe(false);
379
+ expect(tracker.mounts.filter((id) => id === localId)).toHaveLength(
380
+ localMounts,
381
+ );
382
+ handle.destroy();
383
+ });
384
+ });