@ai-group/chat-sdk 2.1.13 → 2.1.14

package/dist/esm/hooks/useADKChat.js

@@ -4,10 +4,10 @@ import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
 import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
 import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
 import _createForOfIteratorHelper from "@babel/runtime/helpers/esm/createForOfIteratorHelper";
-import { useEffect, useCallback, useMemo, useRef, useState } from 'react';
-import { fetchEventSource, EventStreamContentType } from '@microsoft/fetch-event-source';
-import { message } from 'antd';
-import { v4 } from 'uuid';
+import { useEffect, useCallback, useMemo, useRef, useState } from "react";
+import { fetchEventSource, EventStreamContentType } from "@microsoft/fetch-event-source";
+import { message } from "antd";
+import { v4 } from "uuid";
 import { SESSION_PAGE_SIZE, API_SUCCESS_CODE } from "../constants";
 import { SuccessCode, ErrorCode } from "../types";
 import { fetchAppConfig, fetchSessionList, fetchSessionDetail, closeSession, updateSessionTitle } from "../services/api";
@@ -48,7 +48,7 @@ function useADKChat(_ref) {
     _ref$config = _ref.config,
     config = _ref$config === void 0 ? {} : _ref$config,
     _ref$type = _ref.type,
-    type = _ref$type === void 0 ? 'agentDebug' : _ref$type,
+    type = _ref$type === void 0 ? "agentDebug" : _ref$type,
     onError = _ref.onError,
     onMessage = _ref.onMessage,
     onSuccess = _ref.onSuccess,
@@ -77,7 +77,9 @@ function useADKChat(_ref) {
     sessionList = _useState8[0],
     setSessionList = _useState8[1];
   // 当前会话
-  var _useState9 = useState(''),
+  var _useState9 = useState(function () {
+      return v4();
+    }),
     _useState10 = _slicedToArray(_useState9, 2),
     currentSessionId = _useState10[0],
     setCurrentSessionId = _useState10[1];
@@ -91,7 +93,7 @@ function useADKChat(_ref) {
     sessionPagination = _useState12[0],
     setSessionPagination = _useState12[1];
   // 开场白
-  var _useState13 = useState(''),
+  var _useState13 = useState(""),
     _useState14 = _slicedToArray(_useState13, 2),
     prologue = _useState14[0],
     setPrologue = _useState14[1];
@@ -109,14 +111,14 @@ function useADKChat(_ref) {
     messages.forEach(function (msg) {
       if (msg.functionResponse) {
         var _msg$functionResponse;
-        fnResMap[(msg === null || msg === void 0 || (_msg$functionResponse = msg.functionResponse) === null || _msg$functionResponse === void 0 ? void 0 : _msg$functionResponse.id) || ''] = msg.functionResponse;
+        fnResMap[(msg === null || msg === void 0 || (_msg$functionResponse = msg.functionResponse) === null || _msg$functionResponse === void 0 ? void 0 : _msg$functionResponse.id) || ""] = msg.functionResponse;
       }
     });
     return messages.map(function (msg) {
       if (msg.functionCall) {
         var _msg$functionCall;
         return _objectSpread(_objectSpread({}, msg), {}, {
-          functionResponse: fnResMap[(msg === null || msg === void 0 || (_msg$functionCall = msg.functionCall) === null || _msg$functionCall === void 0 ? void 0 : _msg$functionCall.id) || '']
+          functionResponse: fnResMap[(msg === null || msg === void 0 || (_msg$functionCall = msg.functionCall) === null || _msg$functionCall === void 0 ? void 0 : _msg$functionCall.id) || ""]
         });
       }
       return msg;
@@ -144,7 +146,6 @@ function useADKChat(_ref) {
     if (messages.length === 0 && suggested_questions.length) {
       setSuggestedQuestions(suggested_questions);
     }
-    ;
   };
 
   // 初始化开场白
@@ -169,9 +170,9 @@ function useADKChat(_ref) {
   // 存储事件
   var storeEvents = useCallback(function (part, event) {
     var _event$actions;
-    var title = '';
+    var title = "";
     if (part == null && (_event$actions = event.actions) !== null && _event$actions !== void 0 && _event$actions.artifactDelta) {
-      title += 'eventAction: artifact';
+      title += "eventAction: artifact";
     } else if (part) {
       if (part.text) {
         title += "text:".concat(part.text);
@@ -250,7 +251,7 @@ function useADKChat(_ref) {
           invocationId: event.invocationId,
           eventId: event.id,
           timestamp: event.timestamp,
-          role: 'bot',
+          role: "bot",
           text: newChunk,
           thought: part.thought
         };
@@ -274,14 +275,14 @@ function useADKChat(_ref) {
           });
         }
         textMsgRef.current = _objectSpread(_objectSpread({}, textMsgRef.current), {}, {
-          text: (textMsgRef.current.text || '') + newChunk
+          text: (textMsgRef.current.text || "") + newChunk
         });
         updateMessage(textMsgRef.current);
       }
     } else {
       textMsgRef.current = null;
       storeEvents(part, event);
-      storeMessage(part, event, 'bot');
+      storeMessage(part, event, "bot");
     }
   }, [storeEvents, storeMessage, insertMessage, updateMessage]);
 
@@ -290,7 +291,7 @@ function useADKChat(_ref) {
     var _event$content;
     // 获取建议类消息的文本内容
     var part = (event === null || event === void 0 || (_event$content = event.content) === null || _event$content === void 0 || (_event$content = _event$content.parts) === null || _event$content === void 0 ? void 0 : _event$content[0]) || null;
-    var text = (part === null || part === void 0 ? void 0 : part.text) || '';
+    var text = (part === null || part === void 0 ? void 0 : part.text) || "";
     storeEvents(part, event);
     if (text) {
       setSuggestedQuestions(function (prev) {
@@ -305,7 +306,7 @@ function useADKChat(_ref) {
     insertMessage({
       id: v4(),
       text: chunkJson.errorMessage,
-      role: 'bot'
+      role: "bot"
     });
   }, [storeEvents, insertMessage]);
 
@@ -313,7 +314,7 @@ function useADKChat(_ref) {
   var processActionArtifact = useCallback(function (event) {
     if (event.actions && event.actions.artifactDelta) {
       storeEvents(null, event);
-      storeMessage(null, event, 'bot');
+      storeMessage(null, event, "bot");
     }
   }, [storeEvents, storeMessage]);
 
@@ -327,14 +328,14 @@ function useADKChat(_ref) {
       ctrl.current = new AbortController();
       var requestUrl = "".concat(url, "/api/gateway-web/openApi/v1/aizt/app/").concat(appNo, "/sendMsgStreaming");
       fetchEventSource(requestUrl, {
-        method: 'POST',
+        method: "POST",
         signal: ctrl.current.signal,
         body: JSON.stringify(values),
         openWhenHidden: true,
         headers: {
-          'Content-Type': 'application/json',
-          'Cache-Control': 'no-cache',
-          stream: 'true',
+          "Content-Type": "application/json",
+          "Cache-Control": "no-cache",
+          stream: "true",
           Authorization: "Bearer ".concat(token)
         },
         onopen: function onopen(response) {
@@ -342,7 +343,7 @@ function useADKChat(_ref) {
             return _regeneratorRuntime().wrap(function _callee$(_context) {
               while (1) switch (_context.prev = _context.next) {
                 case 0:
-                  if (response.ok && response.headers.get('content-type') === EventStreamContentType) {
+                  if (response.ok && response.headers.get("content-type") === EventStreamContentType) {
                     // everything's good
                   } else if (response.status >= 400 && response.status < 500 && response.status !== 429) {
                     resolve();
@@ -364,8 +365,8 @@ function useADKChat(_ref) {
           }
           onStream === null || onStream === void 0 || onStream(chunkJson);
           if (chunkJson.content) {
-            // 建议类消息单独处理 
-            if (chunkJson.content.role === 'followup') {
+            // 建议类消息单独处理
+            if (chunkJson.content.role === "followup") {
               processFollowupPart(chunkJson);
               return;
             }
@@ -392,7 +393,7 @@ function useADKChat(_ref) {
           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);
+            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();
@@ -400,7 +401,7 @@ function useADKChat(_ref) {
         onerror: function onerror(error) {
           setLoading(false);
           resolve();
-          console.error('EventSource failed:', error);
+          console.error("EventSource failed:", error);
           throw error;
         }
       });
@@ -414,19 +415,20 @@ function useADKChat(_ref) {
       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;
+            _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;
+            console.log(text, loading);
             if (!loading) {
-              _context2.next = 3;
+              _context2.next = 4;
               break;
             }
             return _context2.abrupt("return");
-          case 3:
+          case 4:
             if (!(!text.trim() && !functionResponse)) {
-              _context2.next = 5;
+              _context2.next = 6;
               break;
             }
             return _context2.abrupt("return");
-          case 5:
+          case 6:
             // 清空建议问题
             setSuggestedQuestions([]);
             // 发送消息
@@ -436,7 +438,7 @@ function useADKChat(_ref) {
               if (text.trim()) {
                 temp.push({
                   id: v4(),
-                  role: 'user',
+                  role: "user",
                   text: text
                 });
               }
@@ -444,7 +446,7 @@ function useADKChat(_ref) {
               if (files.length > 0) {
                 temp.push({
                   id: v4(),
-                  role: 'user',
+                  role: "user",
                   fileData: files.map(function (file) {
                     var _file$response, _file$response2;
                     return {
@@ -459,18 +461,18 @@ function useADKChat(_ref) {
               if (functionResponse) {
                 temp.push({
                   id: v4(),
-                  role: 'user',
+                  role: "user",
                   functionResponse: functionResponse
                 });
               }
               return temp;
             });
-            _context2.next = 9;
+            _context2.next = 10;
             return sendMessageRequest({
               stream: true,
               sessionId: currentSessionId,
               content: {
-                role: 'user',
+                role: "user",
                 parts: functionResponse ? [{
                   functionResponse: functionResponse
                 }] : [{
@@ -488,10 +490,10 @@ function useADKChat(_ref) {
                 };
               })
             });
-          case 9:
+          case 10:
             // 获取聊天记录
             initAppConversations();
-          case 10:
+          case 11:
           case "end":
             return _context2.stop();
         }
@@ -510,7 +512,7 @@ function useADKChat(_ref) {
       var newMessages = _toConsumableArray(prev);
       // 删除最后一条 AI 消息
       var lastAIIndex = newMessages.findLastIndex(function (msg) {
-        return msg.role === 'bot';
+        return msg.role === "bot";
       });
       if (lastAIIndex !== -1) {
         newMessages.splice(lastAIIndex, 1);
@@ -519,18 +521,18 @@ function useADKChat(_ref) {
       setSuggestedQuestions([]);
       // 重新发送最后一条用户消息
       var lastUserMessage = newMessages.findLast(function (msg) {
-        return msg.role === 'user';
+        return msg.role === "user";
       });
       if (lastUserMessage) {
         var _lastUserMessage$text = lastUserMessage.text,
-          text = _lastUserMessage$text === void 0 ? '' : _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',
+            role: "user",
             parts: [{
               text: text
             }]
@@ -586,7 +588,7 @@ function useADKChat(_ref) {
     setCurrentSessionId(newSessionId);
     // 重置聊天内容
     stopChat();
-    setPrologue((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn = appInfo.onboardingInfo) === null || _appInfo$onboardingIn === void 0 ? void 0 : _appInfo$onboardingIn.prologue) || '');
+    setPrologue((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn = appInfo.onboardingInfo) === null || _appInfo$onboardingIn === void 0 ? void 0 : _appInfo$onboardingIn.prologue) || "");
     setMessages([]);
     setSuggestedQuestions((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn2 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn2 === void 0 ? void 0 : _appInfo$onboardingIn2.suggested_questions) || []);
   };
@@ -597,7 +599,7 @@ function useADKChat(_ref) {
     var mapped = [];
     messages.forEach(function (item) {
       if (!item || !item.content || !Array.isArray(item.content.parts)) return;
-      var role = (item.content.role || '').toLowerCase() === 'user' ? 'user' : 'bot';
+      var role = (item.content.role || "").toLowerCase() === "user" ? "user" : "bot";
       var parts = Array.isArray(item.content.parts) ? item.content.parts.filter(function (p) {
         if (!p) return false;
         return Boolean(p.text || p.inlineData || p.functionCall || p.functionResponse || p.fileData || p.executableCode || p.codeExecutionResult || p.errorMessage);
@@ -622,12 +624,12 @@ function useADKChat(_ref) {
         }
         if (part.text) {
           msg.text = part.text;
-          if (typeof part.thought !== 'undefined') msg.thought = part.thought;
+          if (typeof part.thought !== "undefined") msg.thought = part.thought;
         }
         if (part.functionCall) msg.functionCall = part.functionCall;
         if (part.fileData) {
           msg.inlineData = {
-            displayName: part.fileData.displayName || '',
+            displayName: part.fileData.displayName || "",
             data: part.fileData.fileUri,
             mimeType: part.fileData.mimeType
           };
@@ -685,7 +687,7 @@ function useADKChat(_ref) {
             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) || '');
+              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);
@@ -748,7 +750,7 @@ function useADKChat(_ref) {
                 formatSessions = data.map(function (item) {
                   return {
                     key: item.sessionId,
-                    label: item.title || '新对话',
+                    label: item.title || "新对话",
                     timestamp: new Date(item.startTime).getTime(),
                     messages: [],
                     meta: _objectSpread({}, item)
@@ -765,14 +767,14 @@ function useADKChat(_ref) {
                 if (showFirstSession) {
                   fetchDetail && setCurrentSessionDetail(sessionId);
                 } else {
-                  setPrologue((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn5 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn5 === void 0 ? void 0 : _appInfo$onboardingIn5.prologue) || '');
+                  setPrologue((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn5 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn5 === void 0 ? void 0 : _appInfo$onboardingIn5.prologue) || "");
                   setSuggestedQuestions((appInfo === null || appInfo === void 0 || (_appInfo$onboardingIn6 = appInfo.onboardingInfo) === null || _appInfo$onboardingIn6 === void 0 ? void 0 : _appInfo$onboardingIn6.suggested_questions) || []);
                 }
                 // 成功回调
                 if (!initialized) {
                   onSuccess === null || onSuccess === void 0 || onSuccess({
                     code: SuccessCode.APP_MESSAGES_INIT_SUCCESS,
-                    message: '初始化聊天记录成功',
+                    message: "初始化聊天记录成功",
                     data: {
                       sessionId: sessionId
                     }
@@ -786,7 +788,7 @@ function useADKChat(_ref) {
             } else {
               setSessionList([]);
               setCurrentSessionId(v4());
-              setPrologue('');
+              setPrologue("");
               setSuggestedQuestions([]);
             }
             _context4.next = 15;
@@ -840,7 +842,7 @@ function useADKChat(_ref) {
                 clearChat();
               }
             } else {
-              message.error('删除会话失败');
+              message.error("删除会话失败");
             }
           case 7:
           case "end":
@@ -881,7 +883,7 @@ function useADKChat(_ref) {
               // 编辑成功后，重新获取会话列表
               initAppConversations();
             } else {
-              message.error('更新会话失败');
+              message.error("更新会话失败");
             }
           case 7:
           case "end":
@@ -917,12 +919,12 @@ function useADKChat(_ref) {
               // 成功回调
               onSuccess === null || onSuccess === void 0 || onSuccess({
                 code: SuccessCode.APP_CONFIG_INIT_SUCCESS,
-                message: '初始化配置成功'
+                message: "初始化配置成功"
               });
             } else {
               onError === null || onError === void 0 || onError({
                 code: result.code || ErrorCode.APP_NOT_FOUND,
-                message: result.message || '获取应用配置失败'
+                message: result.message || "获取应用配置失败"
               });
             }
             _context7.next = 13;
@@ -932,7 +934,7 @@ function useADKChat(_ref) {
             _context7.t0 = _context7["catch"](0);
             onError === null || onError === void 0 || onError({
               code: ErrorCode.API_ERROR,
-              message: '获取应用配置失败'
+              message: "获取应用配置失败"
             });
           case 13:
             _context7.prev = 13;