@ai-group/chat-sdk 0.2.8 → 0.3.0

package/dist/esm/hooks/useAgentGenerator.js

@@ -10,7 +10,7 @@ import { MessageRole, MessageStatus } from "../types/XAiMessage";
 import { ErrorCode } from "../types/XAiProvider";
 import { request } from "../utils/request";
 import { safeJsonParse } from "../utils";
-import { fetchAppConfig } from "../services/api";
+import { fetchAppConfig, fetchSessionList, fetchSessionDetail } from "../services/api";
 // 图标
 import documentIcon from "../assets/document.png";
 import completedIcon from "../assets/completed.png";
@@ -61,14 +61,27 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
     onError = _ref.onError,
     onSuccess = _ref.onSuccess;
   var textSpeed = 60;
+  // 聊天记录
   var _useState = useState([]),
     _useState2 = _slicedToArray(_useState, 2),
     messages = _useState2[0],
     setMessages = _useState2[1];
-  var _useState3 = useState(null),
+  // 会话记录
+  var _useState3 = useState([]),
     _useState4 = _slicedToArray(_useState3, 2),
-    appInfo = _useState4[0],
-    setAppInfo = _useState4[1];
+    sessionsList = _useState4[0],
+    setSessionList = _useState4[1];
+  // 当前会话
+  var _useState5 = useState(''),
+    _useState6 = _slicedToArray(_useState5, 2),
+    currentSession = _useState6[0],
+    setCurrentSession = _useState6[1];
+  // 应用 配置
+  var _useState7 = useState(null),
+    _useState8 = _slicedToArray(_useState7, 2),
+    appInfo = _useState8[0],
+    setAppInfo = _useState8[1];
+  // 请求 配置
   var requestInfo = useRef({});
   var _ref2 = config || {},
     appNo = _ref2.appNo,
@@ -206,6 +219,7 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
               description: content
             }]
           };
