@assistant-ui/react 0.5.21 → 0.5.23

package/dist/index.d.ts

@@ -1,9 +1,7 @@
 import * as react from 'react';
 import { ComponentType, PropsWithChildren, ComponentPropsWithoutRef, FC, ElementType, ReactNode } from 'react';
-import { T as ThreadRuntime, U as Unsubscribe, A as AddToolResultOptions, B as BaseAssistantRuntime, R as ReactThreadRuntime, a as AssistantRuntime, b as ThreadState, c as ThreadActionsState, C as ContentPartState, d as TooltipIconButtonProps, e as ButtonProps } from './Thread-jfAlPLli.js';
-export { i as INTERNAL } from './Thread-jfAlPLli.js';
-import { T as ThreadAssistantContentPart, M as MessageStatus, a as ThreadRoundtrip, b as ThreadMessage, c as ModelConfig, d as TextContentPart, C as ContentPartStatus, I as ImageContentPart, U as UIContentPart, e as ToolCallContentPart, f as ToolCallContentPartStatus, g as CoreMessage, h as ModelConfigProvider, A as AppendMessage, i as Tool, j as CoreToolCallContentPart } from './AssistantTypes-D93BmqD5.js';
-export { p as CoreAssistantContentPart, s as CoreAssistantMessage, q as CoreSystemMessage, o as CoreUserContentPart, r as CoreUserMessage, m as ThreadAssistantMessage, l as ThreadSystemMessage, k as ThreadUserContentPart, n as ThreadUserMessage } from './AssistantTypes-D93BmqD5.js';
+import { T as ThreadAssistantContentPart, M as MessageStatus, a as ThreadRoundtrip, b as ThreadMessage, c as ModelConfig, d as ModelConfigProvider, e as TextContentPart, C as ContentPartStatus, I as ImageContentPart, U as UIContentPart, f as ToolCallContentPart, g as ToolCallContentPartStatus, h as CoreMessage, A as AppendMessage, i as Tool, j as CoreToolCallContentPart } from './AssistantTypes-BNB-knVq.js';
+export { p as CoreAssistantContentPart, s as CoreAssistantMessage, q as CoreSystemMessage, o as CoreUserContentPart, r as CoreUserMessage, m as ThreadAssistantMessage, l as ThreadSystemMessage, k as ThreadUserContentPart, n as ThreadUserMessage } from './AssistantTypes-BNB-knVq.js';
 import { UseBoundStore, StoreApi } from 'zustand';
 import { Primitive } from '@radix-ui/react-primitive';
 import * as PopoverPrimitive from '@radix-ui/react-popover';
@@ -12,11 +10,18 @@ import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { LanguageModelV1Message, LanguageModelV1FunctionTool } from '@ai-sdk/provider';
 import { JSONSchema7 } from 'json-schema';
 import { z } from 'zod';
-import 'class-variance-authority';
-import 'class-variance-authority/types';
+import * as class_variance_authority from 'class-variance-authority';
+import { VariantProps } from 'class-variance-authority';
+import * as class_variance_authority_types from 'class-variance-authority/types';
+
+type Unsubscribe = () => void;
 
 type ReadonlyStore<T> = UseBoundStore<Omit<StoreApi<T>, "setState" | "destroy">>;
 
+type ReactThreadRuntime = ThreadRuntime & {
+    unstable_synchronizer?: ComponentType;
+};
+
 type ChatModelRunUpdate = {
     content: ThreadAssistantContentPart[];
 };
@@ -38,6 +43,18 @@ type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult> | AsyncGenerator<ChatModelRunResult, void>;
 };
 
+declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRuntime> implements AssistantRuntime {
+    private _thread;
+    constructor(_thread: TThreadRuntime);
+    get thread(): TThreadRuntime;
+    set thread(thread: TThreadRuntime);
+    abstract registerModelConfigProvider(provider: ModelConfigProvider): Unsubscribe;
+    abstract switchToThread(threadId: string | null): void;
+    private _subscriptions;
+    subscribe(callback: () => void): Unsubscribe;
+    private subscriptionHandler;
+}
+
 type TextContentPartProps = {
     part: TextContentPart;
     status: ContentPartStatus;
@@ -60,6 +77,33 @@ type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResu
 };
 type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
 
+interface ExportedMessageRepository {
+    headId?: string | null;
+    messages: Array<{
+        message: ThreadMessage;
+        parentId: string | null;
+    }>;
+}
+declare class MessageRepository {
+    private messages;
+    private head;
+    private root;
+    private performOp;
+    getMessages(): ThreadMessage[];
+    addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
+    getMessage(messageId: string): {
+        parentId: string | null;
+        message: ThreadMessage;
+    };
+    appendOptimisticMessage(parentId: string | null, message: CoreMessage): string;
+    deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
+    getBranches(messageId: string): string[];
+    switchToBranch(messageId: string): void;
+    resetHead(messageId: string | null): void;
+    export(): ExportedMessageRepository;
+    import({ headId, messages }: ExportedMessageRepository): void;
+}
+
 type LocalRuntimeOptions = {
     initialMessages?: readonly CoreMessage[] | undefined;
     maxToolRoundtrips?: number;
@@ -92,6 +136,8 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     private notifySubscribers;
     subscribe(callback: () => void): Unsubscribe;
     addToolResult({ messageId, toolCallId, result }: AddToolResultOptions): void;
+    export(): ExportedMessageRepository;
+    import(data: ExportedMessageRepository): void;
 }
 
 declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
