@assistant-ui/react 0.1.5 → 0.1.6

package/dist/{Thread-DcmnmkmV.d.mts → Thread-ZUDFhMtm.d.mts}

@@ -1,5 +1,5 @@
+import { ReactNode, ComponentType } from 'react';
 import { z } from 'zod';
-import { ReactNode } from 'react';
 
 type TextContentPart = {
     type: "text";
@@ -42,6 +42,28 @@ type AppendMessage = {
 };
 type ThreadMessage = UserMessage | AssistantMessage;
 
+type ContentPartStatus = "done" | "in_progress" | "error";
+type TextContentPartProps = {
+    part: TextContentPart;
+    status: ContentPartStatus;
+};
+type TextContentPartComponent = ComponentType<TextContentPartProps>;
+type ImageContentPartProps = {
+    part: ImageContentPart;
+    status: ContentPartStatus;
+};
+type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
+type UIContentPartProps = {
+    part: UIContentPart;
+    status: ContentPartStatus;
+};
+type UIContentPartComponent = ComponentType<UIContentPartProps>;
+type ToolCallContentPartProps<TArgs = any, TResult = any> = {
+    part: ToolCallContentPart<TArgs, TResult>;
+    status: ContentPartStatus;
+};
+type ToolCallContentPartComponent<TArgs = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
 type Tool<TArgs = unknown, TResult = unknown> = {
     description?: string;
@@ -67,4 +89,4 @@ type ThreadState = {
     cancelRun: () => void;
 };
 
-export type { AssistantContentPart as A, ImageContentPart as I, ModelConfigProvider as M, TextContentPart as T, UIContentPart as U, ToolCallContentPart as a, ThreadState as b, Unsubscribe as c, ThreadMessage as d, ModelConfig as e, AppendMessage as f, AssistantMessage as g, UserMessage as h, UserContentPart as i, AppendContentPart as j, Tool as k };
+export type { AssistantContentPart as A, ImageContentPartComponent as I, ModelConfigProvider as M, TextContentPartComponent as T, UIContentPartComponent as U, ToolCallContentPartComponent as a, ToolCallContentPartProps as b, ThreadState as c, Unsubscribe as d, ThreadMessage as e, ModelConfig as f, AppendMessage as g, AssistantMessage as h, UserMessage as i, UserContentPart as j, AppendContentPart as k, TextContentPart as l, Tool as m, ImageContentPart as n, ToolCallContentPart as o, UIContentPart as p, TextContentPartProps as q, ImageContentPartProps as r, UIContentPartProps as s };

package/dist/{Thread-DcmnmkmV.d.ts → Thread-ZUDFhMtm.d.ts}

@@ -1,5 +1,5 @@
+import { ReactNode, ComponentType } from 'react';
 import { z } from 'zod';
-import { ReactNode } from 'react';
 
 type TextContentPart = {
     type: "text";
@@ -42,6 +42,28 @@ type AppendMessage = {
 };
 type ThreadMessage = UserMessage | AssistantMessage;
 
+type ContentPartStatus = "done" | "in_progress" | "error";
+type TextContentPartProps = {
+    part: TextContentPart;
+    status: ContentPartStatus;
+};
+type TextContentPartComponent = ComponentType<TextContentPartProps>;
+type ImageContentPartProps = {
+    part: ImageContentPart;
+    status: ContentPartStatus;
+};
+type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
+type UIContentPartProps = {
+    part: UIContentPart;
+    status: ContentPartStatus;
+};
+type UIContentPartComponent = ComponentType<UIContentPartProps>;
+type ToolCallContentPartProps<TArgs = any, TResult = any> = {
+    part: ToolCallContentPart<TArgs, TResult>;
+    status: ContentPartStatus;
+};
+type ToolCallContentPartComponent<TArgs = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
 type Tool<TArgs = unknown, TResult = unknown> = {
     description?: string;
@@ -67,4 +89,4 @@ type ThreadState = {
     cancelRun: () => void;
 };
 
-export type { AssistantContentPart as A, ImageContentPart as I, ModelConfigProvider as M, TextContentPart as T, UIContentPart as U, ToolCallContentPart as a, ThreadState as b, Unsubscribe as c, ThreadMessage as d, ModelConfig as e, AppendMessage as f, AssistantMessage as g, UserMessage as h, UserContentPart as i, AppendContentPart as j, Tool as k };
+export type { AssistantContentPart as A, ImageContentPartComponent as I, ModelConfigProvider as M, TextContentPartComponent as T, UIContentPartComponent as U, ToolCallContentPartComponent as a, ToolCallContentPartProps as b, ThreadState as c, Unsubscribe as d, ThreadMessage as e, ModelConfig as f, AppendMessage as g, AssistantMessage as h, UserMessage as i, UserContentPart as j, AppendContentPart as k, TextContentPart as l, Tool as m, ImageContentPart as n, ToolCallContentPart as o, UIContentPart as p, TextContentPartProps as q, ImageContentPartProps as r, UIContentPartProps as s };

package/dist/experimental.d.mts

@@ -1,7 +1,7 @@
-import { M as ModelConfigProvider, d as ThreadMessage, c as Unsubscribe, k as Tool, a as ToolCallContentPart, b as ThreadState } from './Thread-DcmnmkmV.mjs';
-export { I as ImageContentPart, e as ModelConfig, U as UIContentPart } from './Thread-DcmnmkmV.mjs';
+import { M as ModelConfigProvider, e as ThreadMessage, d as Unsubscribe, a as ToolCallContentPartComponent, c as ThreadState, m as Tool } from './Thread-ZUDFhMtm.mjs';
+export { n as ImageContentPart, I as ImageContentPartComponent, r as ImageContentPartProps, f as ModelConfig, T as TextContentPartComponent, q as TextContentPartProps, o as ToolCallContentPart, b as ToolCallContentPartProps, p as UIContentPart, U as UIContentPartComponent, s as UIContentPartProps } from './Thread-ZUDFhMtm.mjs';
+import { ReactNode } from 'react';
 import { UseBoundStore, StoreApi } from 'zustand';
-import { ReactNode, ComponentType } from 'react';
 import 'zod';
 
 type AssistantModelConfigState = {
@@ -51,28 +51,9 @@ type ThreadViewportState = {
     onScrollToBottom: (callback: () => void) => Unsubscribe;
 };
 
-declare const useAssistantInstructions: (instruction: string) => void;
-
-type UseAssistantTool<TArgs, TResult> = Tool<TArgs, TResult> & {
-    name: string;
-    render?: ToolRenderComponent<TArgs, TResult>;
-};
-declare const useAssistantTool: <TArgs, TResult>(tool: UseAssistantTool<TArgs, TResult>) => void;
-
-type UseAssistantToolRenderer<TArgs, TResult> = {
-    name: string;
-    render: ToolRenderComponent<TArgs, TResult>;
-};
-declare const useAssistantToolRenderer: (tool: UseAssistantToolRenderer<any, any> | null) => void;
-
-type ToolRenderComponent<TArgs, TResult> = ComponentType<{
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: "done" | "in_progress" | "error";
-}>;
-
 type AssistantToolRenderersState = {
-    getToolRenderer: (name: string) => ToolRenderComponent<any, any> | null;
-    setToolRenderer: (name: string, render: ToolRenderComponent<any, any>) => () => void;
+    getToolRenderer: (name: string) => ToolCallContentPartComponent | null;
+    setToolRenderer: (name: string, render: ToolCallContentPartComponent) => () => void;
 };
 
 type AssistantContextValue = {
@@ -105,4 +86,22 @@ type ContentPartContextValue = {
 };
 declare const useContentPartContext: () => ContentPartContextValue;
 
-export { type AssistantContextValue, type AssistantModelConfigState, type ComposerContextValue, type ComposerState, type ContentPartContextValue, type ContentPartState, type EditComposerState, type MessageContextValue, type MessageState, ModelConfigProvider, type ThreadContextValue, ThreadState, type ThreadViewportState, ToolCallContentPart, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolRenderer, useComposerContext, useContentPartContext, useMessageContext, useThreadContext };
+declare const useAssistantInstructions: (instruction: string) => void;
+
+type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {
+    name: string;
+    render?: ToolCallContentPartComponent<TArgs, TResult>;
+};
+declare const useAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => void;
+
+type AssistantToolRendererProps<TArgs, TResult> = {
+    name: string;
+    render: ToolCallContentPartComponent<TArgs, TResult>;
+};
+declare const useAssistantToolRenderer: (tool: AssistantToolRendererProps<any, any> | null) => void;
+
+declare const makeTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => () => null;
+
+declare const makeToolRenderer: <TArgs, TResult>(tool: AssistantToolRendererProps<TArgs, TResult>) => () => null;
+
+export { type AssistantContextValue, type AssistantModelConfigState, type AssistantToolProps, type AssistantToolRendererProps, type ComposerContextValue, type ComposerState, type ContentPartContextValue, type ContentPartState, type EditComposerState, type MessageContextValue, type MessageState, ModelConfigProvider, type ThreadContextValue, ThreadState, type ThreadViewportState, ToolCallContentPartComponent, makeTool, makeToolRenderer, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolRenderer, useComposerContext, useContentPartContext, useMessageContext, useThreadContext };

package/dist/experimental.d.ts

@@ -1,7 +1,7 @@
-import { M as ModelConfigProvider, d as ThreadMessage, c as Unsubscribe, k as Tool, a as ToolCallContentPart, b as ThreadState } from './Thread-DcmnmkmV.js';
-export { I as ImageContentPart, e as ModelConfig, U as UIContentPart } from './Thread-DcmnmkmV.js';
+import { M as ModelConfigProvider, e as ThreadMessage, d as Unsubscribe, a as ToolCallContentPartComponent, c as ThreadState, m as Tool } from './Thread-ZUDFhMtm.js';
+export { n as ImageContentPart, I as ImageContentPartComponent, r as ImageContentPartProps, f as ModelConfig, T as TextContentPartComponent, q as TextContentPartProps, o as ToolCallContentPart, b as ToolCallContentPartProps, p as UIContentPart, U as UIContentPartComponent, s as UIContentPartProps } from './Thread-ZUDFhMtm.js';
+import { ReactNode } from 'react';
 import { UseBoundStore, StoreApi } from 'zustand';
-import { ReactNode, ComponentType } from 'react';
 import 'zod';
 
 type AssistantModelConfigState = {
@@ -51,28 +51,9 @@ type ThreadViewportState = {
     onScrollToBottom: (callback: () => void) => Unsubscribe;
 };
 
-declare const useAssistantInstructions: (instruction: string) => void;
-
-type UseAssistantTool<TArgs, TResult> = Tool<TArgs, TResult> & {
-    name: string;
-    render?: ToolRenderComponent<TArgs, TResult>;
-};
-declare const useAssistantTool: <TArgs, TResult>(tool: UseAssistantTool<TArgs, TResult>) => void;
-
-type UseAssistantToolRenderer<TArgs, TResult> = {
-    name: string;
-    render: ToolRenderComponent<TArgs, TResult>;
-};
-declare const useAssistantToolRenderer: (tool: UseAssistantToolRenderer<any, any> | null) => void;
-
-type ToolRenderComponent<TArgs, TResult> = ComponentType<{
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: "done" | "in_progress" | "error";
-}>;
-
 type AssistantToolRenderersState = {
-    getToolRenderer: (name: string) => ToolRenderComponent<any, any> | null;
-    setToolRenderer: (name: string, render: ToolRenderComponent<any, any>) => () => void;
+    getToolRenderer: (name: string) => ToolCallContentPartComponent | null;
+    setToolRenderer: (name: string, render: ToolCallContentPartComponent) => () => void;
 };
 
 type AssistantContextValue = {
@@ -105,4 +86,22 @@ type ContentPartContextValue = {
 };
 declare const useContentPartContext: () => ContentPartContextValue;
 
-export { type AssistantContextValue, type AssistantModelConfigState, type ComposerContextValue, type ComposerState, type ContentPartContextValue, type ContentPartState, type EditComposerState, type MessageContextValue, type MessageState, ModelConfigProvider, type ThreadContextValue, ThreadState, type ThreadViewportState, ToolCallContentPart, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolRenderer, useComposerContext, useContentPartContext, useMessageContext, useThreadContext };
+declare const useAssistantInstructions: (instruction: string) => void;
+
+type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {
+    name: string;
+    render?: ToolCallContentPartComponent<TArgs, TResult>;
+};
+declare const useAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => void;
+
+type AssistantToolRendererProps<TArgs, TResult> = {
+    name: string;
+    render: ToolCallContentPartComponent<TArgs, TResult>;
+};
+declare const useAssistantToolRenderer: (tool: AssistantToolRendererProps<any, any> | null) => void;
+
+declare const makeTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => () => null;
+
+declare const makeToolRenderer: <TArgs, TResult>(tool: AssistantToolRendererProps<TArgs, TResult>) => () => null;
+
+export { type AssistantContextValue, type AssistantModelConfigState, type AssistantToolProps, type AssistantToolRendererProps, type ComposerContextValue, type ComposerState, type ContentPartContextValue, type ContentPartState, type EditComposerState, type MessageContextValue, type MessageState, ModelConfigProvider, type ThreadContextValue, ThreadState, type ThreadViewportState, ToolCallContentPartComponent, makeTool, makeToolRenderer, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolRenderer, useComposerContext, useContentPartContext, useMessageContext, useThreadContext };

package/dist/experimental.js

@@ -20,6 +20,8 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/experimental.ts
 var experimental_exports = {};
 __export(experimental_exports, {
+  makeTool: () => makeTool,
+  makeToolRenderer: () => makeToolRenderer,
   useAssistantContext: () => useAssistantContext,
   useAssistantInstructions: () => useAssistantInstructions,
   useAssistantTool: () => useAssistantTool,
@@ -147,8 +149,28 @@ var useAssistantToolRenderer = (tool) => {
     return setToolRenderer(name, render);
   }, [setToolRenderer, tool]);
 };
+
+// src/model-config/makeTool.tsx
+var makeTool = (tool) => {
+  const Tool = () => {
+    useAssistantTool(tool);
+    return null;
+  };
+  return Tool;
+};
+
+// src/model-config/makeToolRenderer.tsx
+var makeToolRenderer = (tool) => {
+  const ToolRenderer = () => {
+    useAssistantToolRenderer(tool);
+    return null;
+  };
+  return ToolRenderer;
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  makeTool,
+  makeToolRenderer,
   useAssistantContext,
   useAssistantInstructions,
   useAssistantTool,

package/dist/experimental.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/experimental.ts","../src/context/AssistantContext.ts","../src/context/ThreadContext.ts","../src/context/ComposerContext.ts","../src/context/MessageContext.ts","../src/context/ContentPartContext.ts","../src/model-config/useAssistantInstructions.tsx","../src/model-config/useAssistantTool.tsx","../src/model-config/useAssistantToolRenderer.tsx"],"sourcesContent":["export type {\n  ImageContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n} from \"./utils/AssistantTypes\";\n\nexport type {\n  ModelConfigProvider,\n  ModelConfig,\n} from \"./utils/ModelConfigTypes\";\n\nexport * from \"./context\";\nexport { useAssistantInstructions } from \"./model-config/useAssistantInstructions\";\nexport { useAssistantTool } from \"./model-config/useAssistantTool\";\nexport { useAssistantToolRenderer } from \"./model-config/useAssistantToolRenderer\";\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { AssistantModelConfigState } from \"./stores/AssistantModelConfig\";\nimport type { AssistantToolRenderersState } from \"./stores/AssistantToolRenderers\";\n\nexport type AssistantContextValue = {\n  useModelConfig: UseBoundStore<StoreApi<AssistantModelConfigState>>;\n  useToolRenderers: UseBoundStore<StoreApi<AssistantToolRenderersState>>;\n};\n\nexport const AssistantContext = createContext<AssistantContextValue | null>(\n  null,\n);\n\nexport const useAssistantContext = (): AssistantContextValue => {\n  const context = useContext(AssistantContext);\n  if (!context)\n    throw new Error(\n      \"This component must be used within an AssistantRuntimeProvider.\",\n    );\n  return context;\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { ComposerState } from \"./stores/Composer\";\nimport type { ThreadState } from \"./stores/Thread\";\nimport type { ThreadViewportState } from \"./stores/ThreadViewport\";\n\nexport type ThreadContextValue = {\n  useThread: UseBoundStore<StoreApi<ThreadState>>;\n  useComposer: UseBoundStore<StoreApi<ComposerState>>;\n  useViewport: UseBoundStore<StoreApi<ThreadViewportState>>;\n};\n\nexport const ThreadContext = createContext<ThreadContextValue | null>(null);\n\nexport const useThreadContext = (): ThreadContextValue => {\n  const context = useContext(ThreadContext);\n  if (!context)\n    throw new Error(\"This component must be used within an AssistantRuntimeProvider.\");\n  return context;\n};\n","import { useContext, useMemo } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport { MessageContext } from \"./MessageContext\";\nimport { useThreadContext } from \"./ThreadContext\";\nimport type { ComposerState } from \"./stores/Composer\";\nimport type { EditComposerState } from \"./stores/MessageComposer\";\n\nexport type ComposerContextValue = {\n  useComposer: UseBoundStore<StoreApi<EditComposerState | ComposerState>>;\n  type: \"edit\" | \"new\";\n};\n\nexport const useComposerContext = (): ComposerContextValue => {\n  const { useComposer } = useThreadContext();\n  const { useComposer: useEditComposer } = useContext(MessageContext) ?? {};\n  return useMemo(\n    () => ({\n      useComposer: (useEditComposer ?? useComposer) as UseBoundStore<\n        StoreApi<EditComposerState | ComposerState>\n      >,\n      type: useEditComposer ? (\"edit\" as const) : (\"new\" as const),\n    }),\n    [useEditComposer, useComposer],\n  );\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { MessageState } from \"./stores/Message\";\nimport type { EditComposerState } from \"./stores/MessageComposer\";\n\nexport type MessageContextValue = {\n  useMessage: UseBoundStore<StoreApi<MessageState>>;\n  useComposer: UseBoundStore<StoreApi<EditComposerState>>;\n};\n\nexport const MessageContext = createContext<MessageContextValue | null>(null);\n\nexport const useMessageContext = () => {\n  const context = useContext(MessageContext);\n  if (!context)\n    throw new Error(\n      \"This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />.\",\n    );\n  return context;\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { ContentPartState } from \"./stores/ContentPart\";\n\nexport type ContentPartContextValue = {\n  useContentPart: UseBoundStore<StoreApi<ContentPartState>>;\n};\n\nexport const ContentPartContext = createContext<ContentPartContextValue | null>(\n  null,\n);\n\nexport const useContentPartContext = (): ContentPartContextValue => {\n  const context = useContext(ContentPartContext);\n  if (!context)\n    throw new Error(\n      \"This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >.\",\n    );\n  return context;\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\n\nexport const useAssistantInstructions = (instruction: string) => {\n  const { useModelConfig } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  useEffect(() => {\n    const config = {\n      system: instruction,\n    };\n    return registerModelConfigProvider(() => config);\n  }, [registerModelConfigProvider, instruction]);\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { Tool } from \"../utils/ModelConfigTypes\";\nimport type { ToolRenderComponent } from \"./ToolRenderComponent\";\n\nexport type UseAssistantTool<TArgs, TResult> = Tool<TArgs, TResult> & {\n  name: string;\n  render?: ToolRenderComponent<TArgs, TResult>;\n};\n\nexport const useAssistantTool = <TArgs, TResult>(\n  tool: UseAssistantTool<TArgs, TResult>,\n) => {\n  const { useModelConfig, useToolRenderers } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    const { name, render, ...rest } = tool;\n    const config = {\n      tools: {\n        [tool.name]: rest,\n      },\n    };\n    const unsub1 = registerModelConfigProvider(() => config);\n    const unsub2 = render ? setToolRenderer(name, render) : undefined;\n    return () => {\n      unsub1();\n      unsub2?.();\n    };\n  }, [registerModelConfigProvider, setToolRenderer, tool]);\n};\n","\"use client\";\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { ToolRenderComponent } from \"./ToolRenderComponent\";\n\ntype UseAssistantToolRenderer<TArgs, TResult> = {\n  name: string;\n  render: ToolRenderComponent<TArgs, TResult>;\n};\n\nexport const useAssistantToolRenderer = (\n  // biome-ignore lint/suspicious/noExplicitAny: intentional any\n  tool: UseAssistantToolRenderer<any, any> | null,\n) => {\n  const { useToolRenderers } = useAssistantContext();\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    if (!tool) return;\n    const { name, render } = tool;\n    return setToolRenderer(name, render);\n  }, [setToolRenderer, tool]);\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,mBAA0C;AAUnC,IAAM,uBAAmB;AAAA,EAC9B;AACF;AAEO,IAAM,sBAAsB,MAA6B;AAC9D,QAAM,cAAU,yBAAW,gBAAgB;AAC3C,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ACrBA,IAAAA,gBAA0C;AAYnC,IAAM,oBAAgB,6BAAyC,IAAI;AAEnE,IAAM,mBAAmB,MAA0B;AACxD,QAAM,cAAU,0BAAW,aAAa;AACxC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,iEAAiE;AACnF,SAAO;AACT;;;ACnBA,IAAAC,gBAAoC;;;ACApC,IAAAC,gBAA0C;AAUnC,IAAM,qBAAiB,6BAA0C,IAAI;AAErE,IAAM,oBAAoB,MAAM;AACrC,QAAM,cAAU,0BAAW,cAAc;AACzC,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ADPO,IAAM,qBAAqB,MAA4B;AAC5D,QAAM,EAAE,YAAY,IAAI,iBAAiB;AACzC,QAAM,EAAE,aAAa,gBAAgB,QAAI,0BAAW,cAAc,KAAK,CAAC;AACxE,aAAO;AAAA,IACL,OAAO;AAAA,MACL,aAAc,mBAAmB;AAAA,MAGjC,MAAM,kBAAmB,SAAoB;AAAA,IAC/C;AAAA,IACA,CAAC,iBAAiB,WAAW;AAAA,EAC/B;AACF;;;AExBA,IAAAC,gBAA0C;AAQnC,IAAM,yBAAqB;AAAA,EAChC;AACF;AAEO,IAAM,wBAAwB,MAA+B;AAClE,QAAM,cAAU,0BAAW,kBAAkB;AAC7C,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ACjBA,IAAAC,gBAA0B;AAGnB,IAAM,2BAA2B,CAAC,gBAAwB;AAC/D,QAAM,EAAE,eAAe,IAAI,oBAAoB;AAC/C,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,+BAAU,MAAM;AACd,UAAM,SAAS;AAAA,MACb,QAAQ;AAAA,IACV;AACA,WAAO,4BAA4B,MAAM,MAAM;AAAA,EACjD,GAAG,CAAC,6BAA6B,WAAW,CAAC;AAC/C;;;ACdA,IAAAC,gBAA0B;AAUnB,IAAM,mBAAmB,CAC9B,SACG;AACH,QAAM,EAAE,gBAAgB,iBAAiB,IAAI,oBAAoB;AACjE,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,+BAAU,MAAM;AACd,UAAM,EAAE,MAAM,QAAQ,GAAG,KAAK,IAAI;AAClC,UAAM,SAAS;AAAA,MACb,OAAO;AAAA,QACL,CAAC,KAAK,IAAI,GAAG;AAAA,MACf;AAAA,IACF;AACA,UAAM,SAAS,4BAA4B,MAAM,MAAM;AACvD,UAAM,SAAS,SAAS,gBAAgB,MAAM,MAAM,IAAI;AACxD,WAAO,MAAM;AACX,aAAO;AACP,eAAS;AAAA,IACX;AAAA,EACF,GAAG,CAAC,6BAA6B,iBAAiB,IAAI,CAAC;AACzD;;;ACjCA,IAAAC,gBAA0B;AASnB,IAAM,2BAA2B,CAEtC,SACG;AACH,QAAM,EAAE,iBAAiB,IAAI,oBAAoB;AACjD,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,+BAAU,MAAM;AACd,QAAI,CAAC,KAAM;AACX,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,WAAO,gBAAgB,MAAM,MAAM;AAAA,EACrC,GAAG,CAAC,iBAAiB,IAAI,CAAC;AAC5B;","names":["import_react","import_react","import_react","import_react","import_react","import_react","import_react"]}
+{"version":3,"sources":["../src/experimental.ts","../src/context/AssistantContext.ts","../src/context/ThreadContext.ts","../src/context/ComposerContext.ts","../src/context/MessageContext.ts","../src/context/ContentPartContext.ts","../src/model-config/useAssistantInstructions.tsx","../src/model-config/useAssistantTool.tsx","../src/model-config/useAssistantToolRenderer.tsx","../src/model-config/makeTool.tsx","../src/model-config/makeToolRenderer.tsx"],"sourcesContent":["export type {\n  ImageContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n} from \"./utils/AssistantTypes\";\n\nexport type {\n  ModelConfigProvider,\n  ModelConfig,\n} from \"./utils/ModelConfigTypes\";\n\nexport type {\n  TextContentPartProps,\n  TextContentPartComponent,\n  ImageContentPartProps,\n  ImageContentPartComponent,\n  UIContentPartProps,\n  UIContentPartComponent,\n  ToolCallContentPartProps,\n  ToolCallContentPartComponent,\n} from \"./primitives/message/ContentPartComponentTypes\";\n\nexport * from \"./context\";\nexport { useAssistantInstructions } from \"./model-config/useAssistantInstructions\";\nexport {\n  useAssistantTool,\n  type AssistantToolProps,\n} from \"./model-config/useAssistantTool\";\nexport {\n  useAssistantToolRenderer,\n  type AssistantToolRendererProps,\n} from \"./model-config/useAssistantToolRenderer\";\nexport { makeTool } from \"./model-config/makeTool\";\nexport { makeToolRenderer } from \"./model-config/makeToolRenderer\";\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { AssistantModelConfigState } from \"./stores/AssistantModelConfig\";\nimport type { AssistantToolRenderersState } from \"./stores/AssistantToolRenderers\";\n\nexport type AssistantContextValue = {\n  useModelConfig: UseBoundStore<StoreApi<AssistantModelConfigState>>;\n  useToolRenderers: UseBoundStore<StoreApi<AssistantToolRenderersState>>;\n};\n\nexport const AssistantContext = createContext<AssistantContextValue | null>(\n  null,\n);\n\nexport const useAssistantContext = (): AssistantContextValue => {\n  const context = useContext(AssistantContext);\n  if (!context)\n    throw new Error(\n      \"This component must be used within an AssistantRuntimeProvider.\",\n    );\n  return context;\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { ComposerState } from \"./stores/Composer\";\nimport type { ThreadState } from \"./stores/Thread\";\nimport type { ThreadViewportState } from \"./stores/ThreadViewport\";\n\nexport type ThreadContextValue = {\n  useThread: UseBoundStore<StoreApi<ThreadState>>;\n  useComposer: UseBoundStore<StoreApi<ComposerState>>;\n  useViewport: UseBoundStore<StoreApi<ThreadViewportState>>;\n};\n\nexport const ThreadContext = createContext<ThreadContextValue | null>(null);\n\nexport const useThreadContext = (): ThreadContextValue => {\n  const context = useContext(ThreadContext);\n  if (!context)\n    throw new Error(\"This component must be used within an AssistantRuntimeProvider.\");\n  return context;\n};\n","import { useContext, useMemo } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport { MessageContext } from \"./MessageContext\";\nimport { useThreadContext } from \"./ThreadContext\";\nimport type { ComposerState } from \"./stores/Composer\";\nimport type { EditComposerState } from \"./stores/MessageComposer\";\n\nexport type ComposerContextValue = {\n  useComposer: UseBoundStore<StoreApi<EditComposerState | ComposerState>>;\n  type: \"edit\" | \"new\";\n};\n\nexport const useComposerContext = (): ComposerContextValue => {\n  const { useComposer } = useThreadContext();\n  const { useComposer: useEditComposer } = useContext(MessageContext) ?? {};\n  return useMemo(\n    () => ({\n      useComposer: (useEditComposer ?? useComposer) as UseBoundStore<\n        StoreApi<EditComposerState | ComposerState>\n      >,\n      type: useEditComposer ? (\"edit\" as const) : (\"new\" as const),\n    }),\n    [useEditComposer, useComposer],\n  );\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { MessageState } from \"./stores/Message\";\nimport type { EditComposerState } from \"./stores/MessageComposer\";\n\nexport type MessageContextValue = {\n  useMessage: UseBoundStore<StoreApi<MessageState>>;\n  useComposer: UseBoundStore<StoreApi<EditComposerState>>;\n};\n\nexport const MessageContext = createContext<MessageContextValue | null>(null);\n\nexport const useMessageContext = () => {\n  const context = useContext(MessageContext);\n  if (!context)\n    throw new Error(\n      \"This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />.\",\n    );\n  return context;\n};\n","import { createContext, useContext } from \"react\";\nimport type { StoreApi, UseBoundStore } from \"zustand\";\nimport type { ContentPartState } from \"./stores/ContentPart\";\n\nexport type ContentPartContextValue = {\n  useContentPart: UseBoundStore<StoreApi<ContentPartState>>;\n};\n\nexport const ContentPartContext = createContext<ContentPartContextValue | null>(\n  null,\n);\n\nexport const useContentPartContext = (): ContentPartContextValue => {\n  const context = useContext(ContentPartContext);\n  if (!context)\n    throw new Error(\n      \"This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >.\",\n    );\n  return context;\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\n\nexport const useAssistantInstructions = (instruction: string) => {\n  const { useModelConfig } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  useEffect(() => {\n    const config = {\n      system: instruction,\n    };\n    return registerModelConfigProvider(() => config);\n  }, [registerModelConfigProvider, instruction]);\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { ToolCallContentPartComponent } from \"../primitives/message/ContentPartComponentTypes\";\nimport type { Tool } from \"../utils/ModelConfigTypes\";\n\nexport type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {\n  name: string;\n  render?: ToolCallContentPartComponent<TArgs, TResult>;\n};\n\nexport const useAssistantTool = <TArgs, TResult>(\n  tool: AssistantToolProps<TArgs, TResult>,\n) => {\n  const { useModelConfig, useToolRenderers } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    const { name, render, ...rest } = tool;\n    const config = {\n      tools: {\n        [tool.name]: rest,\n      },\n    };\n    const unsub1 = registerModelConfigProvider(() => config);\n    const unsub2 = render ? setToolRenderer(name, render) : undefined;\n    return () => {\n      unsub1();\n      unsub2?.();\n    };\n  }, [registerModelConfigProvider, setToolRenderer, tool]);\n};\n","\"use client\";\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { ToolCallContentPartComponent } from \"../primitives/message/ContentPartComponentTypes\";\n\nexport type AssistantToolRendererProps<TArgs, TResult> = {\n  name: string;\n  render: ToolCallContentPartComponent<TArgs, TResult>;\n};\n\nexport const useAssistantToolRenderer = (\n  // biome-ignore lint/suspicious/noExplicitAny: intentional any\n  tool: AssistantToolRendererProps<any, any> | null,\n) => {\n  const { useToolRenderers } = useAssistantContext();\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    if (!tool) return;\n    const { name, render } = tool;\n    return setToolRenderer(name, render);\n  }, [setToolRenderer, tool]);\n};\n","\"use client\";\nimport { type AssistantToolProps, useAssistantTool } from \"./useAssistantTool\";\n\nexport const makeTool = <TArgs, TResult>(\n  tool: AssistantToolProps<TArgs, TResult>,\n) => {\n  const Tool = () => {\n    useAssistantTool(tool);\n    return null;\n  };\n  return Tool;\n};\n","\"use client\";\nimport {\n  type AssistantToolRendererProps,\n  useAssistantToolRenderer,\n} from \"./useAssistantToolRenderer\";\n\nexport const makeToolRenderer = <TArgs, TResult>(\n  tool: AssistantToolRendererProps<TArgs, TResult>,\n) => {\n  const ToolRenderer = () => {\n    useAssistantToolRenderer(tool);\n    return null;\n  };\n  return ToolRenderer;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,mBAA0C;AAUnC,IAAM,uBAAmB;AAAA,EAC9B;AACF;AAEO,IAAM,sBAAsB,MAA6B;AAC9D,QAAM,cAAU,yBAAW,gBAAgB;AAC3C,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ACrBA,IAAAA,gBAA0C;AAYnC,IAAM,oBAAgB,6BAAyC,IAAI;AAEnE,IAAM,mBAAmB,MAA0B;AACxD,QAAM,cAAU,0BAAW,aAAa;AACxC,MAAI,CAAC;AACH,UAAM,IAAI,MAAM,iEAAiE;AACnF,SAAO;AACT;;;ACnBA,IAAAC,gBAAoC;;;ACApC,IAAAC,gBAA0C;AAUnC,IAAM,qBAAiB,6BAA0C,IAAI;AAErE,IAAM,oBAAoB,MAAM;AACrC,QAAM,cAAU,0BAAW,cAAc;AACzC,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ADPO,IAAM,qBAAqB,MAA4B;AAC5D,QAAM,EAAE,YAAY,IAAI,iBAAiB;AACzC,QAAM,EAAE,aAAa,gBAAgB,QAAI,0BAAW,cAAc,KAAK,CAAC;AACxE,aAAO;AAAA,IACL,OAAO;AAAA,MACL,aAAc,mBAAmB;AAAA,MAGjC,MAAM,kBAAmB,SAAoB;AAAA,IAC/C;AAAA,IACA,CAAC,iBAAiB,WAAW;AAAA,EAC/B;AACF;;;AExBA,IAAAC,gBAA0C;AAQnC,IAAM,yBAAqB;AAAA,EAChC;AACF;AAEO,IAAM,wBAAwB,MAA+B;AAClE,QAAM,cAAU,0BAAW,kBAAkB;AAC7C,MAAI,CAAC;AACH,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;;;ACjBA,IAAAC,gBAA0B;AAGnB,IAAM,2BAA2B,CAAC,gBAAwB;AAC/D,QAAM,EAAE,eAAe,IAAI,oBAAoB;AAC/C,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,+BAAU,MAAM;AACd,UAAM,SAAS;AAAA,MACb,QAAQ;AAAA,IACV;AACA,WAAO,4BAA4B,MAAM,MAAM;AAAA,EACjD,GAAG,CAAC,6BAA6B,WAAW,CAAC;AAC/C;;;ACdA,IAAAC,gBAA0B;AAUnB,IAAM,mBAAmB,CAC9B,SACG;AACH,QAAM,EAAE,gBAAgB,iBAAiB,IAAI,oBAAoB;AACjE,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,+BAAU,MAAM;AACd,UAAM,EAAE,MAAM,QAAQ,GAAG,KAAK,IAAI;AAClC,UAAM,SAAS;AAAA,MACb,OAAO;AAAA,QACL,CAAC,KAAK,IAAI,GAAG;AAAA,MACf;AAAA,IACF;AACA,UAAM,SAAS,4BAA4B,MAAM,MAAM;AACvD,UAAM,SAAS,SAAS,gBAAgB,MAAM,MAAM,IAAI;AACxD,WAAO,MAAM;AACX,aAAO;AACP,eAAS;AAAA,IACX;AAAA,EACF,GAAG,CAAC,6BAA6B,iBAAiB,IAAI,CAAC;AACzD;;;ACjCA,IAAAC,gBAA0B;AASnB,IAAM,2BAA2B,CAEtC,SACG;AACH,QAAM,EAAE,iBAAiB,IAAI,oBAAoB;AACjD,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,+BAAU,MAAM;AACd,QAAI,CAAC,KAAM;AACX,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,WAAO,gBAAgB,MAAM,MAAM;AAAA,EACrC,GAAG,CAAC,iBAAiB,IAAI,CAAC;AAC5B;;;AClBO,IAAM,WAAW,CACtB,SACG;AACH,QAAM,OAAO,MAAM;AACjB,qBAAiB,IAAI;AACrB,WAAO;AAAA,EACT;AACA,SAAO;AACT;;;ACLO,IAAM,mBAAmB,CAC9B,SACG;AACH,QAAM,eAAe,MAAM;AACzB,6BAAyB,IAAI;AAC7B,WAAO;AAAA,EACT;AACA,SAAO;AACT;","names":["import_react","import_react","import_react","import_react","import_react","import_react","import_react"]}

package/dist/experimental.mjs

@@ -56,7 +56,27 @@ var useAssistantToolRenderer = (tool) => {
     return setToolRenderer(name, render);
   }, [setToolRenderer, tool]);
 };
+
+// src/model-config/makeTool.tsx
+var makeTool = (tool) => {
+  const Tool = () => {
+    useAssistantTool(tool);
+    return null;
+  };
+  return Tool;
+};
+
+// src/model-config/makeToolRenderer.tsx
+var makeToolRenderer = (tool) => {
+  const ToolRenderer = () => {
+    useAssistantToolRenderer(tool);
+    return null;
+  };
+  return ToolRenderer;
+};
 export {
+  makeTool,
+  makeToolRenderer,
   useAssistantContext,
   useAssistantInstructions,
   useAssistantTool,

package/dist/experimental.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/model-config/useAssistantInstructions.tsx","../src/model-config/useAssistantTool.tsx","../src/model-config/useAssistantToolRenderer.tsx"],"sourcesContent":["\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\n\nexport const useAssistantInstructions = (instruction: string) => {\n  const { useModelConfig } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  useEffect(() => {\n    const config = {\n      system: instruction,\n    };\n    return registerModelConfigProvider(() => config);\n  }, [registerModelConfigProvider, instruction]);\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { Tool } from \"../utils/ModelConfigTypes\";\nimport type { ToolRenderComponent } from \"./ToolRenderComponent\";\n\nexport type UseAssistantTool<TArgs, TResult> = Tool<TArgs, TResult> & {\n  name: string;\n  render?: ToolRenderComponent<TArgs, TResult>;\n};\n\nexport const useAssistantTool = <TArgs, TResult>(\n  tool: UseAssistantTool<TArgs, TResult>,\n) => {\n  const { useModelConfig, useToolRenderers } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    const { name, render, ...rest } = tool;\n    const config = {\n      tools: {\n        [tool.name]: rest,\n      },\n    };\n    const unsub1 = registerModelConfigProvider(() => config);\n    const unsub2 = render ? setToolRenderer(name, render) : undefined;\n    return () => {\n      unsub1();\n      unsub2?.();\n    };\n  }, [registerModelConfigProvider, setToolRenderer, tool]);\n};\n","\"use client\";\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { ToolRenderComponent } from \"./ToolRenderComponent\";\n\ntype UseAssistantToolRenderer<TArgs, TResult> = {\n  name: string;\n  render: ToolRenderComponent<TArgs, TResult>;\n};\n\nexport const useAssistantToolRenderer = (\n  // biome-ignore lint/suspicious/noExplicitAny: intentional any\n  tool: UseAssistantToolRenderer<any, any> | null,\n) => {\n  const { useToolRenderers } = useAssistantContext();\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    if (!tool) return;\n    const { name, render } = tool;\n    return setToolRenderer(name, render);\n  }, [setToolRenderer, tool]);\n};\n"],"mappings":";;;;;;;;;AAEA,SAAS,iBAAiB;AAGnB,IAAM,2BAA2B,CAAC,gBAAwB;AAC/D,QAAM,EAAE,eAAe,IAAI,oBAAoB;AAC/C,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,YAAU,MAAM;AACd,UAAM,SAAS;AAAA,MACb,QAAQ;AAAA,IACV;AACA,WAAO,4BAA4B,MAAM,MAAM;AAAA,EACjD,GAAG,CAAC,6BAA6B,WAAW,CAAC;AAC/C;;;ACdA,SAAS,aAAAA,kBAAiB;AAUnB,IAAM,mBAAmB,CAC9B,SACG;AACH,QAAM,EAAE,gBAAgB,iBAAiB,IAAI,oBAAoB;AACjE,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,EAAAC,WAAU,MAAM;AACd,UAAM,EAAE,MAAM,QAAQ,GAAG,KAAK,IAAI;AAClC,UAAM,SAAS;AAAA,MACb,OAAO;AAAA,QACL,CAAC,KAAK,IAAI,GAAG;AAAA,MACf;AAAA,IACF;AACA,UAAM,SAAS,4BAA4B,MAAM,MAAM;AACvD,UAAM,SAAS,SAAS,gBAAgB,MAAM,MAAM,IAAI;AACxD,WAAO,MAAM;AACX,aAAO;AACP,eAAS;AAAA,IACX;AAAA,EACF,GAAG,CAAC,6BAA6B,iBAAiB,IAAI,CAAC;AACzD;;;ACjCA,SAAS,aAAAC,kBAAiB;AASnB,IAAM,2BAA2B,CAEtC,SACG;AACH,QAAM,EAAE,iBAAiB,IAAI,oBAAoB;AACjD,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,EAAAC,WAAU,MAAM;AACd,QAAI,CAAC,KAAM;AACX,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,WAAO,gBAAgB,MAAM,MAAM;AAAA,EACrC,GAAG,CAAC,iBAAiB,IAAI,CAAC;AAC5B;","names":["useEffect","useEffect","useEffect","useEffect"]}
+{"version":3,"sources":["../src/model-config/useAssistantInstructions.tsx","../src/model-config/useAssistantTool.tsx","../src/model-config/useAssistantToolRenderer.tsx","../src/model-config/makeTool.tsx","../src/model-config/makeToolRenderer.tsx"],"sourcesContent":["\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\n\nexport const useAssistantInstructions = (instruction: string) => {\n  const { useModelConfig } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  useEffect(() => {\n    const config = {\n      system: instruction,\n    };\n    return registerModelConfigProvider(() => config);\n  }, [registerModelConfigProvider, instruction]);\n};\n","\"use client\";\n\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { ToolCallContentPartComponent } from \"../primitives/message/ContentPartComponentTypes\";\nimport type { Tool } from \"../utils/ModelConfigTypes\";\n\nexport type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {\n  name: string;\n  render?: ToolCallContentPartComponent<TArgs, TResult>;\n};\n\nexport const useAssistantTool = <TArgs, TResult>(\n  tool: AssistantToolProps<TArgs, TResult>,\n) => {\n  const { useModelConfig, useToolRenderers } = useAssistantContext();\n  const registerModelConfigProvider = useModelConfig(\n    (s) => s.registerModelConfigProvider,\n  );\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    const { name, render, ...rest } = tool;\n    const config = {\n      tools: {\n        [tool.name]: rest,\n      },\n    };\n    const unsub1 = registerModelConfigProvider(() => config);\n    const unsub2 = render ? setToolRenderer(name, render) : undefined;\n    return () => {\n      unsub1();\n      unsub2?.();\n    };\n  }, [registerModelConfigProvider, setToolRenderer, tool]);\n};\n","\"use client\";\nimport { useEffect } from \"react\";\nimport { useAssistantContext } from \"../context/AssistantContext\";\nimport type { ToolCallContentPartComponent } from \"../primitives/message/ContentPartComponentTypes\";\n\nexport type AssistantToolRendererProps<TArgs, TResult> = {\n  name: string;\n  render: ToolCallContentPartComponent<TArgs, TResult>;\n};\n\nexport const useAssistantToolRenderer = (\n  // biome-ignore lint/suspicious/noExplicitAny: intentional any\n  tool: AssistantToolRendererProps<any, any> | null,\n) => {\n  const { useToolRenderers } = useAssistantContext();\n  const setToolRenderer = useToolRenderers((s) => s.setToolRenderer);\n  useEffect(() => {\n    if (!tool) return;\n    const { name, render } = tool;\n    return setToolRenderer(name, render);\n  }, [setToolRenderer, tool]);\n};\n","\"use client\";\nimport { type AssistantToolProps, useAssistantTool } from \"./useAssistantTool\";\n\nexport const makeTool = <TArgs, TResult>(\n  tool: AssistantToolProps<TArgs, TResult>,\n) => {\n  const Tool = () => {\n    useAssistantTool(tool);\n    return null;\n  };\n  return Tool;\n};\n","\"use client\";\nimport {\n  type AssistantToolRendererProps,\n  useAssistantToolRenderer,\n} from \"./useAssistantToolRenderer\";\n\nexport const makeToolRenderer = <TArgs, TResult>(\n  tool: AssistantToolRendererProps<TArgs, TResult>,\n) => {\n  const ToolRenderer = () => {\n    useAssistantToolRenderer(tool);\n    return null;\n  };\n  return ToolRenderer;\n};\n"],"mappings":";;;;;;;;;AAEA,SAAS,iBAAiB;AAGnB,IAAM,2BAA2B,CAAC,gBAAwB;AAC/D,QAAM,EAAE,eAAe,IAAI,oBAAoB;AAC/C,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,YAAU,MAAM;AACd,UAAM,SAAS;AAAA,MACb,QAAQ;AAAA,IACV;AACA,WAAO,4BAA4B,MAAM,MAAM;AAAA,EACjD,GAAG,CAAC,6BAA6B,WAAW,CAAC;AAC/C;;;ACdA,SAAS,aAAAA,kBAAiB;AAUnB,IAAM,mBAAmB,CAC9B,SACG;AACH,QAAM,EAAE,gBAAgB,iBAAiB,IAAI,oBAAoB;AACjE,QAAM,8BAA8B;AAAA,IAClC,CAAC,MAAM,EAAE;AAAA,EACX;AACA,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,EAAAC,WAAU,MAAM;AACd,UAAM,EAAE,MAAM,QAAQ,GAAG,KAAK,IAAI;AAClC,UAAM,SAAS;AAAA,MACb,OAAO;AAAA,QACL,CAAC,KAAK,IAAI,GAAG;AAAA,MACf;AAAA,IACF;AACA,UAAM,SAAS,4BAA4B,MAAM,MAAM;AACvD,UAAM,SAAS,SAAS,gBAAgB,MAAM,MAAM,IAAI;AACxD,WAAO,MAAM;AACX,aAAO;AACP,eAAS;AAAA,IACX;AAAA,EACF,GAAG,CAAC,6BAA6B,iBAAiB,IAAI,CAAC;AACzD;;;ACjCA,SAAS,aAAAC,kBAAiB;AASnB,IAAM,2BAA2B,CAEtC,SACG;AACH,QAAM,EAAE,iBAAiB,IAAI,oBAAoB;AACjD,QAAM,kBAAkB,iBAAiB,CAAC,MAAM,EAAE,eAAe;AACjE,EAAAC,WAAU,MAAM;AACd,QAAI,CAAC,KAAM;AACX,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,WAAO,gBAAgB,MAAM,MAAM;AAAA,EACrC,GAAG,CAAC,iBAAiB,IAAI,CAAC;AAC5B;;;AClBO,IAAM,WAAW,CACtB,SACG;AACH,QAAM,OAAO,MAAM;AACjB,qBAAiB,IAAI;AACrB,WAAO;AAAA,EACT;AACA,SAAO;AACT;;;ACLO,IAAM,mBAAmB,CAC9B,SACG;AACH,QAAM,eAAe,MAAM;AACzB,6BAAyB,IAAI;AAC7B,WAAO;AAAA,EACT;AACA,SAAO;AACT;","names":["useEffect","useEffect","useEffect","useEffect"]}

package/dist/index.d.mts

@@ -1,8 +1,8 @@
 import * as react from 'react';
 import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
-import { T as TextContentPart, I as ImageContentPart, U as UIContentPart, a as ToolCallContentPart, b as ThreadState, c as Unsubscribe, M as ModelConfigProvider, d as ThreadMessage, e as ModelConfig, A as AssistantContentPart, f as AppendMessage } from './Thread-DcmnmkmV.mjs';
-export { j as AppendContentPart, g as AssistantMessage, i as UserContentPart, h as UserMessage } from './Thread-DcmnmkmV.mjs';
+import { T as TextContentPartComponent, I as ImageContentPartComponent, U as UIContentPartComponent, a as ToolCallContentPartComponent, b as ToolCallContentPartProps, c as ThreadState, d as Unsubscribe, M as ModelConfigProvider, e as ThreadMessage, f as ModelConfig, A as AssistantContentPart, g as AppendMessage } from './Thread-ZUDFhMtm.mjs';
+export { k as AppendContentPart, h as AssistantMessage, l as TextContentPart, j as UserContentPart, i as UserMessage } from './Thread-ZUDFhMtm.mjs';
 import { ComponentPropsWithoutRef, Primitive } from '@radix-ui/react-primitive';
 import 'zod';
 
@@ -135,27 +135,12 @@ declare const MessageIf: FC<MessageIfProps>;
 
 type MessageContentProps = {
     components?: {
-        Text?: ComponentType<{
-            part: TextContentPart;
-            status: "done" | "in_progress" | "error";
-        }>;
-        Image?: ComponentType<{
-            part: ImageContentPart;
-            status: "done" | "in_progress" | "error";
-        }>;
-        UI?: ComponentType<{
-            part: UIContentPart;
-            status: "done" | "in_progress" | "error";
-        }>;
+        Text?: TextContentPartComponent;
+        Image?: ImageContentPartComponent;
+        UI?: UIContentPartComponent;
         tools?: {
-            by_name?: Record<string, ComponentType<{
-                part: ToolCallContentPart;
-                status: "done" | "in_progress" | "error";
-            }>>;
-            Fallback?: ComponentType<{
-                part: ToolCallContentPart;
-                status: "done" | "in_progress" | "error";
-            }>;
+            by_name?: Record<string, ToolCallContentPartComponent>;
+            Fallback?: ComponentType<ToolCallContentPartProps>;
         };
     };
 };
@@ -331,4 +316,4 @@ declare namespace internal {
   export { internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider };
 }
 
-export { index$1 as ActionBarPrimitive, AppendMessage, AssistantContentPart, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$4 as ComposerPrimitive, index as ContentPartPrimitive, internal as INTERNAL, index$3 as MessagePrimitive, type ReactThreadRuntime, TextContentPart, ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useLocalRuntime, useReloadMessage };
+export { index$1 as ActionBarPrimitive, AppendMessage, AssistantContentPart, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$4 as ComposerPrimitive, index as ContentPartPrimitive, internal as INTERNAL, index$3 as MessagePrimitive, type ReactThreadRuntime, ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useLocalRuntime, useReloadMessage };

package/dist/index.d.ts

@@ -1,8 +1,8 @@
 import * as react from 'react';
 import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
-import { T as TextContentPart, I as ImageContentPart, U as UIContentPart, a as ToolCallContentPart, b as ThreadState, c as Unsubscribe, M as ModelConfigProvider, d as ThreadMessage, e as ModelConfig, A as AssistantContentPart, f as AppendMessage } from './Thread-DcmnmkmV.js';
-export { j as AppendContentPart, g as AssistantMessage, i as UserContentPart, h as UserMessage } from './Thread-DcmnmkmV.js';
+import { T as TextContentPartComponent, I as ImageContentPartComponent, U as UIContentPartComponent, a as ToolCallContentPartComponent, b as ToolCallContentPartProps, c as ThreadState, d as Unsubscribe, M as ModelConfigProvider, e as ThreadMessage, f as ModelConfig, A as AssistantContentPart, g as AppendMessage } from './Thread-ZUDFhMtm.js';
+export { k as AppendContentPart, h as AssistantMessage, l as TextContentPart, j as UserContentPart, i as UserMessage } from './Thread-ZUDFhMtm.js';
 import { ComponentPropsWithoutRef, Primitive } from '@radix-ui/react-primitive';
 import 'zod';
 
@@ -135,27 +135,12 @@ declare const MessageIf: FC<MessageIfProps>;
 
 type MessageContentProps = {
     components?: {
-        Text?: ComponentType<{
-            part: TextContentPart;
-            status: "done" | "in_progress" | "error";
-        }>;
-        Image?: ComponentType<{
-            part: ImageContentPart;
-            status: "done" | "in_progress" | "error";
-        }>;
-        UI?: ComponentType<{
-            part: UIContentPart;
-            status: "done" | "in_progress" | "error";
-        }>;
+        Text?: TextContentPartComponent;
+        Image?: ImageContentPartComponent;
+        UI?: UIContentPartComponent;
         tools?: {
-            by_name?: Record<string, ComponentType<{
-                part: ToolCallContentPart;
-                status: "done" | "in_progress" | "error";
-            }>>;
-            Fallback?: ComponentType<{
-                part: ToolCallContentPart;
-                status: "done" | "in_progress" | "error";
-            }>;
+            by_name?: Record<string, ToolCallContentPartComponent>;
+            Fallback?: ComponentType<ToolCallContentPartProps>;
         };
     };
 };
@@ -331,4 +316,4 @@ declare namespace internal {
   export { internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider };
 }
 
-export { index$1 as ActionBarPrimitive, AppendMessage, AssistantContentPart, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$4 as ComposerPrimitive, index as ContentPartPrimitive, internal as INTERNAL, index$3 as MessagePrimitive, type ReactThreadRuntime, TextContentPart, ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useLocalRuntime, useReloadMessage };
+export { index$1 as ActionBarPrimitive, AppendMessage, AssistantContentPart, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$4 as ComposerPrimitive, index as ContentPartPrimitive, internal as INTERNAL, index$3 as MessagePrimitive, type ReactThreadRuntime, ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useLocalRuntime, useReloadMessage };

package/dist/index.js

@@ -693,6 +693,7 @@ var ComposerInput = (0, import_react20.forwardRef)(
     const focus = (0, import_react20.useCallback)(() => {
       const textarea = textareaRef.current;
       if (!textarea || !autoFocusEnabled) return;
+      console.log("focus");
       textarea.focus();
       textarea.setSelectionRange(
         textareaRef.current.value.length,
@@ -711,6 +712,7 @@ var ComposerInput = (0, import_react20.forwardRef)(
         value,
         ...rest,
         ref,
+        autoFocus,
         disabled,
         onChange: (0, import_primitive5.composeEventHandlers)(onChange, (e) => {
           const composerState = useComposer.getState();