@assistant-ui/core 0.3.16 → 0.3.18

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 (474) hide show
  1. package/dist/adapters/attachment.d.ts.map +1 -1
  2. package/dist/adapters/attachment.js +5 -4
  3. package/dist/adapters/attachment.js.map +1 -1
  4. package/dist/index.d.ts +3 -3
  5. package/dist/index.js +2 -2
  6. package/dist/index.js.map +1 -1
  7. package/dist/internal.d.ts +4 -2
  8. package/dist/internal.js +4 -2
  9. package/dist/model-context/frame/host.d.ts.map +1 -1
  10. package/dist/model-context/frame/host.js +10 -4
  11. package/dist/model-context/frame/host.js.map +1 -1
  12. package/dist/model-context/frame/provider.d.ts +5 -0
  13. package/dist/model-context/frame/provider.d.ts.map +1 -1
  14. package/dist/model-context/frame/provider.js +101 -28
  15. package/dist/model-context/frame/provider.js.map +1 -1
  16. package/dist/model-context/interactable-composer-metadata.d.ts.map +1 -1
  17. package/dist/model-context/interactable-composer-metadata.js +16 -9
  18. package/dist/model-context/interactable-composer-metadata.js.map +1 -1
  19. package/dist/model-context/registry.d.ts.map +1 -1
  20. package/dist/model-context/registry.js +3 -5
  21. package/dist/model-context/registry.js.map +1 -1
  22. package/dist/model-context/tool-call-text.d.ts +26 -0
  23. package/dist/model-context/tool-call-text.d.ts.map +1 -0
  24. package/dist/model-context/tool-call-text.js +18 -0
  25. package/dist/model-context/tool-call-text.js.map +1 -0
  26. package/dist/model-context/types.d.ts.map +1 -1
  27. package/dist/model-context/types.js +5 -4
  28. package/dist/model-context/types.js.map +1 -1
  29. package/dist/react/RuntimeAdapter.js +1 -1
  30. package/dist/react/adapters/LocalStorageThreadListAdapter.d.ts.map +1 -1
  31. package/dist/react/adapters/LocalStorageThreadListAdapter.js +1 -1
  32. package/dist/react/adapters/LocalStorageThreadListAdapter.js.map +1 -1
  33. package/dist/react/client/AdaptedRemoteThread.js +1 -1
  34. package/dist/react/client/InMemoryThreadList.d.ts.map +1 -1
  35. package/dist/react/client/InMemoryThreadList.js +63 -57
  36. package/dist/react/client/InMemoryThreadList.js.map +1 -1
  37. package/dist/react/client/Interactables.d.ts.map +1 -1
  38. package/dist/react/client/Interactables.js +25 -28
  39. package/dist/react/client/Interactables.js.map +1 -1
  40. package/dist/react/client/RemoteThreadList.d.ts +16 -0
  41. package/dist/react/client/RemoteThreadList.d.ts.map +1 -1
  42. package/dist/react/client/RemoteThreadList.js +354 -242
  43. package/dist/react/client/RemoteThreadList.js.map +1 -1
  44. package/dist/react/client/Tools.d.ts.map +1 -1
  45. package/dist/react/client/Tools.js +24 -18
  46. package/dist/react/client/Tools.js.map +1 -1
  47. package/dist/react/index.d.ts +2 -1
  48. package/dist/react/index.js +2 -1
  49. package/dist/react/interactables-legacy/Interactables.js +246 -125
  50. package/dist/react/interactables-legacy/Interactables.js.map +1 -1
  51. package/dist/react/interactables-shared/useInteractablePersistenceQueue.d.ts.map +1 -1
  52. package/dist/react/interactables-shared/useInteractablePersistenceQueue.js +11 -9
  53. package/dist/react/interactables-shared/useInteractablePersistenceQueue.js.map +1 -1
  54. package/dist/react/model-context/makeAssistantTool.d.ts +2 -2
  55. package/dist/react/model-context/makeAssistantTool.js +1 -1
  56. package/dist/react/model-context/makeAssistantTool.js.map +1 -1
  57. package/dist/react/model-context/toolbox.d.ts +2 -14
  58. package/dist/react/model-context/toolbox.d.ts.map +1 -1
  59. package/dist/react/model-context/toolbox.js +1 -13
  60. package/dist/react/model-context/toolbox.js.map +1 -1
  61. package/dist/react/model-context/useAssistantContext.js +37 -7
  62. package/dist/react/model-context/useAssistantContext.js.map +1 -1
  63. package/dist/react/model-context/useAssistantTool.d.ts +2 -2
  64. package/dist/react/model-context/useAssistantTool.js +1 -1
  65. package/dist/react/model-context/useAssistantTool.js.map +1 -1
  66. package/dist/react/model-context/useAuiToolOverrides.js +31 -5
  67. package/dist/react/model-context/useAuiToolOverrides.js.map +1 -1
  68. package/dist/react/model-context/useInlineRender.d.ts.map +1 -1
  69. package/dist/react/model-context/useInlineRender.js +10 -36
  70. package/dist/react/model-context/useInlineRender.js.map +1 -1
  71. package/dist/react/model-context/useToolArgsStatus.d.ts.map +1 -1
  72. package/dist/react/model-context/useToolArgsStatus.js +2 -1
  73. package/dist/react/model-context/useToolArgsStatus.js.map +1 -1
  74. package/dist/react/primitive-hooks/useMessageError.d.ts.map +1 -1
  75. package/dist/react/primitive-hooks/useMessageError.js +2 -8
  76. package/dist/react/primitive-hooks/useMessageError.js.map +1 -1
  77. package/dist/react/primitive-hooks/useThreadListLoadMore.d.ts.map +1 -1
  78. package/dist/react/primitive-hooks/useThreadListLoadMore.js +2 -4
  79. package/dist/react/primitive-hooks/useThreadListLoadMore.js.map +1 -1
  80. package/dist/react/primitives/composer/ComposerAttachments.d.ts.map +1 -1
  81. package/dist/react/primitives/composer/ComposerAttachments.js +4 -3
  82. package/dist/react/primitives/composer/ComposerAttachments.js.map +1 -1
  83. package/dist/react/primitives/composer/ComposerQueue.d.ts.map +1 -1
  84. package/dist/react/primitives/composer/ComposerQueue.js +4 -3
  85. package/dist/react/primitives/composer/ComposerQueue.js.map +1 -1
  86. package/dist/react/primitives/message/MessageAttachments.d.ts.map +1 -1
  87. package/dist/react/primitives/message/MessageAttachments.js +7 -6
  88. package/dist/react/primitives/message/MessageAttachments.js.map +1 -1
  89. package/dist/react/primitives/message/MessageGroupedParts.js +2 -2
  90. package/dist/react/primitives/message/MessageGroupedParts.js.map +1 -1
  91. package/dist/react/primitives/message/MessageParts.js +3 -3
  92. package/dist/react/primitives/message/MessageParts.js.map +1 -1
  93. package/dist/react/primitives/threadList/ThreadListItems.d.ts.map +1 -1
  94. package/dist/react/primitives/threadList/ThreadListItems.js +5 -4
  95. package/dist/react/primitives/threadList/ThreadListItems.js.map +1 -1
  96. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.d.ts +6 -4
  97. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.d.ts.map +1 -1
  98. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.js +59 -29
  99. package/dist/react/runtimes/RemoteThreadListHookInstanceManager.js.map +1 -1
  100. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.d.ts +11 -6
  101. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.d.ts.map +1 -1
  102. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.js +160 -143
  103. package/dist/react/runtimes/RemoteThreadListThreadListRuntimeCore.js.map +1 -1
  104. package/dist/react/runtimes/RemoteThreadResource.d.ts +2 -6
  105. package/dist/react/runtimes/RemoteThreadResource.d.ts.map +1 -1
  106. package/dist/react/runtimes/RemoteThreadResource.js +15 -13
  107. package/dist/react/runtimes/RemoteThreadResource.js.map +1 -1
  108. package/dist/react/runtimes/assistant-transport/useLatestRef.js +2 -2
  109. package/dist/react/runtimes/assistant-transport/useLatestRef.js.map +1 -1
  110. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.d.ts +4 -1
  111. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.d.ts.map +1 -1
  112. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.js +424 -52
  113. package/dist/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.js.map +1 -1
  114. package/dist/react/runtimes/cloud/auiV0.d.ts +5 -1
  115. package/dist/react/runtimes/cloud/auiV0.d.ts.map +1 -1
  116. package/dist/react/runtimes/cloud/auiV0.js.map +1 -1
  117. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.d.ts +29 -0
  118. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.d.ts.map +1 -0
  119. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.js +160 -0
  120. package/dist/react/runtimes/cloud/createCloudThreadListAdapter.js.map +1 -0
  121. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.d.ts +1 -9
  122. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.d.ts.map +1 -1
  123. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.js +16 -153
  124. package/dist/react/runtimes/cloud/useCloudThreadListAdapter.js.map +1 -1
  125. package/dist/react/runtimes/external-message-converter.js +1 -1
  126. package/dist/react/runtimes/external-message-converter.js.map +1 -1
  127. package/dist/react/runtimes/useExternalStoreRuntime.js +75 -46
  128. package/dist/react/runtimes/useExternalStoreRuntime.js.map +1 -1
  129. package/dist/react/runtimes/useRemoteThreadListRuntime.d.ts.map +1 -1
  130. package/dist/react/runtimes/useRemoteThreadListRuntime.js +77 -29
  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/message-part-runtime.d.ts +2 -2
  137. package/dist/runtime/api/message-part-runtime.d.ts.map +1 -1
  138. package/dist/runtime/api/message-part-runtime.js +1 -1
  139. package/dist/runtime/api/message-part-runtime.js.map +1 -1
  140. package/dist/runtime/api/thread-list-item-runtime.d.ts +7 -3
  141. package/dist/runtime/api/thread-list-item-runtime.d.ts.map +1 -1
  142. package/dist/runtime/api/thread-list-item-runtime.js +2 -2
  143. package/dist/runtime/api/thread-list-item-runtime.js.map +1 -1
  144. package/dist/runtime/api/thread-list-runtime.d.ts +12 -1
  145. package/dist/runtime/api/thread-list-runtime.d.ts.map +1 -1
  146. package/dist/runtime/api/thread-list-runtime.js +23 -14
  147. package/dist/runtime/api/thread-list-runtime.js.map +1 -1
  148. package/dist/runtime/api/thread-runtime.d.ts +5 -1
  149. package/dist/runtime/api/thread-runtime.d.ts.map +1 -1
  150. package/dist/runtime/api/thread-runtime.js +1 -1
  151. package/dist/runtime/api/thread-runtime.js.map +1 -1
  152. package/dist/runtime/base/base-composer-runtime-core.d.ts.map +1 -1
  153. package/dist/runtime/base/base-composer-runtime-core.js +16 -12
  154. package/dist/runtime/base/base-composer-runtime-core.js.map +1 -1
  155. package/dist/runtime/base/base-thread-runtime-core.d.ts +2 -2
  156. package/dist/runtime/base/base-thread-runtime-core.d.ts.map +1 -1
  157. package/dist/runtime/base/base-thread-runtime-core.js +8 -7
  158. package/dist/runtime/base/base-thread-runtime-core.js.map +1 -1
  159. package/dist/runtime/interfaces/composer-runtime-core.d.ts +12 -5
  160. package/dist/runtime/interfaces/composer-runtime-core.d.ts.map +1 -1
  161. package/dist/runtime/interfaces/thread-list-runtime-core.d.ts +18 -3
  162. package/dist/runtime/interfaces/thread-list-runtime-core.d.ts.map +1 -1
  163. package/dist/runtime/interfaces/thread-runtime-core.d.ts +15 -1
  164. package/dist/runtime/interfaces/thread-runtime-core.d.ts.map +1 -1
  165. package/dist/runtime/utils/auto-status.d.ts +1 -1
  166. package/dist/runtime/utils/auto-status.d.ts.map +1 -1
  167. package/dist/runtime/utils/auto-status.js +6 -2
  168. package/dist/runtime/utils/auto-status.js.map +1 -1
  169. package/dist/runtime/utils/external-message-conversion.d.ts +3 -2
  170. package/dist/runtime/utils/external-message-conversion.d.ts.map +1 -1
  171. package/dist/runtime/utils/external-message-conversion.js +5 -4
  172. package/dist/runtime/utils/external-message-conversion.js.map +1 -1
  173. package/dist/runtime/utils/message-repository.d.ts +1 -0
  174. package/dist/runtime/utils/message-repository.d.ts.map +1 -1
  175. package/dist/runtime/utils/message-repository.js +15 -6
  176. package/dist/runtime/utils/message-repository.js.map +1 -1
  177. package/dist/runtime/utils/resolveToolApprovalResponse.d.ts +6 -3
  178. package/dist/runtime/utils/resolveToolApprovalResponse.d.ts.map +1 -1
  179. package/dist/runtime/utils/resolveToolApprovalResponse.js +12 -3
  180. package/dist/runtime/utils/resolveToolApprovalResponse.js.map +1 -1
  181. package/dist/runtime/utils/thread-message-like.d.ts +2 -10
  182. package/dist/runtime/utils/thread-message-like.d.ts.map +1 -1
  183. package/dist/runtime/utils/thread-message-like.js +1 -1
  184. package/dist/runtime/utils/thread-message-like.js.map +1 -1
  185. package/dist/runtime/utils/tool-call-tree.d.ts +39 -0
  186. package/dist/runtime/utils/tool-call-tree.d.ts.map +1 -0
  187. package/dist/runtime/utils/tool-call-tree.js +85 -0
  188. package/dist/runtime/utils/tool-call-tree.js.map +1 -0
  189. package/dist/runtimes/external-store/external-store-adapter.d.ts +29 -2
  190. package/dist/runtimes/external-store/external-store-adapter.d.ts.map +1 -1
  191. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.d.ts.map +1 -1
  192. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.js +21 -27
  193. package/dist/runtimes/external-store/external-store-thread-list-runtime-core.js.map +1 -1
  194. package/dist/runtimes/external-store/external-store-thread-runtime-core.d.ts +2 -1
  195. package/dist/runtimes/external-store/external-store-thread-runtime-core.d.ts.map +1 -1
  196. package/dist/runtimes/external-store/external-store-thread-runtime-core.js +34 -28
  197. package/dist/runtimes/external-store/external-store-thread-runtime-core.js.map +1 -1
  198. package/dist/runtimes/inert/InertThreadRuntimeCore.d.ts +1 -1
  199. package/dist/runtimes/inert/InertThreadRuntimeCore.d.ts.map +1 -1
  200. package/dist/runtimes/inert/InertThreadRuntimeCore.js.map +1 -1
  201. package/dist/runtimes/internal.d.ts +2 -2
  202. package/dist/runtimes/internal.js +2 -2
  203. package/dist/runtimes/local/local-thread-runtime-core.d.ts +1 -1
  204. package/dist/runtimes/local/local-thread-runtime-core.d.ts.map +1 -1
  205. package/dist/runtimes/local/local-thread-runtime-core.js +3 -1
  206. package/dist/runtimes/local/local-thread-runtime-core.js.map +1 -1
  207. package/dist/runtimes/remote-thread-list/remote-thread-state.d.ts +18 -1
  208. package/dist/runtimes/remote-thread-list/remote-thread-state.d.ts.map +1 -1
  209. package/dist/runtimes/remote-thread-list/remote-thread-state.js +148 -43
  210. package/dist/runtimes/remote-thread-list/remote-thread-state.js.map +1 -1
  211. package/dist/runtimes/remote-thread-list/title-generation.d.ts +49 -0
  212. package/dist/runtimes/remote-thread-list/title-generation.d.ts.map +1 -0
  213. package/dist/runtimes/remote-thread-list/title-generation.js +197 -0
  214. package/dist/runtimes/remote-thread-list/title-generation.js.map +1 -0
  215. package/dist/runtimes/remote-thread-list/title.d.ts +11 -0
  216. package/dist/runtimes/remote-thread-list/title.d.ts.map +1 -0
  217. package/dist/runtimes/remote-thread-list/title.js +14 -0
  218. package/dist/runtimes/remote-thread-list/title.js.map +1 -0
  219. package/dist/runtimes/remote-thread-list/types.d.ts +16 -5
  220. package/dist/runtimes/remote-thread-list/types.d.ts.map +1 -1
  221. package/dist/runtimes/tool-invocations/ToolInvocationTracker.d.ts +31 -8
  222. package/dist/runtimes/tool-invocations/ToolInvocationTracker.d.ts.map +1 -1
  223. package/dist/runtimes/tool-invocations/ToolInvocationTracker.js +97 -60
  224. package/dist/runtimes/tool-invocations/ToolInvocationTracker.js.map +1 -1
  225. package/dist/store/clients/external-thread.d.ts +1 -1
  226. package/dist/store/clients/external-thread.d.ts.map +1 -1
  227. package/dist/store/clients/external-thread.js +30 -7
  228. package/dist/store/clients/external-thread.js.map +1 -1
  229. package/dist/store/clients/model-context-client.d.ts.map +1 -1
  230. package/dist/store/clients/model-context-client.js +2 -1
  231. package/dist/store/clients/model-context-client.js.map +1 -1
  232. package/dist/store/clients/single-thread-list.d.ts.map +1 -1
  233. package/dist/store/clients/single-thread-list.js +2 -1
  234. package/dist/store/clients/single-thread-list.js.map +1 -1
  235. package/dist/store/clients/suggestions.d.ts.map +1 -1
  236. package/dist/store/clients/suggestions.js +19 -6
  237. package/dist/store/clients/suggestions.js.map +1 -1
  238. package/dist/store/clients/thread-message-client.js +1 -1
  239. package/dist/store/clients/thread-selection-events.js +1 -1
  240. package/dist/store/index.d.ts +2 -1
  241. package/dist/store/index.js +2 -1
  242. package/dist/store/internal.d.ts +2 -2
  243. package/dist/store/internal.js +2 -2
  244. package/dist/store/primitive-predicates.d.ts +4 -1
  245. package/dist/store/primitive-predicates.d.ts.map +1 -1
  246. package/dist/store/primitive-predicates.js +9 -1
  247. package/dist/store/primitive-predicates.js.map +1 -1
  248. package/dist/store/runtime-clients/composer-runtime-client.d.ts +1 -0
  249. package/dist/store/runtime-clients/composer-runtime-client.d.ts.map +1 -1
  250. package/dist/store/runtime-clients/composer-runtime-client.js +98 -62
  251. package/dist/store/runtime-clients/composer-runtime-client.js.map +1 -1
  252. package/dist/store/runtime-clients/message-runtime-client.d.ts +1 -0
  253. package/dist/store/runtime-clients/message-runtime-client.d.ts.map +1 -1
  254. package/dist/store/runtime-clients/message-runtime-client.js +221 -144
  255. package/dist/store/runtime-clients/message-runtime-client.js.map +1 -1
  256. package/dist/store/runtime-clients/thread-list-item-runtime-client.d.ts.map +1 -1
  257. package/dist/store/runtime-clients/thread-list-item-runtime-client.js +2 -2
  258. package/dist/store/runtime-clients/thread-list-item-runtime-client.js.map +1 -1
  259. package/dist/store/runtime-clients/thread-list-runtime-client.d.ts.map +1 -1
  260. package/dist/store/runtime-clients/thread-list-runtime-client.js +128 -107
  261. package/dist/store/runtime-clients/thread-list-runtime-client.js.map +1 -1
  262. package/dist/store/runtime-clients/thread-runtime-client.d.ts.map +1 -1
  263. package/dist/store/runtime-clients/thread-runtime-client.js +194 -122
  264. package/dist/store/runtime-clients/thread-runtime-client.js.map +1 -1
  265. package/dist/store/runtime-clients/useSubscribable.d.ts +3 -1
  266. package/dist/store/runtime-clients/useSubscribable.d.ts.map +1 -1
  267. package/dist/store/runtime-clients/useSubscribable.js +1 -1
  268. package/dist/store/runtime-clients/useSubscribable.js.map +1 -1
  269. package/dist/store/scopes/composer.d.ts +10 -2
  270. package/dist/store/scopes/composer.d.ts.map +1 -1
  271. package/dist/store/scopes/message.d.ts +30 -1
  272. package/dist/store/scopes/message.d.ts.map +1 -1
  273. package/dist/store/scopes/part.d.ts +6 -3
  274. package/dist/store/scopes/part.d.ts.map +1 -1
  275. package/dist/store/scopes/thread-list-item.d.ts +2 -2
  276. package/dist/store/scopes/thread-list-item.d.ts.map +1 -1
  277. package/dist/store/scopes/thread.d.ts +8 -0
  278. package/dist/store/scopes/thread.d.ts.map +1 -1
  279. package/dist/store/scopes/threads.d.ts +1 -0
  280. package/dist/store/scopes/threads.d.ts.map +1 -1
  281. package/dist/subscribable/subscribable.d.ts +8 -1
  282. package/dist/subscribable/subscribable.d.ts.map +1 -1
  283. package/dist/subscribable/subscribable.js +24 -16
  284. package/dist/subscribable/subscribable.js.map +1 -1
  285. package/dist/types/attachment.d.ts +2 -1
  286. package/dist/types/attachment.d.ts.map +1 -1
  287. package/dist/types/attachment.js +2 -1
  288. package/dist/types/attachment.js.map +1 -1
  289. package/dist/types/message.d.ts +40 -1
  290. package/dist/types/message.d.ts.map +1 -1
  291. package/dist/types/message.js +7 -1
  292. package/dist/types/message.js.map +1 -1
  293. package/dist/utils/abortable-iterable.d.ts +19 -0
  294. package/dist/utils/abortable-iterable.d.ts.map +1 -0
  295. package/dist/utils/abortable-iterable.js +72 -0
  296. package/dist/utils/abortable-iterable.js.map +1 -0
  297. package/dist/utils/composite-context-provider.d.ts.map +1 -1
  298. package/dist/utils/composite-context-provider.js +20 -2
  299. package/dist/utils/composite-context-provider.js.map +1 -1
  300. package/dist/utils/record.d.ts +11 -0
  301. package/dist/utils/record.d.ts.map +1 -0
  302. package/dist/utils/record.js +12 -0
  303. package/dist/utils/record.js.map +1 -0
  304. package/package.json +12 -17
  305. package/src/adapters/attachment.test.ts +50 -1
  306. package/src/adapters/attachment.ts +7 -4
  307. package/src/index.ts +7 -1
  308. package/src/internal.ts +12 -0
  309. package/src/model-context/frame/host.test.ts +24 -0
  310. package/src/model-context/frame/host.ts +10 -4
  311. package/src/model-context/frame/provider.test.ts +288 -1
  312. package/src/model-context/frame/provider.ts +148 -30
  313. package/src/model-context/interactable-composer-metadata.test.ts +28 -0
  314. package/src/model-context/interactable-composer-metadata.ts +9 -3
  315. package/src/model-context/registry.test.ts +12 -0
  316. package/src/model-context/registry.ts +3 -2
  317. package/src/model-context/tool-call-text.test.ts +67 -0
  318. package/src/model-context/tool-call-text.ts +67 -0
  319. package/src/model-context/types.test.ts +25 -0
  320. package/src/model-context/types.ts +9 -4
  321. package/src/react/AssistantRuntimeProvider.test.tsx +66 -2
  322. package/src/react/RuntimeAdapter.test.ts +34 -0
  323. package/src/react/adapters/LocalStorageThreadListAdapter.tsx +1 -3
  324. package/src/react/client/InMemoryThreadList.ts +1 -0
  325. package/src/react/client/Interactables.test.ts +28 -0
  326. package/src/react/client/Interactables.ts +15 -13
  327. package/src/react/client/RemoteThreadList.background.test.ts +526 -0
  328. package/src/react/client/RemoteThreadList.concurrent.test.tsx +315 -0
  329. package/src/react/client/RemoteThreadList.test.ts +617 -3
  330. package/src/react/client/RemoteThreadList.ts +433 -167
  331. package/src/react/client/Tools.ts +29 -12
  332. package/src/react/index.ts +4 -0
  333. package/src/react/interactables-legacy/Interactables.ts +16 -16
  334. package/src/react/interactables-shared/useInteractablePersistenceQueue.ts +9 -10
  335. package/src/react/model-context/makeAssistantTool.ts +2 -2
  336. package/src/react/model-context/modelContextHooks.test.tsx +131 -0
  337. package/src/react/model-context/toolbox.test.ts +9 -9
  338. package/src/react/model-context/toolbox.ts +9 -36
  339. package/src/react/model-context/useAssistantContext.ts +4 -2
  340. package/src/react/model-context/useAssistantTool.ts +2 -2
  341. package/src/react/model-context/useAuiToolOverrides.ts +4 -2
  342. package/src/react/model-context/useInlineRender.test.tsx +48 -0
  343. package/src/react/model-context/useInlineRender.ts +14 -10
  344. package/src/react/model-context/useToolArgsStatus.test.tsx +30 -0
  345. package/src/react/model-context/useToolArgsStatus.ts +2 -1
  346. package/src/react/primitive-hooks/useMessageError.ts +2 -19
  347. package/src/react/primitive-hooks/useThreadListLoadMore.ts +2 -3
  348. package/src/react/primitives/composer/ComposerAttachments.test.tsx +76 -0
  349. package/src/react/primitives/composer/ComposerAttachments.tsx +9 -4
  350. package/src/react/primitives/composer/ComposerQueue.test.tsx +72 -0
  351. package/src/react/primitives/composer/ComposerQueue.tsx +7 -4
  352. package/src/react/primitives/message/MessageAttachments.test.tsx +107 -37
  353. package/src/react/primitives/message/MessageAttachments.tsx +10 -7
  354. package/src/react/primitives/message/MessageGroupedParts.tsx +2 -2
  355. package/src/react/primitives/message/MessageParts.tsx +3 -3
  356. package/src/react/primitives/threadList/ThreadListItems.test.tsx +99 -0
  357. package/src/react/primitives/threadList/ThreadListItems.tsx +8 -5
  358. package/src/react/runtimes/RemoteThreadListHookInstanceManager.destroy.test.tsx +172 -0
  359. package/src/react/runtimes/RemoteThreadListHookInstanceManager.running.test.tsx +127 -1
  360. package/src/react/runtimes/RemoteThreadListHookInstanceManager.test.tsx +7 -1
  361. package/src/react/runtimes/RemoteThreadListHookInstanceManager.tsx +99 -35
  362. package/src/react/runtimes/RemoteThreadListThreadListRuntimeCore.load-error.test.ts +132 -0
  363. package/src/react/runtimes/RemoteThreadListThreadListRuntimeCore.switch-delete.test.ts +274 -0
  364. package/src/react/runtimes/RemoteThreadListThreadListRuntimeCore.tsx +223 -152
  365. package/src/react/runtimes/RemoteThreadResource.test.ts +9 -0
  366. package/src/react/runtimes/RemoteThreadResource.ts +20 -16
  367. package/src/react/runtimes/assistant-transport/transport-scheduling.test.ts +51 -2
  368. package/src/react/runtimes/assistant-transport/useLatestRef.ts +2 -2
  369. package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.test.ts +45 -0
  370. package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.test.tsx +546 -5
  371. package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.ts +546 -103
  372. package/src/react/runtimes/cloud/auiV0.ts +5 -0
  373. package/src/react/runtimes/cloud/createCloudThreadListAdapter.test.ts +103 -0
  374. package/src/react/runtimes/cloud/createCloudThreadListAdapter.ts +237 -0
  375. package/src/react/runtimes/cloud/useCloudThreadListAdapter.test.tsx +90 -1
  376. package/src/react/runtimes/cloud/useCloudThreadListAdapter.tsx +33 -238
  377. package/src/react/runtimes/external-message-converter.ts +1 -0
  378. package/src/react/runtimes/useExternalStoreRuntime.lifecycle.test.tsx +41 -0
  379. package/src/react/runtimes/useExternalStoreRuntime.ts +10 -4
  380. package/src/react/runtimes/useLocalRuntime.test.tsx +20 -0
  381. package/src/react/runtimes/useRemoteThreadListRuntime.concurrent.test.tsx +216 -0
  382. package/src/react/runtimes/useRemoteThreadListRuntime.load-error.test.tsx +108 -0
  383. package/src/react/runtimes/useRemoteThreadListRuntime.ts +57 -6
  384. package/src/react/types/MessagePartComponentTypes.ts +7 -3
  385. package/src/react/types/scopes/tools.ts +2 -0
  386. package/src/runtime/api/message-part-runtime.ts +3 -3
  387. package/src/runtime/api/message-runtime.test.ts +75 -0
  388. package/src/runtime/api/thread-list-item-runtime.ts +10 -3
  389. package/src/runtime/api/thread-list-runtime.ts +46 -12
  390. package/src/runtime/api/thread-runtime.test.ts +78 -0
  391. package/src/runtime/api/thread-runtime.ts +8 -1
  392. package/src/runtime/base/base-composer-runtime-core.ts +36 -8
  393. package/src/runtime/base/base-thread-runtime-core.test.ts +149 -4
  394. package/src/runtime/base/base-thread-runtime-core.ts +16 -8
  395. package/src/runtime/interfaces/composer-runtime-core.ts +10 -5
  396. package/src/runtime/interfaces/thread-list-runtime-core.ts +25 -2
  397. package/src/runtime/interfaces/thread-runtime-core.ts +17 -1
  398. package/src/runtime/utils/auto-status.test.ts +57 -0
  399. package/src/runtime/utils/auto-status.ts +13 -1
  400. package/src/runtime/utils/external-message-conversion.test.ts +165 -0
  401. package/src/runtime/utils/external-message-conversion.ts +18 -2
  402. package/src/runtime/utils/message-repository.ts +29 -17
  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 +3 -11
  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 +18 -13
  410. package/src/runtimes/external-store/external-store-thread-runtime-core.ts +56 -33
  411. package/src/runtimes/inert/InertThreadRuntimeCore.ts +1 -1
  412. package/src/runtimes/internal.ts +1 -0
  413. package/src/runtimes/local/local-thread-runtime-core.test.ts +29 -3
  414. package/src/runtimes/local/local-thread-runtime-core.ts +5 -1
  415. package/src/runtimes/remote-thread-list/remote-thread-state.test.ts +238 -0
  416. package/src/runtimes/remote-thread-list/remote-thread-state.ts +223 -44
  417. package/src/runtimes/remote-thread-list/title-generation.test.ts +602 -0
  418. package/src/runtimes/remote-thread-list/title-generation.ts +329 -0
  419. package/src/runtimes/remote-thread-list/title.test.ts +62 -0
  420. package/src/runtimes/remote-thread-list/title.ts +16 -0
  421. package/src/runtimes/remote-thread-list/types.ts +16 -5
  422. package/src/runtimes/tool-invocations/EDGE_CASES.md +110 -5
  423. package/src/runtimes/tool-invocations/ToolInvocationTracker.test.ts +750 -136
  424. package/src/runtimes/tool-invocations/ToolInvocationTracker.ts +144 -91
  425. package/src/store/clients/external-thread.ts +62 -11
  426. package/src/store/clients/model-context-client.ts +5 -1
  427. package/src/store/clients/single-thread-list.ts +1 -0
  428. package/src/store/clients/suggestions.test.ts +195 -1
  429. package/src/store/clients/suggestions.ts +45 -34
  430. package/src/store/index.ts +4 -0
  431. package/src/store/internal.ts +2 -0
  432. package/src/store/primitive-predicates.ts +35 -0
  433. package/src/store/runtime-clients/composer-runtime-client.ts +42 -9
  434. package/src/store/runtime-clients/message-runtime-client.ts +54 -6
  435. package/src/store/runtime-clients/thread-list-item-runtime-client.ts +4 -2
  436. package/src/store/runtime-clients/thread-list-runtime-client.ts +17 -2
  437. package/src/store/runtime-clients/thread-runtime-client.ts +58 -10
  438. package/src/store/runtime-clients/useSubscribable.ts +8 -2
  439. package/src/store/scopes/composer.ts +16 -4
  440. package/src/store/scopes/message.ts +14 -0
  441. package/src/store/scopes/part.ts +6 -3
  442. package/src/store/scopes/thread-list-item.ts +5 -2
  443. package/src/store/scopes/thread.ts +4 -0
  444. package/src/store/scopes/threads.ts +1 -0
  445. package/src/subscribable/subscribable.test.ts +50 -0
  446. package/src/subscribable/subscribable.ts +33 -21
  447. package/src/tests/MessageRepository.test.ts +116 -0
  448. package/src/tests/OptimisticState-list-race.test.ts +1 -0
  449. package/src/tests/RemoteThreadListThreadListRuntimeCore-initialize-adapter-change.test.ts +38 -0
  450. package/src/tests/RemoteThreadListThreadListRuntimeCore-load-race.test.ts +158 -0
  451. package/src/tests/RemoteThreadListThreadListRuntimeCore-loadMore.test.ts +78 -0
  452. package/src/tests/RemoteThreadListThreadListRuntimeCore-reload.test.ts +52 -0
  453. package/src/tests/RemoteThreadListThreadListRuntimeCore-title-race.test.ts +89 -0
  454. package/src/tests/attachment-adapters.test.ts +14 -0
  455. package/src/tests/base-composer-runtime-core.test.ts +39 -0
  456. package/src/tests/external-store-optimistic-id.test.ts +79 -0
  457. package/src/tests/external-store-repository-instance.test.ts +166 -0
  458. package/src/tests/external-store-thread-list-runtime-core.test.ts +83 -0
  459. package/src/tests/external-store-thread-runtime-core-adapter.test.ts +49 -34
  460. package/src/tests/external-thread-attachments.test.tsx +224 -1
  461. package/src/tests/external-thread-parity.test.tsx +23 -0
  462. package/src/tests/remote-thread-list-adapter-switch.test.ts +66 -1
  463. package/src/tests/remote-thread-list-isLoading.test.ts +1 -0
  464. package/src/tests/thread-list-runtime-getLoadThreadsPromise.test.ts +1 -0
  465. package/src/tests/thread-switch-events.test.tsx +475 -2
  466. package/src/tests/tools-scope-migration.test.tsx +28 -0
  467. package/src/types/attachment.ts +4 -0
  468. package/src/types/message.ts +50 -2
  469. package/src/utils/abortable-iterable.test.ts +213 -0
  470. package/src/utils/abortable-iterable.ts +93 -0
  471. package/src/utils/composite-context-provider.test.ts +25 -0
  472. package/src/utils/composite-context-provider.ts +23 -2
  473. package/src/utils/record.test.ts +48 -0
  474. package/src/utils/record.ts +11 -0
