@assistant-ui/react 0.12.11 → 0.12.12

package/dist/types/MessagePartComponentTypes.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"MessagePartComponentTypes.d.ts","sourceRoot":"","sources":["../../src/types/MessagePartComponentTypes.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,OAAO,CAAC;AAC9D,OAAO,KAAK,EACV,iBAAiB,EACjB,eAAe,EACf,eAAe,EACf,gBAAgB,EAChB,oBAAoB,EACpB,iBAAiB,EACjB,eAAe,EACf,mBAAmB,EACnB,yBAAyB,EAC1B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEhD,MAAM,MAAM,qBAAqB,GAAG;IAClC,MAAM,EAAE,iBAAiB,CAAC;CAC3B,CAAC;AACF,MAAM,MAAM,yBAAyB,GAAG,aAAa,CAAC,qBAAqB,CAAC,CAAC;AAE7E,MAAM,MAAM,oBAAoB,GAAG,gBAAgB,GAAG,eAAe,CAAC;AACtE,MAAM,MAAM,wBAAwB,GAAG,aAAa,CAAC,oBAAoB,CAAC,CAAC;AAE3E,MAAM,MAAM,yBAAyB,GAAG,gBAAgB,GAAG,oBAAoB,CAAC;AAChF,MAAM,MAAM,6BAA6B,GACvC,aAAa,CAAC,yBAAyB,CAAC,CAAC;AAE3C,MAAM,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;IAClD,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB,CAAC,CAAC;AACH,MAAM,MAAM,uBAAuB,GAAG,aAAa,CAAC,mBAAmB,CAAC,CAAC;AAEzE,MAAM,MAAM,sBAAsB,GAAG,gBAAgB,GAAG,iBAAiB,CAAC;AAC1E,MAAM,MAAM,0BAA0B,GAAG,aAAa,CAAC,sBAAsB,CAAC,CAAC;AAE/E,MAAM,MAAM,qBAAqB,GAAG,gBAAgB,GAAG,gBAAgB,CAAC;AACxE,MAAM,MAAM,yBAAyB,GAAG,aAAa,CAAC,qBAAqB,CAAC,CAAC;AAE7E,MAAM,MAAM,oBAAoB,GAAG,gBAAgB,GAAG,eAAe,CAAC;AACtE,MAAM,MAAM,wBAAwB,GAAG,aAAa,CAAC,oBAAoB,CAAC,CAAC;AAE3E,MAAM,MAAM,8BAA8B,GAAG,gBAAgB,GAC3D,yBAAyB,CAAC;AAC5B,MAAM,MAAM,kCAAkC,GAC5C,aAAa,CAAC,8BAA8B,CAAC,CAAC;AAEhD,MAAM,MAAM,oBAAoB,CAAC,CAAC,GAAG,GAAG,IAAI,gBAAgB,GAC1D,eAAe,CAAC,CAAC,CAAC,CAAC;AACrB,MAAM,MAAM,wBAAwB,CAAC,CAAC,GAAG,GAAG,IAAI,aAAa,CAC3D,oBAAoB,CAAC,CAAC,CAAC,CACxB,CAAC;AAEF,MAAM,MAAM,wBAAwB,CAClC,KAAK,GAAG,GAAG,EACX,OAAO,GAAG,OAAO,IACf,gBAAgB,GAClB,mBAAmB,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG;IACpC,SAAS,EAAE,CAAC,MAAM,EAAE,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,IAAI,CAAC;IAC7D,MAAM,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;CACpC,CAAC;AAEJ,MAAM,MAAM,4BAA4B,CACtC,KAAK,GAAG,GAAG,EACX,OAAO,GAAG,GAAG,IACX,aAAa,CAAC,wBAAwB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC"}

package/dist/types/MessagePartComponentTypes.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=MessagePartComponentTypes.js.map

package/dist/types/MessagePartComponentTypes.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"MessagePartComponentTypes.js","sourceRoot":"","sources":["../../src/types/MessagePartComponentTypes.ts"],"names":[],"mappings":""}