@@ -416,6 +462,76 @@ type ExternalStoreAdapterBase<T> = {
 };
 type ExternalStoreAdapter<T = ThreadMessage> = ExternalStoreAdapterBase<T> & (T extends ThreadMessage ? object : ExternalStoreMessageConverterAdapter<T>);
 
+declare class ProxyConfigProvider implements ModelConfigProvider {
+    private _providers;
+    getModelConfig(): ModelConfig;
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+}
+
+type TextContentPartState = Readonly<{
+    status: ContentPartStatus;
+    part: TextContentPart;
+}>;
+type ContentPartState = Readonly<{
+    status: ContentPartStatus | ToolCallContentPartStatus;
+    part: TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
+}>;
+
+declare const useSmooth: (state: TextContentPartState, smooth?: boolean) => TextContentPartState;
+
+declare const withSmoothContextProvider: <C extends React.ComponentType<any>>(Component: C) => C;
+declare const useSmoothStatus: () => {
+    type: "running";
+} | {
+    type: "complete";
+} | {
+    type: "incomplete";
+    reason: "cancelled" | "length" | "content-filter" | "other" | "error";
+    error?: unknown;
+} | {
+    type: "requires-action";
+    reason: "tool-calls";
+};
+
+declare const buttonVariants: (props?: ({
+    variant?: "default" | "outline" | "ghost" | null | undefined;
+    size?: "default" | "icon" | null | undefined;
+} & class_variance_authority_types.ClassProp) | undefined) => string;
+type ButtonProps = React.ComponentPropsWithoutRef<typeof Primitive.button> & VariantProps<typeof buttonVariants>;
+
+type TooltipIconButtonProps = ButtonProps & {
+    tooltip: string;
+    side?: "top" | "bottom" | "left" | "right";
+};
+declare const TooltipIconButton: react.ForwardRefExoticComponent<Omit<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
+    ref?: ((instance: HTMLButtonElement | null) => void | react.DO_NOT_USE_OR_YOU_WILL_BE_FIRED_CALLBACK_REF_RETURN_VALUES[keyof react.DO_NOT_USE_OR_YOU_WILL_BE_FIRED_CALLBACK_REF_RETURN_VALUES]) | react.RefObject<HTMLButtonElement> | null | undefined;
+} & {
+    asChild?: boolean;
+}, "ref"> & class_variance_authority.VariantProps<(props?: ({
+    variant?: "default" | "outline" | "ghost" | null | undefined;
+    size?: "default" | "icon" | null | undefined;
+} & class_variance_authority_types.ClassProp) | undefined) => string> & {
+    tooltip: string;
+    side?: "top" | "bottom" | "left" | "right";
+} & react.RefAttributes<HTMLButtonElement>>;
+
+declare const generateId: (size?: number) => string;
+
+type internal_BaseAssistantRuntime<TThreadRuntime extends ReactThreadRuntime> = BaseAssistantRuntime<TThreadRuntime>;
+declare const internal_BaseAssistantRuntime: typeof BaseAssistantRuntime;
+type internal_MessageRepository = MessageRepository;
+declare const internal_MessageRepository: typeof MessageRepository;
+type internal_ProxyConfigProvider = ProxyConfigProvider;
+declare const internal_ProxyConfigProvider: typeof ProxyConfigProvider;
+declare const internal_TooltipIconButton: typeof TooltipIconButton;
+declare const internal_generateId: typeof generateId;
+declare const internal_useSmooth: typeof useSmooth;
+declare const internal_useSmoothStatus: typeof useSmoothStatus;
+declare const internal_withSmoothContextProvider: typeof withSmoothContextProvider;
+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, internal_useSmoothStatus as useSmoothStatus, internal_withSmoothContextProvider as withSmoothContextProvider };
+}
+
 declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     store: ExternalStoreAdapter<any>;
     private _subscriptions;
@@ -462,6 +578,42 @@ declare const getExternalStoreMessage: <T>(message: ThreadMessage) => T | undefi
 
 type ThreadRuntimeStore = ThreadRuntime;
 
