p-api-agent 0.0.1 → 0.0.3

package/dist/index.cjs

@@ -1,10 +1,10 @@
-"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 __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 });
@@ -26,6 +26,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
   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 = {};
@@ -98,109 +99,191 @@ 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();
+  constructor(options = {}) {
+    __publicField(this, "function_call", null);
+    __publicField(this, "llm_chat_func", null);
+    __publicField(this, "max_loop");
+    __publicField(this, "retry_times");
+    __publicField(this, "custom_system_prompt", null);
+    this.max_loop = options.maxLoop ?? 20;
+    this.retry_times = options.retryTimes ?? 2;
+    if (options.customSystemPrompt) {
+      this.custom_system_prompt = options.customSystemPrompt;
+    }
   }
-  /**
-   * 注册工具函数
-   */
-  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;
+  // ─────────── Registration ───────────
+  /** 注册工具函数集合，支持链式调用 */
+  register_function_call(fc) {
+    this.function_call = fc;
+    return this;
   }
-  /**
-   * 注册LLM模型文字能力
-   */
+  /** 注册 LLM 文字能力，支持链式调用 */
   register_llm_text_ability(func) {
     this.llm_chat_func = func;
+    return this;
   }
-  /**
-  * 注册LLM模型图片能力
-  */
-  register_llm_image_ability() {
-    return true;
+  /** 覆盖系统提示词 */
+  set_system_prompt(prompt) {
+    this.custom_system_prompt = prompt;
+    return this;
   }
+  /** 动态修改最大循环次数 */
+  set_max_loop(n) {
+    this.max_loop = n;
+    return this;
+  }
+  // ─────────── Main ReAct Loop ───────────
   /**
-   * 创建聊天
+   * 发起对话。
+   *
+   * 架构：ReAct 单调用循环
+   *   每轮只调用一次 LLM，LLM 在拥有完整工具 schema 的系统提示词下
+   *   直接输出 end / use_tool / no_ability 三种指令，
+   *   彻底消除了旧版 "判断→查文档→生成参数→再生成答案" 的多次调用开销。
    */
   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.llm_chat_func) {
+      return { result: "\u672A\u6CE8\u518CLLM\u80FD\u529B", use_tools: [] };
     }
