@assistant-ui/react 0.0.25 → 0.0.26

package/dist/index.mjs

@@ -1787,15 +1787,17 @@ var LocalRuntime = class {
     this.abortController = new AbortController();
     this.notifySubscribers();
     try {
-      await this.adapter.run({
+      const updateHandler = ({ content }) => {
+        message.content = content;
+        this.repository.addOrUpdateMessage(parentId, { ...message });
+        this.notifySubscribers();
+      };
+      const result = await this.adapter.run({
         messages,
         abortSignal: this.abortController.signal,
-        onUpdate: ({ content }) => {
-          message.content = content;
-          this.repository.addOrUpdateMessage(parentId, { ...message });
-          this.notifySubscribers();
-        }
+        onUpdate: updateHandler
       });
+      updateHandler(result);
       message.status = "done";
       this.repository.addOrUpdateMessage(parentId, { ...message });
     } catch (e) {