@the-open-engine/zeroshot 6.7.0 → 6.7.2

package/src/isolation-manager.js

@@ -28,6 +28,7 @@ const { readRepoSettings } = require('../lib/repo-settings');
 const { provisionClaudeCredentials } = require('./claude-credentials');
 
 const DEFAULT_WORKTREE_SETUP_TIMEOUT_MS = 15 * 60 * 1000;
+const FRESH_BASE_REF_PREFIX = 'refs/zeroshot/base-fetch';
 
 function runSync(command, args, options = {}) {
   const timeout = options.timeout ?? 30000;
@@ -48,6 +49,61 @@ function runShellSync(command, options = {}) {
   return runSync('/bin/bash', ['-lc', command], options);
 }
 
+function resolveCommit(repoRoot, ref) {
+  return runSync('git', ['rev-parse', '--verify', `${ref}^{commit}`], {
+    cwd: repoRoot,
+    encoding: 'utf8',
+    stdio: 'pipe',
+  }).trim();
+}
+
+function deleteTemporaryBaseRef(repoRoot, temporaryRef) {
+  try {
+    runSync('git', ['update-ref', '-d', temporaryRef], {
+      cwd: repoRoot,
+      encoding: 'utf8',
+      stdio: 'pipe',
+    });
+  } catch (err) {
+    console.warn(
+      `[IsolationManager] Warning: failed to remove temporary base ref ${temporaryRef}: ${err.message}`
+    );
+  }
+}
+
+function fetchFreshRemoteBase(repoRoot, remoteName, branch) {
+  const temporaryRef = `${FRESH_BASE_REF_PREFIX}/${crypto.randomBytes(16).toString('hex')}`;
+
+  try {
+    runSync(
+      'git',
+      [
+        'fetch',
+        '--atomic',
+        '--no-tags',
+        '--no-write-fetch-head',
+        '--refmap=',
+        '--',
+        remoteName,
+        `+refs/heads/${branch}:${temporaryRef}`,
+      ],
+      {
+        cwd: repoRoot,
+        encoding: 'utf8',
+        stdio: 'pipe',
+      }
+    );
+
+    return {
+      baseSha: resolveCommit(repoRoot, temporaryRef),
+      temporaryRef,
+    };
+  } catch (err) {
+    deleteTemporaryBaseRef(repoRoot, temporaryRef);
+    throw err;
+  }
+}
+
 function expandHomePath(value) {
   if (!value) return value;
   if (value === '~') return os.homedir();
@@ -127,7 +183,7 @@ class IsolationManager {
     this.containers = new Map(); // clusterId -> containerId
     this.isolatedDirs = new Map(); // clusterId -> { path, originalDir }
     this.clusterConfigDirs = new Map(); // clusterId -> configDirPath
-    this.worktrees = new Map(); // clusterId -> { path, branch, repoRoot }
+    this.worktrees = new Map(); // clusterId -> { path, branch, repoRoot, baseRef, baseSha }
     this._exitWatchers = new Map(); // clusterId -> ChildProcess
   }
 
@@ -1443,7 +1499,7 @@ class IsolationManager {
    * Creates a git worktree at ~/.zeroshot/worktrees/{clusterId}
    * @param {string} clusterId - Cluster ID
    * @param {string} workDir - Original working directory (must be a git repo)
-   * @returns {{ path: string, branch: string, repoRoot: string }}
+   * @returns {{ path: string, branch: string, repoRoot: string, baseRef: string, baseSha: string }}
    */
   createWorktreeIsolation(clusterId, workDir, options = {}) {
     if (!this._isGitRepo(workDir)) {
@@ -1485,8 +1541,10 @@ class IsolationManager {
    * @param {string} workDir - Original working directory
    * @param {object} [options] - Worktree creation options
    * @param {string} [options.baseRef] - Git ref to base the worktree branch on
+   * @param {string} [options.remoteName=origin] - Remote used by a remote base ref
+   * @param {boolean} [options.requireFreshBase=false] - Require a freshly fetched remote base
    * @param {number} [options.worktreeSetupTimeoutMs] - Setup command timeout in milliseconds
-   * @returns {{ path: string, branch: string, repoRoot: string }}
+   * @returns {{ path: string, branch: string, repoRoot: string, baseRef: string, baseSha: string }}
    */
   createWorktree(clusterId, workDir, options = {}) {
     const repoRoot = this._getGitRoot(workDir);
@@ -1560,20 +1618,42 @@ class IsolationManager {
     }
     const worktreeSetupTimeoutMs = resolveWorktreeSetupTimeoutMs(repoSettings, options);
 
-    // Best-effort ensure origin/<branch> exists locally if requested.
-    if (worktreeBaseRef && worktreeBaseRef.startsWith('origin/')) {
-      const branch = worktreeBaseRef.slice('origin/'.length);
-      try {
-        runSync('git', ['fetch', 'origin', branch], {
-          cwd: repoRoot,
-          encoding: 'utf8',
-          stdio: 'pipe',
-        });
-      } catch {
-        // ignore
+    const baseRef = worktreeBaseRef || 'HEAD';
+    const remoteName = options.remoteName || 'origin';
+    let baseSha;
+    let temporaryBaseRef = null;
+
+    if (worktreeBaseRef && worktreeBaseRef.startsWith(`${remoteName}/`)) {
+      const branch = worktreeBaseRef.slice(remoteName.length + 1);
+      const remoteTrackingRef = `refs/remotes/${remoteName}/${branch}`;
+
+      if (options.requireFreshBase === true) {
+        const freshBase = fetchFreshRemoteBase(repoRoot, remoteName, branch);
+        baseSha = freshBase.baseSha;
+        temporaryBaseRef = freshBase.temporaryRef;
+      } else {
+        try {
+          runSync(
+            'git',
+            ['fetch', '--no-tags', '--', remoteName, `+refs/heads/${branch}:${remoteTrackingRef}`],
+            {
+              cwd: repoRoot,
+              encoding: 'utf8',
+              stdio: 'pipe',
+            }
+          );
+        } catch {
+          // Repository worktree settings historically allowed an offline cached remote ref.
+        }
+        baseSha = resolveCommit(repoRoot, remoteTrackingRef);
       }
+    } else {
+      baseSha = resolveCommit(repoRoot, baseRef);
     }
 
+    console.log(`[IsolationManager] Worktree base ref: ${baseRef}`);
+    console.log(`[IsolationManager] Worktree base commit: ${baseSha}`);
+
     // Create branch name from cluster ID (e.g., cluster-cosmic-meteor-87 -> zeroshot/cosmic-meteor-87)
     const baseBranchName = `zeroshot/${clusterId.replace(/^cluster-/, '')}`;
     let branchName = baseBranchName;
@@ -1581,44 +1661,18 @@ class IsolationManager {
     // Worktree path in persistent location (survives reboots)
     const worktreePath = path.join(os.homedir(), '.zeroshot', 'worktrees', clusterId);
 
-    // Ensure parent directory exists
-    const parentDir = path.dirname(worktreePath);
-    if (!fs.existsSync(parentDir)) {
-      fs.mkdirSync(parentDir, { recursive: true });
-    }
-
-    // Best-effort cleanup of stale worktree metadata and directory.
-    // IMPORTANT: If a previous run deleted the directory without deregistering the worktree,
-    // git may keep the branch "checked out" and block deletion/reuse.
     try {
-      runSync('git', ['worktree', 'remove', '--force', worktreePath], {
-        cwd: repoRoot,
-        encoding: 'utf8',
-        stdio: 'pipe',
-      });
-    } catch {
-      // ignore
-    }
-    try {
-      runSync('git', ['worktree', 'prune'], { cwd: repoRoot, encoding: 'utf8', stdio: 'pipe' });
-    } catch {
-      // ignore
-    }
-    try {
-      fs.rmSync(worktreePath, { recursive: true, force: true });
-    } catch {
-      // ignore
-    }
-
-    const baseRef = worktreeBaseRef || 'HEAD';
-    console.log(`[IsolationManager] Worktree base ref: ${baseRef}`);
-    console.log(`[IsolationManager] Worktree path: ${worktreePath}`);
+      // Ensure parent directory exists
+      const parentDir = path.dirname(worktreePath);
+      if (!fs.existsSync(parentDir)) {
+        fs.mkdirSync(parentDir, { recursive: true });
+      }
 
-    // Create worktree with new branch based on baseRef (retry on branch collision/in-use)
-    for (let attempt = 0; attempt < 10; attempt++) {
-      // Best-effort delete if branch exists and is not in use by another worktree.
+      // Best-effort cleanup of stale worktree metadata and directory.
+      // IMPORTANT: If a previous run deleted the directory without deregistering the worktree,
+      // git may keep the branch "checked out" and block deletion/reuse.
       try {
-        runSync('git', ['branch', '-D', branchName], {
+        runSync('git', ['worktree', 'remove', '--force', worktreePath], {
           cwd: repoRoot,
           encoding: 'utf8',
           stdio: 'pipe',
@@ -1626,38 +1680,72 @@ class IsolationManager {
       } catch {
         // ignore
       }
-
       try {
-        runSync('git', ['worktree', 'add', '-b', branchName, worktreePath, baseRef], {
+        runSync('git', ['worktree', 'prune'], {
           cwd: repoRoot,
           encoding: 'utf8',
           stdio: 'pipe',
         });
-        console.log(`[IsolationManager] Worktree setup phase: created branch ${branchName}`);
-        break;
-      } catch (err) {
-        const stderr = (
-          err && (err.stderr || err.message) ? String(err.stderr || err.message) : ''
-        ).toLowerCase();
-        const isBranchCollision =
-          stderr.includes('already exists') ||
-          stderr.includes('cannot delete branch') ||
-          stderr.includes('checked out');
-
-        if (attempt < 9 && isBranchCollision) {
-          branchName = `${baseBranchName}-${crypto.randomBytes(3).toString('hex')}`;
-          try {
-            runSync('git', ['worktree', 'prune'], {
-              cwd: repoRoot,
-              encoding: 'utf8',
-              stdio: 'pipe',
-            });
-          } catch {
-            // ignore
+      } catch {
+        // ignore
+      }
+      try {
+        fs.rmSync(worktreePath, { recursive: true, force: true });
+      } catch {
+        // ignore
+      }
+
+      console.log(`[IsolationManager] Worktree path: ${worktreePath}`);
+
+      // Create worktree with new branch based on baseRef (retry on branch collision/in-use)
+      for (let attempt = 0; attempt < 10; attempt++) {
+        // Best-effort delete if branch exists and is not in use by another worktree.
+        try {
+          runSync('git', ['branch', '-D', branchName], {
+            cwd: repoRoot,
+            encoding: 'utf8',
+            stdio: 'pipe',
+          });
+        } catch {
+          // ignore
+        }
+
+        try {
+          runSync('git', ['worktree', 'add', '-b', branchName, worktreePath, baseSha], {
+            cwd: repoRoot,
+            encoding: 'utf8',
+            stdio: 'pipe',
+          });
+          console.log(`[IsolationManager] Worktree setup phase: created branch ${branchName}`);
+          break;
+        } catch (err) {
+          const stderr = (
+            err && (err.stderr || err.message) ? String(err.stderr || err.message) : ''
+          ).toLowerCase();
+          const isBranchCollision =
+            stderr.includes('already exists') ||
+            stderr.includes('cannot delete branch') ||
+            stderr.includes('checked out');
+
+          if (attempt < 9 && isBranchCollision) {
+            branchName = `${baseBranchName}-${crypto.randomBytes(3).toString('hex')}`;
+            try {
+              runSync('git', ['worktree', 'prune'], {
+                cwd: repoRoot,
+                encoding: 'utf8',
+                stdio: 'pipe',
+              });
+            } catch {
+              // ignore
+            }
+            continue;
           }
-          continue;
+          throw err;
         }
-        throw err;
+      }
+    } finally {
+      if (temporaryBaseRef) {
+        deleteTemporaryBaseRef(repoRoot, temporaryBaseRef);
       }
     }
 
@@ -1684,6 +1772,8 @@ class IsolationManager {
       path: worktreePath,
       branch: branchName,
       repoRoot,
+      baseRef,
+      baseSha,
     };
   }
 

package/src/orchestrator.js

@@ -224,6 +224,7 @@ function buildPrOptions(options, requiredQualityGates) {
     prBase: options.prBase || null,
     mergeQueue: options.mergeQueue || false,
     closeIssue: options.closeIssue || null,
+    gitRemote: options.gitRemote || null,
     autoMerge,
     ...(requiredQualityGates.length > 0 ? { requiredQualityGates } : {}),
     cwd: options.cwd || process.cwd(),
@@ -1266,6 +1267,8 @@ class Orchestrator {
         const { detectGitContext } = require('../lib/git-remote-utils');
         const gitContext = detectGitContext(options.cwd);
         cluster.gitPlatform = gitContext?.provider || null;
+        options.gitRemote = gitContext?.remote || options.gitRemote || null;
+        cluster.prOptions = buildPrOptions(options, requiredQualityGates);
 
         if (cluster.gitPlatform) {
           this._log(`[Orchestrator] Git platform detected: ${cluster.gitPlatform.toUpperCase()}`);
@@ -1535,6 +1538,7 @@ class Orchestrator {
       const agentRole = message.content?.data?.role;
       const attempts = message.content?.data?.attempts || 1;
       const hookFailure = message.content?.data?.hookFailure === true;
+      const restartExhausted = message.content?.data?.restartExhausted === true;
 
       await this._saveClusters();
 
@@ -1542,7 +1546,7 @@ class Orchestrator {
         agentRole === 'implementation' ||
         agentRole === 'coordinator' ||
         message.sender === 'consensus-coordinator';
-      const shouldStop = shouldStopForRole && (hookFailure || attempts >= 3);
+      const shouldStop = shouldStopForRole && (hookFailure || restartExhausted || attempts >= 3);
 
       if (shouldStop) {
         this._log(`\n${'='.repeat(80)}`);
@@ -1551,6 +1555,7 @@ class Orchestrator {
         this._log(
           `${message.sender} (${agentRole || 'unknown role'}) failed` +
             (hookFailure ? ` (hookFailure=true)` : ``) +
+            (restartExhausted ? ` (restartExhausted=true)` : ``) +
             ` after ${attempts} attempts`
         );
         this._log(`Error: ${message.content?.data?.error || 'unknown'}`);
@@ -1573,7 +1578,7 @@ class Orchestrator {
     });
   }
 
-  _registerAgentLifecycleHandlers(messageBus, clusterId) {
+  _registerAgentLifecycleHandlers(messageBus, _clusterId) {
     messageBus.on('topic:AGENT_LIFECYCLE', async (message) => {
       const event = message.content?.data?.event;
       if (
@@ -1596,13 +1601,14 @@ class Orchestrator {
       const timeSinceLastOutput = message.content?.data?.timeSinceLastOutput;
       const analysis = message.content?.data?.analysis || 'No analysis available';
 
+      const consecutiveWarnings = message.content?.data?.consecutiveWarnings;
+      const warningsBeforeKill = message.content?.data?.warningsBeforeKill;
       this._log(
-        `⚠️  Orchestrator: Agent ${agentId} appears stale (${Math.round(timeSinceLastOutput / 1000)}s no output) but will NOT be killed`
+        `⚠️  Orchestrator: Agent ${agentId} has produced no output for ${Math.round(timeSinceLastOutput / 1000)}s ` +
+          `(warning ${consecutiveWarnings}/${warningsBeforeKill})`
       );
       this._log(`    Analysis: ${analysis}`);
-      this._log(
-        `    Manual intervention may be needed - use 'zeroshot resume ${clusterId}' if stuck`
-      );
+      this._log(`    Zeroshot will terminate and restart the task if inactivity persists`);
     });
   }
 
@@ -1793,18 +1799,30 @@ class Orchestrator {
       const workDir = options.cwd || process.cwd();
 
       isolationManager = new IsolationManager({});
-      // `prBase` is the PR target branch. Use that same local ref as the
-      // worktree base so repo-local integration commits are not skipped.
+      const { detectGitContext } = require('../lib/git-remote-utils');
+      const gitContext = detectGitContext(workDir);
+      if (gitContext?.remote) {
+        options.gitRemote = gitContext.remote;
+      }
+
+      // `prBase` is the PR target branch. Base isolated work on its refreshed
+      // remote-tracking ref so a stale local branch cannot change the plan.
       const worktreeOptions = {};
       if (options.prBase) {
-        worktreeOptions.baseRef = options.prBase;
-        this._log(`[Orchestrator] Using worktree base ref: ${options.prBase}`);
+        const remoteName = options.gitRemote || 'origin';
+        worktreeOptions.baseRef = `${remoteName}/${options.prBase}`;
+        if (remoteName !== 'origin') {
+          worktreeOptions.remoteName = remoteName;
+        }
+        worktreeOptions.requireFreshBase = true;
+        this._log(`[Orchestrator] Using worktree base ref: ${worktreeOptions.baseRef}`);
       }
       worktreeInfo = isolationManager.createWorktreeIsolation(clusterId, workDir, worktreeOptions);
 
       this._log(`[Orchestrator] Starting cluster in worktree isolation mode`);
       this._log(`[Orchestrator] Worktree: ${worktreeInfo.path}`);
       this._log(`[Orchestrator] Branch: ${worktreeInfo.branch}`);
+      this._log(`[Orchestrator] Worktree base commit: ${worktreeInfo.baseSha}`);
     }
 
     return { isolationManager, containerId, worktreeInfo, image: isolationImage };
@@ -1867,16 +1885,13 @@ class Orchestrator {
       closeIssue: options.closeIssue,
       requiredQualityGates: options.requiredQualityGates,
       autoMerge: resolveRunPlan(options).autoMerge,
+      gitRemote: gitContext?.remote || options.gitRemote,
+      issueNumber: inputData.number,
+      issueTitle: inputData.title,
+      includeIssueReference: !skipCloseIssue,
       cwd: options.cwd,
     });
 
-    // Template replacement for issue context
-    const issueRef = skipCloseIssue ? '' : `Closes #${inputData.number || 'unknown'}`;
-    gitPusherConfig.prompt = gitPusherConfig.prompt
-      .replace(/\{\{issue_number\}\}/g, inputData.number || 'unknown')
-      .replace(/\{\{issue_title\}\}/g, inputData.title || 'Implementation')
-      .replace(/Closes #\{\{issue_number\}\}/g, issueRef);
-
     config.agents.push(gitPusherConfig);
     this._log(
       `[Orchestrator] Injected ${platform}-git-pusher agent (issue: ${inputData.number || 'N/A'}, PR platform: ${platform.toUpperCase()})`
@@ -2361,6 +2376,15 @@ class Orchestrator {
       return;
     }
     this.closed = true;
+
+    for (const cluster of this.clusters.values()) {
+      if (typeof cluster.snapshotter?.stop === 'function') {
+        cluster.snapshotter.stop();
+      }
+      if (typeof cluster.messageBus?.close === 'function') {
+        cluster.messageBus.close();
+      }
+    }
   }
 
   /**
@@ -2637,31 +2661,68 @@ class Orchestrator {
   }
 
   _resolveFailureInfo(cluster, clusterId) {
-    if (cluster.failureInfo) {
-      return cluster.failureInfo.agentId ? cluster.failureInfo : null;
+    const persistedFailure = cluster.failureInfo?.agentId ? cluster.failureInfo : null;
+    if (persistedFailure) {
+      if (!this._hasDurableProgressAfterFailure(cluster, clusterId, persistedFailure)) {
+        return persistedFailure;
+      }
+      this._log(
+        `[Orchestrator] Ignoring recovered registry failure from ${persistedFailure.agentId}; durable task progress followed it`
+      );
     }
 
     const errors = cluster.messageBus.query({
       cluster_id: clusterId,
       topic: 'AGENT_ERROR',
-      limit: 10,
       order: 'desc',
     });
 
-    if (errors.length === 0) {
-      return null;
+    for (const error of errors) {
+      const failureInfo = {
+        agentId: error.sender,
+        taskId: error.content?.data?.taskId || null,
+        iteration: error.content?.data?.iteration || 0,
+        error: error.content?.data?.error || error.content?.text,
+        timestamp: error.timestamp,
+      };
+      if (this._hasDurableProgressAfterFailure(cluster, clusterId, failureInfo)) {
+        this._log(
+          `[Orchestrator] Ignoring recovered ledger failure from ${error.sender}; durable task progress followed it`
+        );
+        continue;
+      }
+
+      this._log(`[Orchestrator] Found failure from ledger: ${failureInfo.agentId}`);
+      return failureInfo;
     }
 
-    const firstError = errors[0];
-    const failureInfo = {
-      agentId: firstError.sender,
-      taskId: firstError.content?.data?.taskId || null,
-      iteration: firstError.content?.data?.iteration || 0,
-      error: firstError.content?.data?.error || firstError.content?.text,
-      timestamp: firstError.timestamp,
-    };
-    this._log(`[Orchestrator] Found failure from ledger: ${failureInfo.agentId}`);
-    return failureInfo;
+    return null;
+  }
+
+  _hasDurableProgressAfterFailure(cluster, clusterId, failureInfo) {
+    const failureIteration = failureInfo.iteration;
+    const failureTimestamp = failureInfo.timestamp;
+    return cluster.messageBus
+      .query({
+        cluster_id: clusterId,
+        topic: 'AGENT_LIFECYCLE',
+        sender: failureInfo.agentId,
+        since: failureTimestamp,
+      })
+      .some((message) => {
+        const event = message.content?.data?.event;
+        if (!['TASK_STARTED', 'TASK_COMPLETED'].includes(event)) {
+          return false;
+        }
+
+        const iteration = message.content?.data?.iteration;
+        return (
+          (Number.isFinite(failureTimestamp) && message.timestamp > failureTimestamp) ||
+          (Number.isInteger(failureIteration) &&
+            Number.isInteger(iteration) &&
+            iteration > failureIteration)
+        );
+      });
   }
 
   _requireFailedAgent(clusterId, cluster, failureInfo) {
@@ -3984,7 +4045,6 @@ Continue from where you left off. Review your previous output to understand what
         }
       }
 
-      // Generate platform-specific git-pusher agent from template
       const {
         generateGitPusherAgent,
         isPlatformSupported,
@@ -3996,18 +4056,19 @@ Continue from where you left off. Review your previous output to understand what
         );
       }
 
-      // Use persisted PR options from cluster state (or empty for repo settings fallback)
-      const gitPusherConfig = generateGitPusherAgent(platform, cluster.prOptions || {});
-
       // Get issue context from ledger
       const issueMsg = cluster.messageBus.ledger.findLast({ topic: 'ISSUE_OPENED' });
       const issueNumber = issueMsg?.content?.data?.number || 'unknown';
       const issueTitle = issueMsg?.content?.data?.title || 'Implementation';
 
-      // Inject issue context into prompt
-      gitPusherConfig.prompt = gitPusherConfig.prompt
-        .replace(/\{\{issue_number\}\}/g, issueNumber)
-        .replace(/\{\{issue_title\}\}/g, issueTitle);
+      // Generate the final prompt in one typed assembly pass. Issue values are
+      // resolved before the shell-quoted remote is inserted, so placeholder-like
+      // remote names cannot be rewritten by issue context.
+      const gitPusherConfig = generateGitPusherAgent(platform, {
+        ...(cluster.prOptions || {}),
+        issueNumber,
+        issueTitle,
+      });
 
       await this._opAddAgents(cluster, { agents: [gitPusherConfig] }, context);
       this._log(`    [--pr mode] Injected ${platform}-git-pusher agent`);

package/task-lib/attachable-watcher.js

@@ -5,10 +5,8 @@
  * Runs detached from parent, provides Unix socket for attach clients.
  */
 
-import { appendFileSync, existsSync, mkdirSync, unlinkSync } from 'fs';
+import { appendFileSync, unlinkSync } from 'fs';
 import { unlink } from 'fs/promises';
-import { join } from 'path';
-import { homedir } from 'os';
 import { updateTask } from './store.js';
 import {
   detectProviderFatalError,
@@ -72,6 +70,7 @@ process.on('unhandledRejection', (reason) => {
 
 const require = createRequire(import.meta.url);
 const { AttachServer } = require('../src/attach');
+const { getTaskSocketPath } = require('../src/attach/socket-paths');
 const { normalizeProviderName } = require('../lib/provider-names');
 
 const taskId = taskIdArg;
@@ -88,12 +87,7 @@ const commandSpec = config.commandSpec || {
 commandSpecCleanup = commandSpec.cleanup || [];
 let server = null;
 
-const SOCKET_DIR = join(homedir(), '.zeroshot', 'sockets');
-const socketPath = join(SOCKET_DIR, `${taskId}.sock`);
-
-if (!existsSync(SOCKET_DIR)) {
-  mkdirSync(SOCKET_DIR, { recursive: true });
-}
+const socketPath = getTaskSocketPath(taskId);
 
 function log(msg) {
   appendFileSync(logFile, msg);
@@ -319,6 +313,8 @@ server.on('exit', async ({ exitCode, signal }) => {
   try {
     await updateTask(taskId, {
       status,
+      pid: null,
+      processGroupId: null,
       exitCode: resolvedCode,
       error: fatalError || (resolvedCode !== 0 && signal ? `Killed by ${signal}` : null),
       socketPath: null,
@@ -336,7 +332,12 @@ server.on('error', async (err) => {
   log(`\nError: ${err.message}\n`);
   await cleanupCommandSpec();
   try {
-    await updateTask(taskId, { status: 'failed', error: err.message });
+    await updateTask(taskId, {
+      status: 'failed',
+      pid: null,
+      processGroupId: null,
+      error: err.message,
+    });
   } catch (updateError) {
     log(`[${Date.now()}][ERROR] Failed to update task status: ${updateError.message}\n`);
   }
@@ -358,6 +359,8 @@ try {
     pid: server.pid,
     socketPath,
     attachable: true,
+    processGroupId: process.platform === 'win32' ? null : server.pid,
+    terminationStrategy: process.platform === 'win32' ? 'process-tree' : 'process-group',
   });
 
   log(`[${Date.now()}][SYSTEM] Started with PTY (attachable)\n`);