@mndrk/agx 2.0.16 → 2.0.18

package/cloud-runtime/standalone/Projects/Agents/agx-cloud/lib/stream-multiplexer.ts

@@ -1,6 +1,7 @@
 import { readFile } from "fs/promises";
 import type { Participant, ChatEvent } from "./types";
 import { runCliResponse } from "./cli-runner";
+import { register, update as updateProcess, remove as removeProcess } from "./agent-process-registry";
 import { saveMessages, setReaction, saveLogs } from "./history-store";
 import { parseReactionSignals } from "./reaction-protocol";
 import { getAttachmentMeta } from "./attachment-store";
@@ -532,6 +533,17 @@ async function runAgent(
     }, null, 2)}`,
   });
 
+  const sinceMessageId = currentUserMessageId || "";
+  register({
+    threadId,
+    agentId: p.id,
+    pid: 0,
+    state: "spawning",
+    sinceMessageId,
+    startedAt: Date.now(),
+    lastActivity: Date.now(),
+  });
+
   try {
     await runCliResponse({
       provider: p.provider,
@@ -542,17 +554,29 @@ async function runAgent(
       skills: skillsContent,
       systemContext,
       signal,
+      onSpawn: (pid) => {
+        updateProcess(threadId, p.id, { pid, state: "running", lastActivity: Date.now() });
+      },
       onLog: (stream, line) => {
         write({ type: "log", participantId: p.id, stream, line });
       },
       onDelta: (delta) => {
         fullResponse += delta;
+        updateProcess(threadId, p.id, { lastActivity: Date.now() });
       },
     });
+    updateProcess(threadId, p.id, { state: "done", lastActivity: Date.now() });
   } catch (err) {
     const message = err instanceof Error ? err.message : String(err);
+    const isAbort = signal?.aborted;
+    updateProcess(threadId, p.id, {
+      state: isAbort ? "killed" : "error",
+      lastActivity: Date.now(),
+    });
     logActivity(p.id, "error", { thread: rootMessageId || threadId, error: message });
     write({ type: "participant-error", participantId: p.id, error: message });
+  } finally {
+    removeProcess(threadId, p.id);
   }
 
   const parsed = parseReactionSignals(fullResponse);