@assistant-ui/react 0.5.22 → 0.5.24

package/dist/index.d.mts

@@ -77,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;
@@ -109,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> {
@@ -439,24 +468,6 @@ declare class ProxyConfigProvider implements ModelConfigProvider {
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
 }
 
-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;
-}
-
 type TextContentPartState = Readonly<{
     status: ContentPartStatus;
     part: TextContentPart;
@@ -521,12 +532,16 @@ 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 };
 }
 
+type ConverterCallback<TIn> = (cache: ThreadMessage | undefined, message: TIn, idx: number) => ThreadMessage;
+declare class ThreadMessageConverter {
+    private readonly cache;
+    convertMessages<TIn extends WeakKey>(messages: TIn[], converter: ConverterCallback<TIn>, keyMapper?: (m: TIn) => WeakKey): ThreadMessage[];
+}
+
 declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
-    store: ExternalStoreAdapter<any>;
     private _subscriptions;
     private repository;
     private assistantOptimisticId;
-    private useStore;
     get capabilities(): {
         switchToBranch: boolean;
         edit: boolean;
@@ -537,7 +552,10 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     messages: ThreadMessage[];
     isDisabled: boolean;
     isRunning: boolean;
+    converter: ThreadMessageConverter;
+    private _store;
     constructor(store: ExternalStoreAdapter<any>);
+    set store(store: ExternalStoreAdapter<any>);
     getBranches(messageId: string): string[];
     switchToBranch(branchId: string): void;
     append(message: AppendMessage): Promise<void>;
@@ -545,9 +563,6 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     cancelRun(): void;
     subscribe(callback: () => void): Unsubscribe;
     private updateMessages;
-    onStoreUpdated(): void;
-    private updateData;
-    unstable_synchronizer: () => null;
     addToolResult(options: AddToolResultOptions): void;
 }
 
@@ -555,13 +570,12 @@ declare class ExternalStoreRuntime extends BaseAssistantRuntime<ExternalStoreThr
     private readonly _proxyConfigProvider;
     constructor(store: ExternalStoreAdapter<any>);
     set store(store: ExternalStoreAdapter<any>);
-    onStoreUpdated(): void;
     getModelConfig(): ModelConfig;
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
     switchToThread(threadId: string | null): void | Promise<void>;
 }
 
-declare const useExternalStoreRuntime: (store: ExternalStoreAdapter<any>) => ExternalStoreRuntime;
+declare const useExternalStoreRuntime: <T>(store: ExternalStoreAdapter<T>) => ExternalStoreRuntime;
 
 declare const getExternalStoreMessage: <T>(message: ThreadMessage) => T | undefined;
 

package/dist/index.d.ts

@@ -77,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;
@@ -109,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> {
@@ -439,24 +468,6 @@ declare class ProxyConfigProvider implements ModelConfigProvider {
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
 }
 
-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;
-}
-
 type TextContentPartState = Readonly<{
     status: ContentPartStatus;
     part: TextContentPart;
@@ -521,12 +532,16 @@ 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 };
 }
 
+type ConverterCallback<TIn> = (cache: ThreadMessage | undefined, message: TIn, idx: number) => ThreadMessage;
+declare class ThreadMessageConverter {
+    private readonly cache;
+    convertMessages<TIn extends WeakKey>(messages: TIn[], converter: ConverterCallback<TIn>, keyMapper?: (m: TIn) => WeakKey): ThreadMessage[];
+}
+
 declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
-    store: ExternalStoreAdapter<any>;
     private _subscriptions;
     private repository;
     private assistantOptimisticId;
-    private useStore;
     get capabilities(): {
         switchToBranch: boolean;
         edit: boolean;
@@ -537,7 +552,10 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     messages: ThreadMessage[];
     isDisabled: boolean;
     isRunning: boolean;
+    converter: ThreadMessageConverter;
+    private _store;
     constructor(store: ExternalStoreAdapter<any>);
+    set store(store: ExternalStoreAdapter<any>);
     getBranches(messageId: string): string[];
     switchToBranch(branchId: string): void;
     append(message: AppendMessage): Promise<void>;
@@ -545,9 +563,6 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     cancelRun(): void;
     subscribe(callback: () => void): Unsubscribe;
     private updateMessages;
-    onStoreUpdated(): void;
-    private updateData;
-    unstable_synchronizer: () => null;
     addToolResult(options: AddToolResultOptions): void;
 }
 
@@ -555,13 +570,12 @@ declare class ExternalStoreRuntime extends BaseAssistantRuntime<ExternalStoreThr
     private readonly _proxyConfigProvider;
     constructor(store: ExternalStoreAdapter<any>);
     set store(store: ExternalStoreAdapter<any>);
-    onStoreUpdated(): void;
     getModelConfig(): ModelConfig;
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
     switchToThread(threadId: string | null): void | Promise<void>;
 }
 
-declare const useExternalStoreRuntime: (store: ExternalStoreAdapter<any>) => ExternalStoreRuntime;
+declare const useExternalStoreRuntime: <T>(store: ExternalStoreAdapter<T>) => ExternalStoreRuntime;
 
 declare const getExternalStoreMessage: <T>(message: ThreadMessage) => T | undefined;