@assistant-ui/react 0.11.35 → 0.11.36

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 (199) hide show
  1. package/dist/client/AssistantClient.d.ts +13 -57
  2. package/dist/client/AssistantClient.d.ts.map +1 -1
  3. package/dist/client/AssistantClient.js +49 -26
  4. package/dist/client/AssistantClient.js.map +1 -1
  5. package/dist/client/ModelContext.d.ts +8 -0
  6. package/dist/client/ModelContext.d.ts.map +1 -0
  7. package/dist/client/ModelContext.js +21 -0
  8. package/dist/client/ModelContext.js.map +1 -0
  9. package/dist/client/ModelContextClient.d.ts +7 -0
  10. package/dist/client/ModelContextClient.d.ts.map +1 -0
  11. package/dist/client/ModelContextClient.js +18 -0
  12. package/dist/client/ModelContextClient.js.map +1 -0
  13. package/dist/client/ToolUIClient.d.ts.map +1 -1
  14. package/dist/client/ToolUIClient.js +45 -3
  15. package/dist/client/ToolUIClient.js.map +1 -1
  16. package/dist/client/ToolUIContext.d.ts +4 -0
  17. package/dist/client/ToolUIContext.d.ts.map +1 -0
  18. package/dist/client/ToolUIContext.js +20 -0
  19. package/dist/client/ToolUIContext.js.map +1 -0
  20. package/dist/client/Tools.d.ts +10 -0
  21. package/dist/client/Tools.d.ts.map +1 -0
  22. package/dist/client/Tools.js +55 -0
  23. package/dist/client/Tools.js.map +1 -0
  24. package/dist/client/types/ModelContext.d.ts +12 -0
  25. package/dist/client/types/ModelContext.d.ts.map +1 -0
  26. package/dist/client/types/ModelContext.js +1 -0
  27. package/dist/client/types/ModelContext.js.map +1 -0
  28. package/dist/client/types/ThreadList.d.ts.map +1 -1
  29. package/dist/client/types/ToolUI.d.ts +12 -1
  30. package/dist/client/types/ToolUI.d.ts.map +1 -1
  31. package/dist/client/types/Tools.d.ts +9 -0
  32. package/dist/client/types/Tools.d.ts.map +1 -0
  33. package/dist/client/types/Tools.js +1 -0
  34. package/dist/client/types/Tools.js.map +1 -0
  35. package/dist/context/providers/AttachmentByIndexProvider.d.ts.map +1 -1
  36. package/dist/context/providers/AttachmentByIndexProvider.js +20 -24
  37. package/dist/context/providers/AttachmentByIndexProvider.js.map +1 -1
  38. package/dist/context/providers/MessageByIndexProvider.d.ts.map +1 -1
  39. package/dist/context/providers/MessageByIndexProvider.js +27 -29
  40. package/dist/context/providers/MessageByIndexProvider.js.map +1 -1
  41. package/dist/context/providers/MessageProvider.d.ts.map +1 -1
  42. package/dist/context/providers/MessageProvider.js +11 -13
  43. package/dist/context/providers/MessageProvider.js.map +1 -1
  44. package/dist/context/providers/PartByIndexProvider.d.ts.map +1 -1
  45. package/dist/context/providers/PartByIndexProvider.js +11 -13
  46. package/dist/context/providers/PartByIndexProvider.js.map +1 -1
  47. package/dist/context/providers/TextMessagePartProvider.d.ts.map +1 -1
  48. package/dist/context/providers/TextMessagePartProvider.js +11 -13
  49. package/dist/context/providers/TextMessagePartProvider.js.map +1 -1
  50. package/dist/context/providers/ThreadListItemProvider.d.ts.map +1 -1
  51. package/dist/context/providers/ThreadListItemProvider.js +41 -45
  52. package/dist/context/providers/ThreadListItemProvider.js.map +1 -1
  53. package/dist/context/react/AssistantApiContext.d.ts +33 -4
  54. package/dist/context/react/AssistantApiContext.d.ts.map +1 -1
  55. package/dist/context/react/AssistantApiContext.js +35 -17
  56. package/dist/context/react/AssistantApiContext.js.map +1 -1
  57. package/dist/context/react/index.d.ts +1 -1
  58. package/dist/context/react/index.d.ts.map +1 -1
  59. package/dist/context/react/index.js +5 -1
  60. package/dist/context/react/index.js.map +1 -1
  61. package/dist/legacy-runtime/AssistantRuntimeProvider.d.ts.map +1 -1
  62. package/dist/legacy-runtime/AssistantRuntimeProvider.js +7 -9
  63. package/dist/legacy-runtime/AssistantRuntimeProvider.js.map +1 -1
  64. package/dist/legacy-runtime/RuntimeAdapter.d.ts +7 -0
  65. package/dist/legacy-runtime/RuntimeAdapter.d.ts.map +1 -0
  66. package/dist/legacy-runtime/RuntimeAdapter.js +20 -0
  67. package/dist/legacy-runtime/RuntimeAdapter.js.map +1 -0
  68. package/dist/legacy-runtime/client/ThreadListRuntimeClient.d.ts +2 -0
  69. package/dist/legacy-runtime/client/ThreadListRuntimeClient.d.ts.map +1 -1
  70. package/dist/legacy-runtime/client/ThreadListRuntimeClient.js +6 -2
  71. package/dist/legacy-runtime/client/ThreadListRuntimeClient.js.map +1 -1
  72. package/dist/legacy-runtime/hooks/AssistantContext.d.ts +14 -1
  73. package/dist/legacy-runtime/hooks/AssistantContext.d.ts.map +1 -1
  74. package/dist/legacy-runtime/hooks/AssistantContext.js +1 -1
  75. package/dist/legacy-runtime/hooks/AssistantContext.js.map +1 -1
  76. package/dist/legacy-runtime/hooks/AttachmentContext.d.ts +6 -0
  77. package/dist/legacy-runtime/hooks/AttachmentContext.d.ts.map +1 -1
  78. package/dist/legacy-runtime/hooks/AttachmentContext.js.map +1 -1
  79. package/dist/legacy-runtime/hooks/ComposerContext.d.ts +41 -2
  80. package/dist/legacy-runtime/hooks/ComposerContext.d.ts.map +1 -1
  81. package/dist/legacy-runtime/hooks/ComposerContext.js.map +1 -1
  82. package/dist/legacy-runtime/hooks/MessageContext.d.ts +36 -2
  83. package/dist/legacy-runtime/hooks/MessageContext.d.ts.map +1 -1
  84. package/dist/legacy-runtime/hooks/MessageContext.js.map +1 -1
  85. package/dist/legacy-runtime/hooks/MessagePartContext.d.ts +6 -0
  86. package/dist/legacy-runtime/hooks/MessagePartContext.d.ts.map +1 -1
  87. package/dist/legacy-runtime/hooks/MessagePartContext.js.map +1 -1
  88. package/dist/legacy-runtime/hooks/ThreadContext.d.ts +26 -1
  89. package/dist/legacy-runtime/hooks/ThreadContext.d.ts.map +1 -1
  90. package/dist/legacy-runtime/hooks/ThreadContext.js.map +1 -1
  91. package/dist/legacy-runtime/hooks/ThreadListItemContext.d.ts +6 -0
  92. package/dist/legacy-runtime/hooks/ThreadListItemContext.d.ts.map +1 -1
  93. package/dist/legacy-runtime/hooks/ThreadListItemContext.js.map +1 -1
  94. package/dist/legacy-runtime/runtime-cores/core/AssistantRuntimeCore.d.ts +1 -0
  95. package/dist/legacy-runtime/runtime-cores/core/AssistantRuntimeCore.d.ts.map +1 -1
  96. package/dist/legacy-runtime/runtime-cores/core/BaseAssistantRuntimeCore.d.ts +1 -0
  97. package/dist/legacy-runtime/runtime-cores/core/BaseAssistantRuntimeCore.d.ts.map +1 -1
  98. package/dist/legacy-runtime/runtime-cores/core/BaseAssistantRuntimeCore.js +3 -0
  99. package/dist/legacy-runtime/runtime-cores/core/BaseAssistantRuntimeCore.js.map +1 -1
  100. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadListRuntimeCore.d.ts +4 -1
  101. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadListRuntimeCore.d.ts.map +1 -1
  102. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadListRuntimeCore.js +2 -2
  103. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadListRuntimeCore.js.map +1 -1
  104. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadRuntimeCore.d.ts.map +1 -1
  105. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadRuntimeCore.js +7 -1
  106. package/dist/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadRuntimeCore.js.map +1 -1
  107. package/dist/legacy-runtime/runtime-cores/external-store/auto-status.d.ts +2 -12
  108. package/dist/legacy-runtime/runtime-cores/external-store/auto-status.d.ts.map +1 -1
  109. package/dist/legacy-runtime/runtime-cores/external-store/auto-status.js +45 -15
  110. package/dist/legacy-runtime/runtime-cores/external-store/auto-status.js.map +1 -1
  111. package/dist/legacy-runtime/runtime-cores/external-store/external-message-converter.d.ts +2 -0
  112. package/dist/legacy-runtime/runtime-cores/external-store/external-message-converter.d.ts.map +1 -1
  113. package/dist/legacy-runtime/runtime-cores/external-store/external-message-converter.js +4 -2
  114. package/dist/legacy-runtime/runtime-cores/external-store/external-message-converter.js.map +1 -1
  115. package/dist/legacy-runtime/runtime-cores/local/LocalThreadListRuntimeCore.d.ts +4 -1
  116. package/dist/legacy-runtime/runtime-cores/local/LocalThreadListRuntimeCore.d.ts.map +1 -1
  117. package/dist/legacy-runtime/runtime-cores/local/LocalThreadListRuntimeCore.js +2 -2
  118. package/dist/legacy-runtime/runtime-cores/local/LocalThreadListRuntimeCore.js.map +1 -1
  119. package/dist/legacy-runtime/runtime-cores/utils/MessageRepository.js +1 -1
  120. package/dist/legacy-runtime/runtime-cores/utils/MessageRepository.js.map +1 -1
  121. package/dist/model-context/index.d.ts +2 -0
  122. package/dist/model-context/index.d.ts.map +1 -1
  123. package/dist/model-context/index.js +6 -0
  124. package/dist/model-context/index.js.map +1 -1
  125. package/dist/model-context/makeAssistantVisible.js +1 -1
  126. package/dist/model-context/makeAssistantVisible.js.map +1 -1
  127. package/dist/model-context/toolbox.d.ts +28 -0
  128. package/dist/model-context/toolbox.d.ts.map +1 -0
  129. package/dist/model-context/toolbox.js +15 -0
  130. package/dist/model-context/toolbox.js.map +1 -0
  131. package/dist/model-context/useAssistantInstructions.js +1 -1
  132. package/dist/model-context/useAssistantInstructions.js.map +1 -1
  133. package/dist/model-context/useAssistantTool.js +1 -1
  134. package/dist/model-context/useAssistantTool.js.map +1 -1
  135. package/dist/primitives/message/MessageParts.d.ts.map +1 -1
  136. package/dist/primitives/message/MessageParts.js +2 -2
  137. package/dist/primitives/message/MessageParts.js.map +1 -1
  138. package/dist/primitives/message/MessagePartsGrouped.d.ts.map +1 -1
  139. package/dist/primitives/message/MessagePartsGrouped.js +2 -2
  140. package/dist/primitives/message/MessagePartsGrouped.js.map +1 -1
  141. package/dist/tests/setup.js +7 -7
  142. package/dist/tests/setup.js.map +1 -1
  143. package/dist/utils/tap-store/derived-scopes.d.ts +81 -0
  144. package/dist/utils/tap-store/derived-scopes.d.ts.map +1 -0
  145. package/dist/utils/tap-store/derived-scopes.js +67 -0
  146. package/dist/utils/tap-store/derived-scopes.js.map +1 -0
  147. package/dist/utils/tap-store/index.d.ts +1 -0
  148. package/dist/utils/tap-store/index.d.ts.map +1 -1
  149. package/dist/utils/tap-store/index.js +4 -0
  150. package/dist/utils/tap-store/index.js.map +1 -1
  151. package/dist/utils/tap-store/tap-api.js +1 -1
  152. package/dist/utils/tap-store/tap-api.js.map +1 -1
  153. package/package.json +9 -9
  154. package/src/client/AssistantClient.ts +64 -38
  155. package/src/client/ModelContext.ts +28 -0
  156. package/src/client/ModelContextClient.ts +16 -0
  157. package/src/client/ToolUIClient.ts +50 -3
  158. package/src/client/ToolUIContext.ts +22 -0
  159. package/src/client/Tools.ts +68 -0
  160. package/src/client/types/ModelContext.ts +15 -0
  161. package/src/client/types/ThreadList.ts +4 -0
  162. package/src/client/types/ToolUI.ts +10 -1
  163. package/src/client/types/Tools.ts +11 -0
  164. package/src/context/providers/AttachmentByIndexProvider.tsx +21 -25
  165. package/src/context/providers/MessageByIndexProvider.tsx +31 -33
  166. package/src/context/providers/MessageProvider.tsx +12 -14
  167. package/src/context/providers/PartByIndexProvider.tsx +12 -14
  168. package/src/context/providers/TextMessagePartProvider.tsx +12 -14
  169. package/src/context/providers/ThreadListItemProvider.tsx +42 -46
  170. package/src/context/react/AssistantApiContext.tsx +69 -29
  171. package/src/context/react/index.ts +5 -1
  172. package/src/legacy-runtime/AssistantRuntimeProvider.tsx +7 -9
  173. package/src/legacy-runtime/RuntimeAdapter.ts +19 -0
  174. package/src/legacy-runtime/client/ThreadListRuntimeClient.ts +10 -1
  175. package/src/legacy-runtime/hooks/AssistantContext.ts +16 -2
  176. package/src/legacy-runtime/hooks/AttachmentContext.ts +6 -0
  177. package/src/legacy-runtime/hooks/ComposerContext.ts +41 -2
  178. package/src/legacy-runtime/hooks/MessageContext.ts +37 -2
  179. package/src/legacy-runtime/hooks/MessagePartContext.ts +6 -0
  180. package/src/legacy-runtime/hooks/ThreadContext.ts +27 -1
  181. package/src/legacy-runtime/hooks/ThreadListItemContext.ts +6 -0
  182. package/src/legacy-runtime/runtime-cores/core/AssistantRuntimeCore.tsx +1 -0
  183. package/src/legacy-runtime/runtime-cores/core/BaseAssistantRuntimeCore.tsx +4 -0
  184. package/src/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadListRuntimeCore.tsx +4 -2
  185. package/src/legacy-runtime/runtime-cores/external-store/ExternalStoreThreadRuntimeCore.tsx +7 -1
  186. package/src/legacy-runtime/runtime-cores/external-store/auto-status.tsx +49 -16
  187. package/src/legacy-runtime/runtime-cores/external-store/external-message-converter.tsx +4 -0
  188. package/src/legacy-runtime/runtime-cores/local/LocalThreadListRuntimeCore.tsx +4 -2
  189. package/src/legacy-runtime/runtime-cores/utils/MessageRepository.tsx +1 -1
  190. package/src/model-context/index.ts +9 -0
  191. package/src/model-context/makeAssistantVisible.tsx +1 -1
  192. package/src/model-context/toolbox.tsx +37 -0
  193. package/src/model-context/useAssistantInstructions.tsx +1 -1
  194. package/src/model-context/useAssistantTool.tsx +1 -1
  195. package/src/primitives/message/MessageParts.tsx +3 -2
  196. package/src/primitives/message/MessagePartsGrouped.tsx +3 -2
  197. package/src/utils/tap-store/derived-scopes.ts +185 -0
  198. package/src/utils/tap-store/index.ts +9 -0
  199. package/src/utils/tap-store/tap-api.ts +1 -1