@@ -20,11 +20,16 @@ var AssistantFrameProvider = class AssistantFrameProvider {
20
20
  _targetOrigin;
21
21
  _strictRegistrations = 0;
22
22
  _wildcardRegistrations = 0;
23
+ _startupTimer;
24
+ _disposed = false;
23
25
  constructor(targetOrigin = getDefaultTargetOrigin()) {
24
26
  this._targetOrigin = targetOrigin;
25
27
  this.handleMessage = this.handleMessage.bind(this);
26
28
  window.addEventListener("message", this.handleMessage);
27
- setTimeout(() => this.broadcastUpdate(), 0);
29
+ this._startupTimer = setTimeout(() => {
30
+ this._startupTimer = void 0;
31
+ this.broadcastUpdate();
32
+ }, 0);
28
33
  }
29
34
  static getInstance(targetOrigin) {
30
35
  if (!AssistantFrameProvider._instance) AssistantFrameProvider._instance = new AssistantFrameProvider(targetOrigin);
@@ -57,18 +62,22 @@ var AssistantFrameProvider = class AssistantFrameProvider {
57
62
  });
58
63
  break;
59
64
  case "tool-call":
60
- this.handleToolCall(message, event);
65
+ this.handleToolCall(message, event).catch((error) => {
66
+ console.error("[assistant-ui] AssistantFrame tool call failed.", error);
67
+ });
61
68
  break;
62
69
  case "tool-cancel": this.cancelToolCall(message.id);
63
70
  }
64
71
  }
65
72
  async handleToolCall(message, event) {
66
- const tool = this.getModelContext().tools?.[message.toolName];
73
+ const resolvedTool = this.getTool(message.toolName);
74
+ const tool = resolvedTool?.tool;
67
75
  const abortController = new AbortController();
68
76
  this._activeToolCalls.get(message.id)?.abortController.abort();
69
77
  const activeCall = {
70
78
  abortController,
71
- event
79
+ event,
80
+ provider: resolvedTool?.provider
72
81
  };
73
82
  this._activeToolCalls.set(message.id, activeCall);
74
83
  let result;
@@ -87,11 +96,21 @@ var AssistantFrameProvider = class AssistantFrameProvider {
87
96
  }
88
97
  if (this._activeToolCalls.get(message.id) !== activeCall) return;
89
98
  this._activeToolCalls.delete(message.id);
90
- this.sendMessage(event, {
91
- type: "tool-result",
92
- id: message.id,
93
- ...error !== void 0 ? { error } : { result }
94
- });
99
+ try {
100
+ this.sendMessage(event, {
101
+ type: "tool-result",
102
+ id: message.id,
103
+ ...error !== void 0 ? { error } : { result }
104
+ });
105
+ } catch (sendError) {
106
+ if (error !== void 0) throw sendError;
107
+ console.error("[assistant-ui] AssistantFrame tool result could not be sent.", sendError);
108
+ this.sendMessage(event, {
109
+ type: "tool-result",
110
+ id: message.id,
111
+ error: "Tool result could not be sent across the frame boundary"
112
+ });
113
+ }
95
114
  }
