bunnyquery 1.4.3 → 1.4.5

package/dist/engine.mjs

@@ -1122,11 +1122,17 @@ var ChatSession = class {
   }
   dispatchAgentRequest(params) {
     var self = this;
+    var dispatchItemId;
     var sendAndPoll = function() {
       return Promise.resolve(
         self._callProviderFor(params.aiPlatform, params.text, params.boundedMessages, params.systemPrompt, params.aiModel, params.userId, params.extractContent)
       ).then(function(initial) {
         if (initial && initial.poll && (initial.status === "pending" || initial.status === "running")) {
+          if (initial.id) {
+            if (dispatchItemId && dispatchItemId !== initial.id) self.historyItemPolls.delete(dispatchItemId);
+            dispatchItemId = initial.id;
+            self.historyItemPolls.set(initial.id, true);
+          }
           return initial.poll({ latency: POLL_INTERVAL });
         }
         return initial;
@@ -1148,13 +1154,39 @@ 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];
+      if (dispatchItemId) self.historyItemPolls.delete(dispatchItemId);
+      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;
@@ -1207,6 +1239,7 @@ var ChatSession = class {
           self.host.notify();
         }
         if (result && result.poll && (result.status === "pending" || result.status === "running")) {
+          if (serverId) self.historyItemPolls.set(serverId, true);
           return result.poll({ latency: POLL_INTERVAL }).then(function(res) {
             return self.onQueuedSendResponse(capturedComposed, res, capturedPlatform, serverId);
           }).catch(function(err) {
@@ -1244,7 +1277,6 @@ var ChatSession = class {
       serviceId: id.serviceId,
       history: historyForLlm
     });
-    var requestToken = this.state.gateRefreshToken;
     var run = this.dispatchAgentRequest({
       key,
       serviceId: id.serviceId,
@@ -1259,7 +1291,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);
@@ -1347,6 +1379,7 @@ var ChatSession = class {
     else this.state.messages.push(msg);
   }
   onQueuedSendResponse(_composed, response, platform, serverId) {
+    if (serverId) this.historyItemPolls.delete(serverId);
     var targetIdx = this.resolveQueuedUserBubble(serverId);
     if (targetIdx === void 0) {
       this.host.notify();
@@ -1381,6 +1414,7 @@ var ChatSession = class {
     this.host.scrollToBottom(true);
   }
   onQueuedSendError(_composed, err, serverId) {
+    if (serverId) this.historyItemPolls.delete(serverId);
     var isNotExists = err && (err.code === "NOT_EXISTS" || err.body && err.body.code === "NOT_EXISTS");
     if (isNotExists) {
       var userIdx = serverId ? this.state.messages.findIndex(function(m) {
@@ -1873,7 +1907,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({