-    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({
+    const history = this.normalize_input(user_input);
+    const system_msg = {
       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}` }
-          ]
+      content: [{ type: "text", text: this.build_system_prompt() }]
+    };
+    const result = { result: "", use_tools: [] };
+    const tool_fail_count = {};
+    for (let turn = 0; turn < this.max_loop; turn++) {
+      let raw;
+      try {
+        raw = await this.call_llm_with_retry([system_msg, ...history]);
+      } catch (err) {
+        result.result = `LLM\u8C03\u7528\u5931\u8D25: ${err?.message ?? String(err)}`;
+        return result;
+      }
+      const cmd = this.parse_command(raw);
+      if (!cmd) {
+        result.result = raw;
+        return result;
+      }
+      if (cmd.command === "end") {
+        result.result = cmd.result;
+        return result;
+      }
+      if (cmd.command === "no_ability") {
+        result.result = cmd.result || "\u65E0\u6CD5\u6267\u884C\u6B64\u64CD\u4F5C";
+        return result;
+      }
+      if (cmd.command === "use_tool") {
+        const { tool_name, params } = cmd;
+        if (!this.function_call) {
+          result.result = "\u672A\u6CE8\u518C\u5DE5\u5177\u51FD\u6570\u96C6\u5408\uFF0C\u65E0\u6CD5\u6267\u884C\u5DE5\u5177\u8C03\u7528";
+          return result;
+        }
+        history.push({
+          role: "assistant",
+          content: [{ type: "text", text: raw }]
         });
-      } 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({
+        let tool_result;
+        try {
+          tool_result = await this.function_call.exec_function(tool_name, params);
+        } catch (err) {
+          const err_msg = err?.message ?? String(err);
+          tool_fail_count[tool_name] = (tool_fail_count[tool_name] ?? 0) + 1;
+          history.push({
+            role: "user",
+            content: [{
+              type: "text",
+              text: `[\u5DE5\u5177\u6267\u884C\u5931\u8D25] ${tool_name}
+\u9519\u8BEF\u4FE1\u606F: ${err_msg}
+\u8BF7\u6839\u636E\u9519\u8BEF\u8C03\u6574\u7B56\u7565\u540E\u7EE7\u7EED`
+            }]
+          });
+          if (tool_fail_count[tool_name] >= 2) {
+            result.result = `\u5DE5\u5177 "${tool_name}" \u8FDE\u7EED\u8C03\u7528\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5`;
+            return result;
+          }
+          continue;
+        }
+        history.push({
           role: "user",
-          content: [
-            { type: "text", text: `\u5DE5\u5177\u51FD\u6570${tool_name}\u7684\u8C03\u7528\u7ED3\u679C:
-${JSON.stringify(tool_result)}` }
-          ]
+          content: [{
+            type: "text",
+            text: `[\u5DE5\u5177\u8C03\u7528\u7ED3\u679C] ${tool_name}
+${JSON.stringify(tool_result, null, 2)}`
+          }]
         });
+        result.use_tools.push({ tool_name, params, exec_result: tool_result });
+        continue;
       }
-      loop_count++;
     }
+    result.result = "\u8D85\u51FA\u6700\u5927\u63A8\u7406\u8F6E\u6B21\uFF0C\u8BF7\u7CBE\u7B80\u95EE\u9898\u540E\u91CD\u8BD5";
+    return result;
   }
+  // ─────────── System Prompt ───────────
   /**
-   * agent预置提示词
+   * 构建系统提示词。
+   * 将所有工具的完整 schema（含参数类型、必填项、示例值）
+   * 直接注入系统提示词，让 LLM 在单次调用中就能做出准确的工具选择和参数填写。
    */
-  get_preset_prompt() {
-    const prompt = import_fs.default.readFileSync(import_path.default.join(__dirname, "./preset_prompt.md"), "utf-8");
-    return prompt;
+  build_system_prompt() {
+    if (this.custom_system_prompt) return this.custom_system_prompt;
+    const base = this.load_preset_prompt();
+    if (!this.function_call) return base;
+    const tool_schemas = this.function_call.get_tools_with_schema().map((t) => {
+      const props = Object.entries(t.input_schema.properties).map(([k, v]) => {
+        const req = t.input_schema.required?.includes(k) ? "\u5FC5\u586B" : "\u53EF\u9009";
+        const ex = v.examples?.length ? `\uFF0C\u793A\u4F8B: ${v.examples.join(" / ")}` : "";
+        return `    - ${k} (${v.type}, ${req}): ${v.description}${ex}`;
+      }).join("\n");
+      return `### ${t.name}
+\u63CF\u8FF0: ${t.description}
+\u53C2\u6570:
+${props}`;
+    }).join("\n\n");
+    return `${base}
+
+## \u53EF\u7528\u5DE5\u5177\u5217\u8868
+
+${tool_schemas}`;
+  }
+  load_preset_prompt() {
+    try {
+      return import_fs.default.readFileSync(import_path.default.join(__dirname, "./preset_prompt.md"), "utf-8");
+    } catch {
+      return "";
+    }
+  }
+  // ─────────── Helpers ───────────
+  normalize_input(input) {
+    if (typeof input === "string") {
+      return [{ role: "user", content: [{ type: "text", text: input }] }];
+    }
+    return [...input];
+  }
+  parse_command(raw) {
+    const json = this.safe_extract_json(raw);
+    if (!json || typeof json.command !== "string") return null;
+    return json;
+  }
+  safe_extract_json(content) {
+    try {
+      return LLM_Utils.extract_json(content);
+    } catch {
+      return null;
+    }
+  }
+  async call_llm_with_retry(input) {
+    let last_error;
+    for (let i = 0; i <= this.retry_times; i++) {
+      try {
+        const res = await this.llm_chat_func(input);
+        if (typeof res === "string" && res.trim()) return res;
+        throw new Error("\u7A7A\u54CD\u5E94");
+      } catch (err) {
+        last_error = err;
+      }
+    }
+    throw last_error ?? new Error("LLM\u8C03\u7528\u5931\u8D25");
   }
 };
 
@@ -210,7 +293,7 @@ var import_fs2 = __toESM(require("fs"), 1);
 var FunctionCall = class {
   constructor(tool_path) {
     this.tool_path = tool_path;
-    this.tools = {};
+    __publicField(this, "tools", {});
     this.init();
   }
   init() {
@@ -226,6 +309,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) => {
@@ -243,7 +329,7 @@ var FunctionCall = class {
 ${params}`;
   }
   /**
-   * 获取工具函数列表
+   * 获取工具函数列表（仅名称 + 描述）
    */
   get_tools_list() {
     return Object.keys(this.tools).map((key) => {
@@ -254,6 +340,19 @@ ${params}`;
       };
     });
   }
+  /**
+   * 获取工具函数列表（含完整 input_schema），供 Agent 注入系统提示词
+   */
+  get_tools_with_schema() {
+    return Object.keys(this.tools).map((key) => {
+      const info = this.tools[key];
+      return {
+        name: info.name,
+        description: info.description,
+        input_schema: info.input_schema
+      };
+    });
+  }
   /**
    * 执行工具函数
    */

package/dist/index.d.cts

@@ -24,12 +24,30 @@ declare class FunctionCall {
      */
     gen_tool_doc(name: string): string;
     /**
-     * 获取工具函数列表
+     * 获取工具函数列表（仅名称 + 描述）
      */
     get_tools_list(): {
         name: string;
         description: string;
     }[];
+    /**
+     * 获取工具函数列表（含完整 input_schema），供 Agent 注入系统提示词
+     */
+    get_tools_with_schema(): {
+        name: string;
+        description: string;
+        input_schema: {
+            type: string;
+            properties: {
+                [key: string]: {
+                    type: string;
+                    description: string;
+                    examples?: string[];
+                };
+            };
+            required?: string[];
+        };
+    }[];
     /**
      * 执行工具函数
      */
@@ -47,36 +65,63 @@ interface ImageContent {
     };
 }
 type Content = TextContent | ImageContent;
-type UserChatInput = string | {
+type Message = {
     role: 'system' | 'user' | 'assistant';
     content: Content[];
-}[];
+};
+type UserChatInput = Message[];
+interface ToolRecord {
+    tool_name: string;
+    params: any;
+    exec_result: any;
+}
+interface CreateChatResult {
+    result: string;
+    use_tools: ToolRecord[];
+}
+interface AgentOptions {
+    /** 最大推理循环次数，默认 20 */
+    maxLoop?: number;
+    /** LLM 失败重试次数，默认 2 */
+    retryTimes?: number;
+    /** 完全覆盖系统提示词 */
+    customSystemPrompt?: string;
+}
 declare class Agent {
     private function_call;
-    private function_call_doc;
     private llm_chat_func;
-    preset_prompt: string;
-    constructor();
-    /**
-     * 注册工具函数
-     */
-    register_function_call(function_call: FunctionCall): boolean;
-    /**
-     * 注册LLM模型文字能力
-     */
-    register_llm_text_ability(func: (user_input: UserChatInput) => Promise<string>): void;
-    /**
-    * 注册LLM模型图片能力
-    */
-    register_llm_image_ability(): boolean;
+    private max_loop;
+    private retry_times;
+    private custom_system_prompt;
+    constructor(options?: AgentOptions);
+    /** 注册工具函数集合，支持链式调用 */
+    register_function_call(fc: FunctionCall): this;
+    /** 注册 LLM 文字能力，支持链式调用 */
+    register_llm_text_ability(func: (input: UserChatInput) => Promise<string>): this;
+    /** 覆盖系统提示词 */
+    set_system_prompt(prompt: string): this;
+    /** 动态修改最大循环次数 */
+    set_max_loop(n: number): this;
     /**
-     * 创建聊天
+     * 发起对话。
+     *
+     * 架构：ReAct 单调用循环
+     *   每轮只调用一次 LLM，LLM 在拥有完整工具 schema 的系统提示词下
+     *   直接输出 end / use_tool / no_ability 三种指令，
+     *   彻底消除了旧版 "判断→查文档→生成参数→再生成答案" 的多次调用开销。
      */
-    create_chat(user_input: UserChatInput): Promise<any>;
+    create_chat(user_input: string | UserChatInput): Promise<CreateChatResult>;
     /**
-     * agent预置提示词
+     * 构建系统提示词。
+     * 将所有工具的完整 schema（含参数类型、必填项、示例值）
+     * 直接注入系统提示词，让 LLM 在单次调用中就能做出准确的工具选择和参数填写。
      */
-    get_preset_prompt(): string;
+    private build_system_prompt;
+    private load_preset_prompt;
+    private normalize_input;
+    private parse_command;
+    private safe_extract_json;
+    private call_llm_with_retry;
 }
 
 declare class _LLM_Utils {
@@ -100,4 +145,4 @@ declare class _LLM_Utils {
 }
 declare const LLM_Utils: _LLM_Utils;
 
-export { Agent, FunctionCall, LLM_Utils, type RegisterInfo };
+export { Agent, type AgentOptions, type CreateChatResult, FunctionCall, LLM_Utils, type Message, type RegisterInfo, type ToolRecord, type UserChatInput };

package/dist/index.d.ts

@@ -24,12 +24,30 @@ declare class FunctionCall {
      */
     gen_tool_doc(name: string): string;
     /**
-     * 获取工具函数列表
+     * 获取工具函数列表（仅名称 + 描述）
      */
     get_tools_list(): {
         name: string;
         description: string;
     }[];
+    /**
+     * 获取工具函数列表（含完整 input_schema），供 Agent 注入系统提示词
+     */
+    get_tools_with_schema(): {
+        name: string;
+        description: string;
+        input_schema: {
+            type: string;
+            properties: {
+                [key: string]: {
+                    type: string;
+                    description: string;
+                    examples?: string[];
+                };
+            };
+            required?: string[];
+        };
+    }[];
     /**
      * 执行工具函数
      */
@@ -47,36 +65,63 @@ interface ImageContent {
     };
 }
 type Content = TextContent | ImageContent;
-type UserChatInput = string | {
+type Message = {
     role: 'system' | 'user' | 'assistant';
     content: Content[];
-}[];
+};
+type UserChatInput = Message[];
+interface ToolRecord {
+    tool_name: string;
+    params: any;
+    exec_result: any;
+}
+interface CreateChatResult {
+    result: string;
+    use_tools: ToolRecord[];
+}
+interface AgentOptions {
+    /** 最大推理循环次数，默认 20 */
+    maxLoop?: number;
+    /** LLM 失败重试次数，默认 2 */
+    retryTimes?: number;
+    /** 完全覆盖系统提示词 */
+    customSystemPrompt?: string;
+}
 declare class Agent {
     private function_call;
-    private function_call_doc;
     private llm_chat_func;
-    preset_prompt: string;
-    constructor();
-    /**
-     * 注册工具函数
-     */
-    register_function_call(function_call: FunctionCall): boolean;
-    /**
-     * 注册LLM模型文字能力
-     */
-    register_llm_text_ability(func: (user_input: UserChatInput) => Promise<string>): void;
-    /**
-    * 注册LLM模型图片能力
-    */
-    register_llm_image_ability(): boolean;
+    private max_loop;
+    private retry_times;
+    private custom_system_prompt;
+    constructor(options?: AgentOptions);
+    /** 注册工具函数集合，支持链式调用 */
+    register_function_call(fc: FunctionCall): this;
+    /** 注册 LLM 文字能力，支持链式调用 */
+    register_llm_text_ability(func: (input: UserChatInput) => Promise<string>): this;
+    /** 覆盖系统提示词 */
+    set_system_prompt(prompt: string): this;
+    /** 动态修改最大循环次数 */
+    set_max_loop(n: number): this;
     /**
-     * 创建聊天
+     * 发起对话。
+     *
+     * 架构：ReAct 单调用循环
+     *   每轮只调用一次 LLM，LLM 在拥有完整工具 schema 的系统提示词下
+     *   直接输出 end / use_tool / no_ability 三种指令，
+     *   彻底消除了旧版 "判断→查文档→生成参数→再生成答案" 的多次调用开销。
      */
-    create_chat(user_input: UserChatInput): Promise<any>;
+    create_chat(user_input: string | UserChatInput): Promise<CreateChatResult>;
     /**
-     * agent预置提示词
+     * 构建系统提示词。
+     * 将所有工具的完整 schema（含参数类型、必填项、示例值）
+     * 直接注入系统提示词，让 LLM 在单次调用中就能做出准确的工具选择和参数填写。
      */
-    get_preset_prompt(): string;
+    private build_system_prompt;
+    private load_preset_prompt;
+    private normalize_input;
+    private parse_command;
+    private safe_extract_json;
+    private call_llm_with_retry;
 }
 
 declare class _LLM_Utils {
@@ -100,4 +145,4 @@ declare class _LLM_Utils {
 }
 declare const LLM_Utils: _LLM_Utils;
 
-export { Agent, FunctionCall, LLM_Utils, type RegisterInfo };
+export { Agent, type AgentOptions, type CreateChatResult, FunctionCall, LLM_Utils, type Message, type RegisterInfo, type ToolRecord, type UserChatInput };

package/dist/index.js

@@ -1,9 +1,12 @@
+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";
@@ -67,109 +70,191 @@ var LLM_Utils = new _LLM_Utils();
 // src/agent/index.ts
 import path from "path";
 var Agent = class {
-  // 角色定义的提示词
-  constructor() {
-    this.function_call = null;
-    // 工具函数的说明文档, 供模型调用时参考
-    this.function_call_doc = "";
-    // 模型的聊天能力
-    this.llm_chat_func = null;
-    // 预置提示词
-    this.preset_prompt = this.get_preset_prompt();
+  constructor(options = {}) {
+    __publicField(this, "function_call", null);
+    __publicField(this, "llm_chat_func", null);
+    __publicField(this, "max_loop");
+    __publicField(this, "retry_times");
+    __publicField(this, "custom_system_prompt", null);
+    this.max_loop = options.maxLoop ?? 20;
+    this.retry_times = options.retryTimes ?? 2;
+    if (options.customSystemPrompt) {
+      this.custom_system_prompt = options.customSystemPrompt;
+    }
   }
-  /**
-   * 注册工具函数
-   */
-  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;
+  // ─────────── Registration ───────────
+  /** 注册工具函数集合，支持链式调用 */
+  register_function_call(fc) {
+    this.function_call = fc;
+    return this;
   }
-  /**
-   * 注册LLM模型文字能力
-   */
+  /** 注册 LLM 文字能力，支持链式调用 */
   register_llm_text_ability(func) {
     this.llm_chat_func = func;
+    return this;
   }
-  /**
-  * 注册LLM模型图片能力
-  */
-  register_llm_image_ability() {
-    return true;
+  /** 覆盖系统提示词 */
+  set_system_prompt(prompt) {
+    this.custom_system_prompt = prompt;
+    return this;
   }
+  /** 动态修改最大循环次数 */
+  set_max_loop(n) {
+    this.max_loop = n;
+    return this;
+  }
+  // ─────────── Main ReAct Loop ───────────
   /**
-   * 创建聊天
+   * 发起对话。
+   *
+   * 架构：ReAct 单调用循环
+   *   每轮只调用一次 LLM，LLM 在拥有完整工具 schema 的系统提示词下
+   *   直接输出 end / use_tool / no_ability 三种指令，
+   *   彻底消除了旧版 "判断→查文档→生成参数→再生成答案" 的多次调用开销。
    */
   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.llm_chat_func) {
+      return { result: "\u672A\u6CE8\u518CLLM\u80FD\u529B", use_tools: [] };
     }
-    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({
+    const history = this.normalize_input(user_input);
+    const system_msg = {
       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}` }
-          ]
+      content: [{ type: "text", text: this.build_system_prompt() }]
+    };
+    const result = { result: "", use_tools: [] };
+    const tool_fail_count = {};
+    for (let turn = 0; turn < this.max_loop; turn++) {
+      let raw;
+      try {
+        raw = await this.call_llm_with_retry([system_msg, ...history]);
+      } catch (err) {
+        result.result = `LLM\u8C03\u7528\u5931\u8D25: ${err?.message ?? String(err)}`;
+        return result;
+      }
+      const cmd = this.parse_command(raw);
+      if (!cmd) {
+        result.result = raw;
+        return result;
+      }
+      if (cmd.command === "end") {
+        result.result = cmd.result;
+        return result;
+      }
+      if (cmd.command === "no_ability") {
+        result.result = cmd.result || "\u65E0\u6CD5\u6267\u884C\u6B64\u64CD\u4F5C";
+        return result;
+      }
+      if (cmd.command === "use_tool") {
+        const { tool_name, params } = cmd;
+        if (!this.function_call) {
+          result.result = "\u672A\u6CE8\u518C\u5DE5\u5177\u51FD\u6570\u96C6\u5408\uFF0C\u65E0\u6CD5\u6267\u884C\u5DE5\u5177\u8C03\u7528";
+          return result;
+        }
+        history.push({
+          role: "assistant",
+          content: [{ type: "text", text: raw }]
         });
-      } 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({
+        let tool_result;
+        try {
+          tool_result = await this.function_call.exec_function(tool_name, params);
+        } catch (err) {
+          const err_msg = err?.message ?? String(err);
+          tool_fail_count[tool_name] = (tool_fail_count[tool_name] ?? 0) + 1;
+          history.push({
+            role: "user",
+            content: [{
+              type: "text",
+              text: `[\u5DE5\u5177\u6267\u884C\u5931\u8D25] ${tool_name}
+\u9519\u8BEF\u4FE1\u606F: ${err_msg}
+\u8BF7\u6839\u636E\u9519\u8BEF\u8C03\u6574\u7B56\u7565\u540E\u7EE7\u7EED`
+            }]
+          });
+          if (tool_fail_count[tool_name] >= 2) {
+            result.result = `\u5DE5\u5177 "${tool_name}" \u8FDE\u7EED\u8C03\u7528\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5`;
+            return result;
+          }
+          continue;
+        }
+        history.push({
           role: "user",
-          content: [
-            { type: "text", text: `\u5DE5\u5177\u51FD\u6570${tool_name}\u7684\u8C03\u7528\u7ED3\u679C:
-${JSON.stringify(tool_result)}` }
-          ]
+          content: [{
+            type: "text",
+            text: `[\u5DE5\u5177\u8C03\u7528\u7ED3\u679C] ${tool_name}
+${JSON.stringify(tool_result, null, 2)}`
+          }]
         });
+        result.use_tools.push({ tool_name, params, exec_result: tool_result });
+        continue;
       }
-      loop_count++;
     }
+    result.result = "\u8D85\u51FA\u6700\u5927\u63A8\u7406\u8F6E\u6B21\uFF0C\u8BF7\u7CBE\u7B80\u95EE\u9898\u540E\u91CD\u8BD5";
+    return result;
   }
+  // ─────────── System Prompt ───────────
   /**
-   * agent预置提示词
+   * 构建系统提示词。
+   * 将所有工具的完整 schema（含参数类型、必填项、示例值）
+   * 直接注入系统提示词，让 LLM 在单次调用中就能做出准确的工具选择和参数填写。
    */
-  get_preset_prompt() {
-    const prompt = fs.readFileSync(path.join(__dirname, "./preset_prompt.md"), "utf-8");
-    return prompt;
+  build_system_prompt() {
+    if (this.custom_system_prompt) return this.custom_system_prompt;
+    const base = this.load_preset_prompt();
+    if (!this.function_call) return base;
+    const tool_schemas = this.function_call.get_tools_with_schema().map((t) => {
+      const props = Object.entries(t.input_schema.properties).map(([k, v]) => {
+        const req = t.input_schema.required?.includes(k) ? "\u5FC5\u586B" : "\u53EF\u9009";
+        const ex = v.examples?.length ? `\uFF0C\u793A\u4F8B: ${v.examples.join(" / ")}` : "";
+        return `    - ${k} (${v.type}, ${req}): ${v.description}${ex}`;
+      }).join("\n");
+      return `### ${t.name}
+\u63CF\u8FF0: ${t.description}
+\u53C2\u6570:
+${props}`;
+    }).join("\n\n");
+    return `${base}
+
+## \u53EF\u7528\u5DE5\u5177\u5217\u8868
+
+${tool_schemas}`;
+  }
+  load_preset_prompt() {
+    try {
+      return fs.readFileSync(path.join(__dirname, "./preset_prompt.md"), "utf-8");
+    } catch {
+      return "";
+    }
+  }
+  // ─────────── Helpers ───────────
+  normalize_input(input) {
+    if (typeof input === "string") {
+      return [{ role: "user", content: [{ type: "text", text: input }] }];
+    }
+    return [...input];
+  }
+  parse_command(raw) {
+    const json = this.safe_extract_json(raw);
+    if (!json || typeof json.command !== "string") return null;
+    return json;
+  }
+  safe_extract_json(content) {
+    try {
+      return LLM_Utils.extract_json(content);
+    } catch {
+      return null;
+    }
+  }
+  async call_llm_with_retry(input) {
+    let last_error;
+    for (let i = 0; i <= this.retry_times; i++) {
+      try {
+        const res = await this.llm_chat_func(input);
+        if (typeof res === "string" && res.trim()) return res;
+        throw new Error("\u7A7A\u54CD\u5E94");
+      } catch (err) {
+        last_error = err;
+      }
+    }
+    throw last_error ?? new Error("LLM\u8C03\u7528\u5931\u8D25");
   }
 };
 
@@ -179,7 +264,7 @@ import fs2 from "fs";
 var FunctionCall = class {
   constructor(tool_path) {
     this.tool_path = tool_path;
-    this.tools = {};
+    __publicField(this, "tools", {});
     this.init();
   }
   init() {
@@ -195,6 +280,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) => {
@@ -212,7 +300,7 @@ var FunctionCall = class {
 ${params}`;
   }
   /**
-   * 获取工具函数列表
+   * 获取工具函数列表（仅名称 + 描述）
    */
   get_tools_list() {
     return Object.keys(this.tools).map((key) => {
@@ -223,6 +311,19 @@ ${params}`;
       };
     });
   }
+  /**
+   * 获取工具函数列表（含完整 input_schema），供 Agent 注入系统提示词
+   */
+  get_tools_with_schema() {
+    return Object.keys(this.tools).map((key) => {
+      const info = this.tools[key];
+      return {
+        name: info.name,
+        description: info.description,
+        input_schema: info.input_schema
+      };
+    });
+  }
   /**
    * 执行工具函数
    */

package/package.json

@@ -1,7 +1,6 @@
 {
-  "type": "module",
   "name": "p-api-agent",
-  "version": "0.0.1",
+  "version": "0.0.3",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
   "files": [
@@ -24,11 +23,15 @@
     "@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": {
     "axios": "^1.13.6",
     "jsonrepair": "^3.13.2"
-  }
+  },
+  "type": "module"
 }