ai-project-manage-cli 8.1.0 → 8.1.2

package/dist/index.js

@@ -1043,6 +1043,8 @@ function validateWebIdeMessagePush(o) {
   if (!nonEmptyString(o.user)) {
     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()) : void 0;
   return {
     ok: true,
     data: {
@@ -1054,7 +1056,9 @@ function validateWebIdeMessagePush(o) {
       model: o.model.trim(),
       apiKey: o.apiKey.trim(),
       workdir: o.workdir.trim(),
-      user: o.user.trim()
+      user: o.user.trim(),
+      ...useNewAgent ? { useNewAgent: true } : {},
+      ...tools && tools.length > 0 ? { 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) {
+    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,
@@ -3902,10 +3928,12 @@ function markBranchDone(taskId, workdir) {
 }
 
 // src/commands/connect/webide-pull-requests.ts
-var PRE_AGENT_PR_ACTIONS = /* @__PURE__ */ new Set(["request-accept"]);
-var POST_AGENT_PR_ACTIONS = /* @__PURE__ */ new Set(["request-organize-code"]);
-function shouldEnsureWebIdePullRequests(action) {
-  return PRE_AGENT_PR_ACTIONS.has(action);
+var POST_AGENT_PR_ACTIONS = /* @__PURE__ */ new Set([
+  "request-organize-code",
+  "request-patch"
+]);
+function shouldEnsureWebIdePullRequests(_action) {
+  return false;
 }
 function shouldEnsureWebIdePullRequestsAfterAgent(action) {
   return POST_AGENT_PR_ACTIONS.has(action);
@@ -4396,10 +4424,6 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
   );
   await updateStatus2(cfg2, messageId, "TYPING");
   syncLocalTaskStatus(workdir, taskId, msg2, "TYPING");
-  const startAction = startActionForInbound(msg2.action);
-  if (startAction) {
-    await webIdeRunAction(cfg2, taskId, startAction);
-  }
   const eventSession = new EventSession(msg2.content);
   const prepAgentId = loadWebIdeAgentId(workdir, taskId) ?? "webide-cli-prep";
   const logSyncRef = {
@@ -4435,7 +4459,12 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
       throw err;
     }
   };
+  let pausedForAssumptions = false;
   try {
+    const startAction = startActionForInbound(msg2.action);
+    if (startAction) {
+      await webIdeRunAction(cfg2, taskId, startAction);
+    }
     if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
     let assumptionAnswersText;
     if (msg2.action === "confirm-assumptions") {
@@ -4443,7 +4472,6 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
         { cfg: cfg2, taskId }
       );
     }
-    let pausedForAssumptions = false;
     await runPrepStep(
       "\u521D\u59CB\u5316\u5DE5\u4F5C\u533A",
       async () => {
@@ -4505,7 +4533,7 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
     if (shouldEnsureWebIdePullRequests(msg2.action)) {
       if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
       await runPrepStep(
-        "\u6D4B\u8BD5\u524D\u63A8\u9001",
+        "\u9A8C\u6536\u524D\u63A8\u9001",
         async () => pushWorkspaceRepos(workdir),
         (pushed) => `\u5DF2 push ${pushed} \u4E2A\u4ED3\u5E93`
       );
@@ -4611,7 +4639,15 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
     const startLocalServices = isStartLocalServicesAction(msg2.action);
     const designAction = isDesignAction(msg2.action);
     const executeSql = isExecuteSqlAction(msg2.action);
-    const savedAgentId = startLocalServices ? void 0 : designAction ? loadWebIdeDesignAgentId(workdir, taskId) : loadWebIdeAgentId(workdir, taskId);
+    const useNewAgent = msg2.useNewAgent === true;
+    if (useNewAgent && !startLocalServices) {
+      if (designAction) {
+        clearWebIdeDesignAgentId(workdir, taskId);
+      } else {
+        clearWebIdeAgentId(workdir, taskId);
+      }
+    }
+    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 persistAgentId = (agentId) => {
@@ -4653,6 +4689,7 @@ ${assumptionAnswersText}` : msg2.content,
         enablePtySessionMcp: startLocalServices,
         enableMysqlTools: executeSql,
         sqlExecutionId,
+        toolNames: Array.isArray(msg2.tools) && msg2.tools.length > 0 ? msg2.tools : void 0,
         enableSandbox: false,
         onInvalidatePersistedAgentId: startLocalServices ? void 0 : () => {
           if (designAction) {
@@ -4911,6 +4948,22 @@ ${assumptionAnswersText}` : msg2.content,
         statusErr instanceof Error ? statusErr.message : statusErr
       );
     }
+    const completeAction = completeActionForInbound(msg2.action);
+    if (completeAction && !pausedForAssumptions) {
+      try {
+        await webIdeRunAction(
+          cfg2,
+          taskId,
+          completeAction,
+          completeAction === "complete-assessment" ? { force: true } : void 0
+        );
+      } catch (completeErr) {
+        console.warn(
+          `[apm] \u5931\u8D25\u540E\u56DE\u5199 complete \u8DF3\u8FC7 action=${completeAction}:`,
+          completeErr instanceof Error ? completeErr.message : completeErr
+        );
+      }
+    }
     syncLocalTaskStatus(workdir, taskId, msg2, "FAILED");
   }
 }

package/package.json

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