@@ -1,65 +1,21 @@
1
- import { Unsubscribe, ResourceElement } from "@assistant-ui/tap";
1
+ import { ResourceElement } from "@assistant-ui/tap";
2
2
  import { ThreadListClientApi, ThreadListClientState } from "./types/ThreadList";
3
- import { AssistantRuntime } from "../legacy-runtime/runtime/AssistantRuntime";
4
- import { ModelContextProvider } from "../model-context";
5
- import { AssistantEvent, AssistantEventCallback } from "../types/EventTypes";
6
- import { ToolUIApi } from "./types/ToolUI";
3
+ import { AssistantApi } from "../context/react/AssistantApiContext";
4
+ import { ToolsApi, ToolsState } from "./types/Tools";
5
+ import { ModelContextApi, ModelContextState } from "./types/ModelContext";
7
6
  export type AssistantClientProps = {
8
7
  threads: ResourceElement<{
9
8
  state: ThreadListClientState;
10
9
  api: ThreadListClientApi;
11
10
  }>;
12
- registerModelContextProvider?: (provider: ModelContextProvider) => Unsubscribe;
13
- };
14
- export declare const useAssistantClient: (props: AssistantClientProps) => {
15
- on<TEvent extends AssistantEvent>(selector: import("..").AssistantEventSelector<TEvent>, callback: AssistantEventCallback<TEvent>): Unsubscribe;
16
- subscribe: (listener: () => void) => Unsubscribe;
17
- flushSync: () => void;
18
- __internal_getRuntime?: () => AssistantRuntime;
19
- threads: (() => ThreadListClientApi) & ({
20
- source: null;
21
- query: Record<string, never>;
22
- } | {
23
- source: "root";
24
- query: {};
25
- get: () => ThreadListClientApi;
26
- });
27
- toolUIs: (() => ToolUIApi) & ({
28
- source: null;
29
- query: Record<string, never>;
30
- } | {
31
- source: "root";
32
- query: {};
33
- get: () => ToolUIApi;
34
- });
35
- thread: (() => import("./types/Thread").ThreadClientApi) & ({
36
- source: null;
37
- query: Record<string, never>;
38
- } | {
39
- source: "threads";
40
- query: {
41
- type: "main";
42
- };
43
- get: () => import("./types/Thread").ThreadClientApi;
44
- });
45
- threadListItem: (() => import("./types/ThreadListItem").ThreadListItemClientApi) & ({
46
- source: null;
47
- query: Record<string, never>;
48
- } | {
49
- source: "threads";
50
- query: {
51
- type: "main";
52
- };
53
- get: () => import("./types/ThreadListItem").ThreadListItemClientApi;
54
- });
55
- composer: (() => import("./types/Composer").ComposerClientApi) & ({
56
- source: null;
57
- query: Record<string, never>;
58
- } | {
59
- source: "thread";
60
- query: {};
61
- get: () => import("./types/Composer").ComposerClientApi;
62
- });
63
- registerModelContextProvider(provider: ModelContextProvider): Unsubscribe;
11
+ modelContext?: ResourceElement<{
12
+ state: ModelContextState;
13
+ api: ModelContextApi;
14
+ }>;
15
+ tools?: ResourceElement<{
16
+ state: ToolsState;
17
+ api: ToolsApi;
18
+ }> | undefined;
64
19
  };
20
+ export declare const useAssistantClient: (props: AssistantClientProps) => AssistantApi;
65
21
  //# sourceMappingURL=AssistantClient.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"AssistantClient.d.ts","sourceRoot":"","sources":["../../src/client/AssistantClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,WAAW,EAEX,eAAe,EAEhB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,gBAAgB,EAAE,MAAM,4CAA4C,CAAC;AAC9E,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AAIxD,OAAO,EACL,cAAc,EACd,sBAAsB,EAGvB,MAAM,qBAAqB,CAAC;AAQ7B,OAAO,EAAE,SAAS,EAAe,MAAM,gBAAgB,CAAC;AA6HxD,MAAM,MAAM,oBAAoB,GAAG;IACjC,OAAO,EAAE,eAAe,CAAC;QACvB,KAAK,EAAE,qBAAqB,CAAC;QAC7B,GAAG,EAAE,mBAAmB,CAAC;KAC1B,CAAC,CAAC;IACH,4BAA4B,CAAC,EAAE,CAC7B,QAAQ,EAAE,oBAAoB,KAC3B,WAAW,CAAC;CAIlB,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2CA7CnB,oBAAoB;CAgD9D,CAAC"}