package/dist/types/scopes/dataRenderers.d.ts

@@ -1,13 +0,0 @@
-import type { DataMessagePartComponent } from "../MessagePartComponentTypes.js";
-import type { Unsubscribe } from "@assistant-ui/core";
-export type DataRenderersState = {
-    renderers: Record<string, DataMessagePartComponent[]>;
-};
-export type DataRenderersMethods = {
-    getState(): DataRenderersState;
-    setDataUI(name: string, render: DataMessagePartComponent): Unsubscribe;
-};
-export type DataRenderersClientSchema = {
-    methods: DataRenderersMethods;
-};
-//# sourceMappingURL=dataRenderers.d.ts.map

package/dist/types/scopes/dataRenderers.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"dataRenderers.d.ts","sourceRoot":"","sources":["../../../src/types/scopes/dataRenderers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAE,wCAAqC;AAC7E,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;CACvD,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,IAAI,kBAAkB,CAAC;IAC/B,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,wBAAwB,GAAG,WAAW,CAAC;CACxE,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,EAAE,oBAAoB,CAAC;CAC/B,CAAC"}

package/dist/types/scopes/dataRenderers.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=dataRenderers.js.map

package/dist/types/scopes/dataRenderers.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"dataRenderers.js","sourceRoot":"","sources":["../../../src/types/scopes/dataRenderers.ts"],"names":[],"mappings":""}

package/dist/types/scopes/tools.d.ts

@@ -1,13 +0,0 @@
-import type { ToolCallMessagePartComponent } from "../MessagePartComponentTypes.js";
-import type { Unsubscribe } from "@assistant-ui/core";
-export type ToolsState = {
-    tools: Record<string, ToolCallMessagePartComponent[]>;
-};
-export type ToolsMethods = {
-    getState(): ToolsState;
-    setToolUI(toolName: string, render: ToolCallMessagePartComponent): Unsubscribe;
-};
-export type ToolsClientSchema = {
-    methods: ToolsMethods;
-};
-//# sourceMappingURL=tools.d.ts.map

package/dist/types/scopes/tools.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../../src/types/scopes/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,4BAA4B,EAAE,wCAAqC;AACjF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEtD,MAAM,MAAM,UAAU,GAAG;IACvB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,4BAA4B,EAAE,CAAC,CAAC;CACvD,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,IAAI,UAAU,CAAC;IACvB,SAAS,CACP,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,4BAA4B,GACnC,WAAW,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,OAAO,EAAE,YAAY,CAAC;CACvB,CAAC"}

package/dist/types/scopes/tools.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=tools.js.map

package/dist/types/scopes/tools.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../src/types/scopes/tools.ts"],"names":[],"mappings":""}

package/dist/types/store-augmentation.d.ts

@@ -1,10 +0,0 @@
-import "@assistant-ui/core/store";
-import type { ToolsClientSchema } from "./scopes/tools.js";
-import type { DataRenderersClientSchema } from "./scopes/dataRenderers.js";
-declare module "@assistant-ui/core/store" {
-    interface ScopeRegistry {
-        tools: ToolsClientSchema;
-        dataRenderers: DataRenderersClientSchema;
-    }
-}
-//# sourceMappingURL=store-augmentation.d.ts.map

package/dist/types/store-augmentation.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"store-augmentation.d.ts","sourceRoot":"","sources":["../../src/types/store-augmentation.ts"],"names":[],"mappings":"AAAA,OAAO,0BAA0B,CAAC;AAElC,OAAO,KAAK,EAAE,iBAAiB,EAAE,0BAAuB;AACxD,OAAO,KAAK,EAAE,yBAAyB,EAAE,kCAA+B;AAExE,OAAO,QAAQ,0BAA0B,CAAC;IACxC,UAAU,aAAa;QACrB,KAAK,EAAE,iBAAiB,CAAC;QACzB,aAAa,EAAE,yBAAyB,CAAC;KAC1C;CACF"}

package/dist/types/store-augmentation.js

