@assistant-ui/react 0.11.38 → 0.11.41

package/src/model-context/index.ts

@@ -19,12 +19,7 @@ export { tool } from "./tool";
 
 export { makeAssistantVisible } from "./makeAssistantVisible";
 
-export {
-  Toolkit,
-  type ToolDefinition,
-  type ToolkitFallback,
-  type ToolkitLayout,
-} from "./toolbox";
+export type { Toolkit, ToolDefinition } from "./toolbox";
 
 export { Tools } from "../client/Tools";
 

package/src/model-context/makeAssistantVisible.tsx

@@ -11,18 +11,21 @@ import {
   createContext,
   useContext,
 } from "react";
-import { z } from "zod";
 import { useAssistantApi } from "../context/react/AssistantApiContext";
 import { useComposedRefs } from "@radix-ui/react-compose-refs";
 import { tool } from "./tool";
 
-// TODO replace zod with json-schema so we can drop the zod dep
-
 const click = tool({
-  parameters: z.object({
-    clickId: z.string(),
-  }),
-  execute: async ({ clickId }) => {
+  parameters: {
+    type: "object",
+    properties: {
+      clickId: {
+        type: "string",
+      },
+    },
+    required: ["clickId"],
+  },
+  execute: async ({ clickId }: { clickId: string }) => {
     const escapedClickId = CSS.escape(clickId);
     const el = document.querySelector(`[data-click-id='${escapedClickId}']`);
     if (el instanceof HTMLElement) {
@@ -38,11 +41,19 @@ const click = tool({
 });
 
 const edit = tool({
-  parameters: z.object({
-    editId: z.string(),
-    value: z.string(),
-  }),
-  execute: async ({ editId, value }) => {
+  parameters: {
+    type: "object",
+    properties: {
+      editId: {
+        type: "string",
+      },
+      value: {
+        type: "string",
+      },
+    },
+    required: ["editId", "value"],
+  },
+  execute: async ({ editId, value }: { editId: string; value: string }) => {
     const escapedEditId = CSS.escape(editId);
     const el = document.querySelector(`[data-edit-id='${escapedEditId}']`);
     if (el instanceof HTMLInputElement || el instanceof HTMLTextAreaElement) {

package/src/model-context/toolbox.tsx

@@ -1,6 +1,5 @@
 "use client";
 
-import type { ComponentType, ReactNode } from "react";
 import type { Tool } from "assistant-stream";
 import type { ToolCallMessagePartComponent } from "../types/MessagePartComponentTypes";
 
@@ -11,26 +10,7 @@ export type ToolDefinition<
   render?: ToolCallMessagePartComponent<TArgs, TResult> | undefined;
 };
 
-export const FallbackSymbol = Symbol("Toolkit.Fallback");
-export const LayoutSymbol = Symbol("Toolkit.Layout");
-
-export type ToolkitFallback = {
-  render: ToolCallMessagePartComponent<unknown, unknown>;
-};
-
-export type ToolkitLayout = {
-  render: ComponentType<{ children: ReactNode }>;
-};
-
-export type Toolkit = Record<string, ToolDefinition<any, any>> & {
-  [FallbackSymbol]?: ToolkitFallback;
-  [LayoutSymbol]?: ToolkitLayout;
-};
-
-export const Toolkit = {
-  Fallback: FallbackSymbol,
-  Layout: LayoutSymbol,
-} as const;
+export type Toolkit = Record<string, ToolDefinition<any, any>>;
 
 export type ToolsConfig = {
   toolkit: Toolkit;

package/src/model-context/useAssistantTool.tsx

@@ -23,7 +23,7 @@ export const useAssistantTool = <
 
   useEffect(() => {
     if (!tool.render) return undefined;
-    return api.toolUIs().setToolUI(tool.toolName, tool.render);
+    return api.tools().setToolUI(tool.toolName, tool.render);
   }, [api, tool.toolName, tool.render]);
 
   useEffect(() => {

package/src/model-context/useAssistantToolUI.tsx

@@ -15,6 +15,6 @@ export const useAssistantToolUI = (
   const api = useAssistantApi();
   useEffect(() => {
     if (!tool?.toolName || !tool?.render) return undefined;
-    return api.toolUIs().setToolUI(tool.toolName, tool.render);
+    return api.tools().setToolUI(tool.toolName, tool.render);
   }, [api, tool?.toolName, tool?.render]);
 };

package/src/primitives/index.ts

@@ -15,5 +15,6 @@ export { useMessagePartReasoning } from "./messagePart/useMessagePartReasoning";
 export { useMessagePartSource } from "./messagePart/useMessagePartSource";
 export { useMessagePartFile } from "./messagePart/useMessagePartFile";
 export { useMessagePartImage } from "./messagePart/useMessagePartImage";
+export { useMessagePartData } from "./messagePart/useMessagePartData";
 export { useThreadViewportAutoScroll } from "./thread/useThreadViewportAutoScroll";
 export { useScrollLock } from "./reasoning";

package/src/primitives/message/MessageParts.tsx

@@ -252,9 +252,8 @@ const ToolUIDisplay = ({
 }: {
   Fallback: ToolCallMessagePartComponent | undefined;
 } & ToolCallMessagePartProps) => {
-  const Render = useAssistantState(({ toolUIs }) => {
-    const Render =
-      toolUIs.tools[props.toolName] ?? toolUIs.fallback ?? Fallback;
+  const Render = useAssistantState(({ tools }) => {
+    const Render = tools.tools[props.toolName] ?? Fallback;
     if (Array.isArray(Render)) return Render[0] ?? Fallback;
     return Render;
   });
@@ -338,6 +337,9 @@ const MessagePartComponent: FC<MessagePartComponentProps> = ({
     case "audio":
       return <Audio {...part} />;
 
+    case "data":
+      return null;
+
     default:
       const unhandledType: never = type;
       throw new Error(`Unknown message part type: ${unhandledType}`);

package/src/primitives/message/MessagePartsGrouped.tsx

@@ -220,9 +220,8 @@ const ToolUIDisplay = ({
 }: {
   Fallback: ToolCallMessagePartComponent | undefined;
 } & ToolCallMessagePartProps) => {
-  const Render = useAssistantState(({ toolUIs }) => {
-    const Render =
-      toolUIs.tools[props.toolName] ?? toolUIs.fallback ?? Fallback;
+  const Render = useAssistantState(({ tools }) => {
+    const Render = tools.tools[props.toolName] ?? Fallback;
     if (Array.isArray(Render)) return Render[0] ?? Fallback;
     return Render;
   });
@@ -305,6 +304,9 @@ const MessagePartComponent: FC<MessagePartComponentProps> = ({
     case "audio":
       return <Audio {...part} />;
 
+    case "data":
+      return null;
+
     default:
       const unhandledType: never = type;
       throw new Error(`Unknown message part type: ${unhandledType}`);

package/src/primitives/messagePart/useMessagePartData.tsx

@@ -0,0 +1,23 @@
+"use client";
+
+import { useAssistantState } from "../../context";
+import { DataMessagePart } from "../../types";
+
+export const useMessagePartData = <T = any,>(name?: string) => {
+  const part = useAssistantState(({ part }) => {
+    if (part.type !== "data") {
+      return null;
+    }
+    return part as DataMessagePart<T>;
+  });
+
+  if (!part) {
+    return null;
+  }
+
+  if (name && part.name !== name) {
+    return null;
+  }
+
+  return part;
+};

package/src/types/AssistantTypes.ts

@@ -6,6 +6,7 @@ import type {
   SourceMessagePart,
   ImageMessagePart,
   FileMessagePart,
+  DataMessagePart,
   Unstable_AudioMessagePart,
   ToolCallMessagePart,
   ThreadUserMessagePart,
@@ -19,6 +20,7 @@ export type {
   SourceMessagePart,
   ImageMessagePart,
   FileMessagePart,
+  DataMessagePart,
   Unstable_AudioMessagePart,
   ToolCallMessagePart,
   ThreadUserMessagePart,

package/src/types/MessagePartTypes.ts

@@ -43,6 +43,12 @@ export type Unstable_AudioMessagePart = {
   };
 };
 
+export type DataMessagePart<T = any> = {
+  readonly type: "data";
+  readonly name: string;
+  readonly data: T;
+};
+
 export type ToolCallMessagePart<
   TArgs = ReadonlyJSONObject,
   TResult = unknown,
@@ -72,4 +78,5 @@ export type ThreadAssistantMessagePart =
   | ToolCallMessagePart
   | SourceMessagePart
   | FileMessagePart
-  | ImageMessagePart;
+  | ImageMessagePart
+  | DataMessagePart;

package/src/types/index.ts

@@ -12,6 +12,7 @@ export type {
   SourceMessagePart,
   ImageMessagePart,
   FileMessagePart,
+  DataMessagePart,
   Unstable_AudioMessagePart,
   ToolCallMessagePart,
   MessageStatus,

package/dist/client/ToolUIClient.d.ts

@@ -1,7 +0,0 @@
-import { ToolUIApi } from "./types/ToolUI";
-export declare const ToolUIClient: import("@assistant-ui/tap").ResourceElementConstructor<{
-    key: string | undefined;
-    state: any;
-    api: ToolUIApi;
-}, undefined>;
-//# sourceMappingURL=ToolUIClient.d.ts.map

package/dist/client/ToolUIClient.d.ts.map

@@ -1 +0,0 @@
-{"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"}

package/dist/client/ToolUIClient.js

@@ -1,71 +0,0 @@
-// src/client/ToolUIClient.ts
-import { resource, tapState } from "@assistant-ui/tap";
-import { tapApi } from "../utils/tap-store/index.js";
-var ToolUIClient = resource(() => {
-  const [state, setState] = tapState(() => ({
-    tools: {},
-    fallback: [],
-    layout: []
-  }));
-  return tapApi({
-    getState: () => state,
-    setToolUI: (toolName, render) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          tools: {
-            ...prev.tools,
-            [toolName]: [...prev.tools[toolName] ?? [], render]
-          }
-        };
-      });
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            tools: {
-              ...prev.tools,
-              [toolName]: prev.tools[toolName]?.filter((r) => r !== render) ?? []
-            }
-          };
-        });
-      };
-    },
-    setFallbackToolUI: (render) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          fallback: [...prev.fallback, render]
-        };
-      });
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            fallback: prev.fallback.filter((r) => r !== render)
-          };
-        });
-      };
-    },
-    setToolUILayout: (render) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          layout: [...prev.layout, render]
-        };
-      });
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            layout: prev.layout.filter((r) => r !== render)
-          };
-        });
-      };
-    }
-  });
-});
-export {
-  ToolUIClient
-};
-//# sourceMappingURL=ToolUIClient.js.map