+type AddToolResultOptions = {
+    messageId: string;
+    toolCallId: string;
+    result: any;
+};
+type RuntimeCapabilities = {
+    switchToBranch: boolean;
+    edit: boolean;
+    reload: boolean;
+    cancel: boolean;
+    copy: boolean;
+};
+type ThreadActionsState = Readonly<{
+    capabilities: Readonly<RuntimeCapabilities>;
+    getBranches: (messageId: string) => readonly string[];
+    switchToBranch: (branchId: string) => void;
+    append: (message: AppendMessage) => void;
+    startRun: (parentId: string | null) => void;
+    cancelRun: () => void;
+    addToolResult: (options: AddToolResultOptions) => void;
+}>;
+
+type ThreadRuntime = Readonly<ThreadState & Omit<ThreadActionsState, "getRuntime"> & {
+    messages: readonly ThreadMessage[];
+    subscribe: (callback: () => void) => Unsubscribe;
+}>;
+
+type ThreadRuntimeWithSubscribe = {
+    readonly thread: ThreadRuntime;
+    subscribe: (callback: () => void) => Unsubscribe;
+};
+type AssistantRuntime = ThreadRuntimeWithSubscribe & {
+    switchToThread: (threadId: string | null) => void;
+    registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
+};
+
 type AssistantRuntimeProviderProps = {
     runtime: AssistantRuntime;
 };
@@ -496,6 +648,11 @@ declare function useAssistantContext(options: {
     optional: true;
 }): AssistantContextValue | null;
 
+type ThreadState = Readonly<{
+    isRunning: boolean;
+    isDisabled: boolean;
+}>;
+
 type ThreadViewportState = Readonly<{
     isAtBottom: boolean;
     scrollToBottom: () => void;
@@ -1214,4 +1371,4 @@ declare const exports: {
     Text: FC;
 };
 
-export { index$6 as ActionBarPrimitive, type ActionBarPrimitiveCopyProps, type ActionBarPrimitiveEditProps, type ActionBarPrimitiveReloadProps, type ActionBarPrimitiveRootProps, AddToolResultOptions, 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, 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, ContentPartState, CoreMessage, EdgeChatAdapter, type EdgeRuntimeOptions, type EdgeRuntimeRequestOptions, _default$4 as EditComposer, type EditComposerState, type ExternalStoreAdapter, type ExternalStoreMessageConverter, ExternalStoreRuntime, ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, MessageStatus, type MessageUtilsState, ModelConfig, ModelConfigProvider, ReactThreadRuntime, type StringsConfig, type SuggestionConfig, TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, ThreadActionsState, ThreadAssistantContentPart, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, ThreadMessage, type ThreadMessageLike, type ThreadMessagesState, index as ThreadPrimitive, type ThreadPrimitiveEmptyProps, type ThreadPrimitiveIfProps, type ThreadPrimitiveMessagesProps, type ThreadPrimitiveRootProps, type ThreadPrimitiveScrollToBottomProps, type ThreadPrimitiveSuggestionProps, type ThreadPrimitiveViewportProps, type ThreadRootProps, ThreadRuntime, ThreadState, type ThreadViewportState, _default as ThreadWelcome, type ThreadWelcomeConfig, type ThreadWelcomeMessageProps, type ThreadWelcomeSuggestionProps, Tool, ToolCallContentPart, type ToolCallContentPartComponent, type ToolCallContentPartProps, UIContentPart, type UIContentPartComponent, type UIContentPartProps, Unsubscribe, type UseActionBarCopyProps, _default$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, fromCoreMessage, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, getExternalStoreMessage, 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, useExternalStoreRuntime, 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, 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, CoreMessage, EdgeChatAdapter, type EdgeRuntimeOptions, type EdgeRuntimeRequestOptions, _default$4 as EditComposer, type EditComposerState, type ExternalStoreAdapter, type ExternalStoreMessageConverter, ExternalStoreRuntime, internal as INTERNAL, ImageContentPart, type ImageContentPartComponent, type ImageContentPartProps, type LocalRuntimeOptions, type MessageContextValue, index$1 as MessagePrimitive, type MessagePrimitiveContentProps, type MessagePrimitiveIfProps, type MessagePrimitiveInProgressProps, type MessagePrimitiveRootProps, type MessageState, MessageStatus, type MessageUtilsState, ModelConfig, ModelConfigProvider, type ReactThreadRuntime, type StringsConfig, type SuggestionConfig, TextContentPart, type TextContentPartComponent, type TextContentPartProps, _default$3 as Thread, type ThreadActionsState, ThreadAssistantContentPart, type ThreadConfig, ThreadConfigProvider, type ThreadConfigProviderProps, type ThreadContextValue, ThreadMessage, type ThreadMessageLike, 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 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$1 as UserActionBar, _default$2 as UserMessage, type UserMessageConfig, type UserMessageContentProps, fromCoreMessage, fromCoreMessages, fromLanguageModelMessages, fromLanguageModelTools, getExternalStoreMessage, 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, useExternalStoreRuntime, useLocalRuntime, useMessageContext, useMessageIf, useSwitchToNewThread, useThreadConfig, useThreadContext, useThreadEmpty, useThreadIf, useThreadScrollToBottom, useThreadSuggestion };