@ai-group/chat-sdk 3.5.0 → 3.5.1

package/dist/esm/hooks/useADKChat.js

@@ -44,6 +44,9 @@ var combineTextParts = function combineTextParts(parts) {
   }
   return result;
 };
+
+// 后台会话连接状态
+
 function useADKChat(_ref) {
   var _ref$url = _ref.url,
     url = _ref$url === void 0 ? window.location.origin : _ref$url,
@@ -119,6 +122,13 @@ function useADKChat(_ref) {
     setMessages = _useState20[1];
   // 用户自定义业务参数 (stateDelta)，每次发送请求时携带
   var stateDeltaRef = useRef(undefined);
+
+  // ===== 多会话后台连接管理 =====
+  var backgroundRef = useRef(new Map());
+  var messagesRef = useRef(messages);
+  messagesRef.current = messages;
+  var currentSessionIdRef = useRef(currentSessionId);
+  currentSessionIdRef.current = currentSessionId;
   var mergedMessages = useMemo(function () {
     var fnResMap = {};
     messages.forEach(function (msg) {
@@ -366,15 +376,24 @@ function useADKChat(_ref) {
   // 发送消息请求
   var sendMessageRequest = useCallback(function (values) {
     return new Promise(function (resolve) {
-      if (ctrl.current) {
+      var targetSessionId = values.sessionId || currentSessionIdRef.current;
+
+      // 只 abort 同一 session 的旧连接
+      if (targetSessionId === currentSessionIdRef.current && ctrl.current) {
         ctrl.current.abort();
       }
-      setLoading(true);
-      ctrl.current = new AbortController();
+      var newCtrl = new AbortController();
+      if (targetSessionId === currentSessionIdRef.current) {
+        setLoading(true);
+        ctrl.current = newCtrl;
+      }
       var requestUrl = "".concat(url, "/api/gateway-web/openApi/v1/aizt/app/").concat(appNo, "/sendMsgStreaming");
+
+      // 后台流式文本拼接
+      var bgTextMsg = null;
       fetchEventSource(requestUrl, {
         method: "POST",
-        signal: ctrl.current.signal,
+        signal: newCtrl.signal,
         body: JSON.stringify(values),
         openWhenHidden: true,
         headers: {
@@ -418,7 +437,6 @@ function useADKChat(_ref) {
             // 转换为旧格式
             chunkJson = _objectSpread(_objectSpread({}, rawData.data), {}, {
               content: rawData.data,
-              // data 本身就是 content
               invocationId: rawData.result.invocationId,
               sessionId: rawData.result.sessionId,
               id: rawData.result.invocationId
@@ -428,6 +446,84 @@ function useADKChat(_ref) {
             message.warning(chunkJson.error);
             return;
           }
+
+          // 判断是否为后台 session
+          var isBackground = currentSessionIdRef.current !== targetSessionId;
+          if (isBackground) {
+            // 后台 session：更新 backgroundRef 中的消息
+            var bg = backgroundRef.current.get(targetSessionId);
+            if (!bg) return;
+            if (chunkJson.content) {
+              if (chunkJson.content.role === "followup") return;
+              var _iterator2 = _createForOfIteratorHelper(combineTextParts(chunkJson.content.parts)),
+                _step2;
+              try {
+                for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+                  var part = _step2.value;
+                  var isPureText = part.text && !part.fileData;
+                  if (isPureText) {
+                    var _chunkJson, _bgTextMsg;
+                    if (((_chunkJson = chunkJson) === null || _chunkJson === void 0 ? void 0 : _chunkJson.partial) === false) continue;
+                    var newChunk = part.text || "";
+                    if (!bgTextMsg || ((_bgTextMsg = bgTextMsg) === null || _bgTextMsg === void 0 ? void 0 : _bgTextMsg.thought) !== part.thought) {
+                      var newMessage = {
+                        id: v4(),
+                        author: chunkJson.author,
+                        invocationId: chunkJson.invocationId,
+                        eventId: chunkJson.id,
+                        timestamp: chunkJson.timestamp,
+                        role: "bot",
+                        text: newChunk,
+                        thought: part.thought
+                      };
+                      bgTextMsg = newMessage;
+                      bg.messages = [].concat(_toConsumableArray(bg.messages), [newMessage]);
+                    } else {
+                      bgTextMsg = _objectSpread(_objectSpread({}, bgTextMsg), {}, {
+                        text: (bgTextMsg.text || "") + newChunk
+                      });
+                      bg.messages = bg.messages.map(function (m) {
+                        return m.id === bgTextMsg.id ? bgTextMsg : m;
+                      });
+                      if (chunkJson.finishReason === "stop") {
+                        bgTextMsg = null;
+                      }
+                    }
+                  } else {
+                    bgTextMsg = null;
+                    var msg = {
+                      id: v4(),
+                      author: chunkJson.author,
+                      invocationId: chunkJson.invocationId,
+                      eventId: chunkJson.id,
+                      timestamp: chunkJson.timestamp,
+                      role: "bot"
+                    };
+                    if (part.functionCall) msg.functionCall = part.functionCall;else if (part.functionResponse) msg.functionResponse = part.functionResponse;else if (part.executableCode) msg.executableCode = part.executableCode;else if (part.codeExecutionResult) msg.codeExecutionResult = part.codeExecutionResult;else {
+                      msg.text = part.text;
+                      msg.thought = part.thought;
+                      if (part.fileData) msg.fileData = [part.fileData];
+                    }
+                    bg.messages = [].concat(_toConsumableArray(bg.messages), [msg]);
+                  }
+                }
+              } catch (err) {
+                _iterator2.e(err);
+              } finally {
+                _iterator2.f();
+              }
+            } else if (chunkJson.errorMessage) {
+              bg.messages = [].concat(_toConsumableArray(bg.messages), [{
+                id: v4(),
+                text: chunkJson.errorMessage,
+                role: "bot"
+              }]);
+            }
+            bg.textMsgRef = bgTextMsg;
+            return;
+          }
+
+          // 前台 session：正常处理
           onStream === null || onStream === void 0 || onStream(chunkJson);
           updateLastBotEventMeta(chunkJson);
           if (chunkJson.content) {
@@ -437,19 +533,19 @@ function useADKChat(_ref) {
               return;
             }
             // eslint-disable-next-line no-restricted-syntax
-            var _iterator2 = _createForOfIteratorHelper(combineTextParts(chunkJson.content.parts)),
-              _step2;
+            var _iterator3 = _createForOfIteratorHelper(combineTextParts(chunkJson.content.parts)),
+              _step3;
             try {
-              for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
-                var _chunkJson;
-                var part = _step2.value;
+              for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
+                var _chunkJson2;
+                var _part = _step3.value;
                 // partial为 true 时忽略该条消息文本
-                processPart(chunkJson, part, ((_chunkJson = chunkJson) === null || _chunkJson === void 0 ? void 0 : _chunkJson.partial) === false);
+                processPart(chunkJson, _part, ((_chunkJson2 = chunkJson) === null || _chunkJson2 === void 0 ? void 0 : _chunkJson2.partial) === false);
               }
             } catch (err) {
-              _iterator2.e(err);
+              _iterator3.e(err);
             } finally {
-              _iterator2.f();
+              _iterator3.f();
             }
           } else if (chunkJson.errorMessage) {
             processErrorMessage(chunkJson);
@@ -458,16 +554,31 @@ function useADKChat(_ref) {
           }
         },
         onclose: function onclose() {
-          setLoading(false);
-          if (textMsgRef.current) {
-            var _textMsgRef$current;
-            onMessage === null || onMessage === void 0 || onMessage(((_textMsgRef$current = textMsgRef.current) === null || _textMsgRef$current === void 0 ? void 0 : _textMsgRef$current.text) || "", textMsgRef.current);
+          var isBackground = currentSessionIdRef.current !== targetSessionId;
+          if (isBackground) {
+            var bg = backgroundRef.current.get(targetSessionId);
+            if (bg) {
+              bg.loading = false;
+              bg.textMsgRef = null;
+            }
+          } else {
+            setLoading(false);
+            if (textMsgRef.current) {
+              var _textMsgRef$current;
+              onMessage === null || onMessage === void 0 || onMessage(((_textMsgRef$current = textMsgRef.current) === null || _textMsgRef$current === void 0 ? void 0 : _textMsgRef$current.text) || "", textMsgRef.current);
+            }
+            textMsgRef.current = null;
           }
-          textMsgRef.current = null;
           resolve();
         },
         onerror: function onerror(error) {
-          setLoading(false);
+          var isBackground = currentSessionIdRef.current !== targetSessionId;
+          if (isBackground) {
+            var bg = backgroundRef.current.get(targetSessionId);
+            if (bg) bg.loading = false;
+          } else {
+            setLoading(false);
+          }
           resolve();
           console.error("EventSource failed:", error);
           throw error;
@@ -678,13 +789,14 @@ function useADKChat(_ref) {
     });
   };
 
-  // 停止会话
+  // 停止会话（只中断前台连接）
   var stopChat = useCallback(function () {
     var _ctrl$current;
     (_ctrl$current = ctrl.current) === null || _ctrl$current === void 0 || _ctrl$current.abort();
+    ctrl.current = null;
     setLoading(false);
     textMsgRef.current = null;
-  }, [ctrl]);
+  }, []);
 
   // 清除/重置会话
   var clearChat = function clearChat() {
@@ -737,11 +849,11 @@ function useADKChat(_ref) {
           finishReason: item.finishReason,
           raw: item
         };
-        var _iterator3 = _createForOfIteratorHelper(textParts),
-          _step3;
+        var _iterator4 = _createForOfIteratorHelper(textParts),
+          _step4;
         try {
-          for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
-            var part = _step3.value;
+          for (_iterator4.s(); !(_step4 = _iterator4.n()).done;) {
+            var part = _step4.value;
             if (part.inlineData) {
               msg.inlineData = {
                 displayName: part.inlineData.displayName,
@@ -758,9 +870,9 @@ function useADKChat(_ref) {
             }
           }
         } catch (err) {
-          _iterator3.e(err);
+          _iterator4.e(err);
         } finally {
-          _iterator3.f();
+          _iterator4.f();
         }
         if (fileDataParts.length > 0) {
           msg.fileData = fileDataParts.map(function (part) {
@@ -815,12 +927,12 @@ function useADKChat(_ref) {
   // 设置当前会话详情
   var setCurrentSession = /*#__PURE__*/function () {
     var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4(sessionId) {
-      var _yield$fetchSessionDe, data, result, _appInfo$onboardingIn3, _appInfo$onboardingIn4;
+      var prevSessionId, bg, _yield$fetchSessionDe, data, result, _appInfo$onboardingIn3, _appInfo$onboardingIn4;
       return _regeneratorRuntime().wrap(function _callee4$(_context4) {
         while (1) switch (_context4.prev = _context4.next) {
           case 0:
             if (!sessionId) {
-              _context4.next = 15;
+              _context4.next = 26;
               break;
             }
             if (!(sessionId === currentSessionId)) {
@@ -829,39 +941,72 @@ function useADKChat(_ref) {
             }
             return _context4.abrupt("return");
           case 3:
-            // 切换会话时，abort 当前流式请求，防止数据污染和 loading 状态泄漏
-            stopChat();
-            // 重置 initialized 以便 XAdkChatbot 能重新触发置底滚动
+            // 1. 将当前会话挂起到后台（不中断 SSE 连接）
+            prevSessionId = currentSessionId;
+            if (ctrl.current) {
+              // 有活跃流式连接，保存到后台
+              backgroundRef.current.set(prevSessionId, {
+                ctrl: ctrl.current,
+                messages: _toConsumableArray(messagesRef.current),
+                loading: true,
+                textMsgRef: textMsgRef.current
+              });
+              ctrl.current = null;
+              textMsgRef.current = null;
+              setLoading(false);
+            } else if (messagesRef.current.length > 0) {
+              // 无活跃连接但有消息，缓存用于切回
+              backgroundRef.current.set(prevSessionId, {
+                ctrl: new AbortController(),
+                messages: _toConsumableArray(messagesRef.current),
+                loading: false,
+                textMsgRef: null
+              });
+            }
+
+            // 2. 尝试从后台恢复目标会话
+            bg = backgroundRef.current.get(sessionId);
+            if (!bg) {
+              _context4.next = 15;
+              break;
+            }
+            setMessages(bg.messages);
+            setLoading(bg.loading);
+            textMsgRef.current = bg.textMsgRef;
+            ctrl.current = bg.loading ? bg.ctrl : null;
+            setCurrentSessionId(sessionId);
+            setInitialized(true);
+            backgroundRef.current.delete(sessionId);
+            return _context4.abrupt("return");
+          case 15:
+            // 3. 后台没有目标 session，走 API 加载（原有逻辑）
             setInitialized(false);
-            setCurrentSessionId(sessionId); // 设置当前会话 id;
-            _context4.next = 8;
+            setCurrentSessionId(sessionId);
+            _context4.next = 19;
             return fetchSessionDetail({
               url: url,
               appNo: appNo,
               sessionId: sessionId,
               token: token
             });
-          case 8:
+          case 19:
             _yield$fetchSessionDe = _context4.sent;
             data = _yield$fetchSessionDe.data;
             result = _yield$fetchSessionDe.result;
             if ((result === null || result === void 0 ? void 0 : result.code) === API_SUCCESS_CODE) {
               setPrologue((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn3 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn3 === void 0 ? void 0 : _appInfo$onboardingIn3.prologue) || "");
-              // 新接口直接返回按 event/item 结构的数组，传入 formatMessages 处理
               if (Array.isArray(data) && data.length > 0) {
                 formatMessages(data, true);
               } else {
                 setSuggestedQuestions((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn4 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn4 === void 0 ? void 0 : _appInfo$onboardingIn4.suggested_questions) || []);
               }
             }
-            // 恢复 initialized，触发 XAdkChatbot 的初始化置底逻辑
             setInitialized(true);
-            _context4.next = 16;
+            _context4.next = 27;
             break;
-          case 15:
-            // 兼容 无 session异常
+          case 26:
             setCurrentSessionId(v4());
-          case 16:
+          case 27:
           case "end":
             return _context4.stop();
         }
@@ -1179,6 +1324,19 @@ function useADKChat(_ref) {
       initAppConversations(true);
     }
   }, [appInfo]);
+
+  // 组件卸载时清理所有连接（前台 + 后台）
+  useEffect(function () {
+    return function () {
+      var _ctrl$current2;
+      (_ctrl$current2 = ctrl.current) === null || _ctrl$current2 === void 0 || _ctrl$current2.abort();
+      backgroundRef.current.forEach(function (bg) {
+        var _bg$ctrl;
+        return (_bg$ctrl = bg.ctrl) === null || _bg$ctrl === void 0 ? void 0 : _bg$ctrl.abort();
+      });
+      backgroundRef.current.clear();
+    };
+  }, []);
   return {
     appInfo: appInfo,
     startChat: startChat,