@ai-group/chat-sdk 1.0.9 → 1.0.11

package/dist/cjs/hooks/useAgentGenerator.js

@@ -44,7 +44,10 @@ var import_request = require("../utils/request");
 var import_utils = require("../utils/index");
 var import_constants = require("../constants");
 var import_api = require("../services/api");
+var import_workflowNode = require("../utils/workflowNode");
 var import_document = __toESM(require("../assets/document.png"));
+var import_completed = __toESM(require("../assets/completed.png"));
+var import_completed_black = __toESM(require("../assets/completed-black.png"));
 var import_document_black = __toESM(require("../assets/document-black.png"));
 var import_qa = __toESM(require("../assets/qa.png"));
 var import_qa_black = __toESM(require("../assets/qa-black.png"));
@@ -58,7 +61,11 @@ var ActionType = /* @__PURE__ */ ((ActionType2) => {
   ActionType2["invokeToolStart"] = "INVOKE_AGENT_TOOL_START";
   ActionType2["invokeToolEnd"] = "INVOKE_AGENT_TOOL_END";
   ActionType2["reasoning"] = "REASONING";
+  ActionType2["reasoningStart"] = "REASONING_START";
   ActionType2["reasoningEnd"] = "REASONING_END";
+  ActionType2["componentStart"] = "COMPONENT_START";
+  ActionType2["componentStream"] = "COMPONENT_STREAM";
+  ActionType2["componentEnd"] = "COMPONENT_END";
   ActionType2["suggest"] = "FOLLOW_UP";
   ActionType2["response"] = "RESPONSE";
   ActionType2["finish"] = "FINISHED";
@@ -75,14 +82,303 @@ var ActionAvailability = /* @__PURE__ */ ((ActionAvailability2) => {
   ActionAvailability2["remote"] = "remote";
   return ActionAvailability2;
 })(ActionAvailability || {});
-var useAgentGenerator = ({
-  url,
-  token,
-  config,
-  onError,
-  onMessage,
-  onSuccess
-}) => {
+var MessageProcessor = class {
+  constructor(textSpeed = 60) {
+    this.textSpeed = textSpeed;
+  }
+  // 获取知识图标
+  getKnowledgeIcons(knowledgeType) {
+    return knowledgeType === "DOCUMENT" ? { icon: import_document.default, expandIcon: import_document_black.default } : { icon: import_qa.default, expandIcon: import_qa_black.default };
+  }
+  // 获取工具图标
+  getToolIcons(toolType) {
+    return toolType === "SKILL" ? { icon: import_skillNo.default, expandIcon: import_skillNo_black.default } : { icon: import_tools.default, expandIcon: import_tools_black.default };
+  }
+  // 获取工作流图标(暂时不做区分)
+  getWorkflowIcon() {
+    return { icon: import_skillNo.default, expandIcon: import_skillNo_black.default };
+  }
+  // 更新执行时间
+  updateExecutionCost(messages, messageIndex, uniqueId, cost) {
+    const msg = messages[messageIndex];
+    if (msg == null ? void 0 : msg.execute) {
+      const execution = msg.execute.find((item) => item.id === uniqueId);
+      if (execution) {
+        execution.extra.cost = cost ? (cost / 1e3).toFixed(2) : 0;
+      }
+    }
+  }
+  // 查找最后一个待处理的AI消息
+  findLastPendingAIMessage(messages) {
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const msg = messages[i];
+      if (msg.role === import_XAiMessage.MessageRole.assistant && msg.type === "TextMessage" && (msg.status === "init" || msg.status === "pending")) {
+        return [i, msg];
+      }
+    }
+    return [-1, null];
+  }
+  // 创建新的AI消息
+  createNewAIMessage() {
+    return {
+      id: (0, import_uuid.v4)(),
+      type: "TextMessage",
+      createdAt: /* @__PURE__ */ new Date(),
+      role: import_XAiMessage.MessageRole.assistant,
+      content: { text: "" },
+      execute: [],
+      thinks: "",
+      status: import_XAiMessage.MessageStatus.init
+    };
+  }
+  // 确保存在待处理的AI消息
+  ensureAIMessageExists(messages) {
+    const [index, existingMsg] = this.findLastPendingAIMessage(messages);
+    if (index !== -1 && existingMsg) {
+      return [[...messages], index, { ...existingMsg }];
+    }
+    const newMessage = this.createNewAIMessage();
+    const newMessages = [...messages, newMessage];
+    return [newMessages, newMessages.length - 1, newMessage];
+  }
+  // 处理建议
+  handleSuggestion(messages, data) {
+    var _a;
+    const { content } = data;
+    const newMessages = [...messages];
+    const suggestIndex = newMessages.findIndex((item) => item.type === "SuggestionMessage");
+    if (suggestIndex === -1) {
+      const newSuggestion = {
+        id: (0, import_uuid.v4)(),
+        type: "SuggestionMessage",
+        createdAt: /* @__PURE__ */ new Date(),
+        status: import_XAiMessage.MessageStatus.done,
+        role: import_XAiMessage.MessageRole.suggestion,
+        content: [{ key: (0, import_uuid.v4)(), description: content }]
+      };
+      newMessages.push(newSuggestion);
+    } else {
+      const suggestion = newMessages[suggestIndex];
+      (_a = suggestion == null ? void 0 : suggestion.content) == null ? void 0 : _a.push({
+        key: (0, import_uuid.v4)(),
+        description: content
+      });
+    }
+    return newMessages;
+  }
+  // 处理思考过程
+  handleReasoning(messages, data) {
+    const { content } = data;
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const updatedMsg = {
+      ...aiMsg,
+      thinks: (aiMsg.thinks || "") + content,
+      status: aiMsg.status === import_XAiMessage.MessageStatus.init ? import_XAiMessage.MessageStatus.pending : aiMsg.status
+    };
+    const updatedMessages = [...newMessages];
+    updatedMessages[index] = updatedMsg;
+    return updatedMessages;
+  }
+  // 处理思考过程结束
+  handleReasoningEnd(messages) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const updatedMsg = {
+      ...aiMsg,
+      thinks: (aiMsg.thinks || "") + "\n",
+      status: aiMsg.status === import_XAiMessage.MessageStatus.init ? import_XAiMessage.MessageStatus.pending : aiMsg.status
+    };
+    const updatedMessages = [...newMessages];
+    updatedMessages[index] = updatedMsg;
+    return updatedMessages;
+  }
+  // 处理知识库调用开始
+  handleRecallKnowledgeStart(messages, data) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { uniqueId = (0, import_uuid.v4)(), args, knowledgeType } = (data == null ? void 0 : data.data) || {};
+    aiMsg.execute = aiMsg.execute || [];
+    const icons = this.getKnowledgeIcons(knowledgeType);
+    aiMsg.execute.push({
+      id: uniqueId,
+      createdAt: /* @__PURE__ */ new Date(),
+      type: "ActionExecutionMessage",
+      name: knowledgeType === "DOCUMENT" ? "已搜索文档知识库" : "已搜索问答知识库",
+      arguments: args,
+      extra: {
+        ...data,
+        ...icons
+      }
+    });
+    newMessages[index] = aiMsg;
+    return newMessages;
+  }
+  // 处理知识库调用结束
+  handleRecallKnowledgeEnd(messages, data) {
+    var _a;
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    this.updateExecutionCost(newMessages, index, (_a = data == null ? void 0 : data.data) == null ? void 0 : _a.uniqueId, data.cost);
+    return newMessages;
+  }
+  // 处理工具调用开始
+  handleInvokeToolStart(messages, data) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { uniqueId = (0, import_uuid.v4)(), args, toolType } = (data == null ? void 0 : data.data) || {};
+    aiMsg.execute = aiMsg.execute || [];
+    const icons = this.getToolIcons(toolType);
+    aiMsg.execute.push({
+      id: uniqueId,
+      createdAt: /* @__PURE__ */ new Date(),
+      type: "ActionExecutionMessage",
+      name: toolType === "SKILL" ? "已调用工作流" : "已调用工具",
+      arguments: args,
+      extra: {
+        ...data,
+        ...icons
+      }
+    });
+    newMessages[index] = aiMsg;
+    return newMessages;
+  }
+  // 处理工具调用结束
+  handleInvokeToolEnd(messages, data) {
+    var _a;
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    this.updateExecutionCost(newMessages, index, (_a = data == null ? void 0 : data.data) == null ? void 0 : _a.uniqueId, data.cost);
+    return newMessages;
+  }
+  // 处理工作流组件开始
+  handleComponentStart(messages, data) {
+    var _a;
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { content } = data;
+    const { nodeId, args, componentType } = content;
+    const { name } = (0, import_workflowNode.getNodeTemplate)(componentType);
+    (_a = aiMsg == null ? void 0 : aiMsg.execute) == null ? void 0 : _a.push({
+      id: nodeId || (0, import_uuid.v4)(),
+      createdAt: /* @__PURE__ */ new Date(),
+      type: "ActionExecutionMessage",
+      name: name ? `已调用${name}组件` : "已调用工作流组件",
+      arguments: args,
+      extra: {
+        ...data,
+        ...this.getWorkflowIcon()
+      }
+    });
+    newMessages[index] = aiMsg;
+    return newMessages;
+  }
+  // 处理工作流组件结束
+  handleComponentEnd(messages, data) {
+    var _a;
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    this.updateExecutionCost(newMessages, index, (_a = data == null ? void 0 : data.content) == null ? void 0 : _a.nodeId, data.cost);
+    return newMessages;
+  }
+  // 处理响应正文
+  handleResponse(messages, data) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { contentType, content, success, errorMsg } = data;
+    if (success) {
+      if (aiMsg.status === import_XAiMessage.MessageStatus.init) {
+        aiMsg.status = import_XAiMessage.MessageStatus.pending;
+      }
+      if (contentType === "TEXT" && "text" in aiMsg.content) {
+        aiMsg.content.text += content;
+      }
+    } else {
+      aiMsg.status = import_XAiMessage.MessageStatus.failed;
+      aiMsg.content.text = `<span style="color: red;">❌ ${errorMsg}</span>`;
+    }
+    newMessages[index] = aiMsg;
+    return newMessages;
+  }
+  // 处理完成
+  handleFinish(messages, data) {
+    var _a;
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    const msg = newMessages[index];
+    msg.execute = msg.execute || [];
+    msg.execute.push({
+      id: ((_a = data == null ? void 0 : data.data) == null ? void 0 : _a.uniqueId) || (0, import_uuid.v4)(),
+      createdAt: /* @__PURE__ */ new Date(),
+      type: "ActionExecutionMessage",
+      name: "运行完毕",
+      extra: {
+        ...data,
+        cost: (data == null ? void 0 : data.cost) ? ((data == null ? void 0 : data.cost) / 1e3).toFixed(2) : 0,
+        icon: import_completed.default,
+        expandIcon: import_completed_black.default
+      }
+    });
+    msg.status = import_XAiMessage.MessageStatus.done;
+    return newMessages;
+  }
+  // 处理响应数据
+  processResponse(messages, responseData) {
+    if (!(responseData == null ? void 0 : responseData.content) && !(responseData == null ? void 0 : responseData.data))
+      return messages;
+    let response = responseData.data || responseData.content;
+    if (typeof response === "string") {
+      response = (0, import_utils.safeJsonParse)(response, {
+        action: "RESPONSE" /* response */,
+        data: {
+          contentType: "TEXT",
+          content: "解析响应数据时出错，请重试",
+          success: false,
+          errorMsg: "响应数据格式错误"
+        }
+      });
+    }
+    const { action, data = {} } = response;
+    switch (action) {
+      case "FOLLOW_UP" /* suggest */:
+        return this.handleSuggestion(messages, data);
+      case "REASONING" /* reasoning */:
+        return this.handleReasoning(messages, data);
+      case "REASONING_END" /* reasoningEnd */:
+        return this.handleReasoningEnd(messages);
+      case "COMPONENT_START" /* componentStart */:
+        return this.handleComponentStart(messages, data);
+      case "COMPONENT_END" /* componentEnd */:
+        return this.handleComponentEnd(messages, data);
+      case "RECALL_KNOWLEDGE_START" /* recallKnowledgeStart */:
+        return this.handleRecallKnowledgeStart(messages, data);
+      case "RECALL_KNOWLEDGE_END" /* recallKnowledgeEnd */:
+        return this.handleRecallKnowledgeEnd(messages, data);
+      case "INVOKE_AGENT_TOOL_START" /* invokeToolStart */:
+        return this.handleInvokeToolStart(messages, data);
+      case "INVOKE_AGENT_TOOL_END" /* invokeToolEnd */:
+        return this.handleInvokeToolEnd(messages, data);
+      case "RESPONSE" /* response */:
+        return this.handleResponse(messages, data);
+      case "FINISHED" /* finish */:
+        return this.handleFinish(messages, data);
+      default:
+        return messages;
+    }
+  }
+};
+var useAgentGenerator = (params) => {
+  const {
+    url,
+    token,
+    config,
+    onError,
+    onMessage,
+    onSuccess,
+    ...rest
+  } = params;
   const textSpeed = 60;
   const [messages, setMessages] = (0, import_react.useState)([]);
   const [sessionList, setSessionList] = (0, import_react.useState)([]);
@@ -91,6 +387,7 @@ var useAgentGenerator = ({
   const [appInfo, setAppInfo] = (0, import_react.useState)(null);
   const requestInfo = (0, import_react.useRef)({});
   const { appNo, pt, tc } = config || {};
+  const messageProcessor = (0, import_react.useMemo)(() => new MessageProcessor(textSpeed), [textSpeed]);
   (0, import_react.useEffect)(() => {
     if (url && config) {
       const initConfig = async () => {
@@ -134,321 +431,142 @@ var useAgentGenerator = ({
     }
   });
   (0, import_react.useEffect)(() => {
-    if (!responseData || !responseData.content || !responseData.data)
+    if (!responseData)
       return;
-    let response = responseData.data || responseData.content;
-    if (typeof response === "string") {
-      response = (0, import_utils.safeJsonParse)(response, {
-        action: "RESPONSE" /* response */,
-        data: {
-          contentType: "TEXT",
-          content: "解析响应数据时出错，请重试",
-          success: false,
-          errorMsg: "响应数据格式错误"
-        }
-      });
-    }
-    setMessages((prevMessages) => {
-      var _a, _b, _c;
-      let newMessages = [...prevMessages];
-      const { action, data } = response;
-      const { uniqueId, contentType, content, toolType, args, knowledgeType, success = true, errorMsg } = data || {};
-      if (action === "FOLLOW_UP" /* suggest */) {
-        const suggestMessageIndex = messages.findIndex((item) => item.type === "SuggestionMessage");
-        if (suggestMessageIndex === -1) {
-          const newMessage = {
-            id: (0, import_uuid.v4)(),
-            type: "SuggestionMessage",
-            createdAt: /* @__PURE__ */ new Date(),
-            status: import_XAiMessage.MessageStatus.done,
-            role: import_XAiMessage.MessageRole.suggestion,
-            content: [{
-              key: (0, import_uuid.v4)(),
-              description: content
-            }]
-          };
-          newMessages.push(newMessage);
-        } else {
-          const suggestMsg = newMessages[suggestMessageIndex];
-          if (suggestMsg == null ? void 0 : suggestMsg.content) {
-            suggestMsg.content.push({
-              key: (0, import_uuid.v4)(),
-              description: content
-            });
-          }
-        }
-        return newMessages;
-      }
-      const lastAIMsgIndex = [...prevMessages].findLastIndex(
-        (m) => m.role === import_XAiMessage.MessageRole.assistant && m.type === "TextMessage" && (m.status === "init" || m.status === "pending")
-      );
-      const realIndex = lastAIMsgIndex === -1 ? -1 : prevMessages.length - 1 - lastAIMsgIndex;
-      const lastAIMsg = realIndex !== -1 ? prevMessages[realIndex] : null;
-      let aiMsg;
-      if (!lastAIMsg) {
-        aiMsg = {
-          id: (0, import_uuid.v4)(),
-          type: "TextMessage",
-          createdAt: /* @__PURE__ */ new Date(),
-          role: import_XAiMessage.MessageRole.assistant,
-          content: { text: "" },
-          execute: [],
-          status: import_XAiMessage.MessageStatus.init
-        };
-        newMessages.push(aiMsg);
-      } else if (lastAIMsg.type === "TextMessage") {
-        aiMsg = { ...lastAIMsg };
-      } else if (lastAIMsg.type === "ImageMessage") {
-        aiMsg = void 0;
-      }
-      if (!aiMsg) {
-        return newMessages;
-      }
-      switch (action) {
-        case "REASONING_END" /* reasoningEnd */:
-          aiMsg.thinks += "/n";
-          break;
-        case "REASONING" /* reasoning */:
-          aiMsg.thinks += content;
-          break;
-        case "RECALL_KNOWLEDGE_START" /* recallKnowledgeStart */:
-          aiMsg.execute = aiMsg.execute || [];
-          aiMsg.execute.push({
-            id: uniqueId,
-            createdAt: /* @__PURE__ */ new Date(),
-            type: "ActionExecutionMessage",
-            name: knowledgeType === "DOCUMENT" ? "已搜索文档知识库" : "已搜索问答知识库",
-            arguments: args,
-            extra: {
-              ...data,
-              icon: knowledgeType === "DOCUMENT" ? import_document.default : import_qa.default,
-              expandIcon: knowledgeType === "DOCUMENT" ? import_document_black.default : import_qa_black.default
-            }
-          });
-          break;
-        case "RECALL_KNOWLEDGE_END" /* recallKnowledgeEnd */: {
-          const recallKnowledgeMsg = (_a = aiMsg.execute) == null ? void 0 : _a.find((item) => item.id === uniqueId);
-          if (recallKnowledgeMsg) {
-            recallKnowledgeMsg.extra.cost = (data == null ? void 0 : data.cost) ? ((data == null ? void 0 : data.cost) / 1e3).toFixed(2) : 0;
-          }
-          break;
-        }
-        case "INVOKE_AGENT_TOOL_START" /* invokeToolStart */:
-          aiMsg.execute = aiMsg.execute || [];
-          aiMsg.execute.push({
-            id: uniqueId,
-            createdAt: /* @__PURE__ */ new Date(),
-            type: "ActionExecutionMessage",
-            name: toolType === "SKILL" ? "已调用工作流" : "已调用工具",
-            arguments: args,
-            extra: {
-              ...data,
-              icon: toolType === "SKILL" ? import_skillNo.default : import_tools.default,
-              expandIcon: toolType === "SKILL" ? import_skillNo_black.default : import_tools_black.default
-            }
-          });
-          break;
-        case "INVOKE_AGENT_TOOL_END" /* invokeToolEnd */: {
-          const invokeToolMsg = (_b = aiMsg.execute) == null ? void 0 : _b.find((item) => item.id === uniqueId);
-          if (invokeToolMsg) {
-            invokeToolMsg.extra.cost = (data == null ? void 0 : data.cost) ? ((data == null ? void 0 : data.cost) / 1e3).toFixed(2) : 0;
-          }
-          break;
-        }
-        case "RESPONSE" /* response */:
-          if (success) {
-            if (aiMsg.status === import_XAiMessage.MessageStatus.init) {
-              aiMsg.status = import_XAiMessage.MessageStatus.pending;
-            }
-            if (contentType === "TEXT" && "text" in aiMsg.content) {
-              aiMsg.content.text += content;
-            }
-          } else {
-            aiMsg.status = import_XAiMessage.MessageStatus.failed;
-            aiMsg.content.text = `<span style="color: red;">❌ ${errorMsg}</span>`;
-          }
-          break;
-        case "FINISHED" /* finish */:
-          aiMsg.status = import_XAiMessage.MessageStatus.done;
-          onMessage == null ? void 0 : onMessage((_c = aiMsg == null ? void 0 : aiMsg.content) == null ? void 0 : _c.text, aiMsg);
-          initAppConversations(false);
-          break;
-      }
-      if (lastAIMsg && lastAIMsg.type === "TextMessage") {
-        newMessages[realIndex] = aiMsg;
-      } else if (!lastAIMsg) {
-        if (newMessages.length > 0 && newMessages[newMessages.length - 1].type === "TextMessage") {
-          newMessages[newMessages.length - 1] = aiMsg;
-        } else {
-          newMessages = [...newMessages, aiMsg];
-        }
-      }
-      return newMessages;
-    });
-  }, [responseData]);
-  const setMessage = async (messageType, content) => {
+    setMessages(
+      (prev) => messageProcessor.processResponse(prev, responseData)
+    );
+  }, [responseData, messageProcessor]);
+  const setMessage = (0, import_react.useCallback)(async (messageType, content) => {
     switch (messageType) {
       case "prologue": {
         setMessages((prev) => {
-          const prologueMessageIndex = prev.findIndex((item) => {
+          const prologueIndex = prev.findIndex((item) => {
             var _a;
             return (_a = item.extra) == null ? void 0 : _a.prologue;
           });
-          if (prologueMessageIndex === -1) {
+          if (prologueIndex === -1) {
             const newMessage = {
               type: "TextMessage",
               id: (0, import_uuid.v4)(),
               createdAt: /* @__PURE__ */ new Date(),
               role: import_XAiMessage.MessageRole.assistant,
               status: import_XAiMessage.MessageStatus.done,
-              content: {
-                text: content
-              },
-              extra: {
-                prologue: true,
-                noFooter: true
-              }
+              content: { text: content },
+              extra: { prologue: true, noFooter: true }
             };
             return [newMessage, ...prev];
           }
-          return prev.map((item) => {
-            var _a;
-            if (((_a = item.extra) == null ? void 0 : _a.prologue) && item.content && "text" in item.content) {
-              item.content.text = content;
+          return prev.map(
+            (item) => {
+              var _a;
+              return ((_a = item.extra) == null ? void 0 : _a.prologue) && item.content && "text" in item.content ? { ...item, content: { text: content } } : item;
             }
-            return item;
-          });
+          );
         });
         break;
       }
       case "suggest": {
         setMessages((prev) => {
-          const suggestMessageIndex = prev.findIndex((item) => item.type === "SuggestionMessage");
-          if (suggestMessageIndex === -1) {
-            const prologueMessageIndex = prev.findIndex((item) => {
-              var _a;
-              return (_a = item.extra) == null ? void 0 : _a.prologue;
-            });
-            const newMessage = {
-              id: (0, import_uuid.v4)(),
-              type: "SuggestionMessage",
-              createdAt: /* @__PURE__ */ new Date(),
-              status: import_XAiMessage.MessageStatus.done,
-              role: import_XAiMessage.MessageRole.suggestion,
-              content
-            };
-            if (prologueMessageIndex === -1) {
-              return [newMessage, ...prev];
-            }
+          const suggestIndex = prev.findIndex((item) => item.type === "SuggestionMessage");
+          const prologueIndex = prev.findIndex((item) => {
+            var _a;
+            return (_a = item.extra) == null ? void 0 : _a.prologue;
+          });
+          const newSuggestion = {
+            id: (0, import_uuid.v4)(),
+            type: "SuggestionMessage",
+            createdAt: /* @__PURE__ */ new Date(),
+            status: import_XAiMessage.MessageStatus.done,
+            role: import_XAiMessage.MessageRole.suggestion,
+            content
+          };
+          if (suggestIndex === -1) {
             const newMessages = [...prev];
-            newMessages.splice(prologueMessageIndex + 1, 0, newMessage);
+            const insertIndex = prologueIndex === -1 ? 0 : prologueIndex + 1;
+            newMessages.splice(insertIndex, 0, newSuggestion);
             return newMessages;
           }
-          return prev.map((item) => {
-            if (item.type === "SuggestionMessage") {
-              item.content = content;
-            }
-            return item;
-          });
+          return prev.map(
+            (item) => item.type === "SuggestionMessage" ? { ...item, content } : item
+          );
         });
         break;
       }
       default:
-        break;
+        console.warn(`Unknown message type: ${messageType}`);
     }
-  };
-  const chat = async (content) => {
-    let newMessage;
-    let inputs_msg;
-    if (typeof content === "string") {
-      newMessage = {
+  }, []);
+  const chat = async (content, extra, insert = true) => {
+    if (insert) {
+      const newMessage = {
         type: "TextMessage",
         id: (0, import_uuid.v4)(),
         createdAt: /* @__PURE__ */ new Date(),
         role: import_XAiMessage.MessageRole.user,
         status: import_XAiMessage.MessageStatus.done,
-        content: {
-          text: content
-        }
+        extra,
+        content: { text: content }
       };
-    } else {
-      newMessage = {
-        type: "TextMessage",
-        id: (0, import_uuid.v4)(),
-        createdAt: /* @__PURE__ */ new Date(),
-        role: import_XAiMessage.MessageRole.user,
-        status: import_XAiMessage.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((prev) => {
+        const newMessages = prev.filter((item) => item.type !== "SuggestionMessage");
+        newMessages.push(newMessage);
+        return newMessages;
+      });
     }
-    setMessages((prev) => {
-      const suggestMessageIndex = messages.findIndex((item) => item.type === "SuggestionMessage");
-      if (suggestMessageIndex !== -1) {
-        prev.splice(suggestMessageIndex, 1);
-      }
-      prev.push(newMessage);
-      return prev;
-    });
+    let requestBody = {
+      appParams: {
+        inputs_msg: content
+      },
+      sessionId: currentSessionId
+    };
     start(
       `${url}/appouter/${appNo}/sendMsg?pt=${pt}&tc=${tc}`,
-      {
-        appParams: {
-          inputs_msg
-        },
-        sessionId: currentSessionId
-      },
+      requestBody,
       {
         Authorization: `Bearer ${token}`
       }
     );
   };
-  const reChat = () => {
-    setMessages((prevMessages) => {
-      const lastAIMsgIndex = [...prevMessages].findLastIndex(
+  const reChat = (0, import_react.useCallback)(() => {
+    setMessages((prev) => {
+      var _a;
+      const newMessages = [...prev];
+      const lastAIIndex = newMessages == null ? void 0 : newMessages.findLastIndex(
         (m) => m.role === import_XAiMessage.MessageRole.assistant && m.type === "TextMessage"
       );
-      const newMessages = [...prevMessages];
-      if (lastAIMsgIndex !== -1) {
-        const realIndex = newMessages.length - 1 - lastAIMsgIndex;
-        newMessages.splice(realIndex, 1);
+      if (lastAIIndex !== -1) {
+        newMessages.splice(lastAIIndex, 1);
       }
-      const suggestMessageIndex = newMessages.findIndex((item) => item.type === "SuggestionMessage");
-      if (suggestMessageIndex !== -1) {
-        newMessages.splice(suggestMessageIndex, 1);
+      const suggestIndex = newMessages.findIndex((item) => item.type === "SuggestionMessage");
+      if (suggestIndex !== -1) {
+        newMessages.splice(suggestIndex, 1);
       }
-      const lastUserMsg = [...newMessages].findLast(
+      const lastUserMsg = newMessages.findLast(
         (m) => m.role === import_XAiMessage.MessageRole.user && m.type === "TextMessage"
       );
-      if (lastUserMsg && lastUserMsg.content && typeof lastUserMsg.content.text === "string") {
-        chat(lastUserMsg.content.text);
+      if ((_a = lastUserMsg == null ? void 0 : lastUserMsg.content) == null ? void 0 : _a.text) {
+        setTimeout(() => {
+          chat(lastUserMsg.content.text, lastUserMsg.extra, false);
+        }, 0);
       }
       return newMessages;
     });
-  };
+  }, [chat]);
   const stopChat = () => {
     stop();
-    setMessages((prevMessages) => {
-      const newMessages = [...prevMessages];
-      newMessages.forEach((m) => {
-        if (m.status === import_XAiMessage.MessageStatus.pending) {
-          m.status = import_XAiMessage.MessageStatus.done;
-        } else if (m.status === import_XAiMessage.MessageStatus.init) {
-          if (m.type === "TextMessage") {
-            m.content.text = '<span style="color: #cbcbcb;">用户已取消</span>';
+    setMessages(
+      (prev) => prev.map((m) => {
+        if (m.status === import_XAiMessage.MessageStatus.pending || m.status === import_XAiMessage.MessageStatus.init) {
+          const newMsg = { ...m, status: import_XAiMessage.MessageStatus.done };
+          if (m.type === "TextMessage" && m.status === import_XAiMessage.MessageStatus.init) {
+            newMsg.content.text = '<span style="color: #cbcbcb;">用户已取消</span>';
           }
-          m.status = import_XAiMessage.MessageStatus.done;
+          return newMsg;
         }
-      });
-      return newMessages;
-    });
+        return m;
+      })
+    );
   };
   const formatMessages = (data, init = false) => {
     setMessages((prev) => {