bunnyquery 1.4.2 → 1.4.4

package/dist/engine.mjs

@@ -1148,13 +1148,38 @@ var ChatSession = class {
     }).catch(function(err) {
       return { content: getErrorMessage(err), isError: true };
     }).then(function(result) {
-      var existing = self.aiChatHistoryCache[params.key] || { messages: [], endOfList: false, startKeyHistory: [] };
-      self.aiChatHistoryCache[params.key] = {
-        messages: existing.messages.concat([{ role: "assistant", content: result.content, isError: result.isError }]),
-        endOfList: existing.endOfList,
-        startKeyHistory: existing.startKeyHistory
-      };
       delete self.pendingAgentRequests[params.key];
+      var existing = self.aiChatHistoryCache[params.key] || { messages: [], endOfList: false, startKeyHistory: [] };
+      var reply = { role: "assistant", content: result.content, isError: result.isError };
+      var onThisVisibleChat = self.host.isViewMounted() && self.getHistoryCacheKey() === params.key;
+      if (onThisVisibleChat) {
+        self.aiChatHistoryCache[params.key] = {
+          messages: existing.messages.concat([reply]),
+          endOfList: existing.endOfList,
+          startKeyHistory: existing.startKeyHistory
+        };
+      } else {
+        var msgs = existing.messages.slice();
+        var idx = -1;
+        for (var i = msgs.length - 1; i >= 0; i--) {
+          var m = msgs[i];
+          if (m && m.isPending && m.role === "assistant" && !m.isBackgroundTask) {
+            idx = i;
+            break;
+          }
+        }
+        if (idx !== -1) {
+          reply._serverItemId = msgs[idx]._serverItemId;
+          msgs[idx] = reply;
+        } else {
+          msgs.push(reply);
+        }
+        self.aiChatHistoryCache[params.key] = {
+          messages: msgs,
+          endOfList: existing.endOfList,
+          startKeyHistory: existing.startKeyHistory
+        };
+      }
       return result;
     });
     this.pendingAgentRequests[params.key] = run;
@@ -1244,7 +1269,6 @@ var ChatSession = class {
       serviceId: id.serviceId,
       history: historyForLlm
     });
-    var requestToken = this.state.gateRefreshToken;
     var run = this.dispatchAgentRequest({
       key,
       serviceId: id.serviceId,
@@ -1259,7 +1283,7 @@ var ChatSession = class {
     });
     Promise.resolve(run).catch(function() {
     }).then(function() {
-      if (requestToken !== self.state.gateRefreshToken || self.getHistoryCacheKey() !== key) return;
+      if (!(self.host.isViewMounted() && self.getHistoryCacheKey() === key)) return;
       self.state.sending = false;
       return Promise.resolve(self.typewriteLatestReply(key)).then(function() {
         self.host.scrollToBottom(true);
@@ -1873,7 +1897,7 @@ var ChatSession = class {
           if (item.status !== "running" && item.status !== "pending") return;
           if (!item.poll || !item.id) return;
           if (self.historyItemPolls.has(item.id)) return;
-          if (item.status === "running" && self.pendingAgentRequests[self.getHistoryCacheKey()]) return;
+          if ((item.status === "running" || item.status === "pending") && self.pendingAgentRequests[self.getHistoryCacheKey()]) return;
           self.historyItemPolls.set(item.id, true);
           var capturedId = item.id;
           var pp = item.poll({