+          // 插到末尾
           newMessages.push(newMessage);
         } else {
           var suggestMsg = newMessages[suggestMessageIndex];
@@ -356,92 +370,82 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
   // 添加对话(非普通对话)
   var setMessage = /*#__PURE__*/function () {
     var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(messageType, content) {
-      var prologueMessageIndex, newMessage, suggestMessageIndex, _prologueMessageIndex, _newMessage;
       return _regeneratorRuntime().wrap(function _callee2$(_context2) {
         while (1) switch (_context2.prev = _context2.next) {
           case 0:
             _context2.t0 = messageType;
-            _context2.next = _context2.t0 === 'prologue' ? 3 : _context2.t0 === 'suggest' ? 6 : 9;
+            _context2.next = _context2.t0 === 'prologue' ? 3 : _context2.t0 === 'suggest' ? 5 : 7;
             break;
           case 3:
-            prologueMessageIndex = messages.findIndex(function (item) {
-              var _item$extra;
-              return (_item$extra = item.extra) === null || _item$extra === void 0 ? void 0 : _item$extra.prologue;
-            });
-            if (prologueMessageIndex === -1) {
-              newMessage = {
-                type: 'TextMessage',
-                id: uuidV4(),
-                createdAt: new Date(),
-                role: MessageRole.assistant,
-                status: MessageStatus.done,
-                content: {
-                  text: content
-                },
-                extra: {
-                  prologue: true,
-                  noFooter: true
-                }
-              };
-              setMessages(function (prev) {
-                return [newMessage].concat(_toConsumableArray(prev));
+            setMessages(function (prev) {
+              var prologueMessageIndex = prev.findIndex(function (item) {
+                var _item$extra;
+                return (_item$extra = item.extra) === null || _item$extra === void 0 ? void 0 : _item$extra.prologue;
               });
-            } else {
-              setMessages(function (prev) {
-                return prev.map(function (item) {
-                  var _item$extra2;
-                  if ((_item$extra2 = item.extra) !== null && _item$extra2 !== void 0 && _item$extra2.prologue && item.content && 'text' in item.content) {
-                    item.content.text = content;
+              if (prologueMessageIndex === -1) {
+                var newMessage = {
+                  type: 'TextMessage',
+                  id: uuidV4(),
+                  createdAt: new Date(),
+                  role: MessageRole.assistant,
+                  status: MessageStatus.done,
+                  content: {
+                    text: content
+                  },
+                  extra: {
+                    prologue: true,
+                    noFooter: true
                   }
-                  return item;
-                });
+                };
+                return [newMessage].concat(_toConsumableArray(prev));
+              }
+              return prev.map(function (item) {
+                var _item$extra2;
+                if ((_item$extra2 = item.extra) !== null && _item$extra2 !== void 0 && _item$extra2.prologue && item.content && 'text' in item.content) {
+                  item.content.text = content;
+                }
+                return item;
               });
-            }
-            return _context2.abrupt("break", 10);
-          case 6:
-            suggestMessageIndex = messages.findIndex(function (item) {
-              return item.type === 'SuggestionMessage';
             });
-            if (suggestMessageIndex === -1) {
-              _prologueMessageIndex = messages.findIndex(function (item) {
-                var _item$extra3;
-                return (_item$extra3 = item.extra) === null || _item$extra3 === void 0 ? void 0 : _item$extra3.prologue;
+            return _context2.abrupt("break", 8);
+          case 5:
+            setMessages(function (prev) {
+              var suggestMessageIndex = prev.findIndex(function (item) {
+                return item.type === 'SuggestionMessage';
               });
-              _newMessage = {
-                id: uuidV4(),
-                type: 'SuggestionMessage',
-                createdAt: new Date(),
-                status: MessageStatus.done,
-                role: MessageRole.suggestion,
-                content: content
-              };
-              if (_prologueMessageIndex === -1) {
-                // 如果没有开场白，则插入到最前面
-                setMessages(function (prev) {
-                  return [_newMessage].concat(_toConsumableArray(prev));
+              if (suggestMessageIndex === -1) {
+                var prologueMessageIndex = prev.findIndex(function (item) {
+                  var _item$extra3;
+                  return (_item$extra3 = item.extra) === null || _item$extra3 === void 0 ? void 0 : _item$extra3.prologue;
                 });
-              } else {
+                var newMessage = {
+                  id: uuidV4(),
+                  type: 'SuggestionMessage',
+                  createdAt: new Date(),
+                  status: MessageStatus.done,
+                  role: MessageRole.suggestion,
+                  content: content
+                };
+                if (prologueMessageIndex === -1) {
+                  // 如果没有开场白，则插入到最前面
+                  return [newMessage].concat(_toConsumableArray(prev));
+                }
                 // 如果有开场白，则插到开场白后面
-                setMessages(function (prev) {
-                  var newMessages = _toConsumableArray(prev);
-                  newMessages.splice(_prologueMessageIndex + 1, 0, _newMessage);
-                  return newMessages;
-                });
+                var newMessages = _toConsumableArray(prev);
+                newMessages.splice(prologueMessageIndex + 1, 0, newMessage);
+                return newMessages;
               }
-            } else {
-              setMessages(function (prev) {
-                return prev.map(function (item) {
-                  if (item.type === 'SuggestionMessage') {
-                    item.content = content;
-                  }
-                  return item;
-                });
+              return prev.map(function (item) {
+                if (item.type === 'SuggestionMessage') {
+                  item.content = content;
+                }
+                return item;
               });
-            }
-            return _context2.abrupt("break", 10);
-          case 9:
-            return _context2.abrupt("break", 10);
-          case 10:
+            });
+            return _context2.abrupt("break", 8);
+          case 7:
+            return _context2.abrupt("break", 8);
+          case 8:
           case "end":
             return _context2.stop();
         }
@@ -484,7 +488,8 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
             start("".concat(url, "/appouter/").concat(appNo, "/sendMsg?pt=").concat(pt, "&tc=").concat(tc), {
               appParams: {
                 inputs_msg: content
-              }
+              },
+              sessionId: currentSession
             }, {
               Authorization: "Bearer ".concat(token)
             });
@@ -550,47 +555,193 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
     });
   };
 
+  // 初始化历史记录
+  var formatMessages = function formatMessages(data) {
+    setMessages(function (prev) {
+      var initMessagesList = [];
+      data.forEach(function (item) {
+        var _ref7 = item || {},
+          content = _ref7.content,
+          messageTime = _ref7.messageTime,
+          role = _ref7.role;
+        if (role === 'USER') {
+          var newMessage = {
+            type: 'TextMessage',
+            id: uuidV4(),
+            createdAt: messageTime,
+            role: MessageRole.user,
+            status: MessageStatus.done,
+            content: {
+              text: content
+            }
+          };
+          initMessagesList.push(newMessage);
+        } else if (role === 'AI') {
+          var _contentObj$result;
+          var contentObj = safeJsonParse(content);
+          var _newMessage = {
+            id: uuidV4(),
+            type: 'TextMessage',
+            createdAt: messageTime,
+            role: MessageRole.assistant,
+            content: {
+              text: contentObj === null || contentObj === void 0 || (_contentObj$result = contentObj.result) === null || _contentObj$result === void 0 ? void 0 : _contentObj$result.result
+            },
+            thinks: [],
+            status: MessageStatus.done
+          };
+          initMessagesList.push(_newMessage);
+        }
+      });
+      return [].concat(_toConsumableArray(prev), initMessagesList);
+    });
+  };
+
+  // 初始化开场白
+  var initPrologue = /*#__PURE__*/function () {
+    var _ref8 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4() {
+      var _appInfo$onboardingIn, onboardingInfo, _ref9, _ref9$suggested_quest, suggested_questions, _ref9$prologue, prologue, formatSuggests, suggestMessages, hasMessages;
+      return _regeneratorRuntime().wrap(function _callee4$(_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
+          case 0:
+            _appInfo$onboardingIn = appInfo.onboardingInfo, onboardingInfo = _appInfo$onboardingIn === void 0 ? {} : _appInfo$onboardingIn; // eslint-disable-next-line camelcase
+            _ref9 = onboardingInfo || {}, _ref9$suggested_quest = _ref9.suggested_questions, suggested_questions = _ref9$suggested_quest === void 0 ? [] : _ref9$suggested_quest, _ref9$prologue = _ref9.prologue, prologue = _ref9$prologue === void 0 ? '' : _ref9$prologue; // 初始化开场白
+            if (prologue) {
+              setMessage('prologue', prologue);
+            } else {
+              setMessages(function (prev) {
+                return prev.filter(function (item) {
+                  var _item$extra4;
+                  return !((_item$extra4 = item.extra) !== null && _item$extra4 !== void 0 && _item$extra4.prologue);
+                });
+              });
+            }
+
+            // 过滤空内容
+            formatSuggests = suggested_questions.filter(function (item) {
+              return item;
+            });
+            if (formatSuggests.length > 0) {
+              suggestMessages = formatSuggests.map(function (item, index) {
+                return {
+                  key: index + 1,
+                  description: item
+                };
+              }); // 页面为空时展示开场白预设问题
+              hasMessages = messages.some(function (item) {
+                var _item$extra5;
+                return !((_item$extra5 = item.extra) !== null && _item$extra5 !== void 0 && _item$extra5.prologue) && item.type !== 'SuggestionMessage';
+              });
+              if (!hasMessages) {
+                setMessage('suggest', suggestMessages);
+              }
+            } else {
+              setMessages(function (prev) {
+                return prev.filter(function (item) {
+                  return !(item.type === 'SuggestionMessage');
+                });
+              });
+            }
+          case 5:
+          case "end":
+            return _context4.stop();
+        }
+      }, _callee4);
+    }));
+    return function initPrologue() {
+      return _ref8.apply(this, arguments);
+    };
+  }();
+
+  // 获取聊天记录
+  var initAppConversations = /*#__PURE__*/function () {
+    var _ref10 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
+      var result, _result$data2, sessionId, sessionResult, _sessionResult$data, sessionsData;
+      return _regeneratorRuntime().wrap(function _callee5$(_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
+          case 0:
+            _context5.prev = 0;
+            _context5.next = 3;
+            return fetchSessionList({
+              url: url,
+              appNo: appNo,
+              pt: pt,
+              tc: tc,
+              token: token
+            });
+          case 3:
+            result = _context5.sent;
+            if (!(result !== null && result !== void 0 && result.success)) {
+              _context5.next = 24;
+              break;
+            }
+            if (!((result === null || result === void 0 || (_result$data2 = result.data) === null || _result$data2 === void 0 ? void 0 : _result$data2.length) > 0)) {
+              _context5.next = 19;
+              break;
+            }
+            setSessionList(result.data);
+            sessionId = result.data[0].sessionId;
+            if (!sessionId) {
+              _context5.next = 16;
+              break;
+            }
+            setCurrentSession(sessionId); // 设置当前会话 id;
+            _context5.next = 12;
+            return fetchSessionDetail({
+              url: url,
+              appNo: appNo,
+              pt: pt,
+              tc: tc,
+              sessionId: sessionId,
+              token: token
+            });
+          case 12:
+            sessionResult = _context5.sent;
+            if (sessionResult !== null && sessionResult !== void 0 && sessionResult.success) {
+              sessionsData = (sessionResult === null || sessionResult === void 0 || (_sessionResult$data = sessionResult.data) === null || _sessionResult$data === void 0 || (_sessionResult$data = _sessionResult$data.list) === null || _sessionResult$data === void 0 || (_sessionResult$data = _sessionResult$data[0]) === null || _sessionResult$data === void 0 ? void 0 : _sessionResult$data.sessionMessages) || [];
+              if (sessionsData.length > 0) {
+                formatMessages(sessionsData);
+              }
+            }
+            _context5.next = 17;
+            break;
+          case 16:
+            // 兼容 无 session异常
+            setCurrentSession(uuidV4());
+          case 17:
+            _context5.next = 22;
+            break;
+          case 19:
+            setSessionList([]);
+            setCurrentSession(uuidV4());
+            initPrologue();
+          case 22:
+            _context5.next = 25;
+            break;
+          case 24:
+            initPrologue();
+          case 25:
+            _context5.next = 29;
+            break;
+          case 27:
+            _context5.prev = 27;
+            _context5.t0 = _context5["catch"](0);
+          case 29:
+          case "end":
+            return _context5.stop();
+        }
+      }, _callee5, null, [[0, 27]]);
+    }));
+    return function initAppConversations() {
+      return _ref10.apply(this, arguments);
+    };
+  }();
+
   // 初始化应用
   useEffect(function () {
     if (appInfo) {
-      var _appInfo$onboardingIn = appInfo.onboardingInfo,
-        onboardingInfo = _appInfo$onboardingIn === void 0 ? {} : _appInfo$onboardingIn;
-      // eslint-disable-next-line camelcase
-      var _ref7 = onboardingInfo || {},
-        _ref7$suggested_quest = _ref7.suggested_questions,
-        suggested_questions = _ref7$suggested_quest === void 0 ? [] : _ref7$suggested_quest,
-        _ref7$prologue = _ref7.prologue,
-        prologue = _ref7$prologue === void 0 ? '' : _ref7$prologue;
-      if (prologue) {
-        setMessage('prologue', prologue);
-      } else {
-        setMessages(function (prev) {
-          return prev.filter(function (item) {
-            var _item$extra4;
-            return !((_item$extra4 = item.extra) !== null && _item$extra4 !== void 0 && _item$extra4.prologue);
-          });
-        });
-      }
-
-      // 过滤空内容
-      var formatSuggests = suggested_questions.filter(function (item) {
-        return item;
-      });
-      if (formatSuggests.length > 0) {
-        var suggestMessages = formatSuggests.map(function (item, index) {
-          return {
-            key: index + 1,
-            description: item
-          };
-        });
-        setMessage('suggest', suggestMessages);
-      } else {
-        setMessages(function (prev) {
-          return prev.filter(function (item) {
-            return !(item.type === 'SuggestionMessage');
-          });
-        });
-      }
+      // 初始化历史数据
+      initAppConversations();
     }
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [appInfo]);
@@ -599,8 +750,11 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
     messages: messages,
     loading: loading,
     content: responseData.content,
+    sessionsList: sessionsList,
+    currentSession: currentSession,
     reChat: reChat,
     chat: chat,
+    setCurrentSession: setCurrentSession,
     setMessage: setMessage,
     setMessages: setMessages,
     stopChat: stopChat