@assistant-ui/react 0.1.12 → 0.2.0

package/dist/index.d.mts

@@ -1,11 +1,10 @@
 import * as react from 'react';
-import { FC, PropsWithChildren, ComponentPropsWithoutRef, ComponentType, ReactNode } from 'react';
+import { FC, PropsWithChildren, ComponentPropsWithoutRef, ReactNode, ComponentType } from 'react';
 import * as PopoverPrimitive from '@radix-ui/react-popover';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { Primitive } from '@radix-ui/react-primitive';
-import { T as TextContentPartComponent, I as ImageContentPartComponent, U as UIContentPartComponent, a as ToolCallContentPartComponent, b as ToolCallContentPartProps, c as ThreadState, d as ThreadActionsState, e as Unsubscribe, M as ModelConfigProvider, f as ThreadMessage, g as ModelConfig, A as AssistantContentPart, h as AppendMessage } from './ThreadActions-BLcKtagX.mjs';
-export { l as AppendContentPart, i as AssistantMessage, m as TextContentPart, k as UserContentPart, j as UserMessage } from './ThreadActions-BLcKtagX.mjs';
-import 'zod';
+import { z } from 'zod';
+import { UseBoundStore, StoreApi } from 'zustand';
 
 type UseActionBarCopyProps = {
     copiedDuration?: number;
@@ -181,9 +180,9 @@ declare namespace index$3 {
 
 declare const ContentPartInProgressIndicator: FC;
 
-type PrimitiveSpanProps = ComponentPropsWithoutRef<typeof Primitive.span>;
+type PrimitiveSpanProps = ComponentPropsWithoutRef<typeof Primitive.p>;
 type ContentPartTextProps = Omit<PrimitiveSpanProps, "children">;
-declare const ContentPartText: react.ForwardRefExoticComponent<ContentPartTextProps & react.RefAttributes<HTMLSpanElement>>;
+declare const ContentPartText: react.ForwardRefExoticComponent<ContentPartTextProps & react.RefAttributes<HTMLParagraphElement>>;
 
 declare const ContentPartImage: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ImgHTMLAttributes<HTMLImageElement>, HTMLImageElement>, "ref"> & {
     ref?: ((instance: HTMLImageElement | null) => void | react.DO_NOT_USE_OR_YOU_WILL_BE_FIRED_CALLBACK_REF_RETURN_VALUES[keyof react.DO_NOT_USE_OR_YOU_WILL_BE_FIRED_CALLBACK_REF_RETURN_VALUES]) | react.RefObject<HTMLImageElement> | null | undefined;
@@ -206,6 +205,71 @@ declare const MessageRoot: react.ForwardRefExoticComponent<Omit<Omit<react.Detai
 type MessageIfProps = PropsWithChildren<UseMessageIfProps>;
 declare const MessageIf: FC<MessageIfProps>;
 
+type TextContentPart = {
+    type: "text";
+    text: string;
+};
+type ImageContentPart = {
+    type: "image";
+    image: string;
+};
+type UIContentPart = {
+    type: "ui";
+    display: ReactNode;
+};
+type ToolCallContentPart<TArgs = unknown, TResult = unknown> = {
+    type: "tool-call";
+    toolCallId: string;
+    toolName: string;
+    args: TArgs;
+    result?: TResult;
+};
+type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
+type AssistantContentPart = TextContentPart | ToolCallContentPart | UIContentPart;
+type AppendContentPart = TextContentPart | ImageContentPart;
+type BaseMessage = {
+    id: string;
+    createdAt: Date;
+};
+type UserMessage = BaseMessage & {
+    role: "user";
+    content: UserContentPart[];
+};
+type AssistantMessage = BaseMessage & {
+    role: "assistant";
+    content: AssistantContentPart[];
+    status: "in_progress" | "done" | "error";
+};
+type AppendMessage = {
+    parentId: string | null;
+    role: "user";
+    content: AppendContentPart[];
+};
+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;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type MessageContentProps = {
     components?: {
         Text?: TextContentPartComponent;
@@ -288,6 +352,131 @@ declare namespace index {
   export { ThreadEmpty as Empty, ThreadIf as If, ThreadMessages as Messages, ThreadRoot as Root, ThreadScrollToBottom as ScrollToBottom, ThreadSuggestion as Suggestion, ThreadViewport as Viewport };
 }
 
+type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
+type Tool<TArgs = unknown, TResult = unknown> = {
+    description?: string;
+    parameters: z.ZodSchema<TArgs>;
+    execute: ToolExecuteFunction<TArgs, TResult>;
+};
+type ModelConfig = {
+    priority?: number;
+    system?: string;
+    tools?: Record<string, Tool<any, any>>;
+};
+type ModelConfigProvider = () => ModelConfig;
+
+type Unsubscribe = () => void;
+
+type AssistantRuntimeProviderProps = {
+    runtime: AssistantRuntime;
+};
+declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
+
+type AssistantModelConfigState = Readonly<{
+    getModelConfig: ModelConfigProvider;
+    registerModelConfigProvider: (provider: ModelConfigProvider) => () => void;
+}>;
+
+type ContentPartState = Readonly<{
+    status: "in_progress" | "done" | "error";
+    part: ThreadMessage["content"][number];
+}>;
+
+type MessageState = Readonly<{
+    message: Readonly<ThreadMessage>;
+    parentId: string | null;
+    branches: readonly string[];
+    isLast: boolean;
+}>;
+
+type BaseComposerState = Readonly<{
+    value: string;
+    setValue: (value: string) => void;
+}>;
+
+type ReadonlyStore<T> = UseBoundStore<Omit<StoreApi<T>, "setState" | "destroy">>;
+
+type EditComposerState = BaseComposerState & Readonly<{
+    isEditing: boolean;
+    edit: () => void;
+    send: () => void;
+    cancel: () => boolean;
+}>;
+
+type ThreadState = Readonly<{
+    messages: readonly ThreadMessage[];
+    isRunning: boolean;
+}>;
+
+type ThreadActionsState = Readonly<{
+    getBranches: (messageId: string) => readonly string[];
+    switchToBranch: (branchId: string) => void;
+    append: (message: AppendMessage) => void;
+    startRun: (parentId: string | null) => void;
+    cancelRun: () => void;
+    addToolResult: (toolCallId: string, result: any) => void;
+}>;
+
+type ComposerState = BaseComposerState & Readonly<{
+    isEditing: true;
+    send: () => void;
+    cancel: () => boolean;
+    focus: () => void;
+    onFocus: (listener: () => void) => Unsubscribe;
+}>;
+
+type ThreadViewportState = Readonly<{
+    isAtBottom: boolean;
+    scrollToBottom: () => void;
+    onScrollToBottom: (callback: () => void) => Unsubscribe;
+}>;
+
+type AssistantToolUIsState = Readonly<{
+    getToolUI: (toolName: string) => ToolCallContentPartComponent | null;
+    setToolUI: (toolName: string, render: ToolCallContentPartComponent) => () => void;
+}>;
+
+type AssistantContextValue = {
+    useModelConfig: ReadonlyStore<AssistantModelConfigState>;
+    useToolUIs: ReadonlyStore<AssistantToolUIsState>;
+};
+declare const useAssistantContext: () => AssistantContextValue;
+
+type ThreadContextValue = {
+    useThread: ReadonlyStore<ThreadState>;
+    useThreadActions: ReadonlyStore<ThreadActionsState>;
+    useComposer: ReadonlyStore<ComposerState>;
+    useViewport: ReadonlyStore<ThreadViewportState>;
+};
+declare const useThreadContext: () => ThreadContextValue;
+
+type ComposerContextValue = {
+    useComposer: ReadonlyStore<EditComposerState | ComposerState>;
+    type: "edit" | "new";
+};
+declare const useComposerContext: () => ComposerContextValue;
+
+type MessageUtilsState = Readonly<{
+    inProgressIndicator: ReactNode | null;
+    setInProgressIndicator: (value: ReactNode | null) => void;
+    isCopied: boolean;
+    setIsCopied: (value: boolean) => void;
+    isHovering: boolean;
+    setIsHovering: (value: boolean) => void;
+}>;
+
+type MessageContextValue = {
+    useMessage: ReadonlyStore<MessageState>;
+    useMessageUtils: ReadonlyStore<MessageUtilsState>;
+    useEditComposer: ReadonlyStore<EditComposerState>;
+};
+declare const useMessageContext: () => MessageContextValue;
+
+type ContentPartContextValue = {
+    useContentPart: ReadonlyStore<ContentPartState>;
+};
+declare const useContentPartContext: () => ContentPartContextValue;
+
 type ThreadRuntime = Readonly<ThreadState & ThreadActionsState & {
     subscribe: (callback: () => void) => Unsubscribe;
 }>;
@@ -335,10 +524,23 @@ declare class LocalRuntime implements AssistantRuntime {
 
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
 
-type AssistantRuntimeProviderProps = {
-    runtime: AssistantRuntime;
+type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {
+    toolName: string;
+    render?: ToolCallContentPartComponent<TArgs, TResult>;
 };
-declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
+declare const useAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => void;
+
+declare const makeAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => () => null;
+
+type AssistantToolUIProps<TArgs, TResult> = {
+    toolName: string;
+    render: ToolCallContentPartComponent<TArgs, TResult>;
+};
+declare const useAssistantToolUI: (tool: AssistantToolUIProps<any, any> | null) => void;
+
+declare const makeAssistantToolUI: <TArgs, TResult>(tool: AssistantToolUIProps<TArgs, TResult>) => () => null;
+
+declare const useAssistantInstructions: (instruction: string) => void;
 
 declare class ProxyConfigProvider {
     private _providers;
@@ -368,4 +570,4 @@ declare namespace internal {
   export { internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider };
 }
 
-export { index$6 as ActionBarPrimitive, AppendMessage, AssistantContentPart, index$5 as AssistantModalPrimitive, type AssistantRuntime, AssistantRuntimeProvider, index$4 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$3 as ComposerPrimitive, index$2 as ContentPartPrimitive, internal as INTERNAL, index$1 as MessagePrimitive, type ReactThreadRuntime, ThreadMessage, index as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useActionBarCopy, useActionBarEdit, useActionBarReload, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerIf, useComposerSend, useContentPartDisplay, useContentPartImage, useContentPartInProgressIndicator, useContentPartText, useLocalRuntime, useMessageIf, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };
+export { index$6 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, type AssistantContentPart, type AssistantContextValue, type AssistantMessage, index$5 as AssistantModalPrimitive, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, index$4 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, type ComposerContextValue, index$3 as ComposerPrimitive, type ComposerState, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartState, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type MessageContextValue, index$1 as MessagePrimitive, type MessageState, type ModelConfig, type ReactThreadRuntime, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, type ThreadContextValue, type ThreadMessage, index as ThreadPrimitive, type ThreadRuntime, type ThreadState, type ThreadViewportState, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UserContentPart, type UserMessage, makeAssistantTool, makeAssistantToolUI, useActionBarCopy, useActionBarEdit, useActionBarReload, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartInProgressIndicator, useContentPartText, useLocalRuntime, useMessageContext, useMessageIf, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };

package/dist/index.d.ts

@@ -1,11 +1,10 @@
 import * as react from 'react';
-import { FC, PropsWithChildren, ComponentPropsWithoutRef, ComponentType, ReactNode } from 'react';
+import { FC, PropsWithChildren, ComponentPropsWithoutRef, ReactNode, ComponentType } from 'react';
 import * as PopoverPrimitive from '@radix-ui/react-popover';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { Primitive } from '@radix-ui/react-primitive';
-import { T as TextContentPartComponent, I as ImageContentPartComponent, U as UIContentPartComponent, a as ToolCallContentPartComponent, b as ToolCallContentPartProps, c as ThreadState, d as ThreadActionsState, e as Unsubscribe, M as ModelConfigProvider, f as ThreadMessage, g as ModelConfig, A as AssistantContentPart, h as AppendMessage } from './ThreadActions-BLcKtagX.js';
-export { l as AppendContentPart, i as AssistantMessage, m as TextContentPart, k as UserContentPart, j as UserMessage } from './ThreadActions-BLcKtagX.js';
-import 'zod';
+import { z } from 'zod';
+import { UseBoundStore, StoreApi } from 'zustand';
 
 type UseActionBarCopyProps = {
     copiedDuration?: number;
@@ -181,9 +180,9 @@ declare namespace index$3 {
 
 declare const ContentPartInProgressIndicator: FC;
 
-type PrimitiveSpanProps = ComponentPropsWithoutRef<typeof Primitive.span>;
+type PrimitiveSpanProps = ComponentPropsWithoutRef<typeof Primitive.p>;
 type ContentPartTextProps = Omit<PrimitiveSpanProps, "children">;
-declare const ContentPartText: react.ForwardRefExoticComponent<ContentPartTextProps & react.RefAttributes<HTMLSpanElement>>;
+declare const ContentPartText: react.ForwardRefExoticComponent<ContentPartTextProps & react.RefAttributes<HTMLParagraphElement>>;
 
 declare const ContentPartImage: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ImgHTMLAttributes<HTMLImageElement>, HTMLImageElement>, "ref"> & {
     ref?: ((instance: HTMLImageElement | null) => void | react.DO_NOT_USE_OR_YOU_WILL_BE_FIRED_CALLBACK_REF_RETURN_VALUES[keyof react.DO_NOT_USE_OR_YOU_WILL_BE_FIRED_CALLBACK_REF_RETURN_VALUES]) | react.RefObject<HTMLImageElement> | null | undefined;
@@ -206,6 +205,71 @@ declare const MessageRoot: react.ForwardRefExoticComponent<Omit<Omit<react.Detai
 type MessageIfProps = PropsWithChildren<UseMessageIfProps>;
 declare const MessageIf: FC<MessageIfProps>;
 
+type TextContentPart = {
+    type: "text";
+    text: string;
+};
+type ImageContentPart = {
+    type: "image";
+    image: string;
+};
+type UIContentPart = {
+    type: "ui";
+    display: ReactNode;
+};
+type ToolCallContentPart<TArgs = unknown, TResult = unknown> = {
+    type: "tool-call";
+    toolCallId: string;
+    toolName: string;
+    args: TArgs;
+    result?: TResult;
+};
+type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
+type AssistantContentPart = TextContentPart | ToolCallContentPart | UIContentPart;
+type AppendContentPart = TextContentPart | ImageContentPart;
+type BaseMessage = {
+    id: string;
+    createdAt: Date;
+};
+type UserMessage = BaseMessage & {
+    role: "user";
+    content: UserContentPart[];
+};
+type AssistantMessage = BaseMessage & {
+    role: "assistant";
+    content: AssistantContentPart[];
+    status: "in_progress" | "done" | "error";
+};
+type AppendMessage = {
+    parentId: string | null;
+    role: "user";
+    content: AppendContentPart[];
+};
+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;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type MessageContentProps = {
     components?: {
         Text?: TextContentPartComponent;
@@ -288,6 +352,131 @@ declare namespace index {
   export { ThreadEmpty as Empty, ThreadIf as If, ThreadMessages as Messages, ThreadRoot as Root, ThreadScrollToBottom as ScrollToBottom, ThreadSuggestion as Suggestion, ThreadViewport as Viewport };
 }
 
+type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
+type Tool<TArgs = unknown, TResult = unknown> = {
+    description?: string;
+    parameters: z.ZodSchema<TArgs>;
+    execute: ToolExecuteFunction<TArgs, TResult>;
+};
+type ModelConfig = {
+    priority?: number;
+    system?: string;
+    tools?: Record<string, Tool<any, any>>;
+};
+type ModelConfigProvider = () => ModelConfig;
+
+type Unsubscribe = () => void;
+
+type AssistantRuntimeProviderProps = {
+    runtime: AssistantRuntime;
+};
+declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
+
+type AssistantModelConfigState = Readonly<{
+    getModelConfig: ModelConfigProvider;
+    registerModelConfigProvider: (provider: ModelConfigProvider) => () => void;
+}>;
+
+type ContentPartState = Readonly<{
+    status: "in_progress" | "done" | "error";
+    part: ThreadMessage["content"][number];
+}>;
+
+type MessageState = Readonly<{
+    message: Readonly<ThreadMessage>;
+    parentId: string | null;
+    branches: readonly string[];
+    isLast: boolean;
+}>;
+
+type BaseComposerState = Readonly<{
+    value: string;
+    setValue: (value: string) => void;
+}>;
+
+type ReadonlyStore<T> = UseBoundStore<Omit<StoreApi<T>, "setState" | "destroy">>;
+
+type EditComposerState = BaseComposerState & Readonly<{
+    isEditing: boolean;
+    edit: () => void;
+    send: () => void;
+    cancel: () => boolean;
+}>;
+
+type ThreadState = Readonly<{
+    messages: readonly ThreadMessage[];
+    isRunning: boolean;
+}>;
+
+type ThreadActionsState = Readonly<{
+    getBranches: (messageId: string) => readonly string[];
+    switchToBranch: (branchId: string) => void;
+    append: (message: AppendMessage) => void;
+    startRun: (parentId: string | null) => void;
+    cancelRun: () => void;
+    addToolResult: (toolCallId: string, result: any) => void;
+}>;
+
+type ComposerState = BaseComposerState & Readonly<{
+    isEditing: true;
+    send: () => void;
+    cancel: () => boolean;
+    focus: () => void;
+    onFocus: (listener: () => void) => Unsubscribe;
+}>;
+
+type ThreadViewportState = Readonly<{
+    isAtBottom: boolean;
+    scrollToBottom: () => void;
+    onScrollToBottom: (callback: () => void) => Unsubscribe;
+}>;
+
+type AssistantToolUIsState = Readonly<{
+    getToolUI: (toolName: string) => ToolCallContentPartComponent | null;
+    setToolUI: (toolName: string, render: ToolCallContentPartComponent) => () => void;
+}>;
+
+type AssistantContextValue = {
+    useModelConfig: ReadonlyStore<AssistantModelConfigState>;
+    useToolUIs: ReadonlyStore<AssistantToolUIsState>;
+};
+declare const useAssistantContext: () => AssistantContextValue;
+
+type ThreadContextValue = {
+    useThread: ReadonlyStore<ThreadState>;
+    useThreadActions: ReadonlyStore<ThreadActionsState>;
+    useComposer: ReadonlyStore<ComposerState>;
+    useViewport: ReadonlyStore<ThreadViewportState>;
+};
+declare const useThreadContext: () => ThreadContextValue;
+
+type ComposerContextValue = {
+    useComposer: ReadonlyStore<EditComposerState | ComposerState>;
+    type: "edit" | "new";
+};
+declare const useComposerContext: () => ComposerContextValue;
+
+type MessageUtilsState = Readonly<{
+    inProgressIndicator: ReactNode | null;
+    setInProgressIndicator: (value: ReactNode | null) => void;
+    isCopied: boolean;
+    setIsCopied: (value: boolean) => void;
+    isHovering: boolean;
+    setIsHovering: (value: boolean) => void;
+}>;
+
+type MessageContextValue = {
+    useMessage: ReadonlyStore<MessageState>;
+    useMessageUtils: ReadonlyStore<MessageUtilsState>;
+    useEditComposer: ReadonlyStore<EditComposerState>;
+};
+declare const useMessageContext: () => MessageContextValue;
+
+type ContentPartContextValue = {
+    useContentPart: ReadonlyStore<ContentPartState>;
+};
+declare const useContentPartContext: () => ContentPartContextValue;
+
 type ThreadRuntime = Readonly<ThreadState & ThreadActionsState & {
     subscribe: (callback: () => void) => Unsubscribe;
 }>;
@@ -335,10 +524,23 @@ declare class LocalRuntime implements AssistantRuntime {
 
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
 
-type AssistantRuntimeProviderProps = {
-    runtime: AssistantRuntime;
+type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {
+    toolName: string;
+    render?: ToolCallContentPartComponent<TArgs, TResult>;
 };
-declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
+declare const useAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => void;
+
+declare const makeAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => () => null;
+
+type AssistantToolUIProps<TArgs, TResult> = {
+    toolName: string;
+    render: ToolCallContentPartComponent<TArgs, TResult>;
+};
+declare const useAssistantToolUI: (tool: AssistantToolUIProps<any, any> | null) => void;
+
+declare const makeAssistantToolUI: <TArgs, TResult>(tool: AssistantToolUIProps<TArgs, TResult>) => () => null;
+
+declare const useAssistantInstructions: (instruction: string) => void;
 
 declare class ProxyConfigProvider {
     private _providers;
@@ -368,4 +570,4 @@ declare namespace internal {
   export { internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider };
 }
 
-export { index$6 as ActionBarPrimitive, AppendMessage, AssistantContentPart, index$5 as AssistantModalPrimitive, type AssistantRuntime, AssistantRuntimeProvider, index$4 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$3 as ComposerPrimitive, index$2 as ContentPartPrimitive, internal as INTERNAL, index$1 as MessagePrimitive, type ReactThreadRuntime, ThreadMessage, index as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useActionBarCopy, useActionBarEdit, useActionBarReload, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerIf, useComposerSend, useContentPartDisplay, useContentPartImage, useContentPartInProgressIndicator, useContentPartText, useLocalRuntime, useMessageIf, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };
+export { index$6 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, type AssistantContentPart, type AssistantContextValue, type AssistantMessage, index$5 as AssistantModalPrimitive, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, index$4 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, type ComposerContextValue, index$3 as ComposerPrimitive, type ComposerState, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartState, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type MessageContextValue, index$1 as MessagePrimitive, type MessageState, type ModelConfig, type ReactThreadRuntime, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, type ThreadContextValue, type ThreadMessage, index as ThreadPrimitive, type ThreadRuntime, type ThreadState, type ThreadViewportState, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UserContentPart, type UserMessage, makeAssistantTool, makeAssistantToolUI, useActionBarCopy, useActionBarEdit, useActionBarReload, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartInProgressIndicator, useContentPartText, useLocalRuntime, useMessageContext, useMessageIf, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };