@trigger.dev/sdk 0.0.0-prerelease-20260909121655 → 0.0.0-prerelease-20260911094758

package/dist/commonjs/v3/ai.js

@@ -1615,15 +1615,11 @@ async function installChatInputRouter(chatId, options) {
             checkpoint.appliedThrough = Math.max(checkpoint.appliedThrough ?? replayWindowEnd, replayWindowEnd);
         }
     }
-    // A boot that replayed `.in` itself has already answered everything up to
-    // `recoveredThrough`, so the floor has to cover it before the tail opens.
-    if (options?.recoveredThrough !== undefined) {
-        const recovered = options.recoveredThrough;
-        checkpoint.resumeFrom = Math.max(checkpoint.resumeFrom ?? recovered, recovered);
-        checkpoint.appliedThrough = Math.max(checkpoint.appliedThrough ?? checkpoint.resumeFrom, checkpoint.resumeFrom);
-    }
     const router = entry.router;
     router.restore(checkpoint);
+    if (options?.recoveredSeqNums && options.recoveredSeqNums.length > 0) {
+        router.markRecovered(options.recoveredSeqNums);
+    }
     const floor = router.resumeFrom();
     if (floor !== undefined) {
         v3_1.sessionStreams.setLastSeqNum(chatId, "in", floor);
@@ -4074,6 +4070,13 @@ function chatAgent(options) {
             // default, `inFlightUsers`). The turn-loop checks this queue ahead of
             // `messagesInput.waitWithIdleTimeout` so recovered turns fire first.
             const bootInjectedQueue = [];
+            const recoveredSeqByPayload = new WeakMap();
+            const dispatchBootInjected = () => bootInjectedQueue.shift();
+            const settleRecoveredTurn = (wirePayload) => {
+                const settledSeq = recoveredSeqByPayload.get(wirePayload);
+                if (settledSeq !== undefined)
+                    chatInputRouter().settleRecovered(settledSeq);
+            };
             const couldHavePriorState = payload.continuation === true || ctx.attempt.number > 1;
             // `.in` resume cursor, computed at most once per boot. The boot
             // block below resolves it (snapshot field or records scan) and the
@@ -4214,16 +4217,10 @@ function chatAgent(options) {
             }
             // ── session.in router ──────────────────────────────────────────
             //
-            // Reads the turn boundary and subscribes in one call. `bootInCursor` is
-            // only a fallback: the boot block above may already have resolved a
-            // cursor from the snapshot, which is used when the boundary itself
-            // carries none. Everything the boot replayed off `.in` is dispatched from
-            // `bootInjectedQueue` below, so it goes into the floor here — folded in
-            // after the subscription opens, the live tail re-delivers it as a turn.
-            const lastRecoveredInSeq = replayedInTail.length > 0 ? replayedInTail[replayedInTail.length - 1].seqNum : undefined;
+            const recoveredSeqNums = replayedInTail.map((r) => r.seqNum);
             await installChatInputRouter(payload.chatId, {
                 fallbackResumeFrom: bootInCursorResolved ? bootInCursor : undefined,
-                recoveredThrough: lastRecoveredInSeq,
+                recoveredSeqNums,
                 resuming: Boolean(payload.continuation) || ctx.attempt.number > 1,
             });
             // ── Recovery boot + chain reconstruction ────────────────────────
@@ -4306,7 +4303,7 @@ function chatAgent(options) {
                 // branches: at n=1 the orphan partial is dropped and the interrupted
                 // user is re-dispatched as a fresh turn instead.
                 let seedChain;
-                let recoveredTurns;
+                let recoveredEntries;
                 if (hookChain !== undefined) {
                     seedChain = hookChain;
                 }
@@ -4317,13 +4314,26 @@ function chatAgent(options) {
                     seedChain = settledMessages;
                 }
                 if (hookRecoveredTurns !== undefined) {
-                    recoveredTurns = hookRecoveredTurns;
+                    const seqNumsByRecoveredId = new Map();
+                    for (const entry of replayedInTail) {
+                        const existing = seqNumsByRecoveredId.get(entry.message.id);
+                        if (existing)
+                            existing.push(entry.seqNum);
+                        else
+                            seqNumsByRecoveredId.set(entry.message.id, [entry.seqNum]);
+                    }
+                    recoveredEntries = hookRecoveredTurns.map((message) => ({
+                        message,
+                        seqNum: seqNumsByRecoveredId.get(message.id)?.shift(),
+                    }));
                 }
                 else if (partialAssistant !== undefined && inFlightUsers.length > 1) {
-                    recoveredTurns = inFlightUsers.slice(1);
+                    recoveredEntries = replayedInTail
+                        .slice(1)
+                        .map((r) => ({ message: r.message, seqNum: r.seqNum }));
                 }
                 else {
-                    recoveredTurns = inFlightUsers;
+                    recoveredEntries = replayedInTail.map((r) => ({ message: r.message, seqNum: r.seqNum }));
                 }
                 // `beforeBoot` errors bubble — the customer opted into blocking
                 // persistence and a failure there should fail the run rather than
@@ -4353,13 +4363,14 @@ function chatAgent(options) {
                 for (const entry of replayedInTail) {
                     metadataById.set(entry.message.id, entry.metadata);
                 }
-                for (const msg of recoveredTurns) {
+                const dispatchedRecoveredSeqs = new Set();
+                for (const { message: msg, seqNum } of recoveredEntries) {
                     if (wireMessageId && msg.id === wireMessageId)
                         continue;
                     const recoveredMetadata = metadataById.has(msg.id)
                         ? metadataById.get(msg.id)
                         : payload.metadata;
-                    bootInjectedQueue.push({
+                    const injectedPayload = {
                         chatId: payload.chatId,
                         sessionId: payload.sessionId,
                         metadata: recoveredMetadata,
@@ -4368,7 +4379,17 @@ function chatAgent(options) {
                         messageId: msg.id,
                         continuation: payload.continuation,
                         previousRunId: payload.previousRunId,
-                    });
+                    };
+                    bootInjectedQueue.push(injectedPayload);
+                    if (seqNum !== undefined) {
+                        recoveredSeqByPayload.set(injectedPayload, seqNum);
+                        dispatchedRecoveredSeqs.add(seqNum);
+                    }
+                }
+                for (const entry of replayedInTail) {
+                    if (!dispatchedRecoveredSeqs.has(entry.seqNum)) {
+                        chatInputRouter().settleRecovered(entry.seqNum);
+                    }
                 }
                 accumulatedUIMessages = seedChain;
                 // ── Head-start bootstrap ─────────────────────────────────────
@@ -4528,7 +4549,7 @@ function chatAgent(options) {
                  */
                 let dispatchedRecoveredFirstTurn = false;
                 if (preloaded && bootInjectedQueue.length > 0) {
-                    currentWirePayload = bootInjectedQueue.shift();
+                    currentWirePayload = dispatchBootInjected();
                     dispatchedRecoveredFirstTurn = true;
                 }
                 // Handle preloaded runs — fire onPreload, then wait for the first real message
@@ -4736,7 +4757,7 @@ function chatAgent(options) {
                     // waiting on the live session.in. Subsequent recovered turns
                     // get drained by the end-of-turn picker below.
                     if (bootInjectedQueue.length > 0) {
-                        currentWirePayload = bootInjectedQueue.shift();
+                        currentWirePayload = dispatchBootInjected();
                     }
                     else {
                         const effectiveIdleTimeout = idleTimeoutInSeconds ?? payload.idleTimeoutInSeconds;
@@ -5300,6 +5321,7 @@ function chatAgent(options) {
                                     chatId: currentWirePayload.chatId,
                                     messageId: currentWirePayload.messageId,
                                 });
+                                settleRecoveredTurn(currentWirePayload);
                                 await writeTurnCompleteChunk(currentWirePayload.chatId);
                                 // Not a turn — don't consume an iteration.
                                 turn--;
@@ -6015,6 +6037,7 @@ function chatAgent(options) {
                                     }
                                     locals_js_1.locals.set(chatResponsePartsKey, []);
                                 }
+                                settleRecoveredTurn(currentWirePayload);
                                 // Write turn-complete control chunk — closes the frontend stream.
                                 const turnCompleteResult = await writeTurnCompleteChunk(currentWirePayload.chatId, turnAccessToken);
                                 // Fire onTurnComplete — stream is closed, use for persistence.
@@ -6126,7 +6149,7 @@ function chatAgent(options) {
                             // produced these from in-flight user messages on session.in
                             // that the dead predecessor never acknowledged.
                             if (bootInjectedQueue.length > 0) {
-                                currentWirePayload = bootInjectedQueue.shift();
+                                currentWirePayload = dispatchBootInjected();
                                 return "continue";
                             }
                             // chat.requestUpgrade() was called — exit the loop; the handover
@@ -6446,7 +6469,7 @@ function chatAgent(options) {
                         // recovered turn shouldn't strand the rest of the boot queue
                         // until an unrelated live message arrives.
                         if (bootInjectedQueue.length > 0) {
-                            currentWirePayload = bootInjectedQueue.shift();
+                            currentWirePayload = dispatchBootInjected();
                             continue;
                         }
                         // Wait for the next message — same as after a successful turn