@@ -1,2 +0,0 @@
-import "@assistant-ui/core/store";
-//# sourceMappingURL=store-augmentation.js.map

package/dist/types/store-augmentation.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"store-augmentation.js","sourceRoot":"","sources":["../../src/types/store-augmentation.ts"],"names":[],"mappings":"AAAA,OAAO,0BAA0B,CAAC"}

package/src/client/DataRenderers.ts

@@ -1,42 +0,0 @@
-import { resource, tapState, tapCallback } from "@assistant-ui/tap";
-import { type ClientOutput } from "@assistant-ui/store";
-import { DataRenderersState } from "../types/scopes";
-import { DataMessagePartComponent } from "../types";
-
-export const DataRenderers = resource((): ClientOutput<"dataRenderers"> => {
-  const [state, setState] = tapState<DataRenderersState>(() => ({
-    renderers: {},
-  }));
-
-  const setDataUI = tapCallback(
-    (name: string, render: DataMessagePartComponent) => {
-      setState((prev) => {
-        return {
-          ...prev,
-          renderers: {
-            ...prev.renderers,
-            [name]: [...(prev.renderers[name] ?? []), render],
-          },
-        };
-      });
-
-      return () => {
-        setState((prev) => {
-          return {
-            ...prev,
-            renderers: {
-              ...prev.renderers,
-              [name]: prev.renderers[name]?.filter((r) => r !== render) ?? [],
-            },
-          };
-        });
-      };
-    },
-    [],
-  );
-
-  return {
-    getState: () => state,
-    setDataUI,
-  };
-});

package/src/client/Tools.ts

@@ -1,97 +0,0 @@
-import { resource, tapState, tapEffect, tapCallback } from "@assistant-ui/tap";
-import {
-  tapAssistantClientRef,
-  type ClientOutput,
-  attachTransformScopes,
-} from "@assistant-ui/store";
-import { ToolsState } from "../types/scopes";
-import type { Tool } from "assistant-stream";
-import { type Toolkit } from "../model-context/toolbox";
-import { ToolCallMessagePartComponent } from "../types";
-import { ModelContext } from "@assistant-ui/core/store";
-
-export const Tools = resource(
-  ({ toolkit }: { toolkit?: Toolkit }): ClientOutput<"tools"> => {
-    const [state, setState] = tapState<ToolsState>(() => ({
-      tools: {},
-    }));
-
-    const clientRef = tapAssistantClientRef();
-
-    const setToolUI = tapCallback(
-      (toolName: string, render: ToolCallMessagePartComponent) => {
-        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) ?? [],
-              },
-            };
-          });
-        };
-      },
-      [],
-    );
-
-    tapEffect(() => {
-      if (!toolkit) return;
-      const unsubscribes: (() => void)[] = [];
-
-      // Register tool UIs (exclude symbols)
-      for (const [toolName, tool] of Object.entries(toolkit)) {
-        if (tool.render) {
-          unsubscribes.push(setToolUI(toolName, tool.render));
-        }
-      }
-
-      // Register tools with model context (exclude symbols)
-      const toolsWithoutRender = Object.entries(toolkit).reduce(
-        (acc, [name, tool]) => {
-          const { render, ...rest } = tool;
-          acc[name] = rest;
-          return acc;
-        },
-        {} as Record<string, Tool<any, any>>,
-      );
-
-      const modelContextProvider = {
-        getModelContext: () => ({
-          tools: toolsWithoutRender,
-        }),
-      };
-
-      unsubscribes.push(
-        clientRef.current!.modelContext().register(modelContextProvider),
-      );
-
-      return () => {
-        unsubscribes.forEach((fn) => fn());
-      };
-    }, [toolkit, setToolUI, clientRef]);
-
-    return {
-      getState: () => state,
-      setToolUI,
-    };
-  },
-);
-
-attachTransformScopes(Tools, (scopes, parent) => ({
-  ...scopes,
-  ...(scopes.modelContext || parent.modelContext.source !== null
-    ? {}
-    : { modelContext: ModelContext() }),
-}));

package/src/model-context/makeAssistantDataUI.ts

