@amical/aiagent-sdk 0.1.10 → 0.1.12

package/dist/index.js

@@ -136,9 +136,11 @@ return (
   },
   {
     name: "agent_read_file" /* FILE_READ */,
-    description: "Read the contents of a specific file.",
+    description: "Read the contents of a specific file. The returned content includes line numbers (e.g. '10|const x = 1;') so you can reference exact lines when editing.",
     params: {
-      path: "(required) The absolute or relative file path to read."
+      path: "(required) The absolute or relative file path to read.",
+      start_line: "(optional) The 1-based line number to start reading from (inclusive). Defaults to 1.",
+      end_line: "(optional) The 1-based line number to stop reading at (inclusive). Defaults to the last line."
     }
   },
   {
@@ -285,6 +287,44 @@ var READ_ONLY_TOOLS = [
   "agent_load_skill" /* LOAD_SKILL */,
   "agent_use_subagents" /* USE_SUBAGENTS */
 ];
+
+// src/shared/custom-tools.ts
+function isCustomToolGroup(tool) {
+  return Array.isArray(tool.children) && tool.children.length > 0;
+}
+function collectCustomToolNames(tools) {
+  const names = [];
+  const visit = (list) => {
+    for (const tool of list) {
+      names.push(tool.name);
+      if (Array.isArray(tool.children) && tool.children.length > 0) {
+        visit(tool.children);
+      }
+    }
+  };
+  visit(tools);
+  return names;
+}
+function renderCustomToolPrompt(tool, cwd = "") {
+  const params = tool.params ? Object.entries(tool.params) : [];
+  const replaceCwd = (text) => String(text).replaceAll("${context.cwd}", cwd);
+  const paramLines = params.map(([name, description]) => `- ${name}: ${replaceCwd(description)}`).join("\n");
+  const usageLines = params.map(([name, description]) => `<${name}>${replaceCwd(description)}</${name}>`).join("\n");
+  return `### ${tool.name}
+Description: ${tool.description}
+Parameters: 
+${paramLines}
+- task_progress: (optional) A checklist showing task progress after this tool use is completed. The task_progress parameter must be included as a separate parameter inside of the parent tool call, it must be separate from other parameters such as content, arguments, etc. (See 'UPDATING TASK PROGRESS' section for more details)
+Usage: 
+<${tool.name}>
+${usageLines}
+<task_progress>Checklist here (optional)</task_progress>
+</${tool.name}>
+`;
+}
+function renderCustomToolsPrompt(tools, cwd = "") {
+  return tools.map((tool) => renderCustomToolPrompt(tool, cwd)).join("\n");
+}
 function parseAssistantMessageV2(assistantMessage, toolParamNames2) {
   const contentBlocks = [];
   let currentTextContentStart = 0;
@@ -543,7 +583,17 @@ var TEXT_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
   "dockerfile",
   "makefile",
   "properties",
-  "lock"
+  "lock",
+  "node-version",
+  "nvmrc",
+  "editorconfig",
+  "prettierrc",
+  "eslintrc",
+  "babelrc",
+  "stylelintrc",
+  "huskyrc",
+  "commitlintrc",
+  "lintstagedrc"
 ]);
 var AgentState = /* @__PURE__ */ ((AgentState2) => {
   AgentState2["IDLE"] = "idle";
@@ -568,9 +618,19 @@ var ClineAgent = class {
   skillDir;
   rulesDir;
   customTools;
+  /** 当前可用（已展开集合）的顶层工具列表，传给 PromptManager 渲染 */
+  activeCustomTools;
+  /** 所有已知工具名（含未激活 children），便于查找定义 */
+  customToolsByName;
   customToolNames;
   allowDefaultTools;
   customInstructions;
+  /** 已激活的工具集入口名 */
+  activatedGroups = /* @__PURE__ */ new Set();
+  /** activeCustomTools 变更版本号，用于 Prompt 缓存失效 */
+  activeToolsVersion = 0;
+  /** 缓存 sysPrompt 时记录的版本号 */
+  promptActiveToolsVersion = -1;
   sysPrompt;
   noUseTollSysPrompt;
   // 简单的事件回调系统，供宿主环境注册以监听 Agent 内部事件
@@ -590,33 +650,47 @@ var ClineAgent = class {
       this.allowDefaultTools.push("agent_attempt_completion" /* ATTEMPT */);
     }
     this.customInstructions = options.customInstructions;
+    this.customToolsByName = /* @__PURE__ */ new Map();
+    this.activeCustomTools = [...this.customTools];
+    this.customToolNames = new Set(collectCustomToolNames(this.customTools));
     if (this.customTools.length > 0) {
-      const dynamicToolNames = /* @__PURE__ */ new Set();
       const customToolParams = /* @__PURE__ */ new Set();
-      for (const customTool of this.customTools) {
-        dynamicToolNames.add(customTool.name);
-        if (customTool.params) {
-          for (const key in customTool.params) {
-            if (!Object.hasOwn(customTool.params, key)) continue;
-            customToolParams.add(key);
+      const visit = (tools) => {
+        for (const customTool of tools) {
+          this.customToolsByName.set(customTool.name, customTool);
+          if (customTool.params) {
+            for (const key in customTool.params) {
+              if (!Object.hasOwn(customTool.params, key)) continue;
+              customToolParams.add(key);
+            }
+          }
+          if (Array.isArray(customTool.children) && customTool.children.length > 0) {
+            visit(customTool.children);
           }
         }
-      }
+      };
+      visit(this.customTools);
       const defaultToolNames = Object.values(AgentDefaultTool);
-      const overlappingTools = Array.from(dynamicToolNames).filter((name) => defaultToolNames.includes(name));
+      const overlappingTools = Array.from(this.customToolNames).filter((name) => defaultToolNames.includes(name));
       if (overlappingTools.length > 0) {
         throw new Error(`Custom tools and default tools have overlapping names: ${overlappingTools.join(", ")}`);
       }
-      if (dynamicToolNames.size > 0) {
-        setDynamicToolUseNames("customUser", Array.from(dynamicToolNames));
+      if (this.customToolNames.size > 0) {
+        setDynamicToolUseNames("customUser", Array.from(this.customToolNames));
       }
       if (customToolParams.size > 0) {
         this.allToolParamNames = Array.from(/* @__PURE__ */ new Set([...this.allToolParamNames, ...Array.from(customToolParams)]));
       }
     }
-    this.customToolNames = new Set(this.customTools.map((t) => t.name));
     this.skillDir = this.runtime.path.join(this.runtime.env.cwd(), ".agents/skills");
     this.rulesDir = this.runtime.path.join(this.runtime.env.cwd(), ".agents/rules");
+    const originalOnCondense = this.memory.onCondense;
+    this.memory.onCondense = async () => {
+      this.api.clearConversationId();
+      if (originalOnCondense) {
+        await originalOnCondense();
+      }
+    };
   }
   setState(newState) {
     this.state = newState;
@@ -638,10 +712,11 @@ var ClineAgent = class {
       shell: this.runtime.env.shell(),
       skills: await this.getSkills(),
       agentRules: await this.getAgentRules(),
-      customTools: this.customTools,
+      customTools: this.activeCustomTools,
       allowDefaultTools: this.allowDefaultTools,
       customInstructions: this.customInstructions
     };
+    this.promptActiveToolsVersion = this.activeToolsVersion;
     this.sysPrompt = await this.promptManager.getSystemPrompt(context);
     this.noUseTollSysPrompt = this.promptManager.getNoUseToolInstructions(context);
     return { sysPrompt: this.sysPrompt, noUseTollSysPrompt: this.noUseTollSysPrompt };
@@ -673,16 +748,6 @@ var ClineAgent = class {
   async getAgentRules() {
     const cwd = this.runtime.env.cwd();
     const rules = [];
-    const agentsMdPath = this.runtime.path.join(cwd, "Agents.md");
-    try {
-      if (await this.runtime.fs.fileExists(agentsMdPath)) {
-        const content = await this.runtime.fs.readFile(agentsMdPath);
-        if (content && content.trim()) {
-          rules.push(content.trim());
-        }
-      }
-    } catch (e) {
-    }
     try {
       if (await this.runtime.fs.fileExists(this.rulesDir)) {
         const files = await this.runtime.fs.readDir(this.rulesDir);
@@ -708,6 +773,28 @@ ${content.trim()}\`\`\``);
       }
     } catch (e) {
     }
+    try {
+      const getAgentsMdPath = async () => {
+        const agentsMdPathList = [
+          this.runtime.path.join(cwd, "AGENTS.md"),
+          this.runtime.path.join(cwd, "Agents.md"),
+          this.runtime.path.join(cwd, "agents.md")
+        ];
+        for (const item of agentsMdPathList) {
+          if (await this.runtime.fs.fileExists(item)) {
+            return item;
+          }
+        }
+      };
+      const agentsMdPath = await getAgentsMdPath();
+      if (agentsMdPath) {
+        const content = await this.runtime.fs.readFile(agentsMdPath);
+        if (content && content.trim()) {
+          rules.push(content.trim());
+        }
+      }
+    } catch (e) {
+    }
     return rules.join("\n---\n");
   }
   async getSkills() {
@@ -750,9 +837,9 @@ ${content.trim()}\`\`\``);
    * 核心任务循环
    */
   async loop() {
-    const { sysPrompt, noUseTollSysPrompt } = await this.getPrompt(true);
-    const { customTools, customToolNames } = this;
+    const { customToolsByName, customToolNames } = this;
     while (this.state === "thinking" /* THINKING */) {
+      const { sysPrompt, noUseTollSysPrompt } = await this.getCurrentPrompt();
       this.loopCount++;
       if (this.loopCount > this.maxLoopCount) {
         await this.memory.appendMessage({
@@ -797,7 +884,33 @@ ${content.trim()}\`\`\``);
             }
           }
           if (customToolNames.has(tool.name)) {
-            const customTool = customTools.find((t) => t.name === tool.name);
+            const customTool = customToolsByName.get(tool.name);
+            if (!customTool) {
+              toolResults.push({ tool, result: `Error: Custom tool "${tool.name}" not found.` });
+              continue;
+            }
+            if (isCustomToolGroup(customTool)) {
+              let groupResult;
+              if (customTool.onCall) {
+                try {
+                  groupResult = await customTool.onCall(tool.params);
+                } catch (error) {
+                  toolResults.push({ tool, result: `Error: ${error.message}` });
+                  continue;
+                }
+              }
+              const childrenPrompt = this.activateCustomToolGroup(customTool);
+              const finalPrompt = `# New Available Tools
+
+The following are instructions for using the new available tools.
+
+${childrenPrompt}`;
+              const promptBlock = groupResult !== void 0 ? `${typeof groupResult === "string" ? groupResult : JSON.stringify(groupResult)}
+
+${finalPrompt}` : finalPrompt;
+              toolResults.push({ tool, result: promptBlock });
+              continue;
+            }
             if (customTool.onCall) {
               try {
                 const res = await customTool.onCall(tool.params);
@@ -819,7 +932,8 @@ ${content.trim()}\`\`\``);
             if (path) {
               try {
                 const content = await this.runtime.fs.readFile(path);
-                toolResults.push({ tool, result: content });
+                const result = this.formatFileContentWithLineNumbers(content, tool.params.start_line, tool.params.end_line);
+                toolResults.push({ tool, result });
               } catch (error) {
                 toolResults.push({ tool, result: `Error: ${error.message}` });
               }
@@ -832,6 +946,8 @@ ${content.trim()}\`\`\``);
               const result = await this.runtime.terminal.execCommand(cmd);
               toolResults.push({ tool, result: `STDOUT: ${result.stdout}
 STDERR: ${result.stderr}` });
+            } else {
+              toolResults.push({ tool, result: "Error: No command provided." });
             }
           } else if (tool.name === "agent_write_to_file" /* FILE_NEW */) {
             const path = tool.params.path;
@@ -843,12 +959,18 @@ STDERR: ${result.stderr}` });
               } catch (error) {
                 toolResults.push({ tool, result: `Error: ${error.message}` });
               }
+            } else {
+              toolResults.push({ tool, result: "Error: Missing path or content." });
             }
           } else if (tool.name === "agent_list_files" /* LIST_FILES */) {
             const path = tool.params.path || this.runtime.env.cwd();
+            const recursive = tool.params.recursive === "true";
             try {
-              const files = await this.runtime.fs.readDir(path);
-              toolResults.push({ tool, result: files.map((f) => f.name + (f.isDirectory() ? "/" : "")).join("\n") });
+              const files = await this.runtime.fs.readDir(path, { recursive });
+              toolResults.push({
+                tool,
+                result: files.map((f) => `${f.isDirectory() ? "[DIR] " : "      "}${f.name}${f.isDirectory() ? "/" : ""}`).join("\n")
+              });
             } catch (e) {
               toolResults.push({ tool, result: `Error: ${e.message}` });
             }
@@ -988,10 +1110,15 @@ STDERR: ${result.stderr}` });
               try {
                 const res = await fetch(url);
                 const text = await res.text();
-                toolResults.push({ tool, result: text.substring(0, 3e3) });
+                const MAX_LEN = 8e3;
+                const truncated = text.length > MAX_LEN ? `${text.substring(0, MAX_LEN)}
+...[truncated ${text.length - MAX_LEN} chars]` : text;
+                toolResults.push({ tool, result: truncated });
               } catch (e) {
                 toolResults.push({ tool, result: `Error: ${e.message}` });
               }
+            } else {
+              toolResults.push({ tool, result: "Error: No url provided." });
             }
           } else if (tool.name === "agent_focus_chain" /* TODO */) {
             toolResults.push({ tool, result: `Focus chain updated successfully.` });
@@ -1058,6 +1185,65 @@ ${this.promptManager.getUsedToolResultInstructions()}`;
       }
     }
   }
+  /**
+   * 激活自定义工具集：将入口的 children 注册到活跃工具列表并返回渲染后的 prompt。
+   */
+  activateCustomToolGroup(group) {
+    if (this.activatedGroups.has(group.name)) {
+      return renderCustomToolsPrompt(group.children || [], this.runtime.env.cwd());
+    }
+    this.activatedGroups.add(group.name);
+    this.activeCustomTools = [...this.activeCustomTools, ...group.children || []];
+    this.activeToolsVersion++;
+    return renderCustomToolsPrompt(group.children || [], this.runtime.env.cwd());
+  }
+  /**
+   * 检查自上次获取 prompt 后是否有工具集被激活，若有则刷新缓存。
+   */
+  async getCurrentPrompt() {
+    if (this.promptActiveToolsVersion !== this.activeToolsVersion) {
+      return this.getPrompt(true);
+    }
+    return this.getPrompt();
+  }
+  /**
+   * 将文件内容格式化为带行号文本（模拟编辑器行号显示），
+   * 支持通过 start_line / end_line 截取读取范围。
+   */
+  formatFileContentWithLineNumbers(content, startLineParam, endLineParam) {
+    const lines = content.split(/\r?\n/);
+    const normalized = lines.length > 0 && lines[lines.length - 1] === "" ? lines.slice(0, -1) : lines;
+    const total = normalized.length;
+    if (total === 0) {
+      return "1|(empty file)";
+    }
+    const parseLine = (val, fallback) => {
+      if (val === void 0 || val === null || val.trim() === "") {
+        return fallback;
+      }
+      const num = Number.parseInt(val, 10);
+      if (!Number.isFinite(num) || num <= 0) {
+        return fallback;
+      }
+      return num;
+    };
+    const start = parseLine(startLineParam, 1);
+    const end = parseLine(endLineParam, total);
+    if (start > end) {
+      return `Error: start_line (${start}) cannot be greater than end_line (${end}).`;
+    }
+    if (start > total) {
+      return `Error: start_line (${start}) is out of range. The file has ${total} lines.`;
+    }
+    const from = Math.max(1, start);
+    const to = Math.min(total, end);
+    const width = String(total).length;
+    const selected = normalized.slice(from - 1, to);
+    const numbered = selected.map((line, idx) => `${String(from + idx).padStart(width)}|${line}`).join("\n");
+    const header = from === 1 && to === total ? `1-${total}/${total}` : `${from}-${to}/${total}`;
+    return `${header}
+${numbered}`;
+  }
   /**
    * 简单的 Diff 解析与替换辅助方法
    */
@@ -1230,9 +1416,13 @@ exports.ClineAgent = ClineAgent;
 exports.READ_ONLY_TOOLS = READ_ONLY_TOOLS;
 exports.TOOL_RESULT_BLOCK = TOOL_RESULT_BLOCK;
 exports.VERSION = VERSION;
+exports.collectCustomToolNames = collectCustomToolNames;
 exports.defaultTools = defaultTools;
 exports.getToolUseNames = getToolUseNames;
+exports.isCustomToolGroup = isCustomToolGroup;
 exports.parseAssistantMessageV2 = parseAssistantMessageV2;
+exports.renderCustomToolPrompt = renderCustomToolPrompt;
+exports.renderCustomToolsPrompt = renderCustomToolsPrompt;
 exports.setDynamicToolUseNames = setDynamicToolUseNames;
 exports.toolParamNames = toolParamNames;
 exports.toolUseNames = toolUseNames;