@assistant-ui/react 0.3.4 → 0.4.0

package/dist/index.d.mts

@@ -1,15 +1,21 @@
 import * as react from 'react';
 import { ReactNode, ComponentType, PropsWithChildren, ComponentPropsWithoutRef, FC } from 'react';
 import { z } from 'zod';
+import { JSONSchema7 } from 'json-schema';
+import { LanguageModelV1FinishReason, LanguageModelV1LogProbs } from '@ai-sdk/provider';
 import { UseBoundStore, StoreApi } from 'zustand';
 import { Primitive } from '@radix-ui/react-primitive';
 import * as PopoverPrimitive from '@radix-ui/react-popover';
+import * as react_textarea_autosize from 'react-textarea-autosize';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
+import * as class_variance_authority from 'class-variance-authority';
+import { VariantProps } from 'class-variance-authority';
+import * as class_variance_authority_types from 'class-variance-authority/types';
 
 type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
 type Tool<TArgs = unknown, TResult = unknown> = {
     description?: string | undefined;
-    parameters: z.ZodSchema<TArgs>;
+    parameters: z.ZodSchema<TArgs> | JSONSchema7;
     execute: ToolExecuteFunction<TArgs, TResult>;
 };
 type ModelConfig = {
@@ -40,41 +46,76 @@ type ToolCallContentPart<TArgs = unknown, TResult = unknown> = {
     args: TArgs;
     result?: TResult;
 };
-type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
-type AssistantContentPart = TextContentPart | ToolCallContentPart | UIContentPart;
+type ThreadUserContentPart = TextContentPart | ImageContentPart | UIContentPart;
+type ThreadAssistantContentPart = TextContentPart | ToolCallContentPart | UIContentPart;
 type AppendContentPart = TextContentPart | ImageContentPart;
-type BaseMessage = {
+type MessageCommonProps = {
     id: string;
     createdAt: Date;
 };
-type UserMessage = BaseMessage & {
+type MessageStatus = {
+    type: "in_progress";
+} | {
+    type: "done";
+    finishReason?: LanguageModelV1FinishReason;
+    logprops?: LanguageModelV1LogProbs;
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+    };
+} | {
+    type: "error";
+    error: unknown;
+};
+type ThreadSystemMessage = MessageCommonProps & {
+    role: "system";
+    content: [TextContentPart];
+};
+type ThreadUserMessage = MessageCommonProps & {
     role: "user";
-    content: UserContentPart[];
+    content: ThreadUserContentPart[];
 };
-type AssistantMessage = BaseMessage & {
+type ThreadAssistantMessage = MessageCommonProps & {
     role: "assistant";
-    content: AssistantContentPart[];
-    status: "in_progress" | "done" | "error";
+    content: ThreadAssistantContentPart[];
+    status: MessageStatus;
 };
 type AppendMessage = {
     parentId: string | null;
     role: "user";
     content: AppendContentPart[];
 };
-type ThreadMessage = UserMessage | AssistantMessage;
+type ThreadMessage = ThreadSystemMessage | ThreadUserMessage | ThreadAssistantMessage;
+/** Core Message Types (without UI content parts) */
+type CoreUserContentPart = TextContentPart | ImageContentPart;
+type CoreAssistantContentPart = TextContentPart | ToolCallContentPart;
+type CoreUserMessage = MessageCommonProps & {
+    role: "user";
+    content: CoreUserContentPart[];
+};
+type CoreAssistantMessage = MessageCommonProps & {
+    role: "assistant";
+    content: CoreAssistantContentPart[];
+    status: MessageStatus;
+};
+type CoreThreadMessage = ThreadSystemMessage | CoreUserMessage | CoreAssistantMessage;
 
 type ReactThreadRuntime = ThreadRuntime & {
     unstable_synchronizer?: ComponentType;
 };
 
+type ChatModelRunUpdate = {
+    content: ThreadAssistantContentPart[];
+};
 type ChatModelRunResult = {
-    content: AssistantContentPart[];
+    content: ThreadAssistantContentPart[];
+    status?: MessageStatus;
 };
 type ChatModelRunOptions = {
     messages: ThreadMessage[];
     abortSignal: AbortSignal;
     config: ModelConfig;
-    onUpdate: (result: ChatModelRunResult) => void;
+    onUpdate: (result: ChatModelRunUpdate) => void;
 };
 type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
@@ -101,7 +142,7 @@ declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRu
     append(message: AppendMessage): void;
     startRun(parentId: string | null): void;
     cancelRun(): void;
-    addToolResult(toolCallId: string, result: any): void;
+    addToolResult(options: AddToolResultOptions): void;
     private _subscriptions;
     subscribe(callback: () => void): Unsubscribe;
     private subscriptionHandler;
@@ -137,11 +178,21 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     cancelRun(): void;
     private notifySubscribers;
     subscribe(callback: () => void): Unsubscribe;
-    addToolResult(): void;
+    addToolResult({ messageId, toolCallId, result }: AddToolResultOptions): void;
 }
 
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
 
+type EdgeRuntimeOptions = {
+    api: string;
+};
+declare const useEdgeRuntime: (options: EdgeRuntimeOptions) => LocalRuntime;
+
+type AddToolResultOptions = {
+    messageId: string;
+    toolCallId: string;
+    result: any;
+};
 type ThreadActionsState = Readonly<{
     capabilities: Readonly<{
         edit: boolean;
@@ -154,28 +205,27 @@ type ThreadActionsState = Readonly<{
     append: (message: AppendMessage) => void;
     startRun: (parentId: string | null) => void;
     cancelRun: () => void;
-    addToolResult: (toolCallId: string, result: any) => void;
+    addToolResult: (options: AddToolResultOptions) => void;
 }>;
 
-type ContentPartStatus = "done" | "in_progress" | "error";
 type TextContentPartProps = {
     part: TextContentPart;
-    status: ContentPartStatus;
+    status: MessageStatus;
 };
 type TextContentPartComponent = ComponentType<TextContentPartProps>;
 type ImageContentPartProps = {
     part: ImageContentPart;
-    status: ContentPartStatus;
+    status: MessageStatus;
 };
 type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
 type UIContentPartProps = {
     part: UIContentPart;
-    status: ContentPartStatus;
+    status: MessageStatus;
 };
 type UIContentPartComponent = ComponentType<UIContentPartProps>;
 type ToolCallContentPartProps<TArgs = any, TResult = any> = {
     part: ToolCallContentPart<TArgs, TResult>;
-    status: ContentPartStatus;
+    status: MessageStatus;
     addResult: (result: any) => void;
 };
 type ToolCallContentPartComponent<TArgs = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
@@ -195,21 +245,18 @@ type AssistantRuntimeProviderProps = {
 };
 declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
 
+type AssistantActionsState = Readonly<{
+    switchToThread: (threadId: string | null) => void;
+}>;
+
 type AssistantModelConfigState = Readonly<{
     getModelConfig: ModelConfigProvider;
     registerModelConfigProvider: (provider: ModelConfigProvider) => () => void;
 }>;
 
-type ContentPartState = Readonly<{
-    status: "in_progress" | "done" | "error";
-    part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-}>;
-
-type MessageState = Readonly<{
-    message: Readonly<ThreadMessage>;
-    parentId: string | null;
-    branches: readonly string[];
-    isLast: boolean;
+type AssistantToolUIsState = Readonly<{
+    getToolUI: (toolName: string) => ToolCallContentPartComponent | null;
+    setToolUI: (toolName: string, render: ToolCallContentPartComponent) => () => void;
 }>;
 
 type BaseComposerState = Readonly<{
@@ -219,33 +266,19 @@ type BaseComposerState = Readonly<{
 
 type ReadonlyStore<T> = UseBoundStore<Omit<StoreApi<T>, "setState" | "destroy">>;
 
-type EditComposerState = BaseComposerState & Readonly<{
-    canCancel: boolean;
-    isEditing: boolean;
-    edit: () => void;
-    send: () => void;
-    cancel: () => void;
-}>;
-
-type ThreadState = Readonly<{
-    isRunning: boolean;
-}>;
-
-type AssistantToolUIsState = Readonly<{
-    getToolUI: (toolName: string) => ToolCallContentPartComponent | null;
-    setToolUI: (toolName: string, render: ToolCallContentPartComponent) => () => void;
-}>;
-
-type AssistantActionsState = Readonly<{
-    switchToThread: (threadId: string | null) => void;
-}>;
-
 type AssistantContextValue = {
     useModelConfig: ReadonlyStore<AssistantModelConfigState>;
     useToolUIs: ReadonlyStore<AssistantToolUIsState>;
     useAssistantActions: ReadonlyStore<AssistantActionsState>;
 };
-declare const useAssistantContext: () => AssistantContextValue;
+declare function useAssistantContext(): AssistantContextValue;
+declare function useAssistantContext(options: {
+    optional: true;
+}): AssistantContextValue | null;
+
+type ThreadState = Readonly<{
+    isRunning: boolean;
+}>;
 
 type ThreadViewportState = Readonly<{
     isAtBottom: boolean;
@@ -262,7 +295,18 @@ type ThreadContextValue = {
     useComposer: ReadonlyStore<ComposerState>;
     useViewport: ReadonlyStore<ThreadViewportState>;
 };
-declare const useThreadContext: () => ThreadContextValue;
+declare function useThreadContext(): ThreadContextValue;
+declare function useThreadContext(options: {
+    optional: true;
+}): ThreadContextValue | null;
+
+type EditComposerState = BaseComposerState & Readonly<{
+    canCancel: boolean;
+    isEditing: boolean;
+    edit: () => void;
+    send: () => void;
+    cancel: () => void;
+}>;
 
 type ComposerContextValue = {
     useComposer: ReadonlyStore<EditComposerState | ComposerState>;
@@ -270,6 +314,13 @@ type ComposerContextValue = {
 };
 declare const useComposerContext: () => ComposerContextValue;
 
+type MessageState = Readonly<{
+    message: Readonly<ThreadMessage>;
+    parentId: string | null;
+    branches: readonly string[];
+    isLast: boolean;
+}>;
+
 type MessageUtilsState = Readonly<{
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
@@ -282,12 +333,23 @@ type MessageContextValue = {
     useMessageUtils: ReadonlyStore<MessageUtilsState>;
     useEditComposer: ReadonlyStore<EditComposerState>;
 };
-declare const useMessageContext: () => MessageContextValue;
+declare function useMessageContext(): MessageContextValue;
+declare function useMessageContext(options: {
+    optional: true;
+}): MessageContextValue | null;
+
+type ContentPartState = Readonly<{
+    status: MessageStatus;
+    part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
+}>;
 
 type ContentPartContextValue = {
     useContentPart: ReadonlyStore<ContentPartState>;
 };
-declare const useContentPartContext: () => ContentPartContextValue;
+declare function useContentPartContext(): ContentPartContextValue;
+declare function useContentPartContext(options: {
+    optional: true;
+}): ContentPartContextValue | null;
 
 type ComposerState = BaseComposerState & Readonly<{
     canCancel: boolean;
@@ -357,23 +419,24 @@ declare const useComposerIf: (props: UseComposerIfProps) => boolean;
 declare const useComposerSend: () => (() => void) | null;
 
 declare const useContentPartDisplay: () => Readonly<{
-    status: "in_progress" | "done" | "error";
+    status: MessageStatus;
     part: UIContentPart;
 }>;
 
 declare const useContentPartImage: () => Readonly<{
-    status: "in_progress" | "done" | "error";
+    status: MessageStatus;
     part: ImageContentPart;
 }>;
 
 declare const useContentPartText: () => Readonly<{
-    status: "in_progress" | "done" | "error";
+    status: MessageStatus;
     part: TextContentPart;
 }>;
 
 type MessageIfFilters = {
     user: boolean | undefined;
     assistant: boolean | undefined;
+    system: boolean | undefined;
     hasBranches: boolean | undefined;
     copied: boolean | undefined;
     lastOrHover: boolean | undefined;
@@ -632,11 +695,13 @@ type ThreadPrimitiveMessagesProps = {
         UserMessage?: ComponentType | undefined;
         EditComposer?: ComponentType | undefined;
         AssistantMessage?: ComponentType | undefined;
+        SystemMessage?: ComponentType | undefined;
     } | {
         Message?: ComponentType | undefined;
         UserMessage: ComponentType;
         EditComposer?: ComponentType | undefined;
         AssistantMessage: ComponentType;
+        SystemMessage?: ComponentType | undefined;
     };
 };
 declare const ThreadPrimitiveMessages: react.NamedExoticComponent<ThreadPrimitiveMessagesProps>;
@@ -659,6 +724,298 @@ declare namespace index {
   export { ThreadPrimitiveEmpty as Empty, ThreadPrimitiveIf as If, ThreadPrimitiveMessages as Messages, ThreadPrimitiveRoot as Root, ThreadPrimitiveScrollToBottom as ScrollToBottom, ThreadPrimitiveSuggestion as Suggestion, ThreadPrimitiveViewport as Viewport };
 }
 
+type AvatarProps = {
+    src?: string | undefined;
+    alt?: string | undefined;
+    fallback?: string | undefined;
+};
+
+type SuggestionConfig = {
+    text: string;
+    prompt?: string;
+};
+type ThreadWelcomeConfig = {
+    message?: string | null | undefined;
+    suggestions?: SuggestionConfig[] | undefined;
+};
+type UserMessageConfig = {
+    allowEdit?: boolean | undefined;
+};
+type AssistantMessageConfig = {
+    allowReload?: boolean | undefined;
+    allowCopy?: boolean | undefined;
+    components?: {
+        Text?: TextContentPartComponent | undefined;
+    } | undefined;
+};
+type StringsConfig = {
+    assistantModal?: {
+        open: {
+            button: {
+                tooltip?: string | undefined;
+            };
+        };
+        closed: {
+            button: {
+                tooltip?: string | undefined;
+            };
+        };
+    };
+    thread?: {
+        scrollToBottom?: {
+            tooltip?: string | undefined;
+        };
+    };
+    userMessage?: {
+        edit?: {
+            tooltip?: string | undefined;
+        };
+    };
+    assistantMessage?: {
+        reload?: {
+            tooltip?: string | undefined;
+        };
+        copy?: {
+            tooltip?: string | undefined;
+        };
+    };
+    branchPicker?: {
+        previous?: {
+            tooltip?: string | undefined;
+        };
+        next?: {
+            tooltip?: string | undefined;
+        };
+    };
+    composer?: {
+        send?: {
+            tooltip?: string | undefined;
+        } | undefined;
+        cancel?: {
+            tooltip?: string | undefined;
+        } | undefined;
+        input?: {
+            placeholder?: string | undefined;
+        };
+    };
+    editComposer?: {
+        send?: {
+            label?: string | undefined;
+        };
+        cancel?: {
+            label?: string | undefined;
+        };
+    };
+    code?: {
+        header?: {
+            copy?: {
+                tooltip?: string | undefined;
+            };
+        };
+    };
+};
+type ThreadConfig = {
+    runtime?: AssistantRuntime;
+    assistantAvatar?: AvatarProps;
+    welcome?: ThreadWelcomeConfig;
+    assistantMessage?: AssistantMessageConfig;
+    userMessage?: UserMessageConfig;
+    strings?: StringsConfig;
+};
+declare const useThreadConfig: () => Omit<ThreadConfig, "runtime">;
+type ThreadConfigProviderProps = PropsWithChildren<{
+    config?: ThreadConfig | undefined;
+}>;
+declare const ThreadConfigProvider: FC<ThreadConfigProviderProps>;
+
+declare const buttonVariants: (props?: ({
+    variant?: "default" | "outline" | "ghost" | null | undefined;
+    size?: "default" | "icon" | null | undefined;
+} & class_variance_authority_types.ClassProp) | undefined) => string;
+type ButtonProps = React.ComponentPropsWithoutRef<typeof Primitive.button> & VariantProps<typeof buttonVariants>;
+
+type TooltipIconButtonProps = ButtonProps & {
+    tooltip: string;
+    side?: "top" | "bottom" | "left" | "right";
+};
+declare const TooltipIconButton: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
+    ref?: ((instance: HTMLButtonElement | 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<HTMLButtonElement> | null | undefined;
+} & {
+    asChild?: boolean;
+}, "ref"> & class_variance_authority.VariantProps<(props?: ({
+    variant?: "default" | "outline" | "ghost" | null | undefined;
+    size?: "default" | "icon" | null | undefined;
+} & class_variance_authority_types.ClassProp) | undefined) => string> & {
+    tooltip: string;
+    side?: "top" | "bottom" | "left" | "right";
+} & react.RefAttributes<HTMLButtonElement>>;
+
+declare const AssistantActionBar: FC;
+declare const exports$a: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & UseActionBarFloatStatusProps & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Reload: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+    Copy: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+};
+declare const _default$9: typeof AssistantActionBar & typeof exports$a;
+
+declare const AssistantMessage: FC;
+type AssistantMessageContentProps = MessagePrimitiveContentProps & ComponentPropsWithoutRef<"div">;
+declare const exports$9: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Avatar: FC;
+    Content: react.ForwardRefExoticComponent<MessagePrimitiveContentProps & Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & react.RefAttributes<HTMLDivElement>>;
+};
+declare const _default$8: typeof AssistantMessage & typeof exports$9;
+
+declare const AssistantModal: FC<ThreadConfig>;
+declare const exports$8: {
+    Root: FC<PopoverPrimitive.PopoverProps & {
+        config?: ThreadConfig | undefined;
+    } & {
+        children?: react.ReactNode | undefined;
+    }>;
+    Trigger: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+    Content: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<PopoverPrimitive.PopoverContentProps & react.RefAttributes<HTMLDivElement>, "ref"> & {
+        dissmissOnInteractOutside?: boolean | undefined;
+    } & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+};
+declare const _default$7: typeof AssistantModal & typeof exports$8;
+
+declare const BranchPicker: FC;
+declare const exports$7: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & {
+        hideWhenSingleBranch?: boolean | undefined;
+    } & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Previous: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+    Next: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+};
+declare const _default$6: typeof BranchPicker & typeof exports$7;
+
+declare const Composer: FC;
+declare const ComposerInputStyled: react.ForwardRefExoticComponent<Partial<Omit<Omit<react_textarea_autosize.TextareaAutosizeProps & {
+    asChild?: boolean | undefined;
+} & react.RefAttributes<HTMLTextAreaElement>, "ref">, "asChild">> & react.RefAttributes<HTMLTextAreaElement>>;
+type ComposerInputProps = ComponentPropsWithoutRef<typeof ComposerInputStyled>;
+declare const exports$6: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.FormHTMLAttributes<HTMLFormElement>, HTMLFormElement>, "ref"> & {
+        ref?: ((instance: HTMLFormElement | 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<HTMLFormElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & react.RefAttributes<HTMLFormElement>, "ref">, "asChild">> & react.RefAttributes<HTMLFormElement>>;
+    Input: react.ForwardRefExoticComponent<Omit<Partial<Omit<Omit<react_textarea_autosize.TextareaAutosizeProps & {
+        asChild?: boolean | undefined;
+    } & react.RefAttributes<HTMLTextAreaElement>, "ref">, "asChild">> & react.RefAttributes<HTMLTextAreaElement>, "ref"> & react.RefAttributes<HTMLTextAreaElement>>;
+    Action: FC;
+    Send: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+    Cancel: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+};
+declare const _default$5: typeof Composer & typeof exports$6;
+
+declare const EditComposer: FC;
+declare const exports$5: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.FormHTMLAttributes<HTMLFormElement>, HTMLFormElement>, "ref"> & {
+        ref?: ((instance: HTMLFormElement | 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<HTMLFormElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & react.RefAttributes<HTMLFormElement>, "ref">, "asChild">> & react.RefAttributes<HTMLFormElement>>;
+    Input: react.ForwardRefExoticComponent<Partial<Omit<Omit<react_textarea_autosize.TextareaAutosizeProps & {
+        asChild?: boolean | undefined;
+    } & react.RefAttributes<HTMLTextAreaElement>, "ref">, "asChild">> & react.RefAttributes<HTMLTextAreaElement>>;
+    Footer: react.ForwardRefExoticComponent<Partial<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Cancel: react.ForwardRefExoticComponent<Partial<ButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+    Send: react.ForwardRefExoticComponent<Partial<ButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+};
+declare const _default$4: typeof EditComposer & typeof exports$5;
+
+declare const Thread: FC<ThreadConfig>;
+type ThreadRootProps = ThreadPrimitiveRootProps & ThreadConfigProviderProps;
+declare const exports$4: {
+    Root: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & {
+        config?: ThreadConfig | undefined;
+    } & {
+        children?: react.ReactNode | undefined;
+    } & react.RefAttributes<HTMLDivElement>>;
+    Viewport: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & UseThreadViewportAutoScrollProps & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Messages: FC<{
+        components?: {
+            UserMessage?: ComponentType | undefined;
+            EditComposer?: ComponentType | undefined;
+            AssistantMessage?: ComponentType | undefined;
+            SystemMessage?: ComponentType | undefined;
+        };
+    }>;
+    ScrollToBottom: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+    ViewportFooter: react.ForwardRefExoticComponent<Partial<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+};
+declare const _default$3: typeof Thread & typeof exports$4;
+
+declare const UserMessage: FC;
+type UserMessageContentProps = MessagePrimitiveContentProps & ComponentPropsWithoutRef<"div">;
+declare const exports$3: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Content: react.ForwardRefExoticComponent<MessagePrimitiveContentProps & Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & react.RefAttributes<HTMLDivElement>>;
+};
+declare const _default$2: typeof UserMessage & typeof exports$3;
+
+declare const UserActionBar: FC;
+declare const exports$2: {
+    Root: react.ForwardRefExoticComponent<Partial<Omit<Omit<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+        ref?: ((instance: HTMLDivElement | 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<HTMLDivElement> | null | undefined;
+    } & {
+        asChild?: boolean;
+    }, "ref"> & UseActionBarFloatStatusProps & react.RefAttributes<HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Edit: react.ForwardRefExoticComponent<Partial<TooltipIconButtonProps> & react.RefAttributes<HTMLButtonElement>>;
+};
+declare const _default$1: typeof UserActionBar & typeof exports$2;
+
+declare const ThreadWelcome: FC;
+declare const ThreadWelcomeMessageStyled: react.ForwardRefExoticComponent<Partial<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLParagraphElement>, HTMLParagraphElement>, "ref">, "asChild">> & react.RefAttributes<HTMLParagraphElement>>;
+type ThreadWelcomeMessageProps = Omit<ComponentPropsWithoutRef<typeof ThreadWelcomeMessageStyled>, "children"> & {
+    message?: string | undefined;
+};
+type ThreadWelcomeSuggestionProps = {
+    suggestion: SuggestionConfig;
+};
+declare const exports$1: {
+    Root: react.ForwardRefExoticComponent<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & react.RefAttributes<HTMLDivElement>>;
+    Center: react.ForwardRefExoticComponent<Partial<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref">, "asChild">> & react.RefAttributes<HTMLDivElement>>;
+    Avatar: FC;
+    Message: react.ForwardRefExoticComponent<Omit<Omit<Partial<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLParagraphElement>, HTMLParagraphElement>, "ref">, "asChild">> & react.RefAttributes<HTMLParagraphElement>, "ref">, "children"> & {
+        message?: string | undefined;
+    } & react.RefAttributes<HTMLParagraphElement>>;
+    Suggestions: FC;
+    Suggestion: FC<ThreadWelcomeSuggestionProps>;
+};
+declare const _default: typeof ThreadWelcome & typeof exports$1;
+
+declare const exports: {
+    Text: FC<TextContentPartProps>;
+};
+
 declare class ProxyConfigProvider {
     private _providers;
     getModelConfig(): ModelConfig;
@@ -672,6 +1029,10 @@ declare class MessageRepository {
     private performOp;
     getMessages(): ThreadMessage[];
     addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
+    getMessage(messageId: string): {
+        parentId: string | null;
+        message: ThreadMessage;
+    };
     appendOptimisticMessage(parentId: string | null, message: Omit<ThreadMessage, "id" | "createdAt">): string;
     deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
     getBranches(messageId: string): string[];
@@ -687,9 +1048,10 @@ type internal_MessageRepository = MessageRepository;
 declare const internal_MessageRepository: typeof MessageRepository;
 type internal_ProxyConfigProvider = ProxyConfigProvider;
 declare const internal_ProxyConfigProvider: typeof ProxyConfigProvider;
+declare const internal_TooltipIconButton: typeof TooltipIconButton;
 declare const internal_useSmooth: typeof useSmooth;
 declare namespace internal {
-  export { internal_BaseAssistantRuntime as BaseAssistantRuntime, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, internal_useSmooth as useSmooth };
+  export { internal_BaseAssistantRuntime as BaseAssistantRuntime, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, internal_TooltipIconButton as TooltipIconButton, internal_useSmooth as useSmooth };
 }
 
-export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type AppendContentPart, type AppendMessage, type AssistantContentPart, type AssistantContextValue, type AssistantMessage, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ComposerContextValue, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, type ComposerState, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, type ContentPartState, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, type ModelConfig, type ModelConfigProvider, type ReactThreadRuntime, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, type ThreadContextValue, type ThreadMessage, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, 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, useAppendMessage, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartText, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };
+export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type AddToolResultOptions, type AppendContentPart, type AppendMessage, _default$9 as AssistantActionBar, type AssistantActionsState, type AssistantContextValue, _default$8 as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$7 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, type AssistantToolUIsState, _default$6 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, _default$5 as Composer, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, type ComposerState, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, type ContentPartState, type CoreAssistantContentPart, type CoreAssistantMessage, type CoreThreadMessage, type CoreUserContentPart, type CoreUserMessage, _default$4 as EditComposer, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, type MessageStatus, type MessageUtilsState, type ModelConfig, type ModelConfigProvider, type ReactThreadRuntime, type StringsConfig, type SuggestionConfig, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, type ThreadActionsState, type ThreadAssistantContentPart, type ThreadAssistantMessage, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, type ThreadMessage, type ThreadMessagesState, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, type ThreadRuntime, type ThreadState, type ThreadSystemMessage, type ThreadUserContentPart, type ThreadUserMessage, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UseActionBarCopyProps, _default$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, makeAssistantTool, makeAssistantToolUI, useActionBarCopy, useActionBarEdit, useActionBarReload, useAppendMessage, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartText, useEdgeRuntime, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };