@assistant-ui/react 0.0.29 → 0.1.1

package/dist/index.d.ts

@@ -1,10 +1,10 @@
 import * as react from 'react';
 import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
-import { UseChatHelpers, UseAssistantHelpers } from '@ai-sdk/react';
-import * as react_jsx_runtime from 'react/jsx-runtime';
-import { UseBoundStore, StoreApi } from 'zustand';
-import { Message } from 'ai';
+import { T as TextContentPart, I as ImageContentPart, U as UIContentPart, a as ToolCallContentPart, M as ModelConfigProvider, b as ThreadMessage, c as ModelConfig, A as AssistantContentPart, d as AppendMessage } from './ModelConfigTypes-CzmXY3sn.js';
+export { h as AppendContentPart, e as AssistantMessage, g as UserContentPart, f as UserMessage } from './ModelConfigTypes-CzmXY3sn.js';
+import { T as ThreadState, U as Unsubscribe } from './Thread-BMASJT4a.js';
+import 'zod';
 
 declare const useCopyMessage: ({ copiedDuration }: {
     copiedDuration?: number | undefined;
@@ -133,46 +133,6 @@ type MessageIfProps = RequireAtLeastOne<MessageIfFilters> & {
 };
 declare const MessageIf: FC<MessageIfProps>;
 
-type TextContentPart = {
-    type: "text";
-    text: string;
-};
-type ImageContentPart = {
-    type: "image";
-    image: string;
-};
-type UIContentPart = {
-    type: "ui";
-    display: ReactNode;
-};
-type ToolCallContentPart = {
-    type: "tool-call";
-    name: string;
-    args: object;
-    result?: object;
-};
-type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
-type AssistantContentPart = TextContentPart | UIContentPart | ToolCallContentPart;
-type AppendContentPart = TextContentPart | ImageContentPart;
-type BaseMessage = {
-    id: string;
-    createdAt: Date;
-};
-type UserMessage = BaseMessage & {
-    role: "user";
-    content: UserContentPart[];
-};
-type AssistantMessage = BaseMessage & {
-    role: "assistant";
-    content: AssistantContentPart[];
-    status: "in_progress" | "done" | "error";
-};
-type AppendMessage = {
-    parentId: string | null;
-    content: AppendContentPart[];
-};
-type ThreadMessage = UserMessage | AssistantMessage;
-
 type MessageContentProps = {
     components?: {
         Text?: ComponentType<{
@@ -275,206 +235,25 @@ declare namespace index {
   export { ContentPartInProgressIndicator as InProgressIndicator };
 }
 
-type VercelRSCMessage = {
-    id: string;
-    role: "user" | "assistant";
-    display: ReactNode;
-    createdAt?: Date;
-};
-
-type RSCMessageConverter<T> = {
-    convertMessage: (message: T) => VercelRSCMessage;
-};
-type VercelRSCAdapterBase<T> = {
-    messages: T[];
-    append: (message: AppendMessage) => Promise<void>;
-    edit?: (message: AppendMessage) => Promise<void>;
-    reload?: (parentId: string | null) => Promise<void>;
-    convertMessage?: (message: T) => VercelRSCMessage;
-};
-type VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
-
-declare class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage> implements AssistantRuntime, ReactThreadRuntime {
-    adapter: VercelRSCAdapter<T>;
-    private useAdapter;
-    private _subscriptions;
-    isRunning: boolean;
-    messages: ThreadMessage[];
-    constructor(adapter: VercelRSCAdapter<T>);
-    private withRunning;
-    getBranches(): readonly string[];
-    switchToBranch(): void;
-    append(message: AppendMessage): Promise<void>;
-    startRun(parentId: string | null): Promise<void>;
-    cancelRun(): void;
-    subscribe(callback: () => void): () => void;
-    onAdapterUpdated(): void;
-    private updateData;
-    unstable_synchronizer: () => null;
-}
-
-declare const useVercelRSCRuntime: <T extends WeakKey>(adapter: VercelRSCAdapter<T>) => VercelRSCRuntime<T>;
-
-declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
-
-type VercelHelpers = UseChatHelpers | UseAssistantHelpers;
-
-declare class VercelAIRuntime implements AssistantRuntime, ReactThreadRuntime {
-    vercel: VercelHelpers;
-    private _subscriptions;
-    private repository;
-    private assistantOptimisticId;
-    private useVercel;
-    messages: ThreadMessage[];
-    isRunning: boolean;
-    constructor(vercel: VercelHelpers);
-    getBranches(messageId: string): string[];
-    switchToBranch(branchId: string): void;
-    append(message: AppendMessage): Promise<void>;
-    startRun(parentId: string | null): Promise<void>;
-    cancelRun(): void;
-    subscribe(callback: () => void): () => void;
-    private updateVercelMessages;
-    onVercelUpdated(): void;
-    private updateData;
-    unstable_synchronizer: () => null;
-}
-
-declare const useVercelUseChatRuntime: (chatHelpers: UseChatHelpers) => VercelAIRuntime;
-
-declare const useVercelUseAssistantRuntime: (assistantHelpers: UseAssistantHelpers) => VercelAIRuntime;
-
-declare const getVercelAIMessage: (message: ThreadMessage) => Message[] | undefined;
-
-type ContentPartState = Readonly<{
-    status: "in_progress" | "done" | "error";
-    part: ThreadMessage["content"][number];
-}>;
-
-type MessageState = Readonly<{
-    message: Readonly<ThreadMessage>;
-    parentId: string | null;
-    branches: readonly string[];
-    isLast: boolean;
-    inProgressIndicator: ReactNode | null;
-    setInProgressIndicator: (value: ReactNode | null) => void;
-    isCopied: boolean;
-    setIsCopied: (value: boolean) => void;
-    isHovering: boolean;
-    setIsHovering: (value: boolean) => void;
-}>;
-
-type BaseComposerState = Readonly<{
-    value: string;
-    setValue: (value: string) => void;
-}>;
-
-type EditComposerState = BaseComposerState & Readonly<{
-    isEditing: boolean;
-    edit: () => void;
-    send: () => void;
-    cancel: () => boolean;
-}>;
-
-type ThreadState = {
-    messages: readonly ThreadMessage[];
-    isRunning: boolean;
-    getBranches: (messageId: string) => readonly string[];
-    switchToBranch: (branchId: string) => void;
-    append: (message: AppendMessage) => void;
-    startRun: (parentId: string | null) => void;
-    cancelRun: () => void;
-};
-
-type ComposerState = BaseComposerState & Readonly<{
-    isEditing: true;
-    send: () => void;
-    cancel: () => boolean;
-}>;
-
-type ThreadViewportState = {
-    isAtBottom: boolean;
-    scrollToBottom: () => void;
-    onScrollToBottom: (callback: () => void) => () => void;
-};
-
-type ThreadContextValue = {
-    useThread: UseBoundStore<StoreApi<ThreadState>>;
-    useComposer: UseBoundStore<StoreApi<ComposerState>>;
-    useViewport: UseBoundStore<StoreApi<ThreadViewportState>>;
-};
-declare const useThreadContext: () => ThreadContextValue;
-
-type ComposerContextValue = {
-    useComposer: UseBoundStore<StoreApi<EditComposerState | ComposerState>>;
-    type: "edit" | "new";
-};
-declare const useComposerContext: () => ComposerContextValue;
-
-type MessageContextValue = {
-    useMessage: UseBoundStore<StoreApi<MessageState>>;
-    useComposer: UseBoundStore<StoreApi<EditComposerState>>;
-};
-declare const useMessageContext: () => MessageContextValue;
-
-type ContentPartContextValue = {
-    useContentPart: UseBoundStore<StoreApi<ContentPartState>>;
-};
-declare const useContentPartContext: () => ContentPartContextValue;
-
-type Unsubscribe = () => void;
 type ThreadRuntime = Readonly<ThreadState & {
     subscribe: (callback: () => void) => Unsubscribe;
 }>;
 
-type AssistantRuntime = ThreadRuntime;
+type AssistantRuntime = ThreadRuntime & {
+    registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
+};
 
 type ReactThreadRuntime = ThreadRuntime & {
     unstable_synchronizer?: ComponentType;
 };
 
-declare class MessageRepository {
-    private messages;
-    private head;
-    private root;
-    private performOp;
-    getMessages(): ThreadMessage[];
-    addOrUpdateMessage(parentId: string | null, message: ThreadMessage): void;
-    appendOptimisticMessage(parentId: string | null, message: Omit<ThreadMessage, "id" | "createdAt">): string;
-    deleteMessage(messageId: string, replacementId?: string | null | undefined): void;
-    getBranches(messageId: string): string[];
-    switchToBranch(messageId: string): void;
-    resetHead(messageId: string | null): void;
-}
-
-/**
- * @deprecated Will be removed in 0.1.0.
- */
-type VercelAIAssistantProviderProps = PropsWithChildren<{
-    chat: UseChatHelpers;
-} | {
-    assistant: UseAssistantHelpers;
-}>;
-/**
- * @deprecated `const runtime = useVercelUseChatRuntime(chat)` and `<AssistantRuntimeProvider runtime={...} />`. Will be removed in 0.1.0.
- */
-declare const VercelAIAssistantProvider: FC<VercelAIAssistantProviderProps>;
-
-/**
- * @deprecated Will be removed in 0.1.0.
- */
-type VercelRSCAssistantProviderProps<T> = PropsWithChildren<VercelRSCAdapter<T>>;
-/**
- * @deprecated Replaced with `const runtime = useVercelRSCRuntime({ messages, append })` and `<AssistantRuntimeProvider runtime={runtime} />`. Will be removed in 0.1.0.
- */
-declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, ...adapter }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
-
 type ChatModelRunResult = {
     content: AssistantContentPart[];
 };
 type ChatModelRunOptions = {
     messages: ThreadMessage[];
     abortSignal: AbortSignal;
+    config: ModelConfig;
     onUpdate: (result: ChatModelRunResult) => void;
 };
 type ChatModelAdapter = {
@@ -484,6 +263,7 @@ type ChatModelAdapter = {
 declare class LocalRuntime implements AssistantRuntime {
     adapter: ChatModelAdapter;
     private _subscriptions;
+    private _configProviders;
     private abortController;
     private repository;
     get messages(): ThreadMessage[];
@@ -496,6 +276,7 @@ declare class LocalRuntime implements AssistantRuntime {
     cancelRun(): void;
     private notifySubscribers;
     subscribe(callback: () => void): Unsubscribe;
+    registerModelConfigProvider(provider: ModelConfigProvider): () => boolean;
 }
 
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
@@ -505,4 +286,4 @@ type AssistantRuntimeProviderProps = {
 };
 declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChildren<AssistantRuntimeProviderProps>>;
 
-export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, type AssistantContentPart, type AssistantMessage, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type ReactThreadRuntime, type TextContentPart, type ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, type UserContentPart, type UserMessage, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, type VercelRSCAdapter, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, type VercelRSCMessage, getVercelAIMessage, getVercelRSCMessage, type ChatModelAdapter as unstable_ChatModelAdapter, type ChatModelRunOptions as unstable_ChatModelRunOptions, type ComposerContextValue as unstable_ComposerContextValue, type ComposerState as unstable_ComposerState, type ContentPartContextValue as unstable_ContentPartContextValue, type ContentPartState as unstable_ContentPartState, type EditComposerState as unstable_EditComposerState, type ImageContentPart as unstable_ImageContentPart, type MessageContextValue as unstable_MessageContextValue, MessageRepository as unstable_MessageRepository, type MessageState as unstable_MessageState, type ThreadContextValue as unstable_ThreadContextValue, type ThreadState as unstable_ThreadState, type ThreadViewportState as unstable_ThreadViewportState, type ToolCallContentPart as unstable_ToolCallContentPart, type UIContentPart as unstable_UIContentPart, useComposerContext as unstable_useComposerContext, useContentPartContext as unstable_useContentPartContext, useLocalRuntime as unstable_useLocalRuntime, useMessageContext as unstable_useMessageContext, useThreadContext as unstable_useThreadContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage, useVercelRSCRuntime, useVercelUseAssistantRuntime, useVercelUseChatRuntime };
+export { index$1 as ActionBarPrimitive, AppendMessage, AssistantContentPart, type AssistantRuntime, AssistantRuntimeProvider, index$2 as BranchPickerPrimitive, type ChatModelAdapter, type ChatModelRunOptions, index$4 as ComposerPrimitive, index as ContentPartPrimitive, index$3 as MessagePrimitive, type ReactThreadRuntime, TextContentPart, ThreadMessage, index$5 as ThreadPrimitive, type ThreadRuntime, Unsubscribe, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useLocalRuntime, useReloadMessage };