package/dist/client/ToolUIClient.js.map

@@ -1 +0,0 @@
-{"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":[]}

package/dist/client/ToolUIContext.d.ts

@@ -1,4 +0,0 @@
-import type { ToolUIApi } from "./types/ToolUI";
-export declare const withToolUIProvider: <TResult>(toolUIs: ToolUIApi, fn: () => TResult) => TResult;
-export declare const tapToolUI: () => ToolUIApi;
-//# sourceMappingURL=ToolUIContext.d.ts.map

package/dist/client/ToolUIContext.d.ts.map

@@ -1 +0,0 @@
-{"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"}

package/dist/client/ToolUIContext.js

@@ -1,20 +0,0 @@
-// src/client/ToolUIContext.ts
-import {
-  createContext,
-  tapContext,
-  withContextProvider
-} from "@assistant-ui/tap";
-var ToolUIContext = createContext(null);
-var withToolUIProvider = (toolUIs, fn) => {
-  return withContextProvider(ToolUIContext, toolUIs, fn);
-};
-var tapToolUI = () => {
-  const toolUIs = tapContext(ToolUIContext);
-  if (!toolUIs) throw new Error("ToolUI context is not available");
-  return toolUIs;
-};
-export {
-  tapToolUI,
-  withToolUIProvider
-};
-//# sourceMappingURL=ToolUIContext.js.map

