@mybricks/plugin-ai 0.0.1 → 0.0.3

package/dist/rxai/src/agent/planning.js

@@ -0,0 +1,1076 @@
+import { defineProperty as _defineProperty, asyncToGenerator as _asyncToGenerator, objectSpread2 as _objectSpread2, objectWithoutProperties as _objectWithoutProperties } from '../../../_virtual/_rollupPluginBabelHelpers.js';
+import { getSystemPrompt } from '../prompt/planning.js';
+import { BaseAgent } from './base.js';
+import { parseFileBlocks } from '../tool/util.js';
+import { getToolPrompt } from '../prompt/tool.js';
+import { Events } from '../utils/events.js';
+import { uuid } from '../utils/uuid.js';
+import { RxaiError } from '../error/base.js';
+import { retry } from '../utils/retry.js';
+
+var _excluded = ["events"];
+/** 没有response */
+var CATCH_EMPTY = Symbol("CATCH_EMPTY");
+/**
+ * 分析计划
+ * 执行计划
+ */
+class PlanningAgent extends BaseAgent {
+  constructor(options) {
+    super(options);
+    // 设置UUID
+    _defineProperty(this, "startTime", 0);
+    _defineProperty(this, "endTime", 0);
+    _defineProperty(this, "llmContent", "");
+    _defineProperty(this, "loading", false);
+    _defineProperty(this, "commands", []);
+    _defineProperty(this, "filenames", []);
+    _defineProperty(this, "defaultPlanList", false);
+    /** 事件 */
+    _defineProperty(this, "events", new Events());
+    /** 整体运行状态 */
+    _defineProperty(this, "status", "pending");
+    /** error信息 */
+    _defineProperty(this, "error", null);
+    _defineProperty(this, "enableRetry", true);
+    _defineProperty(this, "summaryMessage", "");
+    /** 当前请求的cancel */
+    _defineProperty(this, "currentRequestCancel", () => {});
+    _defineProperty(this, "summaryLoading", false);
+    _defineProperty(this, "getHistoryMessages", filenames => {
+      return this.options.historyMessages(filenames);
+    });
+    this.options = options;
+    this.uuid = options.uuid || uuid();
+    if (options.planList) {
+      // 配置默认的规划列表
+      var time = new Date().getTime();
+      var llmContent = "```bash" + "\n".concat(options.planList.reduce((pre, cur) => {
+        return (pre ? pre + " && " : pre) + "node ".concat(cur);
+      }, "")) + "\n```";
+      this.setStartTime(time);
+      this.setEndTime(time);
+      this.setLlmContent(llmContent);
+      this.setLoading(false);
+      this.setCommands(parseBashCommands(llmContent).map(argv => {
+        return {
+          startTime: 0,
+          endTime: 0,
+          argv,
+          status: null,
+          tool: {
+            name: argv[1],
+            displayName: argv[1]
+          },
+          content: {
+            llm: "",
+            display: "",
+            response: ""
+          }
+        };
+      }), false);
+      this.defaultPlanList = true;
+    }
+    // 设置userMessage
+    this.events.emit("userMessage", this.getUserMessage());
+  }
+
+  /** 随机ID，保证唯一性 */
+
+  get id() {
+    return this.uuid;
+  }
+
+  /** 开始执行 */
+  run() {
+    var _this = this;
+    return _asyncToGenerator(function* () {
+      // 记录开始时间
+      _this.setStartTime(new Date().getTime());
+      yield _this.start();
+    })();
+  }
+  start() {
+    var _this2 = this;
+    return _asyncToGenerator(function* () {
+      _this2.setStatus("pending");
+      if (!_this2.commands.length) {
+        // 没有commands，需要规划
+        _this2.setLoading(true);
+        yield _this2.tryCatch(() => retry(() => {
+          return _this2.planning();
+        }, _this2.options.requestInstance.maxRetries, error => {
+          return error instanceof RxaiError && error.type === "request";
+        }), true);
+        _this2.setLoading(false);
+      }
+      yield _this2.executeCommands();
+
+      // 执行结束，调用emits回调通知调用方
+      if (_this2.status === "success") {
+        _this2.options.emits.complete("");
+      } else if (_this2.status === "error") {
+        _this2.options.emits.error("");
+      }
+      _this2.summary();
+    })();
+  }
+
+  /** 设置需缓存的值 */
+  setStartTime(startTime) {
+    this.startTime = startTime;
+    this.idbPubContent("startTime", startTime);
+  }
+  setEndTime(endTime) {
+    this.endTime = endTime;
+    this.idbPubContent("endTime", endTime);
+  }
+  setLlmContent(llmContent) {
+    this.llmContent = llmContent;
+    this.idbPubContent("llmContent", llmContent);
+  }
+  setLoading(loading) {
+    this.loading = loading;
+    this.events.emit("loading", loading);
+  }
+
+  /** 构建用户偏好信息提示词 */
+  buildGuidePromptSection() {
+    return this.options.guidePrompt ? "<\u7528\u6237\u504F\u597D\u4FE1\u606F>\n\u5173\u4E8E\u5F53\u524D\u9879\u76EE\uFF0C\u7528\u6237\u63D0\u4F9B\u4E86\u4ED6\u7684\u504F\u597D\u4FE1\u606F\uFF0C\u8BF7\u6CE8\u610F\u53C2\u8003\u504F\u597D\u4FE1\u606F\u6765\u5B8C\u6210\u4EFB\u52A1\u3002\n".concat(this.options.guidePrompt, "\n</\u7528\u6237\u504F\u597D\u4FE1\u606F>\n") : "";
+  }
+  setCommands(commands, sync) {
+    this.commands = commands.map(command => {
+      if (!command.events) {
+        command.events = new Events();
+      }
+      return _objectSpread2({}, command);
+    });
+    this.events.emit("commands", commands);
+    if (sync) {
+      this.idbPubContent("commands", commands.map(command => {
+        var {
+            events
+          } = command,
+          other = _objectWithoutProperties(command, _excluded);
+        return other;
+      }));
+    }
+  }
+  idbPubContent(type, content) {
+    var _this$error, _this$options$idb;
+    // TODO error类型扩展
+    if (((_this$error = this.error) === null || _this$error === void 0 ? void 0 : _this$error.message) === "已销毁") {
+      return;
+    }
+    (_this$options$idb = this.options.idb) === null || _this$options$idb === void 0 || _this$options$idb.putContent({
+      id: this.uuid,
+      type,
+      content
+    });
+  }
+
+  /** 规划 */
+  planning() {
+    var _this3 = this;
+    return _asyncToGenerator(function* () {
+      var {
+        options
+      } = _this3;
+      var planningStream = getPlanningStream(message => {
+        _this3.events.emit("planningMessage", message);
+      });
+      var planningResponse = yield _this3.request({
+        messages: yield _this3.getLLMMessages({
+          start: [{
+            role: "system",
+            content: getSystemPrompt({
+              title: "MyBricks.ai",
+              tools: options.tools,
+              prompt: _this3.system.prompt,
+              guidePromptSection: _this3.buildGuidePromptSection() 
+            })
+          }, ..._this3.getHistoryMessages()]
+        }),
+        emits: _this3.getEmits({
+          write: chunk => {
+            planningStream(chunk);
+          }
+        })
+      });
+      if (planningResponse instanceof RxaiError) {
+        // 规划出错
+        return;
+      }
+      _this3.setLlmContent(planningResponse);
+      var bashCommands = parseBashCommands(planningResponse);
+      if (!bashCommands.length) {
+        // 说明没有规划
+        // this.events.emit("summary", planningResponse);
+        _this3.setStatus("success");
+      } else {
+        var {
+          planningCheck
+        } = _this3.options;
+        if (planningCheck) {
+          var check = planningCheck(bashCommands);
+          if (!check) {
+            throw new RxaiError("规划结果不符合预期", "request");
+          }
+          bashCommands = check;
+        }
+        if (bashCommands.length === 1 && bashCommands[0][1] === "get-history-records") {
+          bashCommands.push(["node", "analyse-and-answer", {}]);
+        }
+        _this3.setCommands(bashCommands.map(argv => {
+          return {
+            startTime: 0,
+            endTime: 0,
+            argv,
+            status: null,
+            tool: {
+              name: argv[1],
+              displayName: argv[1]
+            },
+            content: {
+              llm: "",
+              display: "",
+              response: ""
+            }
+          };
+        }), true);
+      }
+      _this3.setEndTime(new Date().getTime());
+    })();
+  }
+
+  /** 执行规划的脚本 */
+  executeCommands() {
+    var _this4 = this;
+    return _asyncToGenerator(function* () {
+      // 命令
+      var commands = _this4.commands;
+      if (!commands.length) {
+        // 没有命令
+        return;
+      }
+
+      // 当前执行到第几个
+      var index = commands.findIndex(command => command.status !== "success");
+      if (index === -1) {
+        // 命令全部执行完成
+        return;
+      }
+      var _loop = function* _loop() {
+        var command = commands[index];
+
+        // 变更状态，开始执行
+        command.status = "pending";
+        command.startTime = new Date().getTime();
+        _this4.setCommands(commands, false);
+        var [error, response] = yield _this4.tryCatch(/*#__PURE__*/_asyncToGenerator(function* () {
+          return yield retry(() => {
+            return _this4.executeCommand(command);
+          }, _this4.requestInstance.maxRetries,
+          // 暂时都用request配置的maxRetries
+          error => {
+            return error instanceof RxaiError && error.type === "request";
+          });
+        }), true);
+        if (response === CATCH_EMPTY) {
+          command.status = "error";
+          if (error instanceof RxaiError && error.type === "tool") {
+            Object.assign(command.content, {
+              llm: error.message,
+              display: error.display
+            });
+          } else if (error instanceof RxaiError && error.type === "request" || error instanceof Error) {
+            var message = error.message;
+            Object.assign(command.content, {
+              llm: message,
+              display: message
+            });
+          } else {
+            var _message = "工具调用错误";
+            Object.assign(command.content, {
+              llm: _message,
+              display: _message
+            });
+          }
+          _this4.setError(error);
+        } else {
+          command.status = "success";
+          Object.assign(command.content, response);
+          _this4.setCommands(commands, true);
+        }
+        command.endTime = new Date().getTime();
+        if (_this4.status === "pending") {
+          index++;
+          // TODO: idb记录状态
+          if (commands.length === index) {
+            // 所有工具都执行完成，设置完成状态
+            _this4.setStatus("success");
+
+            // this.events.emit(
+            //   "summary",
+            //   command.content.display,
+            // );
+          }
+        }
+      };
+      while (commands.length !== index && _this4.status === "pending") {
+        yield* _loop();
+      }
+    })();
+  }
+
+  /**
+   * 执行命令
+   * 目前均为node命令，后续可能扩展
+   */
+  executeCommand(command) {
+    var _this5 = this;
+    return _asyncToGenerator(function* () {
+      var {
+        argv
+      } = command;
+      var [, name, params = {}] = argv;
+
+      // 已经前置校验过工具合法性，所以tool一定是有的
+      var tool = _this5.options.tools.find(tool => {
+        return tool.name === name;
+      });
+      command.tool.displayName = tool.displayName;
+
+      /** 工具提示词 */
+      var toolPrompt = getToolPrompt(tool, {
+        attachments: _this5.options.attachments,
+        params,
+        guidePromptSection: _this5.buildGuidePromptSection() 
+      });
+      var content = {
+        llm: "",
+        display: "",
+        response: ""
+      };
+      var toolExecute = /*#__PURE__*/function () {
+        var _ref2 = _asyncToGenerator(function* (tool, params) {
+          var [error, response] = yield _this5.tryCatch(() => {
+            if (tool.name === "get-history-records") {
+              // @ts-ignore
+              _this5.filenames = tool.execute(params);
+              return "已读取历史对话记录";
+            }
+            return tool.execute(params);
+          });
+          if (response === CATCH_EMPTY) {
+            throw error;
+          }
+          if (typeof response === "string") {
+            return {
+              llm: response,
+              display: response
+            };
+          } else {
+            return {
+              llm: response.llmContent,
+              display: response.displayContent
+            };
+          }
+        });
+        return function toolExecute(_x, _x2) {
+          return _ref2.apply(this, arguments);
+        };
+      }();
+      if (!toolPrompt) {
+        // 没有提示词，走本地调用，执行execute
+        Object.assign(content, yield toolExecute(tool, {
+          params,
+          files: [],
+          content: "",
+          replaceContent: "",
+          getUserMessage: () => _this5.getUserMessage()
+        }));
+      } else {
+        var _tool$hooks, _this5$options$attach, _tool$aiRole;
+        var streamMessage = "";
+        var streamError = null;
+        var stream = tool.stream ? (content, status) => {
+          if (streamError) {
+            return;
+          }
+          if (_this5.status === "error") {
+            var _this5$currentRequest;
+            (_this5$currentRequest = _this5.currentRequestCancel) === null || _this5$currentRequest === void 0 || _this5$currentRequest.call(_this5);
+            streamError = _this5.error;
+            return;
+          }
+          try {
+            var {
+              content: _replaceContent,
+              files: _files
+            } = parseFileBlocks(content);
+            var res = tool.stream({
+              files: _files,
+              status,
+              replaceContent: _replaceContent
+            });
+            if (typeof res === "string") {
+              var _command$events;
+              (_command$events = command.events) === null || _command$events === void 0 || _command$events.emit("streamMessage", {
+                message: res,
+                status
+              });
+            }
+          } catch (e) {
+            var _this5$currentRequest2;
+            streamError = e;
+            (_this5$currentRequest2 = _this5.currentRequestCancel) === null || _this5$currentRequest2 === void 0 || _this5$currentRequest2.call(_this5);
+          }
+        } : (content, status) => {
+          var _command$events2;
+          (_command$events2 = command.events) === null || _command$events2 === void 0 || _command$events2.emit("streamMessage", {
+            message: content,
+            status
+          });
+        };
+        stream === null || stream === void 0 || stream("", "start");
+
+        // 执行 before 钩子
+        if ((_tool$hooks = tool.hooks) !== null && _tool$hooks !== void 0 && _tool$hooks.before && typeof tool.hooks.before === "function") {
+          yield tool.hooks.before({
+            params
+          });
+        }
+        var historyMessages = _this5.getHistoryMessages(_this5.filenames);
+        var llmMessages = yield _this5.getLLMMessages({
+          start: [{
+            role: "system",
+            content: getToolPrompt(tool, {
+              attachments: _this5.options.attachments,
+              params,
+              guidePromptSection: _this5.buildGuidePromptSection() 
+            })
+          }, ...historyMessages]
+        });
+
+        // 检查是否有附件：当前请求的附件 + 历史记录中带进来的附件
+        // 只要 content 是数组结构就说明包含了附件
+        var hasHistoryAttachments = historyMessages.some(msg => {
+          return Array.isArray(msg.content);
+        });
+        var hasAttachments = !!((_this5$options$attach = _this5.options.attachments) !== null && _this5$options$attach !== void 0 && _this5$options$attach.length || hasHistoryAttachments);
+        var response = yield _this5.request({
+          messages: llmMessages,
+          emits: _this5.getEmits({
+            write: chunk => {
+              // if (tool.streamThoughts) {
+              //   this.events.emit("streamMessage", chunk);
+              // }
+
+              streamMessage += chunk;
+              command.content.response = streamMessage;
+              stream === null || stream === void 0 || stream(streamMessage, "ing");
+              // if (!stream) {
+              //   command.events?.emit("streamMessage", {
+              //     message: streamMessage,
+              //     status: "ing",
+              //   });
+              //   this.events.emit("streamMessage", chunk);
+              // }
+            },
+            complete: content => {
+              stream === null || stream === void 0 || stream(content, "complete");
+            }
+          }),
+          aiRole: typeof tool.aiRole === "function" ? (_tool$aiRole = tool.aiRole) === null || _tool$aiRole === void 0 ? void 0 : _tool$aiRole.call(tool, {
+            params,
+            hasAttachments
+          }) : tool.aiRole
+        });
+        if (streamError) {
+          throw streamError;
+        }
+        if (response instanceof RxaiError) {
+          throw response;
+        }
+
+        // 解析文件
+        var {
+          files,
+          content: replaceContent
+        } = parseFileBlocks(response);
+        Object.assign(content, yield toolExecute(tool, {
+          params,
+          files,
+          content: response,
+          replaceContent,
+          getUserMessage: () => _this5.getUserMessage()
+        }), {
+          response
+        });
+      }
+      return content;
+    })();
+  }
+
+  /** emits代理 */
+  getEmits(emits) {
+    var {
+      options
+    } = this;
+    return {
+      write: chunk => {
+        var _emits$write;
+        options.emits.write(chunk);
+        emits === null || emits === void 0 || (_emits$write = emits.write) === null || _emits$write === void 0 || _emits$write.call(emits, chunk);
+      },
+      complete: content => {
+        var _emits$complete;
+        emits === null || emits === void 0 || (_emits$complete = emits.complete) === null || _emits$complete === void 0 || _emits$complete.call(emits, content);
+      },
+      error: _error => {
+        var _emits$error;
+        options.emits.error(_error);
+        // this.setStatus("error");
+        console.error(_error);
+        emits === null || emits === void 0 || (_emits$error = emits.error) === null || _emits$error === void 0 || _emits$error.call(emits, _error);
+      },
+      cancel: fn => {
+        var _emits$cancel;
+        options.emits.cancel(fn);
+        emits === null || emits === void 0 || (_emits$cancel = emits.cancel) === null || _emits$cancel === void 0 || _emits$cancel.call(emits, fn);
+        this.currentRequestCancel = fn;
+      }
+    };
+  }
+
+  /** 获取用户需求 */
+  getUserMessage() {
+    var _options$attachments;
+    var {
+      options
+    } = this;
+    return {
+      role: "user",
+      content: (_options$attachments = options.attachments) !== null && _options$attachments !== void 0 && _options$attachments.length ? [{
+        type: "text",
+        text: options.message
+      }, ...options.attachments.filter(attachement => {
+        return attachement.type === "image";
+      }).map(attachement => {
+        return {
+          type: "image_url",
+          image_url: {
+            url: attachement.content
+          }
+        };
+      })] : options.message
+    };
+  }
+  formatUserMessage(userMessage, formatFunction) {
+    var userTextMessage;
+    var newUserMessage;
+
+    // 提取用户消息内容并创建新的消息对象
+    if (typeof (userMessage === null || userMessage === void 0 ? void 0 : userMessage.content) === "string") {
+      newUserMessage = _objectSpread2({}, userMessage);
+      userTextMessage = userMessage.content;
+    } else if (Array.isArray(userMessage === null || userMessage === void 0 ? void 0 : userMessage.content)) {
+      var _userMessage$content$;
+      newUserMessage = _objectSpread2(_objectSpread2({}, userMessage), {}, {
+        content: [...userMessage.content]
+      });
+      var idx = userMessage.content.findIndex(item => item.type === "text");
+      userTextMessage = (_userMessage$content$ = userMessage.content[idx]) === null || _userMessage$content$ === void 0 ? void 0 : _userMessage$content$.text;
+    }
+
+    // 通过format函数处理文本内容
+    var formattedText = formatFunction(userTextMessage);
+
+    // 将格式化后的内容塞回新消息对象
+    if (typeof (userMessage === null || userMessage === void 0 ? void 0 : userMessage.content) === "string") {
+      newUserMessage.content = formattedText;
+    } else if (Array.isArray(userMessage === null || userMessage === void 0 ? void 0 : userMessage.content)) {
+      var _idx = userMessage.content.findIndex(item => item.type === "text");
+      if (_idx !== -1) {
+        newUserMessage.content[_idx].text = formattedText;
+      }
+    }
+    return newUserMessage;
+  }
+
+  /** 获取对话消息列表 */
+  getLLMMessages(params) {
+    var _this6 = this;
+    return _asyncToGenerator(function* () {
+      var {
+        options
+      } = _this6;
+      var {
+        start,
+        end
+      } = params;
+      // 辅助函数，用于从 argv 构建命令字符串
+      var buildCommandString = argv => {
+        var [bash, name, params = {}] = argv;
+        return "".concat(name, " ").concat(Object.entries(params).reduce((acc, _ref3) => {
+          var [key, value] = _ref3;
+          return acc + "-".concat(key, " ").concat(value, " ");
+        }, "")).trim();
+      };
+
+      // 获取用户原始需求
+      var userMessage = typeof options.formatUserMessage === "function" ? _this6.formatUserMessage(_this6.getUserMessage(), options.formatUserMessage) : _this6.getUserMessage();
+
+      // 如果存在命令，则构建“工具规划”和“执行进度”
+      if (_this6.commands.length > 0) {
+        // 预先找到当前正在执行的命令
+        var currentCommand = _this6.commands.find(c => c.status === "pending");
+        var currentCommandString = currentCommand ? buildCommandString(currentCommand.argv) : "无";
+
+        // --- 构建“工具规划”部分 ---
+        var planningContent = "\n\n---\n## \u5DE5\u5177\u89C4\u5212";
+        planningContent += "\n\u4E3A\u4E86\u5E2E\u52A9\u7528\u6237\u8FBE\u6210\u4E0A\u8FF0\u76EE\u7684\uFF0C\u7CFB\u7EDF\u89C4\u5212\u4E86\u4EE5\u4E0B\u5DE5\u5177\u6765\u5904\u7406\uFF0C\u5F53\u524D\u6B63\u5728\u6267\u884C ".concat(currentCommandString, "\u3002");
+        for (var command of _this6.commands) {
+          if (command.status === null) continue;
+          var statusMap = {
+            success: "[已完成]",
+            pending: "[正在执行]"
+          };
+          var status = statusMap[command.status] || "[待执行]";
+          planningContent += "\n".concat(status, " ").concat(buildCommandString(command.argv));
+        }
+
+        // --- 构建“执行进度”部分 ---
+        var progressContent = "\n\n## \u6267\u884C\u8FDB\u5EA6";
+        for (var _command of _this6.commands) {
+          if (_command.status === null) continue;
+          var commandStr = buildCommandString(_command.argv);
+          switch (_command.status) {
+            case "success":
+              {
+                progressContent += "\n\n[\u5DF2\u5B8C\u6210] ".concat(commandStr);
+                var output = _command.content.llm || _command.content.display;
+                progressContent += "\n----- \u8F93\u51FA\u5185\u5BB9 -----\n".concat(output, "\n----- \u8F93\u51FA\u5185\u5BB9 -----");
+                break;
+              }
+            case "pending":
+              progressContent += "\n\n[\u6B63\u5728\u6267\u884C] ".concat(commandStr);
+              if (_this6.error instanceof RxaiError && _this6.error.type === "tool") {
+                progressContent += "\n\u6267\u884C\u65F6\u51FA\u9519: ".concat(_this6.error.message, "\n\u8BF7\u5206\u6790\u9519\u8BEF\u539F\u56E0\uFF0C\u4FEE\u6B63\u4E0A\u8FF0\u547D\u4EE4\u6216\u91CD\u65B0\u89C4\u5212\u3002");
+              } else {
+                progressContent += "\n请根据工具描述、用户消息、以及前置工具的执行结果，为当前步骤提供输出。";
+              }
+              break;
+            default:
+              progressContent += "\n\n[\u5F85\u6267\u884C] ".concat(commandStr);
+              break;
+          }
+        }
+        // 组合所有内容
+        userMessage = _this6.formatUserMessage(userMessage, msg => msg + planningContent + progressContent);
+      }
+
+      // 附加重试错误信息
+      var retryMessage = [];
+      if (_this6.error instanceof RxaiError && _this6.error.type === "retry") {
+        retryMessage.push({
+          role: "user",
+          // @ts-ignore
+          content: "[\u7CFB\u7EDF\u63D0\u793A] \u4E0A\u6B21\u89C4\u5212\u51FA\u9519\uFF0C\u9519\u8BEF\u4FE1\u606F: ".concat(_this6.error.message, "\u3002\u8BF7\u57FA\u4E8E\u7528\u6237\u9700\u6C42\u91CD\u65B0\u89C4\u5212\u3002")
+        });
+      }
+      var guideMessage = [];
+
+      // 组装最终消息列表
+      var presetMsgs = typeof options.presetMessages === "function" ? yield options.presetMessages() : options.presetMessages;
+      var messages = [...presetMsgs, ...guideMessage, userMessage, ...retryMessage];
+      if (start) {
+        messages.unshift(...start);
+      }
+      if (end) {
+        messages.push(...end);
+      }
+      _this6.setError(null);
+      return messages;
+    })();
+  }
+
+  /** 请求统一封装 */
+  request(params) {
+    var _this7 = this;
+    return _asyncToGenerator(function* () {
+      var response = yield _this7.requestInstance.requestAsStream(_objectSpread2(_objectSpread2({}, params), {}, {
+        enableLog: _this7.enableLog
+      }));
+      if (response.type === "error") {
+        _this7.setError(response.content);
+        return response.content;
+      } else if (response.type === "cancel") {
+        var error = new RxaiError("已取消执行", "request");
+        _this7.setError(error);
+        return error;
+      }
+      return response.content;
+    })();
+  }
+
+  /** 统一错误处理 */
+  setError(error) {
+    if (!error) {
+      // error清理
+      this.error = null;
+      this.events.emit("error", "");
+      this.idbPubContent("error", null);
+      return;
+    }
+    this.setStatus("error");
+    if (error instanceof RxaiError) {
+      this.error = error;
+    } else {
+      // 默认为ToolError
+      this.error = new RxaiError((error === null || error === void 0 ? void 0 : error.message) || "工具调用错误", "tool");
+    }
+    this.idbPubContent("error", this.error.toJSON());
+    this.events.emit("error", this.error.display);
+  }
+
+  /** 设置状态 */
+  setStatus(status) {
+    this.status = status;
+    this.idbPubContent("status", status);
+    this.events.emit("status", status);
+  }
+
+  /** TODO: 获取DB存储的plan静态数据 */
+  getDBContent() {
+    var _this8 = this;
+    return _asyncToGenerator(function* () {
+      var {
+        options
+      } = _this8;
+      var presetMsgs = typeof options.presetMessages === "function" ? yield options.presetMessages() : options.presetMessages;
+      return {
+        uuid: _this8.uuid,
+        extension: options.extension,
+        enableLog: options.enableLog,
+        attachments: options.attachments,
+        message: options.message,
+        presetHistoryMessages: options.presetHistoryMessages,
+        presetMessages: presetMsgs,
+        planList: options.planList,
+        blockId: options.blockId
+      };
+    })();
+  }
+
+  /** TODO: 从DB恢复 */
+  recover(params) {
+    this.enableRetry = false;
+    params.forEach(_ref4 => {
+      var {
+        type,
+        content
+      } = _ref4;
+      if (type === "error") {
+        if (content) {
+          this.error = new RxaiError(content.message, content.type, content.display).recover(content);
+        }
+      } else {
+        // @ts-ignore
+        this[type] = content;
+      }
+    });
+    if (this.status === "pending") {
+      // 未正常完成，设置为取消状态
+      this.setStatus("aborted");
+    }
+    this.setLoading(false);
+    var commands = this.commands;
+    commands.forEach(command => {
+      if (command.status === "pending") {
+        command.status = null;
+      }
+    });
+    this.setCommands(commands, false);
+    if (this.error) {
+      this.events.emit("error", this.error.display);
+    } else {
+      if (this.status === "aborted") {
+        this.events.emit("summary", "已取消");
+      } else if (!commands.length) {
+        this.events.emit("summary", this.llmContent);
+      }
+    }
+    if (commands.length) {
+      var bashIndex = this.llmContent.indexOf("```bash");
+      this.events.emit("planningMessage", this.llmContent.slice(0, bashIndex));
+    }
+  }
+
+  /** 获取扩展参数 */
+  get extension() {
+    return this.options.extension;
+  }
+  get blockId() {
+    return this.options.blockId;
+  }
+
+  /** TODO: 获取当前plan的总结信息 */
+  getMessages() {
+    var _this$error2;
+    if (this.loading || this.status === "pending" || this.messages.length) {
+      return null;
+    }
+
+    // if (!this.summaryMessage) {
+    //   this.summary();
+    // }
+    var message = "";
+    var presetHistoryMessages = this.options.presetHistoryMessages;
+    if (presetHistoryMessages !== null && presetHistoryMessages !== void 0 && presetHistoryMessages.length) {
+      message += "### 系统信息" + "".concat(presetHistoryMessages.reduce((pre, cur) => {
+        return pre + "\n".concat(cur.content);
+      }, ""));
+    }
+    message += "\n\n### 用户消息" + "\n".concat(this.options.message);
+    if (this.commands.length) {
+      message += "\n\n### 工具调用记录" + "".concat(this.commands.reduce((pre, command, index) => {
+        if (!command.status) {
+          return pre;
+        }
+
+        // const isLast = this.commands.length - 1 === index;
+
+        var [bash, name, params = {}] = command.argv;
+        if (command.status === "success") {
+          return pre + "\n- [x] ".concat(bash, " ").concat(name, " ").concat(Object.entries(params).reduce((pre, _ref5) => {
+            var [key, value] = _ref5;
+            return pre + "-".concat(key, " ").concat(value, " ");
+          }, "")) + // (isLast
+          //   ? `\n${command.content.llm || command.content.display}`
+          //   : "")
+          "\nstdout\uFF1A".concat(command.content.llm || command.content.display);
+        } else if (command.status === "error") {
+          return pre + "\n- [] ".concat(bash, " ").concat(name, " ").concat(Object.entries(params).reduce((pre, _ref6) => {
+            var [key, value] = _ref6;
+            return pre + "-".concat(key, " ").concat(value, " ");
+          }, "")) + "\nzsh\uFF1A".concat(command.content.llm || command.content.display);
+        }
+        return pre;
+      }, ""));
+    }
+    message += "\n\n### 状态" + (this.status === "success" ? "\nsuccess".concat(!this.commands.length ? "\uFF1A".concat(this.llmContent) : "") : "") + (this.status === "aborted" ? "\naborted：执行中断" : "") + (this.status === "error" ? "\nerror\uFF1A".concat((_this$error2 = this.error) === null || _this$error2 === void 0 ? void 0 : _this$error2.message) : "");
+    return {
+      message,
+      summaryMessage: this.summaryMessage ? "".concat(presetHistoryMessages !== null && presetHistoryMessages !== void 0 && presetHistoryMessages.length ? presetHistoryMessages.reduce((pre, cur) => {
+        return pre + "\n".concat(cur.content);
+      }, "") + "\n" : "") + this.summaryMessage : "",
+      attachments: this.options.attachments
+    };
+  }
+
+  /** 安全执行 */
+  tryCatch(task) {
+    var _arguments = arguments,
+      _this9 = this;
+    return _asyncToGenerator(function* () {
+      var log = _arguments.length > 1 && _arguments[1] !== undefined ? _arguments[1] : false;
+      try {
+        var result = task();
+        if (result instanceof Promise) {
+          return [undefined, yield result];
+        }
+        return [undefined, result];
+      } catch (e) {
+        var _this9$error;
+        // TODO error类型扩展
+        if (log && ((_this9$error = _this9.error) === null || _this9$error === void 0 ? void 0 : _this9$error.message) !== "已销毁") {
+          console.error("[Rxai - planning - error]", e);
+        }
+        return [e, CATCH_EMPTY];
+      }
+    })();
+  }
+  retry() {
+    var _this0 = this;
+    return _asyncToGenerator(function* () {
+      if (_this0.error instanceof RxaiError && _this0.error.type === "retry") {
+        // 从意图识别开始
+        if (_this0.defaultPlanList) {
+          // 有默认配置，重制commands
+          _this0.setCommands(parseBashCommands(_this0.llmContent).map(argv => {
+            return {
+              startTime: 0,
+              endTime: 0,
+              argv,
+              status: null,
+              tool: {
+                name: argv[1],
+                displayName: argv[1]
+              },
+              content: {
+                llm: "",
+                display: "",
+                response: ""
+              }
+            };
+          }), true);
+        } else {
+          // 清空规划
+          _this0.setLlmContent("");
+          _this0.setCommands([], true);
+        }
+
+        // this.setError(null);
+      }
+      yield _this0.start();
+    })();
+  }
+  export() {
+    var _this1 = this;
+    return _asyncToGenerator(function* () {
+      var {
+        options
+      } = _this1;
+      var presetMsgs = typeof options.presetMessages === "function" ? yield options.presetMessages() : options.presetMessages;
+      return {
+        uuid: _this1.uuid,
+        options: {
+          attachments: options.attachments,
+          extension: options.extension,
+          message: options.message,
+          presetHistoryMessages: options.presetHistoryMessages,
+          presetMessages: presetMsgs
+        },
+        commands: _this1.commands,
+        defaultPlanList: _this1.defaultPlanList,
+        enableLog: _this1.enableLog,
+        enableRetry: _this1.enableRetry,
+        endTime: _this1.endTime,
+        error: _this1.error ? _this1.error.toJSON() : undefined,
+        llmContent: _this1.llmContent,
+        startTime: _this1.startTime,
+        status: _this1.status,
+        summaryMessage: _this1.summaryMessage
+      };
+    })();
+  }
+  summary() {
+    if (this.summaryLoading) {
+      return;
+    }
+    if (this.summaryMessage) {
+      return;
+    }
+    this.summaryLoading = true;
+    var historyMessage = this.getMessages();
+    if (!historyMessage) {
+      return;
+    }
+    var {
+      message
+    } = historyMessage;
+    this.requestInstance.requestAsStream({
+      messages: [{
+        role: "system",
+        content: "\u8BF7\u5BF9\u4EE5\u4E0B\u5BF9\u8BDD\u5386\u53F2\u8BB0\u5F55\u8FDB\u884C**\u4E13\u4E1A\u6458\u8981**\uFF0C\u63D0\u53D6\u6838\u5FC3\u64CD\u4F5C\u4E0E\u7ED3\u679C\uFF0C\u8BED\u8A00\u7CBE\u70BC\uFF0C\u9002\u5408\u5B58\u6863\u548C\u5FEB\u901F\u67E5\u9605\u3002\n\n---\n\n**\u6458\u8981\u8981\u6C42\uFF1A**\n1. **\u7528\u6237\u610F\u56FE**\uFF1A\u7528\u4E00\u53E5\u8BDD\u603B\u7ED3\u7528\u6237\u7684\u539F\u59CB\u6307\u4EE4\u6216\u8BF7\u6C42\u3002\n2. **\u5173\u952E\u64CD\u4F5C**\uFF1A\u5217\u51FA\u6267\u884C\u7684\u4E3B\u8981\u52A8\u4F5C\uFF08\u5DE5\u5177\u8C03\u7528\u548C\u91CD\u8981\u4FEE\u6539\uFF09\u3002\n3. **\u6267\u884C\u7ED3\u679C**\uFF1A\u6982\u62EC\u4EFB\u52A1\u6267\u884C\u7684\u5173\u952E\u4EA7\u51FA\u6216\u53D8\u5316\u3002\n4. **\u6700\u7EC8\u72B6\u6001**\uFF1A\u8BF4\u660E\u4EFB\u52A1\u5B8C\u6210\u60C5\u51B5\u3002\n\n---\n\n**\u8F93\u51FA\u683C\u5F0F\u793A\u4F8B\uFF1A**\n```\n\u3010\u7528\u6237\u610F\u56FE\u3011\u7528\u6237\u5E0C\u671B\u5B9E\u73B0\u67D0\u4E2A\u5177\u4F53\u76EE\u6807\u3002\n\u3010\u5173\u952E\u64CD\u4F5C\u3011\u8C03\u7528\u4E86\u5DE5\u5177A\u548CB\u3002\n\u3010\u6267\u884C\u7ED3\u679C\u3011\u4EA7\u751F\u4E86Z\u53D8\u5316\u6216\u8FBE\u6210\u4E86W\u6548\u679C\u3002\n\u3010\u6700\u7EC8\u72B6\u6001\u3011\u6210\u529F\u5B8C\u6210/\u90E8\u5206\u5B8C\u6210/\u5931\u8D25\u3002\n```\n\n---\n\n**\u6CE8\u610F\uFF1A**\n1. \u6458\u8981\u7ED3\u679C\u5FC5\u987B\u7CBE\u7B80\uFF0C\u603B\u5B57\u6570\u4E0D\u8D85\u8FC750\u5B57\u3002\n\n---\n\n\u8BF7\u6839\u636E\u4EE5\u4E0A\u8981\u6C42\uFF0C\u76F4\u63A5\u7ED9\u51FA\u6E05\u6670\u3001\u7B80\u660E\u7684\u6458\u8981\u7ED3\u679C\u3002"
+      }, {
+        role: "user",
+        content: "<\u5BF9\u8BDD\u5386\u53F2\u8BB0\u5F55>\n".concat(message, "\n</\u5BF9\u8BDD\u5386\u53F2\u8BB0\u5F55>")
+      }],
+      emits: {
+        write() {},
+        cancel() {},
+        error() {},
+        complete() {}
+      }
+    }).then(res => {
+      if (res.type === "complete") {
+        this.summaryMessage = res.content;
+        this.idbPubContent("summaryMessage", res.content);
+      }
+    }).catch(e => {
+      console.error(e);
+    }).finally(() => {
+      this.summaryLoading = false;
+    });
+    return {
+      message,
+      attachments: this.options.attachments
+    };
+  }
+  destroy() {
+    var _this$currentRequestC;
+    this.status = "error";
+    this.error = new RxaiError("已销毁", "tool");
+    (_this$currentRequestC = this.currentRequestCancel) === null || _this$currentRequestC === void 0 || _this$currentRequestC.call(this);
+  }
+}
+function parseBashCommands(string) {
+  var bashCodeRegex = /```\s*bash([\s\S]*?)```/;
+  var matchResult = string.match(bashCodeRegex);
+  if (!matchResult) return [];
+  var bashContent = matchResult[1].trim();
+  var subCommands = bashContent.split("&&").map(cmd => cmd.trim());
+  var commandArray = subCommands.map(cmd => cmd.split(/\s+/).filter(Boolean));
+  var result = [];
+  commandArray.forEach(command => {
+    var [node, filename, ...args] = command;
+    var params = {};
+    var key = "";
+    args.forEach(arg => {
+      if (arg.startsWith("-")) {
+        // 支持 -v 或 --option 格式，移除前面的 - 或 --
+        key = arg.replace(/^--?/, "");
+      } else if (key) {
+        // 当前面有key时，当前值作为参数值
+        params[key] = arg;
+        key = ""; // 重置key，准备接收下一个参数
+      }
+    });
+    result.push([node, filename, params]);
+  });
+  return result;
+}
+function getPlanningStream(write) {
+  var stopWrite = false;
+  var planningMessage = "";
+  var temp = "";
+  return chunk => {
+    if (stopWrite) {
+      return;
+    }
+    var tempChunk = temp + chunk;
+    var backticksIndex = tempChunk.indexOf("`");
+    if (temp && tempChunk.length >= 7) {
+      if (new RegExp("^".concat(tempChunk.slice(0, 7))).test("```bash")) {
+        stopWrite = true;
+      } else {
+        temp = "";
+        planningMessage += tempChunk;
+      }
+    } else {
+      if (backticksIndex !== -1) {
+        if (tempChunk.slice(backticksIndex, backticksIndex + 7) === "```bash") {
+          stopWrite = true;
+          planningMessage += tempChunk.slice(0, backticksIndex);
+        } else {
+          planningMessage += tempChunk.slice(0, backticksIndex);
+          temp = tempChunk.slice(backticksIndex);
+        }
+      } else {
+        planningMessage += chunk;
+      }
+    }
+    var bashIndex = planningMessage.indexOf("```bash");
+    if (bashIndex !== -1) {
+      stopWrite = true;
+      planningMessage = planningMessage.slice(0, bashIndex);
+    }
+    write(planningMessage);
+  };
+}
+
+export { PlanningAgent };
+//# sourceMappingURL=planning.js.map