@ai-group/chat-sdk 3.5.5 → 3.5.6

package/dist/cjs/hooks/useAgentGenerator.js

@@ -0,0 +1,804 @@
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/hooks/useAgentGenerator.ts
+var useAgentGenerator_exports = {};
+__export(useAgentGenerator_exports, {
+  ActionAvailability: () => ActionAvailability,
+  ActionType: () => ActionType,
+  ProcessStatus: () => ProcessStatus,
+  default: () => useAgentGenerator_default
+});
+module.exports = __toCommonJS(useAgentGenerator_exports);
+var import_react = require("react");
+var import_uuid = require("uuid");
+var import_useEventStreamRequest = require("./useEventStreamRequest");
+var import_XAiMessage = require("../types/XAiMessage");
+var import_types = require("../types");
+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"));
+var import_skillNo = __toESM(require("../assets/skillNo.png"));
+var import_skillNo_black = __toESM(require("../assets/skillNo-black.png"));
+var import_tools = __toESM(require("../assets/tools.png"));
+var import_tools_black = __toESM(require("../assets/tools-black.png"));
+var ActionType = /* @__PURE__ */ ((ActionType2) => {
+  ActionType2["recallKnowledgeStart"] = "RECALL_KNOWLEDGE_START";
+  ActionType2["recallKnowledgeEnd"] = "RECALL_KNOWLEDGE_END";
+  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";
+  return ActionType2;
+})(ActionType || {});
+var ProcessStatus = /* @__PURE__ */ ((ProcessStatus2) => {
+  ProcessStatus2["start"] = "START";
+  ProcessStatus2["end"] = "END";
+  return ProcessStatus2;
+})(ProcessStatus || {});
+var ActionAvailability = /* @__PURE__ */ ((ActionAvailability2) => {
+  ActionAvailability2["disabled"] = "disabled";
+  ActionAvailability2["enabled"] = "enabled";
+  ActionAvailability2["remote"] = "remote";
+  return ActionAvailability2;
+})(ActionAvailability || {});
+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 };
+  }
+  // 更新工具执行事件
+  updateExecution(messages, messageIndex, uniqueId, cost, name) {
+    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;
+        if (name)
+          execution.name = name;
+      }
+    }
+  }
+  // 查找最后一个待处理的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 || ""}
+`,
+      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)() } = data || {};
+    const { args } = (data == null ? void 0 : data.data) || {};
+    aiMsg.execute = aiMsg.execute || [];
+    aiMsg.execute.push({
+      id: uniqueId,
+      createdAt: /* @__PURE__ */ new Date(),
+      type: "ActionExecutionMessage",
+      name: "开始搜索知识库",
+      arguments: args,
+      extra: {
+        ...data,
+        icon: import_document.default,
+        expandIcon: import_document_black.default
+      }
+    });
+    newMessages[index] = aiMsg;
+    return newMessages;
+  }
+  // 处理知识库调用结束
+  handleRecallKnowledgeEnd(messages, data) {
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    if (index === -1)
+      return messages;
+    const { uniqueId = (0, import_uuid.v4)(), cost } = data || {};
+    const newMessages = [...messages];
+    this.updateExecution(newMessages, index, uniqueId, cost, "已搜索知识库");
+    return newMessages;
+  }
+  // 处理工具调用开始
+  handleInvokeToolStart(messages, data) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { uniqueId = (0, import_uuid.v4)() } = data || {};
+    const { 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) {
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    const { uniqueId = (0, import_uuid.v4)() } = data || {};
+    const { toolType } = (data == null ? void 0 : data.data) || {};
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    this.updateExecution(newMessages, index, uniqueId, data.cost, toolType === "SKILL" ? "已调用工作流组件" : "已调用工具");
+    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;
+  }
+  // 处理工作流组件流消息
+  handleComponentStream(messages, data) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { nodeId = import_uuid.v4, content } = data || {};
+    const { processData } = content || {};
+    aiMsg.raw = data.raw;
+    aiMsg.stepContent = aiMsg.stepContent || [];
+    aiMsg.stepContent.push({
+      id: nodeId,
+      content: processData
+    });
+    if (aiMsg.status === import_XAiMessage.MessageStatus.init) {
+      aiMsg.status = import_XAiMessage.MessageStatus.pending;
+    }
+    newMessages[index] = aiMsg;
+    return newMessages;
+  }
+  // 处理工作流组件结束
+  handleComponentEnd(messages, data) {
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    const { content } = data || {};
+    const { nodeId, componentType } = content || {};
+    const { name } = (0, import_workflowNode.getNodeTemplate)(componentType);
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    this.updateExecution(newMessages, index, nodeId, data.cost, name ? `已调用完成${name}组件` : "已调用工作流组件");
+    return newMessages;
+  }
+  // 处理响应正文
+  handleResponse(messages, data) {
+    const [newMessages, index, aiMsg] = this.ensureAIMessageExists(messages);
+    const { contentType, content, success, errorMsg } = data;
+    if (success) {
+      aiMsg.raw = data.raw;
+      if (aiMsg.status === import_XAiMessage.MessageStatus.init) {
+        aiMsg.status = import_XAiMessage.MessageStatus.pending;
+      }
+      if (contentType === "TEXT" && "text" in aiMsg.content) {
+        aiMsg.content.text += content;
+      }
+      if (contentType === "JSON") {
+        aiMsg.content.text += JSON.stringify(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) {
+    const [index, _aiMsg] = this.findLastPendingAIMessage(messages);
+    const { data: { uniqueId = (0, import_uuid.v4)() } = {}, cost, needExecuteDetail } = data || {};
+    if (index === -1)
+      return messages;
+    const newMessages = [...messages];
+    const msg = newMessages[index];
+    if (needExecuteDetail) {
+      msg.execute = msg.execute || [];
+      msg.execute.push({
+        id: uniqueId || (0, import_uuid.v4)(),
+        createdAt: /* @__PURE__ */ new Date(),
+        type: "ActionExecutionMessage",
+        name: "运行完毕",
+        extra: {
+          ...data,
+          cost: cost ? (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) {
+    var _a;
+    if (!((_a = responseData == null ? void 0 : responseData.data) == null ? void 0 : _a.outputs))
+      return messages;
+    let response = responseData.data.outputs;
+    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_STREAM" /* componentStream */:
+        return this.handleComponentStream(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,
+    textSpeed = 55,
+    ...rest
+  } = params;
+  const [messages, setMessages] = (0, import_react.useState)([]);
+  const [sessionList, setSessionList] = (0, import_react.useState)([]);
+  const [sessionPagination, setSessionPagination] = (0, import_react.useState)({ total: 0, pageNum: 1, pageSize: import_constants.SESSION_PAGE_SIZE });
+  const [currentSessionId, setCurrentSessionId] = (0, import_react.useState)("");
+  const [appInfo, setAppInfo] = (0, import_react.useState)(null);
+  const requestInfo = (0, import_react.useRef)({});
+  const { appNo } = config || {};
+  const messageProcessor = (0, import_react.useMemo)(() => new MessageProcessor(textSpeed), [textSpeed]);
+  (0, import_react.useEffect)(() => {
+    if (url && config) {
+      const initConfig = async () => {
+        try {
+          const { data, result } = await (0, import_api.fetchAppConfig)({ url, ...config });
+          if (result.code === import_constants.API_SUCCESS_CODE) {
+            setAppInfo(data);
+            onSuccess == null ? void 0 : onSuccess({
+              code: import_types.SuccessCode.APP_CONFIG_INIT_SUCCESS,
+              message: "初始化配置成功"
+            });
+          } else {
+            onError == null ? void 0 : onError({
+              code: result.code || import_types.ErrorCode.APP_NOT_FOUND,
+              message: result.message || "获取应用配置失败"
+            });
+          }
+        } catch (error) {
+          onError == null ? void 0 : onError({
+            code: import_types.ErrorCode.API_ERROR,
+            message: "获取应用配置失败"
+          });
+        }
+      };
+      initConfig();
+    }
+  }, [url, config]);
+  const { start, loading, responseData, stop } = (0, import_useEventStreamRequest.useEventStreamRequest)({
+    streaming: true,
+    textSpeed,
+    onClose: () => {
+      setTimeout(() => {
+        var _a, _b;
+        if (((_a = requestInfo.current) == null ? void 0 : _a.sensitiveWords) && ((_b = requestInfo.current) == null ? void 0 : _b.requestId)) {
+          import_request.request.post(url);
+        }
+      }, 500);
+    },
+    onOpen(res) {
+      console.log("res-----open:", res);
+    }
+  });
+  (0, import_react.useEffect)(() => {
+    if (!responseData)
+      return;
+    setMessages((prev) => messageProcessor.processResponse(prev, responseData));
+  }, [responseData, messageProcessor]);
+  const setMessage = (0, import_react.useCallback)(async (messageType, content) => {
+    switch (messageType) {
+      case "prologue": {
+        setMessages((prev) => {
+          const prologueIndex = prev.findIndex((item) => {
+            var _a;
+            return (_a = item.extra) == null ? void 0 : _a.prologue;
+          });
+          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 }
+            };
+            return [newMessage, ...prev];
+          }
+          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;
+          });
+        });
+        break;
+      }
+      case "suggest": {
+        setMessages((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];
+            const insertIndex = prologueIndex === -1 ? 0 : prologueIndex + 1;
+            newMessages.splice(insertIndex, 0, newSuggestion);
+            return newMessages;
+          }
+          return prev.map((item) => item.type === "SuggestionMessage" ? { ...item, content } : item);
+        });
+        break;
+      }
+      default:
+        console.warn(`Unknown message type: ${messageType}`);
+    }
+  }, []);
+  const chat = (0, import_react.useCallback)(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,
+        extra,
+        content
+      };
+      setMessages((prev) => {
+        const newMessages = prev.filter((item) => item.type !== "SuggestionMessage");
+        newMessages.push(newMessage);
+        return newMessages;
+      });
+    }
+    const requestBody = {
+      appParams: {
+        inputs_msg: content.text
+      },
+      sessionId: currentSessionId
+    };
+    start(
+      `${url}/api/gateway-web/openApi/v1/aizt/app/${appNo}/sendMsgStreaming`,
+      requestBody,
+      {
+        Authorization: `Bearer ${token}`
+      }
+    );
+  }, [appNo, currentSessionId, start, token, url]);
+  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"
+      );
+      if (lastAIIndex !== -1) {
+        newMessages.splice(lastAIIndex, 1);
+      }
+      const suggestIndex = newMessages.findIndex((item) => item.type === "SuggestionMessage");
+      if (suggestIndex !== -1) {
+        newMessages.splice(suggestIndex, 1);
+      }
+      const lastUserMsg = newMessages.findLast(
+        (m) => m.role === import_XAiMessage.MessageRole.user && m.type === "TextMessage"
+      );
+      if ((_a = lastUserMsg == null ? void 0 : lastUserMsg.content) == null ? void 0 : _a.text) {
+        setTimeout(() => {
+          chat(lastUserMsg.content, lastUserMsg.extra, false);
+        }, 0);
+      }
+      return newMessages;
+    });
+  }, [chat]);
+  const stopChat = () => {
+    stop();
+    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>';
+        }
+        return newMsg;
+      }
+      return m;
+    }));
+  };
+  const formatMessages = (data, init = false) => {
+    setMessages((prev) => {
+      const initMessagesList = [];
+      data.forEach((item) => {
+        const { content, messageTime, role } = item || {};
+        if (role === "USER") {
+          const newMessage = {
+            type: "TextMessage",
+            id: (0, import_uuid.v4)(),
+            createdAt: messageTime,
+            role: import_XAiMessage.MessageRole.user,
+            status: import_XAiMessage.MessageStatus.done,
+            content: {
+              text: content
+            }
+          };
+          initMessagesList.push(newMessage);
+        } else if (role === "AI") {
+          const newMessage = {
+            id: (0, import_uuid.v4)(),
+            type: "TextMessage",
+            createdAt: messageTime,
+            role: import_XAiMessage.MessageRole.assistant,
+            content: { text: content },
+            execute: [],
+            status: import_XAiMessage.MessageStatus.done,
+            isLike: item.isLike,
+            raw: item.raw
+          };
+          initMessagesList.push(newMessage);
+        }
+      });
+      if (init) {
+        return [...initMessagesList];
+      }
+      return [...prev, ...initMessagesList];
+    });
+  };
+  const initPrologue = async () => {
+    const { onboardingInfo = {} } = appInfo;
+    const { suggested_questions = [], prologue = "" } = onboardingInfo || {};
+    if (prologue) {
+      setMessage("prologue", prologue);
+    } else {
+      setMessages((prev) => prev.filter((item) => {
+        var _a;
+        return !((_a = item.extra) == null ? void 0 : _a.prologue);
+      }));
+    }
+    const formatSuggests = suggested_questions.filter((item) => item);
+    if (formatSuggests.length > 0) {
+      const suggestMessages = formatSuggests.map((item, index) => ({
+        key: index + 1,
+        description: item
+      }));
+      const hasMessages = messages.some((item) => {
+        var _a;
+        return !((_a = item.extra) == null ? void 0 : _a.prologue) && item.type !== "SuggestionMessage";
+      });
+      if (!hasMessages) {
+        setMessage("suggest", suggestMessages);
+      }
+    } else {
+      setMessages((prev) => prev.filter((item) => !(item.type === "SuggestionMessage")));
+    }
+  };
+  const initAppConversations = async (fetchDetail = false) => {
+    try {
+      const { data = [], pagination, result } = await (0, import_api.fetchSessionList)({
+        url,
+        appNo,
+        token,
+        pageNum: sessionPagination.pageNum,
+        pageSize: sessionPagination.pageSize
+      });
+      if ((result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE) {
+        if (data.length > 0) {
+          const formatSessions = data.map((item) => {
+            return {
+              key: item.sessionId,
+              label: item.title || "新对话",
+              timestamp: new Date(item.startTime).getTime(),
+              messages: [],
+              meta: {
+                ...item
+              }
+            };
+          });
+          setSessionList(formatSessions);
+          setSessionPagination({ pageNum: sessionPagination.pageNum, pageSize: sessionPagination.pageSize, total: (pagination == null ? void 0 : pagination.total) || 0 });
+          const { sessionId } = data == null ? void 0 : data[0];
+          fetchDetail && setCurrentSession(sessionId);
+        } else {
+          setSessionList([]);
+          setCurrentSessionId((0, import_uuid.v4)());
+          initPrologue();
+        }
+        onSuccess == null ? void 0 : onSuccess({
+          code: import_types.SuccessCode.APP_MESSAGES_INIT_SUCCESS,
+          message: "初始化聊天记录成功"
+        });
+      } else {
+        initPrologue();
+      }
+    } catch {
+    }
+  };
+  const setCurrentSession = async (sessionId) => {
+    if (sessionId) {
+      if (sessionId === currentSessionId) {
+        return;
+      }
+      setCurrentSessionId(sessionId);
+      const { data, result } = await (0, import_api.fetchSessionDetail)({
+        url,
+        appNo,
+        sessionId,
+        token
+      });
+      if ((result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE) {
+        const sessionsData = data == null ? void 0 : data.reduce((acc, cur) => {
+          const sessionMessages = Array.isArray(cur.sessionMessages) ? cur.sessionMessages : [];
+          const msgs = sessionMessages.map((msg) => ({
+            ...msg,
+            raw: cur,
+            isLike: cur == null ? void 0 : cur.isLike
+          }));
+          return [...acc, ...msgs];
+        }, []);
+        if (sessionsData.length > 0) {
+          formatMessages(sessionsData, true);
+        }
+      }
+    } else {
+      setCurrentSessionId((0, import_uuid.v4)());
+    }
+  };
+  const deleteSession = async (sessionId) => {
+    if (!sessionId)
+      return;
+    const { result } = await (0, import_api.closeSession)({
+      url,
+      appNo,
+      sessionId,
+      token
+    });
+    if ((result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE) {
+      initAppConversations(false);
+      if (sessionId === currentSessionId) {
+        createNewChat();
+      }
+    }
+  };
+  const updateSession = async (sessionId, title) => {
+    if (!sessionId && !title)
+      return;
+    const { result } = await (0, import_api.updateSessionTitle)({
+      url,
+      appNo,
+      title,
+      sessionId,
+      token
+    });
+    if ((result == null ? void 0 : result.code) === import_constants.API_SUCCESS_CODE) {
+      initAppConversations(false);
+    }
+  };
+  const createNewChat = () => {
+    stopChat();
+    setMessages([]);
+    initPrologue();
+    setCurrentSessionId((0, import_uuid.v4)());
+  };
+  (0, import_react.useEffect)(() => {
+    if (appInfo) {
+      initAppConversations(true);
+    }
+  }, [appInfo]);
+  return {
+    appInfo,
+    messages,
+    loading,
+    content: responseData.content,
+    sessionList,
+    currentSessionId,
+    sessionPagination,
+    initAppConversations,
+    reChat,
+    chat,
+    createNewChat,
+    setCurrentSessionId,
+    setCurrentSession,
+    updateSession,
+    deleteSession,
+    setMessage,
+    setMessages,
+    stopChat,
+    initPrologue
+  };
+};
+var useAgentGenerator_default = useAgentGenerator;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ActionAvailability,
+  ActionType,
+  ProcessStatus
+});
+//# sourceMappingURL=useAgentGenerator.js.map