p-api-agent 0.0.1 → 0.0.2

package/dist/{index.d.cts → index.d.mts}

@@ -47,7 +47,7 @@ interface ImageContent {
     };
 }
 type Content = TextContent | ImageContent;
-type UserChatInput = string | {
+type UserChatInput = {
     role: 'system' | 'user' | 'assistant';
     content: Content[];
 }[];
@@ -69,14 +69,29 @@ declare class Agent {
     * 注册LLM模型图片能力
     */
     register_llm_image_ability(): boolean;
+    /**
+     * 判断是否需要调用function_call, 以及调用哪个工具函数
+     */
+    check_whether_use_function_call(user_input: UserChatInput): Promise<any>;
     /**
      * 创建聊天
      */
-    create_chat(user_input: UserChatInput): Promise<any>;
+    create_chat(user_input: string | UserChatInput): Promise<{
+        result: string;
+        use_tools: any[];
+    }>;
     /**
      * agent预置提示词
      */
     get_preset_prompt(): string;
+    /**
+     * 预处理用户输入
+     */
+    pre_dispose_user_input(user_input: string | UserChatInput): UserChatInput;
+    /**
+     * 深拷贝方法
+     */
+    private deepClone;
 }
 
 declare class _LLM_Utils {

package/dist/index.d.ts

@@ -47,7 +47,7 @@ interface ImageContent {
     };
 }
 type Content = TextContent | ImageContent;
-type UserChatInput = string | {
+type UserChatInput = {
     role: 'system' | 'user' | 'assistant';
     content: Content[];
 }[];
@@ -69,14 +69,29 @@ declare class Agent {
     * 注册LLM模型图片能力
     */
     register_llm_image_ability(): boolean;
+    /**
+     * 判断是否需要调用function_call, 以及调用哪个工具函数
+     */
+    check_whether_use_function_call(user_input: UserChatInput): Promise<any>;
     /**
      * 创建聊天
      */
-    create_chat(user_input: UserChatInput): Promise<any>;
+    create_chat(user_input: string | UserChatInput): Promise<{
+        result: string;
+        use_tools: any[];
+    }>;
     /**
      * agent预置提示词
      */
     get_preset_prompt(): string;
+    /**
+     * 预处理用户输入
+     */
+    pre_dispose_user_input(user_input: string | UserChatInput): UserChatInput;
+    /**
+     * 深拷贝方法
+     */
+    private deepClone;
 }
 
 declare class _LLM_Utils {

package/dist/index.js

@@ -1,15 +1,47 @@
-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 __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 __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+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);
+var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  Agent: () => Agent,
+  FunctionCall: () => FunctionCall,
+  LLM_Utils: () => LLM_Utils
 });
+module.exports = __toCommonJS(index_exports);
 
 // src/agent/index.ts
-import fs from "fs";
+var import_fs = __toESM(require("fs"));
 
 // src/utils/llm_utils.ts
