@assistant-ui/react 0.5.66 → 0.5.67

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-import { C as ContentPartStatus, T as TextContentPart, I as ImageContentPart, U as UIContentPart, a as ToolCallContentPart, b as ToolCallContentPartStatus, c as ThreadMessage, d as CoreMessage, e as ThreadComposerAttachment, A as AppendMessage, M as ModelConfig, f as ModelConfigProvider, g as ThreadAssistantContentPart, h as MessageStatus, i as ThreadRoundtrip, j as MessageAttachment, k as Tool, l as CoreToolCallContentPart, m as CreateEdgeRuntimeAPIOptions, n as ThreadUserContentPart } from './edge-D7ch2oPK.mjs';
-export { s as CoreAssistantContentPart, v as CoreAssistantMessage, t as CoreSystemMessage, r as CoreUserContentPart, u as CoreUserMessage, E as EdgeRuntimeRequestOptions, p as ThreadAssistantMessage, o as ThreadSystemMessage, q as ThreadUserMessage } from './edge-D7ch2oPK.mjs';
+import { T as ThreadComposerAttachment, a as ThreadMessage, C as CoreMessage, A as AppendMessage, M as ModelConfig, b as ModelConfigProvider, c as ThreadAssistantContentPart, d as MessageStatus, e as ThreadRoundtrip, f as MessageAttachment, g as Tool, h as TextContentPart, I as ImageContentPart, i as ToolCallContentPart, j as CoreToolCallContentPart, U as UIContentPart, k as CreateEdgeRuntimeAPIOptions, l as ThreadUserContentPart, m as ContentPartStatus, n as ToolCallContentPartStatus } from './edge-BDAlTCC1.mjs';
+export { s as CoreAssistantContentPart, v as CoreAssistantMessage, t as CoreSystemMessage, r as CoreUserContentPart, u as CoreUserMessage, E as EdgeRuntimeRequestOptions, p as ThreadAssistantMessage, o as ThreadSystemMessage, q as ThreadUserMessage } from './edge-BDAlTCC1.mjs';
 import * as react from 'react';
 import { ComponentType, PropsWithChildren, FC, ComponentPropsWithoutRef, ElementType, ReactNode } from 'react';
 import { StoreApi, UseBoundStore } from 'zustand';
@@ -14,31 +14,24 @@ import * as class_variance_authority_types from 'class-variance-authority/types'
 import 'json-schema';
 import 'zod';
 
-type EmptyContentPartProps = {
-    status: ContentPartStatus;
-};
-type EmptyContentPartComponent = ComponentType<EmptyContentPartProps>;
-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 extends Record<string, unknown> = any, TResult = unknown> = {
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: ToolCallContentPartStatus;
-    addResult: (result: any) => void;
-};
-type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+type ComposerRuntimeCore = Readonly<{
+    attachmentAccept: string;
+    attachments: readonly ThreadComposerAttachment[];
+    addAttachment: (file: File) => Promise<void>;
+    removeAttachment: (attachmentId: string) => Promise<void>;
+    isEditing: boolean;
+    canCancel: boolean;
+    isEmpty: boolean;
+    text: string;
+    setText: (value: string) => void;
+    /**
+     * @deprecated This method will be removed in 0.6.0. Submit feedback if you need this functionality.
+     */
+    reset: () => void;
+    send: () => void;
+    cancel: () => void;
+    subscribe: (callback: () => void) => Unsubscribe;
+}>;
 
 type Unsubscribe = () => void;
 
@@ -91,21 +84,6 @@ declare class MessageRepository {
     import({ headId, messages }: ExportedMessageRepository): void;
 }
 
