ai-project-manage-cli 4.0.16 → 4.0.18

package/dist/index.js

@@ -154,6 +154,16 @@ var requestConfig = {
       path: "/requirement-artifacts/delete"
     })
   },
+  theater: {
+    reportMemberAgentProgress: defineEndpoint({
+      method: "POST",
+      path: "/theater/sessions/report-member-agent-progress"
+    }),
+    submitMemberResponse: defineEndpoint({
+      method: "POST",
+      path: "/theater/sessions/submit-member-response"
+    })
+  },
   spaceWorkflow: {
     aiStartNode: defineEndpoint({
       method: "POST",
@@ -459,6 +469,10 @@ var EventSession = class {
         return { type: "status", status: event.status, message: event.message };
     }
   }
+  /** 合并所有 assistant 片段，供剧场成员回传等场景使用 */
+  getAssistantText() {
+    return this.events.filter((e) => e.type === "assistant").map((e) => String(e.content ?? "")).join("\n").trim();
+  }
   writeToFile(cwd, requirementId, agentId) {
     const sessionsDir = resolve2(
       cwd,
@@ -768,6 +782,158 @@ async function runPull(requirementId, workspaceDir) {
 }
 
 // src/commands/connect.ts
+async function createAgentForJob(payload) {
+  const options = {
+    apiKey: payload.apiKey,
+    model: { id: payload.model ?? "default" },
+    local: { cwd: payload.cwd }
+  };
+  if (payload.agentId) {
+    return Agent.resume(payload.agentId, options);
+  }
+  return Agent.create(options);
+}
+async function runAgentStream(payload, session, agent) {
+  const run = await agent.send(payload.prompt);
+  let failed = false;
+  let failReason = "";
+  for await (const event of run.stream()) {
+    if (event.type === "status") {
+      console.log(`[Status]`, event.message || event.status);
+      if (event.status === "ERROR") {
+        failed = true;
+        failReason = event.message?.trim() ?? "";
+        console.error("[apm] Agent \u6267\u884C\u5931\u8D25:", failReason || "(\u65E0\u5931\u8D25\u539F\u56E0)");
+        break;
+      }
+      continue;
+    }
+    if (event.type === "assistant") {
+      const output = event.message.content[0].text;
+      console.log(`[Output]`, output);
+      session.addEvent(event);
+      continue;
+    }
+    if (event.type === "thinking") {
+      console.log(`[Thinking]`, event.text);
+      continue;
+    }
+    if (event.type === "tool_call") {
+      if (event.status === "completed" || event.status === "error") {
+        console.log(`[ToolCall(${event.call_id}) Result]`);
+      }
+      session.addEvent(event);
+      continue;
+    }
+    if (event.type === "task") {
+      console.log(`[Task:${event.status}]`);
+      session.addEvent(event);
+      continue;
+    }
+    if (event.type === "request") {
+      console.log(JSON.stringify(event, null, 2));
+      session.addEvent(event);
+      continue;
+    }
+    console.log("\u672A\u77E5\u4E8B\u4EF6:", JSON.stringify(event, null, 2));
+  }
+  return { agentId: run.agentId, failed, failReason };
+}
+async function handleTheaterJob(api, payload) {
+  const sessionId = payload.theaterSessionId.trim();
+  const memberKey = payload.memberKey.trim();
+  if (!sessionId) {
+    throw new Error("THEATER_JOB \u7F3A\u5C11 theaterSessionId");
+  }
+  if (!memberKey) {
+    throw new Error("THEATER_JOB \u7F3A\u5C11 memberKey");
+  }
+  console.log("[apm] \u5267\u573A\u4F1A\u8BDD:", sessionId, "\u6210\u5458:", memberKey);
+  await api.theater.reportMemberAgentProgress({
+    sessionId,
+    memberKey,
+    status: "IN_PROGRESS",
+    logId: payload.logId?.trim() || void 0
+  });
+  const eventSession = new EventSession(payload.prompt);
+  const agent = await createAgentForJob(payload);
+  const { agentId, failed, failReason } = await runAgentStream(
+    payload,
+    eventSession,
+    agent
+  );
+  const assistantText = eventSession.getAssistantText();
+  const content = failed ? failReason || assistantText || "\u672A\u77E5\u9519\u8BEF" : assistantText || "[Agent \u672A\u4EA7\u751F\u6587\u672C\u8F93\u51FA]";
+  const result = await api.theater.submitMemberResponse({
+    sessionId,
+    memberKey,
+    content,
+    agentId,
+    agentStatus: failed ? "FAILED" : "SUCCESS",
+    logId: payload.logId?.trim() || void 0
+  });
+  console.log(
+    "[apm] \u5267\u573A\u6210\u5458\u56DE\u590D\u5DF2\u63D0\u4EA4",
+    failed ? "(\u542B\u5931\u8D25\u8BF4\u660E)" : "",
+    result.clearedPending ? "\xB7 \u5DF2\u6E05\u9664\u5F85\u56DE\u590D" : ""
+  );
+  if (result.canAdvance) {
+    console.log("[apm] \u53EF\u624B\u52A8\u63A8\u8FDB\u4E3B\u6301\u8F6E\u6B21\uFF08autoPlay \u5173\u95ED\uFF09");
+  }
+}
+async function handleWorkflowJob(api, payload) {
+  const apmRoot = join6(payload.cwd, ".apm");
+  console.log("[apm] ROOT:", apmRoot);
+  console.log("[apm] workflow node:", payload.nodeId);
+  try {
+    await runPull(payload.requirementId, apmRoot);
+  } catch (pullErr) {
+    console.error("[apm] apm pull \u5931\u8D25:", pullErr);
+    throw pullErr;
+  }
+  const agent = await createAgentForJob(payload);
+  await api.cliRequirements.updateDevStatus({
+    requirementId: payload.requirementId,
+    status: "WORKING"
+  });
+  await api.spaceWorkflow.aiStartNode({
+    requirementId: payload.requirementId,
+    nodeId: payload.nodeId,
+    agentId: agent.agentId
+  });
+  const session = new EventSession(payload.prompt);
+  const { agentId, failed, failReason } = await runAgentStream(
+    payload,
+    session,
+    agent
+  );
+  await api.cliRequirements.updateDevStatus({
+    requirementId: payload.requirementId,
+    status: "IDLE"
+  });
+  session.writeToFile(payload.cwd, payload.requirementId, agentId);
+  if (failed) {
+    const failResult = await api.spaceWorkflow.aiFailNode({
+      requirementId: payload.requirementId,
+      nodeId: payload.nodeId,
+      error: failReason
+    });
+    console.log("[apm] \u5DE5\u4F5C\u6D41\u8282\u70B9\u72B6\u6001:", failResult.nodeStatus);
+    return;
+  }
+  console.log("[Done]");
+  try {
+    await runUploadArtifact(payload.requirementId, apmRoot);
+  } catch (pullErr) {
+    console.error("[apm] apm upload-artifact \u5931\u8D25:", pullErr);
+    throw pullErr;
+  }
+  const confirmResult = await api.spaceWorkflow.aiConfirmNode({
+    requirementId: payload.requirementId,
+    nodeId: payload.nodeId
+  });
+  console.log("[apm] \u5DE5\u4F5C\u6D41\u8282\u70B9\u72B6\u6001:", confirmResult.nodeStatus);
+}
 function runConnect(opts) {
   void (async () => {
     const cfg = await ensureApmConfig();
@@ -806,113 +972,14 @@ function runConnect(opts) {
       const text = typeof data === "string" ? data : data.toString();
       try {
         const msg = JSON.parse(text);
-        if (msg.type !== "JOB") {
+        if (msg.type === "THEATER_JOB") {
+          await handleTheaterJob(api, msg.payload);
           return;
         }
-        const payload = msg.payload;
-        const apmRoot = join6(payload.cwd, ".apm");
-        console.log("[apm] ROOT:", apmRoot);
-        console.log("[apm] workflow node:", payload.nodeId);
-        try {
-          await runPull(payload.requirementId, apmRoot);
-        } catch (pullErr) {
-          console.error("[apm] apm pull \u5931\u8D25:", pullErr);
-          throw pullErr;
-        }
-        const agent = await (async () => {
-          const options = {
-            apiKey: payload.apiKey,
-            model: { id: payload.model ?? "default" },
-            local: { cwd: payload.cwd }
-          };
-          if (payload.agentId) {
-            return Agent.resume(payload.agentId, options);
-          }
-          return Agent.create(options);
-        })();
-        await api.cliRequirements.updateDevStatus({
-          requirementId: payload.requirementId,
-          status: "WORKING"
-        });
-        await api.spaceWorkflow.aiStartNode({
-          requirementId: payload.requirementId,
-          nodeId: payload.nodeId,
-          agentId: agent.agentId
-        });
-        const session = new EventSession(payload.prompt);
-        const run = await agent.send(payload.prompt);
-        let agentRunFailed = false;
-        let agentFailReason = "";
-        for await (const event of run.stream()) {
-          if (event.type === "status") {
-            console.log(`[Status]`, event.message || event.status);
-            if (event.status === "ERROR") {
-              agentRunFailed = true;
-              agentFailReason = event.message?.trim() ?? "";
-              console.error(
-                "[apm] Agent \u6267\u884C\u5931\u8D25:",
-                agentFailReason || "(\u65E0\u5931\u8D25\u539F\u56E0)"
-              );
-              break;
-            }
-            continue;
-          }
-          if (event.type === "assistant") {
-            const output = event.message.content[0].text;
-            console.log(`[Output]`, output);
-            session.addEvent(event);
-            continue;
-          }
-          if (event.type === "thinking") {
-            const thinking = event.text;
-            console.log(`[Thinking]`, thinking);
-            continue;
-          }
-          if (event.type === "tool_call") {
-            if (event.status === "completed" || event.status === "error") {
-              console.log(`[ToolCall(${event.call_id}) Result]`);
-            }
-            session.addEvent(event);
-            continue;
-          }
-          if (event.type === "task") {
-            console.log(`[Task:${event.status}]`);
-            session.addEvent(event);
-            continue;
-          }
-          if (event.type === "request") {
-            console.log(JSON.stringify(event, null, 2));
-            session.addEvent(event);
-            continue;
-          }
-          console.log("\u672A\u77E5\u4E8B\u4EF6:", JSON.stringify(event, null, 2));
-        }
-        await api.cliRequirements.updateDevStatus({
-          requirementId: payload.requirementId,
-          status: "IDLE"
-        });
-        session.writeToFile(payload.cwd, payload.requirementId, run.agentId);
-        if (agentRunFailed) {
-          const failResult = await api.spaceWorkflow.aiFailNode({
-            requirementId: payload.requirementId,
-            nodeId: payload.nodeId,
-            error: agentFailReason
-          });
-          console.log("[apm] \u5DE5\u4F5C\u6D41\u8282\u70B9\u72B6\u6001:", failResult.nodeStatus);
+        if (msg.type === "JOB") {
+          await handleWorkflowJob(api, msg.payload);
           return;
         }
-        console.log("[Done]");
-        try {
-          await runUploadArtifact(payload.requirementId, apmRoot);
-        } catch (pullErr) {
-          console.error("[apm] apm upload-artifact \u5931\u8D25:", pullErr);
-          throw pullErr;
-        }
-        const confirmResult = await api.spaceWorkflow.aiConfirmNode({
-          requirementId: payload.requirementId,
-          nodeId: payload.nodeId
-        });
-        console.log("[apm] \u5DE5\u4F5C\u6D41\u8282\u70B9\u72B6\u6001:", confirmResult.nodeStatus);
       } catch (err) {
         console.error(
           "[apm] \u65E0\u6CD5\u89E3\u6790 WebSocket \u6D88\u606F:",

package/package.json

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