@assistant-ui/react 0.5.22 → 0.5.23

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;

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;

package/dist/index.js

@@ -2273,6 +2273,25 @@ var MessageRepository = (_class4 = class {constructor() { _class4.prototype.__in
       }
     }
   }
+  export() {
+    const exportItems = [];
+    for (const [, message] of this.messages) {
+      exportItems.push({
+        message: message.current,
+        parentId: _nullishCoalesce(_optionalChain([message, 'access', _84 => _84.prev, 'optionalAccess', _85 => _85.current, 'access', _86 => _86.id]), () => ( null))
+      });
+    }
+    return {
+      headId: _nullishCoalesce(_optionalChain([this, 'access', _87 => _87.head, 'optionalAccess', _88 => _88.current, 'access', _89 => _89.id]), () => ( null)),
+      messages: exportItems
+    };
+  }
+  import({ headId, messages }) {
+    for (const { message, parentId } of messages) {
+      this.addOrUpdateMessage(parentId, message);
+    }
+    this.resetHead(_nullishCoalesce(_nullishCoalesce(headId, () => ( _optionalChain([messages, 'access', _90 => _90.at, 'call', _91 => _91(-1), 'optionalAccess', _92 => _92.message, 'access', _93 => _93.id]))), () => ( null)));
+  }
 }, _class4);
 
 // src/ui/base/tooltip-icon-button.tsx