-type ThreadComposerRuntimeCore = Readonly<{
-    attachmentAccept: string;
-    attachments: ThreadComposerAttachment[];
-    addAttachment: (file: File) => Promise<void>;
-    removeAttachment: (attachmentId: string) => Promise<void>;
-    canCancel: boolean;
-    isEmpty: boolean;
-    text: string;
-    setText: (value: string) => void;
-    reset: () => void;
-    send: () => void;
-    cancel: () => void;
-    subscribe: (callback: () => void) => Unsubscribe;
-}>;
-
 type RuntimeCapabilities = Readonly<{
     switchToBranch: boolean;
     edit: boolean;
@@ -136,7 +114,9 @@ type ThreadRuntimeCore = Readonly<{
     speak: (messageId: string) => SpeechSynthesisAdapter.Utterance;
     submitFeedback: (feedback: SubmitFeedbackOptions) => void;
     getModelConfig: () => ModelConfig;
-    composer: ThreadComposerRuntimeCore;
+    composer: ComposerRuntimeCore;
+    getEditComposer: (messageId: string) => ComposerRuntimeCore | undefined;
+    beginEdit: (messageId: string) => void;
     capabilities: Readonly<RuntimeCapabilities>;
     threadId: string;
     isDisabled: boolean;
@@ -214,37 +194,69 @@ type AttachmentAdapter = {
     send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
 };
 
-declare class BaseThreadComposerRuntimeCore implements ThreadComposerRuntimeCore {
-    private runtime;
-    private _attachmentAdapter?;
+declare class SimpleImageAttachmentAdapter implements AttachmentAdapter {
+    accept: string;
+    add(state: {
+        file: File;
+    }): Promise<ThreadComposerAttachment>;
+    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
+    remove(): Promise<void>;
+}
+
+declare class SimpleTextAttachmentAdapter implements AttachmentAdapter {
+    accept: string;
+    add(state: {
+        file: File;
+    }): Promise<ThreadComposerAttachment>;
+    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
+    remove(): Promise<void>;
+}
+
+declare class CompositeAttachmentAdapter implements AttachmentAdapter {
+    private _adapters;
+    accept: string;
+    constructor(adapters: AttachmentAdapter[]);
+    add(state: {
+        file: File;
+    }): Promise<ThreadComposerAttachment>;
+    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
+    remove(attachment: ThreadComposerAttachment): Promise<void>;
+}
+
+declare abstract class BaseComposerRuntimeCore implements ComposerRuntimeCore {
+    readonly isEditing = true;
     attachmentAccept: string;
-    get isEmpty(): boolean;
-    private _canCancel;
-    get canCancel(): boolean;
-    constructor(runtime: {
-        capabilities: Readonly<RuntimeCapabilities>;
-        messages: ThreadRuntimeCore["messages"];
-        append: (message: AppendMessage) => void;
-        cancelRun: () => void;
-        subscribe: (callback: () => void) => Unsubscribe;
-    });
-    connect(): Unsubscribe;
-    setAttachmentAdapter(adapter: AttachmentAdapter | undefined): boolean;
     private _attachments;
-    get attachments(): ThreadComposerAttachment[];
-    addAttachment(file: File): Promise<void>;
-    removeAttachment(attachmentId: string): Promise<void>;
+    protected set attachments(value: readonly ThreadComposerAttachment[]);
+    get attachments(): readonly ThreadComposerAttachment[];
+    abstract get canCancel(): boolean;
+    get isEmpty(): boolean;
     private _text;
     get text(): string;
     setText(value: string): void;
     reset(): void;
     send(): Promise<void>;
-    cancel(): Promise<void>;
+    abstract handleSend(message: Omit<AppendMessage, "parentId">): void;
+    abstract cancel(): void;
+    protected _attachmentAdapter?: AttachmentAdapter | undefined;
+    setAttachmentAdapter(adapter: AttachmentAdapter | undefined): void;
+    addAttachment(file: File): Promise<void>;
+    removeAttachment(attachmentId: string): Promise<void>;
     private _subscriptions;
-    private notifySubscribers;
+    protected notifySubscribers(): void;
     subscribe(callback: () => void): Unsubscribe;
 }
 
+declare class DefaultThreadComposerRuntimeCore extends BaseComposerRuntimeCore {
+    private runtime;
+    private _canCancel;
+    get canCancel(): boolean;
+    constructor(runtime: Omit<ThreadRuntimeCore, "composer">);
+    connect(): Unsubscribe;
+    handleSend(message: Omit<AppendMessage, "parentId">): Promise<void>;
+    cancel(): Promise<void>;
+}
+
 type FeedbackAdapterFeedback = {
     message: ThreadMessage;
     type: "positive" | "negative";
@@ -263,6 +275,21 @@ type LocalRuntimeOptions = {
     } | undefined;
 };
 
+declare class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
+    private runtime;
+    private endEditCallback;
+    get canCancel(): boolean;
+    private _nonTextParts;
+    private _previousText;
+    private _parentId;
+    constructor(runtime: Omit<ThreadRuntimeCore, "composer">, endEditCallback: () => void, { parentId, message }: {
+        parentId: string | null;
+        message: ThreadMessage;
+    });
+    handleSend(message: Omit<AppendMessage, "parentId">): Promise<void>;
+    cancel(): Promise<void>;
+}
+
 declare class LocalThreadRuntimeCore implements ThreadRuntimeCore {
     private configProvider;
     adapter: ChatModelAdapter;
@@ -282,12 +309,15 @@ declare class LocalThreadRuntimeCore implements ThreadRuntimeCore {
     readonly threadId: string;
     readonly isDisabled = false;
     get messages(): ThreadMessage[];
-    readonly composer: BaseThreadComposerRuntimeCore;
+    readonly composer: DefaultThreadComposerRuntimeCore;
     constructor(configProvider: ModelConfigProvider, adapter: ChatModelAdapter, { initialMessages, ...options }: LocalRuntimeOptions);
     getModelConfig(): ModelConfig;
     private _options;
     get options(): LocalRuntimeOptions;
     set options({ initialMessages, ...options }: LocalRuntimeOptions);
+    private _editComposers;
+    getEditComposer(messageId: string): DefaultEditComposerRuntimeCore | undefined;
+    beginEdit(messageId: string): void;
     getBranches(messageId: string): string[];
     switchToBranch(branchId: string): void;
     append(message: AppendMessage): Promise<void>;
@@ -315,6 +345,23 @@ declare class LocalRuntimeCore extends BaseAssistantRuntimeCore<LocalThreadRunti
     }): void;
 }
 
+declare class AssistantRuntime<TThreadRuntime extends ThreadRuntime = ThreadRuntime> implements AssistantRuntimeCore {
+    private _core;
+    constructor(_core: AssistantRuntimeCore, CustomThreadRuntime: new (binding: ThreadRuntimeCoreBinding) => TThreadRuntime);
+    readonly thread: TThreadRuntime;
+    switchToNewThread(): void;
+    switchToThread(threadId: string): void;
+    /**
+     * @deprecated Use `switchToNewThread` instead. This will be removed in 0.6.0.
+     */
+    switchToThread(threadId: string | null): void;
+    registerModelConfigProvider(provider: ModelConfigProvider): Unsubscribe;
+    /**
+     * @deprecated Thread is now static and never gets updated. This will be removed in 0.6.0.
+     */
+    subscribe(callback: () => void): Unsubscribe;
+}
+
 declare class LocalRuntime extends AssistantRuntime {
     private core;
     constructor(core: LocalRuntimeCore);
@@ -381,35 +428,6 @@ declare class EdgeChatAdapter implements ChatModelAdapter {
 type EdgeRuntimeOptions = EdgeChatAdapterOptions & LocalRuntimeOptions;
 declare const useEdgeRuntime: ({ initialMessages, maxToolRoundtrips, adapters, ...options }: EdgeRuntimeOptions) => LocalRuntime;
 
-declare class SimpleImageAttachmentAdapter implements AttachmentAdapter {
-    accept: string;
-    add(state: {
-        file: File;
-    }): Promise<ThreadComposerAttachment>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
-    remove(): Promise<void>;
-}
-
-declare class SimpleTextAttachmentAdapter implements AttachmentAdapter {
-    accept: string;
-    add(state: {
-        file: File;
-    }): Promise<ThreadComposerAttachment>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
-    remove(): Promise<void>;
-}
-
-declare class CompositeAttachmentAdapter implements AttachmentAdapter {
-    private _adapters;
-    accept: string;
-    constructor(adapters: AttachmentAdapter[]);
-    add(state: {
-        file: File;
-    }): Promise<ThreadComposerAttachment>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
-    remove(attachment: ThreadComposerAttachment): Promise<void>;
-}
-
 type ThreadMessageLike = {
     role: "assistant" | "user" | "system";
     content: string | (TextContentPart | ImageContentPart | ToolCallContentPart<any, any> | CoreToolCallContentPart<any, any> | UIContentPart)[];
@@ -479,65 +497,87 @@ type SubscribableWithState<TState> = Subscribable & {
     getState: () => TState;
 };
 
-type ContentPartSnapshot = {
-    part: ThreadUserContentPart | ThreadAssistantContentPart;
-    status: ContentPartStatus | ToolCallContentPartStatus;
-};
-type ContentPartSnapshotBinding = SubscribableWithState<ContentPartSnapshot | undefined>;
-declare class ContentPartRuntime {
-    private contentBinding;
-    private messageApi;
-    private threadApi;
-    constructor(contentBinding: ContentPartSnapshotBinding, messageApi: MessageSnapshotBinding, threadApi: ThreadRuntimeCoreBinding);
-    getState(): ContentPartSnapshot | undefined;
-    addToolResult(result: any): void;
-}
-
-type MessageState = ThreadMessage & {
+type ThreadComposerRuntimeCoreBinding = SubscribableWithState<ComposerRuntimeCore | undefined>;
+type LegacyEditComposerState = Readonly<{
+    type: "edit" | "thread";
+    /** @deprecated Use `text` instead. This will be removed in 0.6.0. */
+    value: string;
+    /** @deprecated Use `useComposerRuntime().setText()` instead. This will be removed in 0.6.0. */
+    setValue: (value: string) => void;
+    text: string;
     /**
-     * @deprecated You can directly access message fields in the state. Replace `.message.content` with `.content` etc. This will be removed in 0.6.0.
+     * @deprecated Use `useComposerRuntime().setText()` instead. This will be removed in 0.6.0.
      */
-    message: ThreadMessage;
-    parentId: string | null;
-    isLast: boolean;
+    setText: (value: string) => void;
+    canCancel: boolean;
+    isEditing: boolean;
+    isEmpty: boolean;
     /**
-     * @deprecated Use `branchNumber` and `branchCount` instead. This will be removed in 0.6.0.
+     * @deprecated Use useComposerRuntime().beginEdit() instead. This will be removed in 0.6.0.
      */
-    branches: readonly string[];
-    branchNumber: number;
-    branchCount: number;
-};
-type MessageSnapshotBinding = SubscribableWithState<MessageState>;
-declare class MessageRuntime {
-    private _core;
-    private _threadBinding;
-    constructor(_core: MessageSnapshotBinding, _threadBinding: ThreadRuntimeCoreBinding);
-    getState(): MessageState;
-    edit(message: Omit<AppendMessage, "parentId">): void;
-    reload(): void;
-    speak(): void;
-    submitFeedback({ type }: {
-        type: "positive" | "negative";
-    }): void;
-    switchToBranch({ position, branchId, }: {
-        position?: "previous" | "next" | undefined;
-        branchId?: string | undefined;
-    }): void;
-    subscribe(callback: () => void): Unsubscribe;
-    getContentPartByIdx(idx: number): ContentPartRuntime;
-}
-
-type ThreadComposerRuntimeCoreBinding = SubscribableWithState<ThreadComposerRuntimeCore>;
-type UnstableThreadComposerStateV2 = Readonly<{
+    edit: () => void;
+    /**
+     * @deprecated Use `useComposerRuntime().send()` instead. This will be removed in 0.6.0.
+     */
+    send: () => void;
+    /**
+     * @deprecated Use `useComposerRuntime().cancel()` instead. This will be removed in 0.6.0.
+     */
+    cancel: () => void;
+}>;
+type LegacyThreadComposerState = Readonly<{
+    type: "thread" | "edit";
+    /** @deprecated Use `text` instead. This will be removed in 0.6.0. */
+    value: string;
+    /** @deprecated Use `useComposerRuntime().setText` instead. This will be removed in 0.6.0. */
+    setValue: (value: string) => void;
+    attachmentAccept: string;
+    attachments: readonly ThreadComposerAttachment[];
+    /** @deprecated Use `useComposerRuntime().addAttachment` instead. This will be removed in 0.6.0. */
+    addAttachment: (file: File) => void;
+    /** @deprecated Use `useComposerRuntime().removeAttachment` instead. This will be removed in 0.6.0. */
+    removeAttachment: (attachmentId: string) => void;
+    text: string;
+    /** @deprecated Use `useComposerRuntime().setText` instead. This will be removed in 0.6.0. */
+    setText: (value: string) => void;
+    /** @deprecated Use `useComposerRuntime().reset` instead. This will be removed in 0.6.0. */
+    reset: () => void;
+    canCancel: boolean;
+    isEditing: boolean;
     isEmpty: boolean;
+    /**
+     * @deprecated Use `useComposerRuntime().send` instead. This will be removed in 0.6.0.
+     **/
+    send: () => void;
+    /** @deprecated Use `useComposerRuntime().cancel` instead. This will be removed in 0.6.0. */
+    cancel: () => void;
+    /** @deprecated This feature is being removed in 0.6.0. Submit feedback if you need it. */
+    focus: () => void;
+    /** @deprecated This feature is being removed in 0.6.0. Submit feedback if you need it. */
+    onFocus: (listener: () => void) => Unsubscribe;
+}>;
+type ComposerState = LegacyThreadComposerState & LegacyEditComposerState & Readonly<{
+    type: "thread" | "edit";
     text: string;
-    attachments: readonly ThreadComposerAttachment[];
     attachmentAccept: string;
+    attachments: readonly ThreadComposerAttachment[];
     canCancel: boolean;
+    isEditing: boolean;
+    isEmpty: boolean;
 }>;
-declare class ThreadComposerRuntime implements ThreadComposerRuntimeCore {
+/** @deprecated Use `ComposerState` instead. */
+type ThreadComposerState = ComposerState;
+/** @deprecated Use `ComposerState` instead. */
+type EditComposerState = ComposerState;
+declare class ComposerRuntime implements ComposerRuntimeCore {
     private _core;
-    constructor(_core: ThreadComposerRuntimeCoreBinding);
+    private _beginEdit?;
+    get type(): "edit" | "thread";
+    constructor(_core: ThreadComposerRuntimeCoreBinding, _beginEdit?: (() => void) | undefined);
+    /**
+     * @deprecated Use `getState().isEditing` instead. This will be removed in 0.6.0.
+     */
+    get isEditing(): boolean;
     /**
      * @deprecated Use `getState().isEmpty` instead. This will be removed in 0.6.0.
      */
@@ -557,15 +597,14 @@ declare class ThreadComposerRuntime implements ThreadComposerRuntimeCore {
     /**
      * @deprecated Use `getState().attachments` instead. This will be removed in 0.6.0.
      */
-    get attachments(): ThreadComposerAttachment[];
-    getState(): Readonly<{
-        isEmpty: boolean;
-        text: string;
-        attachments: readonly ThreadComposerAttachment[];
-        attachmentAccept: string;
-        canCancel: boolean;
-    }>;
+    get attachments(): readonly ThreadComposerAttachment[];
+    /**
+     * @deprecated Use `getState().text` instead. This will be removed in 0.6.0.
+     */
+    get value(): string;
+    getState(): ComposerState;
     setText(text: string): void;
+    setValue(text: string): void;
     addAttachment(file: File): Promise<void>;
     removeAttachment(attachmentId: string): Promise<void>;
     /**
@@ -574,7 +613,52 @@ declare class ThreadComposerRuntime implements ThreadComposerRuntimeCore {
     reset(): void;
     send(): void;
     cancel(): void;
+    beginEdit(): void;
+    /**
+     * @deprecated Use `beginEdit()` instead. This will be removed in 0.6.0.
+     */
+    edit(): void;
+    subscribe(callback: () => void): Unsubscribe;
+    private _focusListeners;
+    private focus;
+    private onFocus;
+}
+
+type MessageState = ThreadMessage & {
+    /**
+     * @deprecated You can directly access message fields in the state. Replace `.message.content` with `.content` etc. This will be removed in 0.6.0.
+     */
+    message: ThreadMessage;
+    parentId: string | null;
+    isLast: boolean;
+    /**
+     * @deprecated Use `branchNumber` and `branchCount` instead. This will be removed in 0.6.0.
+     */
+    branches: readonly string[];
+    branchNumber: number;
+    branchCount: number;
+};
+type MessageStateBinding = SubscribableWithState<MessageState>;
+declare class MessageRuntime {
+    private _core;
+    private _threadBinding;
+    constructor(_core: MessageStateBinding, _threadBinding: ThreadRuntimeCoreBinding);
+    composer: ComposerRuntime & {
+        type: "edit";
+    };
+    getState(): MessageState;
+    unstable_edit(message: Omit<AppendMessage, "parentId">): void;
+    reload(): void;
+    speak(): SpeechSynthesisAdapter.Utterance;
+    submitFeedback({ type }: {
+        type: "positive" | "negative";
+    }): void;
+    switchToBranch({ position, branchId, }: {
+        position?: "previous" | "next" | undefined;
+        branchId?: string | undefined;
+    }): void;
     subscribe(callback: () => void): Unsubscribe;
+    unstable_getContentPartByIndex(idx: number): ContentPartRuntime;
 }
 
 type CreateAppendMessage = string | {
@@ -589,9 +673,9 @@ type ThreadState = Readonly<{
     isDisabled: boolean;
     isRunning: boolean;
     capabilities: RuntimeCapabilities;
+    messages: readonly ThreadMessage[];
 }>;
 declare class ThreadRuntime implements ThreadRuntimeCore {
-    private _threadBinding;
     /**
      * @deprecated Use `getState().threadId` instead. This will be removed in 0.6.0.
      */
@@ -617,6 +701,9 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
         attachments: boolean;
         feedback: boolean;
     }>;
+    /**
+     * @deprecated Use `getState().messages` instead. This will be removed in 0.6.0.
+     */
     get messages(): readonly ThreadMessage[];
     unstable_getCore(): Readonly<{
         getBranches: (messageId: string) => readonly string[];
@@ -628,7 +715,9 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
         speak: (messageId: string) => SpeechSynthesisAdapter.Utterance;
         submitFeedback: (feedback: SubmitFeedbackOptions) => void;
         getModelConfig: () => ModelConfig;
-        composer: ThreadComposerRuntimeCore;
+        composer: ComposerRuntimeCore;
+        getEditComposer: (messageId: string) => ComposerRuntimeCore | undefined;
+        beginEdit: (messageId: string) => void;
         capabilities: Readonly<RuntimeCapabilities>;
         threadId: string;
         isDisabled: boolean;
@@ -637,13 +726,15 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
         import(repository: ExportedMessageRepository): void;
         export(): ExportedMessageRepository;
     }>;
-    constructor(_threadBinding: ThreadRuntimeCoreBinding);
-    readonly composer: ThreadComposerRuntime;
+    private _threadBinding;
+    constructor(threadBinding: ThreadRuntimeCoreBinding);
+    readonly composer: ComposerRuntime;
     getState(): Readonly<{
         threadId: string;
         isDisabled: boolean;
         isRunning: boolean;
         capabilities: RuntimeCapabilities;
+        messages: readonly ThreadMessage[];
     }>;
     append(message: CreateAppendMessage): void;
     subscribe(callback: () => void): Unsubscribe;
@@ -655,28 +746,85 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
     switchToBranch(branchId: string): void;
     speak(messageId: string): SpeechSynthesisAdapter.Utterance;
     submitFeedback(options: SubmitFeedbackOptions): void;
+    /**
+     * @deprecated This is a temporary API. This will be removed in 0.6.0.
+     */
+    getEditComposer(messageId: string): Readonly<{
+        attachmentAccept: string;
+        attachments: readonly ThreadComposerAttachment[];
+        addAttachment: (file: File) => Promise<void>;
+        removeAttachment: (attachmentId: string) => Promise<void>;
+        isEditing: boolean;
+        canCancel: boolean;
+        isEmpty: boolean;
+        text: string;
+        setText: (value: string) => void;
+        reset: () => void;
+        send: () => void;
+        cancel: () => void;
+        subscribe: (callback: () => void) => Unsubscribe;
+    }> | undefined;
+    /**
+     * @deprecated This is a temporary API. This will be removed in 0.6.0.
+     */
+    beginEdit(messageId: string): void;
     export(): ExportedMessageRepository;
     import(data: ExportedMessageRepository): void;
     unstable_getMesssageByIndex(idx: number): MessageRuntime;
 }
 
-declare class AssistantRuntime<TThreadRuntime extends ThreadRuntime = ThreadRuntime> implements AssistantRuntimeCore {
-    private _core;
-    constructor(_core: AssistantRuntimeCore, CustomThreadRuntime: new (binding: ThreadRuntimeCoreBinding) => TThreadRuntime);
-    readonly thread: TThreadRuntime;
-    switchToNewThread(): void;
-    switchToThread(threadId: string): void;
+type ContentPartState = (ThreadUserContentPart | ThreadAssistantContentPart) & {
     /**
-     * @deprecated Use `switchToNewThread` instead. This will be removed in 0.6.0.
-     */
-    switchToThread(threadId: string | null): void;
-    registerModelConfigProvider(provider: ModelConfigProvider): Unsubscribe;
-    /**
-     * @deprecated Thread is now static and never gets updated. This will be removed in 0.6.0.
+     * @deprecated You can directly access content part fields in the state. Replace `.part.type` with `.type` etc. This will be removed in 0.6.0.
      */
+    part: ThreadUserContentPart | ThreadAssistantContentPart;
+    status: ContentPartStatus | ToolCallContentPartStatus;
+};
+type ContentPartSnapshotBinding = SubscribableWithState<ContentPartState>;
+declare class ContentPartRuntime {
+    private contentBinding;
+    private messageApi;
+    private threadApi;
+    constructor(contentBinding: ContentPartSnapshotBinding, messageApi: MessageStateBinding, threadApi: ThreadRuntimeCoreBinding);
+    getState(): ContentPartState;
+    addToolResult(result: any): void;
     subscribe(callback: () => void): Unsubscribe;
 }
 
+type EmptyContentPartProps = {
+    status: ContentPartStatus;
+};
+type EmptyContentPartComponent = ComponentType<EmptyContentPartProps>;
+type TextContentPartProps = ContentPartState & TextContentPart & {
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.part.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    part: TextContentPart;
+};
+type TextContentPartComponent = ComponentType<TextContentPartProps>;
+type ImageContentPartProps = ContentPartState & ImageContentPart & {
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.part.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    part: ImageContentPart;
+};
+type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
+type UIContentPartProps = ContentPartState & UIContentPart & {
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.part.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    part: UIContentPart;
+};
+type UIContentPartComponent = ComponentType<UIContentPartProps>;
+type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = ContentPartState & ToolCallContentPart<TArgs, TResult> & {
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.part.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    part: ToolCallContentPart<TArgs, TResult>;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type AssistantRuntimeProviderProps = {
     runtime: AssistantRuntime;
 };
@@ -686,66 +834,11 @@ type TextContentPartProviderProps = {
     text: string;
     isRunning?: boolean | undefined;
 };
-declare const TextContentPartProvider: FC<PropsWithChildren<TextContentPartProviderProps>>;
-
-type AssistantToolUIsState = Readonly<{
-    getToolUI: (toolName: string) => ToolCallContentPartComponent | null;
-    setToolUI: (toolName: string, render: ToolCallContentPartComponent) => () => void;
-}>;
-
-type ThreadComposerState = Readonly<{
-    type: "thread";
-    /** @deprecated Use `text` instead. This will be removed in 0.6.0. */
-    value: string;
-    /** @deprecated Use `useComposerRuntime().setText` instead. This will be removed in 0.6.0. */
-    setValue: (value: string) => void;
-    attachmentAccept: string;
-    attachments: readonly ThreadComposerAttachment[];
-    /** @deprecated Use `useComposerRuntime().addAttachment` instead. This will be removed in 0.6.0. */
-    addAttachment: (file: File) => void;
-    /** @deprecated Use `useComposerRuntime().removeAttachment` instead. This will be removed in 0.6.0. */
-    removeAttachment: (attachmentId: string) => void;
-    text: string;
-    /** @deprecated Use `useComposerRuntime().setText` instead. This will be removed in 0.6.0. */
-    setText: (value: string) => void;
-    /** @deprecated Use `useComposerRuntime().reset` instead. This will be removed in 0.6.0. */
-    reset: () => void;
-    canCancel: boolean;
-    isEditing: true;
-    isEmpty: boolean;
-    /** @deprecated Use `useComposerRuntime().send` instead. This will be removed in 0.6.0. */
-    send: () => void;
-    /** @deprecated Use `useComposerRuntime().cancel` instead. This will be removed in 0.6.0. */
-    cancel: () => void;
-    focus: () => void;
-    onFocus: (listener: () => void) => Unsubscribe;
-}>;
-
-type TextContentPartState = Readonly<{
-    status: ContentPartStatus;
-    part: TextContentPart;
-}>;
-type ContentPartState = Readonly<{
-    status: ContentPartStatus | ToolCallContentPartStatus;
-    part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-}>;
-
-type ReadonlyStore<T> = Omit<StoreApi<T>, "setState" | "destroy">;
-
-type EditComposerState = Readonly<{
-    type: "edit";
-    /** @deprecated Use `text` instead. This will be removed in 0.6.0. */
-    value: string;
-    /** @deprecated Use `setText` instead. This will be removed in 0.6.0. */
-    setValue: (value: string) => void;
-    text: string;
-    setText: (value: string) => void;
-    canCancel: boolean;
-    isEditing: boolean;
-    isEmpty: boolean;
-    edit: () => void;
-    send: () => void;
-    cancel: () => void;
+declare const TextContentPartProvider: FC<PropsWithChildren<TextContentPartProviderProps>>;
+
+type AssistantToolUIsState = Readonly<{
+    getToolUI: (toolName: string) => ToolCallContentPartComponent | null;
+    setToolUI: (toolName: string, render: ToolCallContentPartComponent) => () => void;
 }>;
 
 type MessageUtilsState = Readonly<{
@@ -766,6 +859,8 @@ type ThreadViewportState = Readonly<{
     onScrollToBottom: (callback: () => void) => Unsubscribe;
 }>;
 
+type ReadonlyStore<T> = Omit<StoreApi<T>, "setState" | "destroy">;
+
 type AssistantContextValue = {
     useToolUIs: UseBoundStore<ReadonlyStore<AssistantToolUIsState>>;
     useAssistantRuntime: UseBoundStore<ReadonlyStore<AssistantRuntime>>;
@@ -928,6 +1023,7 @@ declare const useThread: {
             attachments: boolean;
             feedback: boolean;
         }>;
+        messages: readonly ThreadMessage[];
     }>;
     <TSelected>(selector: (state: Readonly<{
         threadId: string;
@@ -943,6 +1039,7 @@ declare const useThread: {
             attachments: boolean;
             feedback: boolean;
         }>;
+        messages: readonly ThreadMessage[];
     }>) => TSelected): TSelected;
     (options: {
         optional: true;
@@ -960,6 +1057,7 @@ declare const useThread: {
             attachments: boolean;
             feedback: boolean;
         }>;
+        messages: readonly ThreadMessage[];
     }> | null;
     <TSelected>(options: {
         optional: true;
@@ -977,6 +1075,7 @@ declare const useThread: {
                 attachments: boolean;
                 feedback: boolean;
             }>;
+            messages: readonly ThreadMessage[];
         }>) => TSelected;
     }): TSelected | null;
 };
@@ -995,6 +1094,7 @@ declare const useThreadStore: {
             attachments: boolean;
             feedback: boolean;
         }>;
+        messages: readonly ThreadMessage[];
     }>>;
     (options: {
         optional: true;
@@ -1012,8 +1112,12 @@ declare const useThreadStore: {
             attachments: boolean;
             feedback: boolean;
         }>;
+        messages: readonly ThreadMessage[];
     }>> | null;
 };
+/**
+ * @deprecated Use `useThread().messages` instead. This will be removed in 0.6.0.
+ */
 declare const useThreadMessages: {
     (): readonly ThreadMessage[];
     <TSelected>(selector: (state: readonly ThreadMessage[]) => TSelected): TSelected;
@@ -1025,6 +1129,9 @@ declare const useThreadMessages: {
         selector?: (state: readonly ThreadMessage[]) => TSelected;
     }): TSelected | null;
 };
+/**
+ * @deprecated Use `useThreadRuntime().getState().messages` instead. This will be removed in 0.6.0.
+ */
 declare const useThreadMessagesStore: {
     (): ReadonlyStore<readonly ThreadMessage[]>;
     (options: {
@@ -1032,129 +1139,21 @@ declare const useThreadMessagesStore: {
     }): ReadonlyStore<readonly ThreadMessage[]> | null;
 };
 declare const useThreadComposer: {
-    (): Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>;
-    <TSelected>(selector: (state: Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>) => TSelected): TSelected;
+    (): ComposerState;
+    <TSelected>(selector: (state: ComposerState) => TSelected): TSelected;
     (options: {
         optional: true;
-    }): Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }> | null;
+    }): ComposerState | null;
     <TSelected>(options: {
         optional: true;
-        selector?: (state: Readonly<{
-            type: "thread";
-            value: string;
-            setValue: (value: string) => void;
-            attachmentAccept: string;
-            attachments: readonly ThreadComposerAttachment[];
-            addAttachment: (file: File) => void;
-            removeAttachment: (attachmentId: string) => void;
-            text: string;
-            setText: (value: string) => void;
-            reset: () => void;
-            canCancel: boolean;
-            isEditing: true;
-            isEmpty: boolean;
-            send: () => void;
-            cancel: () => void;
-            focus: () => void;
-            onFocus: (listener: () => void) => Unsubscribe;
-        }>) => TSelected;
+        selector?: (state: ComposerState) => TSelected;
     }): TSelected | null;
 };
 declare const useThreadComposerStore: {
-    (): ReadonlyStore<Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>>;
+    (): ReadonlyStore<ComposerState>;
     (options: {
         optional: true;
-    }): ReadonlyStore<Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>> | null;
+    }): ReadonlyStore<ComposerState> | null;
 };
 declare const useThreadViewport: {
     (): Readonly<{
@@ -1313,96 +1312,25 @@ declare const useMessageUtilsStore: {
     }>> | null;
 };
 declare const useEditComposer: {
-    (): Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }>;
-    <TSelected>(selector: (state: Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }>) => TSelected): TSelected;
+    (): ComposerState;
+    <TSelected>(selector: (state: ComposerState) => TSelected): TSelected;
     (options: {
         optional: true;
-    }): Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }> | null;
+    }): ComposerState | null;
     <TSelected>(options: {
         optional: true;
-        selector?: (state: Readonly<{
-            type: "edit";
-            value: string;
-            setValue: (value: string) => void;
-            text: string;
-            setText: (value: string) => void;
-            canCancel: boolean;
-            isEditing: boolean;
-            isEmpty: boolean;
-            edit: () => void;
-            send: () => void;
-            cancel: () => void;
-        }>) => TSelected;
+        selector?: (state: ComposerState) => TSelected;
     }): TSelected | null;
 };
 declare const useEditComposerStore: {
-    (): ReadonlyStore<Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }>>;
+    (): ReadonlyStore<ComposerState>;
     (options: {
         optional: true;
-    }): ReadonlyStore<Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }>> | null;
+    }): ReadonlyStore<ComposerState> | null;
 };
 
 type ContentPartContextValue = {
+    useContentPartRuntime: UseBoundStore<ReadonlyStore<ContentPartRuntime>>;
     useContentPart: UseBoundStore<ReadonlyStore<ContentPartState>>;
 };
 declare const useContentPartContext: {
@@ -1413,244 +1341,58 @@ declare const useContentPartContext: {
         optional?: boolean | undefined;
     } | undefined): ContentPartContextValue | null;
 };
