ai-project-manage-cli 8.1.1 → 8.1.3

package/dist/index.js

@@ -1044,6 +1044,7 @@ function validateWebIdeMessagePush(o) {
     return { ok: false, reason: "webide-message \u7F3A\u5C11 user" };
   }
   const useNewAgent = o.useNewAgent === true;
+  const tools = Array.isArray(o.tools) ? o.tools.filter((t) => typeof t === "string" && t.trim().length > 0).map((t) => t.trim()) : [];
   return {
     ok: true,
     data: {
@@ -1056,7 +1057,8 @@ function validateWebIdeMessagePush(o) {
       apiKey: o.apiKey.trim(),
       workdir: o.workdir.trim(),
       user: o.user.trim(),
-      ...useNewAgent ? { useNewAgent: true } : {}
+      ...useNewAgent ? { useNewAgent: true } : {},
+      tools
     }
   };
 }

package/dist/worker-script.js

@@ -2250,12 +2250,22 @@ AskQuestion \u5DF2\u901A\u8FC7 MCP \u670D\u52A1\u5668 custom-user-tools \u6CE8\u
 var WORKSPACE_BOUNDARY_HINT = `[\u5DE5\u4F5C\u533A\u8FB9\u754C]
 \u6240\u6709\u6587\u4EF6\u8BFB\u5199\u3001\u641C\u7D22\uFF08Grep/Glob/Read\uFF09\u3001Shell \u5FC5\u987B\u4E25\u683C\u9650\u5236\u5728\u5F53\u524D\u5DE5\u4F5C\u533A cwd \u5185\u3002
 \u7981\u6B62\u8BBF\u95EE\u5DE5\u4F5C\u533A\u5916\u8DEF\u5F84\uFF08\u5982\u5176\u5B83\u9879\u76EE\u3001~/.cursor\u3001agent-transcripts\u3001\u7CFB\u7EDF\u76EE\u5F55\u7B49\uFF09\u3002`;