@@ -2434,7 +2453,7 @@ var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
         });
         if (mergeRoundtrips) {
           const previousMessage = messages[messages.length - 1];
-          if (_optionalChain([previousMessage, 'optionalAccess', _84 => _84.role]) === "assistant") {
+          if (_optionalChain([previousMessage, 'optionalAccess', _94 => _94.role]) === "assistant") {
             previousMessage.content.push(...newContent);
             break;
           }
@@ -2447,7 +2466,7 @@ var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
       }
       case "tool": {
         const previousMessage = messages[messages.length - 1];
-        if (_optionalChain([previousMessage, 'optionalAccess', _85 => _85.role]) !== "assistant")
+        if (_optionalChain([previousMessage, 'optionalAccess', _95 => _95.role]) !== "assistant")
           throw new Error(
             "A tool message must be preceded by an assistant message."
           );
@@ -2651,7 +2670,7 @@ var useEdgeRuntime = ({
 };
 
 // src/runtimes/local/shouldContinue.tsx
-var shouldContinue = (result) => _optionalChain([result, 'access', _86 => _86.status, 'optionalAccess', _87 => _87.type]) === "requires-action" && result.status.reason === "tool-calls" && result.content.every((c) => c.type !== "tool-call" || !!c.result);
+var shouldContinue = (result) => _optionalChain([result, 'access', _96 => _96.status, 'optionalAccess', _97 => _97.type]) === "requires-action" && result.status.reason === "tool-calls" && result.content.every((c) => c.type !== "tool-call" || !!c.result);
 
 // src/runtimes/local/LocalThreadRuntime.tsx
 var CAPABILITIES = Object.freeze({
@@ -2666,7 +2685,7 @@ var LocalThreadRuntime = (_class5 = class {
     this.configProvider = configProvider;
     this.adapter = adapter;
     this.options = options;
-    if (_optionalChain([options, 'optionalAccess', _88 => _88.initialMessages])) {
+    if (_optionalChain([options, 'optionalAccess', _98 => _98.initialMessages])) {
       let parentId = null;
       const messages = fromCoreMessages(options.initialMessages);
       for (const message of messages) {
@@ -2724,7 +2743,7 @@ var LocalThreadRuntime = (_class5 = class {
   }
   async performRoundtrip(parentId, message) {
     const messages = this.repository.getMessages();
-    _optionalChain([this, 'access', _89 => _89.abortController, 'optionalAccess', _90 => _90.abort, 'call', _91 => _91()]);
+    _optionalChain([this, 'access', _99 => _99.abortController, 'optionalAccess', _100 => _100.abort, 'call', _101 => _101()]);
     this.abortController = new AbortController();
     const initialContent = message.content;
     const initialRoundtrips = message.roundtrips;
@@ -2733,13 +2752,13 @@ var LocalThreadRuntime = (_class5 = class {
         ...message,
         ...m.content ? { content: [...initialContent, ..._nullishCoalesce(m.content, () => ( []))] } : void 0,
         status: _nullishCoalesce(m.status, () => ( message.status)),
-        ..._optionalChain([m, 'access', _92 => _92.roundtrips, 'optionalAccess', _93 => _93.length]) ? { roundtrips: [..._nullishCoalesce(initialRoundtrips, () => ( [])), ...m.roundtrips] } : void 0
+        ..._optionalChain([m, 'access', _102 => _102.roundtrips, 'optionalAccess', _103 => _103.length]) ? { roundtrips: [..._nullishCoalesce(initialRoundtrips, () => ( [])), ...m.roundtrips] } : void 0
       };
       this.repository.addOrUpdateMessage(parentId, message);
       this.notifySubscribers();
     };
-    const maxToolRoundtrips = _nullishCoalesce(_optionalChain([this, 'access', _94 => _94.options, 'optionalAccess', _95 => _95.maxToolRoundtrips]), () => ( 1));
-    const toolRoundtrips = _nullishCoalesce(_optionalChain([message, 'access', _96 => _96.roundtrips, 'optionalAccess', _97 => _97.length]), () => ( 0));
+    const maxToolRoundtrips = _nullishCoalesce(_optionalChain([this, 'access', _104 => _104.options, 'optionalAccess', _105 => _105.maxToolRoundtrips]), () => ( 1));
+    const toolRoundtrips = _nullishCoalesce(_optionalChain([message, 'access', _106 => _106.roundtrips, 'optionalAccess', _107 => _107.length]), () => ( 0));
     if (toolRoundtrips > maxToolRoundtrips) {
       updateMessage({
         status: {
@@ -2831,6 +2850,13 @@ var LocalThreadRuntime = (_class5 = class {
       this.performRoundtrip(parentId, message);
     }
   }
+  export() {
+    return this.repository.export();
+  }
+  import(data) {
+    this.repository.import(data);
+    this.notifySubscribers();
+  }
 }, _class5);
 
 // src/runtimes/local/LocalRuntime.tsx
@@ -3012,7 +3038,7 @@ var useExternalStoreSync = (adapter, updateData) => {
 
 // src/runtimes/external-store/ExternalStoreThreadRuntime.tsx
 var hasUpcomingMessage = (isRunning, messages) => {
-  return isRunning && _optionalChain([messages, 'access', _98 => _98[messages.length - 1], 'optionalAccess', _99 => _99.role]) !== "assistant";
+  return isRunning && _optionalChain([messages, 'access', _108 => _108[messages.length - 1], 'optionalAccess', _109 => _109.role]) !== "assistant";
 };
 var ExternalStoreThreadRuntime = (_class7 = class {
   constructor(store) {;_class7.prototype.__init17.call(this);_class7.prototype.__init18.call(this);_class7.prototype.__init19.call(this);_class7.prototype.__init20.call(this);_class7.prototype.__init21.call(this);_class7.prototype.__init22.call(this);_class7.prototype.__init23.call(this);_class7.prototype.__init24.call(this);_class7.prototype.__init25.call(this);
@@ -3052,7 +3078,7 @@ var ExternalStoreThreadRuntime = (_class7 = class {
     this.updateMessages(this.repository.getMessages());
   }
   async append(message) {
-    if (message.parentId !== (_nullishCoalesce(_optionalChain([this, 'access', _110 => _110.messages, 'access', _111 => _111.at, 'call', _112 => _112(-1), 'optionalAccess', _113 => _113.id]), () => ( null)))) {
+    if (message.parentId !== (_nullishCoalesce(_optionalChain([this, 'access', _120 => _120.messages, 'access', _121 => _121.at, 'call', _122 => _122(-1), 'optionalAccess', _123 => _123.id]), () => ( null)))) {
       if (!this.store.onEdit)
         throw new Error("Runtime does not support editing messages.");
       await this.store.onEdit(message);
@@ -3083,7 +3109,7 @@ var ExternalStoreThreadRuntime = (_class7 = class {
     return () => this._subscriptions.delete(callback);
   }
   __init23() {this.updateMessages = (messages) => {
-    _optionalChain([this, 'access', _114 => _114.store, 'access', _115 => _115.setMessages, 'optionalCall', _116 => _116(
+    _optionalChain([this, 'access', _124 => _124.store, 'access', _125 => _125.setMessages, 'optionalCall', _126 => _126(
       messages.flatMap(getExternalStoreMessage).filter((m) => m != null)
     )]);
   }}
@@ -3096,7 +3122,7 @@ var ExternalStoreThreadRuntime = (_class7 = class {
     for (let i = 0; i < vm.length; i++) {
       const message = vm[i];
       const parent = vm[i - 1];
-      this.repository.addOrUpdateMessage(_nullishCoalesce(_optionalChain([parent, 'optionalAccess', _117 => _117.id]), () => ( null)), message);
+      this.repository.addOrUpdateMessage(_nullishCoalesce(_optionalChain([parent, 'optionalAccess', _127 => _127.id]), () => ( null)), message);
     }
     if (this.assistantOptimisticId) {
       this.repository.deleteMessage(this.assistantOptimisticId);
@@ -3104,7 +3130,7 @@ var ExternalStoreThreadRuntime = (_class7 = class {
     }
     if (hasUpcomingMessage(isRunning, vm)) {
       this.assistantOptimisticId = this.repository.appendOptimisticMessage(
-        _nullishCoalesce(_optionalChain([vm, 'access', _118 => _118.at, 'call', _119 => _119(-1), 'optionalAccess', _120 => _120.id]), () => ( null)),
+        _nullishCoalesce(_optionalChain([vm, 'access', _128 => _128.at, 'call', _129 => _129(-1), 'optionalAccess', _130 => _130.id]), () => ( null)),
         {
           role: "assistant",
           content: []
@@ -3112,7 +3138,7 @@ var ExternalStoreThreadRuntime = (_class7 = class {
       );
     }
     this.repository.resetHead(
-      _nullishCoalesce(_nullishCoalesce(this.assistantOptimisticId, () => ( _optionalChain([vm, 'access', _121 => _121.at, 'call', _122 => _122(-1), 'optionalAccess', _123 => _123.id]))), () => ( null))
+      _nullishCoalesce(_nullishCoalesce(this.assistantOptimisticId, () => ( _optionalChain([vm, 'access', _131 => _131.at, 'call', _132 => _132(-1), 'optionalAccess', _133 => _133.id]))), () => ( null))
     );
     this.messages = this.repository.getMessages();
     this.isDisabled = isDisabled;
@@ -3191,7 +3217,7 @@ var ThreadConfigProvider = ({
 }) => {
   const assistant = useAssistantContext({ optional: true });
   const configProvider = config && Object.keys(_nullishCoalesce(config, () => ( {}))).length > 0 ? /* @__PURE__ */ _jsxruntime.jsx.call(void 0, ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ _jsxruntime.jsx.call(void 0, _jsxruntime.Fragment, { children });
-  if (!_optionalChain([config, 'optionalAccess', _124 => _124.runtime])) return configProvider;
+  if (!_optionalChain([config, 'optionalAccess', _134 => _134.runtime])) return configProvider;
   if (assistant) {
     throw new Error(
       "You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed."
@@ -3407,7 +3433,7 @@ var AssistantMessageContent = _react.forwardRef.call(void 0, ({ components: comp
     {
       components: {
         ...componentsProp,
-        Text: _nullishCoalesce(_nullishCoalesce(_optionalChain([componentsProp, 'optionalAccess', _125 => _125.Text]), () => ( components.Text)), () => ( content_part_default.Text))
+        Text: _nullishCoalesce(_nullishCoalesce(_optionalChain([componentsProp, 'optionalAccess', _135 => _135.Text]), () => ( components.Text)), () => ( content_part_default.Text))
       }
     }
   ) });
@@ -3582,7 +3608,7 @@ var ThreadWelcomeSuggestion = ({
 };
 var ThreadWelcomeSuggestions = () => {
   const { welcome: { suggestions } = {} } = useThreadConfig();
-  return /* @__PURE__ */ _jsxruntime.jsx.call(void 0, ThreadWelcomeSuggestionContainer, { children: _optionalChain([suggestions, 'optionalAccess', _126 => _126.map, 'call', _127 => _127((suggestion, idx) => {
+  return /* @__PURE__ */ _jsxruntime.jsx.call(void 0, ThreadWelcomeSuggestionContainer, { children: _optionalChain([suggestions, 'optionalAccess', _136 => _136.map, 'call', _137 => _137((suggestion, idx) => {
     const key = `${suggestion.prompt}-${idx}`;
     return /* @__PURE__ */ _jsxruntime.jsx.call(void 0, ThreadWelcomeSuggestion, { suggestion }, key);
   })]) });
@@ -3660,7 +3686,7 @@ var UserMessageContent = _react.forwardRef.call(void 0,
       {
         components: {
           ...components,
-          Text: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _128 => _128.Text]), () => ( content_part_default.Text))
+          Text: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _138 => _138.Text]), () => ( content_part_default.Text))
         }
       }
     ) });
@@ -3762,10 +3788,10 @@ var ThreadMessages = ({ components, ...rest }) => {
     thread_exports.Messages,
     {
       components: {
-        UserMessage: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _129 => _129.UserMessage]), () => ( user_message_default)),
-        EditComposer: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _130 => _130.EditComposer]), () => ( edit_composer_default)),
-        AssistantMessage: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _131 => _131.AssistantMessage]), () => ( assistant_message_default)),
-        SystemMessage: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _132 => _132.SystemMessage]), () => ( SystemMessage))
+        UserMessage: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _139 => _139.UserMessage]), () => ( user_message_default)),
+        EditComposer: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _140 => _140.EditComposer]), () => ( edit_composer_default)),
+        AssistantMessage: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _141 => _141.AssistantMessage]), () => ( assistant_message_default)),
+        SystemMessage: _nullishCoalesce(_optionalChain([components, 'optionalAccess', _142 => _142.SystemMessage]), () => ( SystemMessage))
       },
       ...rest
     }