-import { jsonrepair } from "jsonrepair";
+var import_jsonrepair = require("jsonrepair");
 var _LLM_Utils = class {
   /**
    * 提取大模型返回的markdown
@@ -30,7 +62,7 @@ var _LLM_Utils = class {
     try {
       return JSON.parse(json_content);
     } catch (err) {
-      return JSON.parse(jsonrepair(json_content));
+      return JSON.parse((0, import_jsonrepair.jsonrepair)(json_content));
     }
   }
   /**
@@ -58,24 +90,24 @@ var _LLM_Utils = class {
     try {
       return JSON.parse(content);
     } catch (err) {
-      return JSON.parse(jsonrepair(content));
+      return JSON.parse((0, import_jsonrepair.jsonrepair)(content));
     }
   }
 };
 var LLM_Utils = new _LLM_Utils();
 
 // src/agent/index.ts
-import path from "path";
+var import_path = __toESM(require("path"));
 var Agent = class {
   // 角色定义的提示词
   constructor() {
-    this.function_call = null;
+    __publicField(this, "function_call", null);
     // 工具函数的说明文档, 供模型调用时参考
-    this.function_call_doc = "";
+    __publicField(this, "function_call_doc", "");
     // 模型的聊天能力
-    this.llm_chat_func = null;
+    __publicField(this, "llm_chat_func", null);
     // 预置提示词
-    this.preset_prompt = this.get_preset_prompt();
+    __publicField(this, "preset_prompt", this.get_preset_prompt());
   }
   /**
    * 注册工具函数
@@ -100,92 +132,214 @@ var Agent = class {
     return true;
   }
   /**
-   * 创建聊天
+   * 判断是否需要调用function_call, 以及调用哪个工具函数
    */
-  async create_chat(user_input) {
-    let dispose_user_input;
-    if (typeof user_input === "string") {
-      dispose_user_input = [
-        {
-          role: "user",
-          content: [
-            { type: "text", text: user_input }
-          ]
-        }
-      ];
-    } else {
-      dispose_user_input = JSON.parse(JSON.stringify(user_input));
-    }
-    if (this.function_call_doc) {
-      dispose_user_input.unshift({
+  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: `# \u5DE5\u5177\u51FD\u6570\u8BF4\u660E\u6587\u6863:
-${this.function_call_doc}` }
+          { 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;
     }
-    dispose_user_input.unshift({
-      role: "system",
+    check_use_prompt.push({
+      role: "assistant",
       content: [
-        { type: "text", text: this.preset_prompt }
+        { 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 = 0;
-    while (loop_count < max_loop) {
-      const llm_res_str = await this.llm_chat_func(dispose_user_input);
-      const llm_res = LLM_Utils.extract_json(llm_res_str);
-      if (!llm_res.command) throw "\u5F02\u5E38\u6307\u4EE4, \u672A\u5305\u542Bcommand\u5B57\u6BB5";
-      if (llm_res.command === "end") {
-        return llm_res.result;
-      } else if (llm_res.command === "get_tool_doc") {
-        const tool_name = llm_res.tool_name;
-        const tool_doc = this.function_call.gen_tool_doc(tool_name);
+    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: "user",
+          role: "assistant",
           content: [
-            { type: "text", text: `\u5DE5\u5177\u51FD\u6570${tool_name}\u7684\u8BF4\u660E\u6587\u6863:
-${tool_doc}` }
+            {
+              type: "text",
+              text: `\u6211\u8981\u8C03\u7528, \u5DE5\u5177\u51FD\u6570${tool_name}
+                        \u8C03\u7528\u53C2\u6570\u4E3A:
+${JSON.stringify(params)}`
+            }
           ]
         });
-      } else if (llm_res.command === "use_tool") {
-        const tool_name = llm_res.tool_name;
-        const tool_params = LLM_Utils.parse_json(llm_res.params);
-        const tool_result = await this.function_call.exec_function(tool_name, tool_params);
         dispose_user_input.push({
           role: "user",
           content: [
-            { type: "text", text: `\u5DE5\u5177\u51FD\u6570${tool_name}\u7684\u8C03\u7528\u7ED3\u679C:
+            { 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;
       }
-      loop_count++;
     }
   }
   /**
    * agent预置提示词
    */
   get_preset_prompt() {
-    const prompt = fs.readFileSync(path.join(__dirname, "./preset_prompt.md"), "utf-8");
+    const prompt = import_fs.default.readFileSync(import_path.default.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 import_path2 = __toESM(require("path"));
+var import_fs2 = __toESM(require("fs"));
 var FunctionCall = class {
   constructor(tool_path) {
     this.tool_path = tool_path;
-    this.tools = {};
+    __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);
+    for (const file_name of import_fs2.default.readdirSync(this.tool_path)) {
+      const route_path = import_path2.default.join(this.tool_path, file_name);
+      const mcp = require(route_path);
       const info = mcp.register();
       this.tools[info.name] = info;
     }
@@ -195,6 +349,9 @@ var FunctionCall = class {
    */
   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) => {
@@ -234,8 +391,9 @@ ${params}`;
     return await info.register_func(params);
   }
 };
-export {
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
   Agent,
   FunctionCall,
   LLM_Utils
-};
+});

package/dist/index.mjs

@@ -0,0 +1,369 @@
+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
+};

package/package.json

@@ -1,7 +1,7 @@
 {
-  "type": "module",
+  "type": "commonjs",
   "name": "p-api-agent",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
   "files": [
@@ -24,7 +24,10 @@
     "@types/lodash": "4.17.16",
     "@types/node": "^25.3.5",
     "javascript-obfuscator": "^4.1.1",
+    "lodash": "^4.17.23",
+    "ts-node": "^10.9.2",
     "tsup": "^8.4.0",
+    "tsx": "^4.21.0",
     "typescript": "5.8.3"
   },
   "dependencies": {

package/dist/index.cjs

@@ -1,273 +0,0 @@
-"use strict";
-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/index.ts
-var index_exports = {};
-__export(index_exports, {
-  Agent: () => Agent,
-  FunctionCall: () => FunctionCall,
-  LLM_Utils: () => LLM_Utils
-});
-module.exports = __toCommonJS(index_exports);
-
-// src/agent/index.ts
-var import_fs = __toESM(require("fs"), 1);
-
-// src/utils/llm_utils.ts
-var import_jsonrepair = require("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((0, import_jsonrepair.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((0, import_jsonrepair.jsonrepair)(content));
-    }
-  }
-};
-var LLM_Utils = new _LLM_Utils();
-
-// src/agent/index.ts
-var import_path = __toESM(require("path"), 1);
-var Agent = class {
-  // 角色定义的提示词
-  constructor() {
-    this.function_call = null;
-    // 工具函数的说明文档, 供模型调用时参考
-    this.function_call_doc = "";
-    // 模型的聊天能力
-    this.llm_chat_func = null;
-    // 预置提示词
-    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;
-  }
-  /**
-   * 创建聊天
-   */
-  async create_chat(user_input) {
-    let dispose_user_input;
-    if (typeof user_input === "string") {
-      dispose_user_input = [
-        {
-          role: "user",
-          content: [
-            { type: "text", text: user_input }
-          ]
-        }
-      ];
-    } else {
-      dispose_user_input = JSON.parse(JSON.stringify(user_input));
-    }
-    if (this.function_call_doc) {
-      dispose_user_input.unshift({
-        role: "system",
-        content: [
-          { type: "text", text: `# \u5DE5\u5177\u51FD\u6570\u8BF4\u660E\u6587\u6863:
-${this.function_call_doc}` }
-        ]
-      });
-    }
-    dispose_user_input.unshift({
-      role: "system",
-      content: [
-        { type: "text", text: this.preset_prompt }
-      ]
-    });
-    let max_loop = 20;
-    let loop_count = 0;
-    while (loop_count < max_loop) {
-      const llm_res_str = await this.llm_chat_func(dispose_user_input);
-      const llm_res = LLM_Utils.extract_json(llm_res_str);
-      if (!llm_res.command) throw "\u5F02\u5E38\u6307\u4EE4, \u672A\u5305\u542Bcommand\u5B57\u6BB5";
-      if (llm_res.command === "end") {
-        return llm_res.result;
-      } else if (llm_res.command === "get_tool_doc") {
-        const tool_name = llm_res.tool_name;
-        const tool_doc = this.function_call.gen_tool_doc(tool_name);
-        dispose_user_input.push({
-          role: "user",
-          content: [
-            { type: "text", text: `\u5DE5\u5177\u51FD\u6570${tool_name}\u7684\u8BF4\u660E\u6587\u6863:
-${tool_doc}` }
-          ]
-        });
-      } else if (llm_res.command === "use_tool") {
-        const tool_name = llm_res.tool_name;
-        const tool_params = LLM_Utils.parse_json(llm_res.params);
-        const tool_result = await this.function_call.exec_function(tool_name, tool_params);
-        dispose_user_input.push({
-          role: "user",
-          content: [
-            { type: "text", text: `\u5DE5\u5177\u51FD\u6570${tool_name}\u7684\u8C03\u7528\u7ED3\u679C:
-${JSON.stringify(tool_result)}` }
-          ]
-        });
-      }
-      loop_count++;
-    }
-  }
-  /**
-   * agent预置提示词
-   */
-  get_preset_prompt() {
-    const prompt = import_fs.default.readFileSync(import_path.default.join(__dirname, "./preset_prompt.md"), "utf-8");
-    return prompt;
-  }
-};
-
-// src/function_call/index.ts
-var import_path2 = __toESM(require("path"), 1);
-var import_fs2 = __toESM(require("fs"), 1);
-var FunctionCall = class {
-  constructor(tool_path) {
-    this.tool_path = tool_path;
-    this.tools = {};
-    this.init();
-  }
-  init() {
-    for (const file_name of import_fs2.default.readdirSync(this.tool_path)) {
-      const route_path = import_path2.default.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];
-    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);
-  }
-};
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  Agent,
-  FunctionCall,
-  LLM_Utils
-});