@assistant-ui/react 0.5.67 → 0.5.68

package/dist/{edge-BDAlTCC1.d.mts → edge-BhYWMe-K.d.mts}

@@ -3,18 +3,38 @@ import { ReactNode } from 'react';
 import { JSONSchema7 } from 'json-schema';
 import { z } from 'zod';
 
+type PendingAttachmentStatus = {
+    type: "running";
+    reason: "uploading";
+    progress: number;
+} | {
+    type: "requires-action";
+    reason: "composer-send";
+} | {
+    type: "incomplete";
+    reason: "error" | "upload-paused";
+};
+type CompleteAttachmentStatus = {
+    type: "complete";
+};
+type AttachmentStatus = PendingAttachmentStatus | CompleteAttachmentStatus;
 type BaseAttachment = {
     id: string;
     type: "image" | "document" | "file";
     name: string;
+    contentType?: string;
+    file?: File;
+    content?: CoreUserContentPart[];
 };
-type ThreadComposerAttachment = BaseAttachment & {
+type PendingAttachment = BaseAttachment & {
+    status: PendingAttachmentStatus;
     file: File;
 };
-type MessageAttachment = BaseAttachment & {
-    file?: File;
+type CompleteAttachment = BaseAttachment & {
+    status: CompleteAttachmentStatus;
     content: CoreUserContentPart[];
 };
+type Attachment = PendingAttachment | CompleteAttachment;
 
 type TextContentPart = {
     type: "text";
@@ -88,7 +108,7 @@ type ThreadSystemMessage = MessageCommonProps & {
 type ThreadUserMessage = MessageCommonProps & {
     role: "user";
     content: ThreadUserContentPart[];
-    attachments: readonly MessageAttachment[];
+    attachments: readonly CompleteAttachment[];
     status?: undefined;
     metadata?: undefined;
 };
@@ -108,7 +128,7 @@ type ThreadAssistantMessage = MessageCommonProps & {
 };
 type AppendMessage = CoreMessage & {
     parentId: string | null;
-    attachments?: readonly MessageAttachment[] | undefined;
+    attachments?: readonly CompleteAttachment[] | undefined;
 };
 type ThreadMessage = ThreadSystemMessage | ThreadUserMessage | ThreadAssistantMessage;
 /** Core Message Types (without UI content parts) */
@@ -464,4 +484,4 @@ declare const createEdgeRuntimeAPI: (options: CreateEdgeRuntimeAPIOptions) => {
     POST: (request: Request) => Promise<Response>;
 };
 
-export { type AppendMessage as A, type CoreMessage as C, type EdgeRuntimeRequestOptions as E, type ImageContentPart as I, type ModelConfig as M, type ThreadComposerAttachment as T, type UIContentPart as U, type ThreadMessage as a, type ModelConfigProvider as b, type ThreadAssistantContentPart as c, type MessageStatus as d, type ThreadRoundtrip as e, type MessageAttachment as f, type Tool as g, type TextContentPart as h, type ToolCallContentPart as i, type CoreToolCallContentPart as j, type CreateEdgeRuntimeAPIOptions as k, type ThreadUserContentPart as l, type ContentPartStatus as m, type ToolCallContentPartStatus as n, type ThreadSystemMessage as o, type ThreadAssistantMessage as p, type ThreadUserMessage as q, type CoreUserContentPart as r, type CoreAssistantContentPart as s, type CoreSystemMessage as t, type CoreUserMessage as u, type CoreAssistantMessage as v, createEdgeRuntimeAPI as w, getEdgeRuntimeResponse as x };
+export { type Attachment as A, getEdgeRuntimeResponse as B, type CoreMessage as C, type EdgeRuntimeRequestOptions as E, type ImageContentPart as I, type ModelConfig as M, type PendingAttachment as P, type ThreadMessage as T, type UIContentPart as U, type AppendMessage as a, type ModelConfigProvider as b, type ThreadAssistantContentPart as c, type MessageStatus as d, type ThreadRoundtrip as e, type PendingAttachmentStatus as f, type CompleteAttachment as g, type CompleteAttachmentStatus as h, type Tool as i, type TextContentPart as j, type ToolCallContentPart as k, type CoreToolCallContentPart as l, type CreateEdgeRuntimeAPIOptions as m, type ThreadUserContentPart as n, type ContentPartStatus as o, type ToolCallContentPartStatus as p, type AttachmentStatus as q, type ThreadSystemMessage as r, type ThreadAssistantMessage as s, type ThreadUserMessage as t, type CoreUserContentPart as u, type CoreAssistantContentPart as v, type CoreSystemMessage as w, type CoreUserMessage as x, type CoreAssistantMessage as y, createEdgeRuntimeAPI as z };

package/dist/{edge-BDAlTCC1.d.ts → edge-BhYWMe-K.d.ts}

@@ -3,18 +3,38 @@ import { ReactNode } from 'react';
 import { JSONSchema7 } from 'json-schema';
 import { z } from 'zod';
 
+type PendingAttachmentStatus = {
+    type: "running";
+    reason: "uploading";
+    progress: number;
+} | {
+    type: "requires-action";
+    reason: "composer-send";
+} | {
+    type: "incomplete";
+    reason: "error" | "upload-paused";
+};
+type CompleteAttachmentStatus = {
+    type: "complete";
+};
+type AttachmentStatus = PendingAttachmentStatus | CompleteAttachmentStatus;
 type BaseAttachment = {
     id: string;
     type: "image" | "document" | "file";
     name: string;
+    contentType?: string;
+    file?: File;
+    content?: CoreUserContentPart[];
 };
-type ThreadComposerAttachment = BaseAttachment & {
+type PendingAttachment = BaseAttachment & {
+    status: PendingAttachmentStatus;
     file: File;
 };
-type MessageAttachment = BaseAttachment & {
-    file?: File;
+type CompleteAttachment = BaseAttachment & {
+    status: CompleteAttachmentStatus;
     content: CoreUserContentPart[];
 };
+type Attachment = PendingAttachment | CompleteAttachment;
 
 type TextContentPart = {
     type: "text";
@@ -88,7 +108,7 @@ type ThreadSystemMessage = MessageCommonProps & {
 type ThreadUserMessage = MessageCommonProps & {
     role: "user";
     content: ThreadUserContentPart[];
-    attachments: readonly MessageAttachment[];
+    attachments: readonly CompleteAttachment[];
     status?: undefined;
     metadata?: undefined;
 };
@@ -108,7 +128,7 @@ type ThreadAssistantMessage = MessageCommonProps & {
 };
 type AppendMessage = CoreMessage & {
     parentId: string | null;
-    attachments?: readonly MessageAttachment[] | undefined;
+    attachments?: readonly CompleteAttachment[] | undefined;
 };
 type ThreadMessage = ThreadSystemMessage | ThreadUserMessage | ThreadAssistantMessage;
 /** Core Message Types (without UI content parts) */
@@ -464,4 +484,4 @@ declare const createEdgeRuntimeAPI: (options: CreateEdgeRuntimeAPIOptions) => {
     POST: (request: Request) => Promise<Response>;
 };
 
-export { type AppendMessage as A, type CoreMessage as C, type EdgeRuntimeRequestOptions as E, type ImageContentPart as I, type ModelConfig as M, type ThreadComposerAttachment as T, type UIContentPart as U, type ThreadMessage as a, type ModelConfigProvider as b, type ThreadAssistantContentPart as c, type MessageStatus as d, type ThreadRoundtrip as e, type MessageAttachment as f, type Tool as g, type TextContentPart as h, type ToolCallContentPart as i, type CoreToolCallContentPart as j, type CreateEdgeRuntimeAPIOptions as k, type ThreadUserContentPart as l, type ContentPartStatus as m, type ToolCallContentPartStatus as n, type ThreadSystemMessage as o, type ThreadAssistantMessage as p, type ThreadUserMessage as q, type CoreUserContentPart as r, type CoreAssistantContentPart as s, type CoreSystemMessage as t, type CoreUserMessage as u, type CoreAssistantMessage as v, createEdgeRuntimeAPI as w, getEdgeRuntimeResponse as x };
+export { type Attachment as A, getEdgeRuntimeResponse as B, type CoreMessage as C, type EdgeRuntimeRequestOptions as E, type ImageContentPart as I, type ModelConfig as M, type PendingAttachment as P, type ThreadMessage as T, type UIContentPart as U, type AppendMessage as a, type ModelConfigProvider as b, type ThreadAssistantContentPart as c, type MessageStatus as d, type ThreadRoundtrip as e, type PendingAttachmentStatus as f, type CompleteAttachment as g, type CompleteAttachmentStatus as h, type Tool as i, type TextContentPart as j, type ToolCallContentPart as k, type CoreToolCallContentPart as l, type CreateEdgeRuntimeAPIOptions as m, type ThreadUserContentPart as n, type ContentPartStatus as o, type ToolCallContentPartStatus as p, type AttachmentStatus as q, type ThreadSystemMessage as r, type ThreadAssistantMessage as s, type ThreadUserMessage as t, type CoreUserContentPart as u, type CoreAssistantContentPart as v, type CoreSystemMessage as w, type CoreUserMessage as x, type CoreAssistantMessage as y, createEdgeRuntimeAPI as z };

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-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 { A as Attachment, P as PendingAttachment, T 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 PendingAttachmentStatus, g as CompleteAttachment, h as CompleteAttachmentStatus, i as Tool, j as TextContentPart, I as ImageContentPart, k as ToolCallContentPart, l as CoreToolCallContentPart, U as UIContentPart, m as CreateEdgeRuntimeAPIOptions, n as ThreadUserContentPart, o as ContentPartStatus, p as ToolCallContentPartStatus } from './edge-BhYWMe-K.mjs';
+export { q as AttachmentStatus, v as CoreAssistantContentPart, y as CoreAssistantMessage, w as CoreSystemMessage, u as CoreUserContentPart, x as CoreUserMessage, E as EdgeRuntimeRequestOptions, s as ThreadAssistantMessage, r as ThreadSystemMessage, t as ThreadUserMessage } from './edge-BhYWMe-K.mjs';
 import * as react from 'react';
 import { ComponentType, PropsWithChildren, FC, ComponentPropsWithoutRef, ElementType, ReactNode } from 'react';
 import { StoreApi, UseBoundStore } from 'zustand';
@@ -16,7 +16,7 @@ import 'zod';
 
 type ComposerRuntimeCore = Readonly<{
     attachmentAccept: string;
-    attachments: readonly ThreadComposerAttachment[];
+    attachments: readonly Attachment[];
     addAttachment: (file: File) => Promise<void>;
     removeAttachment: (attachmentId: string) => Promise<void>;
     isEditing: boolean;
@@ -32,6 +32,9 @@ type ComposerRuntimeCore = Readonly<{
     cancel: () => void;
     subscribe: (callback: () => void) => Unsubscribe;
 }>;
+type ThreadComposerRuntimeCore = ComposerRuntimeCore & Readonly<{
+    attachments: readonly PendingAttachment[];
+}>;
 
 type Unsubscribe = () => void;
 
@@ -114,7 +117,7 @@ type ThreadRuntimeCore = Readonly<{
     speak: (messageId: string) => SpeechSynthesisAdapter.Utterance;
     submitFeedback: (feedback: SubmitFeedbackOptions) => void;
     getModelConfig: () => ModelConfig;
-    composer: ComposerRuntimeCore;
+    composer: ThreadComposerRuntimeCore;
     getEditComposer: (messageId: string) => ComposerRuntimeCore | undefined;
     beginEdit: (messageId: string) => void;
     capabilities: Readonly<RuntimeCapabilities>;
@@ -168,11 +171,7 @@ type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult> | AsyncGenerator<ChatModelRunResult, void>;
 };
 
-type ReactThreadRuntimeCore = ThreadRuntimeCore & {
-    unstable_synchronizer?: ComponentType;
-};
-
-declare abstract class BaseAssistantRuntimeCore<TThreadRuntime extends ReactThreadRuntimeCore> implements AssistantRuntimeCore {
+declare abstract class BaseAssistantRuntimeCore<TThreadRuntime extends ThreadRuntimeCore> implements AssistantRuntimeCore {
     private _thread;
     constructor(_thread: TThreadRuntime);
     get thread(): TThreadRuntime;
@@ -189,17 +188,21 @@ type AttachmentAdapter = {
     accept: string;
     add(state: {
         file: File;
-    }): Promise<ThreadComposerAttachment>;
-    remove(attachment: ThreadComposerAttachment): Promise<void>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
+    }): Promise<Omit<PendingAttachment, "status"> & {
+        status?: PendingAttachmentStatus;
+    }>;
+    remove(attachment: Attachment): Promise<void>;
+    send(attachment: PendingAttachment): Promise<Omit<CompleteAttachment, "status"> & {
+        status?: CompleteAttachmentStatus;
+    }>;
 };
 
 declare class SimpleImageAttachmentAdapter implements AttachmentAdapter {
     accept: string;
     add(state: {
         file: File;
-    }): Promise<ThreadComposerAttachment>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
+    }): Promise<PendingAttachment>;
+    send(attachment: PendingAttachment): Promise<CompleteAttachment>;
     remove(): Promise<void>;
 }
 
@@ -207,8 +210,8 @@ declare class SimpleTextAttachmentAdapter implements AttachmentAdapter {
     accept: string;
     add(state: {
         file: File;
-    }): Promise<ThreadComposerAttachment>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
+    }): Promise<PendingAttachment>;
+    send(attachment: PendingAttachment): Promise<CompleteAttachment>;
     remove(): Promise<void>;
 }
 
@@ -218,17 +221,21 @@ declare class CompositeAttachmentAdapter implements AttachmentAdapter {
     constructor(adapters: AttachmentAdapter[]);
     add(state: {
         file: File;
-    }): Promise<ThreadComposerAttachment>;
-    send(attachment: ThreadComposerAttachment): Promise<MessageAttachment>;
-    remove(attachment: ThreadComposerAttachment): Promise<void>;
+    }): Promise<Omit<PendingAttachment, "status"> & {
+        status?: PendingAttachmentStatus;
+    }>;
+    send(attachment: PendingAttachment): Promise<Omit<CompleteAttachment, "status"> & {
+        status?: CompleteAttachmentStatus;
+    }>;
+    remove(attachment: Attachment): Promise<void>;
 }
 
 declare abstract class BaseComposerRuntimeCore implements ComposerRuntimeCore {
     readonly isEditing = true;
     attachmentAccept: string;
     private _attachments;
-    protected set attachments(value: readonly ThreadComposerAttachment[]);
-    get attachments(): readonly ThreadComposerAttachment[];
+    protected set attachments(value: readonly Attachment[]);
+    get attachments(): readonly Attachment[];
     abstract get canCancel(): boolean;
     get isEmpty(): boolean;
     private _text;
@@ -247,10 +254,11 @@ declare abstract class BaseComposerRuntimeCore implements ComposerRuntimeCore {
     subscribe(callback: () => void): Unsubscribe;
 }
 
-declare class DefaultThreadComposerRuntimeCore extends BaseComposerRuntimeCore {
+declare class DefaultThreadComposerRuntimeCore extends BaseComposerRuntimeCore implements ThreadComposerRuntimeCore {
     private runtime;
     private _canCancel;
     get canCancel(): boolean;
+    get attachments(): readonly PendingAttachment[];
     constructor(runtime: Omit<ThreadRuntimeCore, "composer">);
     connect(): Unsubscribe;
     handleSend(message: Omit<AppendMessage, "parentId">): Promise<void>;
@@ -380,7 +388,7 @@ declare const fromCoreMessages: (message: readonly CoreMessage[]) => ThreadMessa
 declare const fromCoreMessage: (message: CoreMessage, { id, status, attachments, }?: {
     id?: string | undefined;
     status?: MessageStatus | undefined;
-    attachments?: readonly MessageAttachment[] | undefined;
+    attachments?: readonly CompleteAttachment[] | undefined;
 }) => ThreadMessage;
 
 declare const toCoreMessages: (message: ThreadMessage[]) => CoreMessage[];
@@ -434,7 +442,7 @@ type ThreadMessageLike = {
     id?: string | undefined;
     createdAt?: Date | undefined;
     status?: MessageStatus | undefined;
-    attachments?: MessageAttachment[] | undefined;
+    attachments?: CompleteAttachment[] | undefined;
 };
 
 type ExternalStoreMessageConverter<T> = (message: T, idx: number) => ThreadMessageLike;
@@ -497,9 +505,10 @@ type SubscribableWithState<TState> = Subscribable & {
     getState: () => TState;
 };
 
-type ThreadComposerRuntimeCoreBinding = SubscribableWithState<ComposerRuntimeCore | undefined>;
+type ThreadComposerRuntimeCoreBinding = SubscribableWithState<ThreadComposerRuntimeCore | undefined>;
+type ComposerRuntimeCoreBinding = SubscribableWithState<ComposerRuntimeCore | undefined>;
 type LegacyEditComposerState = Readonly<{
-    type: "edit" | "thread";
+    type: "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. */
@@ -526,13 +535,13 @@ type LegacyEditComposerState = Readonly<{
     cancel: () => void;
 }>;
 type LegacyThreadComposerState = Readonly<{
-    type: "thread" | "edit";
+    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[];
+    attachments: readonly Attachment[];
     /** @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. */
@@ -556,24 +565,26 @@ type LegacyThreadComposerState = Readonly<{
     /** @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";
+type BaseComposerState = {
     text: string;
     attachmentAccept: string;
-    attachments: readonly ThreadComposerAttachment[];
+    attachments: readonly Attachment[];
     canCancel: boolean;
     isEditing: boolean;
     isEmpty: boolean;
-}>;
-/** @deprecated Use `ComposerState` instead. */
-type ThreadComposerState = ComposerState;
-/** @deprecated Use `ComposerState` instead. */
-type EditComposerState = ComposerState;
-declare class ComposerRuntime implements ComposerRuntimeCore {
-    private _core;
-    private _beginEdit?;
-    get type(): "edit" | "thread";
-    constructor(_core: ThreadComposerRuntimeCoreBinding, _beginEdit?: (() => void) | undefined);
+};
+type ThreadComposerState = LegacyThreadComposerState & BaseComposerState & {
+    type: "thread";
+    attachments: readonly PendingAttachment[];
+};
+type EditComposerState = LegacyEditComposerState & BaseComposerState & {
+    type: "edit";
+};
+type ComposerState = ThreadComposerState | EditComposerState;
+declare abstract class ComposerRuntime implements ComposerRuntimeCore {
+    protected _core: ComposerRuntimeCoreBinding;
+    abstract get type(): "edit" | "thread";
+    constructor(_core: ComposerRuntimeCoreBinding);
     /**
      * @deprecated Use `getState().isEditing` instead. This will be removed in 0.6.0.
      */
@@ -597,12 +608,12 @@ declare class ComposerRuntime implements ComposerRuntimeCore {
     /**
      * @deprecated Use `getState().attachments` instead. This will be removed in 0.6.0.
      */
-    get attachments(): readonly ThreadComposerAttachment[];
+    get attachments(): readonly Attachment[] | (readonly Attachment[] & readonly PendingAttachment[]);
     /**
      * @deprecated Use `getState().text` instead. This will be removed in 0.6.0.
      */
     get value(): string;
-    getState(): ComposerState;
+    abstract getState(): ComposerState;
     setText(text: string): void;
     setValue(text: string): void;
     addAttachment(file: File): Promise<void>;
@@ -613,15 +624,85 @@ declare class ComposerRuntime implements ComposerRuntimeCore {
     reset(): void;
     send(): void;
     cancel(): void;
+    subscribe(callback: () => void): Unsubscribe;
+    abstract unstable_getAttachmentByIndex(idx: number): AttachmentRuntime;
+}
+declare class ThreadComposerRuntime extends ComposerRuntime implements ThreadComposerState {
+    get type(): "thread";
+    private _getState;
+    constructor(core: ThreadComposerRuntimeCoreBinding);
+    get attachments(): readonly Attachment[] & readonly PendingAttachment[];
+    getState(): ThreadComposerState;
+    private _focusListeners;
+    focus(): void;
+    onFocus(callback: () => void): () => boolean;
+    unstable_getAttachmentByIndex(idx: number): ThreadComposerAttachmentRuntime;
+}
+declare class EditComposerRuntime extends ComposerRuntime implements EditComposerState {
+    private _beginEdit;
+    get type(): "edit";
+    private _getState;
+    constructor(core: ComposerRuntimeCoreBinding, _beginEdit: () => void);
+    getState(): EditComposerState;
     beginEdit(): void;
     /**
      * @deprecated Use `beginEdit()` instead. This will be removed in 0.6.0.
      */
     edit(): void;
+    unstable_getAttachmentByIndex(idx: number): EditComposerAttachmentRuntime;
+}
+
+type MessageAttachmentState = CompleteAttachment & {
+    source: "message";
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.attachment.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    attachment: CompleteAttachment;
+};
+type ThreadComposerAttachmentState = PendingAttachment & {
+    source: "thread-composer";
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.attachment.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    attachment: PendingAttachment;
+};
+type EditComposerAttachmentState = Attachment & {
+    source: "edit-composer";
+    /**
+     * @deprecated You can directly access content part fields in the state. Replace `.attachment.type` with `.type` etc. This will be removed in 0.6.0.
+     */
+    attachment: Attachment;
+};
+type AttachmentState = ThreadComposerAttachmentState | EditComposerAttachmentState | MessageAttachmentState;
+type AttachmentSnapshotBinding<Source extends AttachmentRuntimeSource> = SubscribableWithState<AttachmentState & {
+    source: Source;
+}>;
+type AttachmentRuntimeSource = AttachmentState["source"];
+declare abstract class AttachmentRuntime<Source extends AttachmentRuntimeSource = AttachmentRuntimeSource> {
+    private _core;
+    abstract get source(): Source;
+    constructor(_core: AttachmentSnapshotBinding<Source>);
+    getState(): AttachmentState & {
+        source: Source;
+    };
+    abstract remove(): Promise<void>;
     subscribe(callback: () => void): Unsubscribe;
-    private _focusListeners;
-    private focus;
-    private onFocus;
+}
+declare abstract class ComposerAttachmentRuntime<Source extends "thread-composer" | "edit-composer"> extends AttachmentRuntime<Source> {
+    private _composerApi;
+    constructor(core: AttachmentSnapshotBinding<Source>, _composerApi: ComposerRuntimeCoreBinding);
+    remove(): Promise<void>;
+}
+declare class ThreadComposerAttachmentRuntime extends ComposerAttachmentRuntime<"thread-composer"> {
+    get source(): "thread-composer";
+}
+declare class EditComposerAttachmentRuntime extends ComposerAttachmentRuntime<"edit-composer"> {
+    get source(): "edit-composer";
+}
+declare class MessageAttachmentRuntime extends AttachmentRuntime<"message"> {
+    get source(): "message";
+    constructor(core: AttachmentSnapshotBinding<"message">);
+    remove(): never;
 }
 
 type MessageState = ThreadMessage & {
@@ -643,9 +724,7 @@ declare class MessageRuntime {
     private _core;
     private _threadBinding;
     constructor(_core: MessageStateBinding, _threadBinding: ThreadRuntimeCoreBinding);
-    composer: ComposerRuntime & {
-        type: "edit";
-    };
+    composer: EditComposerRuntime;
     getState(): MessageState;
     unstable_edit(message: Omit<AppendMessage, "parentId">): void;
     reload(): void;
@@ -659,6 +738,7 @@ declare class MessageRuntime {
     }): void;
     subscribe(callback: () => void): Unsubscribe;
     unstable_getContentPartByIndex(idx: number): ContentPartRuntime;
+    unstable_getAttachmentByIndex(idx: number): MessageAttachmentRuntime;
 }
 
 type CreateAppendMessage = string | {
@@ -715,7 +795,7 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
         speak: (messageId: string) => SpeechSynthesisAdapter.Utterance;
         submitFeedback: (feedback: SubmitFeedbackOptions) => void;
         getModelConfig: () => ModelConfig;
-        composer: ComposerRuntimeCore;
+        composer: ThreadComposerRuntimeCore;
         getEditComposer: (messageId: string) => ComposerRuntimeCore | undefined;
         beginEdit: (messageId: string) => void;
         capabilities: Readonly<RuntimeCapabilities>;
@@ -728,7 +808,7 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
     }>;
     private _threadBinding;
     constructor(threadBinding: ThreadRuntimeCoreBinding);
-    readonly composer: ComposerRuntime;
+    readonly composer: ThreadComposerRuntime;
     getState(): Readonly<{
         threadId: string;
         isDisabled: boolean;
@@ -747,11 +827,11 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
     speak(messageId: string): SpeechSynthesisAdapter.Utterance;
     submitFeedback(options: SubmitFeedbackOptions): void;
     /**
-     * @deprecated This is a temporary API. This will be removed in 0.6.0.
+     * @deprecated Use `getMesssageById(id).unstable_getMessageByIndex(idx).composer` instead. This will be removed in 0.6.0.
      */
     getEditComposer(messageId: string): Readonly<{
         attachmentAccept: string;
-        attachments: readonly ThreadComposerAttachment[];
+        attachments: readonly Attachment[];
         addAttachment: (file: File) => Promise<void>;
         removeAttachment: (attachmentId: string) => Promise<void>;
         isEditing: boolean;
@@ -765,7 +845,7 @@ declare class ThreadRuntime implements ThreadRuntimeCore {
         subscribe: (callback: () => void) => Unsubscribe;
     }> | undefined;
     /**
-     * @deprecated This is a temporary API. This will be removed in 0.6.0.
+     * @deprecated Use `getMesssageById(id).unstable_getMessageByIndex(idx).composer.beginEdit()` instead. This will be removed in 0.6.0.
      */
     beginEdit(messageId: string): void;
     export(): ExportedMessageRepository;
@@ -1139,21 +1219,21 @@ declare const useThreadMessagesStore: {
     }): ReadonlyStore<readonly ThreadMessage[]> | null;
 };
 declare const useThreadComposer: {
-    (): ComposerState;
-    <TSelected>(selector: (state: ComposerState) => TSelected): TSelected;
+    (): ThreadComposerState;
+    <TSelected>(selector: (state: ThreadComposerState) => TSelected): TSelected;
     (options: {
         optional: true;
-    }): ComposerState | null;
+    }): ThreadComposerState | null;
     <TSelected>(options: {
         optional: true;
-        selector?: (state: ComposerState) => TSelected;
+        selector?: (state: ThreadComposerState) => TSelected;
     }): TSelected | null;
 };
 declare const useThreadComposerStore: {
-    (): ReadonlyStore<ComposerState>;
+    (): ReadonlyStore<ThreadComposerState>;
     (options: {
         optional: true;
-    }): ReadonlyStore<ComposerState> | null;
+    }): ReadonlyStore<ThreadComposerState> | null;
 };
 declare const useThreadViewport: {
     (): Readonly<{
@@ -1312,21 +1392,21 @@ declare const useMessageUtilsStore: {
     }>> | null;
 };
 declare const useEditComposer: {
-    (): ComposerState;
-    <TSelected>(selector: (state: ComposerState) => TSelected): TSelected;
+    (): EditComposerState;
+    <TSelected>(selector: (state: EditComposerState) => TSelected): TSelected;
     (options: {
         optional: true;
-    }): ComposerState | null;
+    }): EditComposerState | null;
     <TSelected>(options: {
         optional: true;
-        selector?: (state: ComposerState) => TSelected;
+        selector?: (state: EditComposerState) => TSelected;
     }): TSelected | null;
 };
 declare const useEditComposerStore: {
-    (): ReadonlyStore<ComposerState>;
+    (): ReadonlyStore<EditComposerState>;
     (options: {
         optional: true;
-    }): ReadonlyStore<ComposerState> | null;
+    }): ReadonlyStore<EditComposerState> | null;
 };
 
 type ContentPartContextValue = {
@@ -1366,7 +1446,7 @@ declare const useContentPartStore: {
 };
 
 type ComposerContextValue = {
-    useComposer: ReadonlyStore<ComposerState>;
+    useComposer: UseBoundStore<ReadonlyStore<ComposerState>>;
     type: "edit" | "new";
 };
 declare const useComposerContext: () => ComposerContextValue;
@@ -2205,6 +2285,8 @@ declare class ProxyConfigProvider implements ModelConfigProvider {
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
 }
 
+declare const generateId: (size?: number) => string;
+
 declare const useSmooth: (state: ContentPartState & TextContentPart, smooth?: boolean) => ContentPartState & TextContentPart;
 
 declare const withSmoothContextProvider: <C extends ComponentType<any>>(Component: C) => C;
@@ -2264,11 +2346,9 @@ declare const useSmoothStatus: {
     }): TSelected | null;
 };
 
-declare const generateId: (size?: number) => string;
-
 type internal_AssistantRuntime<TThreadRuntime extends ThreadRuntime = ThreadRuntime> = AssistantRuntime<TThreadRuntime>;
 declare const internal_AssistantRuntime: typeof AssistantRuntime;
-type internal_BaseAssistantRuntimeCore<TThreadRuntime extends ReactThreadRuntimeCore> = BaseAssistantRuntimeCore<TThreadRuntime>;
+type internal_BaseAssistantRuntimeCore<TThreadRuntime extends ThreadRuntimeCore> = BaseAssistantRuntimeCore<TThreadRuntime>;
 declare const internal_BaseAssistantRuntimeCore: typeof BaseAssistantRuntimeCore;
 type internal_DefaultThreadComposerRuntimeCore = DefaultThreadComposerRuntimeCore;
 declare const internal_DefaultThreadComposerRuntimeCore: typeof DefaultThreadComposerRuntimeCore;
@@ -2276,9 +2356,9 @@ type internal_MessageRepository = MessageRepository;
 declare const internal_MessageRepository: typeof MessageRepository;
 type internal_ProxyConfigProvider = ProxyConfigProvider;
 declare const internal_ProxyConfigProvider: typeof ProxyConfigProvider;
-type internal_ReactThreadRuntimeCore = ReactThreadRuntimeCore;
 type internal_ThreadRuntime = ThreadRuntime;
 declare const internal_ThreadRuntime: typeof ThreadRuntime;
+type internal_ThreadRuntimeCore = ThreadRuntimeCore;
 type internal_ThreadRuntimeCoreBinding = ThreadRuntimeCoreBinding;
 declare const internal_TooltipIconButton: typeof TooltipIconButton;
 declare const internal_generateId: typeof generateId;
@@ -2286,7 +2366,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_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 { internal_AssistantRuntime as AssistantRuntime, internal_BaseAssistantRuntimeCore as BaseAssistantRuntimeCore, internal_DefaultThreadComposerRuntimeCore as DefaultThreadComposerRuntimeCore, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, internal_ThreadRuntime as ThreadRuntime, type internal_ThreadRuntimeCore as ThreadRuntimeCore, 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, 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 };
+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, Attachment, 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, CompleteAttachment, _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, EditComposerRuntime, 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, PendingAttachment, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SpeechSynthesisAdapter, StreamUtils, type StringsConfig, type SubmitFeedbackOptions, type SuggestionConfig, TextContentPart, type TextContentPartComponent, type TextContentPartProps, TextContentPartProvider, _default$4 as Thread, ThreadAssistantContentPart, ThreadComposerRuntime, 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 };