@amical/aiagent-sdk 0.1.10 → 0.1.12

package/dist/runtime/nodeRuntime.js

@@ -167,9 +167,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."
     }
   },
   {
@@ -516,6 +518,44 @@ var toolParamNames = [
   "start_line",
   "end_line"
 ];
+
+// 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");
+}
 var TEXT_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
   "txt",
   "md",
@@ -574,7 +614,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";
@@ -599,9 +649,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 内部事件
@@ -621,33 +681,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;
@@ -669,10 +743,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 };
@@ -704,16 +779,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);
@@ -739,6 +804,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() {
@@ -781,9 +868,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({
@@ -828,7 +915,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);
@@ -850,7 +963,8 @@ ${content.trim()}\`\`\``);
             if (path3) {
               try {
                 const content = await this.runtime.fs.readFile(path3);
-                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}` });
               }
@@ -863,6 +977,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 path3 = tool.params.path;
@@ -874,12 +990,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 path3 = tool.params.path || this.runtime.env.cwd();
+            const recursive = tool.params.recursive === "true";
             try {
-              const files = await this.runtime.fs.readDir(path3);
-              toolResults.push({ tool, result: files.map((f) => f.name + (f.isDirectory() ? "/" : "")).join("\n") });
+              const files = await this.runtime.fs.readDir(path3, { 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}` });
             }
@@ -1019,10 +1141,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.` });
@@ -1089,6 +1216,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 解析与替换辅助方法
    */
@@ -1338,6 +1524,9 @@ var NodeApiHandler = class {
       };
     }
   }
+  clearConversationId() {
+    this.conversationId = void 0;
+  }
 };
 
 // src/prompt/NodePromptManager.ts
@@ -1450,10 +1639,11 @@ ${context.customInstructions}
     if (allowDefaultTools) {
       useDefaultTools = defaultTools.filter((tool) => allowDefaultTools.includes(tool.name));
     }
+    const renderedCustomTools = renderCustomToolsPrompt(customTools, context.cwd);
     return `# Available Tools
 You can use the following tools by formatting your output as XML blocks. Only use ONE tool per message, and ALWAYS wait for the user's response with the execution result before proceeding.
 
