@assistant-ui/react 0.4.8 → 0.5.2

package/dist/index.d.mts

@@ -2,7 +2,7 @@ import * as react from 'react';
 import { ReactNode, ComponentType, PropsWithChildren, ComponentPropsWithoutRef, FC } from 'react';
 import { z } from 'zod';
 import { JSONSchema7 } from 'json-schema';
-import { LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1Message, LanguageModelV1FunctionTool } from '@ai-sdk/provider';
+import { LanguageModelV1LogProbs, LanguageModelV1Message, LanguageModelV1FunctionTool } from '@ai-sdk/provider';
 import { UseBoundStore, StoreApi } from 'zustand';
 import { Primitive } from '@radix-ui/react-primitive';
 import * as PopoverPrimitive from '@radix-ui/react-popover';
@@ -100,19 +100,38 @@ type MessageCommonProps = {
     id: string;
     createdAt: Date;
 };
-type MessageStatus = {
-    type: "in_progress" | "cancelled";
-} | {
-    type: "done";
-    finishReason?: LanguageModelV1FinishReason;
-    logprops?: LanguageModelV1LogProbs;
+type ThreadRoundtrip = {
+    logprobs?: LanguageModelV1LogProbs | undefined;
     usage?: {
         promptTokens: number;
         completionTokens: number;
-    };
+    } | undefined;
+};
+type ContentPartStatus = {
+    type: "running";
+} | {
+    type: "complete";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "length" | "content-filter" | "other" | "error";
+    error?: unknown;
+};
+type ToolContentPartStatus = {
+    type: "requires-action";
+    reason: "tool-calls";
+} | ContentPartStatus;
+type MessageStatus = {
+    type: "running";
+} | {
+    type: "requires-action";
+    reason: "tool-calls";
 } | {
-    type: "error";
-    error: unknown;
+    type: "complete";
+    reason: "stop" | "unknown";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "tool-calls" | "length" | "content-filter" | "other" | "error";
+    error?: unknown;
 };
 type ThreadSystemMessage = MessageCommonProps & {
     role: "system";
@@ -126,6 +145,7 @@ type ThreadAssistantMessage = MessageCommonProps & {
     role: "assistant";
     content: ThreadAssistantContentPart[];
     status: MessageStatus;
+    roundtrips?: ThreadRoundtrip[] | undefined;
 };
 type AppendMessage = CoreMessage & {
     parentId: string | null;
@@ -160,12 +180,13 @@ type ChatModelRunUpdate = {
 type ChatModelRunResult = {
     content: ThreadAssistantContentPart[];
     status?: MessageStatus;
+    roundtrips?: ThreadRoundtrip[];
 };
 type ChatModelRunOptions = {
     messages: ThreadMessage[];
     abortSignal: AbortSignal;
     config: ModelConfig;
-    onUpdate: (result: ChatModelRunUpdate) => ThreadMessage;
+    onUpdate: (result: ChatModelRunUpdate) => void;
 };
 type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
@@ -183,19 +204,37 @@ declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRu
     private subscriptionHandler;
 }
 
+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: ToolContentPartStatus;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type LocalRuntimeOptions = {
     initialMessages?: readonly CoreMessage[] | undefined;
+    maxToolRoundtrips?: number;
 };
-declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
-    private readonly _proxyConfigProvider;
-    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
-    set adapter(adapter: ChatModelAdapter);
-    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
-    switchToThread(threadId: string | null): LocalThreadRuntime;
-}
+
 declare class LocalThreadRuntime implements ThreadRuntime {
     private configProvider;
     adapter: ChatModelAdapter;
+    private options?;
     private _subscriptions;
     private abortController;
     private readonly repository;
@@ -207,51 +246,43 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     }>;
     get messages(): ThreadMessage[];
     get isRunning(): boolean;
-    constructor(configProvider: ModelConfigProvider, adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
+    constructor(configProvider: ModelConfigProvider, adapter: ChatModelAdapter, options?: LocalRuntimeOptions | undefined);
     getBranches(messageId: string): string[];
     switchToBranch(branchId: string): void;
     append(message: AppendMessage): Promise<void>;
     startRun(parentId: string | null): Promise<void>;
+    private performRoundtrip;
     cancelRun(): void;
     private notifySubscribers;
     subscribe(callback: () => void): Unsubscribe;
     addToolResult({ messageId, toolCallId, result }: AddToolResultOptions): void;
 }
 
+declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
+    private readonly _proxyConfigProvider;
+    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
+    set adapter(adapter: ChatModelAdapter);
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+    switchToThread(threadId: string | null): LocalThreadRuntime;
+}
+
 declare const useLocalRuntime: (adapter: ChatModelAdapter, options?: LocalRuntimeOptions) => LocalRuntime;
 
 declare function toLanguageModelMessages(message: readonly CoreMessage[] | readonly ThreadMessage[]): LanguageModelV1Message[];
 