@@ -1,22 +0,0 @@
-"use client";
-
-import type { FC } from "react";
-import {
-  type AssistantDataUIProps,
-  useAssistantDataUI,
-} from "./useAssistantDataUI";
-
-export type AssistantDataUI = FC & {
-  unstable_data: AssistantDataUIProps;
-};
-
-export const makeAssistantDataUI = <T = any>(
-  dataUI: AssistantDataUIProps<T>,
-) => {
-  const DataUI: AssistantDataUI = () => {
-    useAssistantDataUI(dataUI);
-    return null;
-  };
-  DataUI.unstable_data = dataUI;
-  return DataUI;
-};

package/src/model-context/makeAssistantTool.ts

@@ -1,22 +0,0 @@
-"use client";
-
-import { FC } from "react";
-import { type AssistantToolProps, useAssistantTool } from "./useAssistantTool";
-
-export type AssistantTool = FC & {
-  unstable_tool: AssistantToolProps<any, any>;
-};
-
-export const makeAssistantTool = <
-  TArgs extends Record<string, unknown>,
-  TResult,
->(
-  tool: AssistantToolProps<TArgs, TResult>,
-) => {
-  const Tool: AssistantTool = () => {
-    useAssistantTool(tool);
-    return null;
-  };
-  Tool.unstable_tool = tool;
-  return Tool;
-};

package/src/model-context/makeAssistantToolUI.ts

@@ -1,22 +0,0 @@
-"use client";
-
-import { FC } from "react";
-import {
-  type AssistantToolUIProps,
-  useAssistantToolUI,
-} from "./useAssistantToolUI";
-
-export type AssistantToolUI = FC & {
-  unstable_tool: AssistantToolUIProps<any, any>;
-};
-
-export const makeAssistantToolUI = <TArgs, TResult>(
-  tool: AssistantToolUIProps<TArgs, TResult>,
-) => {
-  const ToolUI: AssistantToolUI = () => {
-    useAssistantToolUI(tool);
-    return null;
-  };
-  ToolUI.unstable_tool = tool;
-  return ToolUI;
-};

package/src/model-context/toolbox.ts

@@ -1,17 +0,0 @@
-"use client";
-
-import type { Tool } from "assistant-stream";
-import type { ToolCallMessagePartComponent } from "../types/MessagePartComponentTypes";
-
-export type ToolDefinition<
-  TArgs extends Record<string, unknown>,
-  TResult,
-> = Tool<TArgs, TResult> & {
-  render?: ToolCallMessagePartComponent<TArgs, TResult> | undefined;
-};
-
-export type Toolkit = Record<string, ToolDefinition<any, any>>;
-
-export type ToolsConfig = {
-  toolkit: Toolkit;
-};

package/src/model-context/useAssistantDataUI.ts

@@ -1,18 +0,0 @@
-"use client";
-
-import { useEffect } from "react";
-import { useAui } from "@assistant-ui/store";
-import type { DataMessagePartComponent } from "../types/MessagePartComponentTypes";
-
-export type AssistantDataUIProps<T = any> = {
-  name: string;
-  render: DataMessagePartComponent<T>;
-};
-
-export const useAssistantDataUI = (dataUI: AssistantDataUIProps | null) => {
-  const aui = useAui();
-  useEffect(() => {
-    if (!dataUI?.name || !dataUI?.render) return undefined;
-    return aui.dataRenderers().setDataUI(dataUI.name, dataUI.render);
-  }, [aui, dataUI?.name, dataUI?.render]);
-};

package/src/model-context/useAssistantInstructions.ts

@@ -1,32 +0,0 @@
-"use client";
-
-import { useEffect } from "react";
-import { useAui } from "@assistant-ui/store";
-import type { AssistantInstructionsConfig } from "@assistant-ui/core";
-
-export type { AssistantInstructionsConfig };
-
-const getInstructions = (
-  instruction: string | AssistantInstructionsConfig,
-): AssistantInstructionsConfig => {
-  if (typeof instruction === "string") return { instruction };
-  return instruction;
-};
-
-export const useAssistantInstructions = (
-  config: string | AssistantInstructionsConfig,
-) => {
-  const { instruction, disabled = false } = getInstructions(config);
-  const aui = useAui();
-
-  useEffect(() => {
-    if (disabled) return;
-
-    const config = {
-      system: instruction,
-    };
-    return aui.modelContext().register({
-      getModelContext: () => config,
-    });
-  }, [aui, instruction, disabled]);
-};

