@ai-group/chat-sdk 0.4.2 → 0.5.1

package/dist/esm/hooks/useAgentGenerator.js

@@ -1,3 +1,4 @@
+import _typeof from "@babel/runtime/helpers/esm/typeof";
 import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
 import _regeneratorRuntime from "@babel/runtime/helpers/esm/regeneratorRuntime";
 import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
@@ -7,14 +8,14 @@ import { useState, useEffect, useRef } from 'react';
 import { v4 as uuidV4 } from 'uuid';
 import { useEventStreamRequest } from "./useEventStreamRequest";
 import { MessageRole, MessageStatus } from "../types/XAiMessage";
-import { ErrorCode, SuccessCode } from "../types/XAiProvider";
+import { ErrorCode, SuccessCode } from "../types";
 import { request } from "../utils/request";
 import { safeJsonParse } from "../utils";
 import { fetchAppConfig, fetchSessionList, fetchSessionDetail } from "../services/api";
 // 图标
 import documentIcon from "../assets/document.png";
-import completedIcon from "../assets/completed.png";
-import completedExpandIcon from "../assets/completed-black.png";
+// import completedIcon from '@/assets/completed.png';
+// import completedExpandIcon from '@/assets/completed-black.png';
 import documentExpandIcon from "../assets/document-black.png";
 import qaIcon from "../assets/qa.png";
 import qaExpandIcon from "../assets/qa-black.png";
@@ -59,6 +60,7 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
     token = _ref.token,
     config = _ref.config,
     onError = _ref.onError,
+    onMessage = _ref.onMessage,
     onSuccess = _ref.onSuccess;
   var textSpeed = 60;
   // 聊天记录
@@ -66,21 +68,44 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
     _useState2 = _slicedToArray(_useState, 2),
     messages = _useState2[0],
     setMessages = _useState2[1];
-  // 会话记录
-  var _useState3 = useState([]),
+  // 聊天记录分页设置
+  var _useState3 = useState({
+      total: 0,
+      pageNum: 1,
+      pageSize: 999
+    }),
     _useState4 = _slicedToArray(_useState3, 2),
-    sessionsList = _useState4[0],
-    setSessionList = _useState4[1];
-  // 当前会话
-  var _useState5 = useState(''),
+    messagePagination = _useState4[0],
+    setMessagePagination = _useState4[1];
+  // 会话记录
+  var _useState5 = useState([]),
     _useState6 = _slicedToArray(_useState5, 2),
-    currentSession = _useState6[0],
-    setCurrentSession = _useState6[1];
-  // 应用 配置
-  var _useState7 = useState(null),
+    sessionList = _useState6[0],
+    setSessionList = _useState6[1];
+  // 会话请求是否加载中
+  var _useState7 = useState(false),
     _useState8 = _slicedToArray(_useState7, 2),
-    appInfo = _useState8[0],
-    setAppInfo = _useState8[1];
+    sessionLoading = _useState8[0],
+    setSessionLoading = _useState8[1];
+  // 会话分页设置
+  var _useState9 = useState({
+      total: 0,
+      pageNum: 1,
+      pageSize: 20
+    }),
+    _useState10 = _slicedToArray(_useState9, 2),
+    sessionPagination = _useState10[0],
+    setSessionPagination = _useState10[1];
+  // 当前会话
+  var _useState11 = useState(''),
+    _useState12 = _slicedToArray(_useState11, 2),
+    currentSessionId = _useState12[0],
+    setCurrentSessionId = _useState12[1];
+  // 应用 配置
+  var _useState13 = useState(null),
+    _useState14 = _slicedToArray(_useState13, 2),
+    appInfo = _useState14[0],
+    setAppInfo = _useState14[1];
   // 请求 配置
   var requestInfo = useRef({});
   var _ref2 = config || {},
