@assistant-ui/react 0.0.16 → 0.0.18

package/dist/index.d.mts

@@ -3,6 +3,7 @@ import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { UseChatHelpers, UseAssistantHelpers } from 'ai/react';
 import * as react_jsx_runtime from 'react/jsx-runtime';
+import { LanguageModel, Message } from 'ai';
 import { UseBoundStore, StoreApi } from 'zustand';
 
 declare const ThreadRoot: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
@@ -132,7 +133,7 @@ type ToolCallContentPart = {
     result?: object;
 };
 type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
-type AssistantContentPart = TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
+type AssistantContentPart = TextContentPart | UIContentPart | ToolCallContentPart;
 type AppendContentPart = TextContentPart | ImageContentPart;
 type BaseMessage = {
     id: string;
@@ -145,6 +146,7 @@ type UserMessage = BaseMessage & {
 type AssistantMessage = BaseMessage & {
     role: "assistant";
     content: AssistantContentPart[];
+    status: "in_progress" | "done" | "error";
 };
 type AppendMessage = {
     parentId: string | null;
@@ -199,14 +201,14 @@ type MessageContentProps = {
 };
 declare const MessageContent: FC<MessageContentProps>;
 
-declare const MessageLoading: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+declare const MessageInProgress: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
     ref?: ((instance: HTMLDivElement | null) => void) | react.RefObject<HTMLDivElement> | null | undefined;
 } & {
     asChild?: boolean;
 }, "key" | keyof react.HTMLAttributes<HTMLDivElement> | "asChild"> & react.RefAttributes<HTMLDivElement>>;
 
 declare namespace index$3 {
-  export { MessageContent as Content, MessageIf as If, MessageLoading as Loading, MessageProvider as Provider, MessageRoot as Root };
+  export { MessageContent as Content, MessageIf as If, MessageInProgress as InProgress, MessageProvider as Provider, MessageRoot as Root };
 }
 
 declare const BranchPickerNext: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -274,14 +276,14 @@ declare namespace index$1 {
 
 type ContentPartProviderProps = PropsWithChildren<{
     part: ThreadMessage["content"][number];
-    isLoading: boolean;
+    status: "in_progress" | "done" | "error";
 }>;
 declare const ContentPartProvider: FC<ContentPartProviderProps>;
 
-declare const ContentPartLoadingIndicator: FC;
+declare const ContentPartInProgressIndicator: FC;
 
 declare namespace index {
-  export { ContentPartLoadingIndicator as LoadingIndicator, ContentPartProvider as Provider };
+  export { ContentPartInProgressIndicator as InProgressIndicator, ContentPartProvider as Provider };
 }
 
 type VercelAIAssistantProviderProps = PropsWithChildren<{
@@ -310,13 +312,80 @@ type RSCMessageConverter<T> = {
 type VercelRSCAssistantProviderProps<T = VercelRSCMessage> = VercelRSCAssistantProviderBaseProps<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, convertMessage, messages: vercelMessages, append: appendCallback, edit, reload, }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
 
+type MessageUpdateCallback = (parentId: string | null, message: ThreadMessage) => void;
+type StatusUpdateCallback = (isRunning: boolean) => void;
+type Unsubscribe = () => void;
+type ThreadRuntime = {
+    append(message: AppendMessage): Promise<{
+        parentId: string;
+        id: string;
+    }>;
+    startRun(parentId: string | null): Promise<{
+        id: string;
+    }>;
+    cancelRun(): void;
+    subscribeToMessageUpdates(callback: MessageUpdateCallback): Unsubscribe;
+    subscribeToStatusUpdates(callback: StatusUpdateCallback): Unsubscribe;
+};
+
+type AssistantProviderProps = {
+    runtime: ThreadRuntime;
+};
+declare const AssistantProvider: FC<PropsWithChildren<AssistantProviderProps>>;
+
+type ChatModelRunResult = {
+    content: AssistantContentPart[];
+};
+type ChatModelRunOptions = {
+    messages: ThreadMessage[];
+    abortSignal: AbortSignal;
+    onUpdate: (result: ChatModelRunResult) => void;
+};
+type ChatModelAdapter = {
+    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
+};
+
+declare class LocalRuntime implements ThreadRuntime {
+    adapter: ChatModelAdapter;
+    private _messageUpdateCallbacks;
+    private _statusUpdateCallbacks;
+    private abortController;
+    private repository;
+    constructor(adapter: ChatModelAdapter);
+    append(message: AppendMessage): Promise<{
+        parentId: string;
+        id: string;
+    }>;
+    startRun(parentId: string | null): Promise<{
+        id: string;
+    }>;
+    private addOrUpdateMessage;
+    private run;
+    cancelRun(): void;
+    subscribeToMessageUpdates(callback: MessageUpdateCallback): Unsubscribe;
+    subscribeToStatusUpdates(callback: StatusUpdateCallback): Unsubscribe;
+}
+
+declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
+
+declare class VercelModelAdapter implements ChatModelAdapter {
+    private readonly model;
+    constructor(model: LanguageModel);
+    run({ messages, abortSignal, onUpdate }: ChatModelRunOptions): Promise<{
+        content: AssistantContentPart[];
+    }>;
+}
+
+declare const getVercelMessage: (message: ThreadMessage) => Message | undefined;
+declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
+
 type MessageState = {
     message: ThreadMessage;
     parentId: string | null;
     branches: string[];
     isLast: boolean;
-    loadingIndicator: ReactNode | null;
-    setLoadingIndicator: (value: ReactNode | null) => void;
+    inProgressIndicator: ReactNode | null;
+    setInProgressIndicator: (value: ReactNode | null) => void;
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
@@ -341,4 +410,4 @@ declare const useGoToNextBranch: () => (() => void) | null;
 
 declare const useGoToPreviousBranch: () => (() => void) | null;
 
-export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, type ImageContentPart, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$5 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };
+export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, type ImageContentPart, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$5 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, AssistantProvider as unstable_AssistantProvider, type ChatModelAdapter as unstable_ChatModelAdapter, type ChatModelRunOptions as unstable_ChatModelRunOptions, VercelModelAdapter as unstable_VercelModelAdapter, getVercelMessage as unstable_getVercelMessage, getVercelRSCMessage as unstable_getVercelRSCMessage, useLocalRuntime as unstable_useLocalRuntime, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };

package/dist/index.d.ts

@@ -3,6 +3,7 @@ import { FC, ReactNode, PropsWithChildren, ComponentType } from 'react';
 import { TextareaAutosizeProps } from 'react-textarea-autosize';
 import { UseChatHelpers, UseAssistantHelpers } from 'ai/react';
 import * as react_jsx_runtime from 'react/jsx-runtime';
+import { LanguageModel, Message } from 'ai';
 import { UseBoundStore, StoreApi } from 'zustand';
 
 declare const ThreadRoot: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
@@ -132,7 +133,7 @@ type ToolCallContentPart = {
     result?: object;
 };
 type UserContentPart = TextContentPart | ImageContentPart | UIContentPart;
-type AssistantContentPart = TextContentPart | ImageContentPart | UIContentPart | ToolCallContentPart;
+type AssistantContentPart = TextContentPart | UIContentPart | ToolCallContentPart;
 type AppendContentPart = TextContentPart | ImageContentPart;
 type BaseMessage = {
     id: string;
@@ -145,6 +146,7 @@ type UserMessage = BaseMessage & {
 type AssistantMessage = BaseMessage & {
     role: "assistant";
     content: AssistantContentPart[];
+    status: "in_progress" | "done" | "error";
 };
 type AppendMessage = {
     parentId: string | null;
@@ -199,14 +201,14 @@ type MessageContentProps = {
 };
 declare const MessageContent: FC<MessageContentProps>;
 
-declare const MessageLoading: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
+declare const MessageInProgress: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
     ref?: ((instance: HTMLDivElement | null) => void) | react.RefObject<HTMLDivElement> | null | undefined;
 } & {
     asChild?: boolean;
 }, "key" | keyof react.HTMLAttributes<HTMLDivElement> | "asChild"> & react.RefAttributes<HTMLDivElement>>;
 
 declare namespace index$3 {
-  export { MessageContent as Content, MessageIf as If, MessageLoading as Loading, MessageProvider as Provider, MessageRoot as Root };
+  export { MessageContent as Content, MessageIf as If, MessageInProgress as InProgress, MessageProvider as Provider, MessageRoot as Root };
 }
 
 declare const BranchPickerNext: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.ButtonHTMLAttributes<HTMLButtonElement>, HTMLButtonElement>, "ref"> & {
@@ -274,14 +276,14 @@ declare namespace index$1 {
 
 type ContentPartProviderProps = PropsWithChildren<{
     part: ThreadMessage["content"][number];
-    isLoading: boolean;
+    status: "in_progress" | "done" | "error";
 }>;
 declare const ContentPartProvider: FC<ContentPartProviderProps>;
 
-declare const ContentPartLoadingIndicator: FC;
+declare const ContentPartInProgressIndicator: FC;
 
 declare namespace index {
-  export { ContentPartLoadingIndicator as LoadingIndicator, ContentPartProvider as Provider };
+  export { ContentPartInProgressIndicator as InProgressIndicator, ContentPartProvider as Provider };
 }
 
 type VercelAIAssistantProviderProps = PropsWithChildren<{
@@ -310,13 +312,80 @@ type RSCMessageConverter<T> = {
 type VercelRSCAssistantProviderProps<T = VercelRSCMessage> = VercelRSCAssistantProviderBaseProps<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, convertMessage, messages: vercelMessages, append: appendCallback, edit, reload, }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
 
+type MessageUpdateCallback = (parentId: string | null, message: ThreadMessage) => void;
+type StatusUpdateCallback = (isRunning: boolean) => void;
+type Unsubscribe = () => void;
+type ThreadRuntime = {
+    append(message: AppendMessage): Promise<{
+        parentId: string;
+        id: string;
+    }>;
+    startRun(parentId: string | null): Promise<{
+        id: string;
+    }>;
+    cancelRun(): void;
+    subscribeToMessageUpdates(callback: MessageUpdateCallback): Unsubscribe;
+    subscribeToStatusUpdates(callback: StatusUpdateCallback): Unsubscribe;
+};
+
+type AssistantProviderProps = {
+    runtime: ThreadRuntime;
+};
+declare const AssistantProvider: FC<PropsWithChildren<AssistantProviderProps>>;
+
+type ChatModelRunResult = {
+    content: AssistantContentPart[];
+};
+type ChatModelRunOptions = {
+    messages: ThreadMessage[];
+    abortSignal: AbortSignal;
+    onUpdate: (result: ChatModelRunResult) => void;
+};
+type ChatModelAdapter = {
+    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
+};
+
+declare class LocalRuntime implements ThreadRuntime {
+    adapter: ChatModelAdapter;
+    private _messageUpdateCallbacks;
+    private _statusUpdateCallbacks;
+    private abortController;
+    private repository;
+    constructor(adapter: ChatModelAdapter);
+    append(message: AppendMessage): Promise<{
+        parentId: string;
+        id: string;
+    }>;
+    startRun(parentId: string | null): Promise<{
+        id: string;
+    }>;
+    private addOrUpdateMessage;
+    private run;
+    cancelRun(): void;
+    subscribeToMessageUpdates(callback: MessageUpdateCallback): Unsubscribe;
+    subscribeToStatusUpdates(callback: StatusUpdateCallback): Unsubscribe;
+}
+
+declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
+
+declare class VercelModelAdapter implements ChatModelAdapter {
+    private readonly model;
+    constructor(model: LanguageModel);
+    run({ messages, abortSignal, onUpdate }: ChatModelRunOptions): Promise<{
+        content: AssistantContentPart[];
+    }>;
+}
+
+declare const getVercelMessage: (message: ThreadMessage) => Message | undefined;
+declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
+
 type MessageState = {
     message: ThreadMessage;
     parentId: string | null;
     branches: string[];
     isLast: boolean;
-    loadingIndicator: ReactNode | null;
-    setLoadingIndicator: (value: ReactNode | null) => void;
+    inProgressIndicator: ReactNode | null;
+    setInProgressIndicator: (value: ReactNode | null) => void;
     isCopied: boolean;
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
@@ -341,4 +410,4 @@ declare const useGoToNextBranch: () => (() => void) | null;
 
 declare const useGoToPreviousBranch: () => (() => void) | null;
 
-export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, type ImageContentPart, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$5 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };
+export { index$1 as ActionBarPrimitive, type AppendContentPart, type AppendMessage, index$2 as BranchPickerPrimitive, index$4 as ComposerPrimitive, index as ContentPartPrimitive, type ImageContentPart, index$3 as MessagePrimitive, type VercelRSCMessage as RSCMessage, type TextContentPart, index$5 as ThreadPrimitive, VercelAIAssistantProvider, type VercelAIAssistantProviderProps, VercelRSCAssistantProvider, type VercelRSCAssistantProviderProps, AssistantProvider as unstable_AssistantProvider, type ChatModelAdapter as unstable_ChatModelAdapter, type ChatModelRunOptions as unstable_ChatModelRunOptions, VercelModelAdapter as unstable_VercelModelAdapter, getVercelMessage as unstable_getVercelMessage, getVercelRSCMessage as unstable_getVercelRSCMessage, useLocalRuntime as unstable_useLocalRuntime, useMessageContext as unstable_useMessageContext, useBeginMessageEdit, useCopyMessage, useGoToNextBranch, useGoToPreviousBranch, useReloadMessage };