package/src/model-context/useAssistantTool.ts

@@ -1,39 +0,0 @@
-"use client";
-
-import { useEffect } from "react";
-import { useAui } from "@assistant-ui/store";
-import type { ToolCallMessagePartComponent } from "../types/MessagePartComponentTypes";
-import type { AssistantToolProps as CoreAssistantToolProps } from "@assistant-ui/core";
-
-export type AssistantToolProps<
-  TArgs extends Record<string, unknown>,
-  TResult,
-> = CoreAssistantToolProps<TArgs, TResult> & {
-  render?: ToolCallMessagePartComponent<TArgs, TResult> | undefined;
-};
-
-export const useAssistantTool = <
-  TArgs extends Record<string, unknown>,
-  TResult,
->(
-  tool: AssistantToolProps<TArgs, TResult>,
-) => {
-  const aui = useAui();
-
-  useEffect(() => {
-    if (!tool.render) return undefined;
-    return aui.tools().setToolUI(tool.toolName, tool.render);
-  }, [aui, tool.toolName, tool.render]);
-
-  useEffect(() => {
-    const { toolName, render, ...rest } = tool;
-    const context = {
-      tools: {
-        [toolName]: rest,
-      },
-    };
-    return aui.modelContext().register({
-      getModelContext: () => context,
-    });
-  }, [aui, tool]);
-};

package/src/model-context/useAssistantToolUI.ts

@@ -1,20 +0,0 @@
-"use client";
-
-import { useEffect } from "react";
-import { useAui } from "@assistant-ui/store";
-import type { ToolCallMessagePartComponent } from "../types/MessagePartComponentTypes";
-
-export type AssistantToolUIProps<TArgs, TResult> = {
-  toolName: string;
-  render: ToolCallMessagePartComponent<TArgs, TResult>;
-};
-
-export const useAssistantToolUI = (
-  tool: AssistantToolUIProps<any, any> | null,
-) => {
-  const aui = useAui();
-  useEffect(() => {
-    if (!tool?.toolName || !tool?.render) return undefined;
-    return aui.tools().setToolUI(tool.toolName, tool.render);
-  }, [aui, tool?.toolName, tool?.render]);
-};

package/src/model-context/useInlineRender.ts

@@ -1,27 +0,0 @@
-"use client";
-
-import { FC, useCallback, useEffect, useState } from "react";
-import { ToolCallMessagePartProps } from "../types";
-import { create } from "zustand";
-
-export const useInlineRender = <TArgs, TResult>(
-  toolUI: FC<ToolCallMessagePartProps<TArgs, TResult>>,
-): FC<ToolCallMessagePartProps<TArgs, TResult>> => {
-  const [useToolUIStore] = useState(() =>
-    create(() => ({
-      toolUI,
-    })),
-  );
-
-  useEffect(() => {
-    useToolUIStore.setState({ toolUI });
-  }, [toolUI, useToolUIStore]);
-
-  return useCallback(
-    function ToolUI(args) {
-      const store = useToolUIStore();
-      return store.toolUI(args);
-    },
-    [useToolUIStore],
-  );
-};

package/src/types/MessagePartComponentTypes.ts