@@ -178,20 +203,22 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
       });
     }
     setMessages(function (prevMessages) {
+      var _aiMsg;
       var newMessages = _toConsumableArray(prevMessages);
       // 消息类型判断
       var _response = response,
         action = _response.action,
         data = _response.data;
-      var uniqueId = data.uniqueId,
-        contentType = data.contentType,
-        content = data.content,
-        toolType = data.toolType,
-        args = data.args,
-        knowledgeType = data.knowledgeType,
-        _data$success = data.success,
-        success = _data$success === void 0 ? true : _data$success,
-        errorMsg = data.errorMsg;
+      var _ref4 = data || {},
+        uniqueId = _ref4.uniqueId,
+        contentType = _ref4.contentType,
+        content = _ref4.content,
+        toolType = _ref4.toolType,
+        args = _ref4.args,
+        knowledgeType = _ref4.knowledgeType,
+        _ref4$success = _ref4.success,
+        success = _ref4$success === void 0 ? true : _ref4$success,
+        errorMsg = _ref4.errorMsg;
 
       // 建议类信息 特殊处理
       if (action === ActionType.suggest) {
@@ -323,19 +350,22 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
           }
           break;
         case ActionType.finish:
-          aiMsg.thinks = aiMsg.thinks || [];
-          aiMsg.thinks.push({
-            id: uniqueId,
-            createdAt: new Date(),
-            type: 'ActionExecutionMessage',
-            name: '运行完毕',
-            extra: _objectSpread(_objectSpread({}, data), {}, {
-              cost: data !== null && data !== void 0 && data.cost ? ((data === null || data === void 0 ? void 0 : data.cost) / 1000).toFixed(2) : 0,
-              icon: completedIcon,
-              expandIcon: completedExpandIcon
-            })
-          });
+          // aiMsg.thinks = aiMsg.thinks || [];
+          // aiMsg.thinks.push({
+          //   id: uniqueId,
+          //   createdAt: new Date(),
+          //   type: 'ActionExecutionMessage',
+          //   name: '运行完毕',
+          //   extra: {
+          //     ...data,
+          //     cost: data?.cost ? (data?.cost / 1000).toFixed(2) : 0,
+          //     icon: completedIcon,
+          //     expandIcon: completedExpandIcon,
+          //   },
+          // });
           aiMsg.status = MessageStatus.done;
+          // 收到一条完整，触发收到消息回调
+          onMessage === null || onMessage === void 0 || onMessage((_aiMsg = aiMsg) === null || _aiMsg === void 0 || (_aiMsg = _aiMsg.content) === null || _aiMsg === void 0 ? void 0 : _aiMsg.text, aiMsg);
           break;
         // 其他类型略
       }
@@ -360,7 +390,7 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
 
   // 添加对话(非普通对话)
   var setMessage = /*#__PURE__*/function () {
-    var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(messageType, content) {
+    var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(messageType, content) {
       return _regeneratorRuntime().wrap(function _callee2$(_context2) {
         while (1) switch (_context2.prev = _context2.next) {
           case 0:
@@ -443,27 +473,45 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
       }, _callee2);
     }));
     return function setMessage(_x, _x2) {
-      return _ref4.apply(this, arguments);
+      return _ref5.apply(this, arguments);
     };
   }();
 
   // 简单对话
   var chat = /*#__PURE__*/function () {
-    var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(content) {
-      var newMessage;
+    var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(content) {
+      var newMessage, inputs_msg;
       return _regeneratorRuntime().wrap(function _callee3$(_context3) {
         while (1) switch (_context3.prev = _context3.next) {
           case 0:
-            newMessage = {
-              type: 'TextMessage',
-              id: uuidV4(),
-              createdAt: new Date(),
-              role: MessageRole.user,
-              status: MessageStatus.done,
-              content: {
-                text: content
-              }
-            };
+            if (typeof content === 'string') {
+              newMessage = {
+                type: 'TextMessage',
+                id: uuidV4(),
+                createdAt: new Date(),
+                role: MessageRole.user,
+                status: MessageStatus.done,
+                content: {
+                  text: content
+                }
+              };
+            } else {
+              newMessage = _objectSpread({
+                type: 'TextMessage',
+                id: uuidV4(),
+                createdAt: new Date(),
+                role: MessageRole.user,
+                status: MessageStatus.done,
+                content: {
+                  text: content
+                }
+              }, content);
+            }
+            if (_typeof(newMessage.content) === 'object' && 'text' in newMessage.content && typeof newMessage.content.text === 'string') {
+              inputs_msg = newMessage.content.text;
+            } else {
+              inputs_msg = '';
+            }
             setMessages(function (prev) {
               // 若存在对话建议，先删除对话建议
               var suggestMessageIndex = messages.findIndex(function (item) {
@@ -478,20 +526,20 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
             });
             start("".concat(url, "/appouter/").concat(appNo, "/sendMsg?pt=").concat(pt, "&tc=").concat(tc), {
               appParams: {
-                inputs_msg: content
+                inputs_msg: inputs_msg
               },
-              sessionId: currentSession
+              sessionId: currentSessionId
             }, {
               Authorization: "Bearer ".concat(token)
             });
-          case 3:
+          case 4:
           case "end":
             return _context3.stop();
         }
       }, _callee3);
     }));
     return function chat(_x3) {
-      return _ref5.apply(this, arguments);
+      return _ref6.apply(this, arguments);
     };
   }();
 
@@ -548,13 +596,14 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
 
   // 初始化历史记录
   var formatMessages = function formatMessages(data) {
+    var init = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : false;
     setMessages(function (prev) {
       var initMessagesList = [];
       data.forEach(function (item) {
-        var _ref6 = item || {},
-          content = _ref6.content,
-          messageTime = _ref6.messageTime,
-          role = _ref6.role;
+        var _ref7 = item || {},
+          content = _ref7.content,
+          messageTime = _ref7.messageTime,
+          role = _ref7.role;
         if (role === 'USER') {
           var newMessage = {
             type: 'TextMessage',
@@ -582,19 +631,22 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
           initMessagesList.push(_newMessage);
         }
       });
+      if (init) {
+        return [].concat(initMessagesList);
+      }
       return [].concat(_toConsumableArray(prev), initMessagesList);
     });
   };
 
   // 初始化开场白
   var initPrologue = /*#__PURE__*/function () {
-    var _ref7 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4() {
-      var _appInfo$onboardingIn, onboardingInfo, _ref8, _ref8$suggested_quest, suggested_questions, _ref8$prologue, prologue, formatSuggests, suggestMessages, hasMessages;
+    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
-            _ref8 = onboardingInfo || {}, _ref8$suggested_quest = _ref8.suggested_questions, suggested_questions = _ref8$suggested_quest === void 0 ? [] : _ref8$suggested_quest, _ref8$prologue = _ref8.prologue, prologue = _ref8$prologue === void 0 ? '' : _ref8$prologue; // 初始化开场白
+            _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 {
@@ -638,14 +690,14 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
       }, _callee4);
     }));
     return function initPrologue() {
-      return _ref7.apply(this, arguments);
+      return _ref8.apply(this, arguments);
     };
   }();
 
   // 获取聊天记录
   var initAppConversations = /*#__PURE__*/function () {
-    var _ref9 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
-      var result, _result$data, resultData, sessionId, sessionResult, _sessionResult$data, sessionsData;
+    var _ref10 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
+      var result, _result$data, resultData, _result$data2, _result$data3, _result$data4, formatSessions, sessionId;
       return _regeneratorRuntime().wrap(function _callee5$(_context5) {
         while (1) switch (_context5.prev = _context5.next) {
           case 0:
@@ -660,77 +712,180 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
             });
           case 3:
             result = _context5.sent;
-            if (!(result !== null && result !== void 0 && result.success)) {
-              _context5.next = 26;
-              break;
+            if (result !== null && result !== void 0 && result.success) {
+              resultData = (result === null || result === void 0 || (_result$data = result.data) === null || _result$data === void 0 ? void 0 : _result$data.data) || [];
+              if (resultData.length > 0) {
+                formatSessions = resultData.map(function (item) {
+                  return {
+                    key: item.sessionId,
+                    label: item.title,
+                    timestamp: new Date(item.gmtCreated).getTime(),
+                    messages: [],
+                    meta: _objectSpread({}, item)
+                  };
+                });
+                setSessionList(formatSessions);
+                // 设置分页信息
+                setSessionPagination({
+                  pageNum: result === null || result === void 0 || (_result$data2 = result.data) === null || _result$data2 === void 0 ? void 0 : _result$data2.pageNum,
+                  pageSize: result === null || result === void 0 || (_result$data3 = result.data) === null || _result$data3 === void 0 ? void 0 : _result$data3.pageSize,
+                  total: result === null || result === void 0 || (_result$data4 = result.data) === null || _result$data4 === void 0 ? void 0 : _result$data4.totalCount
+                });
+                sessionId = resultData[0].sessionId;
+                setCurrentSessionDetail(sessionId);
+              } else {
+                // 页面报错 初始化
+                setSessionList([]);
+                setCurrentSessionId(uuidV4());
+                initPrologue();
+              }
+              // 成功回调
+              onSuccess === null || onSuccess === void 0 || onSuccess({
+                code: SuccessCode.APP_MESSAGES_INIT_SUCCESS,
+                message: '初始化聊天记录成功'
+              });
+            } else {
+              initPrologue();
             }
-            resultData = (result === null || result === void 0 || (_result$data = result.data) === null || _result$data === void 0 ? void 0 : _result$data.data) || [];
-            if (!(resultData.length > 0)) {
-              _context5.next = 20;
+            _context5.next = 9;
+            break;
+          case 7:
+            _context5.prev = 7;
+            _context5.t0 = _context5["catch"](0);
+          case 9:
+          case "end":
+            return _context5.stop();
+        }
+      }, _callee5, null, [[0, 7]]);
+    }));
+    return function initAppConversations() {
+      return _ref10.apply(this, arguments);
+    };
+  }();
+
+  // 设置当前会话详情
+  var setCurrentSessionDetail = /*#__PURE__*/function () {
+    var _ref11 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(sessionId) {
+      var sessionResult, _sessionResult$data, sessionsData;
+      return _regeneratorRuntime().wrap(function _callee6$(_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
+          case 0:
+            if (!sessionId) {
+              _context6.next = 10;
               break;
             }
-            setSessionList(resultData);
-            sessionId = resultData[0].sessionId;
-            if (!sessionId) {
-              _context5.next = 17;
+            if (!(sessionId === currentSessionId)) {
+              _context6.next = 3;
               break;
             }
-            setCurrentSession(sessionId); // 设置当前会话 id;
-            _context5.next = 13;
+            return _context6.abrupt("return");
+          case 3:
+            setCurrentSessionId(sessionId); // 设置当前会话 id;
+            _context6.next = 6;
             return fetchSessionDetail({
               url: url,
               appNo: appNo,
               pt: pt,
               tc: tc,
               sessionId: sessionId,
-              token: token
+              token: token,
+              pageNum: sessionPagination.pageNum,
+              pageSize: sessionPagination.pageSize
             });
-          case 13:
-            sessionResult = _context5.sent;
+          case 6:
+            sessionResult = _context6.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 ? void 0 : _sessionResult$data.reduce(function (acc, cur) {
                 return [].concat(_toConsumableArray(acc), _toConsumableArray(cur.sessionMessages || []));
               }, []);
               if (sessionsData.length > 0) {
-                formatMessages(sessionsData);
+                formatMessages(sessionsData, true);
               }
             }
-            _context5.next = 18;
+            _context6.next = 11;
             break;
-          case 17:
+          case 10:
             // 兼容 无 session异常
-            setCurrentSession(uuidV4());
-          case 18:
-            _context5.next = 23;
-            break;
-          case 20:
-            setSessionList([]);
-            setCurrentSession(uuidV4());
-            initPrologue();
-          case 23:
-            // 成功回调
-            onSuccess === null || onSuccess === void 0 || onSuccess({
-              code: SuccessCode.APP_MESSAGES_INIT_SUCCESS,
-              message: '初始化聊天记录成功'
+            setCurrentSessionId(uuidV4());
+          case 11:
+          case "end":
+            return _context6.stop();
+        }
+      }, _callee6);
+    }));
+    return function setCurrentSessionDetail(_x4) {
+      return _ref11.apply(this, arguments);
+    };
+  }();
+
+  // 获取更多会话记录
+  var loadMoreSessions = /*#__PURE__*/function () {
+    var _ref12 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
+      var pageNum, pageSize, result, _result$data5, _result$data6, _result$data7, _result$data8, resultData, formatSessions;
+      return _regeneratorRuntime().wrap(function _callee7$(_context7) {
+        while (1) switch (_context7.prev = _context7.next) {
+          case 0:
+            if (!sessionLoading) {
+              _context7.next = 2;
+              break;
+            }
+            return _context7.abrupt("return");
+          case 2:
+            _context7.prev = 2;
+            setSessionLoading(true);
+            pageNum = sessionPagination.pageNum, pageSize = sessionPagination.pageSize;
+            _context7.next = 7;
+            return fetchSessionList({
+              url: url,
+              appNo: appNo,
+              pt: pt,
+              tc: tc,
+              token: token,
+              pageNum: pageNum + 1,
+              pageSize: pageSize
             });
-            _context5.next = 27;
-            break;
-          case 26:
-            initPrologue();
-          case 27:
-            _context5.next = 31;
+          case 7:
+            result = _context7.sent;
+            if (result !== null && result !== void 0 && result.success) {
+              resultData = (result === null || result === void 0 || (_result$data5 = result.data) === null || _result$data5 === void 0 ? void 0 : _result$data5.data) || [];
+              if (resultData.length > 0) {
+                formatSessions = resultData.map(function (item) {
+                  return {
+                    key: item.sessionId,
+                    label: item.title,
+                    timestamp: new Date(item.gmtCreated).getTime(),
+                    messages: [],
+                    meta: _objectSpread({}, item)
+                  };
+                });
+                setSessionList(function (prevList) {
+                  return [].concat(_toConsumableArray(prevList), _toConsumableArray(formatSessions));
+                });
+              }
+              // 设置分页信息
+              setSessionPagination({
+                pageNum: result === null || result === void 0 || (_result$data6 = result.data) === null || _result$data6 === void 0 ? void 0 : _result$data6.pageNum,
+                pageSize: result === null || result === void 0 || (_result$data7 = result.data) === null || _result$data7 === void 0 ? void 0 : _result$data7.pageSize,
+                total: result === null || result === void 0 || (_result$data8 = result.data) === null || _result$data8 === void 0 ? void 0 : _result$data8.totalCount
+              });
+            }
+            _context7.next = 13;
             break;
-          case 29:
-            _context5.prev = 29;
-            _context5.t0 = _context5["catch"](0);
-          case 31:
+          case 11:
+            _context7.prev = 11;
+            _context7.t0 = _context7["catch"](2);
+          case 13:
+            _context7.prev = 13;
+            setSessionLoading(false);
+            return _context7.finish(13);
+          case 16:
           case "end":
-            return _context5.stop();
+            return _context7.stop();
         }
-      }, _callee5, null, [[0, 29]]);
+      }, _callee7, null, [[2, 11, 13, 16]]);
     }));
-    return function initAppConversations() {
-      return _ref9.apply(this, arguments);
+    return function loadMoreSessions() {
+      return _ref12.apply(this, arguments);
     };
   }();
 
@@ -747,14 +902,18 @@ var useAgentGenerator = function useAgentGenerator(_ref) {
     messages: messages,
     loading: loading,
     content: responseData.content,
-    sessionsList: sessionsList,
-    currentSession: currentSession,
+    sessionList: sessionList,
+    currentSessionId: currentSessionId,
+    sessionPagination: sessionPagination,
+    loadMoreSessions: loadMoreSessions,
     reChat: reChat,
     chat: chat,
-    setCurrentSession: setCurrentSession,
+    setCurrentSessionId: setCurrentSessionId,
+    setCurrentSessionDetail: setCurrentSessionDetail,
     setMessage: setMessage,
     setMessages: setMessages,
-    stopChat: stopChat
+    stopChat: stopChat,
+    initPrologue: initPrologue
   };
 };
 export default useAgentGenerator;