ai-project-manage-cli 8.1.10 → 8.1.12

package/dist/index.js

@@ -1203,6 +1203,25 @@ function validateCancel(o) {
     data: { type: "cancel", messageId: o.messageId.trim() }
   };
 }
+function validateSteer(o) {
+  if (o.type !== "steer") {
+    return { ok: false, reason: "\u671F\u671B steer" };
+  }
+  if (!nonEmptyString(o.messageId)) {
+    return { ok: false, reason: "steer \u7F3A\u5C11 messageId" };
+  }
+  if (!nonEmptyString(o.text)) {
+    return { ok: false, reason: "steer \u7F3A\u5C11 text" };
+  }
+  return {
+    ok: true,
+    data: {
+      type: "steer",
+      messageId: o.messageId.trim(),
+      text: o.text.trim()
+    }
+  };
+}
 function validateWebIdeStopTerminals(o) {
   if (o.type !== "webide-stop-terminals") {
     return { ok: false, reason: "\u671F\u671B webide-stop-terminals" };
@@ -1319,7 +1338,7 @@ function validateAgentWsMessage(value, kind) {
   }
   const o = value;
   const type = o.type;
-  if (type !== "heartbeat" && type !== "webide-message" && type !== "cancel" && type !== "webide-start-project" && type !== "webide-stop-terminals" && type !== "webide-package") {
+  if (type !== "heartbeat" && type !== "webide-message" && type !== "cancel" && type !== "steer" && type !== "webide-start-project" && type !== "webide-stop-terminals" && type !== "webide-package") {
     return { ok: false, reason: `\u672A\u77E5 type: ${String(type)}` };
   }
   if (kind === "heartbeat" || type === "heartbeat") {
@@ -1328,6 +1347,9 @@ function validateAgentWsMessage(value, kind) {
   if (type === "cancel") {
     return validateCancel(o);
   }
+  if (type === "steer") {
+    return validateSteer(o);
+  }
   if (type === "webide-start-project") {
     return validateWebIdeStartProjectPush(o);
   }
@@ -1347,6 +1369,29 @@ function handleCancelAction(payload, ctx) {
   ctx.activeRuns.get(messageId)?.abort();
 }
 
+// src/commands/connect/actions/steer.ts
+function registerRunSteer(ctx, messageId, target) {
+  const send = (text) => target.steer(messageId, text);
+  ctx.steerRuns.set(messageId, send);
+  const queued = ctx.pendingSteers.get(messageId);
+  if (!queued) return;
+  ctx.pendingSteers.delete(messageId);
+  send(queued);
+}
+function unregisterRunSteer(ctx, messageId) {
+  ctx.steerRuns.delete(messageId);
+  ctx.pendingSteers.delete(messageId);
+}
+function handleSteerAction(payload, ctx) {
+  const send = ctx.steerRuns.get(payload.messageId);
+  if (!send) {
+    ctx.pendingSteers.set(payload.messageId, payload.text);
+    console.log(`[apm] steer \u5DF2\u6392\u961F messageId=${payload.messageId}`);
+    return;
+  }
+  send(payload.text);
+}
+
 // src/commands/connect/core/worker/worker.ts
 import { dirname as dirname3, join as join7 } from "path";
 import { fileURLToPath as fileURLToPath2 } from "url";
@@ -1473,6 +1518,9 @@ var ActionWorker = class {
   cancel(msg_id) {
     this.worker.postMessage({ type: "cancel", messageId: msg_id });
   }
+  steer(msg_id, text) {
+    this.worker.postMessage({ type: "steer", messageId: msg_id, text });
+  }
   get done() {
     return this.promise;
   }
@@ -1512,6 +1560,7 @@ function handleWebIdeMessageAction(payload, ctx) {
   }
   const task = (async () => {
     const handle = new ActionWorker(payload);
+    registerRunSteer(ctx, payload.messageId, handle);
     const onAbort = () => handle.cancel(payload.messageId);
     perMessageController.signal.addEventListener("abort", onAbort, {
       once: true
@@ -1520,6 +1569,7 @@ function handleWebIdeMessageAction(payload, ctx) {
       await handle.done;
     } finally {
       perMessageController.signal.removeEventListener("abort", onAbort);
+      unregisterRunSteer(ctx, payload.messageId);
       ctx.activeRuns.delete(payload.messageId);
     }
   })();
@@ -1552,6 +1602,7 @@ function handleStartProjectAction(payload, ctx) {
   }
   const task = (async () => {
     const handle = new ActionWorker(payload);
+    registerRunSteer(ctx, payload.messageId, handle);
     const onAbort = () => handle.cancel(payload.messageId);
     perMessageController.signal.addEventListener("abort", onAbort, {
       once: true
@@ -1560,6 +1611,7 @@ function handleStartProjectAction(payload, ctx) {
       await handle.done;
     } finally {
       perMessageController.signal.removeEventListener("abort", onAbort);
+      unregisterRunSteer(ctx, payload.messageId);
       ctx.activeRuns.delete(payload.messageId);
     }
   })();
@@ -1688,6 +1740,7 @@ function handleWebIdePackageAction(payload, ctx) {
   }
   const task = (async () => {
     const handle = new ActionWorker(payload);
+    registerRunSteer(ctx, payload.messageId, handle);
     const onAbort = () => handle.cancel(payload.messageId);
     perMessageController.signal.addEventListener("abort", onAbort, {
       once: true
@@ -1696,6 +1749,7 @@ function handleWebIdePackageAction(payload, ctx) {
       await handle.done;
     } finally {
       perMessageController.signal.removeEventListener("abort", onAbort);
+      unregisterRunSteer(ctx, payload.messageId);
       ctx.activeRuns.delete(payload.messageId);
     }
   })();
@@ -1710,6 +1764,9 @@ function dispatchConnectAction(payload, ctx) {
     case "cancel":
       handleCancelAction(payload, ctx);
       return;
+    case "steer":
+      handleSteerAction(payload, ctx);
+      return;
     case "webide-message":
       handleWebIdeMessageAction(payload, ctx);
       return;
@@ -1934,6 +1991,8 @@ async function runConnect(options) {
   const activeTasks = /* @__PURE__ */ new Set();
   const activeRuns = /* @__PURE__ */ new Map();
   const pendingCancels = /* @__PURE__ */ new Set();
+  const steerRuns = /* @__PURE__ */ new Map();
+  const pendingSteers = /* @__PURE__ */ new Map();
   let currentConnectionAbort = null;
   const connectCtx = {
     cfg,
@@ -1941,6 +2000,8 @@ async function runConnect(options) {
     activeTasks,
     activeRuns,
     pendingCancels,
+    steerRuns,
+    pendingSteers,
     isShuttingDown: () => shuttingDown
   };
   const drainAndExit = async (code = 0) => {

package/dist/worker-script.js

@@ -1996,6 +1996,14 @@ var MinioClient = class {
   async removeObject(bucket, objectKey) {
     await this.inner.removeObject(bucket, objectKey);
   }
+  async getObjectBuffer(bucket, objectKey) {
+    const stream = await this.inner.getObject(bucket, objectKey);
+    const chunks = [];
+    for await (const chunk of stream) {
+      chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
+    }
+    return Buffer.concat(chunks);
+  }
 };
 
 // src/commands/connect/apm-log-minio.ts
@@ -2460,6 +2468,10 @@ function createPtySessionMcpServers(cfg2) {
 }
 
 // src/commands/connect/remote-local-agent-store.ts
+var INLINE_CHECKPOINT_MAX_BYTES = 256 * 1024;
+function checkpointObjectKey(agentId, blobId) {
+  return `cursor-agents/${agentId}/checkpoints/${blobId}`;
+}
 function asCheckpointRef(value) {
   if (value === void 0) return void 0;
   if (value === null) return null;
@@ -2534,6 +2546,16 @@ function base64ToBytes(dataBase64) {
 }
 function createRemoteLocalAgentStore(cfg2) {
   const api = createApmApiClient(cfg2);
+  let minioCache = null;
+  function getMinio() {
+    if (!minioCache) {
+      minioCache = createApmLogMinioClient(cfg2).then(({ client, bucket }) => ({
+        client,
+        bucket
+      }));
+    }
+    return minioCache;
+  }
   async function upsertAgent(agent) {
     const row = await api.cli.cursorStoreUpsertAgent({
       agentId: agent.agentId,
@@ -2570,13 +2592,44 @@ function createRemoteLocalAgentStore(cfg2) {
     return fromApiRun(row);
   }
   async function putCheckpoint(input) {
+    const createdAt = Date.now();
+    if (input.data.byteLength > INLINE_CHECKPOINT_MAX_BYTES) {
+      const { client, bucket } = await getMinio();
+      const objectKey = checkpointObjectKey(input.agentId, input.blobId);
+      await client.putObject(bucket, objectKey, Buffer.from(input.data), {
+        "Content-Type": "application/octet-stream"
+      });
+      await api.cli.cursorStorePutCheckpoint({
+        agentId: input.agentId,
+        blobId: input.blobId,
+        objectKey,
+        byteSize: input.data.byteLength,
+        createdAt
+      });
+      return;
+    }
     await api.cli.cursorStorePutCheckpoint({
       agentId: input.agentId,
       blobId: input.blobId,
       dataBase64: bytesToBase64(input.data),
       byteSize: input.data.byteLength,
-      createdAt: Date.now()
+      createdAt
+    });
+  }
+  async function readCheckpointBytes(input) {
+    const row = await api.cli.cursorStoreGetCheckpoint({
+      agentId: input.agentId,
+      blobId: input.blobId
     });
+    if (!row) return null;
+    if (row.dataBase64) {
+      return base64ToBytes(row.dataBase64);
+    }
+    const objectKey = row.objectKey?.trim();
+    if (!objectKey) return null;
+    const { client, bucket } = await getMinio();
+    const buf = await client.getObjectBuffer(bucket, objectKey);
+    return new Uint8Array(buf);
   }
   return {
     agents: {
@@ -2732,12 +2785,7 @@ function createRemoteLocalAgentStore(cfg2) {
     },
     checkpoints: {
       async get(input) {
-        const row = await api.cli.cursorStoreGetCheckpoint({
-          agentId: input.agentId,
-          blobId: input.blobId
-        });
-        if (!row?.dataBase64) return null;
-        return base64ToBytes(row.dataBase64);
+        return readCheckpointBytes(input);
       },
       async create(input) {
         await putCheckpoint(input);
@@ -2783,6 +2831,30 @@ function createRemoteLocalAgentStore(cfg2) {
   };
 }
 
+// src/commands/connect/core/worker/run-steer.ts
+var impl = null;
+var pending = null;
+function bindRunSteer(fn) {
+  impl = fn;
+  if (!pending) return;
+  const text = pending;
+  pending = null;
+  void fn(text);
+}
+function clearRunSteer() {
+  impl = null;
+  pending = null;
+}
+function deliverRunSteer(text) {
+  const trimmed = text.trim();
+  if (!trimmed) return;
+  if (impl) {
+    void impl(trimmed);
+    return;
+  }
+  pending = trimmed;
+}
+
 // src/commands/connect/cursor-agent.ts
 var noopRemoteLogSync = {
   schedule(_session) {
@@ -2914,6 +2986,22 @@ async function runCursorAgent(cfg2, ctx, options) {
     });
     activeRun = run;
     console.log(`[apm] Cursor run id=${run.id} agentId=${agent.agentId}`);
+    bindRunSteer(async (text) => {
+      const steer = run.steer;
+      if (!steer) {
+        console.warn(`[apm] \u5F53\u524D run \u4E0D\u652F\u6301 steer runId=${run.id}`);
+        return;
+      }
+      try {
+        const outcome = await steer(text);
+        console.log(`[apm] steer outcome=${outcome} runId=${run.id}`);
+      } catch (err) {
+        console.error(
+          `[apm] steer \u5931\u8D25 runId=${run.id}:`,
+          err instanceof Error ? err.message : err
+        );
+      }
+    });
     await options.onRunStarted?.({ agentId: agent.agentId, runId: run.id });
     let lastRunErrorStatus;
     for await (const event of run.stream()) {
@@ -2999,6 +3087,7 @@ async function runCursorAgent(cfg2, ctx, options) {
     }
     throw err;
   } finally {
+    clearRunSteer();
     signal?.removeEventListener("abort", abortRun);
     await agent[Symbol.asyncDispose]();
   }
@@ -5062,6 +5151,11 @@ 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 sqlExecutionId = toolNames.includes("MysqlExecute") && typeof promptPayload?.executionId === "string" ? promptPayload.executionId.trim() : void 0;
+    if (toolNames.includes("MysqlExecute") && !sqlExecutionId) {
+      console.warn(
+        "[apm] \u5DE5\u5177\u540D\u5355\u542B MysqlExecute\uFF0C\u4F46 prompt \u672B\u5C3E\u6CA1\u6709 executionId\uFF0C\u672C\u8F6E\u4E0D\u4F1A\u6302\u8F7D\u8BE5\u5DE5\u5177"
+      );
+    }
     const persistAgentId = (agentId) => {
       if (startLocalServices) return;
       if (designAction) {
@@ -5171,7 +5265,7 @@ ${assumptionAnswersText}` : msg2.content,
         detail,
         tokenUsage
       );
-      if (sqlExecutionId) {
+      if (sqlExecutionId && msg2.action !== "request-bugfix") {
         try {
           await api.cli.webideFinalizeSqlExecution({
             executionId: sqlExecutionId,
@@ -5196,7 +5290,7 @@ ${assumptionAnswersText}` : msg2.content,
         "\u5DF2\u53D6\u6D88",
         tokenUsage
       );
-      if (sqlExecutionId) {
+      if (sqlExecutionId && msg2.action !== "request-bugfix") {
         try {
           await api.cli.webideFinalizeSqlExecution({
             executionId: sqlExecutionId,
@@ -5417,11 +5511,19 @@ var cfg = tryReadApmConfig();
 var { msg } = workerData;
 if (!cfg || !msg) throw new Error("worker-script \u7F3A\u5C11 workerData");
 var controller = new AbortController();
-port.on("message", (raw) => {
-  if (raw?.type === "cancel" && raw.messageId === msg.messageId) {
-    controller.abort();
+port.on(
+  "message",
+  (raw) => {
+    if (raw?.messageId !== msg.messageId) return;
+    if (raw.type === "cancel") {
+      controller.abort();
+      return;
+    }
+    if (raw.type === "steer" && typeof raw.text === "string") {
+      deliverRunSteer(raw.text);
+    }
   }
-});
+);
 var reply = (body) => port.postMessage(body);
 try {
   await dispatchWorkerJob(cfg, msg, controller.signal);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-project-manage-cli",
-  "version": "8.1.10",
+  "version": "8.1.12",
   "description": "命令行工具：后续用于调用平台后端 API 完成运维与自动化操作",
   "type": "module",
   "private": false,
@@ -31,7 +31,7 @@
     "@bufbuild/protobuf": "1.10.0",
     "@connectrpc/connect": "~1.7.0",
     "@connectrpc/connect-node": "~1.7.0",
-    "@cursor/sdk": "^1.0.22",
+    "@cursor/sdk": "^1.0.32",
     "commander": "~14.0.3",
     "listpage-http": "~0.0.318",
     "minio": "~8.0.7",