+declare function useContentPartRuntime(options?: {
+    optional?: false | undefined;
+}): ContentPartRuntime;
+declare function useContentPartRuntime(options?: {
+    optional?: boolean | undefined;
+}): ContentPartRuntime | null;
 declare const useContentPart: {
-    (): Readonly<{
-        status: ContentPartStatus | ToolCallContentPartStatus;
-        part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-    }>;
-    <TSelected>(selector: (state: Readonly<{
-        status: ContentPartStatus | ToolCallContentPartStatus;
-        part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-    }>) => TSelected): TSelected;
+    (): ContentPartState;
+    <TSelected>(selector: (state: ContentPartState) => TSelected): TSelected;
     (options: {
         optional: true;
-    }): Readonly<{
-        status: ContentPartStatus | ToolCallContentPartStatus;
-        part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-    }> | null;
+    }): ContentPartState | null;
     <TSelected>(options: {
         optional: true;
-        selector?: (state: Readonly<{
-            status: ContentPartStatus | ToolCallContentPartStatus;
-            part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-        }>) => TSelected;
+        selector?: (state: ContentPartState) => TSelected;
     }): TSelected | null;
 };
 declare const useContentPartStore: {
-    (): ReadonlyStore<Readonly<{
-        status: ContentPartStatus | ToolCallContentPartStatus;
-        part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-    }>>;
+    (): ReadonlyStore<ContentPartState>;
     (options: {
         optional: true;
-    }): ReadonlyStore<Readonly<{
-        status: ContentPartStatus | ToolCallContentPartStatus;
-        part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
-    }>> | null;
+    }): ReadonlyStore<ContentPartState> | null;
 };
 
 type ComposerContextValue = {
-    useComposer: ReadonlyStore<EditComposerState | ThreadComposerState>;
+    useComposer: ReadonlyStore<ComposerState>;
     type: "edit" | "new";
 };
 declare const useComposerContext: () => ComposerContextValue;
 declare const useComposer: {
-    (): Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }> | Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>;
-    <TSelected>(selector: (state: Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }> | Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>) => TSelected): TSelected;
+    (): ComposerState;
+    <TSelected>(selector: (state: ComposerState) => TSelected): TSelected;
     (options: {
         optional: true;
-    }): Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }> | Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }> | null;
+    }): ComposerState | null;
     <TSelected>(options: {
         optional: true;
-        selector?: (state: Readonly<{
-            type: "edit";
-            value: string;
-            setValue: (value: string) => void;
-            text: string;
-            setText: (value: string) => void;
-            canCancel: boolean;
-            isEditing: boolean;
-            isEmpty: boolean;
-            edit: () => void;
-            send: () => void;
-            cancel: () => void;
-        }> | Readonly<{
-            type: "thread";
-            value: string;
-            setValue: (value: string) => void;
-            attachmentAccept: string;
-            attachments: readonly ThreadComposerAttachment[];
-            addAttachment: (file: File) => void;
-            removeAttachment: (attachmentId: string) => void;
-            text: string;
-            setText: (value: string) => void;
-            reset: () => void;
-            canCancel: boolean;
-            isEditing: true;
-            isEmpty: boolean;
-            send: () => void;
-            cancel: () => void;
-            focus: () => void;
-            onFocus: (listener: () => void) => Unsubscribe;
-        }>) => TSelected;
+        selector?: (state: ComposerState) => TSelected;
     }): TSelected | null;
 };
 declare const useComposerStore: {
-    (): ReadonlyStore<Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }> | Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>>;
+    (): ReadonlyStore<ComposerState>;
     (options: {
         optional: true;
-    }): ReadonlyStore<Readonly<{
-        type: "edit";
-        value: string;
-        setValue: (value: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        canCancel: boolean;
-        isEditing: boolean;
-        isEmpty: boolean;
-        edit: () => void;
-        send: () => void;
-        cancel: () => void;
-    }> | Readonly<{
-        type: "thread";
-        value: string;
-        setValue: (value: string) => void;
-        attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
-        addAttachment: (file: File) => void;
-        removeAttachment: (attachmentId: string) => void;
-        text: string;
-        setText: (value: string) => void;
-        reset: () => void;
-        canCancel: boolean;
-        isEditing: true;
-        isEmpty: boolean;
-        send: () => void;
-        cancel: () => void;
-        focus: () => void;
-        onFocus: (listener: () => void) => Unsubscribe;
-    }>> | null;
+    }): ReadonlyStore<ComposerState> | null;
 };