package/dist/client/ToolUIContext.js.map

@@ -1 +0,0 @@
-{"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":[]}

package/dist/client/types/ToolUI.d.ts

@@ -1,23 +0,0 @@
-import { Unsubscribe } from "@assistant-ui/tap";
-import type { ComponentType, ReactNode } from "react";
-import { ToolCallMessagePartComponent } from "../../types";
-export type ToolUIState = {
-    tools: Record<string, ToolCallMessagePartComponent[]>;
-    fallback: ToolCallMessagePartComponent[];
-    layout: ComponentType<{
-        children: ReactNode;
-    }>[];
-};
-export type ToolUIApi = {
-    getState(): ToolUIState;
-    setToolUI(toolName: string, render: ToolCallMessagePartComponent): Unsubscribe;
-    setFallbackToolUI(render: ToolCallMessagePartComponent): Unsubscribe;
-    setToolUILayout(render: ComponentType<{
-        children: ReactNode;
-    }>): Unsubscribe;
-};
-export type ToolUIMeta = {
-    source: "root";
-    query: Record<string, never>;
-};
-//# sourceMappingURL=ToolUI.d.ts.map

package/dist/client/types/ToolUI.d.ts.map

@@ -1 +0,0 @@
-{"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"}

package/dist/client/types/ToolUI.js

