@straiffi/archon 1.7.0 → 1.7.2

package/dist/server/lib/agent.js

@@ -313,6 +313,7 @@ const hasMatchingUsageSnapshot = (current, next) => {
         && current.usage_snapshot_reasoning_tokens === next.usage_snapshot_reasoning_tokens;
 };
 const activeBuildProcesses = new Map();
+const activeReviewProcesses = new Map();
 const activeChatProcesses = new Map();
 const activeSpecProcesses = new Map();
 const activeSpecReviewProcesses = new Map();
@@ -1763,6 +1764,24 @@ const untrackSpecReviewProcess = (specReviewRunId, state) => {
     state.metadataRefresher = null;
 };
 export const isBuildAgentRunning = (ticketId) => activeBuildProcesses.has(ticketId);
+export const registerReviewProcess = (ticketId) => {
+    const state = { children: new Set(), stopRequested: false };
+    activeReviewProcesses.set(ticketId, state);
+    return state;
+};
+export const untrackReviewProcess = (ticketId, state) => {
+    if (activeReviewProcesses.get(ticketId) === state) {
+        activeReviewProcesses.delete(ticketId);
+    }
+};
+export const isReviewAgentRunning = (ticketId) => activeReviewProcesses.has(ticketId);
+export const isReviewStopRequested = (ticketId) => {
+    return activeReviewProcesses.get(ticketId)?.stopRequested === true;
+};
+export const stopAllReviewAgents = async () => {
+    const ticketIds = [...activeReviewProcesses.keys()];
+    await Promise.all(ticketIds.map(stopReviewAgent));
+};
 export const stopAllBuildAgents = async () => {
     const ticketIds = [...activeBuildProcesses.keys()];
     await Promise.all(ticketIds.map(stopBuildAgent));
@@ -1781,6 +1800,24 @@ export const stopBuildAgent = async (ticketId) => {
     }
     return true;
 };
+export const stopReviewAgent = async (ticketId) => {
+    const state = activeReviewProcesses.get(ticketId);
+    if (!state) {
+        return false;
+    }
+    state.stopRequested = true;
+    const children = [...state.children];
+    for (const child of children) {
+        killProcess(child, 'SIGTERM');
+    }
+    await Promise.all(children.map(child => waitForProcessExit(child, 5000)));
+    const aliveChildren = children.filter(isProcessAlive);
+    for (const child of aliveChildren) {
+        killProcess(child, 'SIGKILL');
+    }
+    await Promise.all(aliveChildren.map(child => waitForProcessExit(child, 1000)));
+    return true;
+};
 export const stopChatAgent = async (chatSessionId) => {
     const state = activeChatProcesses.get(chatSessionId);
     if (!state?.child) {
@@ -1834,7 +1871,7 @@ export const stopSpecReviewAgent = async (specReviewRunId) => {
     }
     return true;
 };
-export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = null, resumeSessionId = null, sessionField = 'session_id', trackBuildProcess = false, emitSuccessUpdate = true, opencodeAgentMode, runStream, }) => {
+export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = null, resumeSessionId = null, sessionField = 'session_id', trackBuildProcess = false, reviewProcessState, emitSuccessUpdate = true, opencodeAgentMode, runStream, }) => {
     const resolvedModel = resolveAgentModel(tool, model);
     const subagentTracker = createSubagentRunTracker({
         parentType: 'ticket',
@@ -1885,6 +1922,7 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
         let settled = false;
         let startTime = null;
         let firstFinalAnswerTime = null;
+        let reviewChild = null;
         const buildState = trackBuildProcess
             ? { child: null, stopRequested: false, sessionLookupTimer: null }
             : null;
@@ -1901,6 +1939,9 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
             if (trackBuildProcess && buildState) {
                 untrackBuildProcess(ticketId, buildState);
             }
+            if (reviewChild) {
+                reviewProcessState?.children.delete(reviewChild);
+            }
             handler(value);
         };
         const resolveOnce = settle(resolve);
@@ -1943,6 +1984,7 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
             env: { ...process.env, PWD: cwd },
             detached: process.platform !== 'win32',
         });
+        reviewChild = child;
         if (tool === 'claude') {
             const claudePrompt = opencodeAgentMode === 'plan'
                 ? appendClaudeHeadlessPlanPrompt(prompt)
@@ -1953,6 +1995,12 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
             buildState.child = child;
             activeBuildProcesses.set(ticketId, buildState);
         }
+        if (reviewProcessState) {
+            reviewProcessState.children.add(child);
+            if (reviewProcessState.stopRequested) {
+                killProcess(child, 'SIGTERM');
+            }
+        }
         startTime = Date.now();
         if (tool === 'opencode' && spawnTimestamp !== null) {
             opencodeSubagentMonitor = createOpencodeSubagentMonitor({
@@ -2018,7 +2066,7 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
                 const finalResponse = responseLines.join('\n\n') || errOutput;
                 const modelError = isModelResolutionError(finalLog);
                 const timing = startTime ? { totalMs: Date.now() - startTime } : null;
-                if (buildState?.stopRequested) {
+                if (buildState?.stopRequested || reviewProcessState?.stopRequested) {
                     const transcriptRuns = finalizeRun('stopped', finalLog, finalResponse);
                     resolveOnce(withTranscriptRuns({ response: finalResponse, responseTiming: timing, usageSnapshot }, transcriptRuns));
                     return;
@@ -2033,7 +2081,7 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
             });
             child.on('error', err => {
                 progressUpdater.cancel();
-                if (buildState?.stopRequested) {
+                if (buildState?.stopRequested || reviewProcessState?.stopRequested) {
                     const transcriptRuns = finalizeRun('stopped', null, null);
                     resolveOnce(withTranscriptRuns({ response: null, responseTiming: null, usageSnapshot }, transcriptRuns));
                     return;
@@ -2117,8 +2165,8 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
                 ? recoveredResponse
                 : responseLines.join('\n\n') || stderrOutput;
             const modelError = isModelResolutionError(log);
-            const timing = buildResponseTiming({ startTime, firstFinalAnswerTime, isStopped: Boolean(buildState?.stopRequested) });
-            if (buildState?.stopRequested) {
+            const timing = buildResponseTiming({ startTime, firstFinalAnswerTime, isStopped: Boolean(buildState?.stopRequested || reviewProcessState?.stopRequested) });
+            if (buildState?.stopRequested || reviewProcessState?.stopRequested) {
                 const transcriptRuns = finalizeRun('stopped', log, finalResponse);
                 resolveOnce(withTranscriptRuns({ response: finalResponse, responseTiming: timing }, transcriptRuns));
                 return;
@@ -2137,7 +2185,7 @@ export const runAgent = ({ ticketId, tool, prompt, cwd, io, model, variant = nul
                 clearTimeout(sessionLookupTimer);
                 sessionLookupTimer = null;
             }
-            if (buildState?.stopRequested) {
+            if (buildState?.stopRequested || reviewProcessState?.stopRequested) {
                 const transcriptRuns = finalizeRun('stopped', null, null);
                 resolveOnce(withTranscriptRuns({ response: null, responseTiming: null }, transcriptRuns));
                 return;