@@ -1,66 +0,0 @@
-import type { ComponentType, PropsWithChildren } from "react";
-import type {
-  MessagePartStatus,
-  DataMessagePart,
-  FileMessagePart,
-  ImageMessagePart,
-  ReasoningMessagePart,
-  SourceMessagePart,
-  TextMessagePart,
-  ToolCallMessagePart,
-  Unstable_AudioMessagePart,
-} from "@assistant-ui/core";
-import type { MessagePartState } from "@assistant-ui/core";
-import { ToolResponse } from "assistant-stream";
-
-export type EmptyMessagePartProps = {
-  status: MessagePartStatus;
-};
-export type EmptyMessagePartComponent = ComponentType<EmptyMessagePartProps>;
-
-export type TextMessagePartProps = MessagePartState & TextMessagePart;
-export type TextMessagePartComponent = ComponentType<TextMessagePartProps>;
-
-export type ReasoningMessagePartProps = MessagePartState & ReasoningMessagePart;
-export type ReasoningMessagePartComponent =
-  ComponentType<ReasoningMessagePartProps>;
-
-export type ReasoningGroupProps = PropsWithChildren<{
-  startIndex: number;
-  endIndex: number;
-}>;
-export type ReasoningGroupComponent = ComponentType<ReasoningGroupProps>;
-
-export type SourceMessagePartProps = MessagePartState & SourceMessagePart;
-export type SourceMessagePartComponent = ComponentType<SourceMessagePartProps>;
-
-export type ImageMessagePartProps = MessagePartState & ImageMessagePart;
-export type ImageMessagePartComponent = ComponentType<ImageMessagePartProps>;
-
-export type FileMessagePartProps = MessagePartState & FileMessagePart;
-export type FileMessagePartComponent = ComponentType<FileMessagePartProps>;
-
-export type Unstable_AudioMessagePartProps = MessagePartState &
-  Unstable_AudioMessagePart;
-export type Unstable_AudioMessagePartComponent =
-  ComponentType<Unstable_AudioMessagePartProps>;
-
-export type DataMessagePartProps<T = any> = MessagePartState &
-  DataMessagePart<T>;
-export type DataMessagePartComponent<T = any> = ComponentType<
-  DataMessagePartProps<T>
->;
-
-export type ToolCallMessagePartProps<
-  TArgs = any,
-  TResult = unknown,
-> = MessagePartState &
-  ToolCallMessagePart<TArgs, TResult> & {
-    addResult: (result: TResult | ToolResponse<TResult>) => void;
-    resume: (payload: unknown) => void;
-  };
-
-export type ToolCallMessagePartComponent<
-  TArgs = any,
-  TResult = any,
-> = ComponentType<ToolCallMessagePartProps<TArgs, TResult>>;

package/src/types/scopes/dataRenderers.ts

@@ -1,15 +0,0 @@
-import type { DataMessagePartComponent } from "../MessagePartComponentTypes";
-import type { Unsubscribe } from "@assistant-ui/core";
-
-export type DataRenderersState = {
-  renderers: Record<string, DataMessagePartComponent[]>;
-};
-
-export type DataRenderersMethods = {
-  getState(): DataRenderersState;
-  setDataUI(name: string, render: DataMessagePartComponent): Unsubscribe;
-};
-
-export type DataRenderersClientSchema = {
-  methods: DataRenderersMethods;
-};

package/src/types/scopes/tools.ts

@@ -1,18 +0,0 @@
-import type { ToolCallMessagePartComponent } from "../MessagePartComponentTypes";
-import type { Unsubscribe } from "@assistant-ui/core";
-
-export type ToolsState = {
-  tools: Record<string, ToolCallMessagePartComponent[]>;
-};
-
-export type ToolsMethods = {
-  getState(): ToolsState;
-  setToolUI(
-    toolName: string,
-    render: ToolCallMessagePartComponent,
-  ): Unsubscribe;
-};
-
-export type ToolsClientSchema = {
-  methods: ToolsMethods;
-};

package/src/types/store-augmentation.ts

@@ -1,11 +0,0 @@
-import "@assistant-ui/core/store";
-
-import type { ToolsClientSchema } from "./scopes/tools";
-import type { DataRenderersClientSchema } from "./scopes/dataRenderers";
-
-declare module "@assistant-ui/core/store" {
-  interface ScopeRegistry {
-    tools: ToolsClientSchema;
-    dataRenderers: DataRenderersClientSchema;
-  }
-}