96
115
  cancelToolCall(id) {
97
116
  const activeCall = this._activeToolCalls.get(id);
@@ -99,14 +118,51 @@ var AssistantFrameProvider = class AssistantFrameProvider {
99
118
  this._activeToolCalls.delete(id);
100
119
  activeCall.abortController.abort();
101
120
  }
121
+ cancelToolCallsForProvider(provider) {
122
+ const matchingCalls = Array.from(this._activeToolCalls).filter(([, activeCall]) => activeCall.provider === provider);
123
+ for (const [id, activeCall] of matchingCalls) {
124
+ this._activeToolCalls.delete(id);
125
+ activeCall.abortController.abort();
126
+ }
127
+ let sendFailed = false;
128
+ let sendError;
129
+ for (const [id, activeCall] of matchingCalls) try {
130
+ this.sendMessage(activeCall.event, {
131
+ type: "tool-result",
132
+ id,
133
+ error: "AssistantFrame tool provider has been removed"
134
+ });
135
+ } catch (error) {
136
+ if (sendFailed) console.error(error);
137
+ else {
138
+ sendFailed = true;
139
+ sendError = error;
140
+ }
141
+ }
142
+ if (sendFailed) throw sendError;
143
+ }
102
144
  sendMessage(event, message) {
103
145
  event.source?.postMessage({
104
146
  channel: FRAME_MESSAGE_CHANNEL,
105
147
  message
106
148
  }, { targetOrigin: event.origin });
107
149
  }
150
+ getProviders() {
151
+ return Array.from(new Set(this._providers.values()));
152
+ }
153
+ getTool(toolName) {
154
+ let resolved;
155
+ for (const provider of this.getProviders()) {
156
+ const tool = provider.getModelContext().tools?.[toolName];
157
+ if (tool) resolved = {
158
+ provider,
159
+ tool
160
+ };
161
+ }
162
+ return resolved;
163
+ }
108
164
  getModelContext() {
109
- return Array.from(new Set(this._providers.values())).map((p) => p.getModelContext()).reduce((merged, context) => ({
165
+ return this.getProviders().map((provider) => provider.getModelContext()).reduce((merged, context) => ({
110
166
  system: context.system ? merged.system ? `${merged.system}\n\n${context.system}` : context.system : merged.system,
111
167
  tools: {
112
168
  ...merged.tools || {},
@@ -115,6 +171,7 @@ var AssistantFrameProvider = class AssistantFrameProvider {
115
171
  }), {});
116
172
  }
117
173
  broadcastUpdate() {
174
+ if (this._disposed) return;
118
175
  if (window.parent && window.parent !== window) {
119
176
  const updateMessage = {
120
177
  type: "model-context-update",
@@ -127,6 +184,7 @@ var AssistantFrameProvider = class AssistantFrameProvider {
127
184
  }
128
185
  }
129
186
  removeProvider(id, origin) {
187
+ const provider = this._providers.get(id);
130
188
  this._providers.delete(id);
131
189
  const unsubscribe = this._providerUnsubscribes.get(id);
132
190
  this._providerUnsubscribes.delete(id);
@@ -137,7 +195,10 @@ var AssistantFrameProvider = class AssistantFrameProvider {
137
195
  this._strictRegistrations -= 1;
138
196
  if (this._strictRegistrations === 0) this._targetOrigin = this._wildcardRegistrations > 0 ? "*" : getDefaultTargetOrigin();
139
197
  }
140
- return unsubscribe;
198
+ return {
199
+ unsubscribe,
200
+ removedProvider: provider && !this.getProviders().includes(provider) ? provider : void 0
201
+ };
141
202
  }
142
203
  static addModelContextProvider(provider, targetOrigin) {
143
204
  const origin = targetOrigin ?? getDefaultTargetOrigin();
@@ -151,7 +212,12 @@ var AssistantFrameProvider = class AssistantFrameProvider {
151
212
  if (unsubscribe) instance._providerUnsubscribes.set(id, unsubscribe);
152
213
  instance.broadcastUpdate();
153
214
  } catch (error) {
154
- const unsubscribe = instance.removeProvider(id, origin);
215
+ const { unsubscribe, removedProvider } = instance.removeProvider(id, origin);
216
+ try {
217
+ if (removedProvider) instance.cancelToolCallsForProvider(removedProvider);
218
+ } catch (cancelError) {
219
+ console.error(cancelError);
220
+ }
155
221
  try {
156
222
  unsubscribe?.();
157
223
  } catch (unsubscribeError) {
@@ -168,27 +234,34 @@ var AssistantFrameProvider = class AssistantFrameProvider {
168
234
  return () => {
169
235
  if (released) return;
170
236
  released = true;
171
- const unsubscribe = instance.removeProvider(id, origin);
172
- let unsubscribeFailed = false;
173
- let unsubscribeError;
174
- try {
175
- unsubscribe?.();
176
- } catch (error) {
177
- unsubscribeFailed = true;
178
- unsubscribeError = error;
179
- }
180
- try {
181
- instance.broadcastUpdate();
182
- } catch (error) {
183
- if (!unsubscribeFailed) throw error;
184
- console.error(error);
185
- }
186
- if (unsubscribeFailed) throw unsubscribeError;
237
+ const { unsubscribe, removedProvider } = instance.removeProvider(id, origin);
238
+ let cleanupFailed = false;
239
+ let cleanupError;
240
+ const runCleanup = (cleanup) => {
241
+ try {
242
+ cleanup();
243
+ } catch (error) {
244
+ if (cleanupFailed) console.error(error);
245
+ else {
246
+ cleanupFailed = true;
247
+ cleanupError = error;
248
+ }
249
+ }
250
+ };
251
+ if (removedProvider) runCleanup(() => instance.cancelToolCallsForProvider(removedProvider));
252
+ if (unsubscribe) runCleanup(unsubscribe);
253
+ runCleanup(() => instance.broadcastUpdate());
254
+ if (cleanupFailed) throw cleanupError;
187
255
  };
188
256
  }
189
257
  static dispose() {
190
258
  if (AssistantFrameProvider._instance) {
191
259
  const instance = AssistantFrameProvider._instance;
260
+ instance._disposed = true;
261
+ if (instance._startupTimer !== void 0) {
262
+ clearTimeout(instance._startupTimer);
263
+ instance._startupTimer = void 0;
264
+ }
192
265
  window.removeEventListener("message", instance.handleMessage);
193
266
  let cleanupFailed = false;
194
267
  let cleanupError;
@@ -1 +1 @@
1
- {"version":3,"file":"provider.js","names":["ModelContextProvider","ModelContext","Unsubscribe","Tool","toJSONSchema","FrameMessage","FRAME_MESSAGE_CHANNEL","SerializedModelContext","SerializedTool","serializeTool","tool","description","parameters","undefined","disabled","type","serializeModelContext","context","system","tools","Object","fromEntries","entries","map","name","getDefaultTargetOrigin","window","location","origin","AssistantFrameProvider","_instance","_providers","Map","_providerUnsubscribes","_activeToolCalls","abortController","AbortController","event","MessageEvent","_targetOrigin","_strictRegistrations","_wildcardRegistrations","constructor","targetOrigin","handleMessage","bind","addEventListener","setTimeout","broadcastUpdate","getInstance","reconcileTargetOrigin","size","Error","source","parent","data","channel","message","sendMessage","getModelContext","handleToolCall","cancelToolCall","id","Extract","toolName","get","abort","activeCall","set","result","error","execute","args","toolCallId","abortSignal","signal","human","e","String","delete","postMessage","contexts","Array","from","Set","values","p","reduce","merged","updateMessage","removeProvider","unsubscribe","addModelContextProvider","provider","instance","Symbol","subscribe","unsubscribeError","console","broadcastError","released","unsubscribeFailed","dispose","removeEventListener","cleanupFailed","cleanupError","runCleanup","cleanup","forEach","clear"],"sources":["../../../src/model-context/frame/provider.ts"],"sourcesContent":["import type { ModelContextProvider, ModelContext } from \"../types\";\nimport type { Unsubscribe } from \"../../types/unsubscribe\";\nimport { type Tool, toJSONSchema } from \"assistant-stream\";\nimport {\n type FrameMessage,\n FRAME_MESSAGE_CHANNEL,\n type SerializedModelContext,\n type SerializedTool,\n} from \"./types\";\n\nconst serializeTool = (tool: Tool<any, any>): SerializedTool => ({\n ...(tool.description && { description: tool.description }),\n parameters: tool.parameters ? toJSONSchema(tool.parameters) : undefined,\n ...(tool.disabled !== undefined && { disabled: tool.disabled }),\n ...(tool.type && { type: tool.type }),\n});\n\nconst serializeModelContext = (\n context: ModelContext,\n): SerializedModelContext => ({\n ...(context.system !== undefined && { system: context.system }),\n ...(context.tools && {\n tools: Object.fromEntries(\n Object.entries(context.tools).map(([name, tool]) => [\n name,\n serializeTool(tool),\n ]),\n ),\n }),\n});\n\nconst getDefaultTargetOrigin = () => window.location.origin;\n\nexport class AssistantFrameProvider {\n private static _instance: AssistantFrameProvider | null = null;\n\n private _providers = new Map<symbol, ModelContextProvider>();\n private _providerUnsubscribes = new Map<symbol, Unsubscribe | undefined>();\n private _activeToolCalls = new Map<\n string,\n { abortController: AbortController; event: MessageEvent }\n >();\n private _targetOrigin: string;\n private _strictRegistrations = 0;\n private _wildcardRegistrations = 0;\n\n private constructor(targetOrigin: string = getDefaultTargetOrigin()) {\n this._targetOrigin = targetOrigin;\n this.handleMessage = this.handleMessage.bind(this);\n window.addEventListener(\"message\", this.handleMessage);\n\n setTimeout(() => this.broadcastUpdate(), 0);\n }\n\n private static getInstance(targetOrigin?: string): AssistantFrameProvider {\n if (!AssistantFrameProvider._instance) {\n AssistantFrameProvider._instance = new AssistantFrameProvider(\n targetOrigin,\n );\n } else {\n AssistantFrameProvider._instance.reconcileTargetOrigin(targetOrigin);\n }\n return AssistantFrameProvider._instance;\n }\n\n private reconcileTargetOrigin(\n targetOrigin: string = getDefaultTargetOrigin(),\n ) {\n if (targetOrigin === this._targetOrigin) return;\n\n if (this._providers.size === 0) {\n this._targetOrigin = targetOrigin;\n return;\n }\n\n if (targetOrigin === \"*\") return;\n\n if (this._targetOrigin === \"*\") {\n this._targetOrigin = targetOrigin;\n return;\n }\n\n throw new Error(\n `AssistantFrameProvider cannot register conflicting target origins: \"${this._targetOrigin}\" and \"${targetOrigin}\"`,\n );\n }\n\n private handleMessage(event: MessageEvent) {\n if (this._targetOrigin !== \"*\" && event.origin !== this._targetOrigin)\n return;\n if (event.source !== window.parent) return;\n if (event.data?.channel !== FRAME_MESSAGE_CHANNEL) return;\n\n const message = event.data.message as FrameMessage;\n\n switch (message.type) {\n case \"model-context-request\":\n this.sendMessage(event, {\n type: \"model-context-update\",\n context: serializeModelContext(this.getModelContext()),\n });\n break;\n\n case \"tool-call\":\n this.handleToolCall(message, event);\n break;\n\n case \"tool-cancel\":\n this.cancelToolCall(message.id);\n break;\n }\n }\n\n private async handleToolCall(\n message: Extract<FrameMessage, { type: \"tool-call\" }>,\n event: MessageEvent,\n ) {\n const tool = this.getModelContext().tools?.[message.toolName];\n const abortController = new AbortController();\n this._activeToolCalls.get(message.id)?.abortController.abort();\n const activeCall = { abortController, event };\n this._activeToolCalls.set(message.id, activeCall);\n\n let result: any;\n let error: string | undefined;\n\n if (!tool) {\n error = `Tool \"${message.toolName}\" not found`;\n } else {\n try {\n result = tool.execute\n ? await tool.execute(message.args, {\n toolCallId: message.id,\n abortSignal: abortController.signal,\n human: async () => {\n throw new Error(\n \"Tool human input is not supported in frame context\",\n );\n },\n })\n : undefined;\n } catch (e) {\n error = e instanceof Error ? e.message : String(e);\n }\n }\n\n if (this._activeToolCalls.get(message.id) !== activeCall) return;\n this._activeToolCalls.delete(message.id);\n\n this.sendMessage(event, {\n type: \"tool-result\",\n id: message.id,\n ...(error !== undefined ? { error } : { result }),\n });\n }\n\n private cancelToolCall(id: string) {\n const activeCall = this._activeToolCalls.get(id);\n if (!activeCall) return;\n this._activeToolCalls.delete(id);\n activeCall.abortController.abort();\n }\n\n private sendMessage(event: MessageEvent, message: FrameMessage) {\n event.source?.postMessage(\n { channel: FRAME_MESSAGE_CHANNEL, message },\n { targetOrigin: event.origin },\n );\n }\n\n private getModelContext(): ModelContext {\n const contexts = Array.from(new Set(this._providers.values())).map((p) =>\n p.getModelContext(),\n );\n\n return contexts.reduce(\n (merged, context) => ({\n system: context.system\n ? merged.system\n ? `${merged.system}\\n\\n${context.system}`\n : context.system\n : merged.system,\n tools: { ...(merged.tools || {}), ...(context.tools || {}) },\n }),\n {} as ModelContext,\n );\n }\n\n private broadcastUpdate() {\n if (window.parent && window.parent !== window) {\n const updateMessage: FrameMessage = {\n type: \"model-context-update\",\n context: serializeModelContext(this.getModelContext()),\n };\n\n window.parent.postMessage(\n { channel: FRAME_MESSAGE_CHANNEL, message: updateMessage },\n this._targetOrigin,\n );\n }\n }\n\n private removeProvider(id: symbol, origin: string): Unsubscribe | undefined {\n this._providers.delete(id);\n const unsubscribe = this._providerUnsubscribes.get(id);\n this._providerUnsubscribes.delete(id);\n if (origin === \"*\") {\n this._wildcardRegistrations -= 1;\n if (\n this._wildcardRegistrations === 0 &&\n this._strictRegistrations === 0\n ) {\n this._targetOrigin = getDefaultTargetOrigin();\n }\n } else {\n this._strictRegistrations -= 1;\n if (this._strictRegistrations === 0) {\n this._targetOrigin =\n this._wildcardRegistrations > 0 ? \"*\" : getDefaultTargetOrigin();\n }\n }\n return unsubscribe;\n }\n\n static addModelContextProvider(\n provider: ModelContextProvider,\n targetOrigin?: string,\n ): Unsubscribe {\n const origin = targetOrigin ?? getDefaultTargetOrigin();\n const instance = AssistantFrameProvider.getInstance(origin);\n const id = Symbol();\n instance._providers.set(id, provider);\n if (origin === \"*\") {\n instance._wildcardRegistrations += 1;\n } else {\n instance._strictRegistrations += 1;\n }\n\n try {\n const unsubscribe = provider.subscribe?.(() =>\n instance.broadcastUpdate(),\n );\n if (unsubscribe) {\n instance._providerUnsubscribes.set(id, unsubscribe);\n }\n\n instance.broadcastUpdate();\n } catch (error) {\n const unsubscribe = instance.removeProvider(id, origin);\n // Rollback failures must not replace the registration error.\n try {\n unsubscribe?.();\n } catch (unsubscribeError) {\n console.error(unsubscribeError);\n }\n try {\n instance.broadcastUpdate();\n } catch (broadcastError) {\n console.error(broadcastError);\n }\n throw error;\n }\n\n let released = false;\n return () => {\n if (released) return;\n released = true;\n const unsubscribe = instance.removeProvider(id, origin);\n let unsubscribeFailed = false;\n let unsubscribeError: unknown;\n try {\n unsubscribe?.();\n } catch (error) {\n unsubscribeFailed = true;\n unsubscribeError = error;\n }\n try {\n instance.broadcastUpdate();\n } catch (error) {\n if (!unsubscribeFailed) throw error;\n console.error(error);\n }\n if (unsubscribeFailed) throw unsubscribeError;\n };\n }\n\n static dispose() {\n if (AssistantFrameProvider._instance) {\n const instance = AssistantFrameProvider._instance;\n window.removeEventListener(\"message\", instance.handleMessage);\n\n let cleanupFailed = false;\n let cleanupError: unknown;\n const runCleanup = (cleanup: () => void) => {\n try {\n cleanup();\n } catch (error) {\n if (cleanupFailed) {\n console.error(error);\n } else {\n cleanupFailed = true;\n cleanupError = error;\n }\n }\n };\n\n instance._providerUnsubscribes.forEach((unsubscribe) => {\n if (unsubscribe) runCleanup(unsubscribe);\n });\n instance._providerUnsubscribes.clear();\n instance._providers.clear();\n instance._activeToolCalls.forEach(({ abortController, event }, id) => {\n runCleanup(() => {\n abortController.abort();\n instance.sendMessage(event, {\n type: \"tool-result\",\n id,\n error: \"AssistantFrameProvider has been disposed\",\n });\n });\n });\n instance._activeToolCalls.clear();\n\n AssistantFrameProvider._instance = null;\n if (cleanupFailed) throw cleanupError;\n }\n }\n}\n"],"mappings":";;;AAUA,MAAMS,iBAAiBC,UAA0C;CAC/D,GAAIA,KAAKC,eAAe,EAAEA,aAAaD,KAAKC,YAAY;CACxDC,YAAYF,KAAKE,aAAaR,aAAaM,KAAKE,UAAU,IAAIC,KAAAA;CAC9D,GAAIH,KAAKI,aAAaD,KAAAA,KAAa,EAAEC,UAAUJ,KAAKI,SAAS;CAC7D,GAAIJ,KAAKK,QAAQ,EAAEA,MAAML,KAAKK,KAAK;AACrC;AAEA,MAAMC,yBACJC,aAC4B;CAC5B,GAAIA,QAAQC,WAAWL,KAAAA,KAAa,EAAEK,QAAQD,QAAQC,OAAO;CAC7D,GAAID,QAAQE,SAAS,EACnBA,OAAOC,OAAOC,YACZD,OAAOE,QAAQL,QAAQE,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMd,UAAU,CAClDc,MACAf,cAAcC,IAAI,CAAC,CACpB,CACH,EACF;AACF;AAEA,MAAMe,+BAA+BC,OAAOC,SAASC;AAErD,IAAaC,yBAAb,MAAaA,uBAAuB;CAClC,OAAeC,YAA2C;CAE1D,6BAAqB,IAAIE,IAAkC;CAC3D,wCAAgC,IAAIA,IAAqC;CACzE,mCAA2B,IAAIA,IAG7B;CACF;CACA,uBAA+B;CAC/B,yBAAiC;CAEjC,YAAoBW,eAAuBlB,uBAAuB,GAAG;EACnE,KAAKc,gBAAgBI;EACrB,KAAKC,gBAAgB,KAAKA,cAAcC,KAAK,IAAI;EACjDnB,OAAOoB,iBAAiB,WAAW,KAAKF,aAAa;EAErDG,iBAAiB,KAAKC,gBAAgB,GAAG,CAAC;CAC5C;CAEA,OAAeC,YAAYN,cAA+C;EACxE,IAAI,CAACd,uBAAuBC,WAC1BD,uBAAuBC,YAAY,IAAID,uBACrCc,YACF;OAEAd,uBAAuBC,UAAUoB,sBAAsBP,YAAY;EAErE,OAAOd,uBAAuBC;CAChC;CAEA,sBACEa,eAAuBlB,uBAAuB,GAC9C;EACA,IAAIkB,iBAAiB,KAAKJ,eAAe;EAEzC,IAAI,KAAKR,WAAWoB,SAAS,GAAG;GAC9B,KAAKZ,gBAAgBI;GACrB;EACF;EAEA,IAAIA,iBAAiB,KAAK;EAE1B,IAAI,KAAKJ,kBAAkB,KAAK;GAC9B,KAAKA,gBAAgBI;GACrB;EACF;EAEA,MAAM,IAAIS,MACR,uEAAuE,KAAKb,cAAa,SAAUI,aAAY,EACjH;CACF;CAEA,cAAsBN,OAAqB;EACzC,IAAI,KAAKE,kBAAkB,OAAOF,MAAMT,WAAW,KAAKW,eACtD;EACF,IAAIF,MAAMgB,WAAW3B,OAAO4B,QAAQ;EACpC,IAAIjB,MAAMkB,MAAMC,YAAAA,sBAAmC;EAEnD,MAAMC,UAAUpB,MAAMkB,KAAKE;EAE3B,QAAQA,QAAQ1C,MAAhB;GACE,KAAK;IACH,KAAK2C,YAAYrB,OAAO;KACtBtB,MAAM;KACNE,SAASD,sBAAsB,KAAK2C,gBAAgB,CAAC;IACvD,CAAC;IACD;GAEF,KAAK;IACH,KAAKC,eAAeH,SAASpB,KAAK;IAClC;GAEF,KAAK,eACH,KAAKwB,eAAeJ,QAAQK,EAAE;EAElC;CACF;CAEA,MAAcF,eACZH,SACApB,OACA;EACA,MAAM3B,OAAO,KAAKiD,gBAAgB,CAAC,CAACxC,QAAQsC,QAAQO;EACpD,MAAM7B,kBAAkB,IAAIC,gBAAgB;EAC5C,KAAKF,iBAAiB+B,IAAIR,QAAQK,EAAE,CAAC,EAAE3B,gBAAgB+B,MAAM;EAC7D,MAAMC,aAAa;GAAEhC;GAAiBE;EAAM;EAC5C,KAAKH,iBAAiBkC,IAAIX,QAAQK,IAAIK,UAAU;EAEhD,IAAIE;EACJ,IAAIC;EAEJ,IAAI,CAAC5D,MACH4D,QAAQ,SAASb,QAAQO,SAAQ;OAEjC,IAAI;GACFK,SAAS3D,KAAK6D,UACV,MAAM7D,KAAK6D,QAAQd,QAAQe,MAAM;IAC/BC,YAAYhB,QAAQK;IACpBY,aAAavC,gBAAgBwC;IAC7BC,OAAO,YAAY;KACjB,MAAM,IAAIxB,MACR,oDACF;IACF;GACF,CAAC,IACDvC,KAAAA;EACN,SAASgE,GAAG;GACVP,QAAQO,aAAazB,QAAQyB,EAAEpB,UAAUqB,OAAOD,CAAC;EACnD;EAGF,IAAI,KAAK3C,iBAAiB+B,IAAIR,QAAQK,EAAE,MAAMK,YAAY;EAC1D,KAAKjC,iBAAiB6C,OAAOtB,QAAQK,EAAE;EAEvC,KAAKJ,YAAYrB,OAAO;GACtBtB,MAAM;GACN+C,IAAIL,QAAQK;GACZ,GAAIQ,UAAUzD,KAAAA,IAAY,EAAEyD,MAAM,IAAI,EAAED,OAAO;EACjD,CAAC;CACH;CAEA,eAAuBP,IAAY;EACjC,MAAMK,aAAa,KAAKjC,iBAAiB+B,IAAIH,EAAE;EAC/C,IAAI,CAACK,YAAY;EACjB,KAAKjC,iBAAiB6C,OAAOjB,EAAE;EAC/BK,WAAWhC,gBAAgB+B,MAAM;CACnC;CAEA,YAAoB7B,OAAqBoB,SAAuB;EAC9DpB,MAAMgB,QAAQ2B,YACZ;GAAExB,SAASlD;GAAuBmD;EAAQ,GAC1C,EAAEd,cAAcN,MAAMT,OAAO,CAC/B;CACF;CAEA,kBAAwC;EAKtC,OAJiBsD,MAAMC,KAAK,IAAIC,IAAI,KAAKrD,WAAWsD,OAAO,CAAC,CAAC,CAAC,CAAC9D,KAAK+D,MAClEA,EAAE3B,gBAAgB,CAGbsB,CAAQ,CAACM,QACbC,QAAQvE,aAAa;GACpBC,QAAQD,QAAQC,SACZsE,OAAOtE,SACL,GAAGsE,OAAOtE,OAAM,MAAOD,QAAQC,WAC/BD,QAAQC,SACVsE,OAAOtE;GACXC,OAAO;IAAE,GAAIqE,OAAOrE,SAAS,CAAC;IAAI,GAAIF,QAAQE,SAAS,CAAC;GAAG;EAC7D,IACA,CAAC,CACH;CACF;CAEA,kBAA0B;EACxB,IAAIO,OAAO4B,UAAU5B,OAAO4B,WAAW5B,QAAQ;GAC7C,MAAM+D,gBAA8B;IAClC1E,MAAM;IACNE,SAASD,sBAAsB,KAAK2C,gBAAgB,CAAC;GACvD;GAEAjC,OAAO4B,OAAO0B,YACZ;IAAExB,SAASlD;IAAuBmD,SAASgC;GAAc,GACzD,KAAKlD,aACP;EACF;CACF;CAEA,eAAuBuB,IAAYlC,QAAyC;EAC1E,KAAKG,WAAWgD,OAAOjB,EAAE;EACzB,MAAM6B,cAAc,KAAK1D,sBAAsBgC,IAAIH,EAAE;EACrD,KAAK7B,sBAAsB8C,OAAOjB,EAAE;EACpC,IAAIlC,WAAW,KAAK;GAClB,KAAKa,0BAA0B;GAC/B,IACE,KAAKA,2BAA2B,KAChC,KAAKD,yBAAyB,GAE9B,KAAKD,gBAAgBd,uBAAuB;EAEhD,OAAO;GACL,KAAKe,wBAAwB;GAC7B,IAAI,KAAKA,yBAAyB,GAChC,KAAKD,gBACH,KAAKE,yBAAyB,IAAI,MAAMhB,uBAAuB;EAErE;EACA,OAAOkE;CACT;CAEA,OAAOC,wBACLC,UACAlD,cACa;EACb,MAAMf,SAASe,gBAAgBlB,uBAAuB;EACtD,MAAMqE,WAAWjE,uBAAuBoB,YAAYrB,MAAM;EAC1D,MAAMkC,KAAKiC,OAAO;EAClBD,SAAS/D,WAAWqC,IAAIN,IAAI+B,QAAQ;EACpC,IAAIjE,WAAW,KACbkE,SAASrD,0BAA0B;OAEnCqD,SAAStD,wBAAwB;EAGnC,IAAI;GACF,MAAMmD,cAAcE,SAASG,kBAC3BF,SAAS9C,gBAAgB,CAC3B;GACA,IAAI2C,aACFG,SAAS7D,sBAAsBmC,IAAIN,IAAI6B,WAAW;GAGpDG,SAAS9C,gBAAgB;EAC3B,SAASsB,OAAO;GACd,MAAMqB,cAAcG,SAASJ,eAAe5B,IAAIlC,MAAM;GAEtD,IAAI;IACF+D,cAAc;GAChB,SAASM,kBAAkB;IACzBC,QAAQ5B,MAAM2B,gBAAgB;GAChC;GACA,IAAI;IACFH,SAAS9C,gBAAgB;GAC3B,SAASmD,gBAAgB;IACvBD,QAAQ5B,MAAM6B,cAAc;GAC9B;GACA,MAAM7B;EACR;EAEA,IAAI8B,WAAW;EACf,aAAa;GACX,IAAIA,UAAU;GACdA,WAAW;GACX,MAAMT,cAAcG,SAASJ,eAAe5B,IAAIlC,MAAM;GACtD,IAAIyE,oBAAoB;GACxB,IAAIJ;GACJ,IAAI;IACFN,cAAc;GAChB,SAASrB,OAAO;IACd+B,oBAAoB;IACpBJ,mBAAmB3B;GACrB;GACA,IAAI;IACFwB,SAAS9C,gBAAgB;GAC3B,SAASsB,OAAO;IACd,IAAI,CAAC+B,mBAAmB,MAAM/B;IAC9B4B,QAAQ5B,MAAMA,KAAK;GACrB;GACA,IAAI+B,mBAAmB,MAAMJ;EAC/B;CACF;CAEA,OAAOK,UAAU;EACf,IAAIzE,uBAAuBC,WAAW;GACpC,MAAMgE,WAAWjE,uBAAuBC;GACxCJ,OAAO6E,oBAAoB,WAAWT,SAASlD,aAAa;GAE5D,IAAI4D,gBAAgB;GACpB,IAAIC;GACJ,MAAMC,cAAcC,YAAwB;IAC1C,IAAI;KACFA,QAAQ;IACV,SAASrC,OAAO;KACd,IAAIkC,eACFN,QAAQ5B,MAAMA,KAAK;UACd;MACLkC,gBAAgB;MAChBC,eAAenC;KACjB;IACF;GACF;GAEAwB,SAAS7D,sBAAsB2E,SAASjB,gBAAgB;IACtD,IAAIA,aAAae,WAAWf,WAAW;GACzC,CAAC;GACDG,SAAS7D,sBAAsB4E,MAAM;GACrCf,SAAS/D,WAAW8E,MAAM;GAC1Bf,SAAS5D,iBAAiB0E,SAAS,EAAEzE,iBAAiBE,SAASyB,OAAO;IACpE4C,iBAAiB;KACfvE,gBAAgB+B,MAAM;KACtB4B,SAASpC,YAAYrB,OAAO;MAC1BtB,MAAM;MACN+C;MACAQ,OAAO;KACT,CAAC;IACH,CAAC;GACH,CAAC;GACDwB,SAAS5D,iBAAiB2E,MAAM;GAEhChF,uBAAuBC,YAAY;GACnC,IAAI0E,eAAe,MAAMC;EAC3B;CACF;AACF"}
1
+ {"version":3,"file":"provider.js","names":["ModelContextProvider","ModelContext","Unsubscribe","Tool","toJSONSchema","FrameMessage","FRAME_MESSAGE_CHANNEL","SerializedModelContext","SerializedTool","serializeTool","tool","description","parameters","undefined","disabled","type","serializeModelContext","context","system","tools","Object","fromEntries","entries","map","name","getDefaultTargetOrigin","window","location","origin","AssistantFrameProvider","_instance","_providers","Map","_providerUnsubscribes","_activeToolCalls","abortController","AbortController","event","MessageEvent","provider","_targetOrigin","_strictRegistrations","_wildcardRegistrations","_startupTimer","ReturnType","setTimeout","_disposed","constructor","targetOrigin","handleMessage","bind","addEventListener","broadcastUpdate","getInstance","reconcileTargetOrigin","size","Error","source","parent","data","channel","message","sendMessage","getModelContext","handleToolCall","catch","error","console","cancelToolCall","id","Extract","resolvedTool","getTool","toolName","get","abort","activeCall","set","result","execute","args","toolCallId","abortSignal","signal","human","e","String","delete","sendError","cancelToolCallsForProvider","matchingCalls","Array","from","filter","sendFailed","postMessage","getProviders","Set","values","resolved","contexts","reduce","merged","updateMessage","removeProvider","unsubscribe","removedProvider","includes","addModelContextProvider","instance","Symbol","subscribe","cancelError","unsubscribeError","broadcastError","released","cleanupFailed","cleanupError","runCleanup","cleanup","dispose","clearTimeout","removeEventListener","forEach","clear"],"sources":["../../../src/model-context/frame/provider.ts"],"sourcesContent":["import type { ModelContextProvider, ModelContext } from \"../types\";\nimport type { Unsubscribe } from \"../../types/unsubscribe\";\nimport { type Tool, toJSONSchema } from \"assistant-stream\";\nimport {\n type FrameMessage,\n FRAME_MESSAGE_CHANNEL,\n type SerializedModelContext,\n type SerializedTool,\n} from \"./types\";\n\nconst serializeTool = (tool: Tool<any, any>): SerializedTool => ({\n ...(tool.description && { description: tool.description }),\n parameters: tool.parameters ? toJSONSchema(tool.parameters) : undefined,\n ...(tool.disabled !== undefined && { disabled: tool.disabled }),\n ...(tool.type && { type: tool.type }),\n});\n\nconst serializeModelContext = (\n context: ModelContext,\n): SerializedModelContext => ({\n ...(context.system !== undefined && { system: context.system }),\n ...(context.tools && {\n tools: Object.fromEntries(\n Object.entries(context.tools).map(([name, tool]) => [\n name,\n serializeTool(tool),\n ]),\n ),\n }),\n});\n\nconst getDefaultTargetOrigin = () => window.location.origin;\n\nexport class AssistantFrameProvider {\n private static _instance: AssistantFrameProvider | null = null;\n\n private _providers = new Map<symbol, ModelContextProvider>();\n private _providerUnsubscribes = new Map<symbol, Unsubscribe | undefined>();\n private _activeToolCalls = new Map<\n string,\n {\n abortController: AbortController;\n event: MessageEvent;\n provider: ModelContextProvider | undefined;\n }\n >();\n private _targetOrigin: string;\n private _strictRegistrations = 0;\n private _wildcardRegistrations = 0;\n private _startupTimer: ReturnType<typeof setTimeout> | undefined;\n private _disposed = false;\n\n private constructor(targetOrigin: string = getDefaultTargetOrigin()) {\n this._targetOrigin = targetOrigin;\n this.handleMessage = this.handleMessage.bind(this);\n window.addEventListener(\"message\", this.handleMessage);\n\n this._startupTimer = setTimeout(() => {\n this._startupTimer = undefined;\n this.broadcastUpdate();\n }, 0);\n }\n\n private static getInstance(targetOrigin?: string): AssistantFrameProvider {\n if (!AssistantFrameProvider._instance) {\n AssistantFrameProvider._instance = new AssistantFrameProvider(\n targetOrigin,\n );\n } else {\n AssistantFrameProvider._instance.reconcileTargetOrigin(targetOrigin);\n }\n return AssistantFrameProvider._instance;\n }\n\n private reconcileTargetOrigin(\n targetOrigin: string = getDefaultTargetOrigin(),\n ) {\n if (targetOrigin === this._targetOrigin) return;\n\n if (this._providers.size === 0) {\n this._targetOrigin = targetOrigin;\n return;\n }\n\n if (targetOrigin === \"*\") return;\n\n if (this._targetOrigin === \"*\") {\n this._targetOrigin = targetOrigin;\n return;\n }\n\n throw new Error(\n `AssistantFrameProvider cannot register conflicting target origins: \"${this._targetOrigin}\" and \"${targetOrigin}\"`,\n );\n }\n\n private handleMessage(event: MessageEvent) {\n if (this._targetOrigin !== \"*\" && event.origin !== this._targetOrigin)\n return;\n if (event.source !== window.parent) return;\n if (event.data?.channel !== FRAME_MESSAGE_CHANNEL) return;\n\n const message = event.data.message as FrameMessage;\n\n switch (message.type) {\n case \"model-context-request\":\n this.sendMessage(event, {\n type: \"model-context-update\",\n context: serializeModelContext(this.getModelContext()),\n });\n break;\n\n case \"tool-call\":\n void this.handleToolCall(message, event).catch((error: unknown) => {\n console.error(\n \"[assistant-ui] AssistantFrame tool call failed.\",\n error,\n );\n });\n break;\n\n case \"tool-cancel\":\n this.cancelToolCall(message.id);\n break;\n }\n }\n\n private async handleToolCall(\n message: Extract<FrameMessage, { type: \"tool-call\" }>,\n event: MessageEvent,\n ) {\n const resolvedTool = this.getTool(message.toolName);\n const tool = resolvedTool?.tool;\n const abortController = new AbortController();\n this._activeToolCalls.get(message.id)?.abortController.abort();\n const activeCall = {\n abortController,\n event,\n provider: resolvedTool?.provider,\n };\n this._activeToolCalls.set(message.id, activeCall);\n\n let result: any;\n let error: string | undefined;\n\n if (!tool) {\n error = `Tool \"${message.toolName}\" not found`;\n } else {\n try {\n result = tool.execute\n ? await tool.execute(message.args, {\n toolCallId: message.id,\n abortSignal: abortController.signal,\n human: async () => {\n throw new Error(\n \"Tool human input is not supported in frame context\",\n );\n },\n })\n : undefined;\n } catch (e) {\n error = e instanceof Error ? e.message : String(e);\n }\n }\n\n if (this._activeToolCalls.get(message.id) !== activeCall) return;\n this._activeToolCalls.delete(message.id);\n\n try {\n this.sendMessage(event, {\n type: \"tool-result\",\n id: message.id,\n ...(error !== undefined ? { error } : { result }),\n });\n } catch (sendError) {\n if (error !== undefined) throw sendError;\n\n console.error(\n \"[assistant-ui] AssistantFrame tool result could not be sent.\",\n sendError,\n );\n this.sendMessage(event, {\n type: \"tool-result\",\n id: message.id,\n error: \"Tool result could not be sent across the frame boundary\",\n });\n }\n }\n\n private cancelToolCall(id: string) {\n const activeCall = this._activeToolCalls.get(id);\n if (!activeCall) return;\n this._activeToolCalls.delete(id);\n activeCall.abortController.abort();\n }\n\n private cancelToolCallsForProvider(provider: ModelContextProvider) {\n const matchingCalls = Array.from(this._activeToolCalls).filter(\n ([, activeCall]) => activeCall.provider === provider,\n );\n for (const [id, activeCall] of matchingCalls) {\n this._activeToolCalls.delete(id);\n activeCall.abortController.abort();\n }\n\n let sendFailed = false;\n let sendError: unknown;\n for (const [id, activeCall] of matchingCalls) {\n try {\n this.sendMessage(activeCall.event, {\n type: \"tool-result\",\n id,\n error: \"AssistantFrame tool provider has been removed\",\n });\n } catch (error) {\n if (sendFailed) {\n console.error(error);\n } else {\n sendFailed = true;\n sendError = error;\n }\n }\n }\n\n if (sendFailed) throw sendError;\n }\n\n private sendMessage(event: MessageEvent, message: FrameMessage) {\n event.source?.postMessage(\n { channel: FRAME_MESSAGE_CHANNEL, message },\n { targetOrigin: event.origin },\n );\n }\n\n private getProviders() {\n return Array.from(new Set(this._providers.values()));\n }\n\n private getTool(toolName: string) {\n let resolved:\n | { provider: ModelContextProvider; tool: Tool<any, any> }\n | undefined;\n\n for (const provider of this.getProviders()) {\n const tool = provider.getModelContext().tools?.[toolName];\n if (tool) resolved = { provider, tool };\n }\n\n return resolved;\n }\n\n private getModelContext(): ModelContext {\n const contexts = this.getProviders().map((provider) =>\n provider.getModelContext(),\n );\n\n return contexts.reduce(\n (merged, context) => ({\n system: context.system\n ? merged.system\n ? `${merged.system}\\n\\n${context.system}`\n : context.system\n : merged.system,\n tools: { ...(merged.tools || {}), ...(context.tools || {}) },\n }),\n {} as ModelContext,\n );\n }\n\n private broadcastUpdate() {\n if (this._disposed) return;\n if (window.parent && window.parent !== window) {\n const updateMessage: FrameMessage = {\n type: \"model-context-update\",\n context: serializeModelContext(this.getModelContext()),\n };\n\n window.parent.postMessage(\n { channel: FRAME_MESSAGE_CHANNEL, message: updateMessage },\n this._targetOrigin,\n );\n }\n }\n\n private removeProvider(\n id: symbol,\n origin: string,\n ): {\n unsubscribe: Unsubscribe | undefined;\n removedProvider: ModelContextProvider | undefined;\n } {\n const provider = this._providers.get(id);\n this._providers.delete(id);\n const unsubscribe = this._providerUnsubscribes.get(id);\n this._providerUnsubscribes.delete(id);\n if (origin === \"*\") {\n this._wildcardRegistrations -= 1;\n if (\n this._wildcardRegistrations === 0 &&\n this._strictRegistrations === 0\n ) {\n this._targetOrigin = getDefaultTargetOrigin();\n }\n } else {\n this._strictRegistrations -= 1;\n if (this._strictRegistrations === 0) {\n this._targetOrigin =\n this._wildcardRegistrations > 0 ? \"*\" : getDefaultTargetOrigin();\n }\n }\n const removedProvider =\n provider && !this.getProviders().includes(provider)\n ? provider\n : undefined;\n return { unsubscribe, removedProvider };\n }\n\n static addModelContextProvider(\n provider: ModelContextProvider,\n targetOrigin?: string,\n ): Unsubscribe {\n const origin = targetOrigin ?? getDefaultTargetOrigin();\n const instance = AssistantFrameProvider.getInstance(origin);\n const id = Symbol();\n instance._providers.set(id, provider);\n if (origin === \"*\") {\n instance._wildcardRegistrations += 1;\n } else {\n instance._strictRegistrations += 1;\n }\n\n try {\n const unsubscribe = provider.subscribe?.(() =>\n instance.broadcastUpdate(),\n );\n if (unsubscribe) {\n instance._providerUnsubscribes.set(id, unsubscribe);\n }\n\n instance.broadcastUpdate();\n } catch (error) {\n const { unsubscribe, removedProvider } = instance.removeProvider(\n id,\n origin,\n );\n // Rollback failures must not replace the registration error.\n try {\n if (removedProvider) {\n instance.cancelToolCallsForProvider(removedProvider);\n }\n } catch (cancelError) {\n console.error(cancelError);\n }\n try {\n unsubscribe?.();\n } catch (unsubscribeError) {\n console.error(unsubscribeError);\n }\n try {\n instance.broadcastUpdate();\n } catch (broadcastError) {\n console.error(broadcastError);\n }\n throw error;\n }\n\n let released = false;\n return () => {\n if (released) return;\n released = true;\n const { unsubscribe, removedProvider } = instance.removeProvider(\n id,\n origin,\n );\n let cleanupFailed = false;\n let cleanupError: unknown;\n const runCleanup = (cleanup: () => void) => {\n try {\n cleanup();\n } catch (error) {\n if (cleanupFailed) {\n console.error(error);\n } else {\n cleanupFailed = true;\n cleanupError = error;\n }\n }\n };\n\n if (removedProvider) {\n runCleanup(() => instance.cancelToolCallsForProvider(removedProvider));\n }\n if (unsubscribe) runCleanup(unsubscribe);\n runCleanup(() => instance.broadcastUpdate());\n\n if (cleanupFailed) throw cleanupError;\n };\n }\n\n static dispose() {\n if (AssistantFrameProvider._instance) {\n const instance = AssistantFrameProvider._instance;\n instance._disposed = true;\n if (instance._startupTimer !== undefined) {\n clearTimeout(instance._startupTimer);\n instance._startupTimer = undefined;\n }\n window.removeEventListener(\"message\", instance.handleMessage);\n\n let cleanupFailed = false;\n let cleanupError: unknown;\n const runCleanup = (cleanup: () => void) => {\n try {\n cleanup();\n } catch (error) {\n if (cleanupFailed) {\n console.error(error);\n } else {\n cleanupFailed = true;\n cleanupError = error;\n }\n }\n };\n\n instance._providerUnsubscribes.forEach((unsubscribe) => {\n if (unsubscribe) runCleanup(unsubscribe);\n });\n instance._providerUnsubscribes.clear();\n instance._providers.clear();\n instance._activeToolCalls.forEach(({ abortController, event }, id) => {\n runCleanup(() => {\n abortController.abort();\n instance.sendMessage(event, {\n type: \"tool-result\",\n id,\n error: \"AssistantFrameProvider has been disposed\",\n });\n });\n });\n instance._activeToolCalls.clear();\n\n AssistantFrameProvider._instance = null;\n if (cleanupFailed) throw cleanupError;\n }\n }\n}\n"],"mappings":";;;AAUA,MAAMS,iBAAiBC,UAA0C;CAC/D,GAAIA,KAAKC,eAAe,EAAEA,aAAaD,KAAKC,YAAY;CACxDC,YAAYF,KAAKE,aAAaR,aAAaM,KAAKE,UAAU,IAAIC,KAAAA;CAC9D,GAAIH,KAAKI,aAAaD,KAAAA,KAAa,EAAEC,UAAUJ,KAAKI,SAAS;CAC7D,GAAIJ,KAAKK,QAAQ,EAAEA,MAAML,KAAKK,KAAK;AACrC;AAEA,MAAMC,yBACJC,aAC4B;CAC5B,GAAIA,QAAQC,WAAWL,KAAAA,KAAa,EAAEK,QAAQD,QAAQC,OAAO;CAC7D,GAAID,QAAQE,SAAS,EACnBA,OAAOC,OAAOC,YACZD,OAAOE,QAAQL,QAAQE,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMd,UAAU,CAClDc,MACAf,cAAcC,IAAI,CAAC,CACpB,CACH,EACF;AACF;AAEA,MAAMe,+BAA+BC,OAAOC,SAASC;AAErD,IAAaC,yBAAb,MAAaA,uBAAuB;CAClC,OAAeC,YAA2C;CAE1D,6BAAqB,IAAIE,IAAkC;CAC3D,wCAAgC,IAAIA,IAAqC;CACzE,mCAA2B,IAAIA,IAO7B;CACF;CACA,uBAA+B;CAC/B,yBAAiC;CACjC;CACA,YAAoB;CAEpB,YAAoBgB,eAAuBvB,uBAAuB,GAAG;EACnE,KAAKe,gBAAgBQ;EACrB,KAAKC,gBAAgB,KAAKA,cAAcC,KAAK,IAAI;EACjDxB,OAAOyB,iBAAiB,WAAW,KAAKF,aAAa;EAErD,KAAKN,gBAAgBE,iBAAiB;GACpC,KAAKF,gBAAgB9B,KAAAA;GACrB,KAAKuC,gBAAgB;EACvB,GAAG,CAAC;CACN;CAEA,OAAeC,YAAYL,cAA+C;EACxE,IAAI,CAACnB,uBAAuBC,WAC1BD,uBAAuBC,YAAY,IAAID,uBACrCmB,YACF;OAEAnB,uBAAuBC,UAAUwB,sBAAsBN,YAAY;EAErE,OAAOnB,uBAAuBC;CAChC;CAEA,sBACEkB,eAAuBvB,uBAAuB,GAC9C;EACA,IAAIuB,iBAAiB,KAAKR,eAAe;EAEzC,IAAI,KAAKT,WAAWwB,SAAS,GAAG;GAC9B,KAAKf,gBAAgBQ;GACrB;EACF;EAEA,IAAIA,iBAAiB,KAAK;EAE1B,IAAI,KAAKR,kBAAkB,KAAK;GAC9B,KAAKA,gBAAgBQ;GACrB;EACF;EAEA,MAAM,IAAIQ,MACR,uEAAuE,KAAKhB,cAAa,SAAUQ,aAAY,EACjH;CACF;CAEA,cAAsBX,OAAqB;EACzC,IAAI,KAAKG,kBAAkB,OAAOH,MAAMT,WAAW,KAAKY,eACtD;EACF,IAAIH,MAAMoB,WAAW/B,OAAOgC,QAAQ;EACpC,IAAIrB,MAAMsB,MAAMC,YAAAA,sBAAmC;EAEnD,MAAMC,UAAUxB,MAAMsB,KAAKE;EAE3B,QAAQA,QAAQ9C,MAAhB;GACE,KAAK;IACH,KAAK+C,YAAYzB,OAAO;KACtBtB,MAAM;KACNE,SAASD,sBAAsB,KAAK+C,gBAAgB,CAAC;IACvD,CAAC;IACD;GAEF,KAAK;IACH,KAAUC,eAAeH,SAASxB,KAAK,CAAC,CAAC4B,OAAOC,UAAmB;KACjEC,QAAQD,MACN,mDACAA,KACF;IACF,CAAC;IACD;GAEF,KAAK,eACH,KAAKE,eAAeP,QAAQQ,EAAE;EAElC;CACF;CAEA,MAAcL,eACZH,SACAxB,OACA;EACA,MAAMkC,eAAe,KAAKC,QAAQX,QAAQY,QAAQ;EAClD,MAAM/D,OAAO6D,cAAc7D;EAC3B,MAAMyB,kBAAkB,IAAIC,gBAAgB;EAC5C,KAAKF,iBAAiBwC,IAAIb,QAAQQ,EAAE,CAAC,EAAElC,gBAAgBwC,MAAM;EAC7D,MAAMC,aAAa;GACjBzC;GACAE;GACAE,UAAUgC,cAAchC;EAC1B;EACA,KAAKL,iBAAiB2C,IAAIhB,QAAQQ,IAAIO,UAAU;EAEhD,IAAIE;EACJ,IAAIZ;EAEJ,IAAI,CAACxD,MACHwD,QAAQ,SAASL,QAAQY,SAAQ;OAEjC,IAAI;GACFK,SAASpE,KAAKqE,UACV,MAAMrE,KAAKqE,QAAQlB,QAAQmB,MAAM;IAC/BC,YAAYpB,QAAQQ;IACpBa,aAAa/C,gBAAgBgD;IAC7BC,OAAO,YAAY;KACjB,MAAM,IAAI5B,MACR,oDACF;IACF;GACF,CAAC,IACD3C,KAAAA;EACN,SAASwE,GAAG;GACVnB,QAAQmB,aAAa7B,QAAQ6B,EAAExB,UAAUyB,OAAOD,CAAC;EACnD;EAGF,IAAI,KAAKnD,iBAAiBwC,IAAIb,QAAQQ,EAAE,MAAMO,YAAY;EAC1D,KAAK1C,iBAAiBqD,OAAO1B,QAAQQ,EAAE;EAEvC,IAAI;GACF,KAAKP,YAAYzB,OAAO;IACtBtB,MAAM;IACNsD,IAAIR,QAAQQ;IACZ,GAAIH,UAAUrD,KAAAA,IAAY,EAAEqD,MAAM,IAAI,EAAEY,OAAO;GACjD,CAAC;EACH,SAASU,WAAW;GAClB,IAAItB,UAAUrD,KAAAA,GAAW,MAAM2E;GAE/BrB,QAAQD,MACN,gEACAsB,SACF;GACA,KAAK1B,YAAYzB,OAAO;IACtBtB,MAAM;IACNsD,IAAIR,QAAQQ;IACZH,OAAO;GACT,CAAC;EACH;CACF;CAEA,eAAuBG,IAAY;EACjC,MAAMO,aAAa,KAAK1C,iBAAiBwC,IAAIL,EAAE;EAC/C,IAAI,CAACO,YAAY;EACjB,KAAK1C,iBAAiBqD,OAAOlB,EAAE;EAC/BO,WAAWzC,gBAAgBwC,MAAM;CACnC;CAEA,2BAAmCpC,UAAgC;EACjE,MAAMmD,gBAAgBC,MAAMC,KAAK,KAAK1D,gBAAgB,CAAC,CAAC2D,QACrD,GAAGjB,gBAAgBA,WAAWrC,aAAaA,QAC9C;EACA,KAAK,MAAM,CAAC8B,IAAIO,eAAec,eAAe;GAC5C,KAAKxD,iBAAiBqD,OAAOlB,EAAE;GAC/BO,WAAWzC,gBAAgBwC,MAAM;EACnC;EAEA,IAAImB,aAAa;EACjB,IAAIN;EACJ,KAAK,MAAM,CAACnB,IAAIO,eAAec,eAC7B,IAAI;GACF,KAAK5B,YAAYc,WAAWvC,OAAO;IACjCtB,MAAM;IACNsD;IACAH,OAAO;GACT,CAAC;EACH,SAASA,OAAO;GACd,IAAI4B,YACF3B,QAAQD,MAAMA,KAAK;QACd;IACL4B,aAAa;IACbN,YAAYtB;GACd;EACF;EAGF,IAAI4B,YAAY,MAAMN;CACxB;CAEA,YAAoBnD,OAAqBwB,SAAuB;EAC9DxB,MAAMoB,QAAQsC,YACZ;GAAEnC,SAAStD;GAAuBuD;EAAQ,GAC1C,EAAEb,cAAcX,MAAMT,OAAO,CAC/B;CACF;CAEA,eAAuB;EACrB,OAAO+D,MAAMC,KAAK,IAAIK,IAAI,KAAKlE,WAAWmE,OAAO,CAAC,CAAC;CACrD;CAEA,QAAgBzB,UAAkB;EAChC,IAAI0B;EAIJ,KAAK,MAAM5D,YAAY,KAAKyD,aAAa,GAAG;GAC1C,MAAMtF,OAAO6B,SAASwB,gBAAgB,CAAC,CAAC5C,QAAQsD;GAChD,IAAI/D,MAAMyF,WAAW;IAAE5D;IAAU7B;GAAK;EACxC;EAEA,OAAOyF;CACT;CAEA,kBAAwC;EAKtC,OAJiB,KAAKH,aAAa,CAAC,CAACzE,KAAKgB,aACxCA,SAASwB,gBAAgB,CAGpBqC,CAAQ,CAACC,QACbC,QAAQrF,aAAa;GACpBC,QAAQD,QAAQC,SACZoF,OAAOpF,SACL,GAAGoF,OAAOpF,OAAM,MAAOD,QAAQC,WAC/BD,QAAQC,SACVoF,OAAOpF;GACXC,OAAO;IAAE,GAAImF,OAAOnF,SAAS,CAAC;IAAI,GAAIF,QAAQE,SAAS,CAAC;GAAG;EAC7D,IACA,CAAC,CACH;CACF;CAEA,kBAA0B;EACxB,IAAI,KAAK2B,WAAW;EACpB,IAAIpB,OAAOgC,UAAUhC,OAAOgC,WAAWhC,QAAQ;GAC7C,MAAM6E,gBAA8B;IAClCxF,MAAM;IACNE,SAASD,sBAAsB,KAAK+C,gBAAgB,CAAC;GACvD;GAEArC,OAAOgC,OAAOqC,YACZ;IAAEnC,SAAStD;IAAuBuD,SAAS0C;GAAc,GACzD,KAAK/D,aACP;EACF;CACF;CAEA,eACE6B,IACAzC,QAIA;EACA,MAAMW,WAAW,KAAKR,WAAW2C,IAAIL,EAAE;EACvC,KAAKtC,WAAWwD,OAAOlB,EAAE;EACzB,MAAMoC,cAAc,KAAKxE,sBAAsByC,IAAIL,EAAE;EACrD,KAAKpC,sBAAsBsD,OAAOlB,EAAE;EACpC,IAAIzC,WAAW,KAAK;GAClB,KAAKc,0BAA0B;GAC/B,IACE,KAAKA,2BAA2B,KAChC,KAAKD,yBAAyB,GAE9B,KAAKD,gBAAgBf,uBAAuB;EAEhD,OAAO;GACL,KAAKgB,wBAAwB;GAC7B,IAAI,KAAKA,yBAAyB,GAChC,KAAKD,gBACH,KAAKE,yBAAyB,IAAI,MAAMjB,uBAAuB;EAErE;EAKA,OAAO;GAAEgF;GAAaC,iBAHpBnE,YAAY,CAAC,KAAKyD,aAAa,CAAC,CAACW,SAASpE,QAAQ,IAC9CA,WACA1B,KAAAA;EACgC;CACxC;CAEA,OAAO+F,wBACLrE,UACAS,cACa;EACb,MAAMpB,SAASoB,gBAAgBvB,uBAAuB;EACtD,MAAMoF,WAAWhF,uBAAuBwB,YAAYzB,MAAM;EAC1D,MAAMyC,KAAKyC,OAAO;EAClBD,SAAS9E,WAAW8C,IAAIR,IAAI9B,QAAQ;EACpC,IAAIX,WAAW,KACbiF,SAASnE,0BAA0B;OAEnCmE,SAASpE,wBAAwB;EAGnC,IAAI;GACF,MAAMgE,cAAclE,SAASwE,kBAC3BF,SAASzD,gBAAgB,CAC3B;GACA,IAAIqD,aACFI,SAAS5E,sBAAsB4C,IAAIR,IAAIoC,WAAW;GAGpDI,SAASzD,gBAAgB;EAC3B,SAASc,OAAO;GACd,MAAM,EAAEuC,aAAaC,oBAAoBG,SAASL,eAChDnC,IACAzC,MACF;GAEA,IAAI;IACF,IAAI8E,iBACFG,SAASpB,2BAA2BiB,eAAe;GAEvD,SAASM,aAAa;IACpB7C,QAAQD,MAAM8C,WAAW;GAC3B;GACA,IAAI;IACFP,cAAc;GAChB,SAASQ,kBAAkB;IACzB9C,QAAQD,MAAM+C,gBAAgB;GAChC;GACA,IAAI;IACFJ,SAASzD,gBAAgB;GAC3B,SAAS8D,gBAAgB;IACvB/C,QAAQD,MAAMgD,cAAc;GAC9B;GACA,MAAMhD;EACR;EAEA,IAAIiD,WAAW;EACf,aAAa;GACX,IAAIA,UAAU;GACdA,WAAW;GACX,MAAM,EAAEV,aAAaC,oBAAoBG,SAASL,eAChDnC,IACAzC,MACF;GACA,IAAIwF,gBAAgB;GACpB,IAAIC;GACJ,MAAMC,cAAcC,YAAwB;IAC1C,IAAI;KACFA,QAAQ;IACV,SAASrD,OAAO;KACd,IAAIkD,eACFjD,QAAQD,MAAMA,KAAK;UACd;MACLkD,gBAAgB;MAChBC,eAAenD;KACjB;IACF;GACF;GAEA,IAAIwC,iBACFY,iBAAiBT,SAASpB,2BAA2BiB,eAAe,CAAC;GAEvE,IAAID,aAAaa,WAAWb,WAAW;GACvCa,iBAAiBT,SAASzD,gBAAgB,CAAC;GAE3C,IAAIgE,eAAe,MAAMC;EAC3B;CACF;CAEA,OAAOG,UAAU;EACf,IAAI3F,uBAAuBC,WAAW;GACpC,MAAM+E,WAAWhF,uBAAuBC;GACxC+E,SAAS/D,YAAY;GACrB,IAAI+D,SAASlE,kBAAkB9B,KAAAA,GAAW;IACxC4G,aAAaZ,SAASlE,aAAa;IACnCkE,SAASlE,gBAAgB9B,KAAAA;GAC3B;GACAa,OAAOgG,oBAAoB,WAAWb,SAAS5D,aAAa;GAE5D,IAAImE,gBAAgB;GACpB,IAAIC;GACJ,MAAMC,cAAcC,YAAwB;IAC1C,IAAI;KACFA,QAAQ;IACV,SAASrD,OAAO;KACd,IAAIkD,eACFjD,QAAQD,MAAMA,KAAK;UACd;MACLkD,gBAAgB;MAChBC,eAAenD;KACjB;IACF;GACF;GAEA2C,SAAS5E,sBAAsB0F,SAASlB,gBAAgB;IACtD,IAAIA,aAAaa,WAAWb,WAAW;GACzC,CAAC;GACDI,SAAS5E,sBAAsB2F,MAAM;GACrCf,SAAS9E,WAAW6F,MAAM;GAC1Bf,SAAS3E,iBAAiByF,SAAS,EAAExF,iBAAiBE,SAASgC,OAAO;IACpEiD,iBAAiB;KACfnF,gBAAgBwC,MAAM;KACtBkC,SAAS/C,YAAYzB,OAAO;MAC1BtB,MAAM;MACNsD;MACAH,OAAO;KACT,CAAC;IACH,CAAC;GACH,CAAC;GACD2C,SAAS3E,iBAAiB0F,MAAM;GAEhC/F,uBAAuBC,YAAY;GACnC,IAAIsF,eAAe,MAAMC;EAC3B;CACF;AACF"}
@@ -1 +1 @@
1
- {"version":3,"file":"interactable-composer-metadata.d.ts","names":[],"sources":["../../src/model-context/interactable-composer-metadata.ts"],"mappings":";;;;;KAOY;EACV;EACA;EACA;;;;;;EAMA;;;;;;;KAQG;EACH;EACA;EACA;;;;;;;;;;;iBAYc,kCAAkC;EAChD;IACE;;;;;;iBAgBY,oCACd,OAAO;;iBASO,qBAAqB;;;;;;iBA0DrB,8BACd,eACA,kBACA;EAEM;;;;;;EAMA,aAAa;EACb,gBAAgB;;;;;;KAgGZ;;EAEV;EACA;;EAEA;;;;;;;;;;;;;;;;;;iBA0Bc,iCACd,mBAAmB,0BACnB,YACA,eACC;;;;;iBAwEa,oBACd,mBAAmB,0BACnB,YACA;EACG;;;;;;;;;iBAaW,iCACd,MAAM,qCACN,mBAAmB,2BAClB"}
1
+ {"version":3,"file":"interactable-composer-metadata.d.ts","names":[],"sources":["../../src/model-context/interactable-composer-metadata.ts"],"mappings":";;;;;KAOY;EACV;EACA;EACA;;;;;;EAMA;;;;;;;KAQG;EACH;EACA;EACA;;;;;;;;;;;iBAYc,kCAAkC;EAChD;IACE;;;;;;iBAgBY,oCACd,OAAO;;iBASO,qBAAqB;;;;;;iBAgErB,8BACd,eACA,kBACA;EAEM;;;;;;EAMA,aAAa;EACb,gBAAgB;;;;;;KAgGZ;;EAEV;EACA;;EAEA;;;;;;;;;;;;;;;;;;iBA0Bc,iCACd,mBAAmB,0BACnB,YACA,eACC;;;;;iBAwEa,oBACd,mBAAmB,0BACnB,YACA;EACG;;;;;;;;;iBAaW,iCACd,MAAM,qCACN,mBAAmB,2BAClB"}
@@ -47,15 +47,22 @@ function applyArrayUpdate(prev, update, mintId) {
47
47
  });
48
48
  }
49
49
  const patches = update.update;
50
- if (Array.isArray(patches) && patches.length > 0) next = next.map((item) => {
51
- const id = getArrayItemId(item);
52
- if (id === void 0 || !isRecord(item)) return item;
53
- const patch = patches.find((candidate) => isRecord(candidate) && candidate.id === id);
54
- return patch ? {
55
- ...item,
56
- ...patch
57
- } : item;
58
- });
50
+ if (Array.isArray(patches) && patches.length > 0) {
51
+ const patchesById = /* @__PURE__ */ new Map();
52
+ for (const candidate of patches) {
53
+ const id = getArrayItemId(candidate);
54
+ if (id !== void 0 && !Number.isNaN(id) && !patchesById.has(id)) patchesById.set(id, candidate);
55
+ }
56
+ next = next.map((item) => {
57
+ const id = getArrayItemId(item);
58
+ if (id === void 0 || !isRecord(item)) return item;
59
+ const patch = patchesById.get(id);
60
+ return patch ? {
61
+ ...item,
62
+ ...patch
63
+ } : item;
64
+ });
65
+ }
59
66
  if (Array.isArray(update.add) && update.add.length > 0) {
60
67
  const added = mintId ? update.add.map((item) => {
61
68
  if (!isRecord(item) || item.id !== void 0) return item;
@@ -1 +1 @@
1
- {"version":3,"file":"interactable-composer-metadata.js","names":["isJSONValueEqual","isJSONValue","isRecord","Unstable_InteractableSnapshotEntry","id","name","state","partial","SnapshotCarrierMessage","role","metadata","content","unstable_getInteractableSnapshots","message","undefined","custom","Record","items","interactables","Array","isArray","unstable_formatInteractableSnapshot","entry","JSON","stringify","interactableToolName","replace","getArrayItemId","item","applyArrayUpdate","prev","update","mintId","next","set","clear","remove","length","ids","Set","filter","has","patches","map","patch","find","candidate","add","added","shallowMergeInteractableState","options","arrayBaseline","idFactory","field","idKeyedFields","ReadonlySet","baseline","key","value","Object","entries","baseValue","shallowDiffInteractableState","known","keys","diff","changed","ToolCallLikePart","type","toolCallId","toolName","args","result","asToolCallPart","part","p","updateCallTargets","success","argsId","createAddedItemIdFactory","addedItemIds","idsByField","Map","stringIds","size","get","shift","Unstable_InteractableVersion","origin","versionsCache","WeakMap","unstable_getInteractableVersions","messages","byName","byId","cached","versions","current","msg","it","push","_id","findModelKnownState","last","gateInteractableComposerMetadata","meta","rest","gated","pending","process","env","NODE_ENV","console","warn"],"sources":["../../src/model-context/interactable-composer-metadata.ts"],"sourcesContent":["import { isJSONValueEqual } from \"../utils/json/is-json-equal\";\nimport { isJSONValue, isRecord } from \"../utils/json/is-json\";\n\n/**\n * Unstable / Experimental — the interactables API is still evolving and may change in any release.\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport type Unstable_InteractableSnapshotEntry = {\n id: string;\n name: string;\n state: unknown;\n /**\n * When true, `state` carries only the top-level fields that changed since\n * the model's last known state; omitted fields are unchanged. The first\n * snapshot of an instance is always full.\n */\n partial?: boolean | undefined;\n};\n\n/**\n * Minimal message shape needed to read snapshots and `update_*` tool calls out\n * of history. Structurally compatible with both `ThreadMessage` and the AI\n * SDK's `UIMessage`.\n */\ntype SnapshotCarrierMessage = {\n role: string;\n metadata?: unknown;\n content?: readonly unknown[] | undefined;\n};\n\n/**\n * Reads the interactable snapshots stamped on a message's\n * `metadata.custom.interactables`, or `undefined` if none. This is the read\n * half of the snapshot channel — integrations use it to surface interactable\n * state to the model (see `unstable_injectInteractableContext` in\n * `@assistant-ui/ai-sdk` for the AI SDK implementation).\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport function unstable_getInteractableSnapshots(message: {\n metadata?: unknown;\n}): Unstable_InteractableSnapshotEntry[] | undefined {\n const metadata = message.metadata;\n if (!metadata || typeof metadata !== \"object\") return undefined;\n const custom = (metadata as Record<string, unknown>).custom;\n if (!custom || typeof custom !== \"object\") return undefined;\n const items = (custom as Record<string, unknown>).interactables;\n return Array.isArray(items)\n ? (items as Unstable_InteractableSnapshotEntry[])\n : undefined;\n}\n\n/**\n * Canonical model-facing wording for one snapshot entry.\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport function unstable_formatInteractableSnapshot(\n entry: Unstable_InteractableSnapshotEntry,\n): string {\n if (entry.partial) {\n return `[State of \"${entry.name}\" (id: ${JSON.stringify(entry.id)}) changed — updated fields: ${JSON.stringify(entry.state)}; fields not listed are unchanged]`;\n }\n return `[Current state of \"${entry.name}\" (id: ${JSON.stringify(entry.id)}): ${JSON.stringify(entry.state)}]`;\n}\n\n/** The model-facing tool name for an interactable `name`. One tool per name. */\nexport function interactableToolName(name: string): string {\n return `update_${name.replace(/[^a-zA-Z0-9_-]/g, \"_\")}`;\n}\n\nconst getArrayItemId = (item: unknown): string | number | undefined => {\n if (!isRecord(item)) return undefined;\n const id = item.id;\n return typeof id === \"string\" || typeof id === \"number\" ? id : undefined;\n};\n\nfunction applyArrayUpdate(\n prev: unknown[],\n update: Record<string, unknown>,\n mintId?: () => string | undefined,\n) {\n let next = Array.isArray(update.set) ? [...update.set] : [...prev];\n\n if (update.clear === true) next = [];\n\n if (Array.isArray(update.remove) && update.remove.length > 0) {\n const ids = new Set(update.remove);\n next = next.filter((item) => {\n const id = getArrayItemId(item);\n return id !== undefined ? !ids.has(id) : !ids.has(item);\n });\n }\n\n const patches = update.update;\n if (Array.isArray(patches) && patches.length > 0) {\n next = next.map((item) => {\n const id = getArrayItemId(item);\n if (id === undefined || !isRecord(item)) return item;\n const patch = patches.find(\n (candidate) => isRecord(candidate) && candidate.id === id,\n );\n return patch ? { ...item, ...patch } : item;\n });\n }\n\n if (Array.isArray(update.add) && update.add.length > 0) {\n const added = mintId\n ? update.add.map((item) => {\n if (!isRecord(item) || item.id !== undefined) return item;\n const id = mintId();\n return id === undefined ? item : { ...item, id };\n })\n : update.add;\n next = [...next, ...added];\n }\n\n return next;\n}\n\n/**\n * The merge `update_*` tools apply: top-level keys replace, rest preserved.\n * Array fields also accept operation objects (`add`, `update`, `remove`,\n * `clear`, `set`) so collection edits do not need a companion manage_* tool.\n */\nexport function shallowMergeInteractableState(\n prev: unknown,\n partial: unknown,\n options?:\n | {\n arrayBaseline?: unknown;\n /**\n * Supplies ids for added items in id-keyed array fields. The live\n * execute path mints fresh ids; history folds replay ids previously\n * returned by execute.\n */\n idFactory?: (field: string) => string | undefined;\n idKeyedFields?: ReadonlySet<string>;\n }\n | undefined,\n): unknown {\n if (!isRecord(prev) || !isRecord(partial)) return partial;\n const baseline = isRecord(options?.arrayBaseline)\n ? options.arrayBaseline\n : prev;\n const next = { ...prev };\n for (const [key, value] of Object.entries(partial)) {\n const baseValue = baseline[key];\n if (Array.isArray(baseValue) && isRecord(value)) {\n const mintId =\n options?.idFactory &&\n (options.idKeyedFields === undefined || options.idKeyedFields.has(key))\n ? () => options.idFactory?.(key)\n : undefined;\n next[key] = applyArrayUpdate(baseValue, value, mintId);\n } else {\n next[key] = value;\n }\n }\n return next;\n}\n\n/**\n * The top-level fields of `next` that differ from `known`, for stamping a\n * partial snapshot. Returns `undefined` when a partial stamp has no benefit:\n * the change is not expressible as a shallow merge (non-object states, or a\n * top-level key was removed), or every field changed anyway — the caller\n * falls back to a full snapshot.\n */\nfunction shallowDiffInteractableState(\n known: unknown,\n next: unknown,\n): Record<string, unknown> | undefined {\n if (!isRecord(known) || !isRecord(next)) return undefined;\n for (const key of Object.keys(known)) {\n if (!(key in next)) return undefined;\n }\n const diff: Record<string, unknown> = {};\n for (const [key, value] of Object.entries(next)) {\n if (!(key in known) || !isJSONValueEqual(known[key], value)) {\n diff[key] = value;\n }\n }\n const changed = Object.keys(diff).length;\n if (changed === 0 || changed === Object.keys(next).length) return undefined;\n return diff;\n}\n\ntype ToolCallLikePart = {\n type?: string;\n toolCallId?: string;\n toolName?: string;\n args?: unknown;\n result?: unknown;\n};\n\nconst asToolCallPart = (part: unknown): ToolCallLikePart | undefined => {\n if (!part || typeof part !== \"object\") return undefined;\n const p = part as ToolCallLikePart;\n return p.type === \"tool-call\" ? p : undefined;\n};\n\n/** Whether an accepted `update_*` call addressed instance `id`. */\nconst updateCallTargets = (p: ToolCallLikePart, id: string): boolean => {\n if (!p.args || typeof p.args !== \"object\") return false;\n const result = isRecord(p.result) ? p.result : undefined;\n if (result?.success === false) return false;\n if (typeof result?.id === \"string\") return result.id === id;\n const argsId = (p.args as Record<string, unknown>).id;\n return argsId === id || argsId === undefined;\n};\n\nconst createAddedItemIdFactory = (result: unknown) => {\n const value = isRecord(result) ? result.addedItemIds : undefined;\n if (!isRecord(value)) return undefined;\n\n const idsByField = new Map<string, string[]>();\n for (const [field, ids] of Object.entries(value)) {\n if (!Array.isArray(ids)) continue;\n const stringIds = ids.filter(\n (item): item is string => typeof item === \"string\",\n );\n if (stringIds.length > 0) idsByField.set(field, stringIds);\n }\n if (idsByField.size === 0) return undefined;\n\n return (field: string) => idsByField.get(field)?.shift();\n};\n\n/**\n * Unstable / Experimental — the interactables API is still evolving and may change in any release.\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport type Unstable_InteractableVersion = {\n /** The full state as of this version. */\n state: unknown;\n origin: \"create\" | \"update\" | \"user-edit\";\n /** For create/update versions, the tool call that produced this version. */\n toolCallId?: string | undefined;\n};\n\n// Without this, every mounted anchor re-folds the whole thread on each streaming\n// token, since the repository hands out a new messages array per token.\nconst versionsCache = new WeakMap<\n readonly SnapshotCarrierMessage[],\n Map<string, Map<string, Unstable_InteractableVersion[]>>\n>();\n\n/**\n * Every version of interactable `id` recorded in the thread, oldest first,\n * folded chronologically:\n *\n * - the tool call whose `toolCallId` equals `id` seeds the baseline with its\n * args (`origin: \"create\"`) — the `id: toolCallId` convention for\n * tool-created interactables,\n * - a snapshot stamped on a user message is a `\"user-edit\"` version (a full\n * snapshot replaces the state; a partial one shallow-merges),\n * - each accepted `update_*` call shallow-merges an `\"update\"` version.\n *\n * The last entry is the state the model knows. Partial snapshots and update\n * calls with no baseline to merge into are skipped.\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport function unstable_getInteractableVersions(\n messages: readonly SnapshotCarrierMessage[],\n id: string,\n name: string,\n): Unstable_InteractableVersion[] {\n let byName = versionsCache.get(messages);\n if (!byName) {\n byName = new Map();\n versionsCache.set(messages, byName);\n }\n let byId = byName.get(name);\n if (!byId) {\n byId = new Map();\n byName.set(name, byId);\n }\n const cached = byId.get(id);\n if (cached) return cached;\n\n const toolName = interactableToolName(name);\n const versions: Unstable_InteractableVersion[] = [];\n const current = () => versions[versions.length - 1];\n\n for (const msg of messages) {\n if (msg.role === \"user\") {\n const entry = unstable_getInteractableSnapshots(msg)?.find(\n (it) => it.id === id,\n );\n if (!entry) continue;\n if (entry.partial) {\n const prev = current();\n if (prev) {\n versions.push({\n state: shallowMergeInteractableState(prev.state, entry.state),\n origin: \"user-edit\",\n });\n }\n } else {\n versions.push({ state: entry.state, origin: \"user-edit\" });\n }\n continue;\n }\n\n if (msg.role !== \"assistant\") continue;\n for (const part of msg.content ?? []) {\n const p = asToolCallPart(part);\n if (!p) continue;\n if (p.toolCallId === id && p.toolName === name) {\n if (p.args && typeof p.args === \"object\") {\n versions.push({ state: p.args, origin: \"create\", toolCallId: id });\n }\n } else if (p.toolName === toolName && updateCallTargets(p, id)) {\n const prev = current();\n if (prev) {\n const { id: _id, ...partial } = p.args as Record<string, unknown>;\n const idFactory = createAddedItemIdFactory(p.result);\n versions.push({\n state: idFactory\n ? shallowMergeInteractableState(prev.state, partial, {\n idFactory,\n })\n : shallowMergeInteractableState(prev.state, partial),\n origin: \"update\",\n toolCallId: p.toolCallId,\n });\n }\n }\n }\n }\n byId.set(id, versions);\n return versions;\n}\n\n/**\n * The state the model knows for interactable `id` — its latest recorded\n * version, or `undefined` when no baseline exists yet.\n */\nexport function findModelKnownState(\n messages: readonly SnapshotCarrierMessage[],\n id: string,\n name: string,\n): { state: unknown } | undefined {\n const versions = unstable_getInteractableVersions(messages, id, name);\n const last = versions[versions.length - 1];\n return last ? { state: last.state } : undefined;\n}\n\n/**\n * Snapshot gate. Stamps an interactable only when the model doesn't already\n * know its state (no baseline yet, or the live state differs from the folded\n * thread record). A change that fits a shallow merge is stamped as a partial\n * snapshot carrying just the changed fields; otherwise the full state is\n * stamped. Other metadata keys pass through untouched.\n */\nexport function gateInteractableComposerMetadata(\n meta: Record<string, unknown> | undefined,\n messages: readonly SnapshotCarrierMessage[],\n): Record<string, unknown> | undefined {\n if (!meta) return undefined;\n const { interactables, ...rest } = meta as {\n interactables?: Unstable_InteractableSnapshotEntry[];\n } & Record<string, unknown>;\n\n const gated: Record<string, unknown> = { ...rest };\n if (Array.isArray(interactables)) {\n const pending: Unstable_InteractableSnapshotEntry[] = [];\n for (const it of interactables) {\n if (process.env.NODE_ENV !== \"production\" && !isJSONValue(it.state)) {\n console.warn(\n `[Interactables] state for \"${it.name}\" (${it.id}) is not JSON-equatable ` +\n `(an undefined, NaN, Infinity, function, or symbol value?). It will be ` +\n `re-snapshotted on every send, recreating per-message growth. Use plain ` +\n `JSON values.`,\n );\n }\n const known = findModelKnownState(messages, it.id, it.name);\n if (!known) {\n pending.push({ id: it.id, name: it.name, state: it.state });\n continue;\n }\n if (isJSONValueEqual(it.state, known.state)) continue;\n const diff = shallowDiffInteractableState(known.state, it.state);\n pending.push(\n diff\n ? { id: it.id, name: it.name, state: diff, partial: true }\n : { id: it.id, name: it.name, state: it.state },\n );\n }\n if (pending.length) gated.interactables = pending;\n }\n return Object.keys(gated).length ? gated : undefined;\n}\n"],"mappings":";;;;;;;;;;;;AAuCA,SAAgBY,kCAAkCC,SAEG;CACnD,MAAMH,WAAWG,QAAQH;CACzB,IAAI,CAACA,YAAY,OAAOA,aAAa,UAAU,OAAOI,KAAAA;CACtD,MAAMC,SAAUL,SAAqCK;CACrD,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAOD,KAAAA;CAClD,MAAMG,QAASF,OAAmCG;CAClD,OAAOC,MAAMC,QAAQH,KAAK,IACrBA,QACDH,KAAAA;AACN;;;;;;AAOA,SAAgBO,oCACdC,OACQ;CACR,IAAIA,MAAMf,SACR,OAAO,cAAce,MAAMjB,KAAI,SAAUkB,KAAKC,UAAUF,MAAMlB,EAAE,EAAC,8BAA+BmB,KAAKC,UAAUF,MAAMhB,KAAK,EAAC;CAE7H,OAAO,sBAAsBgB,MAAMjB,KAAI,SAAUkB,KAAKC,UAAUF,MAAMlB,EAAE,EAAC,KAAMmB,KAAKC,UAAUF,MAAMhB,KAAK,EAAC;AAC5G;;AAGA,SAAgBmB,qBAAqBpB,MAAsB;CACzD,OAAO,UAAUA,KAAKqB,QAAQ,mBAAmB,GAAG;AACtD;AAEA,MAAMC,kBAAkBC,SAA+C;CACrE,IAAI,CAAC1B,SAAS0B,IAAI,GAAG,OAAOd,KAAAA;CAC5B,MAAMV,KAAKwB,KAAKxB;CAChB,OAAO,OAAOA,OAAO,YAAY,OAAOA,OAAO,WAAWA,KAAKU,KAAAA;AACjE;AAEA,SAASe,iBACPC,MACAC,QACAC,QACA;CACA,IAAIC,OAAOd,MAAMC,QAAQW,OAAOG,GAAG,IAAI,CAAC,GAAGH,OAAOG,GAAG,IAAI,CAAC,GAAGJ,IAAI;CAEjE,IAAIC,OAAOI,UAAU,MAAMF,OAAO,CAAA;CAElC,IAAId,MAAMC,QAAQW,OAAOK,MAAM,KAAKL,OAAOK,OAAOC,SAAS,GAAG;EAC5D,MAAMC,MAAM,IAAIC,IAAIR,OAAOK,MAAM;EACjCH,OAAOA,KAAKO,QAAQZ,SAAS;GAC3B,MAAMxB,KAAKuB,eAAeC,IAAI;GAC9B,OAAOxB,OAAOU,KAAAA,IAAY,CAACwB,IAAIG,IAAIrC,EAAE,IAAI,CAACkC,IAAIG,IAAIb,IAAI;EACxD,CAAC;CACH;CAEA,MAAMc,UAAUX,OAAOA;CACvB,IAAIZ,MAAMC,QAAQsB,OAAO,KAAKA,QAAQL,SAAS,GAC7CJ,OAAOA,KAAKU,KAAKf,SAAS;EACxB,MAAMxB,KAAKuB,eAAeC,IAAI;EAC9B,IAAIxB,OAAOU,KAAAA,KAAa,CAACZ,SAAS0B,IAAI,GAAG,OAAOA;EAChD,MAAMgB,QAAQF,QAAQG,MACnBC,cAAc5C,SAAS4C,SAAS,KAAKA,UAAU1C,OAAOA,EACzD;EACA,OAAOwC,QAAQ;GAAE,GAAGhB;GAAM,GAAGgB;EAAM,IAAIhB;CACzC,CAAC;CAGH,IAAIT,MAAMC,QAAQW,OAAOgB,GAAG,KAAKhB,OAAOgB,IAAIV,SAAS,GAAG;EACtD,MAAMW,QAAQhB,SACVD,OAAOgB,IAAIJ,KAAKf,SAAS;GACvB,IAAI,CAAC1B,SAAS0B,IAAI,KAAKA,KAAKxB,OAAOU,KAAAA,GAAW,OAAOc;GACrD,MAAMxB,KAAK4B,OAAO;GAClB,OAAO5B,OAAOU,KAAAA,IAAYc,OAAO;IAAE,GAAGA;IAAMxB;GAAG;EACjD,CAAC,IACD2B,OAAOgB;EACXd,OAAO,CAAC,GAAGA,MAAM,GAAGe,KAAK;CAC3B;CAEA,OAAOf;AACT;;;;;;AAOA,SAAgBgB,8BACdnB,MACAvB,SACA2C,SAYS;CACT,IAAI,CAAChD,SAAS4B,IAAI,KAAK,CAAC5B,SAASK,OAAO,GAAG,OAAOA;CAClD,MAAMiD,WAAWtD,SAASgD,SAASC,aAAa,IAC5CD,QAAQC,gBACRrB;CACJ,MAAMG,OAAO,EAAE,GAAGH,KAAK;CACvB,KAAK,MAAM,CAAC2B,KAAKC,UAAUC,OAAOC,QAAQrD,OAAO,GAAG;EAClD,MAAMsD,YAAYL,SAASC;EAC3B,IAAItC,MAAMC,QAAQyC,SAAS,KAAK3D,SAASwD,KAAK,GAM5CzB,KAAKwB,OAAO5B,iBAAiBgC,WAAWH,OAJtCR,SAASE,cACRF,QAAQI,kBAAkBxC,KAAAA,KAAaoC,QAAQI,cAAcb,IAAIgB,GAAG,WAC3DP,QAAQE,YAAYK,GAAG,IAC7B3C,KAAAA,CAC+C;OAErDmB,KAAKwB,OAAOC;CAEhB;CACA,OAAOzB;AACT;;;;;;;;AASA,SAAS6B,6BACPC,OACA9B,MACqC;CACrC,IAAI,CAAC/B,SAAS6D,KAAK,KAAK,CAAC7D,SAAS+B,IAAI,GAAG,OAAOnB,KAAAA;CAChD,KAAK,MAAM2C,OAAOE,OAAOK,KAAKD,KAAK,GACjC,IAAI,EAAEN,OAAOxB,OAAO,OAAOnB,KAAAA;CAE7B,MAAMmD,OAAgC,CAAC;CACvC,KAAK,MAAM,CAACR,KAAKC,UAAUC,OAAOC,QAAQ3B,IAAI,GAC5C,IAAI,EAAEwB,OAAOM,UAAU,CAAC/D,iBAAiB+D,MAAMN,MAAMC,KAAK,GACxDO,KAAKR,OAAOC;CAGhB,MAAMQ,UAAUP,OAAOK,KAAKC,IAAI,CAAC,CAAC5B;CAClC,IAAI6B,YAAY,KAAKA,YAAYP,OAAOK,KAAK/B,IAAI,CAAC,CAACI,QAAQ,OAAOvB,KAAAA;CAClE,OAAOmD;AACT;AAUA,MAAMQ,kBAAkBC,SAAgD;CACtE,IAAI,CAACA,QAAQ,OAAOA,SAAS,UAAU,OAAO5D,KAAAA;CAC9C,MAAM6D,IAAID;CACV,OAAOC,EAAEP,SAAS,cAAcO,IAAI7D,KAAAA;AACtC;;AAGA,MAAM8D,qBAAqBD,GAAqBvE,OAAwB;CACtE,IAAI,CAACuE,EAAEJ,QAAQ,OAAOI,EAAEJ,SAAS,UAAU,OAAO;CAClD,MAAMC,SAAStE,SAASyE,EAAEH,MAAM,IAAIG,EAAEH,SAAS1D,KAAAA;CAC/C,IAAI0D,QAAQK,YAAY,OAAO,OAAO;CACtC,IAAI,OAAOL,QAAQpE,OAAO,UAAU,OAAOoE,OAAOpE,OAAOA;CACzD,MAAM0E,SAAUH,EAAEJ,KAAiCnE;CACnD,OAAO0E,WAAW1E,MAAM0E,WAAWhE,KAAAA;AACrC;AAEA,MAAMiE,4BAA4BP,WAAoB;CACpD,MAAMd,QAAQxD,SAASsE,MAAM,IAAIA,OAAOQ,eAAelE,KAAAA;CACvD,IAAI,CAACZ,SAASwD,KAAK,GAAG,OAAO5C,KAAAA;CAE7B,MAAMmE,6BAAa,IAAIC,IAAsB;CAC7C,KAAK,MAAM,CAAC7B,OAAOf,QAAQqB,OAAOC,QAAQF,KAAK,GAAG;EAChD,IAAI,CAACvC,MAAMC,QAAQkB,GAAG,GAAG;EACzB,MAAM6C,YAAY7C,IAAIE,QACnBZ,SAAyB,OAAOA,SAAS,QAC5C;EACA,IAAIuD,UAAU9C,SAAS,GAAG4C,WAAW/C,IAAImB,OAAO8B,SAAS;CAC3D;CACA,IAAIF,WAAWG,SAAS,GAAG,OAAOtE,KAAAA;CAElC,QAAQuC,UAAkB4B,WAAWI,IAAIhC,KAAK,CAAC,EAAEiC,MAAM;AACzD;AAgBA,MAAMG,gCAAgB,IAAIC,QAGxB;;;;;;;;;;;;;;;;;AAkBF,SAAgBC,iCACdC,UACAxF,IACAC,MACgC;CAChC,IAAIwF,SAASJ,cAAcJ,IAAIO,QAAQ;CACvC,IAAI,CAACC,QAAQ;EACXA,yBAAS,IAAIX,IAAI;EACjBO,cAAcvD,IAAI0D,UAAUC,MAAM;CACpC;CACA,IAAIC,OAAOD,OAAOR,IAAIhF,IAAI;CAC1B,IAAI,CAACyF,MAAM;EACTA,uBAAO,IAAIZ,IAAI;EACfW,OAAO3D,IAAI7B,MAAMyF,IAAI;CACvB;CACA,MAAMC,SAASD,KAAKT,IAAIjF,EAAE;CAC1B,IAAI2F,QAAQ,OAAOA;CAEnB,MAAMzB,WAAW7C,qBAAqBpB,IAAI;CAC1C,MAAM2F,WAA2C,CAAA;CACjD,MAAMC,gBAAgBD,SAASA,SAAS3D,SAAS;CAEjD,KAAK,MAAM6D,OAAON,UAAU;EAC1B,IAAIM,IAAIzF,SAAS,QAAQ;GACvB,MAAMa,QAAQV,kCAAkCsF,GAAG,CAAC,EAAErD,MACnDsD,OAAOA,GAAG/F,OAAOA,EACpB;GACA,IAAI,CAACkB,OAAO;GACZ,IAAIA,MAAMf,SAAS;IACjB,MAAMuB,OAAOmE,QAAQ;IACrB,IAAInE,MACFkE,SAASI,KAAK;KACZ9F,OAAO2C,8BAA8BnB,KAAKxB,OAAOgB,MAAMhB,KAAK;KAC5DkF,QAAQ;IACV,CAAC;GAEL,OACEQ,SAASI,KAAK;IAAE9F,OAAOgB,MAAMhB;IAAOkF,QAAQ;GAAY,CAAC;GAE3D;EACF;EAEA,IAAIU,IAAIzF,SAAS,aAAa;EAC9B,KAAK,MAAMiE,QAAQwB,IAAIvF,WAAW,CAAA,GAAI;GACpC,MAAMgE,IAAIF,eAAeC,IAAI;GAC7B,IAAI,CAACC,GAAG;GACR,IAAIA,EAAEN,eAAejE,MAAMuE,EAAEL,aAAajE,MACpCsE;QAAAA,EAAEJ,QAAQ,OAAOI,EAAEJ,SAAS,UAC9ByB,SAASI,KAAK;KAAE9F,OAAOqE,EAAEJ;KAAMiB,QAAQ;KAAUnB,YAAYjE;IAAG,CAAC;GAAA,OAE9D,IAAIuE,EAAEL,aAAaA,YAAYM,kBAAkBD,GAAGvE,EAAE,GAAG;IAC9D,MAAM0B,OAAOmE,QAAQ;IACrB,IAAInE,MAAM;KACR,MAAM,EAAE1B,IAAIiG,KAAK,GAAG9F,YAAYoE,EAAEJ;KAClC,MAAMnB,YAAY2B,yBAAyBJ,EAAEH,MAAM;KACnDwB,SAASI,KAAK;MACZ9F,OAAO8C,YACHH,8BAA8BnB,KAAKxB,OAAOC,SAAS,EACjD6C,UACF,CAAC,IACDH,8BAA8BnB,KAAKxB,OAAOC,OAAO;MACrDiF,QAAQ;MACRnB,YAAYM,EAAEN;KAChB,CAAC;IACH;GACF;EACF;CACF;CACAyB,KAAK5D,IAAI9B,IAAI4F,QAAQ;CACrB,OAAOA;AACT;;;;;AAMA,SAAgBM,oBACdV,UACAxF,IACAC,MACgC;CAChC,MAAM2F,WAAWL,iCAAiCC,UAAUxF,IAAIC,IAAI;CACpE,MAAMkG,OAAOP,SAASA,SAAS3D,SAAS;CACxC,OAAOkE,OAAO,EAAEjG,OAAOiG,KAAKjG,MAAM,IAAIQ,KAAAA;AACxC;;;;;;;;AASA,SAAgB0F,iCACdC,MACAb,UACqC;CACrC,IAAI,CAACa,MAAM,OAAO3F,KAAAA;CAClB,MAAM,EAAEI,eAAe,GAAGwF,SAASD;CAInC,MAAME,QAAiC,EAAE,GAAGD,KAAK;CACjD,IAAIvF,MAAMC,QAAQF,aAAa,GAAG;EAChC,MAAM0F,UAAgD,CAAA;EACtD,KAAK,MAAMT,MAAMjF,eAAe;GAC9B,IAAI2F,QAAQC,IAAIC,aAAa,gBAAgB,CAAC9G,YAAYkG,GAAG7F,KAAK,GAChE0G,QAAQC,KACN,8BAA8Bd,GAAG9F,KAAI,KAAM8F,GAAG/F,GAAE,kLAIlD;GAEF,MAAM2D,QAAQuC,oBAAoBV,UAAUO,GAAG/F,IAAI+F,GAAG9F,IAAI;GAC1D,IAAI,CAAC0D,OAAO;IACV6C,QAAQR,KAAK;KAAEhG,IAAI+F,GAAG/F;KAAIC,MAAM8F,GAAG9F;KAAMC,OAAO6F,GAAG7F;IAAM,CAAC;IAC1D;GACF;GACA,IAAIN,iBAAiBmG,GAAG7F,OAAOyD,MAAMzD,KAAK,GAAG;GAC7C,MAAM2D,OAAOH,6BAA6BC,MAAMzD,OAAO6F,GAAG7F,KAAK;GAC/DsG,QAAQR,KACNnC,OACI;IAAE7D,IAAI+F,GAAG/F;IAAIC,MAAM8F,GAAG9F;IAAMC,OAAO2D;IAAM1D,SAAS;GAAK,IACvD;IAAEH,IAAI+F,GAAG/F;IAAIC,MAAM8F,GAAG9F;IAAMC,OAAO6F,GAAG7F;GAAM,CAClD;EACF;EACA,IAAIsG,QAAQvE,QAAQsE,MAAMzF,gBAAgB0F;CAC5C;CACA,OAAOjD,OAAOK,KAAK2C,KAAK,CAAC,CAACtE,SAASsE,QAAQ7F,KAAAA;AAC7C"}
1
+ {"version":3,"file":"interactable-composer-metadata.js","names":["isJSONValueEqual","isJSONValue","isRecord","Unstable_InteractableSnapshotEntry","id","name","state","partial","SnapshotCarrierMessage","role","metadata","content","unstable_getInteractableSnapshots","message","undefined","custom","Record","items","interactables","Array","isArray","unstable_formatInteractableSnapshot","entry","JSON","stringify","interactableToolName","replace","getArrayItemId","item","applyArrayUpdate","prev","update","mintId","next","set","clear","remove","length","ids","Set","filter","has","patches","patchesById","Map","candidate","Number","isNaN","map","patch","get","add","added","shallowMergeInteractableState","options","arrayBaseline","idFactory","field","idKeyedFields","ReadonlySet","baseline","key","value","Object","entries","baseValue","shallowDiffInteractableState","known","keys","diff","changed","ToolCallLikePart","type","toolCallId","toolName","args","result","asToolCallPart","part","p","updateCallTargets","success","argsId","createAddedItemIdFactory","addedItemIds","idsByField","stringIds","size","shift","Unstable_InteractableVersion","origin","versionsCache","WeakMap","unstable_getInteractableVersions","messages","byName","byId","cached","versions","current","msg","find","it","push","_id","findModelKnownState","last","gateInteractableComposerMetadata","meta","rest","gated","pending","process","env","NODE_ENV","console","warn"],"sources":["../../src/model-context/interactable-composer-metadata.ts"],"sourcesContent":["import { isJSONValueEqual } from \"../utils/json/is-json-equal\";\nimport { isJSONValue, isRecord } from \"../utils/json/is-json\";\n\n/**\n * Unstable / Experimental — the interactables API is still evolving and may change in any release.\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport type Unstable_InteractableSnapshotEntry = {\n id: string;\n name: string;\n state: unknown;\n /**\n * When true, `state` carries only the top-level fields that changed since\n * the model's last known state; omitted fields are unchanged. The first\n * snapshot of an instance is always full.\n */\n partial?: boolean | undefined;\n};\n\n/**\n * Minimal message shape needed to read snapshots and `update_*` tool calls out\n * of history. Structurally compatible with both `ThreadMessage` and the AI\n * SDK's `UIMessage`.\n */\ntype SnapshotCarrierMessage = {\n role: string;\n metadata?: unknown;\n content?: readonly unknown[] | undefined;\n};\n\n/**\n * Reads the interactable snapshots stamped on a message's\n * `metadata.custom.interactables`, or `undefined` if none. This is the read\n * half of the snapshot channel — integrations use it to surface interactable\n * state to the model (see `unstable_injectInteractableContext` in\n * `@assistant-ui/ai-sdk` for the AI SDK implementation).\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport function unstable_getInteractableSnapshots(message: {\n metadata?: unknown;\n}): Unstable_InteractableSnapshotEntry[] | undefined {\n const metadata = message.metadata;\n if (!metadata || typeof metadata !== \"object\") return undefined;\n const custom = (metadata as Record<string, unknown>).custom;\n if (!custom || typeof custom !== \"object\") return undefined;\n const items = (custom as Record<string, unknown>).interactables;\n return Array.isArray(items)\n ? (items as Unstable_InteractableSnapshotEntry[])\n : undefined;\n}\n\n/**\n * Canonical model-facing wording for one snapshot entry.\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport function unstable_formatInteractableSnapshot(\n entry: Unstable_InteractableSnapshotEntry,\n): string {\n if (entry.partial) {\n return `[State of \"${entry.name}\" (id: ${JSON.stringify(entry.id)}) changed — updated fields: ${JSON.stringify(entry.state)}; fields not listed are unchanged]`;\n }\n return `[Current state of \"${entry.name}\" (id: ${JSON.stringify(entry.id)}): ${JSON.stringify(entry.state)}]`;\n}\n\n/** The model-facing tool name for an interactable `name`. One tool per name. */\nexport function interactableToolName(name: string): string {\n return `update_${name.replace(/[^a-zA-Z0-9_-]/g, \"_\")}`;\n}\n\nconst getArrayItemId = (item: unknown): string | number | undefined => {\n if (!isRecord(item)) return undefined;\n const id = item.id;\n return typeof id === \"string\" || typeof id === \"number\" ? id : undefined;\n};\n\nfunction applyArrayUpdate(\n prev: unknown[],\n update: Record<string, unknown>,\n mintId?: () => string | undefined,\n) {\n let next = Array.isArray(update.set) ? [...update.set] : [...prev];\n\n if (update.clear === true) next = [];\n\n if (Array.isArray(update.remove) && update.remove.length > 0) {\n const ids = new Set(update.remove);\n next = next.filter((item) => {\n const id = getArrayItemId(item);\n return id !== undefined ? !ids.has(id) : !ids.has(item);\n });\n }\n\n const patches = update.update;\n if (Array.isArray(patches) && patches.length > 0) {\n const patchesById = new Map<string | number, Record<string, unknown>>();\n for (const candidate of patches) {\n const id = getArrayItemId(candidate);\n if (id !== undefined && !Number.isNaN(id) && !patchesById.has(id)) {\n patchesById.set(id, candidate as Record<string, unknown>);\n }\n }\n\n next = next.map((item) => {\n const id = getArrayItemId(item);\n if (id === undefined || !isRecord(item)) return item;\n const patch = patchesById.get(id);\n return patch ? { ...item, ...patch } : item;\n });\n }\n\n if (Array.isArray(update.add) && update.add.length > 0) {\n const added = mintId\n ? update.add.map((item) => {\n if (!isRecord(item) || item.id !== undefined) return item;\n const id = mintId();\n return id === undefined ? item : { ...item, id };\n })\n : update.add;\n next = [...next, ...added];\n }\n\n return next;\n}\n\n/**\n * The merge `update_*` tools apply: top-level keys replace, rest preserved.\n * Array fields also accept operation objects (`add`, `update`, `remove`,\n * `clear`, `set`) so collection edits do not need a companion manage_* tool.\n */\nexport function shallowMergeInteractableState(\n prev: unknown,\n partial: unknown,\n options?:\n | {\n arrayBaseline?: unknown;\n /**\n * Supplies ids for added items in id-keyed array fields. The live\n * execute path mints fresh ids; history folds replay ids previously\n * returned by execute.\n */\n idFactory?: (field: string) => string | undefined;\n idKeyedFields?: ReadonlySet<string>;\n }\n | undefined,\n): unknown {\n if (!isRecord(prev) || !isRecord(partial)) return partial;\n const baseline = isRecord(options?.arrayBaseline)\n ? options.arrayBaseline\n : prev;\n const next = { ...prev };\n for (const [key, value] of Object.entries(partial)) {\n const baseValue = baseline[key];\n if (Array.isArray(baseValue) && isRecord(value)) {\n const mintId =\n options?.idFactory &&\n (options.idKeyedFields === undefined || options.idKeyedFields.has(key))\n ? () => options.idFactory?.(key)\n : undefined;\n next[key] = applyArrayUpdate(baseValue, value, mintId);\n } else {\n next[key] = value;\n }\n }\n return next;\n}\n\n/**\n * The top-level fields of `next` that differ from `known`, for stamping a\n * partial snapshot. Returns `undefined` when a partial stamp has no benefit:\n * the change is not expressible as a shallow merge (non-object states, or a\n * top-level key was removed), or every field changed anyway — the caller\n * falls back to a full snapshot.\n */\nfunction shallowDiffInteractableState(\n known: unknown,\n next: unknown,\n): Record<string, unknown> | undefined {\n if (!isRecord(known) || !isRecord(next)) return undefined;\n for (const key of Object.keys(known)) {\n if (!(key in next)) return undefined;\n }\n const diff: Record<string, unknown> = {};\n for (const [key, value] of Object.entries(next)) {\n if (!(key in known) || !isJSONValueEqual(known[key], value)) {\n diff[key] = value;\n }\n }\n const changed = Object.keys(diff).length;\n if (changed === 0 || changed === Object.keys(next).length) return undefined;\n return diff;\n}\n\ntype ToolCallLikePart = {\n type?: string;\n toolCallId?: string;\n toolName?: string;\n args?: unknown;\n result?: unknown;\n};\n\nconst asToolCallPart = (part: unknown): ToolCallLikePart | undefined => {\n if (!part || typeof part !== \"object\") return undefined;\n const p = part as ToolCallLikePart;\n return p.type === \"tool-call\" ? p : undefined;\n};\n\n/** Whether an accepted `update_*` call addressed instance `id`. */\nconst updateCallTargets = (p: ToolCallLikePart, id: string): boolean => {\n if (!p.args || typeof p.args !== \"object\") return false;\n const result = isRecord(p.result) ? p.result : undefined;\n if (result?.success === false) return false;\n if (typeof result?.id === \"string\") return result.id === id;\n const argsId = (p.args as Record<string, unknown>).id;\n return argsId === id || argsId === undefined;\n};\n\nconst createAddedItemIdFactory = (result: unknown) => {\n const value = isRecord(result) ? result.addedItemIds : undefined;\n if (!isRecord(value)) return undefined;\n\n const idsByField = new Map<string, string[]>();\n for (const [field, ids] of Object.entries(value)) {\n if (!Array.isArray(ids)) continue;\n const stringIds = ids.filter(\n (item): item is string => typeof item === \"string\",\n );\n if (stringIds.length > 0) idsByField.set(field, stringIds);\n }\n if (idsByField.size === 0) return undefined;\n\n return (field: string) => idsByField.get(field)?.shift();\n};\n\n/**\n * Unstable / Experimental — the interactables API is still evolving and may change in any release.\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport type Unstable_InteractableVersion = {\n /** The full state as of this version. */\n state: unknown;\n origin: \"create\" | \"update\" | \"user-edit\";\n /** For create/update versions, the tool call that produced this version. */\n toolCallId?: string | undefined;\n};\n\n// Without this, every mounted anchor re-folds the whole thread on each streaming\n// token, since the repository hands out a new messages array per token.\nconst versionsCache = new WeakMap<\n readonly SnapshotCarrierMessage[],\n Map<string, Map<string, Unstable_InteractableVersion[]>>\n>();\n\n/**\n * Every version of interactable `id` recorded in the thread, oldest first,\n * folded chronologically:\n *\n * - the tool call whose `toolCallId` equals `id` seeds the baseline with its\n * args (`origin: \"create\"`) — the `id: toolCallId` convention for\n * tool-created interactables,\n * - a snapshot stamped on a user message is a `\"user-edit\"` version (a full\n * snapshot replaces the state; a partial one shallow-merges),\n * - each accepted `update_*` call shallow-merges an `\"update\"` version.\n *\n * The last entry is the state the model knows. Partial snapshots and update\n * calls with no baseline to merge into are skipped.\n *\n * @deprecated Unstable / Experimental (not actually removed).\n */\nexport function unstable_getInteractableVersions(\n messages: readonly SnapshotCarrierMessage[],\n id: string,\n name: string,\n): Unstable_InteractableVersion[] {\n let byName = versionsCache.get(messages);\n if (!byName) {\n byName = new Map();\n versionsCache.set(messages, byName);\n }\n let byId = byName.get(name);\n if (!byId) {\n byId = new Map();\n byName.set(name, byId);\n }\n const cached = byId.get(id);\n if (cached) return cached;\n\n const toolName = interactableToolName(name);\n const versions: Unstable_InteractableVersion[] = [];\n const current = () => versions[versions.length - 1];\n\n for (const msg of messages) {\n if (msg.role === \"user\") {\n const entry = unstable_getInteractableSnapshots(msg)?.find(\n (it) => it.id === id,\n );\n if (!entry) continue;\n if (entry.partial) {\n const prev = current();\n if (prev) {\n versions.push({\n state: shallowMergeInteractableState(prev.state, entry.state),\n origin: \"user-edit\",\n });\n }\n } else {\n versions.push({ state: entry.state, origin: \"user-edit\" });\n }\n continue;\n }\n\n if (msg.role !== \"assistant\") continue;\n for (const part of msg.content ?? []) {\n const p = asToolCallPart(part);\n if (!p) continue;\n if (p.toolCallId === id && p.toolName === name) {\n if (p.args && typeof p.args === \"object\") {\n versions.push({ state: p.args, origin: \"create\", toolCallId: id });\n }\n } else if (p.toolName === toolName && updateCallTargets(p, id)) {\n const prev = current();\n if (prev) {\n const { id: _id, ...partial } = p.args as Record<string, unknown>;\n const idFactory = createAddedItemIdFactory(p.result);\n versions.push({\n state: idFactory\n ? shallowMergeInteractableState(prev.state, partial, {\n idFactory,\n })\n : shallowMergeInteractableState(prev.state, partial),\n origin: \"update\",\n toolCallId: p.toolCallId,\n });\n }\n }\n }\n }\n byId.set(id, versions);\n return versions;\n}\n\n/**\n * The state the model knows for interactable `id` — its latest recorded\n * version, or `undefined` when no baseline exists yet.\n */\nexport function findModelKnownState(\n messages: readonly SnapshotCarrierMessage[],\n id: string,\n name: string,\n): { state: unknown } | undefined {\n const versions = unstable_getInteractableVersions(messages, id, name);\n const last = versions[versions.length - 1];\n return last ? { state: last.state } : undefined;\n}\n\n/**\n * Snapshot gate. Stamps an interactable only when the model doesn't already\n * know its state (no baseline yet, or the live state differs from the folded\n * thread record). A change that fits a shallow merge is stamped as a partial\n * snapshot carrying just the changed fields; otherwise the full state is\n * stamped. Other metadata keys pass through untouched.\n */\nexport function gateInteractableComposerMetadata(\n meta: Record<string, unknown> | undefined,\n messages: readonly SnapshotCarrierMessage[],\n): Record<string, unknown> | undefined {\n if (!meta) return undefined;\n const { interactables, ...rest } = meta as {\n interactables?: Unstable_InteractableSnapshotEntry[];\n } & Record<string, unknown>;\n\n const gated: Record<string, unknown> = { ...rest };\n if (Array.isArray(interactables)) {\n const pending: Unstable_InteractableSnapshotEntry[] = [];\n for (const it of interactables) {\n if (process.env.NODE_ENV !== \"production\" && !isJSONValue(it.state)) {\n console.warn(\n `[Interactables] state for \"${it.name}\" (${it.id}) is not JSON-equatable ` +\n `(an undefined, NaN, Infinity, function, or symbol value?). It will be ` +\n `re-snapshotted on every send, recreating per-message growth. Use plain ` +\n `JSON values.`,\n );\n }\n const known = findModelKnownState(messages, it.id, it.name);\n if (!known) {\n pending.push({ id: it.id, name: it.name, state: it.state });\n continue;\n }\n if (isJSONValueEqual(it.state, known.state)) continue;\n const diff = shallowDiffInteractableState(known.state, it.state);\n pending.push(\n diff\n ? { id: it.id, name: it.name, state: diff, partial: true }\n : { id: it.id, name: it.name, state: it.state },\n );\n }\n if (pending.length) gated.interactables = pending;\n }\n return Object.keys(gated).length ? gated : undefined;\n}\n"],"mappings":";;;;;;;;;;;;AAuCA,SAAgBY,kCAAkCC,SAEG;CACnD,MAAMH,WAAWG,QAAQH;CACzB,IAAI,CAACA,YAAY,OAAOA,aAAa,UAAU,OAAOI,KAAAA;CACtD,MAAMC,SAAUL,SAAqCK;CACrD,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAOD,KAAAA;CAClD,MAAMG,QAASF,OAAmCG;CAClD,OAAOC,MAAMC,QAAQH,KAAK,IACrBA,QACDH,KAAAA;AACN;;;;;;AAOA,SAAgBO,oCACdC,OACQ;CACR,IAAIA,MAAMf,SACR,OAAO,cAAce,MAAMjB,KAAI,SAAUkB,KAAKC,UAAUF,MAAMlB,EAAE,EAAC,8BAA+BmB,KAAKC,UAAUF,MAAMhB,KAAK,EAAC;CAE7H,OAAO,sBAAsBgB,MAAMjB,KAAI,SAAUkB,KAAKC,UAAUF,MAAMlB,EAAE,EAAC,KAAMmB,KAAKC,UAAUF,MAAMhB,KAAK,EAAC;AAC5G;;AAGA,SAAgBmB,qBAAqBpB,MAAsB;CACzD,OAAO,UAAUA,KAAKqB,QAAQ,mBAAmB,GAAG;AACtD;AAEA,MAAMC,kBAAkBC,SAA+C;CACrE,IAAI,CAAC1B,SAAS0B,IAAI,GAAG,OAAOd,KAAAA;CAC5B,MAAMV,KAAKwB,KAAKxB;CAChB,OAAO,OAAOA,OAAO,YAAY,OAAOA,OAAO,WAAWA,KAAKU,KAAAA;AACjE;AAEA,SAASe,iBACPC,MACAC,QACAC,QACA;CACA,IAAIC,OAAOd,MAAMC,QAAQW,OAAOG,GAAG,IAAI,CAAC,GAAGH,OAAOG,GAAG,IAAI,CAAC,GAAGJ,IAAI;CAEjE,IAAIC,OAAOI,UAAU,MAAMF,OAAO,CAAA;CAElC,IAAId,MAAMC,QAAQW,OAAOK,MAAM,KAAKL,OAAOK,OAAOC,SAAS,GAAG;EAC5D,MAAMC,MAAM,IAAIC,IAAIR,OAAOK,MAAM;EACjCH,OAAOA,KAAKO,QAAQZ,SAAS;GAC3B,MAAMxB,KAAKuB,eAAeC,IAAI;GAC9B,OAAOxB,OAAOU,KAAAA,IAAY,CAACwB,IAAIG,IAAIrC,EAAE,IAAI,CAACkC,IAAIG,IAAIb,IAAI;EACxD,CAAC;CACH;CAEA,MAAMc,UAAUX,OAAOA;CACvB,IAAIZ,MAAMC,QAAQsB,OAAO,KAAKA,QAAQL,SAAS,GAAG;EAChD,MAAMM,8BAAc,IAAIC,IAA8C;EACtE,KAAK,MAAMC,aAAaH,SAAS;GAC/B,MAAMtC,KAAKuB,eAAekB,SAAS;GACnC,IAAIzC,OAAOU,KAAAA,KAAa,CAACgC,OAAOC,MAAM3C,EAAE,KAAK,CAACuC,YAAYF,IAAIrC,EAAE,GAC9DuC,YAAYT,IAAI9B,IAAIyC,SAAoC;EAE5D;EAEAZ,OAAOA,KAAKe,KAAKpB,SAAS;GACxB,MAAMxB,KAAKuB,eAAeC,IAAI;GAC9B,IAAIxB,OAAOU,KAAAA,KAAa,CAACZ,SAAS0B,IAAI,GAAG,OAAOA;GAChD,MAAMqB,QAAQN,YAAYO,IAAI9C,EAAE;GAChC,OAAO6C,QAAQ;IAAE,GAAGrB;IAAM,GAAGqB;GAAM,IAAIrB;EACzC,CAAC;CACH;CAEA,IAAIT,MAAMC,QAAQW,OAAOoB,GAAG,KAAKpB,OAAOoB,IAAId,SAAS,GAAG;EACtD,MAAMe,QAAQpB,SACVD,OAAOoB,IAAIH,KAAKpB,SAAS;GACvB,IAAI,CAAC1B,SAAS0B,IAAI,KAAKA,KAAKxB,OAAOU,KAAAA,GAAW,OAAOc;GACrD,MAAMxB,KAAK4B,OAAO;GAClB,OAAO5B,OAAOU,KAAAA,IAAYc,OAAO;IAAE,GAAGA;IAAMxB;GAAG;EACjD,CAAC,IACD2B,OAAOoB;EACXlB,OAAO,CAAC,GAAGA,MAAM,GAAGmB,KAAK;CAC3B;CAEA,OAAOnB;AACT;;;;;;AAOA,SAAgBoB,8BACdvB,MACAvB,SACA+C,SAYS;CACT,IAAI,CAACpD,SAAS4B,IAAI,KAAK,CAAC5B,SAASK,OAAO,GAAG,OAAOA;CAClD,MAAMqD,WAAW1D,SAASoD,SAASC,aAAa,IAC5CD,QAAQC,gBACRzB;CACJ,MAAMG,OAAO,EAAE,GAAGH,KAAK;CACvB,KAAK,MAAM,CAAC+B,KAAKC,UAAUC,OAAOC,QAAQzD,OAAO,GAAG;EAClD,MAAM0D,YAAYL,SAASC;EAC3B,IAAI1C,MAAMC,QAAQ6C,SAAS,KAAK/D,SAAS4D,KAAK,GAM5C7B,KAAK4B,OAAOhC,iBAAiBoC,WAAWH,OAJtCR,SAASE,cACRF,QAAQI,kBAAkB5C,KAAAA,KAAawC,QAAQI,cAAcjB,IAAIoB,GAAG,WAC3DP,QAAQE,YAAYK,GAAG,IAC7B/C,KAAAA,CAC+C;OAErDmB,KAAK4B,OAAOC;CAEhB;CACA,OAAO7B;AACT;;;;;;;;AASA,SAASiC,6BACPC,OACAlC,MACqC;CACrC,IAAI,CAAC/B,SAASiE,KAAK,KAAK,CAACjE,SAAS+B,IAAI,GAAG,OAAOnB,KAAAA;CAChD,KAAK,MAAM+C,OAAOE,OAAOK,KAAKD,KAAK,GACjC,IAAI,EAAEN,OAAO5B,OAAO,OAAOnB,KAAAA;CAE7B,MAAMuD,OAAgC,CAAC;CACvC,KAAK,MAAM,CAACR,KAAKC,UAAUC,OAAOC,QAAQ/B,IAAI,GAC5C,IAAI,EAAE4B,OAAOM,UAAU,CAACnE,iBAAiBmE,MAAMN,MAAMC,KAAK,GACxDO,KAAKR,OAAOC;CAGhB,MAAMQ,UAAUP,OAAOK,KAAKC,IAAI,CAAC,CAAChC;CAClC,IAAIiC,YAAY,KAAKA,YAAYP,OAAOK,KAAKnC,IAAI,CAAC,CAACI,QAAQ,OAAOvB,KAAAA;CAClE,OAAOuD;AACT;AAUA,MAAMQ,kBAAkBC,SAAgD;CACtE,IAAI,CAACA,QAAQ,OAAOA,SAAS,UAAU,OAAOhE,KAAAA;CAC9C,MAAMiE,IAAID;CACV,OAAOC,EAAEP,SAAS,cAAcO,IAAIjE,KAAAA;AACtC;;AAGA,MAAMkE,qBAAqBD,GAAqB3E,OAAwB;CACtE,IAAI,CAAC2E,EAAEJ,QAAQ,OAAOI,EAAEJ,SAAS,UAAU,OAAO;CAClD,MAAMC,SAAS1E,SAAS6E,EAAEH,MAAM,IAAIG,EAAEH,SAAS9D,KAAAA;CAC/C,IAAI8D,QAAQK,YAAY,OAAO,OAAO;CACtC,IAAI,OAAOL,QAAQxE,OAAO,UAAU,OAAOwE,OAAOxE,OAAOA;CACzD,MAAM8E,SAAUH,EAAEJ,KAAiCvE;CACnD,OAAO8E,WAAW9E,MAAM8E,WAAWpE,KAAAA;AACrC;AAEA,MAAMqE,4BAA4BP,WAAoB;CACpD,MAAMd,QAAQ5D,SAAS0E,MAAM,IAAIA,OAAOQ,eAAetE,KAAAA;CACvD,IAAI,CAACZ,SAAS4D,KAAK,GAAG,OAAOhD,KAAAA;CAE7B,MAAMuE,6BAAa,IAAIzC,IAAsB;CAC7C,KAAK,MAAM,CAACa,OAAOnB,QAAQyB,OAAOC,QAAQF,KAAK,GAAG;EAChD,IAAI,CAAC3C,MAAMC,QAAQkB,GAAG,GAAG;EACzB,MAAMgD,YAAYhD,IAAIE,QACnBZ,SAAyB,OAAOA,SAAS,QAC5C;EACA,IAAI0D,UAAUjD,SAAS,GAAGgD,WAAWnD,IAAIuB,OAAO6B,SAAS;CAC3D;CACA,IAAID,WAAWE,SAAS,GAAG,OAAOzE,KAAAA;CAElC,QAAQ2C,UAAkB4B,WAAWnC,IAAIO,KAAK,CAAC,EAAE+B,MAAM;AACzD;AAgBA,MAAMG,gCAAgB,IAAIC,QAGxB;;;;;;;;;;;;;;;;;AAkBF,SAAgBC,iCACdC,UACA1F,IACAC,MACgC;CAChC,IAAI0F,SAASJ,cAAczC,IAAI4C,QAAQ;CACvC,IAAI,CAACC,QAAQ;EACXA,yBAAS,IAAInD,IAAI;EACjB+C,cAAczD,IAAI4D,UAAUC,MAAM;CACpC;CACA,IAAIC,OAAOD,OAAO7C,IAAI7C,IAAI;CAC1B,IAAI,CAAC2F,MAAM;EACTA,uBAAO,IAAIpD,IAAI;EACfmD,OAAO7D,IAAI7B,MAAM2F,IAAI;CACvB;CACA,MAAMC,SAASD,KAAK9C,IAAI9C,EAAE;CAC1B,IAAI6F,QAAQ,OAAOA;CAEnB,MAAMvB,WAAWjD,qBAAqBpB,IAAI;CAC1C,MAAM6F,WAA2C,CAAA;CACjD,MAAMC,gBAAgBD,SAASA,SAAS7D,SAAS;CAEjD,KAAK,MAAM+D,OAAON,UAAU;EAC1B,IAAIM,IAAI3F,SAAS,QAAQ;GACvB,MAAMa,QAAQV,kCAAkCwF,GAAG,CAAC,EAAEC,MACnDC,OAAOA,GAAGlG,OAAOA,EACpB;GACA,IAAI,CAACkB,OAAO;GACZ,IAAIA,MAAMf,SAAS;IACjB,MAAMuB,OAAOqE,QAAQ;IACrB,IAAIrE,MACFoE,SAASK,KAAK;KACZjG,OAAO+C,8BAA8BvB,KAAKxB,OAAOgB,MAAMhB,KAAK;KAC5DoF,QAAQ;IACV,CAAC;GAEL,OACEQ,SAASK,KAAK;IAAEjG,OAAOgB,MAAMhB;IAAOoF,QAAQ;GAAY,CAAC;GAE3D;EACF;EAEA,IAAIU,IAAI3F,SAAS,aAAa;EAC9B,KAAK,MAAMqE,QAAQsB,IAAIzF,WAAW,CAAA,GAAI;GACpC,MAAMoE,IAAIF,eAAeC,IAAI;GAC7B,IAAI,CAACC,GAAG;GACR,IAAIA,EAAEN,eAAerE,MAAM2E,EAAEL,aAAarE,MACpC0E;QAAAA,EAAEJ,QAAQ,OAAOI,EAAEJ,SAAS,UAC9BuB,SAASK,KAAK;KAAEjG,OAAOyE,EAAEJ;KAAMe,QAAQ;KAAUjB,YAAYrE;IAAG,CAAC;GAAA,OAE9D,IAAI2E,EAAEL,aAAaA,YAAYM,kBAAkBD,GAAG3E,EAAE,GAAG;IAC9D,MAAM0B,OAAOqE,QAAQ;IACrB,IAAIrE,MAAM;KACR,MAAM,EAAE1B,IAAIoG,KAAK,GAAGjG,YAAYwE,EAAEJ;KAClC,MAAMnB,YAAY2B,yBAAyBJ,EAAEH,MAAM;KACnDsB,SAASK,KAAK;MACZjG,OAAOkD,YACHH,8BAA8BvB,KAAKxB,OAAOC,SAAS,EACjDiD,UACF,CAAC,IACDH,8BAA8BvB,KAAKxB,OAAOC,OAAO;MACrDmF,QAAQ;MACRjB,YAAYM,EAAEN;KAChB,CAAC;IACH;GACF;EACF;CACF;CACAuB,KAAK9D,IAAI9B,IAAI8F,QAAQ;CACrB,OAAOA;AACT;;;;;AAMA,SAAgBO,oBACdX,UACA1F,IACAC,MACgC;CAChC,MAAM6F,WAAWL,iCAAiCC,UAAU1F,IAAIC,IAAI;CACpE,MAAMqG,OAAOR,SAASA,SAAS7D,SAAS;CACxC,OAAOqE,OAAO,EAAEpG,OAAOoG,KAAKpG,MAAM,IAAIQ,KAAAA;AACxC;;;;;;;;AASA,SAAgB6F,iCACdC,MACAd,UACqC;CACrC,IAAI,CAACc,MAAM,OAAO9F,KAAAA;CAClB,MAAM,EAAEI,eAAe,GAAG2F,SAASD;CAInC,MAAME,QAAiC,EAAE,GAAGD,KAAK;CACjD,IAAI1F,MAAMC,QAAQF,aAAa,GAAG;EAChC,MAAM6F,UAAgD,CAAA;EACtD,KAAK,MAAMT,MAAMpF,eAAe;GAC9B,IAAI8F,QAAQC,IAAIC,aAAa,gBAAgB,CAACjH,YAAYqG,GAAGhG,KAAK,GAChE6G,QAAQC,KACN,8BAA8Bd,GAAGjG,KAAI,KAAMiG,GAAGlG,GAAE,kLAIlD;GAEF,MAAM+D,QAAQsC,oBAAoBX,UAAUQ,GAAGlG,IAAIkG,GAAGjG,IAAI;GAC1D,IAAI,CAAC8D,OAAO;IACV4C,QAAQR,KAAK;KAAEnG,IAAIkG,GAAGlG;KAAIC,MAAMiG,GAAGjG;KAAMC,OAAOgG,GAAGhG;IAAM,CAAC;IAC1D;GACF;GACA,IAAIN,iBAAiBsG,GAAGhG,OAAO6D,MAAM7D,KAAK,GAAG;GAC7C,MAAM+D,OAAOH,6BAA6BC,MAAM7D,OAAOgG,GAAGhG,KAAK;GAC/DyG,QAAQR,KACNlC,OACI;IAAEjE,IAAIkG,GAAGlG;IAAIC,MAAMiG,GAAGjG;IAAMC,OAAO+D;IAAM9D,SAAS;GAAK,IACvD;IAAEH,IAAIkG,GAAGlG;IAAIC,MAAMiG,GAAGjG;IAAMC,OAAOgG,GAAGhG;GAAM,CAClD;EACF;EACA,IAAIyG,QAAQ1E,QAAQyE,MAAM5F,gBAAgB6F;CAC5C;CACA,OAAOhD,OAAOK,KAAK0C,KAAK,CAAC,CAACzE,SAASyE,QAAQhG,KAAAA;AAC7C"}
@@ -1 +1 @@
1
- {"version":3,"file":"registry.d.ts","names":[],"sources":["../../src/model-context/registry.ts"],"mappings":";;;;cAgBa,gCAAgC;UACnC;UACA;UACA;UACA;;EAQR,mBAAmB;EA+CnB,UAAU,uBAAuB;UAKzB;EAIR,QAAQ,cAAc,yBAAyB,SAC7C,MAAM,mBAAmB,OAAO,WAC/B,+BAA+B,OAAO;EAoBzC,eACE,iBAAiB,8BAChB;EAiCH,YACE,UAAU,uBACT"}
1
+ {"version":3,"file":"registry.d.ts","names":[],"sources":["../../src/model-context/registry.ts"],"mappings":";;;;cAiBa,gCAAgC;UACnC;UACA;UACA;UACA;;EAQR,mBAAmB;EA+CnB,UAAU,uBAAuB;UAKzB;EAIR,QAAQ,cAAc,yBAAyB,SAC7C,MAAM,mBAAmB,OAAO,WAC/B,+BAA+B,OAAO;EAoBzC,eACE,iBAAiB,8BAChB;EAiCH,YACE,UAAU,uBACT"}
@@ -1,3 +1,4 @@
1
+ import { nullProtoRecord } from "../utils/record.js";
1
2
  import { notifySubscribers } from "../subscribable/subscribable.js";
2
3
  import { CompositeContextProvider } from "../utils/composite-context-provider.js";
3
4
  //#region src/model-context/registry.ts
@@ -14,7 +15,7 @@ var ModelContextRegistry = class {
14
15
  getModelContext() {
15
16
  const instructions = Array.from(this._instructions.values()).filter(Boolean);
16
17
  const system = instructions.length > 0 ? instructions.join("\n\n") : void 0;
17
- const tools = {};
18
+ const tools = nullProtoRecord();
18
19
  for (const toolProps of this._tools.values()) {
19
20
  const { toolName, render, ...tool } = toolProps;
20
21
  tools[toolName] = tool;
@@ -25,10 +26,7 @@ var ModelContextRegistry = class {
25
26
  tools: Object.keys(tools).length > 0 ? tools : void 0
26
27
  };
27
28
  if (providerContexts.system) context.system = context.system ? `${context.system}\n\n${providerContexts.system}` : providerContexts.system;
28
- if (providerContexts.tools) context.tools = {
29
- ...context.tools || {},
30
- ...providerContexts.tools
31
- };
29
+ if (providerContexts.tools) context.tools = nullProtoRecord(context.tools, providerContexts.tools);
32
30
  if (providerContexts.callSettings) context.callSettings = providerContexts.callSettings;
33
31
  if (providerContexts.config) context.config = providerContexts.config;
34
32
  if (providerContexts.unstable_composerMetadata) context.unstable_composerMetadata = providerContexts.unstable_composerMetadata;
@@ -1 +1 @@
1
- {"version":3,"file":"registry.js","names":["Tool","ModelContext","ModelContextProvider","AssistantToolProps","AssistantInstructionsConfig","notifySubscribers","notifyStateSubscribers","CompositeContextProvider","Unsubscribe","ModelContextRegistryToolHandle","ModelContextRegistryInstructionHandle","ModelContextRegistryProviderHandle","ModelContextRegistry","_tools","Map","_instructions","_contextProviders","_subscribers","Set","constructor","subscribe","getModelContext","instructions","Array","from","values","filter","Boolean","system","length","join","undefined","tools","Record","toolProps","toolName","render","tool","providerContexts","context","Object","keys","callSettings","config","unstable_composerMetadata","callback","add","delete","addTool","TArgs","TResult","id","Symbol","set","update","newTool","has","remove","addInstruction","instruction","disabled","newConfig","newInstruction","newDisabled","addProvider","provider","unregister","registerModelContextProvider"],"sources":["../../src/model-context/registry.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\nimport {\n type ModelContext,\n type ModelContextProvider,\n type AssistantToolProps,\n type AssistantInstructionsConfig,\n} from \"./types\";\nimport { notifySubscribers as notifyStateSubscribers } from \"../subscribable/subscribable\";\nimport { CompositeContextProvider } from \"../utils/composite-context-provider\";\nimport type { Unsubscribe } from \"../types/unsubscribe\";\nimport type {\n ModelContextRegistryToolHandle,\n ModelContextRegistryInstructionHandle,\n ModelContextRegistryProviderHandle,\n} from \"./registry-handles\";\n\nexport class ModelContextRegistry implements ModelContextProvider {\n private _tools = new Map<symbol, AssistantToolProps<any, any>>();\n private _instructions = new Map<symbol, string>();\n private _contextProviders = new CompositeContextProvider();\n private _subscribers = new Set<() => void>();\n\n constructor() {\n this._contextProviders.subscribe(() => {\n this.notifySubscribers();\n });\n }\n\n getModelContext(): ModelContext {\n const instructions = Array.from(this._instructions.values()).filter(\n Boolean,\n );\n\n const system =\n instructions.length > 0 ? instructions.join(\"\\n\\n\") : undefined;\n\n const tools: Record<string, Tool<any, any>> = {};\n for (const toolProps of this._tools.values()) {\n const { toolName, render, ...tool } = toolProps;\n tools[toolName] = tool;\n }\n\n const providerContexts = this._contextProviders.getModelContext();\n\n const context: ModelContext = {\n system,\n tools: Object.keys(tools).length > 0 ? tools : undefined,\n };\n\n if (providerContexts.system) {\n context.system = context.system\n ? `${context.system}\\n\\n${providerContexts.system}`\n : providerContexts.system;\n }\n\n if (providerContexts.tools) {\n context.tools = { ...(context.tools || {}), ...providerContexts.tools };\n }\n\n if (providerContexts.callSettings) {\n context.callSettings = providerContexts.callSettings;\n }\n\n if (providerContexts.config) {\n context.config = providerContexts.config;\n }\n\n if (providerContexts.unstable_composerMetadata) {\n context.unstable_composerMetadata =\n providerContexts.unstable_composerMetadata;\n }\n\n return context;\n }\n\n subscribe(callback: () => void): Unsubscribe {\n this._subscribers.add(callback);\n return () => this._subscribers.delete(callback);\n }\n\n private notifySubscribers(): void {\n notifyStateSubscribers(this._subscribers);\n }\n\n addTool<TArgs extends Record<string, unknown>, TResult>(\n tool: AssistantToolProps<TArgs, TResult>,\n ): ModelContextRegistryToolHandle<TArgs, TResult> {\n const id = Symbol();\n\n this._tools.set(id, tool);\n this.notifySubscribers();\n\n return {\n update: (newTool: AssistantToolProps<TArgs, TResult>) => {\n if (this._tools.has(id)) {\n this._tools.set(id, newTool);\n this.notifySubscribers();\n }\n },\n remove: () => {\n this._tools.delete(id);\n this.notifySubscribers();\n },\n };\n }\n\n addInstruction(\n config: string | AssistantInstructionsConfig,\n ): ModelContextRegistryInstructionHandle {\n const id = Symbol();\n\n const instruction =\n typeof config === \"string\" ? config : config.instruction;\n const disabled = typeof config === \"object\" ? config.disabled : false;\n\n if (!disabled) {\n this._instructions.set(id, instruction);\n this.notifySubscribers();\n }\n\n return {\n update: (newConfig: string | AssistantInstructionsConfig) => {\n const newInstruction =\n typeof newConfig === \"string\" ? newConfig : newConfig.instruction;\n const newDisabled =\n typeof newConfig === \"object\" ? newConfig.disabled : false;\n\n if (newDisabled) {\n this._instructions.delete(id);\n } else {\n this._instructions.set(id, newInstruction);\n }\n this.notifySubscribers();\n },\n remove: () => {\n this._instructions.delete(id);\n this.notifySubscribers();\n },\n };\n }\n\n addProvider(\n provider: ModelContextProvider,\n ): ModelContextRegistryProviderHandle {\n const unregister =\n this._contextProviders.registerModelContextProvider(provider);\n\n return {\n remove: unregister,\n };\n }\n}\n"],"mappings":";;;AAgBA,IAAaY,uBAAb,MAAkE;CAChE,yBAAiB,IAAIE,IAA0C;CAC/D,gCAAwB,IAAIA,IAAoB;CAChD,oBAA4B,IAAIP,yBAAyB;CACzD,+BAAuB,IAAIW,IAAgB;CAE3CC,cAAc;EACZ,KAAKH,kBAAkBI,gBAAgB;GACrC,KAAKf,kBAAkB;EACzB,CAAC;CACH;CAEAgB,kBAAgC;EAC9B,MAAMC,eAAeC,MAAMC,KAAK,KAAKT,cAAcU,OAAO,CAAC,CAAC,CAACC,OAC3DC,OACF;EAEA,MAAMC,SACJN,aAAaO,SAAS,IAAIP,aAAaQ,KAAK,MAAM,IAAIC,KAAAA;EAExD,MAAMC,QAAwC,CAAC;EAC/C,KAAK,MAAME,aAAa,KAAKrB,OAAOY,OAAO,GAAG;GAC5C,MAAM,EAAEU,UAAUC,QAAQ,GAAGC,SAASH;GACtCF,MAAMG,YAAYE;EACpB;EAEA,MAAMC,mBAAmB,KAAKtB,kBAAkBK,gBAAgB;EAEhE,MAAMkB,UAAwB;GAC5BX;GACAI,OAAOQ,OAAOC,KAAKT,KAAK,CAAC,CAACH,SAAS,IAAIG,QAAQD,KAAAA;EACjD;EAEA,IAAIO,iBAAiBV,QACnBW,QAAQX,SAASW,QAAQX,SACrB,GAAGW,QAAQX,OAAM,MAAOU,iBAAiBV,WACzCU,iBAAiBV;EAGvB,IAAIU,iBAAiBN,OACnBO,QAAQP,QAAQ;GAAE,GAAIO,QAAQP,SAAS,CAAC;GAAI,GAAGM,iBAAiBN;EAAM;EAGxE,IAAIM,iBAAiBI,cACnBH,QAAQG,eAAeJ,iBAAiBI;EAG1C,IAAIJ,iBAAiBK,QACnBJ,QAAQI,SAASL,iBAAiBK;EAGpC,IAAIL,iBAAiBM,2BACnBL,QAAQK,4BACNN,iBAAiBM;EAGrB,OAAOL;CACT;CAEAnB,UAAUyB,UAAmC;EAC3C,KAAK5B,aAAa6B,IAAID,QAAQ;EAC9B,aAAa,KAAK5B,aAAa8B,OAAOF,QAAQ;CAChD;CAEA,oBAAkC;EAChCvC,kBAAuB,KAAKW,YAAY;CAC1C;CAEA+B,QACEX,MACgD;EAChD,MAAMc,KAAKC,OAAO;EAElB,KAAKvC,OAAOwC,IAAIF,IAAId,IAAI;EACxB,KAAKhC,kBAAkB;EAEvB,OAAO;GACLiD,SAASC,YAAgD;IACvD,IAAI,KAAK1C,OAAO2C,IAAIL,EAAE,GAAG;KACvB,KAAKtC,OAAOwC,IAAIF,IAAII,OAAO;KAC3B,KAAKlD,kBAAkB;IACzB;GACF;GACAoD,cAAc;IACZ,KAAK5C,OAAOkC,OAAOI,EAAE;IACrB,KAAK9C,kBAAkB;GACzB;EACF;CACF;CAEAqD,eACEf,QACuC;EACvC,MAAMQ,KAAKC,OAAO;EAElB,MAAMO,cACJ,OAAOhB,WAAW,WAAWA,SAASA,OAAOgB;EAG/C,IAAI,EAFa,OAAOhB,WAAW,WAAWA,OAAOiB,WAAW,QAEjD;GACb,KAAK7C,cAAcsC,IAAIF,IAAIQ,WAAW;GACtC,KAAKtD,kBAAkB;EACzB;EAEA,OAAO;GACLiD,SAASO,cAAoD;IAC3D,MAAMC,iBACJ,OAAOD,cAAc,WAAWA,YAAYA,UAAUF;IAIxD,IAFE,OAAOE,cAAc,WAAWA,UAAUD,WAAW,OAGrD,KAAK7C,cAAcgC,OAAOI,EAAE;SAE5B,KAAKpC,cAAcsC,IAAIF,IAAIW,cAAc;IAE3C,KAAKzD,kBAAkB;GACzB;GACAoD,cAAc;IACZ,KAAK1C,cAAcgC,OAAOI,EAAE;IAC5B,KAAK9C,kBAAkB;GACzB;EACF;CACF;CAEA2D,YACEC,UACoC;EAIpC,OAAO,EACLR,QAHA,KAAKzC,kBAAkBmD,6BAA6BF,QAG5CC,EACV;CACF;AACF"}
1
+ {"version":3,"file":"registry.js","names":["Tool","ModelContext","ModelContextProvider","AssistantToolProps","AssistantInstructionsConfig","notifySubscribers","notifyStateSubscribers","CompositeContextProvider","Unsubscribe","ModelContextRegistryToolHandle","ModelContextRegistryInstructionHandle","ModelContextRegistryProviderHandle","nullProtoRecord","ModelContextRegistry","_tools","Map","_instructions","_contextProviders","_subscribers","Set","constructor","subscribe","getModelContext","instructions","Array","from","values","filter","Boolean","system","length","join","undefined","tools","toolProps","toolName","render","tool","providerContexts","context","Object","keys","callSettings","config","unstable_composerMetadata","callback","add","delete","addTool","Record","TArgs","TResult","id","Symbol","set","update","newTool","has","remove","addInstruction","instruction","disabled","newConfig","newInstruction","newDisabled","addProvider","provider","unregister","registerModelContextProvider"],"sources":["../../src/model-context/registry.ts"],"sourcesContent":["import type { Tool } from \"assistant-stream\";\nimport {\n type ModelContext,\n type ModelContextProvider,\n type AssistantToolProps,\n type AssistantInstructionsConfig,\n} from \"./types\";\nimport { notifySubscribers as notifyStateSubscribers } from \"../subscribable/subscribable\";\nimport { CompositeContextProvider } from \"../utils/composite-context-provider\";\nimport type { Unsubscribe } from \"../types/unsubscribe\";\nimport type {\n ModelContextRegistryToolHandle,\n ModelContextRegistryInstructionHandle,\n ModelContextRegistryProviderHandle,\n} from \"./registry-handles\";\nimport { nullProtoRecord } from \"../utils/record\";\n\nexport class ModelContextRegistry implements ModelContextProvider {\n private _tools = new Map<symbol, AssistantToolProps<any, any>>();\n private _instructions = new Map<symbol, string>();\n private _contextProviders = new CompositeContextProvider();\n private _subscribers = new Set<() => void>();\n\n constructor() {\n this._contextProviders.subscribe(() => {\n this.notifySubscribers();\n });\n }\n\n getModelContext(): ModelContext {\n const instructions = Array.from(this._instructions.values()).filter(\n Boolean,\n );\n\n const system =\n instructions.length > 0 ? instructions.join(\"\\n\\n\") : undefined;\n\n const tools = nullProtoRecord<Tool<any, any>>();\n for (const toolProps of this._tools.values()) {\n const { toolName, render, ...tool } = toolProps;\n tools[toolName] = tool;\n }\n\n const providerContexts = this._contextProviders.getModelContext();\n\n const context: ModelContext = {\n system,\n tools: Object.keys(tools).length > 0 ? tools : undefined,\n };\n\n if (providerContexts.system) {\n context.system = context.system\n ? `${context.system}\\n\\n${providerContexts.system}`\n : providerContexts.system;\n }\n\n if (providerContexts.tools) {\n context.tools = nullProtoRecord(context.tools, providerContexts.tools);\n }\n\n if (providerContexts.callSettings) {\n context.callSettings = providerContexts.callSettings;\n }\n\n if (providerContexts.config) {\n context.config = providerContexts.config;\n }\n\n if (providerContexts.unstable_composerMetadata) {\n context.unstable_composerMetadata =\n providerContexts.unstable_composerMetadata;\n }\n\n return context;\n }\n\n subscribe(callback: () => void): Unsubscribe {\n this._subscribers.add(callback);\n return () => this._subscribers.delete(callback);\n }\n\n private notifySubscribers(): void {\n notifyStateSubscribers(this._subscribers);\n }\n\n addTool<TArgs extends Record<string, unknown>, TResult>(\n tool: AssistantToolProps<TArgs, TResult>,\n ): ModelContextRegistryToolHandle<TArgs, TResult> {\n const id = Symbol();\n\n this._tools.set(id, tool);\n this.notifySubscribers();\n\n return {\n update: (newTool: AssistantToolProps<TArgs, TResult>) => {\n if (this._tools.has(id)) {\n this._tools.set(id, newTool);\n this.notifySubscribers();\n }\n },\n remove: () => {\n this._tools.delete(id);\n this.notifySubscribers();\n },\n };\n }\n\n addInstruction(\n config: string | AssistantInstructionsConfig,\n ): ModelContextRegistryInstructionHandle {\n const id = Symbol();\n\n const instruction =\n typeof config === \"string\" ? config : config.instruction;\n const disabled = typeof config === \"object\" ? config.disabled : false;\n\n if (!disabled) {\n this._instructions.set(id, instruction);\n this.notifySubscribers();\n }\n\n return {\n update: (newConfig: string | AssistantInstructionsConfig) => {\n const newInstruction =\n typeof newConfig === \"string\" ? newConfig : newConfig.instruction;\n const newDisabled =\n typeof newConfig === \"object\" ? newConfig.disabled : false;\n\n if (newDisabled) {\n this._instructions.delete(id);\n } else {\n this._instructions.set(id, newInstruction);\n }\n this.notifySubscribers();\n },\n remove: () => {\n this._instructions.delete(id);\n this.notifySubscribers();\n },\n };\n }\n\n addProvider(\n provider: ModelContextProvider,\n ): ModelContextRegistryProviderHandle {\n const unregister =\n this._contextProviders.registerModelContextProvider(provider);\n\n return {\n remove: unregister,\n };\n }\n}\n"],"mappings":";;;;AAiBA,IAAaa,uBAAb,MAAkE;CAChE,yBAAiB,IAAIE,IAA0C;CAC/D,gCAAwB,IAAIA,IAAoB;CAChD,oBAA4B,IAAIR,yBAAyB;CACzD,+BAAuB,IAAIY,IAAgB;CAE3CC,cAAc;EACZ,KAAKH,kBAAkBI,gBAAgB;GACrC,KAAKhB,kBAAkB;EACzB,CAAC;CACH;CAEAiB,kBAAgC;EAC9B,MAAMC,eAAeC,MAAMC,KAAK,KAAKT,cAAcU,OAAO,CAAC,CAAC,CAACC,OAC3DC,OACF;EAEA,MAAMC,SACJN,aAAaO,SAAS,IAAIP,aAAaQ,KAAK,MAAM,IAAIC,KAAAA;EAExD,MAAMC,QAAQrB,gBAAgC;EAC9C,KAAK,MAAMsB,aAAa,KAAKpB,OAAOY,OAAO,GAAG;GAC5C,MAAM,EAAES,UAAUC,QAAQ,GAAGC,SAASH;GACtCD,MAAME,YAAYE;EACpB;EAEA,MAAMC,mBAAmB,KAAKrB,kBAAkBK,gBAAgB;EAEhE,MAAMiB,UAAwB;GAC5BV;GACAI,OAAOO,OAAOC,KAAKR,KAAK,CAAC,CAACH,SAAS,IAAIG,QAAQD,KAAAA;EACjD;EAEA,IAAIM,iBAAiBT,QACnBU,QAAQV,SAASU,QAAQV,SACrB,GAAGU,QAAQV,OAAM,MAAOS,iBAAiBT,WACzCS,iBAAiBT;EAGvB,IAAIS,iBAAiBL,OACnBM,QAAQN,QAAQrB,gBAAgB2B,QAAQN,OAAOK,iBAAiBL,KAAK;EAGvE,IAAIK,iBAAiBI,cACnBH,QAAQG,eAAeJ,iBAAiBI;EAG1C,IAAIJ,iBAAiBK,QACnBJ,QAAQI,SAASL,iBAAiBK;EAGpC,IAAIL,iBAAiBM,2BACnBL,QAAQK,4BACNN,iBAAiBM;EAGrB,OAAOL;CACT;CAEAlB,UAAUwB,UAAmC;EAC3C,KAAK3B,aAAa4B,IAAID,QAAQ;EAC9B,aAAa,KAAK3B,aAAa6B,OAAOF,QAAQ;CAChD;CAEA,oBAAkC;EAChCvC,kBAAuB,KAAKY,YAAY;CAC1C;CAEA8B,QACEX,MACgD;EAChD,MAAMe,KAAKC,OAAO;EAElB,KAAKvC,OAAOwC,IAAIF,IAAIf,IAAI;EACxB,KAAKhC,kBAAkB;EAEvB,OAAO;GACLkD,SAASC,YAAgD;IACvD,IAAI,KAAK1C,OAAO2C,IAAIL,EAAE,GAAG;KACvB,KAAKtC,OAAOwC,IAAIF,IAAII,OAAO;KAC3B,KAAKnD,kBAAkB;IACzB;GACF;GACAqD,cAAc;IACZ,KAAK5C,OAAOiC,OAAOK,EAAE;IACrB,KAAK/C,kBAAkB;GACzB;EACF;CACF;CAEAsD,eACEhB,QACuC;EACvC,MAAMS,KAAKC,OAAO;EAElB,MAAMO,cACJ,OAAOjB,WAAW,WAAWA,SAASA,OAAOiB;EAG/C,IAAI,EAFa,OAAOjB,WAAW,WAAWA,OAAOkB,WAAW,QAEjD;GACb,KAAK7C,cAAcsC,IAAIF,IAAIQ,WAAW;GACtC,KAAKvD,kBAAkB;EACzB;EAEA,OAAO;GACLkD,SAASO,cAAoD;IAC3D,MAAMC,iBACJ,OAAOD,cAAc,WAAWA,YAAYA,UAAUF;IAIxD,IAFE,OAAOE,cAAc,WAAWA,UAAUD,WAAW,OAGrD,KAAK7C,cAAc+B,OAAOK,EAAE;SAE5B,KAAKpC,cAAcsC,IAAIF,IAAIW,cAAc;IAE3C,KAAK1D,kBAAkB;GACzB;GACAqD,cAAc;IACZ,KAAK1C,cAAc+B,OAAOK,EAAE;IAC5B,KAAK/C,kBAAkB;GACzB;EACF;CACF;CAEA4D,YACEC,UACoC;EAIpC,OAAO,EACLR,QAHA,KAAKzC,kBAAkBmD,6BAA6BF,QAG5CC,EACV;CACF;AACF"}
@@ -0,0 +1,26 @@
1
+ //#region src/model-context/tool-call-text.d.ts
2
+ type ToolCallRunningText<TArgs extends Record<string, unknown>, TValue> = TValue | undefined | null | ((options: {
3
+ args: TArgs;
4
+ }) => TValue | undefined | null);
5
+ type ToolCallCompleteText<TArgs extends Record<string, unknown>, TResult, TValue> = TValue | undefined | null | ((options: {
6
+ args: TArgs;
7
+ result: TResult | undefined;
8
+ }) => TValue | undefined | null);
9
+ type ToolCallText<TArgs extends Record<string, unknown>, TResult, TValue = string> = {
10
+ running: ToolCallRunningText<TArgs, TValue>;
11
+ complete?: ToolCallCompleteText<TArgs, TResult, TValue> | undefined;
12
+ } | {
13
+ running?: ToolCallRunningText<TArgs, TValue> | undefined;
14
+ complete: ToolCallCompleteText<TArgs, TResult, TValue>;
15
+ };
16
+ type ToolCallTextPart<TArgs extends Record<string, unknown>, TResult> = {
17
+ readonly args: TArgs;
18
+ readonly result?: TResult | undefined;
19
+ readonly status?: {
20
+ readonly type?: string | undefined;
21
+ } | undefined;
22
+ };
23
+ declare const resolveToolCallText: <TArgs extends Record<string, unknown>, TResult, TValue>(text: ToolCallText<TArgs, TResult, TValue>, part: ToolCallTextPart<TArgs, TResult>) => TValue | undefined | null;
24
+ //#endregion
25
+ export { ToolCallText, resolveToolCallText };
26
+ //# sourceMappingURL=tool-call-text.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tool-call-text.d.ts","names":[],"sources":["../../src/model-context/tool-call-text.ts"],"mappings":";KAAK,oBAAoB,cAAc,yBAAyB,UAC5D,8BAGE;EAAW,MAAM;MAAY;KAE9B,qBACH,cAAc,yBACd,SACA,UAEE,8BAGE;EACA,MAAM;EACN,QAAQ;MACJ;KAEE,aACV,cAAc,yBACd,SACA;EAGI,SAAS,oBAAoB,OAAO;EACpC,WAAW,qBAAqB,OAAO,SAAS;;EAGhD,UAAU,oBAAoB,OAAO;EACrC,UAAU,qBAAqB,OAAO,SAAS;;KAGhD,iBAAiB,cAAc,yBAAyB;WAClD,MAAM;WACN,SAAS;WACT;aAAoB;;;cAGlB,sBACX,cAAc,yBACd,SACA,QAEA,MAAM,aAAa,OAAO,SAAS,SACnC,MAAM,iBAAiB,OAAO,aAC7B"}
@@ -0,0 +1,18 @@
1
+ //#region src/model-context/tool-call-text.ts
2
+ const resolveToolCallText = (text, part) => {
3
+ if (!(part.status?.type === "running" || part.status?.type === "requires-action")) {
4
+ const value = text.complete;
5
+ if (typeof value !== "function") return value ?? null;
6
+ return value({
7
+ args: part.args,
8
+ result: part.result
9
+ });
10
+ }
11
+ const value = text.running;
12
+ if (typeof value !== "function") return value ?? null;
13
+ return value({ args: part.args });
14
+ };
15
+ //#endregion
16
+ export { resolveToolCallText };
17
+
18
+ //# sourceMappingURL=tool-call-text.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tool-call-text.js","names":["ToolCallRunningText","Record","TValue","options","args","TArgs","ToolCallCompleteText","result","TResult","ToolCallText","running","complete","ToolCallTextPart","status","type","resolveToolCallText","text","part","isRunning","value"],"sources":["../../src/model-context/tool-call-text.ts"],"sourcesContent":["type ToolCallRunningText<TArgs extends Record<string, unknown>, TValue> =\n | TValue\n | undefined\n | null\n | ((options: { args: TArgs }) => TValue | undefined | null);\n\ntype ToolCallCompleteText<\n TArgs extends Record<string, unknown>,\n TResult,\n TValue,\n> =\n | TValue\n | undefined\n | null\n | ((options: {\n args: TArgs;\n result: TResult | undefined;\n }) => TValue | undefined | null);\n\nexport type ToolCallText<\n TArgs extends Record<string, unknown>,\n TResult,\n TValue = string,\n> =\n | {\n running: ToolCallRunningText<TArgs, TValue>;\n complete?: ToolCallCompleteText<TArgs, TResult, TValue> | undefined;\n }\n | {\n running?: ToolCallRunningText<TArgs, TValue> | undefined;\n complete: ToolCallCompleteText<TArgs, TResult, TValue>;\n };\n\ntype ToolCallTextPart<TArgs extends Record<string, unknown>, TResult> = {\n readonly args: TArgs;\n readonly result?: TResult | undefined;\n readonly status?: { readonly type?: string | undefined } | undefined;\n};\n\nexport const resolveToolCallText = <\n TArgs extends Record<string, unknown>,\n TResult,\n TValue,\n>(\n text: ToolCallText<TArgs, TResult, TValue>,\n part: ToolCallTextPart<TArgs, TResult>,\n): TValue | undefined | null => {\n const isRunning =\n part.status?.type === \"running\" || part.status?.type === \"requires-action\";\n\n if (!isRunning) {\n const value = text.complete;\n if (typeof value !== \"function\") return value ?? null;\n return (\n value as (options: {\n args: TArgs;\n result: TResult | undefined;\n }) => TValue | undefined | null\n )({ args: part.args, result: part.result });\n }\n\n const value = text.running;\n if (typeof value !== \"function\") return value ?? null;\n return (value as (options: { args: TArgs }) => TValue | undefined | null)({\n args: part.args,\n });\n};\n"],"mappings":";AAuCA,MAAae,uBAKXC,MACAC,SAC8B;CAI9B,IAAI,EAFFA,KAAKJ,QAAQC,SAAS,aAAaG,KAAKJ,QAAQC,SAAS,oBAE3C;EACd,MAAMK,QAAQH,KAAKL;EACnB,IAAI,OAAOQ,UAAU,YAAY,OAAOA,SAAS;EACjD,OACEA,MAIA;GAAEf,MAAMa,KAAKb;GAAMG,QAAQU,KAAKV;EAAO,CAAC;CAC5C;CAEA,MAAMY,QAAQH,KAAKN;CACnB,IAAI,OAAOS,UAAU,YAAY,OAAOA,SAAS;CACjD,OAAQA,MAAkE,EACxEf,MAAMa,KAAKb,KACb,CAAC;AACH"}
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","names":[],"sources":["../../src/model-context/types.ts"],"mappings":";;;KAGY;EACV;EACA;EACA;EACA;EACA;EACA;EACA,UAAU;;KAGA;EACV;EACA;EACA;EACA;;KAGU;EACV;EACA;EACA,QAAQ,eAAe;EACvB,eAAe;EACf,SAAS;;;;;;EAMT,4BAA4B;;KAGlB;EACV,uBAAuB;EACvB,aAAa,yBAAyB;;KAG5B,mBACV,cAAc,yBACd,WACE,KAAK,OAAO;EACd;EACA;;KAGU;EACV;EACA;;KAGU;EACV;EACA;;cASW,qBACX,WAAW,IAAI,0BACd"}
1
+ {"version":3,"file":"types.d.ts","names":[],"sources":["../../src/model-context/types.ts"],"mappings":";;;KAIY;EACV;EACA;EACA;EACA;EACA;EACA;EACA,UAAU;;KAGA;EACV;EACA;EACA;EACA;;KAGU;EACV;EACA;EACA,QAAQ,eAAe;EACvB,eAAe;EACf,SAAS;;;;;;EAMT,4BAA4B;;KAGlB;EACV,uBAAuB;EACvB,aAAa,yBAAyB;;KAG5B,mBACV,cAAc,yBACd,WACE,KAAK,OAAO;EACd;EACA;;KAGU;EACV;EACA;;KAGU;EACV;EACA;;cASW,qBACX,WAAW,IAAI,0BACd"}