+function wantsTool(name, options, legacyDefault) {
+  if (options.toolNames !== void 0) {
+    return options.toolNames.includes(name);
+  }
+  return legacyDefault;
+}
 function createCursorCustomTools(cfg2, options) {
+  const enableAppend = options.enableAppendMessage !== false && Boolean(options.appendMessageContent);
+  const enableAsk = options.enableAskQuestion !== false;
+  const enablePlan = options.enableWebIdePlanTools === true;
+  const enableMysql = options.enableMysqlTools === true;
   const tools = {
-    ...options.enableAppendMessage !== false && options.appendMessageContent ? createAppendMessageCustomTools({
+    ...wantsTool("AppendMessage", options, enableAppend) && options.appendMessageContent ? createAppendMessageCustomTools({
       appendContent: options.appendMessageContent
     }) : {},
-    ...options.enableAskQuestion !== false ? createAskQuestionTool({
+    ...wantsTool("AskQuestion", options, enableAsk) ? createAskQuestionTool({
       onInvoke: options?.onAskQuestion,
       execute: options?.askQuestionExecute
     }) : {}
@@ -2263,24 +2273,39 @@ function createCursorCustomTools(cfg2, options) {
   if (options?.taskId && options.workdir) {
     const { cli } = createApmApiClient(cfg2);
     const taskId = options.taskId;
-    if (options.enableWebIdePlanTools) {
+    if (wantsTool("RecommendPlan", options, enablePlan)) {
       Object.assign(
         tools,
         createRecommendPlanTool({
           webideRecommendPlan: (args) => cli.webideRecommendPlan({ taskId, ...args })
-        }),
+        })
+      );
+    }
+    if (wantsTool("UpsertWebIdePlan", options, enablePlan)) {
+      Object.assign(
+        tools,
         createUpsertWebIdePlanTool({
           webideUpsertPlan: (args) => cli.webideUpsertPlan({ taskId, ...args })
-        }),
+        })
+      );
+    }
+    if (wantsTool("UpsertWebIdeTestCases", options, enablePlan)) {
+      Object.assign(
+        tools,
         createUpsertWebIdeTestCasesTool({
           webideReplaceTestCases: (args) => cli.webideReplaceTestCases({ taskId, ...args })
-        }),
+        })
+      );
+    }
+    if (wantsTool("MergeWebIdePullRequests", options, enablePlan)) {
+      Object.assign(
+        tools,
         createMergeWebIdePullRequestsTool({
           webideMergePullRequests: () => cli.webideMergePullRequests({ taskId })
         })
       );
     }
-    if (options.enableMysqlTools && options.sqlExecutionId && options.workdir) {
+    if (wantsTool("MysqlExecute", options, enableMysql) && options.sqlExecutionId && options.workdir) {
       Object.assign(
         tools,
         createMysqlExecuteTool({
@@ -2440,6 +2465,7 @@ async function runCursorAgent(cfg2, ctx, options) {
     enableWebIdePlanTools: options.enableWebIdePlanTools,
     enableMysqlTools: options.enableMysqlTools,
     sqlExecutionId: options.sqlExecutionId,
+    toolNames: options.toolNames,
     taskId: options.taskId,
     workdir,
     enablePackageStatusTools: options.enablePackageStatusTools,
@@ -4366,9 +4392,6 @@ function isStartLocalServicesAction(_action) {
 function isDesignAction(action) {
   return action === "request-design" || action === "request-revise-design";
 }
-function isExecuteSqlAction(action) {
-  return action === "request-execute-sql";
-}
 function syncLocalTaskStatus(workdir, taskId, msg2, status) {
   syncWebIdeTaskState(workdir, taskId, {
     messageId: msg2.messageId,
@@ -4612,8 +4635,8 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
     await syncPrepLog();
     const startLocalServices = isStartLocalServicesAction(msg2.action);
     const designAction = isDesignAction(msg2.action);
-    const executeSql = isExecuteSqlAction(msg2.action);
     const useNewAgent = msg2.useNewAgent === true;
+    const toolNames = Array.isArray(msg2.tools) ? msg2.tools.map((t) => String(t).trim()).filter(Boolean) : [];
     if (useNewAgent && !startLocalServices) {
       if (designAction) {
         clearWebIdeDesignAgentId(workdir, taskId);
@@ -4622,8 +4645,8 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
       }
     }
     const savedAgentId = startLocalServices || useNewAgent ? void 0 : designAction ? loadWebIdeDesignAgentId(workdir, taskId) : loadWebIdeAgentId(workdir, taskId);
-    const promptPayload = executeSql ? parseWebIdePromptPayload(msg2.content) : null;
-    const sqlExecutionId = executeSql && typeof promptPayload?.executionId === "string" ? promptPayload.executionId.trim() : void 0;
+    const promptPayload = toolNames.includes("MysqlExecute") ? parseWebIdePromptPayload(msg2.content) : null;
+    const sqlExecutionId = toolNames.includes("MysqlExecute") && typeof promptPayload?.executionId === "string" ? promptPayload.executionId.trim() : void 0;
     const persistAgentId = (agentId) => {
       if (startLocalServices) return;
       if (designAction) {
@@ -4651,18 +4674,21 @@ ${assumptionAnswersText}` : msg2.content,
         forceSend: true,
         eventSession,
         appendMessageContent: (content) => appendContent(cfg2, messageId, content),
-        askQuestionExecute: startLocalServices ? void 0 : createWebIdeAskQuestionExecute({
+        askQuestionExecute: !startLocalServices && toolNames.includes("AskQuestion") ? createWebIdeAskQuestionExecute({
           cfg: cfg2,
           taskId,
           onAwaitingAssumptions: () => {
             pausedForAssumptions = true;
           }
-        }),
-        enableAskQuestion: !startLocalServices && !executeSql,
-        enableWebIdePlanTools: !startLocalServices && !executeSql && !designAction,
+        }) : void 0,
+        // 工具名单完全以服务端模板配置为准，不再按 action 硬编码开关
+        enableAppendMessage: false,
+        enableAskQuestion: false,
+        enableWebIdePlanTools: false,
+        enableMysqlTools: false,
         enablePtySessionMcp: startLocalServices,
-        enableMysqlTools: executeSql,
         sqlExecutionId,
+        toolNames,
         enableSandbox: false,
         onInvalidatePersistedAgentId: startLocalServices ? void 0 : () => {
           if (designAction) {
@@ -4721,7 +4747,7 @@ ${assumptionAnswersText}` : msg2.content,
         detail,
         tokenUsage
       );
-      if (executeSql && sqlExecutionId) {
+      if (sqlExecutionId) {
         try {
           await api.cli.webideFinalizeSqlExecution({
             executionId: sqlExecutionId,
@@ -4746,7 +4772,7 @@ ${assumptionAnswersText}` : msg2.content,
         "\u5DF2\u53D6\u6D88",
         tokenUsage
       );
-      if (executeSql && sqlExecutionId) {
+      if (sqlExecutionId) {
         try {
           await api.cli.webideFinalizeSqlExecution({
             executionId: sqlExecutionId,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-project-manage-cli",
-  "version": "8.1.1",
+  "version": "8.1.3",
   "description": "命令行工具：后续用于调用平台后端 API 完成运维与自动化操作",
   "type": "module",
   "private": false,