1
+ {"version":3,"file":"AssistantClient.d.ts","sourceRoot":"","sources":["../../src/client/AssistantClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAKL,eAAe,EAEhB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAYhF,OAAO,EACL,YAAY,EAIb,MAAM,sCAAsC,CAAC;AAM9C,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AA4I1E,MAAM,MAAM,oBAAoB,GAAG;IACjC,OAAO,EAAE,eAAe,CAAC;QACvB,KAAK,EAAE,qBAAqB,CAAC;QAC7B,GAAG,EAAE,mBAAmB,CAAC;KAC1B,CAAC,CAAC;IACH,YAAY,CAAC,EAAE,eAAe,CAAC;QAC7B,KAAK,EAAE,iBAAiB,CAAC;QACzB,GAAG,EAAE,eAAe,CAAC;KACtB,CAAC,CAAC;IACH,KAAK,CAAC,EACF,eAAe,CAAC;QACd,KAAK,EAAE,UAAU,CAAC;QAClB,GAAG,EAAE,QAAQ,CAAC;KACf,CAAC,GACF,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,OAAO,oBAAoB,iBAK7D,CAAC"}
@@ -5,6 +5,7 @@ import {
5
5
  tapInlineResource,
6
6
  tapResource
7
7
  } from "@assistant-ui/tap";
8
+ import { Tools } from "../model-context/index.js";
8
9
  import { asStore, tapApi } from "../utils/tap-store/index.js";
9
10
  import { useResource } from "@assistant-ui/tap/react";
10
11
  import { useMemo } from "react";
@@ -14,40 +15,58 @@ import {
14
15
  } from "../types/EventTypes.js";
15
16
  import { EventManager } from "../legacy-runtime/client/EventManagerRuntimeClient.js";
16
17
  import {
17
- createAssistantApiField
18
+ createAssistantApiField,
19
+ useAssistantApiImpl,
20
+ extendApi
18
21
  } from "../context/react/AssistantApiContext.js";
19
22
  import { ToolUIClient } from "./ToolUIClient.js";
20
23
  import { withEventsProvider } from "./EventContext.js";
24
+ import { withModelContextProvider } from "./ModelContext.js";
25
+ import { withToolUIProvider } from "./ToolUIContext.js";
26
+ import { ModelContext as ModelContextResource } from "./ModelContextClient.js";
21
27
  var AssistantStore = resource(
22
28
  ({
23
29
  threads: threadsEl,
24
- registerModelContextProvider,
25
- __internal_runtime
30
+ modelContext: modelContextEl,
31
+ tools: toolsEl
26
32
  }) => {
27
33
  const events = tapInlineResource(EventManager());
28
- const { threads, toolUIs } = withEventsProvider(events, () => {
29
- return {
30
- toolUIs: tapInlineResource(ToolUIClient()),
31
- threads: tapResource(threadsEl, [threadsEl])
32
- };
33
- });
34
+ const toolUIsResource = tapInlineResource(ToolUIClient());
35
+ const { threads, toolUIs, tools, modelContext } = withEventsProvider(
36
+ events,
37
+ () => {
38
+ const modelContextResource = tapResource(
39
+ modelContextEl ?? ModelContextResource(),
40
+ [modelContextEl]
41
+ );
42
+ return withModelContextProvider(modelContextResource.api, () => {
43
+ return withToolUIProvider(toolUIsResource.api, () => {
44
+ return {
45
+ toolUIs: toolUIsResource,
46
+ modelContext: modelContextResource,
47
+ tools: tapResource(toolsEl ?? Tools({}), [toolsEl]),
48
+ threads: tapResource(threadsEl, [threadsEl])
49
+ };
50
+ });
51
+ });
52
+ }
53
+ );
34
54
  const state = tapMemo(
35
55
  () => ({
36
56
  threads: threads.state,
37
- toolUIs: toolUIs.state
57
+ toolUIs: toolUIs.state,
58
+ tools: tools.state,
59
+ modelContext: modelContext.state
38
60
  }),
39
- [threads.state, toolUIs.state]
61
+ [threads.state, toolUIs.state, tools.state, modelContext.state]
40
62
  );
41
63
  return tapApi({
42
64
  getState: () => state,
43
65
  threads: threads.api,
44
66
  toolUIs: toolUIs.api,
45
- on: events.on,
46
- registerModelContextProvider: registerModelContextProvider ?? (() => () => {
47
- }),
48
- ...__internal_runtime && {
49
- __internal_getRuntime: () => __internal_runtime
50
- }
67
+ tools: tools.api,
68
+ modelContext: modelContext.api,
69
+ on: events.on
51
70
  });
52
71
  }
53
72
  );
@@ -66,6 +85,16 @@ var getClientFromStore = (client) => {
66
85
  query: {},
67
86
  get: () => client.getState().api.toolUIs
68
87
  }),
