p-api-agent 0.0.2 → 0.0.3

package/dist/index.mjs

@@ -1,369 +0,0 @@
-var __defProp = Object.defineProperty;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
-  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
-}) : x)(function(x) {
-  if (typeof require !== "undefined") return require.apply(this, arguments);
-  throw Error('Dynamic require of "' + x + '" is not supported');
-});
-var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
-
-// src/agent/index.ts
-import fs from "fs";
-
-// src/utils/llm_utils.ts
-import { jsonrepair } from "jsonrepair";
-var _LLM_Utils = class {
-  /**
-   * 提取大模型返回的markdown
-   */
-  extract_markdown(content) {
-    const regex = /```markdown\n(.*?)\n```/s;
-    const match = content.match(regex);
-    if (!match) return content;
-    return match[1];
-  }
-  /**
-   * 提取大模型的json格式
-   */
-  extract_json(content) {
-    const regex = /```json\n(.*?)\n```/s;
-    const match = content.match(regex);
-    let json_content = match ? match[1] : content;
-    try {
-      return JSON.parse(json_content);
-    } catch (err) {
-      return JSON.parse(jsonrepair(json_content));
-    }
-  }
-  /**
-   * 提取mermaid格式
-   */
-  extract_mermaid(content) {
-    const regex = /```mermaid\n(.*?)\n```/s;
-    const match = content.match(regex);
-    if (!match) return content;
-    return match[1];
-  }
-  /**
-   * 提取svg格式
-   */
-  extract_svg(content) {
-    const regex = /```svg\n(.*?)\n```/s;
-    const match = content.match(regex);
-    if (!match) return content;
-    return match[1];
-  }
-  parse_json(content) {
-    if (typeof content === "object") {
-      return content;
-    }
-    try {
-      return JSON.parse(content);
-    } catch (err) {
-      return JSON.parse(jsonrepair(content));
-    }
-  }
-};
-var LLM_Utils = new _LLM_Utils();
-
-// src/agent/index.ts
-import path from "path";
-var Agent = class {
-  // 角色定义的提示词
-  constructor() {
-    __publicField(this, "function_call", null);
-    // 工具函数的说明文档, 供模型调用时参考
-    __publicField(this, "function_call_doc", "");
-    // 模型的聊天能力
-    __publicField(this, "llm_chat_func", null);
-    // 预置提示词
-    __publicField(this, "preset_prompt", this.get_preset_prompt());
-  }
-  /**
-   * 注册工具函数
-   */
-  register_function_call(function_call) {
-    const tools_list = function_call.get_tools_list();
-    this.function_call_doc = tools_list.map((val) => `\u5DE5\u5177\u51FD\u6570:${val.name}
-\u5DE5\u5177\u63CF\u8FF0:${val.description}`).join("\n\n");
-    this.function_call = function_call;
-    return true;
-  }
-  /**
-   * 注册LLM模型文字能力
-   */
-  register_llm_text_ability(func) {
-    this.llm_chat_func = func;
-  }
-  /**
-  * 注册LLM模型图片能力
-  */
-  register_llm_image_ability() {
-    return true;
-  }
-  /**
-   * 判断是否需要调用function_call, 以及调用哪个工具函数
-   */
-  async check_whether_use_function_call(user_input) {
-    const role_prompt = `
-        \u6211\u662F\u4E00\u4E2A\u667A\u80FD\u52A9\u624B, \u6211\u4F1A\u5224\u65AD\u7528\u6237\u8F93\u5165\u662F\u5426\u9700\u8981\u8C03\u7528\u5DE5\u5177\u51FD\u6570\u6765\u8F85\u52A9\u5B8C\u6210\u4EFB\u52A1
-        - \u5982\u679C\u9700\u8981\u8C03\u7528, \u6211\u4F1A\u9996\u5148\u5224\u65AD\u81EA\u5DF1\u662F\u5426\u80FD\u89E3\u51B3,
-        - \u5982\u679C\u80FD\u89E3\u51B3, \u6211\u5C31\u8FD4\u56DE {"command": "no_use", "result": "\u4E0D\u9700\u8981\u8C03\u7528\u5DE5\u5177\u51FD\u6570"}
-        - \u5982\u679C\u4E0D\u80FD\u89E3\u51B3, \u6211\u5C31\u4F1A\u6839\u636E\u5DF2\u6709\u7684\u5DE5\u5177\u51FD\u6570\u5217\u8868\u9009\u62E9\u5DE5\u5177\u51FD\u6570 \u5E76\u4E14\u6839\u636E\u7528\u6237\u8F93\u5165\u7684\u5DE5\u5177\u51FD\u6570\u8BF4\u660E\u6587\u6863\u751F\u6210\u8C03\u7528\u53C2\u6570:
-        - \u6211\u7EDD\u5BF9\u4E0D\u4F1A\u65E0\u4E2D\u751F\u6709\u4E00\u4E2A\u5DE5\u5177\u51FD\u6570
-        - \u6211\u4E0D\u4F1A\u9020\u5047\u6570\u636E / \u865A\u62DF\u6570\u636E / \u634F\u9020\u6570\u636E
-        - \u6CA1\u6709\u7684\u5DE5\u5177\u51FD\u6570,\u6211\u4F1A\u544A\u77E5\u4F60, \u65E0\u6CD5\u6EE1\u8DB3\u4F60\u7684\u64CD\u4F5C
-        ## \u6211\u53EF\u4EE5\u8C03\u7528\u5DE5\u5177\u51FD\u6570\u5217\u8868:
-        ---
-        ${this.function_call_doc}
-
-        ## \u8F93\u51FA\u8981\u6C42
-        ---
-        \u5982\u679C\u9700\u8981\u8C03\u7528\u5DE5\u5177\u51FD\u6570, \u6211\u4F1A\u8FD4\u56DE{"command": "get_tool_doc", "tool_name": "\u5DE5\u5177\u51FD\u6570\u540D\u79F0"}
-        \u5982\u679C\u4E0D\u9700\u8981\u8C03\u7528, \u8BF7\u4F1A\u8FD4\u56DE{"command": "no_use", "result": "\u4E0D\u9700\u8981\u8C03\u7528\u5DE5\u5177\u51FD\u6570"}
-        \u5982\u679C\u81EA\u8EAB\u6CA1\u80FD\u529B\u5904\u7406,\u4E14\u6CA1\u6709\u5408\u9002\u7684\u5DE5\u5177\u51FD\u6570, \u8BF7\u4F1A\u8FD4\u56DE{"command": "no_ability", "result": "\u6211\u6CA1\u80FD\u529B\u6267\u884C\u8FD9\u4E2A\u64CD\u4F5C"}
-        ---
-        `;
-    const check_use_prompt = [
-      {
-        role: "system",
-        content: [
-          { type: "text", text: role_prompt }
-        ]
-      },
-      ...this.deepClone(user_input)
-    ];
-    const check_res = await this.llm_chat_func([
-      ...check_use_prompt,
-      {
-        role: "user",
-        content: [
-          {
-            type: "text",
-            text: "\u4F60\u5FC5\u987B\u8FD4\u56DEJSON\u683C\u5F0F, \u4E0D\u80FD\u8FD4\u56DE\u591A\u4F59\u5B57\u7B26, \u4E0D\u7136\u6211\u4F1A\u89E3\u6790\u5931\u8D25\u7684"
-          }
-        ]
-      }
-    ]);
-    const check_res_json = LLM_Utils.extract_json(check_res);
-    if (check_res_json.command === "no_use" || check_res_json.command === "no_ability") {
-      return check_res_json;
-    }
-    check_use_prompt.push({
-      role: "assistant",
-      content: [
-        { type: "text", text: check_res }
-      ]
-    });
-    const tool_name = check_res_json.tool_name;
-    const doc = await this.function_call.gen_tool_doc(tool_name);
-    const gen_params_prompt = `
-        ## \u5DE5\u5177\u51FD\u6570
-        ${tool_name} 
-        ## \u5DE5\u5177\u51FD\u6570\u8BF4\u660E\u6587\u6863:
-        ---
-        ${doc}
-        ---
-        \u8BF7\u4E25\u683C\u6309\u7167\u5DE5\u5177\u51FD\u6570\u7684\u8F93\u5165\u8981\u6C42, \u6839\u636E\u7528\u6237\u8F93\u5165\u548C\u5DE5\u5177\u51FD\u6570\u8BF4\u660E\u6587\u6863\u751F\u6210\u8C03\u7528\u53C2\u6570,
-        \u8F93\u51FA\u683C\u5F0F\u5FC5\u987B\u4E3Ajson\u683C\u5F0F, \u4F8B\u5982{"\u53C2\u65701": "\u503C1", "\u53C2\u65702": "\u503C2"}, \u4E0D\u80FD\u8FD4\u56DE\u591A\u4F59\u5B57\u7B26
-        `;
-    check_use_prompt.push({
-      role: "user",
-      content: [
-        { type: "text", text: gen_params_prompt },
-        { type: "text", text: "\u4F60\u5FC5\u987B\u8FD4\u56DEJSON\u683C\u5F0F, \u4E0D\u80FD\u8FD4\u56DE\u591A\u4F59\u5B57\u7B26, \u4E0D\u7136\u6211\u4F1A\u89E3\u6790\u5931\u8D25\u7684" }
-      ]
-    });
-    const gen_params_res = await this.llm_chat_func(check_use_prompt);
-    const gen_params_json = LLM_Utils.extract_json(gen_params_res);
-    return {
-      command: "use_tool",
-      tool_name,
-      params: gen_params_json
-    };
-  }
-  /**
-   * 创建聊天
-   */
-  async create_chat(user_input) {
-    const dispose_user_input = this.pre_dispose_user_input(user_input);
-    let max_loop = 20;
-    let loop_count = 1;
-    const return_data = {
-      result: "",
-      use_tools: []
-    };
-    while (loop_count <= max_loop) {
-      const check_use_tools = await this.check_whether_use_function_call(dispose_user_input);
-      if (check_use_tools.command == "use_tool") {
-        const { tool_name, params } = check_use_tools;
-        const tool_result = await this.function_call.exec_function(tool_name, params);
-        dispose_user_input.push({
-          role: "assistant",
-          content: [
-            {
-              type: "text",
-              text: `\u6211\u8981\u8C03\u7528, \u5DE5\u5177\u51FD\u6570${tool_name}
-                        \u8C03\u7528\u53C2\u6570\u4E3A:
-${JSON.stringify(params)}`
-            }
-          ]
-        });
-        dispose_user_input.push({
-          role: "user",
-          content: [
-            { type: "text", text: `\u8C03\u7528\u7ED3\u679C: 
-${JSON.stringify(tool_result)}` }
-          ]
-        });
-        loop_count++;
-        return_data.use_tools.push({
-          tool_name,
-          params,
-          exec_result: tool_result
-        });
-      } else if (check_use_tools.command == "no_ability") {
-        return_data.result = "\u65E0\u6CD5\u6267\u884C\u6B64\u64CD\u4F5C";
-        return return_data;
-      } else if (check_use_tools.command == "no_use") {
-        dispose_user_input.push({
-          role: "user",
-          content: [
-            {
-              type: "text",
-              text: "\u4F60\u6839\u636E\u4E0A\u4E0B\u6587\u8FD4\u56DE\u7ED3\u679C\u5C31\u597D, \u4F60\u7684\u601D\u8003\u8FC7\u7A0B\u65E0\u9700\u8FD4\u56DE\u7ED9\u6211, \u4F60\u53EA\u9700\u8981\u56DE\u7B54\u6211\u7684\u95EE\u9898, \u4E0D\u8981\u5C06\u591A\u4F59\u7684\u4FE1\u606F\u8FD4\u56DE\u7ED9\u6211, \u4E0D\u80FD\u6267\u884C\u64CD\u4F5C\u5C31\u56DE\u590D\u65E0\u6CD5\u64CD\u4F5C"
-            }
-          ]
-        });
-        const llm_chat_res = await this.llm_chat_func(dispose_user_input);
-        return_data.result = llm_chat_res;
-        return return_data;
-      } else {
-        return_data.result = "\u5F02\u5E38\u6307\u5B9A";
-        return return_data;
-      }
-    }
-  }
-  /**
-   * agent预置提示词
-   */
-  get_preset_prompt() {
-    const prompt = fs.readFileSync(path.join(__dirname, "./preset_prompt.md"), "utf-8");
-    return prompt;
-  }
-  /**
-   * 预处理用户输入
-   */
-  pre_dispose_user_input(user_input) {
-    if (typeof user_input === "string") {
-      return [
-        {
-          role: "user",
-          content: [
-            { type: "text", text: user_input }
-          ]
-        }
-      ];
-    }
-    return user_input;
-  }
-  /**
-   * 深拷贝方法
-   */
-  deepClone(obj) {
-    if (obj === null || typeof obj !== "object") {
-      return obj;
-    }
-    if (obj instanceof Date) {
-      return new Date(obj.getTime());
-    }
-    if (obj instanceof Array) {
-      const cloneArray = [];
-      obj.forEach((item) => {
-        cloneArray.push(this.deepClone(item));
-      });
-      return cloneArray;
-    }
-    if (obj instanceof Object) {
-      const cloneObj = {};
-      for (const key in obj) {
-        if (obj.hasOwnProperty(key)) {
-          cloneObj[key] = this.deepClone(obj[key]);
-        }
-      }
-      return cloneObj;
-    }
-    throw new Error("Unable to copy object");
-  }
-};
-
-// src/function_call/index.ts
-import path2 from "path";
-import fs2 from "fs";
-var FunctionCall = class {
-  constructor(tool_path) {
-    this.tool_path = tool_path;
-    __publicField(this, "tools", {});
-    this.init();
-  }
-  init() {
-    for (const file_name of fs2.readdirSync(this.tool_path)) {
-      const route_path = path2.join(this.tool_path, file_name);
-      const mcp = __require(route_path);
-      const info = mcp.register();
-      this.tools[info.name] = info;
-    }
-  }
-  /**
-   * 生成完整的工具函数说明文档
-   */
-  gen_tool_doc(name) {
-    const info = this.tools[name];
-    if (!info) {
-      throw `${name}\u5DE5\u5177\u51FD\u6570\u4E0D\u5B58\u5728`;
-    }
-    const schema = info.input_schema;
-    const schemaDesc = `\u7C7B\u578B: ${schema.type}`;
-    const params = Object.keys(schema.properties).map((key) => {
-      const item = schema.properties[key];
-      const required = schema.required?.includes(key) ? "\u5FC5\u586B" : "\u53EF\u9009";
-      const examples = item.examples && item.examples.length > 0 ? `\u793A\u4F8B\u503C: ${item.examples.join(", ")}` : "\u65E0\u793A\u4F8B";
-      return `  - ${key} (${item.type}, ${required})
-    \u8BF4\u660E: ${item.description}
-    ${examples}`;
-    }).join("\n");
-    return `\u5DE5\u5177\u51FD\u6570\u540D\u79F0: ${info.name}
-\u529F\u80FD\u63CF\u8FF0: ${info.description}
-\u8F93\u5165\u7ED3\u6784: ${schemaDesc}
-\u53C2\u6570\u5217\u8868:
-${params}`;
-  }
-  /**
-   * 获取工具函数列表
-   */
-  get_tools_list() {
-    return Object.keys(this.tools).map((key) => {
-      const info = this.tools[key];
-      return {
-        name: info.name,
-        description: info.description
-      };
-    });
-  }
-  /**
-   * 执行工具函数
-   */
-  async exec_function(name, params) {
-    const info = this.tools[name];
-    if (!info) {
-      throw `\u5DE5\u5177\u51FD\u6570${name}\u4E0D\u5B58\u5728`;
-    }
-    return await info.register_func(params);
-  }
-};
-export {
-  Agent,
-  FunctionCall,
-  LLM_Utils
-};