+declare function useComposerRuntime(options?: {
+    optional?: false | undefined;
+}): ComposerRuntime;
+declare function useComposerRuntime(options?: {
+    optional?: boolean | undefined;
+}): ComposerRuntime | null;
 
 declare const useAppendMessage: () => (message: CreateAppendMessage) => void;
 
@@ -1724,20 +1466,20 @@ declare const useComposerSend: () => (() => void) | null;
 
 declare const useComposerAddAttachment: () => (() => void) | null;
 
-declare const useContentPartDisplay: () => Readonly<{
-    status: ContentPartStatus;
-    part: UIContentPart;
-}>;
+declare const useContentPartDisplay: () => UIContentPart & {
+    part: ThreadUserContentPart | ThreadAssistantContentPart;
+    status: ContentPartStatus | ToolCallContentPartStatus;
+};
 
-declare const useContentPartImage: () => Readonly<{
-    status: ContentPartStatus;
-    part: ImageContentPart;
-}>;
+declare const useContentPartImage: () => ImageContentPart & {
+    part: ThreadUserContentPart | ThreadAssistantContentPart;
+    status: ContentPartStatus | ToolCallContentPartStatus;
+};
 
-declare const useContentPartText: () => Readonly<{
-    status: ContentPartStatus;
-    part: TextContentPart;
-}>;
+declare const useContentPartText: () => TextContentPart & {
+    part: ThreadUserContentPart | ThreadAssistantContentPart;
+    status: ContentPartStatus | ToolCallContentPartStatus;
+};
 
 type MessageIfFilters = {
     user: boolean | undefined;
@@ -2463,7 +2205,7 @@ declare class ProxyConfigProvider implements ModelConfigProvider {
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
 }
 
-declare const useSmooth: (state: TextContentPartState, smooth?: boolean) => TextContentPartState;
+declare const useSmooth: (state: ContentPartState & TextContentPart, smooth?: boolean) => ContentPartState & TextContentPart;
 
 declare const withSmoothContextProvider: <C extends ComponentType<any>>(Component: C) => C;
 declare const useSmoothStatus: {
@@ -2528,8 +2270,8 @@ type internal_AssistantRuntime<TThreadRuntime extends ThreadRuntime = ThreadRunt
 declare const internal_AssistantRuntime: typeof AssistantRuntime;
 type internal_BaseAssistantRuntimeCore<TThreadRuntime extends ReactThreadRuntimeCore> = BaseAssistantRuntimeCore<TThreadRuntime>;
 declare const internal_BaseAssistantRuntimeCore: typeof BaseAssistantRuntimeCore;
-type internal_BaseThreadComposerRuntimeCore = BaseThreadComposerRuntimeCore;
-declare const internal_BaseThreadComposerRuntimeCore: typeof BaseThreadComposerRuntimeCore;
+type internal_DefaultThreadComposerRuntimeCore = DefaultThreadComposerRuntimeCore;
+declare const internal_DefaultThreadComposerRuntimeCore: typeof DefaultThreadComposerRuntimeCore;
 type internal_MessageRepository = MessageRepository;
 declare const internal_MessageRepository: typeof MessageRepository;
 type internal_ProxyConfigProvider = ProxyConfigProvider;
@@ -2544,7 +2286,7 @@ declare const internal_useSmooth: typeof useSmooth;
 declare const internal_useSmoothStatus: typeof useSmoothStatus;
 declare const internal_withSmoothContextProvider: typeof withSmoothContextProvider;
 declare namespace internal {
-  export { internal_AssistantRuntime as AssistantRuntime, internal_BaseAssistantRuntimeCore as BaseAssistantRuntimeCore, internal_BaseThreadComposerRuntimeCore as BaseThreadComposerRuntimeCore, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, type internal_ReactThreadRuntimeCore as ReactThreadRuntimeCore, internal_ThreadRuntime as ThreadRuntime, type internal_ThreadRuntimeCoreBinding as ThreadRuntimeCoreBinding, internal_TooltipIconButton as TooltipIconButton, internal_generateId as generateId, internal_useSmooth as useSmooth, internal_useSmoothStatus as useSmoothStatus, internal_withSmoothContextProvider as withSmoothContextProvider };
+  export { internal_AssistantRuntime as AssistantRuntime, internal_BaseAssistantRuntimeCore as BaseAssistantRuntimeCore, internal_DefaultThreadComposerRuntimeCore as DefaultThreadComposerRuntimeCore, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, type internal_ReactThreadRuntimeCore as ReactThreadRuntimeCore, internal_ThreadRuntime as ThreadRuntime, type internal_ThreadRuntimeCoreBinding as ThreadRuntimeCoreBinding, internal_TooltipIconButton as TooltipIconButton, internal_generateId as generateId, internal_useSmooth as useSmooth, internal_useSmoothStatus as useSmoothStatus, internal_withSmoothContextProvider as withSmoothContextProvider };
 }
 
-export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveFeedbackNegativeProps, type ActionBarPrimitiveFeedbackPositiveProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type ActionBarPrimitiveSpeakProps, type ActionBarPrimitiveStopSpeakingProps, type AddToolResultOptions, AppendMessage, _default$b as AssistantActionBar, type AssistantContextValue, _default$a as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$9 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, AssistantRuntime, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantToolUIsState, type AttachmentAdapter, _default$8 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, _default$7 as Composer, _default$6 as ComposerAttachment, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, CompositeAttachmentAdapter, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, ContentPartRuntime, type ContentPartState, CoreMessage, type DangerousInBrowserRuntimeOptions, EdgeChatAdapter, type EdgeRuntimeOptions, _default$5 as EditComposer, type EditComposerState, type EmptyContentPartComponent, type EmptyContentPartProps, type ExternalStoreAdapter, type ExternalStoreMessageConverter, internal as INTERNAL, ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, MessageRuntime, type MessageState, MessageStatus, type MessageUtilsState, ModelConfig, ModelConfigProvider, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SpeechSynthesisAdapter, StreamUtils, type StringsConfig, type SubmitFeedbackOptions, type SuggestionConfig, TextContentPart, type TextContentPartComponent, type TextContentPartProps, TextContentPartProvider, _default$4 as Thread, ThreadAssistantContentPart, ThreadComposerRuntime, type ThreadComposerState, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, ThreadMessage, type ThreadMessageLike, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, ThreadRuntime, type ThreadState, ThreadUserContentPart, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, Tool, ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, UIContentPart, type UIContentPartComponent, type UIContentPartProps, type UnstableThreadComposerStateV2, type Unsubscribe, type UseActionBarCopyProps, _default$2 as UserActionBar, _default$3 as UserMessage, _default$1 as UserMessageAttachment, type UserMessageConfig, type UserMessageContentProps, WebSpeechSynthesisAdapter, fromCoreMessage, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, getExternalStoreMessage, makeAssistantTool, makeAssistantToolUI, streamUtils, subscribeToMainThread, toCoreMessage, toCoreMessages, toLanguageModelMessages, toLanguageModelTools, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAppendMessage, useAssistantActions, useAssistantActionsStore, useAssistantContext, useAssistantInstructions, useAssistantRuntime, useAssistantRuntimeStore, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposer, useComposerAddAttachment, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useComposerStore, useContentPart, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartStore, useContentPartText, useDangerousInBrowserRuntime, useEdgeRuntime, useEditComposer, useEditComposerStore, useExternalMessageConverter, useExternalStoreRuntime, useLocalRuntime, useMessage, useMessageContext, useMessageIf, useMessageRuntime, useMessageStore, useMessageUtils, useMessageUtilsStore, useSwitchToNewThread, useThread, useThreadActions, useThreadActionsStore, useThreadComposer, useThreadComposerStore, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadMessages, useThreadMessagesStore, useThreadRuntime, useThreadRuntimeStore, useThreadScrollToBottom, useThreadStore, useThreadSuggestion, useThreadViewport, useThreadViewportStore, useToolUIs, useToolUIsStore };
+export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveFeedbackNegativeProps, type ActionBarPrimitiveFeedbackPositiveProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type ActionBarPrimitiveSpeakProps, type ActionBarPrimitiveStopSpeakingProps, type AddToolResultOptions, AppendMessage, _default$b as AssistantActionBar, type AssistantContextValue, _default$a as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$9 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, AssistantRuntime, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantToolUIsState, type AttachmentAdapter, _default$8 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, _default$7 as Composer, _default$6 as ComposerAttachment, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, ComposerRuntime, type ComposerState, CompositeAttachmentAdapter, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, ContentPartRuntime, CoreMessage, type DangerousInBrowserRuntimeOptions, EdgeChatAdapter, type EdgeRuntimeOptions, _default$5 as EditComposer, type EditComposerState, type EmptyContentPartComponent, type EmptyContentPartProps, type ExternalStoreAdapter, type ExternalStoreMessageConverter, internal as INTERNAL, ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, MessageRuntime, type MessageState, MessageStatus, type MessageUtilsState, ModelConfig, ModelConfigProvider, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SpeechSynthesisAdapter, StreamUtils, type StringsConfig, type SubmitFeedbackOptions, type SuggestionConfig, TextContentPart, type TextContentPartComponent, type TextContentPartProps, TextContentPartProvider, _default$4 as Thread, ThreadAssistantContentPart, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, ThreadMessage, type ThreadMessageLike, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, ThreadRuntime, type ThreadState, ThreadUserContentPart, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, Tool, ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UseActionBarCopyProps, _default$2 as UserActionBar, _default$3 as UserMessage, _default$1 as UserMessageAttachment, type UserMessageConfig, type UserMessageContentProps, WebSpeechSynthesisAdapter, fromCoreMessage, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, getExternalStoreMessage, makeAssistantTool, makeAssistantToolUI, streamUtils, subscribeToMainThread, toCoreMessage, toCoreMessages, toLanguageModelMessages, toLanguageModelTools, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAppendMessage, useAssistantActions, useAssistantActionsStore, useAssistantContext, useAssistantInstructions, useAssistantRuntime, useAssistantRuntimeStore, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposer, useComposerAddAttachment, useComposerCancel, useComposerContext, useComposerIf, useComposerRuntime, useComposerSend, useComposerStore, useContentPart, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartRuntime, useContentPartStore, useContentPartText, useDangerousInBrowserRuntime, useEdgeRuntime, useEditComposer, useEditComposerStore, useExternalMessageConverter, useExternalStoreRuntime, useLocalRuntime, useMessage, useMessageContext, useMessageIf, useMessageRuntime, useMessageStore, useMessageUtils, useMessageUtilsStore, useSwitchToNewThread, useThread, useThreadActions, useThreadActionsStore, useThreadComposer, useThreadComposerStore, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadMessages, useThreadMessagesStore, useThreadRuntime, useThreadRuntimeStore, useThreadScrollToBottom, useThreadStore, useThreadSuggestion, useThreadViewport, useThreadViewportStore, useToolUIs, useToolUIsStore };