@@ -1 +0,0 @@
-//# sourceMappingURL=ToolUI.js.map

package/dist/client/types/ToolUI.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/src/client/ToolUIClient.ts

@@ -1,76 +0,0 @@
-import { resource, tapState } from "@assistant-ui/tap";
-import { tapApi } from "../utils/tap-store";
-import { ToolUIState, ToolUIApi } from "./types/ToolUI";
-
-export const ToolUIClient = resource(() => {
-  const [state, setState] = tapState<ToolUIState>(() => ({
-    tools: {},
-    fallback: [],
-    layout: [],
-  }));
-
-  return tapApi<ToolUIApi>({
-    getState: () => state,
-
-    setToolUI: (toolName, render) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          tools: {
-            ...prev.tools,
-            [toolName]: [...(prev.tools[toolName] ?? []), render],
-          },
-        };
-      });
-
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            tools: {
-              ...prev.tools,
-              [toolName]:
-                prev.tools[toolName]?.filter((r) => r !== render) ?? [],
-            },
-          };
-        });
-      };
-    },
-
-    setFallbackToolUI: (render) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          fallback: [...prev.fallback, render],
-        };
-      });
-
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            fallback: prev.fallback.filter((r) => r !== render),
-          };
-        });
-      };
-    },
-
-    setToolUILayout: (render) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          layout: [...prev.layout, render],
-        };
-      });
-
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            layout: prev.layout.filter((r) => r !== render),
-          };
-        });
-      };
-    },
-  });
-});

package/src/client/ToolUIContext.ts

@@ -1,22 +0,0 @@
-import {
-  createContext,
-  tapContext,
-  withContextProvider,
-} from "@assistant-ui/tap";
-import type { ToolUIApi } from "./types/ToolUI";
-
-const ToolUIContext = createContext<ToolUIApi | null>(null);
-
-export const withToolUIProvider = <TResult>(
-  toolUIs: ToolUIApi,
-  fn: () => TResult,
-) => {
-  return withContextProvider(ToolUIContext, toolUIs, fn);
-};
-
-export const tapToolUI = () => {
-  const toolUIs = tapContext(ToolUIContext);
-  if (!toolUIs) throw new Error("ToolUI context is not available");
-
-  return toolUIs;
-};

package/src/client/types/ToolUI.ts

@@ -1,27 +0,0 @@
-import { Unsubscribe } from "@assistant-ui/tap";
-import type { ComponentType, ReactNode } from "react";
-import { ToolCallMessagePartComponent } from "../../types";
-
-export type ToolUIState = {
-  tools: Record<string, ToolCallMessagePartComponent[]>;
-  fallback: ToolCallMessagePartComponent[];
-  layout: ComponentType<{ children: ReactNode }>[];
-};
-
-export type ToolUIApi = {
-  getState(): ToolUIState;
-
-  setToolUI(
-    toolName: string,
-    render: ToolCallMessagePartComponent,
-  ): Unsubscribe;
-
-  setFallbackToolUI(render: ToolCallMessagePartComponent): Unsubscribe;
-
-  setToolUILayout(render: ComponentType<{ children: ReactNode }>): Unsubscribe;
-};
-
-export type ToolUIMeta = {
-  source: "root";
-  query: Record<string, never>;
-};