@ai-group/chat-sdk 2.1.1 → 2.1.4

package/dist/esm/hooks/useADKChat.js

@@ -9,7 +9,7 @@ import { fetchEventSource, EventStreamContentType } from '@microsoft/fetch-event
 import { message } from 'antd';
 import { v4 } from 'uuid';
 import { SESSION_PAGE_SIZE, API_SUCCESS_CODE } from "../constants";
-import { ErrorCode, SuccessCode } from "../types";
+import { SuccessCode, ErrorCode } from "../types";
 import { fetchAppConfig, fetchSessionList, fetchSessionDetail, closeSession, updateSessionTitle } from "../services/api";
 import { safeJsonParse } from "../utils";
 var combineTextParts = function combineTextParts(parts) {
@@ -51,7 +51,8 @@ function useADKChat(_ref) {
     type = _ref$type === void 0 ? 'agentDebug' : _ref$type,
     onError = _ref.onError,
     onMessage = _ref.onMessage,
-    onSuccess = _ref.onSuccess;
+    onSuccess = _ref.onSuccess,
+    onStream = _ref.onStream;
   var _useState = useState(false),
     _useState2 = _slicedToArray(_useState, 2),
     loading = _useState2[0],
@@ -61,53 +62,59 @@ function useADKChat(_ref) {
   var _ref2 = config || {},
     appNo = _ref2.appNo;
   // 智能体信息
-  var _useState3 = useState({
-      appNo: appNo,
-      appVersionNo: '',
-      appTheme: '',
-      appName: '',
-      onboardingInfo: {
-        prologue: '',
-        questionFiles: [],
-        suggested_questions: []
-      }
-    }),
+  // const [appInfo, setAppInfo] = useState<ChatConfig>({
+  //   appNo,
+  //   appVersionNo: '',
+  //   appTheme: '',
+  //   appName: '',
+  //   onboardingInfo: {
+  //     prologue: '',
+  //     questionFiles: [],
+  //     suggested_questions: [],
+  //   },
+  // });
+  var _useState3 = useState(null),
     _useState4 = _slicedToArray(_useState3, 2),
     appInfo = _useState4[0],
     setAppInfo = _useState4[1];
-  // 会话记录
-  var _useState5 = useState([]),
+  // 初始化完成
+  var _useState5 = useState(false),
     _useState6 = _slicedToArray(_useState5, 2),
-    sessionList = _useState6[0],
-    setSessionList = _useState6[1];
-  // 当前会话
-  var _useState7 = useState(''),
+    initialized = _useState6[0],
+    setInitialized = _useState6[1];
+  // 会话记录
+  var _useState7 = useState([]),
     _useState8 = _slicedToArray(_useState7, 2),
-    currentSessionId = _useState8[0],
-    setCurrentSessionId = _useState8[1];
+    sessionList = _useState8[0],
+    setSessionList = _useState8[1];
+  // 当前会话
+  var _useState9 = useState(''),
+    _useState10 = _slicedToArray(_useState9, 2),
+    currentSessionId = _useState10[0],
+    setCurrentSessionId = _useState10[1];
   // 会话分页设置
-  var _useState9 = useState({
+  var _useState11 = useState({
       total: 0,
       pageNum: 1,
       pageSize: SESSION_PAGE_SIZE
     }),
-    _useState10 = _slicedToArray(_useState9, 2),
-    sessionPagination = _useState10[0],
-    setSessionPagination = _useState10[1];
-  // 开场白
-  var _useState11 = useState(''),
     _useState12 = _slicedToArray(_useState11, 2),
-    prologue = _useState12[0],
-    setPrologue = _useState12[1];
-  // 建议问题
-  var _useState13 = useState([]),
+    sessionPagination = _useState12[0],
+    setSessionPagination = _useState12[1];
+  // 开场白
+  var _useState13 = useState(''),
     _useState14 = _slicedToArray(_useState13, 2),
-    suggestedQuestions = _useState14[0],
-    setSuggestedQuestions = _useState14[1];
+    prologue = _useState14[0],
+    setPrologue = _useState14[1];
+  // 建议问题
   var _useState15 = useState([]),
     _useState16 = _slicedToArray(_useState15, 2),
-    messages = _useState16[0],
-    setMessages = _useState16[1];
+    suggestedQuestions = _useState16[0],
+    setSuggestedQuestions = _useState16[1];
+  var _useState17 = useState([]),
+    _useState18 = _slicedToArray(_useState17, 2),
+    messages = _useState18[0],
+    setMessages = _useState18[1];
   var mergedMessages = useMemo(function () {
     var fnResMap = {};
     messages.forEach(function (msg) {
@@ -224,6 +231,8 @@ function useADKChat(_ref) {
         if (event !== null && event !== void 0 && (_event$groundingMetad = event.groundingMetadata) !== null && _event$groundingMetad !== void 0 && (_event$groundingMetad = _event$groundingMetad.searchEntryPoint) !== null && _event$groundingMetad !== void 0 && _event$groundingMetad.renderedContent) {
           msg.renderedContent = event.groundingMetadata.searchEntryPoint.renderedContent;
         }
+      } else if (part.fileData) {
+        msg.fileData = [part.fileData];
       } else if (part.functionCall) {
         msg.functionCall = part.functionCall;
       } else if (part.functionResponse) {
@@ -364,6 +373,7 @@ function useADKChat(_ref) {
             message.warning(chunkJson.error);
             return;
           }
+          onStream === null || onStream === void 0 || onStream(chunkJson);
           if (chunkJson.content) {
             // 建议类消息单独处理 
             if (chunkJson.content.role === 'followup') {
@@ -391,6 +401,10 @@ 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);
+          }
           textMsgRef.current = null;
           resolve();
         },
@@ -406,33 +420,63 @@ function useADKChat(_ref) {
 
   // 发送会话
   var startChat = /*#__PURE__*/function () {
-    var _ref3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(text) {
+    var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_ref3) {
+      var _ref3$text, text, _ref3$files, files, functionResponse;
       return _regeneratorRuntime().wrap(function _callee2$(_context2) {
         while (1) switch (_context2.prev = _context2.next) {
           case 0:
+            _ref3$text = _ref3.text, text = _ref3$text === void 0 ? '' : _ref3$text, _ref3$files = _ref3.files, files = _ref3$files === void 0 ? [] : _ref3$files, functionResponse = _ref3.functionResponse;
             if (!loading) {
-              _context2.next = 2;
+              _context2.next = 3;
               break;
             }
             return _context2.abrupt("return");
-          case 2:
-            if (text.trim()) {
-              _context2.next = 4;
+          case 3:
+            if (!(!text.trim() && !functionResponse)) {
+              _context2.next = 5;
               break;
             }
             return _context2.abrupt("return");
-          case 4:
+          case 5:
             // 清空建议问题
             setSuggestedQuestions([]);
             // 发送消息
             setMessages(function (prev) {
-              return [].concat(_toConsumableArray(prev), [{
-                id: v4(),
-                role: 'user',
-                text: text
-              }]);
+              var temp = _toConsumableArray(prev);
+              // 如果 text 存在
+              if (text.trim()) {
+                temp.push({
+                  id: v4(),
+                  role: 'user',
+                  text: text
+                });
+              }
+              // 如果file 存在
+              if (files.length > 0) {
+                temp.push({
+                  id: v4(),
+                  role: 'user',
+                  fileData: files.map(function (file) {
+                    var _file$response, _file$response2;
+                    return {
+                      displayName: file.name,
+                      mimeType: (_file$response = file.response) === null || _file$response === void 0 ? void 0 : _file$response.mimeType,
+                      fileUri: (_file$response2 = file.response) === null || _file$response2 === void 0 ? void 0 : _file$response2.fileUrl
+                    };
+                  })
+                });
+              }
+              // 如果 functionResponse 存在
+              if (functionResponse) {
+                temp.push({
+                  id: v4(),
+                  role: 'user',
+                  functionResponse: functionResponse
+                });
+              }
+              return temp;
             });
-            _context2.next = 8;
+            _context2.next = 9;
             return sendMessageRequest({
               stream: true,
               sessionId: currentSessionId,
@@ -441,27 +485,100 @@ function useADKChat(_ref) {
                 parts: [{
                   text: text
                 }]
-              }
+              },
+              files: files.map(function (file) {
+                var _file$response3, _file$response4, _file$response5, _file$response6, _file$response7;
+                return {
+                  fileName: (_file$response3 = file.response) === null || _file$response3 === void 0 ? void 0 : _file$response3.fileName,
+                  fileId: (_file$response4 = file.response) === null || _file$response4 === void 0 ? void 0 : _file$response4.fileId,
+                  tempUrl: (_file$response5 = file.response) === null || _file$response5 === void 0 ? void 0 : _file$response5.fileUrl,
+                  type: (_file$response6 = file.response) === null || _file$response6 === void 0 ? void 0 : _file$response6.fileType,
+                  mimeType: (_file$response7 = file.response) === null || _file$response7 === void 0 ? void 0 : _file$response7.mimeType
+                };
+              })
             });
-          case 8:
+          case 9:
             // 获取聊天记录
             initAppConversations();
-          case 9:
+          case 10:
           case "end":
             return _context2.stop();
         }
       }, _callee2);
     }));
     return function startChat(_x) {
-      return _ref3.apply(this, arguments);
+      return _ref4.apply(this, arguments);
     };
   }();
+
+  // 重新发送会话
+  var reChat = function reChat() {
+    if (loading) return;
+    if (messages.length === 0) return;
+    setMessages(function (prev) {
+      var newMessages = _toConsumableArray(prev);
+      // 删除最后一条 AI 消息
+      var lastAIIndex = newMessages.findLastIndex(function (msg) {
+        return msg.role === 'bot';
+      });
+      if (lastAIIndex !== -1) {
+        newMessages.splice(lastAIIndex, 1);
+      }
+      // 清空建议问题
+      setSuggestedQuestions([]);
+      // 重新发送最后一条用户消息
+      var lastUserMessage = newMessages.findLast(function (msg) {
+        return msg.role === 'user';
+      });
+      if (lastUserMessage) {
+        var _lastUserMessage$text = lastUserMessage.text,
+          text = _lastUserMessage$text === void 0 ? '' : _lastUserMessage$text,
+          _lastUserMessage$file = lastUserMessage.fileData,
+          fileData = _lastUserMessage$file === void 0 ? [] : _lastUserMessage$file;
+        sendMessageRequest({
+          stream: true,
+          sessionId: currentSessionId,
+          content: {
+            role: 'user',
+            parts: [{
+              text: text
+            }]
+          },
+          files: fileData === null || fileData === void 0 ? void 0 : fileData.map(function (file) {
+            return {
+              fileName: file.displayName,
+              tempUrl: file.fileUri,
+              mimeType: file.mimeType
+            };
+          })
+        });
+      }
+      return newMessages;
+    });
+  };
+
+  // 确认函数调用
+  var confirmFnCall = function confirmFnCall(fnCall, confirmed) {
+    var functionResponse = {
+      id: fnCall.id,
+      name: fnCall.name,
+      response: {
+        confirmed: confirmed
+      }
+    };
+    startChat({
+      functionResponse: functionResponse
+    });
+  };
+
   // 使用建议问题
   var suggestChat = function suggestChat(text) {
     if (loading) return;
     if (!text.trim()) return;
     setSuggestedQuestions([]);
-    startChat(text);
+    startChat({
+      text: text
+    });
   };
   // 停止会话
   var stopChat = useCallback(function () {
@@ -473,12 +590,13 @@ function useADKChat(_ref) {
 
   // 清除会话
   var clearChat = function clearChat() {
-    setCurrentSessionId(v4());
+    var _appInfo$onboardingIn;
+    var newSessionId = v4();
+    setCurrentSessionId(newSessionId);
     // 重置聊天内容
-    setTimeout(function () {
-      stopChat();
-      setMessages([]);
-    });
+    stopChat();
+    setMessages([]);
+    setSuggestedQuestions((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn = appInfo.onboardingInfo) === null || _appInfo$onboardingIn === void 0 ? void 0 : _appInfo$onboardingIn.suggested_questions) || []);
   };
 
   // 格式化会话记录
@@ -536,8 +654,8 @@ function useADKChat(_ref) {
 
   // 设置当前会话详情
   var setCurrentSessionDetail = /*#__PURE__*/function () {
-    var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(sessionId) {
-      var _yield$fetchSessionDe, _data, result, _appInfo$onboardingIn, _appInfo$onboardingIn2;
+    var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(sessionId) {
+      var _yield$fetchSessionDe, data, result, _appInfo$onboardingIn2, _appInfo$onboardingIn3;
       return _regeneratorRuntime().wrap(function _callee3$(_context3) {
         while (1) switch (_context3.prev = _context3.next) {
           case 0:
@@ -561,15 +679,15 @@ function useADKChat(_ref) {
             });
           case 6:
             _yield$fetchSessionDe = _context3.sent;
-            _data = _yield$fetchSessionDe.data;
+            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$onboardingIn2 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn2 === void 0 ? void 0 : _appInfo$onboardingIn2.prologue) || '');
               // 新接口直接返回按 event/item 结构的数组，传入 formatMessages 处理
-              if (Array.isArray(_data) && _data.length > 0) {
-                formatMessages(_data, true);
-                setPrologue((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn = appInfo.onboardingInfo) === null || _appInfo$onboardingIn === void 0 ? void 0 : _appInfo$onboardingIn.prologue) || '');
+              if (Array.isArray(data) && data.length > 0) {
+                formatMessages(data, true);
               } else {
-                setSuggestedQuestions((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn2 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn2 === void 0 ? void 0 : _appInfo$onboardingIn2.suggested_questions) || []);
+                setSuggestedQuestions((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn3 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn3 === void 0 ? void 0 : _appInfo$onboardingIn3.suggested_questions) || []);
               }
             }
             _context3.next = 13;
@@ -584,21 +702,21 @@ function useADKChat(_ref) {
       }, _callee3);
     }));
     return function setCurrentSessionDetail(_x2) {
-      return _ref4.apply(this, arguments);
+      return _ref5.apply(this, arguments);
     };
   }();
 
   // 获取聊天记录
   var initAppConversations = /*#__PURE__*/function () {
-    var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4() {
+    var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4() {
       var fetchDetail,
         _yield$fetchSessionLi,
         _yield$fetchSessionLi2,
-        _data2,
+        data,
         pagination,
         result,
         formatSessions,
-        _data2$,
+        _data$,
         sessionId,
         _args4 = arguments;
       return _regeneratorRuntime().wrap(function _callee4$(_context4) {
@@ -617,16 +735,16 @@ function useADKChat(_ref) {
           case 5:
             _yield$fetchSessionLi = _context4.sent;
             _yield$fetchSessionLi2 = _yield$fetchSessionLi.data;
-            _data2 = _yield$fetchSessionLi2 === void 0 ? [] : _yield$fetchSessionLi2;
+            data = _yield$fetchSessionLi2 === void 0 ? [] : _yield$fetchSessionLi2;
             pagination = _yield$fetchSessionLi.pagination;
             result = _yield$fetchSessionLi.result;
             if ((result === null || result === void 0 ? void 0 : result.code) === API_SUCCESS_CODE) {
-              if (_data2.length > 0) {
-                formatSessions = _data2.map(function (item) {
+              if (data.length > 0) {
+                formatSessions = data.map(function (item) {
                   return {
                     key: item.sessionId,
                     label: item.title || '新对话',
-                    timestamp: new Date(item.sessionTime).getTime(),
+                    timestamp: new Date(item.startTime).getTime(),
                     messages: [],
                     meta: _objectSpread({}, item)
                   };
@@ -638,8 +756,16 @@ function useADKChat(_ref) {
                   pageSize: sessionPagination.pageSize,
                   total: (pagination === null || pagination === void 0 ? void 0 : pagination.total) || 0
                 });
-                _data2$ = _data2 === null || _data2 === void 0 ? void 0 : _data2[0], sessionId = _data2$.sessionId;
+                _data$ = data === null || data === void 0 ? void 0 : data[0], sessionId = _data$.sessionId;
                 fetchDetail && setCurrentSessionDetail(sessionId);
+                // 成功回调
+                if (!initialized) {
+                  onSuccess === null || onSuccess === void 0 || onSuccess({
+                    code: SuccessCode.APP_MESSAGES_INIT_SUCCESS,
+                    message: '初始化聊天记录成功'
+                  });
+                  setInitialized(true);
+                }
               } else {
                 // 页面报错 初始化
                 setSessionList([]);
@@ -647,11 +773,6 @@ function useADKChat(_ref) {
                 setPrologue('');
                 setSuggestedQuestions([]);
               }
-              // 成功回调
-              onSuccess === null || onSuccess === void 0 || onSuccess({
-                code: SuccessCode.APP_MESSAGES_INIT_SUCCESS,
-                message: '初始化聊天记录成功'
-              });
             } else {
               setSessionList([]);
               setCurrentSessionId(v4());
@@ -674,13 +795,13 @@ function useADKChat(_ref) {
       }, _callee4, null, [[1, 13, 15, 18]]);
     }));
     return function initAppConversations() {
-      return _ref5.apply(this, arguments);
+      return _ref6.apply(this, arguments);
     };
   }();
 
   // 删除会话
   var deleteSession = /*#__PURE__*/function () {
-    var _ref6 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5(sessionId) {
+    var _ref7 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5(sessionId) {
       var _yield$closeSession, result;
       return _regeneratorRuntime().wrap(function _callee5$(_context5) {
         while (1) switch (_context5.prev = _context5.next) {
@@ -716,13 +837,13 @@ function useADKChat(_ref) {
       }, _callee5);
     }));
     return function deleteSession(_x3) {
-      return _ref6.apply(this, arguments);
+      return _ref7.apply(this, arguments);
     };
   }();
 
   // 更新会话名称
   var updateSession = /*#__PURE__*/function () {
-    var _ref7 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(sessionId, title) {
+    var _ref8 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee6(sessionId, title) {
       var _yield$updateSessionT, result;
       return _regeneratorRuntime().wrap(function _callee6$(_context6) {
         while (1) switch (_context6.prev = _context6.next) {
@@ -755,14 +876,14 @@ function useADKChat(_ref) {
       }, _callee6);
     }));
     return function updateSession(_x4, _x5) {
-      return _ref7.apply(this, arguments);
+      return _ref8.apply(this, arguments);
     };
   }();
 
   // 初始化应用
   var initConfig = /*#__PURE__*/function () {
-    var _ref8 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
-      var _yield$fetchAppConfig, _data3, result;
+    var _ref9 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee7() {
+      var _yield$fetchAppConfig, data, result;
       return _regeneratorRuntime().wrap(function _callee7$(_context7) {
         while (1) switch (_context7.prev = _context7.next) {
           case 0:
@@ -775,10 +896,10 @@ function useADKChat(_ref) {
             });
           case 4:
             _yield$fetchAppConfig = _context7.sent;
-            _data3 = _yield$fetchAppConfig.data;
+            data = _yield$fetchAppConfig.data;
             result = _yield$fetchAppConfig.result;
             if (result.code === API_SUCCESS_CODE) {
-              setAppInfo(_data3);
+              setAppInfo(data);
               // 成功回调
               onSuccess === null || onSuccess === void 0 || onSuccess({
                 code: SuccessCode.APP_CONFIG_INIT_SUCCESS,
@@ -810,7 +931,7 @@ function useADKChat(_ref) {
       }, _callee7, null, [[0, 10, 13, 16]]);
     }));
     return function initConfig() {
-      return _ref8.apply(this, arguments);
+      return _ref9.apply(this, arguments);
     };
   }();
 
@@ -831,9 +952,12 @@ function useADKChat(_ref) {
   return {
     appInfo: appInfo,
     startChat: startChat,
+    reChat: reChat,
     stopChat: stopChat,
     clearChat: clearChat,
     suggestChat: suggestChat,
+    confirmFnCall: confirmFnCall,
+    initialized: initialized,
     currentSessionId: currentSessionId,
     sessionList: sessionList,
     prologue: prologue,