@assistant-ui/react 0.0.17 → 0.0.19

package/dist/index.d.mts

@@ -3,7 +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 { Message } from 'ai';
+import { LanguageModel, Message } from 'ai';
 import { UseBoundStore, StoreApi } from 'zustand';
 
 declare const ThreadRoot: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
@@ -103,16 +103,16 @@ declare namespace index$4 {
   export { ComposerCancel as Cancel, ComposerIf as If, ComposerInput as Input, ComposerRoot as Root, ComposerSend as Send };
 }
 
-type BaseComposerState = {
+type BaseComposerState = Readonly<{
     value: string;
     setValue: (value: string) => void;
-};
-type MessageComposerState = BaseComposerState & {
+}>;
+type MessageComposerState = BaseComposerState & Readonly<{
     isEditing: boolean;
     edit: () => void;
     send: () => void;
     cancel: () => boolean;
-};
+}>;
 
 type TextContentPart = {
     type: "text";
@@ -133,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;
@@ -312,13 +312,77 @@ 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;
+type MessageState = Readonly<{
+    message: Readonly<ThreadMessage>;
     parentId: string | null;
-    branches: string[];
+    branches: readonly string[];
     isLast: boolean;
     inProgressIndicator: ReactNode | null;
     setInProgressIndicator: (value: ReactNode | null) => void;
@@ -326,7 +390,7 @@ type MessageState = {
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
     setIsHovering: (value: boolean) => void;
-};
+}>;
 type MessageStore = {
     useMessage: UseBoundStore<StoreApi<MessageState>>;
     useComposer: UseBoundStore<StoreApi<MessageComposerState>>;
@@ -346,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, getVercelMessage as unstable_getVercelMessage, getVercelRSCMessage as unstable_getVercelRSCMessage, 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,7 +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 { Message } from 'ai';
+import { LanguageModel, Message } from 'ai';
 import { UseBoundStore, StoreApi } from 'zustand';
 
 declare const ThreadRoot: react.ForwardRefExoticComponent<Pick<Omit<react.DetailedHTMLProps<react.HTMLAttributes<HTMLDivElement>, HTMLDivElement>, "ref"> & {
@@ -103,16 +103,16 @@ declare namespace index$4 {
   export { ComposerCancel as Cancel, ComposerIf as If, ComposerInput as Input, ComposerRoot as Root, ComposerSend as Send };
 }
 
-type BaseComposerState = {
+type BaseComposerState = Readonly<{
     value: string;
     setValue: (value: string) => void;
-};
-type MessageComposerState = BaseComposerState & {
+}>;
+type MessageComposerState = BaseComposerState & Readonly<{
     isEditing: boolean;
     edit: () => void;
     send: () => void;
     cancel: () => boolean;
-};
+}>;
 
 type TextContentPart = {
     type: "text";
@@ -133,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;
@@ -312,13 +312,77 @@ 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;
+type MessageState = Readonly<{
+    message: Readonly<ThreadMessage>;
     parentId: string | null;
-    branches: string[];
+    branches: readonly string[];
     isLast: boolean;
     inProgressIndicator: ReactNode | null;
     setInProgressIndicator: (value: ReactNode | null) => void;
@@ -326,7 +390,7 @@ type MessageState = {
     setIsCopied: (value: boolean) => void;
     isHovering: boolean;
     setIsHovering: (value: boolean) => void;
-};
+}>;
 type MessageStore = {
     useMessage: UseBoundStore<StoreApi<MessageState>>;
     useComposer: UseBoundStore<StoreApi<MessageComposerState>>;
@@ -346,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, getVercelMessage as unstable_getVercelMessage, getVercelRSCMessage as unstable_getVercelRSCMessage, 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 };