-type TextContentPartProps = {
-    part: TextContentPart;
-    status: MessageStatus;
-};
-type TextContentPartComponent = ComponentType<TextContentPartProps>;
-type ImageContentPartProps = {
-    part: ImageContentPart;
-    status: MessageStatus;
-};
-type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
-type UIContentPartProps = {
-    part: UIContentPart;
-    status: MessageStatus;
-};
-type UIContentPartComponent = ComponentType<UIContentPartProps>;
-type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = {
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: MessageStatus;
-    addResult: (result: any) => void;
-};
-type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
-
 type fromLanguageModelMessagesOptions = {
     mergeRoundtrips: boolean;
 };
 declare const fromLanguageModelMessages: (lm: LanguageModelV1Message[], { mergeRoundtrips }: fromLanguageModelMessagesOptions) => CoreMessage[];
 
 declare const fromCoreMessages: (message: readonly CoreMessage[]) => ThreadMessage[];
+declare const fromCoreMessage: (message: CoreMessage, { id, status, }?: {
+    id?: string | undefined;
+    status?: MessageStatus | undefined;
+}) => ThreadMessage;
 
 declare const toCoreMessages: (message: ThreadMessage[]) => CoreMessage[];
+declare const toCoreMessage: (message: ThreadMessage) => CoreMessage;
 
 declare const fromLanguageModelTools: (tools: LanguageModelV1FunctionTool[]) => Record<string, Tool<any, any>>;
 
