@assistant-ui/react 0.0.19 → 0.0.20

package/dist/index.d.mts

@@ -153,6 +153,15 @@ type AppendMessage = {
     content: AppendContentPart[];
 };
 type ThreadMessage = UserMessage | AssistantMessage;
+type ThreadState = {
+    messages: ThreadMessage[];
+    isRunning: boolean;
+    getBranches: (messageId: string) => readonly string[];
+    switchToBranch: (branchId: string) => void;
+    append: (message: AppendMessage) => void;
+    startRun: (parentId: string | null) => void;
+    cancelRun: () => void;
+};
 
 type MessageProviderProps = PropsWithChildren<{
     message: ThreadMessage;
@@ -293,43 +302,34 @@ type VercelAIAssistantProviderProps = PropsWithChildren<{
 }>;
 declare const VercelAIAssistantProvider: FC<VercelAIAssistantProviderProps>;
 
+type Unsubscribe = () => void;
+type AssistantRuntime = Readonly<ThreadState> & {
+    subscribe: (callback: () => void) => Unsubscribe;
+};
+
 type VercelRSCMessage = {
     id: string;
     role: "user" | "assistant";
     display: ReactNode;
     createdAt?: Date;
 };
-type VercelRSCAssistantProviderBaseProps<T> = PropsWithChildren<{
+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 RSCMessageConverter<T> = {
-    convertMessage: (message: T) => VercelRSCMessage;
 };
-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 VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 
-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 VercelRSCAssistantProviderProps<T> = PropsWithChildren<VercelRSCAdapter<T>>;
+declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, ...adapter }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
 
 type AssistantProviderProps = {
-    runtime: ThreadRuntime;
+    runtime: AssistantRuntime;
 };
 declare const AssistantProvider: FC<PropsWithChildren<AssistantProviderProps>>;
 
@@ -345,25 +345,21 @@ type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
 };
 
-declare class LocalRuntime implements ThreadRuntime {
+declare class LocalRuntime implements AssistantRuntime {
     adapter: ChatModelAdapter;
-    private _messageUpdateCallbacks;
-    private _statusUpdateCallbacks;
+    private _subscriptions;
     private abortController;
     private repository;
+    get messages(): ThreadMessage[];
+    get isRunning(): boolean;
     constructor(adapter: ChatModelAdapter);
-    append(message: AppendMessage): Promise<{
-        parentId: string;
-        id: string;
-    }>;
-    startRun(parentId: string | null): Promise<{
-        id: string;
-    }>;
-    private addOrUpdateMessage;
-    private run;
+    getBranches(messageId: string): string[];
+    switchToBranch(branchId: string): void;
+    append(message: AppendMessage): Promise<void>;
+    startRun(parentId: string | null): Promise<void>;
     cancelRun(): void;
-    subscribeToMessageUpdates(callback: MessageUpdateCallback): Unsubscribe;
-    subscribeToStatusUpdates(callback: StatusUpdateCallback): Unsubscribe;
+    private notifySubscribers;
+    subscribe(callback: () => void): Unsubscribe;
 }
 
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
@@ -410,4 +406,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, 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 };
+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, type VercelRSCMessage, 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

@@ -153,6 +153,15 @@ type AppendMessage = {
     content: AppendContentPart[];
 };
 type ThreadMessage = UserMessage | AssistantMessage;
+type ThreadState = {
+    messages: ThreadMessage[];
+    isRunning: boolean;
+    getBranches: (messageId: string) => readonly string[];
+    switchToBranch: (branchId: string) => void;
+    append: (message: AppendMessage) => void;
+    startRun: (parentId: string | null) => void;
+    cancelRun: () => void;
+};
 
 type MessageProviderProps = PropsWithChildren<{
     message: ThreadMessage;
@@ -293,43 +302,34 @@ type VercelAIAssistantProviderProps = PropsWithChildren<{
 }>;
 declare const VercelAIAssistantProvider: FC<VercelAIAssistantProviderProps>;
 
+type Unsubscribe = () => void;
+type AssistantRuntime = Readonly<ThreadState> & {
+    subscribe: (callback: () => void) => Unsubscribe;
+};
+
 type VercelRSCMessage = {
     id: string;
     role: "user" | "assistant";
     display: ReactNode;
     createdAt?: Date;
 };
-type VercelRSCAssistantProviderBaseProps<T> = PropsWithChildren<{
+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 RSCMessageConverter<T> = {
-    convertMessage: (message: T) => VercelRSCMessage;
 };
-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 VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 
-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 VercelRSCAssistantProviderProps<T> = PropsWithChildren<VercelRSCAdapter<T>>;
+declare const VercelRSCAssistantProvider: <T extends WeakKey = VercelRSCMessage>({ children, ...adapter }: VercelRSCAssistantProviderProps<T>) => react_jsx_runtime.JSX.Element;
 
 type AssistantProviderProps = {
-    runtime: ThreadRuntime;
+    runtime: AssistantRuntime;
 };
 declare const AssistantProvider: FC<PropsWithChildren<AssistantProviderProps>>;
 
@@ -345,25 +345,21 @@ type ChatModelAdapter = {
     run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
 };
 
-declare class LocalRuntime implements ThreadRuntime {
+declare class LocalRuntime implements AssistantRuntime {
     adapter: ChatModelAdapter;
-    private _messageUpdateCallbacks;
-    private _statusUpdateCallbacks;
+    private _subscriptions;
     private abortController;
     private repository;
+    get messages(): ThreadMessage[];
+    get isRunning(): boolean;
     constructor(adapter: ChatModelAdapter);
-    append(message: AppendMessage): Promise<{
-        parentId: string;
-        id: string;
-    }>;
-    startRun(parentId: string | null): Promise<{
-        id: string;
-    }>;
-    private addOrUpdateMessage;
-    private run;
+    getBranches(messageId: string): string[];
+    switchToBranch(branchId: string): void;
+    append(message: AppendMessage): Promise<void>;
+    startRun(parentId: string | null): Promise<void>;
     cancelRun(): void;
-    subscribeToMessageUpdates(callback: MessageUpdateCallback): Unsubscribe;
-    subscribeToStatusUpdates(callback: StatusUpdateCallback): Unsubscribe;
+    private notifySubscribers;
+    subscribe(callback: () => void): Unsubscribe;
 }
 
 declare const useLocalRuntime: (adapter: ChatModelAdapter) => LocalRuntime;
@@ -410,4 +406,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, 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 };
+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, type VercelRSCMessage, 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 };