ai-project-manage-cli 8.1.13 → 8.1.14

package/dist/index.js

@@ -133,6 +133,11 @@ var requestConfig = {
       method: "PUT",
       path: "/cli/webide/transition-prep-events"
     }),
+    /** 「继续」开跑后挂新 runId */
+    webideAttachContinueRun: defineEndpoint({
+      method: "PUT",
+      path: "/cli/webide/continue-run"
+    }),
     webideUpsertMessageLog: defineEndpoint({
       method: "PUT",
       path: "/cli/webide/message-logs"

package/dist/worker-script.js

@@ -44,6 +44,11 @@ var requestConfig = {
       method: "PUT",
       path: "/cli/webide/transition-prep-events"
     }),
+    /** 「继续」开跑后挂新 runId */
+    webideAttachContinueRun: defineEndpoint({
+      method: "PUT",
+      path: "/cli/webide/continue-run"
+    }),
     webideUpsertMessageLog: defineEndpoint({
       method: "PUT",
       path: "/cli/webide/message-logs"
@@ -2610,6 +2615,12 @@ function bytesToBase64(data) {
 function base64ToBytes(dataBase64) {
   return new Uint8Array(Buffer.from(dataBase64, "base64"));
 }
+var pendingCheckpointUploads = /* @__PURE__ */ new Map();
+async function flushPendingCheckpointUploads() {
+  const pending2 = [...pendingCheckpointUploads.values()];
+  if (pending2.length === 0) return;
+  await Promise.allSettled(pending2);
+}
 function createRemoteLocalAgentStore(cfg2) {
   const api = createApmApiClient(cfg2);
   let minioCache = null;
@@ -2666,7 +2677,7 @@ function createRemoteLocalAgentStore(cfg2) {
     });
     return fromApiRun(row);
   }
-  async function putCheckpoint(input) {
+  async function uploadCheckpointRemote(input) {
     const createdAt = Date.now();
     if (input.data.byteLength > INLINE_CHECKPOINT_MAX_BYTES) {
       const { client, bucket } = await getMinio();
@@ -2681,7 +2692,6 @@ function createRemoteLocalAgentStore(cfg2) {
         byteSize: input.data.byteLength,
         createdAt
       });
-      rememberCheckpoint(input.agentId, input.blobId, input.data);
       return;
     }
     await api.cli.cursorStorePutCheckpoint({
@@ -2691,7 +2701,30 @@ function createRemoteLocalAgentStore(cfg2) {
       byteSize: input.data.byteLength,
       createdAt
     });
-    rememberCheckpoint(input.agentId, input.blobId, input.data);
+  }
+  async function putCheckpoint(input) {
+    const data = new Uint8Array(input.data);
+    rememberCheckpoint(input.agentId, input.blobId, data);
+    const key = checkpointCacheKey(input.agentId, input.blobId);
+    if (pendingCheckpointUploads.has(key)) return;
+    const startedAt = Date.now();
+    const uploadPromise = uploadCheckpointRemote({
+      agentId: input.agentId,
+      blobId: input.blobId,
+      data
+    }).then(() => {
+      console.log(
+        `[apm] checkpoint \u8FDC\u7A0B\u4E0A\u4F20\u5B8C\u6210 agentId=${input.agentId} blobId=${input.blobId} bytes=${data.byteLength} ms=${Date.now() - startedAt}`
+      );
+    }).catch((err) => {
+      console.warn(
+        `[apm] checkpoint \u8FDC\u7A0B\u4E0A\u4F20\u5931\u8D25\uFF08\u672C\u5730\u5DF2\u7F13\u5B58\uFF09 agentId=${input.agentId} blobId=${input.blobId}:`,
+        err instanceof Error ? err.message : err
+      );
+    }).finally(() => {
+      pendingCheckpointUploads.delete(key);
+    });
+    pendingCheckpointUploads.set(key, uploadPromise);
   }
   async function readCheckpointBytes(input) {
     const memKey = checkpointCacheKey(input.agentId, input.blobId);
@@ -4281,6 +4314,7 @@ import { resolve as resolve7 } from "node:path";
 async function purgeCursorAgentStoreForAgent(cfg2, agentId) {
   const trimmedAgentId = agentId.trim();
   if (!trimmedAgentId) return false;
+  await flushPendingCheckpointUploads();
   const store = createRemoteLocalAgentStore(cfg2);
   const runIds = [];
   let cursor;
@@ -4934,26 +4968,36 @@ function completeActionForInbound(action) {
   switch (action) {
     case "request-assessment":
     case "confirm-assumptions":
+    case "start-assessment":
+    case "continue-assessment":
       return "complete-assessment";
     case "request-design":
     case "request-revise-design":
+    case "start-design":
       return "complete-design";
     case "request-write-plan":
     case "request-revise-plan":
+    case "start-write-plan":
       return "complete-write-plan";
     case "request-develop":
     case "skip-plan":
+    case "start-develop":
       return "complete-develop";
     case "request-start-project":
+    case "start-project":
       return "complete-project";
     case "request-execute-sql":
+    case "start-execute-sql":
       return "complete-execute-sql";
     case "request-bugfix":
+    case "start-bugfix":
       return "complete-bugfix";
     case "request-organize-code":
     case "request-patch":
+    case "start-organize-code":
       return "complete-organize-code";
     case "request-accept":
+    case "start-accept":
       return "complete-accept";
     default:
       return null;
@@ -4964,6 +5008,13 @@ async function webIdeRunAction(cfg2, taskId, action, params) {
   await api.cli.webideRunAction({ taskId, action, params });
   console.log(`[apm] webide runAction ok action=${action} taskId=${taskId}`);
 }
+async function webIdeAttachContinueRun(cfg2, taskId, runId) {
+  const api = createApmApiClient(cfg2);
+  await api.cli.webideAttachContinueRun({ taskId, runId });
+  console.log(
+    `[apm] webide attachContinueRun ok runId=${runId} taskId=${taskId}`
+  );
+}
 async function webIdeSyncTransitionPrepEvents(cfg2, taskId, startAction, prepEvents) {
   try {
     const api = createApmApiClient(cfg2);
@@ -5036,8 +5087,11 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
   const messageId = msg2.messageId;
   const taskId = msg2.taskId;
   const branchCacheKey = taskId;
+  const continuePayload = msg2.action === "continue" ? parseWebIdePromptPayload(msg2.content) : null;
+  const previousAction = typeof continuePayload?.previousAction === "string" ? continuePayload.previousAction.trim() : "";
+  const effectiveAction = msg2.action === "continue" && previousAction ? previousAction : msg2.action;
   console.log(
-    `[apm] webide-message action=${msg2.action} taskId=${taskId} messageId=${messageId}`
+    `[apm] webide-message action=${msg2.action}${previousAction ? ` previousAction=${previousAction}` : ""} taskId=${taskId} messageId=${messageId}`
   );
   await updateStatus2(cfg2, messageId, "TYPING");
   syncLocalTaskStatus(workdir, taskId, msg2, "TYPING");
@@ -5140,7 +5194,7 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
       eventSession.addCliStep("\u51C6\u5907\u7279\u6027\u5206\u652F", "\u5DF2\u7F13\u5B58\uFF0C\u8DF3\u8FC7", "skip");
       await syncPrepLog();
     }
-    if (shouldEnsureWebIdePullRequests(msg2.action)) {
+    if (shouldEnsureWebIdePullRequests(effectiveAction)) {
       if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
       await runPrepStep(
         "\u9A8C\u6536\u524D\u63A8\u9001",
@@ -5246,8 +5300,8 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
       "running"
     );
     await syncPrepLog();
-    const startLocalServices = isStartLocalServicesAction(msg2.action);
-    const designAction = isDesignAction(msg2.action);
+    const startLocalServices = isStartLocalServicesAction(effectiveAction);
+    const designAction = isDesignAction(effectiveAction);
     const useNewAgent = msg2.useNewAgent === true;
     const toolNames = Array.isArray(msg2.tools) ? msg2.tools.map((t) => String(t).trim()).filter(Boolean) : [];
     if (useNewAgent && !startLocalServices) {
@@ -5258,7 +5312,7 @@ async function handleWebIdeInboundMessage(cfg2, msg2, signal) {
       }
     }
     const savedAgentId = startLocalServices || useNewAgent ? void 0 : designAction ? loadWebIdeDesignAgentId(workdir, taskId) : loadWebIdeAgentId(workdir, taskId);
-    const promptPayload = toolNames.includes("MysqlExecute") ? parseWebIdePromptPayload(msg2.content) : null;
+    const promptPayload = toolNames.includes("MysqlExecute") ? continuePayload ?? parseWebIdePromptPayload(msg2.content) : null;
     const sqlExecutionId = toolNames.includes("MysqlExecute") && typeof promptPayload?.executionId === "string" ? promptPayload.executionId.trim() : void 0;
     if (toolNames.includes("MysqlExecute") && !sqlExecutionId) {
       console.warn(
@@ -5348,6 +5402,15 @@ ${assumptionAnswersText}` : msg2.content,
               eventSession.getCliSteps()
             );
             await webIdeRunAction(cfg2, taskId, startAction, { runId });
+          } else if (msg2.action === "continue") {
+            try {
+              await webIdeAttachContinueRun(cfg2, taskId, runId);
+            } catch (err) {
+              console.warn(
+                "[apm] attach continue run failed:",
+                err instanceof Error ? err.message : err
+              );
+            }
           }
         }
       }
@@ -5374,7 +5437,7 @@ ${assumptionAnswersText}` : msg2.content,
         detail,
         tokenUsage
       );
-      if (sqlExecutionId && msg2.action !== "request-bugfix") {
+      if (sqlExecutionId && effectiveAction !== "request-bugfix") {
         try {
           await api.cli.webideFinalizeSqlExecution({
             executionId: sqlExecutionId,
@@ -5399,7 +5462,7 @@ ${assumptionAnswersText}` : msg2.content,
         "\u5DF2\u53D6\u6D88",
         tokenUsage
       );
-      if (sqlExecutionId && msg2.action !== "request-bugfix") {
+      if (sqlExecutionId && effectiveAction !== "request-bugfix") {
         try {
           await api.cli.webideFinalizeSqlExecution({
             executionId: sqlExecutionId,
@@ -5426,7 +5489,7 @@ ${assumptionAnswersText}` : msg2.content,
       id: messageId,
       content: fallback
     });
-    if ((msg2.action === "write-plan" || msg2.action === "revise-plan") && outcome.createPlan && outcome.createPlan.trim()) {
+    if ((effectiveAction === "write-plan" || effectiveAction === "revise-plan" || effectiveAction === "request-write-plan" || effectiveAction === "request-revise-plan" || effectiveAction === "start-write-plan") && outcome.createPlan && outcome.createPlan.trim()) {
       try {
         await api.cli.webideUpsertPlan({
           taskId,
@@ -5434,12 +5497,12 @@ ${assumptionAnswersText}` : msg2.content,
         });
       } catch (err) {
         console.warn(
-          `[apm] ${msg2.action} \u8865\u5199\u8BA1\u5212\u5931\u8D25:`,
+          `[apm] ${effectiveAction} \u8865\u5199\u8BA1\u5212\u5931\u8D25:`,
           err instanceof Error ? err.message : err
         );
       }
     }
-    if (isDesignAction(msg2.action)) {
+    if (isDesignAction(effectiveAction)) {
       if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
       try {
         const uploaded = await uploadWebIdeDesignArtifacts(
@@ -5448,7 +5511,7 @@ ${assumptionAnswersText}` : msg2.content,
           workdir
         );
         console.log(
-          `[apm] webide \u8BBE\u8BA1\u4EA7\u7269\u4E0A\u4F20 action=${msg2.action} uploaded=${uploaded.uploaded} skipped=${uploaded.skipped} files=${uploaded.artifacts.map((a) => a.name).join(",")}`
+          `[apm] webide \u8BBE\u8BA1\u4EA7\u7269\u4E0A\u4F20 action=${effectiveAction} uploaded=${uploaded.uploaded} skipped=${uploaded.skipped} files=${uploaded.artifacts.map((a) => a.name).join(",")}`
         );
       } catch (err) {
         const detail = err instanceof Error ? err.message : String(err);
@@ -5463,27 +5526,27 @@ ${assumptionAnswersText}` : msg2.content,
         return;
       }
     }
-    if (shouldCommitAfterWebIdeMessage(msg2.action)) {
+    if (shouldCommitAfterWebIdeMessage(effectiveAction)) {
       if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
       const committed = await commitWorkspaceReposIfDirty(
         workdir,
-        `chore(webide): sync after ${msg2.action} (${taskId})`
+        `chore(webide): sync after ${effectiveAction} (${taskId})`
       );
       console.log(
-        `[apm] webide \u515C\u5E95 commit action=${msg2.action} repos=${committed}`
+        `[apm] webide \u515C\u5E95 commit action=${effectiveAction} repos=${committed}`
       );
       if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
       const pushed = await pushWorkspaceRepos(workdir);
       console.log(
-        `[apm] webide \u81EA\u52A8 push action=${msg2.action} repos=${pushed}`
+        `[apm] webide \u81EA\u52A8 push action=${effectiveAction} repos=${pushed}`
       );
     }
-    if (shouldEnsureWebIdePullRequestsAfterAgent(msg2.action)) {
+    if (shouldEnsureWebIdePullRequestsAfterAgent(effectiveAction)) {
       if (signal.aborted) throw new Error("\u4EFB\u52A1\u5DF2\u53D6\u6D88");
-      if (!shouldCommitAfterWebIdeMessage(msg2.action)) {
+      if (!shouldCommitAfterWebIdeMessage(effectiveAction)) {
         const pushed = await pushWorkspaceRepos(workdir);
         console.log(
-          `[apm] webide create-pr \u524D push action=${msg2.action} repos=${pushed}`
+          `[apm] webide create-pr \u524D push action=${effectiveAction} repos=${pushed}`
         );
       }
       const prResult = await ensureWebIdePullRequests(cfg2, taskId, workdir);
@@ -5537,20 +5600,20 @@ ${assumptionAnswersText}` : msg2.content,
       null,
       tokenUsage
     );
-    const completeAction = completeActionForInbound(msg2.action);
+    const completeAction = completeActionForInbound(effectiveAction);
     if (completeAction && !pausedForAssumptions) {
       await webIdeRunAction(cfg2, taskId, completeAction);
     } else if (pausedForAssumptions) {
       console.log(
-        `[apm] \u672C\u8F6E\u505C\u5728\u5F85\u786E\u8BA4\u5047\u8BBE\uFF0C\u8DF3\u8FC7 complete action=${msg2.action} taskId=${taskId}`
+        `[apm] \u672C\u8F6E\u505C\u5728\u5F85\u786E\u8BA4\u5047\u8BBE\uFF0C\u8DF3\u8FC7 complete action=${effectiveAction} taskId=${taskId}`
       );
     }
     await updateStatus2(cfg2, messageId, "SUCCESS");
     syncLocalTaskStatus(workdir, taskId, msg2, "SUCCESS");
     console.log(
-      `[apm] webide-message \u5B8C\u6210 action=${msg2.action} messageId=${messageId} agentId=${outcome.agentId}`
+      `[apm] webide-message \u5B8C\u6210 action=${msg2.action}${previousAction ? ` previousAction=${previousAction}` : ""} messageId=${messageId} agentId=${outcome.agentId}`
     );
-    if (msg2.action === "request-accept") {
+    if (effectiveAction === "request-accept" || effectiveAction === "start-accept") {
       try {
         await cleanWebIdeWorkspaceCache(taskId, workdir, cfg2);
       } catch (err) {
@@ -5577,7 +5640,7 @@ ${assumptionAnswersText}` : msg2.content,
         statusErr instanceof Error ? statusErr.message : statusErr
       );
     }
-    const completeAction = completeActionForInbound(msg2.action);
+    const completeAction = completeActionForInbound(effectiveAction);
     if (completeAction === "complete-assessment" && !pausedForAssumptions) {
       try {
         await webIdeRunAction(cfg2, taskId, completeAction, { force: true });

package/package.json

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