@thegitai/cli 1.0.0-preview.52 → 1.0.0-preview.53

package/dist/src/background-jobs.js

@@ -403,6 +403,10 @@ export function drainBackgroundJobNotifications({ sessionId } = {}) {
     const hasReadableFinalOutput = pending.some((snapshot) => snapshot.status !== 'killed');
     return `${lines.join(' ')}${hasReadableFinalOutput ? ' Use shell_job_output to read any final output.' : ''}`;
 }
+export function hasUnreportedBackgroundJobUpdate(id, { sessionId } = {}) {
+    const filter = sessionFilter(sessionId);
+    return pendingModelNotifications.some((snapshot) => snapshot.id === id && (!filter || snapshot.sessionId === filter));
+}
 function unknownJobError(id) {
     const known = listBackgroundJobs().map((job) => job.id);
     const hint = known.length

package/dist/src/job-follow-up.js

@@ -1,11 +1,40 @@
-import { getBackgroundJob } from './background-jobs.js';
+import { getBackgroundJob, hasUnreportedBackgroundJobUpdate } from './background-jobs.js';
 export const MAX_CHAINED_JOB_FOLLOW_UPS = 3;
+function exitedOnItsOwn(id, sessionId) {
+    const job = getBackgroundJob(id, { sessionId });
+    return job !== null && job.status !== 'running' && job.status !== 'killed';
+}
 export function decideJobFollowUp(jobIds, chainedFollowUps, sessionId) {
-    const finished = jobIds.some((id) => {
-        const job = getBackgroundJob(id, { sessionId });
-        return job !== null && job.status !== 'running' && job.status !== 'killed';
-    });
-    if (!finished)
+    if (!jobIds.some((id) => exitedOnItsOwn(id, sessionId)))
         return 'wait';
     return chainedFollowUps >= MAX_CHAINED_JOB_FOLLOW_UPS ? 'stop' : 'start';
 }
+export function remainingJobWait(jobIds, sessionId) {
+    return jobIds.filter((id) => {
+        const job = getBackgroundJob(id, { sessionId });
+        if (job?.status === 'running')
+            return true;
+        return exitedOnItsOwn(id, sessionId) && hasUnreportedBackgroundJobUpdate(id, { sessionId });
+    });
+}
+export function createJobWaitTracker(sessionId) {
+    let waiting = [];
+    let chained = 0;
+    return {
+        get waiting() {
+            return waiting;
+        },
+        turnStarted(jobFollowUp) {
+            chained = jobFollowUp ? chained + 1 : 0;
+        },
+        turnEnded(waitingForJobs) {
+            waiting = waitingForJobs ? [...waitingForJobs] : remainingJobWait(waiting, sessionId());
+        },
+        takeDecision() {
+            const decision = decideJobFollowUp(waiting, chained, sessionId());
+            if (decision !== 'wait')
+                waiting = [];
+            return decision;
+        },
+    };
+}

package/dist/src/ui/repl.js

@@ -14,7 +14,7 @@ import { agentModeLabel, nextAgentMode, } from '../agent-mode.js';
 import { chat, models } from '../api/index.js';
 import { isTurnCancelledError, isTurnOutcomeKnown } from '../api/chat.js';
 import { formatTurnRecoveryStatus } from '../api/turn-recovery.js';
-import { decideJobFollowUp } from '../job-follow-up.js';
+import { createJobWaitTracker } from '../job-follow-up.js';
 import { getJobBufferedOutput, getJobOutputPreview, hasRunningBackgroundJobs, killAllBackgroundJobs, killBackgroundJob, listBackgroundJobs, setBackgroundJobSession, setBackgroundJobUpdateHook, } from '../background-jobs.js';
 import { clearTodos, listTodos, setTodoSession } from '../todo-list.js';
 import { setScratchSession } from '../scratch-dir.js';
@@ -2492,8 +2492,7 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
                 rollback('turn ended — sending as the next prompt');
             }
         };
-        let waitingJobIds = [];
-        let chainedJobFollowUps = 0;
+        const jobWait = createJobWaitTracker(() => session.sessionId);
         const handleSubmit = async (rawInput, { jobFollowUp = false } = {}) => {
             const chunks = jobFollowUp ? [] : store.getState().pastedChunks;
             const expanded = chunks.length
@@ -2795,8 +2794,7 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
                 await flushQueuedMessage();
                 return;
             }
-            waitingJobIds = [];
-            chainedJobFollowUps = jobFollowUp ? chainedJobFollowUps + 1 : 0;
+            jobWait.turnStarted(jobFollowUp);
             latestUsageSummary = null;
             disarmExitConfirm();
             unacknowledged = new Map();
@@ -2917,7 +2915,7 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
                     status: result.waitingForApproval ? 'Awaiting approval' : 'Ready',
                     tokenUsage: formatClientTokenUsage(Date.now() - turnStartedAt, latestUsageSummary),
                 }));
-                waitingJobIds = result.waitingForJobs ?? [];
+                jobWait.turnEnded(result.waitingForJobs);
                 recoverUnacknowledgedMessages();
                 await flushQueuedMessage();
             }
@@ -3110,16 +3108,15 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
             }));
         };
         const scheduleJobFollowUp = () => {
-            if (waitingJobIds.length === 0)
+            if (jobWait.waiting.length === 0)
                 return;
             setTimeout(async () => {
                 const state = store.getState();
                 if (exiting || state.busy || state.modelPickerOpen || state.resumePickerOpen)
                     return;
-                const decision = decideJobFollowUp(waitingJobIds, chainedJobFollowUps, session.sessionId);
+                const decision = jobWait.takeDecision();
                 if (decision === 'wait')
                     return;
-                waitingJobIds = [];
                 if (decision === 'stop') {
                     appendStaticEntry({
                         body: 'A background job finished. Send a message to continue.',
@@ -3212,7 +3209,7 @@ export async function runClientInteractive({ appendPromptHistory, authConfig, de
         };
         const unsubscribe = store.subscribe(() => {
             renderCurrentFrame();
-            if (waitingJobIds.length > 0 && !store.getState().busy)
+            if (jobWait.waiting.length > 0 && !store.getState().busy)
                 scheduleJobFollowUp();
         });
         const spinnerTimer = setInterval(() => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@thegitai/cli",
-  "version": "1.0.0-preview.52",
+  "version": "1.0.0-preview.53",
   "description": "TheGitAI is an agentic AI coding tool for your terminal. It reads and searches your repository, writes and edits files, runs your tests, and verifies the change before handing it back.",
   "keywords": [
     "agentic-ai",
@@ -45,11 +45,11 @@
     "@lydell/node-pty-linux-x64": "1.1.0",
     "@lydell/node-pty-win32-arm64": "1.1.0",
     "@lydell/node-pty-win32-x64": "1.1.0",
-    "@thegitai/tui-darwin-arm64": "1.0.0-preview.52",
-    "@thegitai/tui-darwin-x64": "1.0.0-preview.52",
-    "@thegitai/tui-linux-arm64": "1.0.0-preview.52",
-    "@thegitai/tui-linux-x64": "1.0.0-preview.52",
-    "@thegitai/tui-win32-x64": "1.0.0-preview.52",
+    "@thegitai/tui-darwin-arm64": "1.0.0-preview.53",
+    "@thegitai/tui-darwin-x64": "1.0.0-preview.53",
+    "@thegitai/tui-linux-arm64": "1.0.0-preview.53",
+    "@thegitai/tui-linux-x64": "1.0.0-preview.53",
+    "@thegitai/tui-win32-x64": "1.0.0-preview.53",
     "@vscode/ripgrep": "1.18.0"
   },
   "repository": {