88
+ tools: createAssistantApiField({
89
+ source: "root",
90
+ query: {},
91
+ get: () => client.getState().api.tools
92
+ }),
93
+ modelContext: createAssistantApiField({
94
+ source: "root",
95
+ query: {},
96
+ get: () => client.getState().api.modelContext
97
+ }),
69
98
  thread: createAssistantApiField({
70
99
  source: "threads",
71
100
  query: { type: "main" },
@@ -81,14 +110,6 @@ var getClientFromStore = (client) => {
81
110
  query: {},
82
111
  get: () => client.getState().api.threads.thread("main").composer
83
112
  }),
84
- registerModelContextProvider(provider) {
85
- return client.getState().api.registerModelContextProvider(provider);
86
- },
87
- ...client.getState().api.__internal_getRuntime && {
88
- __internal_getRuntime() {
89
- return client.getState().api.__internal_getRuntime();
90
- }
91
- },
92
113
  on(selector, callback) {
93
114
  const { event, scope } = normalizeEventSelector(selector);
94
115
  if (scope === "*") return client.getState().api.on(event, callback);
@@ -107,8 +128,10 @@ var getClientFromStore = (client) => {
107
128
  };
108
129
  };
109
130
  var useAssistantClient = (props) => {
131
+ const api = useAssistantApiImpl();
110
132
  const client = useResource(asStore(AssistantStore(props)));
111
- return useMemo(() => getClientFromStore(client), [client]);
133
+ const clientApi = useMemo(() => getClientFromStore(client), [client]);
134
+ return useMemo(() => extendApi(api, clientApi), [api, clientApi]);
112
135
  };
113
136
  export {
114
137
  useAssistantClient
@@ -1 +1 @@
1
- {"version":3,"sources":["../../src/client/AssistantClient.ts"],"sourcesContent":["import {\n tapMemo,\n resource,\n Unsubscribe,\n tapInlineResource,\n ResourceElement,\n tapResource,\n} from \"@assistant-ui/tap\";\nimport { ThreadListClientApi, ThreadListClientState } from \"./types/ThreadList\";\nimport { AssistantRuntime } from \"../legacy-runtime/runtime/AssistantRuntime\";\nimport { ModelContextProvider } from \"../model-context\";\nimport { asStore, Store, tapApi } from \"../utils/tap-store\";\nimport { useResource } from \"@assistant-ui/tap/react\";\nimport { useMemo } from \"react\";\nimport {\n AssistantEvent,\n AssistantEventCallback,\n checkEventScope,\n normalizeEventSelector,\n} from \"../types/EventTypes\";\nimport { EventManager } from \"../legacy-runtime/client/EventManagerRuntimeClient\";\nimport {\n AssistantApi,\n createAssistantApiField,\n} from \"../context/react/AssistantApiContext\";\nimport { ToolUIClient } from \"./ToolUIClient\";\nimport { withEventsProvider } from \"./EventContext\";\nimport { ToolUIApi, ToolUIState } from \"./types/ToolUI\";\n\ntype AssistantClientState = {\n readonly threads: ThreadListClientState;\n readonly toolUIs: ToolUIState;\n};\n\ntype AssistantClientApi = {\n getState(): AssistantClientState;\n\n readonly threads: ThreadListClientApi;\n readonly toolUIs: ToolUIApi;\n\n on<TEvent extends AssistantEvent>(\n event: TEvent,\n callback: AssistantEventCallback<TEvent>,\n ): Unsubscribe;\n\n registerModelContextProvider(provider: ModelContextProvider): Unsubscribe;\n\n /** @internal */\n __internal_getRuntime?(): AssistantRuntime;\n};\n\nconst AssistantStore = resource(\n ({\n threads: threadsEl,\n registerModelContextProvider,\n __internal_runtime,\n }: AssistantClientProps) => {\n const events = tapInlineResource(EventManager());\n\n const { threads, toolUIs } = withEventsProvider(events, () => {\n return {\n toolUIs: tapInlineResource(ToolUIClient()),\n threads: tapResource(threadsEl, [threadsEl]),\n };\n });\n\n const state = tapMemo<AssistantClientState>(\n () => ({\n threads: threads.state,\n toolUIs: toolUIs.state,\n }),\n [threads.state, toolUIs.state],\n );\n\n return tapApi<AssistantClientApi>({\n getState: () => state,\n\n threads: threads.api,\n toolUIs: toolUIs.api,\n on: events.on,\n\n registerModelContextProvider:\n registerModelContextProvider ?? (() => () => {}),\n ...(__internal_runtime && {\n __internal_getRuntime: () => __internal_runtime,\n }),\n });\n },\n);\n\nconst getClientFromStore = (client: Store<{ api: AssistantClientApi }>) => {\n const getItem = () => {\n return client.getState().api.threads.item(\"main\");\n };\n return {\n threads: createAssistantApiField({\n source: \"root\",\n query: {},\n get: () => client.getState().api.threads,\n }),\n toolUIs: createAssistantApiField({\n source: \"root\",\n query: {},\n get: () => client.getState().api.toolUIs,\n }),\n thread: createAssistantApiField({\n source: \"threads\",\n query: { type: \"main\" },\n get: () => client.getState().api.threads.thread(\"main\"),\n }),\n threadListItem: createAssistantApiField({\n source: \"threads\",\n query: { type: \"main\" },\n get: () => getItem(),\n }),\n composer: createAssistantApiField({\n source: \"thread\",\n query: {},\n get: () => client.getState().api.threads.thread(\"main\").composer,\n }),\n registerModelContextProvider(provider: ModelContextProvider) {\n return client.getState().api.registerModelContextProvider(provider);\n },\n ...(client.getState().api.__internal_getRuntime && {\n __internal_getRuntime() {\n return client.getState().api.__internal_getRuntime!();\n },\n }),\n on(selector, callback) {\n const { event, scope } = normalizeEventSelector(selector);\n if (scope === \"*\") return client.getState().api.on(event, callback);\n\n if (\n checkEventScope(\"thread\", scope, event) ||\n checkEventScope(\"thread-list-item\", scope, event) ||\n checkEventScope(\"composer\", scope, event)\n ) {\n return client.getState().api.on(event, (e) => {\n if (e.threadId !== getItem().getState().id) return;\n callback(e);\n });\n }\n\n throw new Error(\n `Event scope is not available in this component: ${scope}`,\n );\n },\n subscribe: client.subscribe,\n flushSync: client.flushSync,\n } satisfies Partial<AssistantApi>;\n};\n\nexport type AssistantClientProps = {\n threads: ResourceElement<{\n state: ThreadListClientState;\n api: ThreadListClientApi;\n }>;\n registerModelContextProvider?: (\n provider: ModelContextProvider,\n ) => Unsubscribe;\n\n /** @internal */\n __internal_runtime?: AssistantRuntime;\n};\n\nexport const useAssistantClient = (props: AssistantClientProps) => {\n const client = useResource(asStore(AssistantStore(props)));\n return useMemo(() => getClientFromStore(client), [client]);\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,EAEA;AAAA,OACK;AAIP,SAAS,SAAgB,cAAc;AACvC,SAAS,mBAAmB;AAC5B,SAAS,eAAe;AACxB;AAAA,EAGE;AAAA,EACA;AAAA,OACK;AACP,SAAS,oBAAoB;AAC7B;AAAA,EAEE;AAAA,OACK;AACP,SAAS,oBAAoB;AAC7B,SAAS,0BAA0B;AAyBnC,IAAM,iBAAiB;AAAA,EACrB,CAAC;AAAA,IACC,SAAS;AAAA,IACT;AAAA,IACA;AAAA,EACF,MAA4B;AAC1B,UAAM,SAAS,kBAAkB,aAAa,CAAC;AAE/C,UAAM,EAAE,SAAS,QAAQ,IAAI,mBAAmB,QAAQ,MAAM;AAC5D,aAAO;AAAA,QACL,SAAS,kBAAkB,aAAa,CAAC;AAAA,QACzC,SAAS,YAAY,WAAW,CAAC,SAAS,CAAC;AAAA,MAC7C;AAAA,IACF,CAAC;AAED,UAAM,QAAQ;AAAA,MACZ,OAAO;AAAA,QACL,SAAS,QAAQ;AAAA,QACjB,SAAS,QAAQ;AAAA,MACnB;AAAA,MACA,CAAC,QAAQ,OAAO,QAAQ,KAAK;AAAA,IAC/B;AAEA,WAAO,OAA2B;AAAA,MAChC,UAAU,MAAM;AAAA,MAEhB,SAAS,QAAQ;AAAA,MACjB,SAAS,QAAQ;AAAA,MACjB,IAAI,OAAO;AAAA,MAEX,8BACE,iCAAiC,MAAM,MAAM;AAAA,MAAC;AAAA,MAChD,GAAI,sBAAsB;AAAA,QACxB,uBAAuB,MAAM;AAAA,MAC/B;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEA,IAAM,qBAAqB,CAAC,WAA+C;AACzE,QAAM,UAAU,MAAM;AACpB,WAAO,OAAO,SAAS,EAAE,IAAI,QAAQ,KAAK,MAAM;AAAA,EAClD;AACA,SAAO;AAAA,IACL,SAAS,wBAAwB;AAAA,MAC/B,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI;AAAA,IACnC,CAAC;AAAA,IACD,SAAS,wBAAwB;AAAA,MAC/B,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI;AAAA,IACnC,CAAC;AAAA,IACD,QAAQ,wBAAwB;AAAA,MAC9B,QAAQ;AAAA,MACR,OAAO,EAAE,MAAM,OAAO;AAAA,MACtB,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI,QAAQ,OAAO,MAAM;AAAA,IACxD,CAAC;AAAA,IACD,gBAAgB,wBAAwB;AAAA,MACtC,QAAQ;AAAA,MACR,OAAO,EAAE,MAAM,OAAO;AAAA,MACtB,KAAK,MAAM,QAAQ;AAAA,IACrB,CAAC;AAAA,IACD,UAAU,wBAAwB;AAAA,MAChC,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI,QAAQ,OAAO,MAAM,EAAE;AAAA,IAC1D,CAAC;AAAA,IACD,6BAA6B,UAAgC;AAC3D,aAAO,OAAO,SAAS,EAAE,IAAI,6BAA6B,QAAQ;AAAA,IACpE;AAAA,IACA,GAAI,OAAO,SAAS,EAAE,IAAI,yBAAyB;AAAA,MACjD,wBAAwB;AACtB,eAAO,OAAO,SAAS,EAAE,IAAI,sBAAuB;AAAA,MACtD;AAAA,IACF;AAAA,IACA,GAAG,UAAU,UAAU;AACrB,YAAM,EAAE,OAAO,MAAM,IAAI,uBAAuB,QAAQ;AACxD,UAAI,UAAU,IAAK,QAAO,OAAO,SAAS,EAAE,IAAI,GAAG,OAAO,QAAQ;AAElE,UACE,gBAAgB,UAAU,OAAO,KAAK,KACtC,gBAAgB,oBAAoB,OAAO,KAAK,KAChD,gBAAgB,YAAY,OAAO,KAAK,GACxC;AACA,eAAO,OAAO,SAAS,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM;AAC5C,cAAI,EAAE,aAAa,QAAQ,EAAE,SAAS,EAAE,GAAI;AAC5C,mBAAS,CAAC;AAAA,QACZ,CAAC;AAAA,MACH;AAEA,YAAM,IAAI;AAAA,QACR,mDAAmD,KAAK;AAAA,MAC1D;AAAA,IACF;AAAA,IACA,WAAW,OAAO;AAAA,IAClB,WAAW,OAAO;AAAA,EACpB;AACF;AAeO,IAAM,qBAAqB,CAAC,UAAgC;AACjE,QAAM,SAAS,YAAY,QAAQ,eAAe,KAAK,CAAC,CAAC;AACzD,SAAO,QAAQ,MAAM,mBAAmB,MAAM,GAAG,CAAC,MAAM,CAAC;AAC3D;","names":[]}
1
+ {"version":3,"sources":["../../src/client/AssistantClient.ts"],"sourcesContent":["import {\n tapMemo,\n resource,\n Unsubscribe,\n tapInlineResource,\n ResourceElement,\n tapResource,\n} from \"@assistant-ui/tap\";\nimport { ThreadListClientApi, ThreadListClientState } from \"./types/ThreadList\";\nimport { Tools } from \"../model-context\";\nimport { asStore, Store, tapApi } from \"../utils/tap-store\";\nimport { useResource } from \"@assistant-ui/tap/react\";\nimport { useMemo } from \"react\";\nimport {\n AssistantEvent,\n AssistantEventCallback,\n checkEventScope,\n normalizeEventSelector,\n} from \"../types/EventTypes\";\nimport { EventManager } from \"../legacy-runtime/client/EventManagerRuntimeClient\";\nimport {\n AssistantApi,\n createAssistantApiField,\n useAssistantApiImpl,\n extendApi,\n} from \"../context/react/AssistantApiContext\";\nimport { ToolUIClient } from \"./ToolUIClient\";\nimport { withEventsProvider } from \"./EventContext\";\nimport { withModelContextProvider } from \"./ModelContext\";\nimport { withToolUIProvider } from \"./ToolUIContext\";\nimport { ToolUIApi, ToolUIState } from \"./types/ToolUI\";\nimport { ToolsApi, ToolsState } from \"./types/Tools\";\nimport { ModelContextApi, ModelContextState } from \"./types/ModelContext\";\nimport { ModelContext as ModelContextResource } from \"./ModelContextClient\";\n\ntype AssistantClientState = {\n readonly threads: ThreadListClientState;\n readonly toolUIs: ToolUIState;\n readonly tools: ToolsState;\n readonly modelContext: ModelContextState;\n};\n\ntype AssistantClientApi = {\n getState(): AssistantClientState;\n\n readonly threads: ThreadListClientApi;\n readonly toolUIs: ToolUIApi;\n readonly tools: ToolsApi;\n readonly modelContext: ModelContextApi;\n\n on<TEvent extends AssistantEvent>(\n event: TEvent,\n callback: AssistantEventCallback<TEvent>,\n ): Unsubscribe;\n};\n\nconst AssistantStore = resource(\n ({\n threads: threadsEl,\n modelContext: modelContextEl,\n tools: toolsEl,\n }: AssistantClientProps) => {\n const events = tapInlineResource(EventManager());\n const toolUIsResource = tapInlineResource(ToolUIClient());\n\n const { threads, toolUIs, tools, modelContext } = withEventsProvider(\n events,\n () => {\n const modelContextResource = tapResource(\n modelContextEl ?? ModelContextResource(),\n [modelContextEl],\n );\n\n return withModelContextProvider(modelContextResource.api, () => {\n return withToolUIProvider(toolUIsResource.api, () => {\n return {\n toolUIs: toolUIsResource,\n modelContext: modelContextResource,\n tools: tapResource(toolsEl ?? Tools({}), [toolsEl]),\n threads: tapResource(threadsEl, [threadsEl]),\n };\n });\n });\n },\n );\n\n const state = tapMemo<AssistantClientState>(\n () => ({\n threads: threads.state,\n toolUIs: toolUIs.state,\n tools: tools.state,\n modelContext: modelContext.state,\n }),\n [threads.state, toolUIs.state, tools.state, modelContext.state],\n );\n\n return tapApi<AssistantClientApi>({\n getState: () => state,\n\n threads: threads.api,\n toolUIs: toolUIs.api,\n tools: tools.api,\n modelContext: modelContext.api,\n on: events.on,\n });\n },\n);\n\nconst getClientFromStore = (client: Store<{ api: AssistantClientApi }>) => {\n const getItem = () => {\n return client.getState().api.threads.item(\"main\");\n };\n return {\n threads: createAssistantApiField({\n source: \"root\",\n query: {},\n get: () => client.getState().api.threads,\n }),\n toolUIs: createAssistantApiField({\n source: \"root\",\n query: {},\n get: () => client.getState().api.toolUIs,\n }),\n tools: createAssistantApiField({\n source: \"root\",\n query: {},\n get: () => client.getState().api.tools,\n }),\n modelContext: createAssistantApiField({\n source: \"root\",\n query: {},\n get: () => client.getState().api.modelContext,\n }),\n thread: createAssistantApiField({\n source: \"threads\",\n query: { type: \"main\" },\n get: () => client.getState().api.threads.thread(\"main\"),\n }),\n threadListItem: createAssistantApiField({\n source: \"threads\",\n query: { type: \"main\" },\n get: () => getItem(),\n }),\n composer: createAssistantApiField({\n source: \"thread\",\n query: {},\n get: () => client.getState().api.threads.thread(\"main\").composer,\n }),\n on(selector, callback) {\n const { event, scope } = normalizeEventSelector(selector);\n if (scope === \"*\") return client.getState().api.on(event, callback);\n\n if (\n checkEventScope(\"thread\", scope, event) ||\n checkEventScope(\"thread-list-item\", scope, event) ||\n checkEventScope(\"composer\", scope, event)\n ) {\n return client.getState().api.on(event, (e) => {\n if (e.threadId !== getItem().getState().id) return;\n callback(e);\n });\n }\n\n throw new Error(\n `Event scope is not available in this component: ${scope}`,\n );\n },\n subscribe: client.subscribe,\n flushSync: client.flushSync,\n } satisfies Partial<AssistantApi>;\n};\n\nexport type AssistantClientProps = {\n threads: ResourceElement<{\n state: ThreadListClientState;\n api: ThreadListClientApi;\n }>;\n modelContext?: ResourceElement<{\n state: ModelContextState;\n api: ModelContextApi;\n }>;\n tools?:\n | ResourceElement<{\n state: ToolsState;\n api: ToolsApi;\n }>\n | undefined;\n};\n\nexport const useAssistantClient = (props: AssistantClientProps) => {\n const api = useAssistantApiImpl();\n const client = useResource(asStore(AssistantStore(props)));\n const clientApi = useMemo(() => getClientFromStore(client), [client]);\n return useMemo(() => extendApi(api, clientApi), [api, clientApi]);\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,EAEA;AAAA,OACK;AAEP,SAAS,aAAa;AACtB,SAAS,SAAgB,cAAc;AACvC,SAAS,mBAAmB;AAC5B,SAAS,eAAe;AACxB;AAAA,EAGE;AAAA,EACA;AAAA,OACK;AACP,SAAS,oBAAoB;AAC7B;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,oBAAoB;AAC7B,SAAS,0BAA0B;AACnC,SAAS,gCAAgC;AACzC,SAAS,0BAA0B;AAInC,SAAS,gBAAgB,4BAA4B;AAuBrD,IAAM,iBAAiB;AAAA,EACrB,CAAC;AAAA,IACC,SAAS;AAAA,IACT,cAAc;AAAA,IACd,OAAO;AAAA,EACT,MAA4B;AAC1B,UAAM,SAAS,kBAAkB,aAAa,CAAC;AAC/C,UAAM,kBAAkB,kBAAkB,aAAa,CAAC;AAExD,UAAM,EAAE,SAAS,SAAS,OAAO,aAAa,IAAI;AAAA,MAChD;AAAA,MACA,MAAM;AACJ,cAAM,uBAAuB;AAAA,UAC3B,kBAAkB,qBAAqB;AAAA,UACvC,CAAC,cAAc;AAAA,QACjB;AAEA,eAAO,yBAAyB,qBAAqB,KAAK,MAAM;AAC9D,iBAAO,mBAAmB,gBAAgB,KAAK,MAAM;AACnD,mBAAO;AAAA,cACL,SAAS;AAAA,cACT,cAAc;AAAA,cACd,OAAO,YAAY,WAAW,MAAM,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;AAAA,cAClD,SAAS,YAAY,WAAW,CAAC,SAAS,CAAC;AAAA,YAC7C;AAAA,UACF,CAAC;AAAA,QACH,CAAC;AAAA,MACH;AAAA,IACF;AAEA,UAAM,QAAQ;AAAA,MACZ,OAAO;AAAA,QACL,SAAS,QAAQ;AAAA,QACjB,SAAS,QAAQ;AAAA,QACjB,OAAO,MAAM;AAAA,QACb,cAAc,aAAa;AAAA,MAC7B;AAAA,MACA,CAAC,QAAQ,OAAO,QAAQ,OAAO,MAAM,OAAO,aAAa,KAAK;AAAA,IAChE;AAEA,WAAO,OAA2B;AAAA,MAChC,UAAU,MAAM;AAAA,MAEhB,SAAS,QAAQ;AAAA,MACjB,SAAS,QAAQ;AAAA,MACjB,OAAO,MAAM;AAAA,MACb,cAAc,aAAa;AAAA,MAC3B,IAAI,OAAO;AAAA,IACb,CAAC;AAAA,EACH;AACF;AAEA,IAAM,qBAAqB,CAAC,WAA+C;AACzE,QAAM,UAAU,MAAM;AACpB,WAAO,OAAO,SAAS,EAAE,IAAI,QAAQ,KAAK,MAAM;AAAA,EAClD;AACA,SAAO;AAAA,IACL,SAAS,wBAAwB;AAAA,MAC/B,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI;AAAA,IACnC,CAAC;AAAA,IACD,SAAS,wBAAwB;AAAA,MAC/B,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI;AAAA,IACnC,CAAC;AAAA,IACD,OAAO,wBAAwB;AAAA,MAC7B,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI;AAAA,IACnC,CAAC;AAAA,IACD,cAAc,wBAAwB;AAAA,MACpC,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI;AAAA,IACnC,CAAC;AAAA,IACD,QAAQ,wBAAwB;AAAA,MAC9B,QAAQ;AAAA,MACR,OAAO,EAAE,MAAM,OAAO;AAAA,MACtB,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI,QAAQ,OAAO,MAAM;AAAA,IACxD,CAAC;AAAA,IACD,gBAAgB,wBAAwB;AAAA,MACtC,QAAQ;AAAA,MACR,OAAO,EAAE,MAAM,OAAO;AAAA,MACtB,KAAK,MAAM,QAAQ;AAAA,IACrB,CAAC;AAAA,IACD,UAAU,wBAAwB;AAAA,MAChC,QAAQ;AAAA,MACR,OAAO,CAAC;AAAA,MACR,KAAK,MAAM,OAAO,SAAS,EAAE,IAAI,QAAQ,OAAO,MAAM,EAAE;AAAA,IAC1D,CAAC;AAAA,IACD,GAAG,UAAU,UAAU;AACrB,YAAM,EAAE,OAAO,MAAM,IAAI,uBAAuB,QAAQ;AACxD,UAAI,UAAU,IAAK,QAAO,OAAO,SAAS,EAAE,IAAI,GAAG,OAAO,QAAQ;AAElE,UACE,gBAAgB,UAAU,OAAO,KAAK,KACtC,gBAAgB,oBAAoB,OAAO,KAAK,KAChD,gBAAgB,YAAY,OAAO,KAAK,GACxC;AACA,eAAO,OAAO,SAAS,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM;AAC5C,cAAI,EAAE,aAAa,QAAQ,EAAE,SAAS,EAAE,GAAI;AAC5C,mBAAS,CAAC;AAAA,QACZ,CAAC;AAAA,MACH;AAEA,YAAM,IAAI;AAAA,QACR,mDAAmD,KAAK;AAAA,MAC1D;AAAA,IACF;AAAA,IACA,WAAW,OAAO;AAAA,IAClB,WAAW,OAAO;AAAA,EACpB;AACF;AAmBO,IAAM,qBAAqB,CAAC,UAAgC;AACjE,QAAM,MAAM,oBAAoB;AAChC,QAAM,SAAS,YAAY,QAAQ,eAAe,KAAK,CAAC,CAAC;AACzD,QAAM,YAAY,QAAQ,MAAM,mBAAmB,MAAM,GAAG,CAAC,MAAM,CAAC;AACpE,SAAO,QAAQ,MAAM,UAAU,KAAK,SAAS,GAAG,CAAC,KAAK,SAAS,CAAC;AAClE;","names":[]}
@@ -0,0 +1,8 @@
1
+ import { Unsubscribe } from "@assistant-ui/tap";
2
+ import { ModelContextProvider } from "../model-context/ModelContextTypes";
3
+ export type ModelContextRegistrar = ModelContextProvider & {
4
+ register: (provider: ModelContextProvider) => Unsubscribe;
5
+ };
6
+ export declare const withModelContextProvider: <TResult>(modelContext: ModelContextRegistrar, fn: () => TResult) => TResult;
7
+ export declare const tapModelContext: () => ModelContextRegistrar;
8
+ //# sourceMappingURL=ModelContext.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ModelContext.d.ts","sourceRoot":"","sources":["../../src/client/ModelContext.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAE1E,MAAM,MAAM,qBAAqB,GAAG,oBAAoB,GAAG;IACzD,QAAQ,EAAE,CAAC,QAAQ,EAAE,oBAAoB,KAAK,WAAW,CAAC;CAC3D,CAAC;AAIF,eAAO,MAAM,wBAAwB,GAAI,OAAO,EAC9C,cAAc,qBAAqB,EACnC,IAAI,MAAM,OAAO,YAGlB,CAAC;AAEF,eAAO,MAAM,eAAe,6BAM3B,CAAC"}
@@ -0,0 +1,21 @@
1
+ // src/client/ModelContext.ts
2
+ import {
3
+ createContext,
4
+ tapContext,
5
+ withContextProvider
6
+ } from "@assistant-ui/tap";
7
+ var ModelContextContext = createContext(null);
8
+ var withModelContextProvider = (modelContext, fn) => {
9
+ return withContextProvider(ModelContextContext, modelContext, fn);
10
+ };
11
+ var tapModelContext = () => {
12
+ const modelContext = tapContext(ModelContextContext);
13
+ if (!modelContext)
14
+ throw new Error("Model context is not available in this context");
15
+ return modelContext;
16
+ };
17
+ export {
18
+ tapModelContext,
19
+ withModelContextProvider
20
+ };
21
+ //# sourceMappingURL=ModelContext.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../src/client/ModelContext.ts"],"sourcesContent":["import {\n createContext,\n tapContext,\n withContextProvider,\n Unsubscribe,\n} from \"@assistant-ui/tap\";\nimport { ModelContextProvider } from \"../model-context/ModelContextTypes\";\n\nexport type ModelContextRegistrar = ModelContextProvider & {\n register: (provider: ModelContextProvider) => Unsubscribe;\n};\n\nconst ModelContextContext = createContext<ModelContextRegistrar | null>(null);\n\nexport const withModelContextProvider = <TResult>(\n modelContext: ModelContextRegistrar,\n fn: () => TResult,\n) => {\n return withContextProvider(ModelContextContext, modelContext, fn);\n};\n\nexport const tapModelContext = () => {\n const modelContext = tapContext(ModelContextContext);\n if (!modelContext)\n throw new Error(\"Model context is not available in this context\");\n\n return modelContext;\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAOP,IAAM,sBAAsB,cAA4C,IAAI;AAErE,IAAM,2BAA2B,CACtC,cACA,OACG;AACH,SAAO,oBAAoB,qBAAqB,cAAc,EAAE;AAClE;AAEO,IAAM,kBAAkB,MAAM;AACnC,QAAM,eAAe,WAAW,mBAAmB;AACnD,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,gDAAgD;AAElE,SAAO;AACT;","names":[]}
@@ -0,0 +1,7 @@
1
+ import type { ModelContextApi } from "./types/ModelContext";
2
+ export declare const ModelContext: import("@assistant-ui/tap").ResourceElementConstructor<{
3
+ key: string | undefined;
4
+ state: any;
5
+ api: ModelContextApi;
6
+ }, undefined>;
7
+ //# sourceMappingURL=ModelContextClient.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ModelContextClient.d.ts","sourceRoot":"","sources":["../../src/client/ModelContextClient.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAqB,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAE/E,eAAO,MAAM,YAAY;;;;aAUvB,CAAC"}
@@ -0,0 +1,18 @@
1
+ // src/client/ModelContextClient.ts
2
+ import { resource, tapState } from "@assistant-ui/tap";
3
+ import { tapApi } from "../utils/tap-store/index.js";
4
+ import { CompositeContextProvider } from "../utils/CompositeContextProvider.js";
5
+ var ModelContext = resource(() => {
6
+ const [state] = tapState(() => ({}));
7
+ const composite = new CompositeContextProvider();
8
+ return tapApi({
9
+ getState: () => state,
10
+ getModelContext: () => composite.getModelContext(),
11
+ subscribe: (callback) => composite.subscribe(callback),
12
+ register: (provider) => composite.registerModelContextProvider(provider)
13
+ });
14
+ });
15
+ export {
16
+ ModelContext
17
+ };
18
+ //# sourceMappingURL=ModelContextClient.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../src/client/ModelContextClient.ts"],"sourcesContent":["import { resource, tapState } from \"@assistant-ui/tap\";\nimport { tapApi } from \"../utils/tap-store\";\nimport { CompositeContextProvider } from \"../utils/CompositeContextProvider\";\nimport type { ModelContextState, ModelContextApi } from \"./types/ModelContext\";\n\nexport const ModelContext = resource(() => {\n const [state] = tapState<ModelContextState>(() => ({}));\n const composite = new CompositeContextProvider();\n\n return tapApi<ModelContextApi>({\n getState: () => state,\n getModelContext: () => composite.getModelContext(),\n subscribe: (callback) => composite.subscribe(callback),\n register: (provider) => composite.registerModelContextProvider(provider),\n });\n});\n"],"mappings":";AAAA,SAAS,UAAU,gBAAgB;AACnC,SAAS,cAAc;AACvB,SAAS,gCAAgC;AAGlC,IAAM,eAAe,SAAS,MAAM;AACzC,QAAM,CAAC,KAAK,IAAI,SAA4B,OAAO,CAAC,EAAE;AACtD,QAAM,YAAY,IAAI,yBAAyB;AAE/C,SAAO,OAAwB;AAAA,IAC7B,UAAU,MAAM;AAAA,IAChB,iBAAiB,MAAM,UAAU,gBAAgB;AAAA,IACjD,WAAW,CAAC,aAAa,UAAU,UAAU,QAAQ;AAAA,IACrD,UAAU,CAAC,aAAa,UAAU,6BAA6B,QAAQ;AAAA,EACzE,CAAC;AACH,CAAC;","names":[]}
@@ -1 +1 @@
1
- {"version":3,"file":"ToolUIClient.d.ts","sourceRoot":"","sources":["../../src/client/ToolUIClient.ts"],"names":[],"mappings":"AAEA,OAAO,EAAe,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAExD,eAAO,MAAM,YAAY;;;;aAwBvB,CAAC"}
1
+ {"version":3,"file":"ToolUIClient.d.ts","sourceRoot":"","sources":["../../src/client/ToolUIClient.ts"],"names":[],"mappings":"AAEA,OAAO,EAAe,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAExD,eAAO,MAAM,YAAY;;;;aAuEvB,CAAC"}
@@ -2,21 +2,63 @@
2
2
  import { resource, tapState } from "@assistant-ui/tap";
3
3
  import { tapApi } from "../utils/tap-store/index.js";
4
4
  var ToolUIClient = resource(() => {
5
- const [state, setState] = tapState(() => ({}));
5
+ const [state, setState] = tapState(() => ({
6
+ tools: {},
7
+ fallback: [],
8
+ layout: []
9
+ }));
6
10
  return tapApi({
7
11
  getState: () => state,
8
12
  setToolUI: (toolName, render) => {
9
13
  setState((prev) => {
10
14
  return {
11
15
  ...prev,
12
- [toolName]: [...prev[toolName] ?? [], render]
16
+ tools: {
17
+ ...prev.tools,
18
+ [toolName]: [...prev.tools[toolName] ?? [], render]
19
+ }
13
20
  };
14
21
  });
15
22
  return () => {
16
23
  setState((prev) => {
17
24
  return {
18
25
  ...prev,
19
- [toolName]: prev[toolName]?.filter((r) => r !== render) ?? []
26
+ tools: {
27
+ ...prev.tools,
28
+ [toolName]: prev.tools[toolName]?.filter((r) => r !== render) ?? []
29
+ }
30
+ };
31
+ });
32
+ };
33
+ },
34
+ setFallbackToolUI: (render) => {
35
+ setState((prev) => {
36
+ return {
37
+ ...prev,
38
+ fallback: [...prev.fallback, render]
39
+ };
40
+ });
41
+ return () => {
42
+ setState((prev) => {
43
+ return {
44
+ ...prev,
45
+ fallback: prev.fallback.filter((r) => r !== render)
46
+ };
47
+ });
48
+ };
49
+ },
50
+ setToolUILayout: (render) => {
51
+ setState((prev) => {
52
+ return {
53
+ ...prev,
54
+ layout: [...prev.layout, render]
55
+ };
56
+ });
57
+ return () => {
58
+ setState((prev) => {
59
+ return {
60
+ ...prev,
61
+ layout: prev.layout.filter((r) => r !== render)
20
62
  };
21
63
  });
22
64
  };
@@ -1 +1 @@
1
- {"version":3,"sources":["../../src/client/ToolUIClient.ts"],"sourcesContent":["import { resource, tapState } from \"@assistant-ui/tap\";\nimport { tapApi } from \"../utils/tap-store\";\nimport { ToolUIState, ToolUIApi } from \"./types/ToolUI\";\n\nexport const ToolUIClient = resource(() => {\n const [state, setState] = tapState<ToolUIState>(() => ({}));\n\n return tapApi<ToolUIApi>({\n getState: () => state,\n\n setToolUI: (toolName, render) => {\n setState((prev) => {\n return {\n ...prev,\n [toolName]: [...(prev[toolName] ?? []), render],\n };\n });\n\n return () => {\n setState((prev) => {\n return {\n ...prev,\n [toolName]: prev[toolName]?.filter((r) => r !== render) ?? [],\n };\n });\n };\n },\n });\n});\n"],"mappings":";AAAA,SAAS,UAAU,gBAAgB;AACnC,SAAS,cAAc;AAGhB,IAAM,eAAe,SAAS,MAAM;AACzC,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAsB,OAAO,CAAC,EAAE;AAE1D,SAAO,OAAkB;AAAA,IACvB,UAAU,MAAM;AAAA,IAEhB,WAAW,CAAC,UAAU,WAAW;AAC/B,eAAS,CAAC,SAAS;AACjB,eAAO;AAAA,UACL,GAAG;AAAA,UACH,CAAC,QAAQ,GAAG,CAAC,GAAI,KAAK,QAAQ,KAAK,CAAC,GAAI,MAAM;AAAA,QAChD;AAAA,MACF,CAAC;AAED,aAAO,MAAM;AACX,iBAAS,CAAC,SAAS;AACjB,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,CAAC,QAAQ,GAAG,KAAK,QAAQ,GAAG,OAAO,CAAC,MAAM,MAAM,MAAM,KAAK,CAAC;AAAA,UAC9D;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF,CAAC;AACH,CAAC;","names":[]}
1
+ {"version":3,"sources":["../../src/client/ToolUIClient.ts"],"sourcesContent":["import { resource, tapState } from \"@assistant-ui/tap\";\nimport { tapApi } from \"../utils/tap-store\";\nimport { ToolUIState, ToolUIApi } from \"./types/ToolUI\";\n\nexport const ToolUIClient = resource(() => {\n const [state, setState] = tapState<ToolUIState>(() => ({\n tools: {},\n fallback: [],\n layout: [],\n }));\n\n return tapApi<ToolUIApi>({\n getState: () => state,\n\n setToolUI: (toolName, render) => {\n setState((prev) => {\n return {\n ...prev,\n tools: {\n ...prev.tools,\n [toolName]: [...(prev.tools[toolName] ?? []), render],\n },\n };\n });\n\n return () => {\n setState((prev) => {\n return {\n ...prev,\n tools: {\n ...prev.tools,\n [toolName]:\n prev.tools[toolName]?.filter((r) => r !== render) ?? [],\n },\n };\n });\n };\n },\n\n setFallbackToolUI: (render) => {\n setState((prev) => {\n return {\n ...prev,\n fallback: [...prev.fallback, render],\n };\n });\n\n return () => {\n setState((prev) => {\n return {\n ...prev,\n fallback: prev.fallback.filter((r) => r !== render),\n };\n });\n };\n },\n\n setToolUILayout: (render) => {\n setState((prev) => {\n return {\n ...prev,\n layout: [...prev.layout, render],\n };\n });\n\n return () => {\n setState((prev) => {\n return {\n ...prev,\n layout: prev.layout.filter((r) => r !== render),\n };\n });\n };\n },\n });\n});\n"],"mappings":";AAAA,SAAS,UAAU,gBAAgB;AACnC,SAAS,cAAc;AAGhB,IAAM,eAAe,SAAS,MAAM;AACzC,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAsB,OAAO;AAAA,IACrD,OAAO,CAAC;AAAA,IACR,UAAU,CAAC;AAAA,IACX,QAAQ,CAAC;AAAA,EACX,EAAE;AAEF,SAAO,OAAkB;AAAA,IACvB,UAAU,MAAM;AAAA,IAEhB,WAAW,CAAC,UAAU,WAAW;AAC/B,eAAS,CAAC,SAAS;AACjB,eAAO;AAAA,UACL,GAAG;AAAA,UACH,OAAO;AAAA,YACL,GAAG,KAAK;AAAA,YACR,CAAC,QAAQ,GAAG,CAAC,GAAI,KAAK,MAAM,QAAQ,KAAK,CAAC,GAAI,MAAM;AAAA,UACtD;AAAA,QACF;AAAA,MACF,CAAC;AAED,aAAO,MAAM;AACX,iBAAS,CAAC,SAAS;AACjB,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,OAAO;AAAA,cACL,GAAG,KAAK;AAAA,cACR,CAAC,QAAQ,GACP,KAAK,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,MAAM,MAAM,KAAK,CAAC;AAAA,YAC1D;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IAEA,mBAAmB,CAAC,WAAW;AAC7B,eAAS,CAAC,SAAS;AACjB,eAAO;AAAA,UACL,GAAG;AAAA,UACH,UAAU,CAAC,GAAG,KAAK,UAAU,MAAM;AAAA,QACrC;AAAA,MACF,CAAC;AAED,aAAO,MAAM;AACX,iBAAS,CAAC,SAAS;AACjB,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,UAAU,KAAK,SAAS,OAAO,CAAC,MAAM,MAAM,MAAM;AAAA,UACpD;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IAEA,iBAAiB,CAAC,WAAW;AAC3B,eAAS,CAAC,SAAS;AACjB,eAAO;AAAA,UACL,GAAG;AAAA,UACH,QAAQ,CAAC,GAAG,KAAK,QAAQ,MAAM;AAAA,QACjC;AAAA,MACF,CAAC;AAED,aAAO,MAAM;AACX,iBAAS,CAAC,SAAS;AACjB,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,QAAQ,KAAK,OAAO,OAAO,CAAC,MAAM,MAAM,MAAM;AAAA,UAChD;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF,CAAC;AACH,CAAC;","names":[]}
@@ -0,0 +1,4 @@
1
+ import type { ToolUIApi } from "./types/ToolUI";
2
+ export declare const withToolUIProvider: <TResult>(toolUIs: ToolUIApi, fn: () => TResult) => TResult;
3
+ export declare const tapToolUI: () => ToolUIApi;
4
+ //# sourceMappingURL=ToolUIContext.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ToolUIContext.d.ts","sourceRoot":"","sources":["../../src/client/ToolUIContext.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAIhD,eAAO,MAAM,kBAAkB,GAAI,OAAO,EACxC,SAAS,SAAS,EAClB,IAAI,MAAM,OAAO,YAGlB,CAAC;AAEF,eAAO,MAAM,SAAS,iBAKrB,CAAC"}
@@ -0,0 +1,20 @@
1
+ // src/client/ToolUIContext.ts
2
+ import {
3
+ createContext,
4
+ tapContext,
5
+ withContextProvider
6
+ } from "@assistant-ui/tap";
7
+ var ToolUIContext = createContext(null);
8
+ var withToolUIProvider = (toolUIs, fn) => {
9
+ return withContextProvider(ToolUIContext, toolUIs, fn);
10
+ };
11
+ var tapToolUI = () => {
12
+ const toolUIs = tapContext(ToolUIContext);
13
+ if (!toolUIs) throw new Error("ToolUI context is not available");
14
+ return toolUIs;
15
+ };
16
+ export {
17
+ tapToolUI,
18
+ withToolUIProvider
19
+ };
20
+ //# sourceMappingURL=ToolUIContext.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../src/client/ToolUIContext.ts"],"sourcesContent":["import {\n createContext,\n tapContext,\n withContextProvider,\n} from \"@assistant-ui/tap\";\nimport type { ToolUIApi } from \"./types/ToolUI\";\n\nconst ToolUIContext = createContext<ToolUIApi | null>(null);\n\nexport const withToolUIProvider = <TResult>(\n toolUIs: ToolUIApi,\n fn: () => TResult,\n) => {\n return withContextProvider(ToolUIContext, toolUIs, fn);\n};\n\nexport const tapToolUI = () => {\n const toolUIs = tapContext(ToolUIContext);\n if (!toolUIs) throw new Error(\"ToolUI context is not available\");\n\n return toolUIs;\n};\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAGP,IAAM,gBAAgB,cAAgC,IAAI;AAEnD,IAAM,qBAAqB,CAChC,SACA,OACG;AACH,SAAO,oBAAoB,eAAe,SAAS,EAAE;AACvD;AAEO,IAAM,YAAY,MAAM;AAC7B,QAAM,UAAU,WAAW,aAAa;AACxC,MAAI,CAAC,QAAS,OAAM,IAAI,MAAM,iCAAiC;AAE/D,SAAO;AACT;","names":[]}
@@ -0,0 +1,10 @@
1
+ import { ToolsApi } from "./types/Tools";
2
+ import { type Toolkit } from "../model-context/toolbox";
3
+ export declare const Tools: import("@assistant-ui/tap").ResourceElementConstructor<{
4
+ key: string | undefined;
5
+ state: any;
6
+ api: ToolsApi;
7
+ }, {
8
+ toolkit?: Toolkit;
9
+ }>;
10
+ //# sourceMappingURL=Tools.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"Tools.d.ts","sourceRoot":"","sources":["../../src/client/Tools.ts"],"names":[],"mappings":"AAIA,OAAO,EAAc,QAAQ,EAAE,MAAM,eAAe,CAAC;AAErD,OAAO,EACL,KAAK,OAAO,EAGb,MAAM,0BAA0B,CAAC;AAElC,eAAO,MAAM,KAAK;;;;;cAAsC,OAAO;EAuD7D,CAAC"}
@@ -0,0 +1,55 @@
1
+ // src/client/Tools.ts
2
+ import { resource, tapState, tapEffect } from "@assistant-ui/tap";
3
+ import { tapApi } from "../utils/tap-store/index.js";
4
+ import { tapModelContext } from "./ModelContext.js";
5
+ import { tapToolUI } from "./ToolUIContext.js";
6
+ import {
7
+ FallbackSymbol,
8
+ LayoutSymbol
9
+ } from "../model-context/toolbox.js";
10
+ var Tools = resource(({ toolkit }) => {
11
+ const [state] = tapState(() => ({}));
12
+ const modelContext = tapModelContext();
13
+ const toolUI = tapToolUI();
14
+ tapEffect(() => {
15
+ if (!toolkit) return;
16
+ const unsubscribes = [];
17
+ const fallback = toolkit[FallbackSymbol];
18
+ if (fallback?.render) {
19
+ unsubscribes.push(toolUI.setFallbackToolUI(fallback.render));
20
+ }
21
+ const layout = toolkit[LayoutSymbol];
22
+ if (layout?.render) {
23
+ unsubscribes.push(toolUI.setToolUILayout(layout.render));
24
+ }
25
+ for (const [toolName, tool] of Object.entries(toolkit)) {
26
+ if (tool.render) {
27
+ unsubscribes.push(toolUI.setToolUI(toolName, tool.render));
28
+ }
29
+ }
30
+ const toolsWithoutRender = Object.entries(toolkit).reduce(
31
+ (acc, [name, tool]) => {
32
+ const { render, ...rest } = tool;
33
+ acc[name] = rest;
34
+ return acc;
35
+ },
36
+ {}
37
+ );
38
+ const modelContextProvider = {
39
+ getModelContext: () => ({
40
+ tools: toolsWithoutRender
41
+ })
42
+ };
43
+ unsubscribes.push(modelContext.register(modelContextProvider));
44
+ return () => {
45
+ unsubscribes.forEach((fn) => fn());
46
+ };
47
+ }, [toolkit, modelContext, toolUI]);
48
+ return tapApi({
49
+ getState: () => state
50
+ });
51
+ });
52
+ export {
53
+ Tools
54
+ };
55
+ //# sourceMappingURL=Tools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../src/client/Tools.ts"],"sourcesContent":["import { resource, tapState, tapEffect } from \"@assistant-ui/tap\";\nimport { tapApi } from \"../utils/tap-store\";\nimport { tapModelContext } from \"./ModelContext\";\nimport { tapToolUI } from \"./ToolUIContext\";\nimport { ToolsState, ToolsApi } from \"./types/Tools\";\nimport type { Tool } from \"assistant-stream\";\nimport {\n type Toolkit,\n FallbackSymbol,\n LayoutSymbol,\n} from \"../model-context/toolbox\";\n\nexport const Tools = resource(({ toolkit }: { toolkit?: Toolkit }) => {\n const [state] = tapState<ToolsState>(() => ({}));\n\n const modelContext = tapModelContext();\n const toolUI = tapToolUI();\n\n tapEffect(() => {\n if (!toolkit) return;\n const unsubscribes: (() => void)[] = [];\n\n // Register fallback UI\n const fallback = toolkit[FallbackSymbol];\n if (fallback?.render) {\n unsubscribes.push(toolUI.setFallbackToolUI(fallback.render));\n }\n\n // Register layout\n const layout = toolkit[LayoutSymbol];\n if (layout?.render) {\n unsubscribes.push(toolUI.setToolUILayout(layout.render));\n }\n\n // Register tool UIs (exclude symbols)\n for (const [toolName, tool] of Object.entries(toolkit)) {\n if (tool.render) {\n unsubscribes.push(toolUI.setToolUI(toolName, tool.render));\n }\n }\n\n // Register tools with model context (exclude symbols)\n const toolsWithoutRender = Object.entries(toolkit).reduce(\n (acc, [name, tool]) => {\n const { render, ...rest } = tool;\n acc[name] = rest;\n return acc;\n },\n {} as Record<string, Tool<any, any>>,\n );\n\n const modelContextProvider = {\n getModelContext: () => ({\n tools: toolsWithoutRender,\n }),\n };\n\n unsubscribes.push(modelContext.register(modelContextProvider));\n\n return () => {\n unsubscribes.forEach((fn) => fn());\n };\n }, [toolkit, modelContext, toolUI]);\n\n return tapApi<ToolsApi>({\n getState: () => state,\n });\n});\n"],"mappings":";AAAA,SAAS,UAAU,UAAU,iBAAiB;AAC9C,SAAS,cAAc;AACvB,SAAS,uBAAuB;AAChC,SAAS,iBAAiB;AAG1B;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AAEA,IAAM,QAAQ,SAAS,CAAC,EAAE,QAAQ,MAA6B;AACpE,QAAM,CAAC,KAAK,IAAI,SAAqB,OAAO,CAAC,EAAE;AAE/C,QAAM,eAAe,gBAAgB;AACrC,QAAM,SAAS,UAAU;AAEzB,YAAU,MAAM;AACd,QAAI,CAAC,QAAS;AACd,UAAM,eAA+B,CAAC;AAGtC,UAAM,WAAW,QAAQ,cAAc;AACvC,QAAI,UAAU,QAAQ;AACpB,mBAAa,KAAK,OAAO,kBAAkB,SAAS,MAAM,CAAC;AAAA,IAC7D;AAGA,UAAM,SAAS,QAAQ,YAAY;AACnC,QAAI,QAAQ,QAAQ;AAClB,mBAAa,KAAK,OAAO,gBAAgB,OAAO,MAAM,CAAC;AAAA,IACzD;AAGA,eAAW,CAAC,UAAU,IAAI,KAAK,OAAO,QAAQ,OAAO,GAAG;AACtD,UAAI,KAAK,QAAQ;AACf,qBAAa,KAAK,OAAO,UAAU,UAAU,KAAK,MAAM,CAAC;AAAA,MAC3D;AAAA,IACF;AAGA,UAAM,qBAAqB,OAAO,QAAQ,OAAO,EAAE;AAAA,MACjD,CAAC,KAAK,CAAC,MAAM,IAAI,MAAM;AACrB,cAAM,EAAE,QAAQ,GAAG,KAAK,IAAI;AAC5B,YAAI,IAAI,IAAI;AACZ,eAAO;AAAA,MACT;AAAA,MACA,CAAC;AAAA,IACH;AAEA,UAAM,uBAAuB;AAAA,MAC3B,iBAAiB,OAAO;AAAA,QACtB,OAAO;AAAA,MACT;AAAA,IACF;AAEA,iBAAa,KAAK,aAAa,SAAS,oBAAoB,CAAC;AAE7D,WAAO,MAAM;AACX,mBAAa,QAAQ,CAAC,OAAO,GAAG,CAAC;AAAA,IACnC;AAAA,EACF,GAAG,CAAC,SAAS,cAAc,MAAM,CAAC;AAElC,SAAO,OAAiB;AAAA,IACtB,UAAU,MAAM;AAAA,EAClB,CAAC;AACH,CAAC;","names":[]}
@@ -0,0 +1,12 @@
1
+ import type { Unsubscribe } from "@assistant-ui/tap";
2
+ import type { ModelContextProvider } from "../../model-context/ModelContextTypes";
3
+ export type ModelContextState = {};
4
+ export type ModelContextApi = ModelContextProvider & {
5
+ getState(): ModelContextState;
6
+ register: (provider: ModelContextProvider) => Unsubscribe;
7
+ };
8
+ export type ModelContextMeta = {
9
+ source: "root";
10
+ query: Record<string, never>;
11
+ };
12
+ //# sourceMappingURL=ModelContext.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ModelContext.d.ts","sourceRoot":"","sources":["../../../src/client/types/ModelContext.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uCAAuC,CAAC;AAGlF,MAAM,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAEnC,MAAM,MAAM,eAAe,GAAG,oBAAoB,GAAG;IACnD,QAAQ,IAAI,iBAAiB,CAAC;IAC9B,QAAQ,EAAE,CAAC,QAAQ,EAAE,oBAAoB,KAAK,WAAW,CAAC;CAC3D,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;CAC9B,CAAC"}
@@ -0,0 +1 @@
1
+ //# sourceMappingURL=ModelContext.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
@@ -1 +1 @@
1
- {"version":3,"file":"ThreadList.d.ts","sourceRoot":"","sources":["../../../src/client/types/ThreadList.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,uBAAuB,EACxB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAE9D,MAAM,MAAM,qBAAqB,GAAG;IAClC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IACtC,QAAQ,CAAC,iBAAiB,EAAE,SAAS,MAAM,EAAE,CAAC;IAE9C,QAAQ,CAAC,WAAW,EAAE,SAAS,yBAAyB,EAAE,CAAC;IAE3D,QAAQ,CAAC,IAAI,EAAE,iBAAiB,CAAC;CAClC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,IAAI,qBAAqB,CAAC;IAElC,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACvC,iBAAiB,IAAI,IAAI,CAAC;IAC1B,IAAI,CACF,iBAAiB,EACb,MAAM,GACN;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GACd;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GACxC,uBAAuB,CAAC;IAE3B,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,eAAe,CAAC;CAC3C,CAAC"}
1
+ {"version":3,"file":"ThreadList.d.ts","sourceRoot":"","sources":["../../../src/client/types/ThreadList.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,uBAAuB,EACxB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG9D,MAAM,MAAM,qBAAqB,GAAG;IAClC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAAC;IACtC,QAAQ,CAAC,iBAAiB,EAAE,SAAS,MAAM,EAAE,CAAC;IAE9C,QAAQ,CAAC,WAAW,EAAE,SAAS,yBAAyB,EAAE,CAAC;IAE3D,QAAQ,CAAC,IAAI,EAAE,iBAAiB,CAAC;CAClC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,IAAI,qBAAqB,CAAC;IAElC,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACvC,iBAAiB,IAAI,IAAI,CAAC;IAC1B,IAAI,CACF,iBAAiB,EACb,MAAM,GACN;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,GACd;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;KAAE,GACxC,uBAAuB,CAAC;IAE3B,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,eAAe,CAAC;CAI3C,CAAC"}
@@ -1,9 +1,20 @@
1
1
  import { Unsubscribe } from "@assistant-ui/tap";
2
+ import type { ComponentType, ReactNode } from "react";
2
3
  import { ToolCallMessagePartComponent } from "../../types";
3
- export type ToolUIState = Record<string, ToolCallMessagePartComponent[]>;
4
+ export type ToolUIState = {
5
+ tools: Record<string, ToolCallMessagePartComponent[]>;
6
+ fallback: ToolCallMessagePartComponent[];
7
+ layout: ComponentType<{
8
+ children: ReactNode;
9
+ }>[];
10
+ };
4
11
  export type ToolUIApi = {
5
12
  getState(): ToolUIState;
6
13
  setToolUI(toolName: string, render: ToolCallMessagePartComponent): Unsubscribe;
14
+ setFallbackToolUI(render: ToolCallMessagePartComponent): Unsubscribe;
15
+ setToolUILayout(render: ComponentType<{
16
+ children: ReactNode;
17
+ }>): Unsubscribe;
7
18
  };
8
19
  export type ToolUIMeta = {
9
20
  source: "root";
@@ -1 +1 @@
1
- {"version":3,"file":"ToolUI.d.ts","sourceRoot":"","sources":["../../../src/client/types/ToolUI.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAE3D,MAAM,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,4BAA4B,EAAE,CAAC,CAAC;AAEzE,MAAM,MAAM,SAAS,GAAG;IACtB,QAAQ,IAAI,WAAW,CAAC;IAExB,SAAS,CACP,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,4BAA4B,GACnC,WAAW,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;CAC9B,CAAC"}
1
+ {"version":3,"file":"ToolUI.d.ts","sourceRoot":"","sources":["../../../src/client/types/ToolUI.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACtD,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAE3D,MAAM,MAAM,WAAW,GAAG;IACxB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,4BAA4B,EAAE,CAAC,CAAC;IACtD,QAAQ,EAAE,4BAA4B,EAAE,CAAC;IACzC,MAAM,EAAE,aAAa,CAAC;QAAE,QAAQ,EAAE,SAAS,CAAA;KAAE,CAAC,EAAE,CAAC;CAClD,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG;IACtB,QAAQ,IAAI,WAAW,CAAC;IAExB,SAAS,CACP,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,4BAA4B,GACnC,WAAW,CAAC;IAEf,iBAAiB,CAAC,MAAM,EAAE,4BAA4B,GAAG,WAAW,CAAC;IAErE,eAAe,CAAC,MAAM,EAAE,aAAa,CAAC;QAAE,QAAQ,EAAE,SAAS,CAAA;KAAE,CAAC,GAAG,WAAW,CAAC;CAC9E,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;CAC9B,CAAC"}
@@ -0,0 +1,9 @@
1
+ export type ToolsState = {};
2
+ export type ToolsApi = {
3
+ getState(): ToolsState;
4
+ };
5
+ export type ToolsMeta = {
6
+ source: "root";
7
+ query: Record<string, never>;
8
+ };
9
+ //# sourceMappingURL=Tools.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"Tools.d.ts","sourceRoot":"","sources":["../../../src/client/types/Tools.ts"],"names":[],"mappings":"AACA,MAAM,MAAM,UAAU,GAAG,EAAE,CAAC;AAE5B,MAAM,MAAM,QAAQ,GAAG;IACrB,QAAQ,IAAI,UAAU,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;CAC9B,CAAC"}
@@ -0,0 +1 @@
1
+ //# sourceMappingURL=Tools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}