@assistant-ui/react 0.2.4 → 0.3.1

package/dist/index.d.mts

@@ -8,14 +8,14 @@ import { TextareaAutosizeProps } from 'react-textarea-autosize';
 
 type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
 type Tool<TArgs = unknown, TResult = unknown> = {
-    description?: string;
+    description?: string | undefined;
     parameters: z.ZodSchema<TArgs>;
     execute: ToolExecuteFunction<TArgs, TResult>;
 };
 type ModelConfig = {
-    priority?: number;
-    system?: string;
-    tools?: Record<string, Tool<any, any>>;
+    priority?: number | undefined;
+    system?: string | undefined;
+    tools?: Record<string, Tool<any, any>> | undefined;
 };
 type ModelConfigProvider = () => ModelConfig;
 
@@ -82,6 +82,12 @@ type ChatModelAdapter = {
 
 declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRuntime> implements AssistantRuntime {
     private _thread;
+    get capabilities(): Readonly<{
+        edit: boolean;
+        reload: boolean;
+        cancel: boolean;
+        copy: boolean;
+    }>;
     constructor(_thread: TThreadRuntime);
     private _unsubscribe;
     get thread(): TThreadRuntime;
@@ -114,7 +120,13 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     adapter: ChatModelAdapter;
     private _subscriptions;
     private abortController;
-    private repository;
+    private readonly repository;
+    readonly capabilities: Readonly<{
+        edit: true;
+        reload: true;
+        cancel: true;
+        copy: true;
+    }>;
     get messages(): ThreadMessage[];
     get isRunning(): boolean;
     constructor(_configProviders: Set<ModelConfigProvider>, adapter: ChatModelAdapter);
@@ -131,6 +143,12 @@ declare class LocalThreadRuntime implements ThreadRuntime {
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
 
 type ThreadActionsState = Readonly<{
+    capabilities: Readonly<{
+        edit: boolean;
+        reload: boolean;
+        cancel: boolean;
+        copy: boolean;
+    }>;
     getBranches: (messageId: string) => readonly string[];
     switchToBranch: (branchId: string) => void;
     append: (message: AppendMessage) => void;
@@ -184,7 +202,7 @@ type AssistantModelConfigState = Readonly<{
 
 type ContentPartState = Readonly<{
     status: "in_progress" | "done" | "error";
-    part: ThreadMessage["content"][number];
+    part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
 }>;
 
 type MessageState = Readonly<{
@@ -202,10 +220,11 @@ 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: () => boolean;
+    cancel: () => void;
 }>;
 
 type ThreadState = Readonly<{
@@ -252,7 +271,6 @@ type ComposerContextValue = {
 declare const useComposerContext: () => ComposerContextValue;
 
 type MessageUtilsState = Readonly<{
-    inProgressIndicator: HTMLSpanElement;
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
@@ -272,16 +290,17 @@ type ContentPartContextValue = {
 declare const useContentPartContext: () => ContentPartContextValue;
 
 type ComposerState = BaseComposerState & Readonly<{
+    canCancel: boolean;
     isEditing: true;
     send: () => void;
-    cancel: () => boolean;
+    cancel: () => void;
     focus: () => void;
     onFocus: (listener: () => void) => Unsubscribe;
 }>;
 
 type CreateAppendMessage = string | {
     parentId?: string | null | undefined;
-    role?: AppendMessage["role"];
+    role?: AppendMessage["role"] | undefined;
     content: AppendMessage["content"];
 };
 declare const useAppendMessage: () => (message: CreateAppendMessage) => void;
@@ -290,7 +309,7 @@ declare const useSwitchToNewThread: () => () => void;
 
 type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {
     toolName: string;
-    render?: ToolCallContentPartComponent<TArgs, TResult>;
+    render?: ToolCallContentPartComponent<TArgs, TResult> | undefined;
 };
 declare const useAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => void;
 
@@ -307,7 +326,7 @@ declare const makeAssistantToolUI: <TArgs, TResult>(tool: AssistantToolUIProps<T
 declare const useAssistantInstructions: (instruction: string) => void;
 
 type UseActionBarCopyProps = {
-    copiedDuration?: number;
+    copiedDuration?: number | undefined;
 };
 declare const useActionBarCopy: ({ copiedDuration, }?: UseActionBarCopyProps) => (() => void) | null;
 
@@ -341,7 +360,10 @@ declare const useContentPartDisplay: () => react.ReactNode;
 
 declare const useContentPartImage: () => string;
 
-declare const useContentPartText: () => string;
+declare const useContentPartText: () => Readonly<{
+    status: "in_progress" | "done" | "error";
+    part: TextContentPart;
+}>;
 
 type MessageIfFilters = {
     user: boolean | undefined;
@@ -367,7 +389,7 @@ declare const useThreadScrollToBottom: () => (() => void) | null;
 type UseApplyThreadSuggestionProps = {
     prompt: string;
     method: "replace";
-    autoSend?: boolean;
+    autoSend?: boolean | undefined;
 };
 declare const useThreadSuggestion: ({ prompt, autoSend, }: UseApplyThreadSuggestionProps) => (() => void) | null;
 
@@ -393,9 +415,7 @@ declare const ActionBarPrimitiveCopy: react.ForwardRefExoticComponent<Omit<Omit<
     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"> & {
-    copiedDuration?: number;
-} & react.RefAttributes<HTMLButtonElement>>;
+}, "ref"> & UseActionBarCopyProps & react.RefAttributes<HTMLButtonElement>>;
 
 type ActionBarPrimitiveReloadProps = ActionButtonProps<typeof useActionBarReload>;
 declare const ActionBarPrimitiveReload: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -422,10 +442,10 @@ type AssistantModalPrimitiveTriggerProps = ComponentPropsWithoutRef<typeof Popov
 declare const AssistantModalPrimitiveTrigger: react.ForwardRefExoticComponent<Omit<PopoverPrimitive.PopoverTriggerProps & react.RefAttributes<HTMLButtonElement>, "ref"> & react.RefAttributes<HTMLButtonElement>>;
 
 type AssistantModalPrimitiveContentProps = ComponentPropsWithoutRef<typeof PopoverPrimitive.Content> & {
-    dissmissOnInteractOutside?: boolean;
+    dissmissOnInteractOutside?: boolean | undefined;
 };
 declare const AssistantModalPrimitiveContent: react.ForwardRefExoticComponent<Omit<PopoverPrimitive.PopoverContentProps & react.RefAttributes<HTMLDivElement>, "ref"> & {
-    dissmissOnInteractOutside?: boolean;
+    dissmissOnInteractOutside?: boolean | undefined;
 } & react.RefAttributes<HTMLDivElement>>;
 
 declare const AssistantModalPrimitiveAnchor: react.ForwardRefExoticComponent<Omit<PopoverPrimitive.PopoverAnchorProps & react.RefAttributes<HTMLDivElement>, "ref"> & react.RefAttributes<HTMLDivElement>>;
@@ -456,14 +476,14 @@ declare const BranchPickerPrimitiveNumber: FC<BranchPickerPrimitiveNumberProps>;
 
 type PrimitiveDivProps$3 = ComponentPropsWithoutRef<typeof Primitive.div>;
 type BranchPickerPrimitiveRootProps = PrimitiveDivProps$3 & {
-    hideWhenSingleBranch?: boolean;
+    hideWhenSingleBranch?: boolean | undefined;
 };
 declare const BranchPickerPrimitiveRoot: 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"> & {
-    hideWhenSingleBranch?: boolean;
+    hideWhenSingleBranch?: boolean | undefined;
 } & react.RefAttributes<HTMLDivElement>>;
 
 declare namespace index$4 {
@@ -479,10 +499,10 @@ declare const ComposerPrimitiveRoot: react.ForwardRefExoticComponent<Omit<Omit<r
 }, "ref"> & react.RefAttributes<HTMLFormElement>>;
 
 type ComposerPrimitiveInputProps = TextareaAutosizeProps & {
-    asChild?: boolean;
+    asChild?: boolean | undefined;
 };
 declare const ComposerPrimitiveInput: react.ForwardRefExoticComponent<TextareaAutosizeProps & {
-    asChild?: boolean;
+    asChild?: boolean | undefined;
 } & react.RefAttributes<HTMLTextAreaElement>>;
 
 type PrimitiveButtonProps = ComponentPropsWithoutRef<typeof Primitive.button>;
@@ -507,12 +527,9 @@ declare namespace index$3 {
   export { ComposerPrimitiveCancel as Cancel, ComposerPrimitiveIf as If, ComposerPrimitiveInput as Input, ComposerPrimitiveRoot as Root, ComposerPrimitiveSend as Send };
 }
 
-type ContentPartPrimitiveInProgressIndicatorProps = {};
-declare const ContentPartPrimitiveInProgressIndicator: FC<ContentPartPrimitiveInProgressIndicatorProps>;
-
-type PrimitiveSpanProps$1 = ComponentPropsWithoutRef<typeof Primitive.p>;
+type PrimitiveSpanProps$1 = ComponentPropsWithoutRef<typeof Primitive.span>;
 type ContentPartPrimitiveTextProps = Omit<PrimitiveSpanProps$1, "children">;
-declare const ContentPartPrimitiveText: react.ForwardRefExoticComponent<ContentPartPrimitiveTextProps & react.RefAttributes<HTMLParagraphElement>>;
+declare const ContentPartPrimitiveText: react.ForwardRefExoticComponent<ContentPartPrimitiveTextProps & react.RefAttributes<HTMLSpanElement>>;
 
 type PrimitiveImageProps = ComponentPropsWithoutRef<typeof Primitive.img>;
 type ContentPartPrimitiveImageProps = PrimitiveImageProps;
@@ -525,8 +542,11 @@ declare const ContentPartPrimitiveImage: react.ForwardRefExoticComponent<Omit<Om
 type ContentPartPrimitiveDisplayProps = {};
 declare const ContentPartPrimitiveDisplay: FC<ContentPartPrimitiveDisplayProps>;
 
+type ContentPartPrimitiveInProgressProps = PropsWithChildren;
+declare const ContentPartPrimitiveInProgress: FC<ContentPartPrimitiveInProgressProps>;
+
 declare namespace index$2 {
-  export { ContentPartPrimitiveDisplay as Display, ContentPartPrimitiveImage as Image, ContentPartPrimitiveInProgressIndicator as InProgressIndicator, ContentPartPrimitiveText as Text };
+  export { ContentPartPrimitiveDisplay as Display, ContentPartPrimitiveImage as Image, ContentPartPrimitiveInProgress as InProgress, ContentPartPrimitiveText as Text };
 }
 
 type PrimitiveDivProps$2 = ComponentPropsWithoutRef<typeof Primitive.div>;
@@ -542,24 +562,23 @@ declare const MessagePrimitiveIf: FC<MessagePrimitiveIfProps>;
 
 type MessagePrimitiveContentProps = {
     components?: {
-        Text?: TextContentPartComponent;
-        Image?: ImageContentPartComponent;
-        UI?: UIContentPartComponent;
+        Text?: TextContentPartComponent | undefined;
+        Image?: ImageContentPartComponent | undefined;
+        UI?: UIContentPartComponent | undefined;
         tools?: {
-            by_name?: Record<string, ToolCallContentPartComponent>;
-            Fallback?: ComponentType<ToolCallContentPartProps>;
+            by_name?: Record<string, ToolCallContentPartComponent | undefined> | undefined;
+            Fallback?: ComponentType<ToolCallContentPartProps> | undefined;
         };
-    };
+    } | undefined;
 };
 declare const MessagePrimitiveContent: FC<MessagePrimitiveContentProps>;
 
 type PrimitiveSpanProps = ComponentPropsWithoutRef<typeof Primitive.span>;
 type MessagePrimitiveInProgressProps = PrimitiveSpanProps;
-declare const MessagePrimitiveInProgress: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLSpanElement>, HTMLSpanElement>, "ref"> & {
-    ref?: ((instance: HTMLSpanElement | 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<HTMLSpanElement> | null | undefined;
-} & {
-    asChild?: boolean;
-}, "ref"> & react.RefAttributes<HTMLSpanElement>>;
+/**
+ * @deprecated Define a custom Text renderer via ContentPartPrimitiveInProgress instead.
+ */
+declare const MessagePrimitiveInProgress: FC<MessagePrimitiveInProgressProps>;
 
 declare namespace index$1 {
   export { MessagePrimitiveContent as Content, MessagePrimitiveIf as If, MessagePrimitiveInProgress as InProgress, MessagePrimitiveRoot as Root };
@@ -596,17 +615,17 @@ declare const ThreadPrimitiveViewport: react.ForwardRefExoticComponent<Omit<Omit
 type ThreadPrimitiveMessagesProps = {
     components: {
         Message: ComponentType;
-        UserMessage?: ComponentType;
-        EditComposer?: ComponentType;
-        AssistantMessage?: ComponentType;
+        UserMessage?: ComponentType | undefined;
+        EditComposer?: ComponentType | undefined;
+        AssistantMessage?: ComponentType | undefined;
     } | {
-        Message?: ComponentType;
+        Message?: ComponentType | undefined;
         UserMessage: ComponentType;
-        EditComposer?: ComponentType;
+        EditComposer?: ComponentType | undefined;
         AssistantMessage: ComponentType;
     };
 };
-declare const ThreadPrimitiveMessages: FC<ThreadPrimitiveMessagesProps>;
+declare const ThreadPrimitiveMessages: react.NamedExoticComponent<ThreadPrimitiveMessagesProps>;
 
 type ThreadPrimitiveScrollToBottomProps = ActionButtonProps<typeof useThreadScrollToBottom>;
 declare const ThreadPrimitiveScrollToBottom: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -620,11 +639,7 @@ declare const ThreadPrimitiveSuggestion: react.ForwardRefExoticComponent<Omit<Om
     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"> & {
-    prompt: string;
-    method: "replace";
-    autoSend?: boolean;
-} & react.RefAttributes<HTMLButtonElement>>;
+}, "ref"> & UseApplyThreadSuggestionProps & react.RefAttributes<HTMLButtonElement>>;
 
 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 };
@@ -660,4 +675,4 @@ declare namespace internal {
   export { internal_BaseAssistantRuntime as BaseAssistantRuntime, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider };
 }
 
-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 ContentPartPrimitiveInProgressIndicatorProps, 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 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 };

package/dist/index.d.ts

@@ -8,14 +8,14 @@ import { TextareaAutosizeProps } from 'react-textarea-autosize';
 
 type ToolExecuteFunction<TArgs, TResult> = (args: TArgs) => TResult | Promise<TResult>;
 type Tool<TArgs = unknown, TResult = unknown> = {
-    description?: string;
+    description?: string | undefined;
     parameters: z.ZodSchema<TArgs>;
     execute: ToolExecuteFunction<TArgs, TResult>;
 };
 type ModelConfig = {
-    priority?: number;
-    system?: string;
-    tools?: Record<string, Tool<any, any>>;
+    priority?: number | undefined;
+    system?: string | undefined;
+    tools?: Record<string, Tool<any, any>> | undefined;
 };
 type ModelConfigProvider = () => ModelConfig;
 
@@ -82,6 +82,12 @@ type ChatModelAdapter = {
 
 declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRuntime> implements AssistantRuntime {
     private _thread;
+    get capabilities(): Readonly<{
+        edit: boolean;
+        reload: boolean;
+        cancel: boolean;
+        copy: boolean;
+    }>;
     constructor(_thread: TThreadRuntime);
     private _unsubscribe;
     get thread(): TThreadRuntime;
@@ -114,7 +120,13 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     adapter: ChatModelAdapter;
     private _subscriptions;
     private abortController;
-    private repository;
+    private readonly repository;
+    readonly capabilities: Readonly<{
+        edit: true;
+        reload: true;
+        cancel: true;
+        copy: true;
+    }>;
     get messages(): ThreadMessage[];
     get isRunning(): boolean;
     constructor(_configProviders: Set<ModelConfigProvider>, adapter: ChatModelAdapter);
@@ -131,6 +143,12 @@ declare class LocalThreadRuntime implements ThreadRuntime {
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
 
 type ThreadActionsState = Readonly<{
+    capabilities: Readonly<{
+        edit: boolean;
+        reload: boolean;
+        cancel: boolean;
+        copy: boolean;
+    }>;
     getBranches: (messageId: string) => readonly string[];
     switchToBranch: (branchId: string) => void;
     append: (message: AppendMessage) => void;
@@ -184,7 +202,7 @@ type AssistantModelConfigState = Readonly<{
 
 type ContentPartState = Readonly<{
     status: "in_progress" | "done" | "error";
-    part: ThreadMessage["content"][number];
+    part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
 }>;
 
 type MessageState = Readonly<{
@@ -202,10 +220,11 @@ 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: () => boolean;
+    cancel: () => void;
 }>;
 
 type ThreadState = Readonly<{
@@ -252,7 +271,6 @@ type ComposerContextValue = {
 declare const useComposerContext: () => ComposerContextValue;
 
 type MessageUtilsState = Readonly<{
-    inProgressIndicator: HTMLSpanElement;
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
@@ -272,16 +290,17 @@ type ContentPartContextValue = {
 declare const useContentPartContext: () => ContentPartContextValue;
 
 type ComposerState = BaseComposerState & Readonly<{
+    canCancel: boolean;
     isEditing: true;
     send: () => void;
-    cancel: () => boolean;
+    cancel: () => void;
     focus: () => void;
     onFocus: (listener: () => void) => Unsubscribe;
 }>;
 
 type CreateAppendMessage = string | {
     parentId?: string | null | undefined;
-    role?: AppendMessage["role"];
+    role?: AppendMessage["role"] | undefined;
     content: AppendMessage["content"];
 };
 declare const useAppendMessage: () => (message: CreateAppendMessage) => void;
@@ -290,7 +309,7 @@ declare const useSwitchToNewThread: () => () => void;
 
 type AssistantToolProps<TArgs, TResult> = Tool<TArgs, TResult> & {
     toolName: string;
-    render?: ToolCallContentPartComponent<TArgs, TResult>;
+    render?: ToolCallContentPartComponent<TArgs, TResult> | undefined;
 };
 declare const useAssistantTool: <TArgs, TResult>(tool: AssistantToolProps<TArgs, TResult>) => void;
 
@@ -307,7 +326,7 @@ declare const makeAssistantToolUI: <TArgs, TResult>(tool: AssistantToolUIProps<T
 declare const useAssistantInstructions: (instruction: string) => void;
 
 type UseActionBarCopyProps = {
-    copiedDuration?: number;
+    copiedDuration?: number | undefined;
 };
 declare const useActionBarCopy: ({ copiedDuration, }?: UseActionBarCopyProps) => (() => void) | null;
 
@@ -341,7 +360,10 @@ declare const useContentPartDisplay: () => react.ReactNode;
 
 declare const useContentPartImage: () => string;
 
-declare const useContentPartText: () => string;
+declare const useContentPartText: () => Readonly<{
+    status: "in_progress" | "done" | "error";
+    part: TextContentPart;
+}>;
 
 type MessageIfFilters = {
     user: boolean | undefined;
@@ -367,7 +389,7 @@ declare const useThreadScrollToBottom: () => (() => void) | null;
 type UseApplyThreadSuggestionProps = {
     prompt: string;
     method: "replace";
-    autoSend?: boolean;
+    autoSend?: boolean | undefined;
 };
 declare const useThreadSuggestion: ({ prompt, autoSend, }: UseApplyThreadSuggestionProps) => (() => void) | null;
 
@@ -393,9 +415,7 @@ declare const ActionBarPrimitiveCopy: react.ForwardRefExoticComponent<Omit<Omit<
     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"> & {
-    copiedDuration?: number;
-} & react.RefAttributes<HTMLButtonElement>>;
+}, "ref"> & UseActionBarCopyProps & react.RefAttributes<HTMLButtonElement>>;
 
 type ActionBarPrimitiveReloadProps = ActionButtonProps<typeof useActionBarReload>;
 declare const ActionBarPrimitiveReload: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -422,10 +442,10 @@ type AssistantModalPrimitiveTriggerProps = ComponentPropsWithoutRef<typeof Popov
 declare const AssistantModalPrimitiveTrigger: react.ForwardRefExoticComponent<Omit<PopoverPrimitive.PopoverTriggerProps & react.RefAttributes<HTMLButtonElement>, "ref"> & react.RefAttributes<HTMLButtonElement>>;
 
 type AssistantModalPrimitiveContentProps = ComponentPropsWithoutRef<typeof PopoverPrimitive.Content> & {
-    dissmissOnInteractOutside?: boolean;
+    dissmissOnInteractOutside?: boolean | undefined;
 };
 declare const AssistantModalPrimitiveContent: react.ForwardRefExoticComponent<Omit<PopoverPrimitive.PopoverContentProps & react.RefAttributes<HTMLDivElement>, "ref"> & {
-    dissmissOnInteractOutside?: boolean;
+    dissmissOnInteractOutside?: boolean | undefined;
 } & react.RefAttributes<HTMLDivElement>>;
 
 declare const AssistantModalPrimitiveAnchor: react.ForwardRefExoticComponent<Omit<PopoverPrimitive.PopoverAnchorProps & react.RefAttributes<HTMLDivElement>, "ref"> & react.RefAttributes<HTMLDivElement>>;
@@ -456,14 +476,14 @@ declare const BranchPickerPrimitiveNumber: FC<BranchPickerPrimitiveNumberProps>;
 
 type PrimitiveDivProps$3 = ComponentPropsWithoutRef<typeof Primitive.div>;
 type BranchPickerPrimitiveRootProps = PrimitiveDivProps$3 & {
-    hideWhenSingleBranch?: boolean;
+    hideWhenSingleBranch?: boolean | undefined;
 };
 declare const BranchPickerPrimitiveRoot: 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"> & {
-    hideWhenSingleBranch?: boolean;
+    hideWhenSingleBranch?: boolean | undefined;
 } & react.RefAttributes<HTMLDivElement>>;
 
 declare namespace index$4 {
@@ -479,10 +499,10 @@ declare const ComposerPrimitiveRoot: react.ForwardRefExoticComponent<Omit<Omit<r
 }, "ref"> & react.RefAttributes<HTMLFormElement>>;
 
 type ComposerPrimitiveInputProps = TextareaAutosizeProps & {
-    asChild?: boolean;
+    asChild?: boolean | undefined;
 };
 declare const ComposerPrimitiveInput: react.ForwardRefExoticComponent<TextareaAutosizeProps & {
-    asChild?: boolean;
+    asChild?: boolean | undefined;
 } & react.RefAttributes<HTMLTextAreaElement>>;
 
 type PrimitiveButtonProps = ComponentPropsWithoutRef<typeof Primitive.button>;
@@ -507,12 +527,9 @@ declare namespace index$3 {
   export { ComposerPrimitiveCancel as Cancel, ComposerPrimitiveIf as If, ComposerPrimitiveInput as Input, ComposerPrimitiveRoot as Root, ComposerPrimitiveSend as Send };
 }
 
-type ContentPartPrimitiveInProgressIndicatorProps = {};
-declare const ContentPartPrimitiveInProgressIndicator: FC<ContentPartPrimitiveInProgressIndicatorProps>;
-
-type PrimitiveSpanProps$1 = ComponentPropsWithoutRef<typeof Primitive.p>;
+type PrimitiveSpanProps$1 = ComponentPropsWithoutRef<typeof Primitive.span>;
 type ContentPartPrimitiveTextProps = Omit<PrimitiveSpanProps$1, "children">;
-declare const ContentPartPrimitiveText: react.ForwardRefExoticComponent<ContentPartPrimitiveTextProps & react.RefAttributes<HTMLParagraphElement>>;
+declare const ContentPartPrimitiveText: react.ForwardRefExoticComponent<ContentPartPrimitiveTextProps & react.RefAttributes<HTMLSpanElement>>;
 
 type PrimitiveImageProps = ComponentPropsWithoutRef<typeof Primitive.img>;
 type ContentPartPrimitiveImageProps = PrimitiveImageProps;
@@ -525,8 +542,11 @@ declare const ContentPartPrimitiveImage: react.ForwardRefExoticComponent<Omit<Om
 type ContentPartPrimitiveDisplayProps = {};
 declare const ContentPartPrimitiveDisplay: FC<ContentPartPrimitiveDisplayProps>;
 
+type ContentPartPrimitiveInProgressProps = PropsWithChildren;
+declare const ContentPartPrimitiveInProgress: FC<ContentPartPrimitiveInProgressProps>;
+
 declare namespace index$2 {
-  export { ContentPartPrimitiveDisplay as Display, ContentPartPrimitiveImage as Image, ContentPartPrimitiveInProgressIndicator as InProgressIndicator, ContentPartPrimitiveText as Text };
+  export { ContentPartPrimitiveDisplay as Display, ContentPartPrimitiveImage as Image, ContentPartPrimitiveInProgress as InProgress, ContentPartPrimitiveText as Text };
 }
 
 type PrimitiveDivProps$2 = ComponentPropsWithoutRef<typeof Primitive.div>;
@@ -542,24 +562,23 @@ declare const MessagePrimitiveIf: FC<MessagePrimitiveIfProps>;
 
 type MessagePrimitiveContentProps = {
     components?: {
-        Text?: TextContentPartComponent;
-        Image?: ImageContentPartComponent;
-        UI?: UIContentPartComponent;
+        Text?: TextContentPartComponent | undefined;
+        Image?: ImageContentPartComponent | undefined;
+        UI?: UIContentPartComponent | undefined;
         tools?: {
-            by_name?: Record<string, ToolCallContentPartComponent>;
-            Fallback?: ComponentType<ToolCallContentPartProps>;
+            by_name?: Record<string, ToolCallContentPartComponent | undefined> | undefined;
+            Fallback?: ComponentType<ToolCallContentPartProps> | undefined;
         };
-    };
+    } | undefined;
 };
 declare const MessagePrimitiveContent: FC<MessagePrimitiveContentProps>;
 
 type PrimitiveSpanProps = ComponentPropsWithoutRef<typeof Primitive.span>;
 type MessagePrimitiveInProgressProps = PrimitiveSpanProps;
-declare const MessagePrimitiveInProgress: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLSpanElement>, HTMLSpanElement>, "ref"> & {
-    ref?: ((instance: HTMLSpanElement | 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<HTMLSpanElement> | null | undefined;
-} & {
-    asChild?: boolean;
-}, "ref"> & react.RefAttributes<HTMLSpanElement>>;
+/**
+ * @deprecated Define a custom Text renderer via ContentPartPrimitiveInProgress instead.
+ */
+declare const MessagePrimitiveInProgress: FC<MessagePrimitiveInProgressProps>;
 
 declare namespace index$1 {
   export { MessagePrimitiveContent as Content, MessagePrimitiveIf as If, MessagePrimitiveInProgress as InProgress, MessagePrimitiveRoot as Root };
@@ -596,17 +615,17 @@ declare const ThreadPrimitiveViewport: react.ForwardRefExoticComponent<Omit<Omit
 type ThreadPrimitiveMessagesProps = {
     components: {
         Message: ComponentType;
-        UserMessage?: ComponentType;
-        EditComposer?: ComponentType;
-        AssistantMessage?: ComponentType;
+        UserMessage?: ComponentType | undefined;
+        EditComposer?: ComponentType | undefined;
+        AssistantMessage?: ComponentType | undefined;
     } | {
-        Message?: ComponentType;
+        Message?: ComponentType | undefined;
         UserMessage: ComponentType;
-        EditComposer?: ComponentType;
+        EditComposer?: ComponentType | undefined;
         AssistantMessage: ComponentType;
     };
 };
-declare const ThreadPrimitiveMessages: FC<ThreadPrimitiveMessagesProps>;
+declare const ThreadPrimitiveMessages: react.NamedExoticComponent<ThreadPrimitiveMessagesProps>;
 
 type ThreadPrimitiveScrollToBottomProps = ActionButtonProps<typeof useThreadScrollToBottom>;
 declare const ThreadPrimitiveScrollToBottom: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -620,11 +639,7 @@ declare const ThreadPrimitiveSuggestion: react.ForwardRefExoticComponent<Omit<Om
     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"> & {
-    prompt: string;
-    method: "replace";
-    autoSend?: boolean;
-} & react.RefAttributes<HTMLButtonElement>>;
+}, "ref"> & UseApplyThreadSuggestionProps & react.RefAttributes<HTMLButtonElement>>;
 
 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 };
@@ -660,4 +675,4 @@ declare namespace internal {
   export { internal_BaseAssistantRuntime as BaseAssistantRuntime, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider };
 }
 
-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 ContentPartPrimitiveInProgressIndicatorProps, 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 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 };