-${[...useDefaultTools, ...customTools].map((tool) => {
+${[...useDefaultTools].map((tool) => {
       return `### ${tool.name}
 Description: ${tool.description}
 Parameters: 
@@ -1466,6 +1656,8 @@ ${Object.entries(tool.params).map(([name, description]) => `<${name}>${descripti
 </${tool.name}>
 `;
     }).join("\n")}
+${renderedCustomTools ? `
+${renderedCustomTools}` : ""}
 
 # Tool Use Examples
 ${useDefaultTools.filter((item) => Boolean(item.example)).map((item) => item.example).join("\n")}}
@@ -1639,14 +1831,125 @@ ${this?.getEnvironmentInfoPrompt?.(context)}`;
 4. Ensure the list accurately reflects the current state`;
   }
 };
+
+// src/memory/condense.ts
+var TOOL_RESULT_CONTENT_REGEX = /<tool_call_result[^>]*>[\s\S]*?<\/tool_call_result>/;
+var EXPIRED_MARKER_TEXT = "[The result has expired and has been removed from the context]";
+var ATTEMPT_TOOL_NAME = "agent_attempt_completion" /* ATTEMPT */;
+function getContentString(content) {
+  return typeof content === "string" ? content : "";
+}
+function isToolResultMessage(content) {
+  if (typeof content !== "string") {
+    return false;
+  }
+  return TOOL_RESULT_CONTENT_REGEX.test(content);
+}
+function isAttemptCallMessage(message) {
+  if (message.role !== "assistant") {
+    return false;
+  }
+  const content = getContentString(message.content);
+  return new RegExp(`<${ATTEMPT_TOOL_NAME}[\\s>]`).test(content) || new RegExp(`<name>\\s*${ATTEMPT_TOOL_NAME}\\s*</name>`).test(content);
+}
+function markExpiredToolResult(content) {
+  if (typeof content !== "string") {
+    return content;
+  }
+  return content.replace(
+    /(<tool_call_result[^>]*tool_name="([^"]*)"[^>]*>)[\s\S]*?(<\/tool_call_result>)/g,
+    (_match, openTag, _toolName, closeTag) => {
+      return `${openTag}
+${EXPIRED_MARKER_TEXT}
+${closeTag}`;
+    }
+  );
+}
+function splitConversations(messages) {
+  const conversations = [];
+  let current = [];
+  for (const message of messages) {
+    current.push(message);
+    if (isAttemptCallMessage(message)) {
+      conversations.push(current);
+      current = [];
+    }
+  }
+  if (current.length > 0) {
+    conversations.push(current);
+  }
+  return conversations;
+}
+function isCompleteConversation(conversation) {
+  if (conversation.length === 0) {
+    return false;
+  }
+  return isAttemptCallMessage(conversation[conversation.length - 1]);
+}
+function compressCompleteConversation(conversation) {
+  if (conversation.length <= 2) {
+    return conversation;
+  }
+  const head = conversation[0];
+  const tail = conversation[conversation.length - 1];
+  const mergedMarker = {
+    role: "user",
+    content: "[History Condensed] The intermediate tool calls and tool call results of this conversation have been merged. Please refer to the `agent_attempt_completion` summary below for the final conclusion."
+  };
+  return [head, mergedMarker, tail];
+}
+function markExpiredFirstHalf(messages) {
+  const half = Math.floor(messages.length / 2);
+  return messages.map((message, index) => {
+    if (index < half && isToolResultMessage(message.content)) {
+      return { ...message, content: markExpiredToolResult(message.content) };
+    }
+    return message;
+  });
+}
+function condenseMessages(messages, maxMessages) {
+  if (messages.length <= maxMessages) {
+    return messages;
+  }
+  const systemMessages = messages.filter((message) => message.role === "system");
+  const restMessages = messages.filter((message) => message.role !== "system");
+  const conversations = splitConversations(restMessages);
+  const completeIndices = [];
+  for (let i = 0; i < conversations.length; i++) {
+    if (isCompleteConversation(conversations[i])) {
+      completeIndices.push(i);
+    }
+  }
+  const currentLength = () => systemMessages.length + conversations.reduce((acc, c) => acc + c.length, 0);
+  if (completeIndices.length >= 2) {
+    const lastCompleteIndex = completeIndices[completeIndices.length - 1];
+    for (const idx of completeIndices) {
+      if (idx === lastCompleteIndex) {
+        break;
+      }
+      conversations[idx] = compressCompleteConversation(conversations[idx]);
+      if (currentLength() <= maxMessages) {
+        break;
+      }
+    }
+  }
+  let condensed = conversations.flat();
+  condensed = markExpiredFirstHalf(condensed);
+  if (systemMessages.length + condensed.length > maxMessages) {
+    const nonSystemBudget = Math.max(0, maxMessages - systemMessages.length);
+    condensed = condensed.slice(-nonSystemBudget);
+  }
+  return [...systemMessages, ...condensed];
+}
 var NodeMemoryManager = class {
   filePath;
   maxMessages;
+  onCondense;
   /**
    * @param dbPath 存储历史记录的本地文件路径（例如：'./.cline_memory.json'）
-   * @param maxMessages 保留的最大消息数量，用于简单的历史截断（默认 150）
+   * @param maxMessages 保留的最大消息数量，用于简单的历史截断（默认 200）
    */
-  constructor(dbPath, maxMessages = 150) {
+  constructor(dbPath, maxMessages = 200) {
     this.filePath = path__namespace.resolve(dbPath);
     this.maxMessages = maxMessages;
   }
@@ -1681,19 +1984,22 @@ var NodeMemoryManager = class {
     await this.writeMessages(messages);
     await this.condenseHistory();
   }
+  /**
+   * 压缩历史：
+   * - system 消息保持不变
+   * - 以 agent_attempt_completion 为边界切分完整对话，仅合并“后面还有其他完整对话”的旧完整对话
+   * - 其余需要压缩的场景下，对前半段工具调用结果标记为“结果已过期”
+   * - 同时触发 onCondense 回调（宿主可据此清空 dify conversationId 新开窗口）
+   */
   async condenseHistory() {
     const messages = await this.readMessages();
-    if (messages.length > this.maxMessages) {
-      const firstMessage = messages[0];
-      const isFirstSystem = firstMessage && firstMessage.role === "system";
-      const sliceCount = messages.length - this.maxMessages;
-      let condensed;
-      if (isFirstSystem) {
-        condensed = [firstMessage, ...messages.slice(sliceCount + 1)];
-      } else {
-        condensed = messages.slice(sliceCount);
-      }
-      await this.writeMessages(condensed);
+    const condensed = condenseMessages(messages, this.maxMessages);
+    if (condensed === messages) {
+      return;
+    }
+    await this.writeMessages(condensed);
+    if (this.onCondense) {
+      await this.onCondense();
     }
   }
   async clear() {
@@ -1913,10 +2219,14 @@ exports.NodePromptManager = NodePromptManager;
 exports.READ_ONLY_TOOLS = READ_ONLY_TOOLS;
 exports.TOOL_RESULT_BLOCK = TOOL_RESULT_BLOCK;
 exports.VERSION = VERSION;
+exports.collectCustomToolNames = collectCustomToolNames;
 exports.createNodeAgent = createNodeAgent;
 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;