@trigger.dev/sdk 4.6.2 → 4.6.4

package/dist/commonjs/v3/ai.d.ts

@@ -415,6 +415,8 @@ export type ChatNewToolResult = {
 interface CompactionState {
     summary: string;
     baseResponseMessageCount: number;
+    /** Completed steps summarized this turn; unlike message counts, matches UI step markers. */
+    baseResponseStepCount: number;
 }
 /**
  * Event passed to `summarize` callbacks.
@@ -1454,13 +1456,17 @@ export type TurnCompleteEvent<TClientData = unknown, TUIM extends UIMessage = UI
      */
     uiMessages: TUIM[];
     /**
-     * Only the new model messages from this turn (user message(s) + assistant response).
-     * Useful for appending to an existing conversation record.
+     * Model messages for this turn's user message(s) and complete assistant response,
+     * including steps summarized during the turn. Same-ID approval and handover
+     * continuations include the full replacement response, so these are not always
+     * an append-only delta. Persist `messages` for future model context, or upsert
+     * `newUIMessages` by ID for the visible conversation.
      */
     newMessages: ModelMessage[];
     /**
-     * Only the new UI messages from this turn (user message(s) + assistant response).
-     * Useful for inserting individual message records instead of overwriting the full history.
+     * New or updated UI messages from this turn (user message(s) + assistant response).
+     * Upsert by message ID: approval and handover continuations can replace an
+     * existing assistant message.
      */
     newUIMessages: TUIM[];
     /** The assistant's response for this turn, with aborted parts cleaned up when `stopped` is true. Undefined if `pipeChat` was used manually. */

package/dist/commonjs/v3/ai.js

@@ -18,6 +18,7 @@ const api_1 = require("@opentelemetry/api");
 const sessionTracing_js_1 = require("./sessionTracing.js");
 const ai_runtime_js_1 = require("../imports/ai-runtime.js");
 const transcriptStorage_js_1 = require("./transcriptStorage.js");
+const compactionResponse_js_1 = require("./compactionResponse.js");
 let transcriptStorageOverride;
 /**
  * Test-only override for the storage `chat.agent` persists through, so a
@@ -2803,6 +2804,7 @@ async function chatCompact(messages, steps, options) {
                 locals_js_1.locals.set(chatCompactionStateKey, {
                     summary,
                     baseResponseMessageCount: currentStep.response.messages.length,
+                    baseResponseStepCount: steps.length,
                 });
                 // Set model-only override — UI messages stay intact for persistence.
                 // The summary becomes the model message history for the next turn,
@@ -5912,6 +5914,13 @@ function chatAgent(options) {
                                 // Check if compaction set a model-only override (preserves UI messages).
                                 // Apply compactUIMessages/compactModelMessages callbacks if configured.
                                 const modelOnlyOverride = locals_js_1.locals.get(chatOverrideModelMessagesKey);
+                                const responseCompaction = modelOnlyOverride
+                                    ? locals_js_1.locals.get(chatCompactionStateKey)
+                                    : undefined;
+                                // Capture the original assistant before compactUIMessages can remove it.
+                                const originalResponse = responseCompaction && capturedResponseMessage
+                                    ? accumulatedUIMessages.find((m) => m.id === capturedResponseMessage?.id)
+                                    : undefined;
                                 if (modelOnlyOverride) {
                                     const compactionSummary = locals_js_1.locals.get(chatCompactionStateKey)?.summary ?? "";
                                     const taskCompactionConfig = locals_js_1.locals.get(chatAgentCompactionKey);
@@ -6003,10 +6012,31 @@ function chatAgent(options) {
                                         // rationale (TRI-9137).
                                         recordToolCallIdsFromMessage(capturedResponseMessage);
                                         try {
+                                            const responseForModel = (0, compactionResponse_js_1.responseAfterCompaction)(capturedResponseMessage, responseCompaction?.baseResponseStepCount, originalResponse);
+                                            // Preserve the complete persistence response, including same-ID
+                                            // replacements whose old tool parts can contain new results.
+                                            // Convert prefix and suffix separately so each tool output is
+                                            // converted once, while only the suffix enters model context.
+                                            const responsePrefixMessages = responseCompaction
+                                                ? await toModelMessages([
+                                                    stripProviderMetadata({
+                                                        ...capturedResponseMessage,
+                                                        parts: capturedResponseMessage.parts.slice(0, capturedResponseMessage.parts.length -
+                                                            responseForModel.parts.length),
+                                                    }),
+                                                ])
+                                                : [];
                                             const responseModelMessages = await toModelMessages([
-                                                stripProviderMetadata(capturedResponseMessage),
+                                                stripProviderMetadata(responseForModel),
                                             ]);
-                                            if (existingIdx !== -1) {
+                                            if (responseCompaction) {
+                                                // The summary already replaced the original response, including
+                                                // a same-ID approval/handover prefix. Replacing its old model run
+                                                // would miss and fall back to the full, uncompacted UI history.
+                                                accumulatedMessages.push(...responseModelMessages);
+                                                locals_js_1.locals.set(chatHandoverSplicedRunKey, undefined);
+                                            }
+                                            else if (existingIdx !== -1) {
                                                 const spliced = locals_js_1.locals.get(chatHandoverSplicedRunKey);
                                                 const splicedRun = spliced && previousAtIdx && spliced.id === previousAtIdx.id
                                                     ? spliced.run
@@ -6025,7 +6055,7 @@ function chatAgent(options) {
                                             else {
                                                 accumulatedMessages.push(...responseModelMessages);
                                             }
-                                            turnNewModelMessages.push(...responseModelMessages);
+                                            turnNewModelMessages.push(...responsePrefixMessages, ...responseModelMessages);
                                         }
                                         catch {
                                             // Conversion failed — skip accumulation for this turn