@@ -259,12 +290,10 @@ declare const toLanguageModelTools: (tools: Record<string, Tool<any, any>>) => L
 
 type EdgeChatAdapterOptions = {
     api: string;
-    maxToolRoundtrips?: number;
 };
 declare class EdgeChatAdapter implements ChatModelAdapter {
     private options;
     constructor(options: EdgeChatAdapterOptions);
-    roundtrip(initialContent: ThreadAssistantContentPart[], { messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<readonly [ThreadMessage | undefined, ChatModelRunResult]>;
     run({ messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<ChatModelRunResult>;
 }
 
@@ -655,7 +684,7 @@ declare function useMessageContext(options: {
 }): MessageContextValue | null;
 
 type ContentPartState = Readonly<{
-    status: MessageStatus;
+    status: ToolContentPartStatus;
     part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
 }>;
 
@@ -735,17 +764,17 @@ declare const useComposerIf: (props: UseComposerIfProps) => boolean;
 declare const useComposerSend: () => (() => void) | null;
 
 declare const useContentPartDisplay: () => Readonly<{
-    status: MessageStatus;
+    status: ContentPartStatus;
     part: UIContentPart;
 }>;
 
 declare const useContentPartImage: () => Readonly<{
-    status: MessageStatus;
+    status: ContentPartStatus;
     part: ImageContentPart;
 }>;
 
 declare const useContentPartText: () => Readonly<{
-    status: MessageStatus;
+    status: ContentPartStatus;
     part: TextContentPart;
 }>;
 
@@ -1349,7 +1378,7 @@ declare class MessageRepository {
         parentId: string | null;
         message: ThreadMessage;
     };
-    appendOptimisticMessage(parentId: string | null, message: Omit<ThreadMessage, "id" | "createdAt">): string;
+    appendOptimisticMessage(parentId: string | null, message: CoreMessage): string;
     deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
     getBranches(messageId: string): string[];
     switchToBranch(messageId: string): void;
@@ -1373,4 +1402,4 @@ declare namespace internal {
   export { internal_BaseAssistantRuntime as BaseAssistantRuntime, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, internal_TooltipIconButton as TooltipIconButton, internal_generateId as generateId, internal_useSmooth as useSmooth };
 }
 
-export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type AddToolResultOptions, type AppendMessage, _default$9 as AssistantActionBar, type AssistantActionsState, type AssistantContextValue, _default$8 as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$7 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, type AssistantToolUIsState, _default$6 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, _default$5 as Composer, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, type ComposerState, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, type ContentPartState, type CoreAssistantContentPart, type CoreAssistantMessage, type CoreMessage, type CoreSystemMessage, type CoreUserContentPart, type CoreUserMessage, EdgeChatAdapter, type EdgeRuntimeOptions, type EdgeRuntimeRequestOptions, _default$4 as EditComposer, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, type MessageStatus, type MessageUtilsState, type ModelConfig, type ModelConfigProvider, type ReactThreadRuntime, type StringsConfig, type SuggestionConfig, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, type ThreadActionsState, type ThreadAssistantContentPart, type ThreadAssistantMessage, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, type ThreadMessage, type ThreadMessagesState, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, type ThreadRuntime, type ThreadState, type ThreadSystemMessage, type ThreadUserContentPart, type ThreadUserMessage, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, type Tool, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UseActionBarCopyProps, _default$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, makeAssistantTool, makeAssistantToolUI, toCoreMessages, toLanguageModelMessages, toLanguageModelTools, useActionBarCopy, useActionBarEdit, useActionBarReload, useAppendMessage, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartText, useEdgeRuntime, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };
+export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type AddToolResultOptions, type AppendMessage, _default$9 as AssistantActionBar, type AssistantActionsState, type AssistantContextValue, _default$8 as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$7 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, type AssistantToolUIsState, _default$6 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, _default$5 as Composer, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, type ComposerState, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, type ContentPartState, type CoreAssistantContentPart, type CoreAssistantMessage, type CoreMessage, type CoreSystemMessage, type CoreUserContentPart, type CoreUserMessage, EdgeChatAdapter, type EdgeRuntimeOptions, type EdgeRuntimeRequestOptions, _default$4 as EditComposer, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, type MessageStatus, type MessageUtilsState, type ModelConfig, type ModelConfigProvider, type ReactThreadRuntime, type StringsConfig, type SuggestionConfig, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, type ThreadActionsState, type ThreadAssistantContentPart, type ThreadAssistantMessage, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, type ThreadMessage, type ThreadMessagesState, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, type ThreadRuntime, type ThreadState, type ThreadSystemMessage, type ThreadUserContentPart, type ThreadUserMessage, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, type Tool, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UseActionBarCopyProps, _default$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, fromCoreMessage, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, makeAssistantTool, makeAssistantToolUI, toCoreMessage, toCoreMessages, toLanguageModelMessages, toLanguageModelTools, useActionBarCopy, useActionBarEdit, useActionBarReload, useAppendMessage, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartText, useEdgeRuntime, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };

package/dist/index.d.ts

@@ -2,7 +2,7 @@ import * as react from 'react';
 import { ReactNode, ComponentType, PropsWithChildren, ComponentPropsWithoutRef, FC } from 'react';
 import { z } from 'zod';
 import { JSONSchema7 } from 'json-schema';
-import { LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1Message, LanguageModelV1FunctionTool } from '@ai-sdk/provider';
+import { LanguageModelV1LogProbs, LanguageModelV1Message, LanguageModelV1FunctionTool } from '@ai-sdk/provider';
 import { UseBoundStore, StoreApi } from 'zustand';
 import { Primitive } from '@radix-ui/react-primitive';
 import * as PopoverPrimitive from '@radix-ui/react-popover';
@@ -100,19 +100,38 @@ type MessageCommonProps = {
     id: string;
     createdAt: Date;
 };
-type MessageStatus = {
-    type: "in_progress" | "cancelled";
-} | {
-    type: "done";
-    finishReason?: LanguageModelV1FinishReason;
-    logprops?: LanguageModelV1LogProbs;
+type ThreadRoundtrip = {
+    logprobs?: LanguageModelV1LogProbs | undefined;
     usage?: {
         promptTokens: number;
         completionTokens: number;
-    };
+    } | undefined;
+};
+type ContentPartStatus = {
+    type: "running";
+} | {
+    type: "complete";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "length" | "content-filter" | "other" | "error";
+    error?: unknown;
+};
+type ToolContentPartStatus = {
+    type: "requires-action";
+    reason: "tool-calls";
+} | ContentPartStatus;
+type MessageStatus = {
+    type: "running";
+} | {
+    type: "requires-action";
+    reason: "tool-calls";
 } | {
-    type: "error";
-    error: unknown;
+    type: "complete";
+    reason: "stop" | "unknown";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "tool-calls" | "length" | "content-filter" | "other" | "error";
+    error?: unknown;
 };
 type ThreadSystemMessage = MessageCommonProps & {
     role: "system";
@@ -126,6 +145,7 @@ type ThreadAssistantMessage = MessageCommonProps & {
     role: "assistant";
     content: ThreadAssistantContentPart[];
     status: MessageStatus;
+    roundtrips?: ThreadRoundtrip[] | undefined;
 };
 type AppendMessage = CoreMessage & {
     parentId: string | null;
@@ -160,12 +180,13 @@ type ChatModelRunUpdate = {
 type ChatModelRunResult = {
     content: ThreadAssistantContentPart[];
     status?: MessageStatus;
+    roundtrips?: ThreadRoundtrip[];
 };
 type ChatModelRunOptions = {
     messages: ThreadMessage[];
     abortSignal: AbortSignal;
     config: ModelConfig;
-    onUpdate: (result: ChatModelRunUpdate) => ThreadMessage;
+    onUpdate: (result: ChatModelRunUpdate) => void;
 };
 type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
@@ -183,19 +204,37 @@ declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRu
     private subscriptionHandler;
 }
 
+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: ToolContentPartStatus;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type LocalRuntimeOptions = {
     initialMessages?: readonly CoreMessage[] | undefined;
+    maxToolRoundtrips?: number;
 };
-declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
-    private readonly _proxyConfigProvider;
-    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
-    set adapter(adapter: ChatModelAdapter);
-    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
-    switchToThread(threadId: string | null): LocalThreadRuntime;
-}
+
 declare class LocalThreadRuntime implements ThreadRuntime {
     private configProvider;
     adapter: ChatModelAdapter;
+    private options?;
     private _subscriptions;
     private abortController;
     private readonly repository;
@@ -207,51 +246,43 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     }>;
     get messages(): ThreadMessage[];
     get isRunning(): boolean;
-    constructor(configProvider: ModelConfigProvider, adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
+    constructor(configProvider: ModelConfigProvider, adapter: ChatModelAdapter, options?: LocalRuntimeOptions | undefined);
     getBranches(messageId: string): string[];
     switchToBranch(branchId: string): void;
     append(message: AppendMessage): Promise<void>;
     startRun(parentId: string | null): Promise<void>;
+    private performRoundtrip;
     cancelRun(): void;
     private notifySubscribers;
     subscribe(callback: () => void): Unsubscribe;
     addToolResult({ messageId, toolCallId, result }: AddToolResultOptions): void;
 }
 
+declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
+    private readonly _proxyConfigProvider;
+    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
+    set adapter(adapter: ChatModelAdapter);
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+    switchToThread(threadId: string | null): LocalThreadRuntime;
+}
+
 declare const useLocalRuntime: (adapter: ChatModelAdapter, options?: LocalRuntimeOptions) => LocalRuntime;
 
 declare function toLanguageModelMessages(message: readonly CoreMessage[] | readonly ThreadMessage[]): LanguageModelV1Message[];
 
-type TextContentPartProps = {
-    part: TextContentPart;
-    status: MessageStatus;
-};
-type TextContentPartComponent = ComponentType<TextContentPartProps>;
-type ImageContentPartProps = {
-    part: ImageContentPart;
-    status: MessageStatus;
-};
-type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
-type UIContentPartProps = {
-    part: UIContentPart;
-    status: MessageStatus;
-};
-type UIContentPartComponent = ComponentType<UIContentPartProps>;
-type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = {
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: MessageStatus;
-    addResult: (result: any) => void;
-};
-type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
-
 type fromLanguageModelMessagesOptions = {
     mergeRoundtrips: boolean;
 };
 declare const fromLanguageModelMessages: (lm: LanguageModelV1Message[], { mergeRoundtrips }: fromLanguageModelMessagesOptions) => CoreMessage[];
 
 declare const fromCoreMessages: (message: readonly CoreMessage[]) => ThreadMessage[];
+declare const fromCoreMessage: (message: CoreMessage, { id, status, }?: {
+    id?: string | undefined;
+    status?: MessageStatus | undefined;
+}) => ThreadMessage;
 
 declare const toCoreMessages: (message: ThreadMessage[]) => CoreMessage[];
+declare const toCoreMessage: (message: ThreadMessage) => CoreMessage;
 
 declare const fromLanguageModelTools: (tools: LanguageModelV1FunctionTool[]) => Record<string, Tool<any, any>>;
 
@@ -259,12 +290,10 @@ declare const toLanguageModelTools: (tools: Record<string, Tool<any, any>>) => L
 
 type EdgeChatAdapterOptions = {
     api: string;
-    maxToolRoundtrips?: number;
 };
 declare class EdgeChatAdapter implements ChatModelAdapter {
     private options;
     constructor(options: EdgeChatAdapterOptions);
-    roundtrip(initialContent: ThreadAssistantContentPart[], { messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<readonly [ThreadMessage | undefined, ChatModelRunResult]>;
     run({ messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<ChatModelRunResult>;
 }
 
@@ -655,7 +684,7 @@ declare function useMessageContext(options: {
 }): MessageContextValue | null;
 
 type ContentPartState = Readonly<{
-    status: MessageStatus;
+    status: ToolContentPartStatus;
     part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
 }>;
 
@@ -735,17 +764,17 @@ declare const useComposerIf: (props: UseComposerIfProps) => boolean;
 declare const useComposerSend: () => (() => void) | null;
 
 declare const useContentPartDisplay: () => Readonly<{
-    status: MessageStatus;
+    status: ContentPartStatus;
     part: UIContentPart;
 }>;
 
 declare const useContentPartImage: () => Readonly<{
-    status: MessageStatus;
+    status: ContentPartStatus;
     part: ImageContentPart;
 }>;
 
 declare const useContentPartText: () => Readonly<{
-    status: MessageStatus;
+    status: ContentPartStatus;
     part: TextContentPart;
 }>;
 
@@ -1349,7 +1378,7 @@ declare class MessageRepository {
         parentId: string | null;
         message: ThreadMessage;
     };
-    appendOptimisticMessage(parentId: string | null, message: Omit<ThreadMessage, "id" | "createdAt">): string;
+    appendOptimisticMessage(parentId: string | null, message: CoreMessage): string;
     deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
     getBranches(messageId: string): string[];
     switchToBranch(messageId: string): void;
@@ -1373,4 +1402,4 @@ declare namespace internal {
   export { internal_BaseAssistantRuntime as BaseAssistantRuntime, internal_MessageRepository as MessageRepository, internal_ProxyConfigProvider as ProxyConfigProvider, internal_TooltipIconButton as TooltipIconButton, internal_generateId as generateId, internal_useSmooth as useSmooth };
 }
 
-export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type AddToolResultOptions, type AppendMessage, _default$9 as AssistantActionBar, type AssistantActionsState, type AssistantContextValue, _default$8 as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$7 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, type AssistantToolUIsState, _default$6 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, _default$5 as Composer, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, type ComposerState, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, type ContentPartState, type CoreAssistantContentPart, type CoreAssistantMessage, type CoreMessage, type CoreSystemMessage, type CoreUserContentPart, type CoreUserMessage, EdgeChatAdapter, type EdgeRuntimeOptions, type EdgeRuntimeRequestOptions, _default$4 as EditComposer, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, type MessageStatus, type MessageUtilsState, type ModelConfig, type ModelConfigProvider, type ReactThreadRuntime, type StringsConfig, type SuggestionConfig, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, type ThreadActionsState, type ThreadAssistantContentPart, type ThreadAssistantMessage, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, type ThreadMessage, type ThreadMessagesState, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, type ThreadRuntime, type ThreadState, type ThreadSystemMessage, type ThreadUserContentPart, type ThreadUserMessage, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, type Tool, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UseActionBarCopyProps, _default$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, makeAssistantTool, makeAssistantToolUI, toCoreMessages, toLanguageModelMessages, toLanguageModelTools, useActionBarCopy, useActionBarEdit, useActionBarReload, useAppendMessage, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartText, useEdgeRuntime, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };
+export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, type AddToolResultOptions, type AppendMessage, _default$9 as AssistantActionBar, type AssistantActionsState, type AssistantContextValue, _default$8 as AssistantMessage, type AssistantMessageConfig, type AssistantMessageContentProps, _default$7 as AssistantModal, index$5 as AssistantModalPrimitive, type AssistantModalPrimitiveContentProps, type AssistantModalPrimitiveRootProps, type AssistantModalPrimitiveTriggerProps, type AssistantModelConfigState, type AssistantRuntime, AssistantRuntimeProvider, type AssistantToolProps, type AssistantToolUIProps, type AssistantToolUIsState, _default$6 as BranchPicker, index$4 as BranchPickerPrimitive, type BranchPickerPrimitiveCountProps, type BranchPickerPrimitiveNextProps, type BranchPickerPrimitiveNumberProps, type BranchPickerPrimitivePreviousProps, type BranchPickerPrimitiveRootProps, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, _default$5 as Composer, type ComposerContextValue, type ComposerInputProps, index$3 as ComposerPrimitive, type ComposerPrimitiveCancelProps, type ComposerPrimitiveIfProps, type ComposerPrimitiveInputProps, type ComposerPrimitiveRootProps, type ComposerPrimitiveSendProps, type ComposerState, exports as ContentPart, type ContentPartContextValue, index$2 as ContentPartPrimitive, type ContentPartPrimitiveDisplayProps, type ContentPartPrimitiveImageProps, type ContentPartPrimitiveInProgressProps, type ContentPartPrimitiveTextProps, type ContentPartState, type CoreAssistantContentPart, type CoreAssistantMessage, type CoreMessage, type CoreSystemMessage, type CoreUserContentPart, type CoreUserMessage, EdgeChatAdapter, type EdgeRuntimeOptions, type EdgeRuntimeRequestOptions, _default$4 as EditComposer, type EditComposerState, internal as INTERNAL, type ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, type MessageStatus, type MessageUtilsState, type ModelConfig, type ModelConfigProvider, type ReactThreadRuntime, type StringsConfig, type SuggestionConfig, type TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, type ThreadActionsState, type ThreadAssistantContentPart, type ThreadAssistantMessage, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, type ThreadMessage, type ThreadMessagesState, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, type ThreadRuntime, type ThreadState, type ThreadSystemMessage, type ThreadUserContentPart, type ThreadUserMessage, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, type Tool, type ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, type UIContentPart, type UIContentPartComponent, type UIContentPartProps, type Unsubscribe, type UseActionBarCopyProps, _default$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, fromCoreMessage, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, makeAssistantTool, makeAssistantToolUI, toCoreMessage, toCoreMessages, toLanguageModelMessages, toLanguageModelTools, useActionBarCopy, useActionBarEdit, useActionBarReload, useAppendMessage, useAssistantContext, useAssistantInstructions, useAssistantTool, useAssistantToolUI, useBranchPickerCount, useBranchPickerNext, useBranchPickerNumber, useBranchPickerPrevious, useComposerCancel, useComposerContext, useComposerIf, useComposerSend, useContentPartContext, useContentPartDisplay, useContentPartImage, useContentPartText, useEdgeRuntime, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };