@sentry/junior 0.52.0 → 0.53.0

package/dist/app.js

@@ -494,6 +494,7 @@ function coerceThreadConversationState(value) {
   const processing = {
     activeTurnId: toOptionalString(rawProcessing.activeTurnId),
     lastCompletedAtMs: toOptionalNumber(rawProcessing.lastCompletedAtMs),
+    lastSessionId: toOptionalString(rawProcessing.lastSessionId),
     pendingAuth: coercePendingAuthState(rawProcessing.pendingAuth)
   };
   const rawStats = isRecord(rawConversation.stats) ? rawConversation.stats : {};
@@ -2101,17 +2102,24 @@ function startActiveTurn(args) {
   args.conversation.processing.activeTurnId = args.nextTurnId;
   args.updateConversationStats(args.conversation);
 }
-function markTurnCompleted(args) {
-  if (!args.sessionId || args.conversation.processing.activeTurnId === args.sessionId) {
-    args.conversation.processing.activeTurnId = void 0;
+function clearActiveTurn(conversation, sessionId) {
+  if (!sessionId || conversation.processing.activeTurnId === sessionId) {
+    conversation.processing.activeTurnId = void 0;
   }
+}
+function markTurnClosed(args) {
+  clearActiveTurn(args.conversation, args.sessionId);
+  args.conversation.processing.lastCompletedAtMs = args.nowMs;
+  args.updateConversationStats(args.conversation);
+}
+function markTurnCompleted(args) {
+  clearActiveTurn(args.conversation, args.sessionId);
+  args.conversation.processing.lastSessionId = args.sessionId;
   args.conversation.processing.lastCompletedAtMs = args.nowMs;
   args.updateConversationStats(args.conversation);
 }
 function markTurnFailed(args) {
-  if (!args.sessionId || args.conversation.processing.activeTurnId === args.sessionId) {
-    args.conversation.processing.activeTurnId = void 0;
-  }
+  clearActiveTurn(args.conversation, args.sessionId);
   args.conversation.processing.lastCompletedAtMs = args.nowMs;
   args.markConversationMessage(args.conversation, args.userMessageId, {
     replied: false,
@@ -2450,6 +2458,9 @@ import { Agent as Agent2 } from "@mariozechner/pi-agent-core";
 import fs from "fs";
 import path2 from "path";
 
+// src/chat/turn-context-tag.ts
+var TURN_CONTEXT_TAG = "runtime-turn-context";
+
 // src/chat/interruption-marker.ts
 var INTERRUPTED_MARKER = "\n\n[Response interrupted before completion]";
 function getInterruptionMarker() {
@@ -3020,7 +3031,6 @@ function formatSlackCapabilityNames(capabilities) {
 }
 var HEADER = "You are a Slack-based helper assistant. Follow the personality block for voice and tone in every reply. The behavior and output blocks define platform mechanics and override personality only when those mechanics conflict.";
 var TURN_CONTEXT_HEADER = "Per-turn runtime context for this request. Treat these blocks as trusted runtime facts and skill/provider instructions for the current turn; the static system prompt remains authoritative.";
-var TURN_CONTEXT_TAG = "runtime-turn-context";
 var TOOL_POLICY_RULES = [
   "- Tool schemas are the source of truth for parameters; tool names are case-sensitive, so call tools exactly by their exposed names and do not invent arguments.",
   "- Use tools for actionable work and for facts that are mutable, external, repository-backed, provider-backed, or requested as verified/current. Stable general knowledge and already-provided context may be answered directly.",
@@ -7882,6 +7892,7 @@ import { Type as Type21 } from "@sinclair/typebox";
 
 // src/chat/respond-helpers.ts
 var MAX_INLINE_ATTACHMENT_BASE64_CHARS = 12e4;
+var RUNTIME_TURN_CONTEXT_START = `<${TURN_CONTEXT_TAG}>`;
 function getSessionIdentifiers(context) {
   return {
     conversationId: context.correlation?.conversationId ?? context.correlation?.threadId ?? context.correlation?.runId,
@@ -8044,6 +8055,75 @@ function getPiMessageRole(value) {
   const role = value.role;
   return typeof role === "string" ? role : void 0;
 }
+function getUserMessageContent(message) {
+  const record = message;
+  return record.role === "user" && Array.isArray(record.content) ? record.content : void 0;
+}
+function isRuntimeTurnContextPart(part, marker) {
+  return part !== null && typeof part === "object" && part.type === "text" && typeof part.text === "string" && part.text.startsWith(marker);
+}
+function replaceRuntimeTurnContext(message, turnContextPrompt) {
+  const content = getUserMessageContent(message);
+  if (!content) {
+    return void 0;
+  }
+  const marker = turnContextPrompt.split("\n", 1)[0];
+  const contextIndex = content.findIndex(
+    (part) => isRuntimeTurnContextPart(part, marker)
+  );
+  if (contextIndex < 0) {
+    return void 0;
+  }
+  const nextContent = [...content];
+  nextContent[contextIndex] = {
+    ...nextContent[contextIndex],
+    text: turnContextPrompt
+  };
+  return {
+    ...message,
+    content: nextContent
+  };
+}
+function refreshRuntimeTurnContext(messages, turnContextPrompt) {
+  for (let index = 0; index < messages.length; index += 1) {
+    const updated = replaceRuntimeTurnContext(
+      messages[index],
+      turnContextPrompt
+    );
+    if (!updated) {
+      continue;
+    }
+    const nextMessages = [...messages];
+    nextMessages[index] = updated;
+    return nextMessages;
+  }
+  return [
+    ...messages,
+    {
+      role: "user",
+      content: [{ type: "text", text: turnContextPrompt }],
+      timestamp: Date.now()
+    }
+  ];
+}
+function stripRuntimeTurnContext(messages) {
+  return messages.flatMap((message) => {
+    const content = getUserMessageContent(message);
+    if (!content) {
+      return [message];
+    }
+    const nextContent = content.filter(
+      (part) => !isRuntimeTurnContextPart(part, RUNTIME_TURN_CONTEXT_START)
+    );
+    if (nextContent.length === content.length) {
+      return [message];
+    }
+    if (nextContent.length === 0) {
+      return [];
+    }
+    return [{ ...message, content: nextContent }];
+  });
+}
 function extractAssistantText(message) {
   const content = message.content ?? [];
   return content.filter(
@@ -11326,9 +11406,98 @@ async function unlinkProvider(userId, provider, userTokenStore) {
   ]);
 }
 
+// src/chat/state/turn-session-store.ts
+import { THREAD_STATE_TTL_MS as THREAD_STATE_TTL_MS3 } from "chat";
+
+// src/chat/state/pi-session-message-store.ts
+import { isDeepStrictEqual } from "util";
+var PI_SESSION_MESSAGE_PREFIX = "junior:pi_session_message";
+function piSessionMessageKey(scope, index) {
+  return `${PI_SESSION_MESSAGE_PREFIX}:${scope.conversationId}:${scope.sessionId}:${index}`;
+}
+function parsePiMessage(value) {
+  return isRecord(value) ? value : void 0;
+}
+function normalizeMessageCount(value) {
+  return Number.isFinite(value) ? Math.max(0, Math.floor(value)) : 0;
+}
+function countMatchingPrefix(left, right) {
+  const limit = Math.min(left.length, right.length);
+  for (let index = 0; index < limit; index += 1) {
+    if (!isDeepStrictEqual(left[index], right[index])) {
+      return index;
+    }
+  }
+  return limit;
+}
+async function loadPiSessionMessages(args) {
+  const stateAdapter = getStateAdapter();
+  await stateAdapter.connect();
+  const messageCount = normalizeMessageCount(args.messageCount);
+  if (messageCount === 0) {
+    return [];
+  }
+  const values = await Promise.all(
+    Array.from(
+      { length: messageCount },
+      (_, index) => stateAdapter.get(piSessionMessageKey(args, index))
+    )
+  );
+  const messages = [];
+  for (const value of values) {
+    const message = parsePiMessage(value);
+    if (!message) {
+      break;
+    }
+    messages.push(message);
+  }
+  return messages.length === messageCount ? messages : void 0;
+}
+async function loadExistingPiSessionMessages(scope, maxCount) {
+  const count = normalizeMessageCount(maxCount);
+  if (count === 0) {
+    return [];
+  }
+  const stateAdapter = getStateAdapter();
+  await stateAdapter.connect();
+  const values = await Promise.all(
+    Array.from(
+      { length: count },
+      (_, index) => stateAdapter.get(piSessionMessageKey(scope, index))
+    )
+  );
+  const messages = [];
+  for (const value of values) {
+    const message = parsePiMessage(value);
+    if (!message) {
+      break;
+    }
+    messages.push(message);
+  }
+  return messages;
+}
+async function commitPiSessionMessages(args) {
+  const stateAdapter = getStateAdapter();
+  await stateAdapter.connect();
+  const existingMessages = await loadExistingPiSessionMessages(
+    { conversationId: args.conversationId, sessionId: args.sessionId },
+    args.messages.length
+  );
+  const writeFromIndex = countMatchingPrefix(existingMessages, args.messages);
+  await Promise.all(
+    args.messages.slice(writeFromIndex).map(
+      (message, offset) => stateAdapter.set(
+        piSessionMessageKey(args, writeFromIndex + offset),
+        message,
+        args.ttlMs
+      )
+    )
+  );
+}
+
 // src/chat/state/turn-session-store.ts
 var AGENT_TURN_SESSION_PREFIX = "junior:agent_turn_session";
-var AGENT_TURN_SESSION_TTL_MS = 24 * 60 * 60 * 1e3;
+var AGENT_TURN_SESSION_TTL_MS = THREAD_STATE_TTL_MS3;
 function agentTurnSessionKey(conversationId, sessionId) {
   return `${AGENT_TURN_SESSION_PREFIX}:${conversationId}:${sessionId}`;
 }
@@ -11354,41 +11523,55 @@ function parseAgentTurnUsage(value) {
   }
   return Object.keys(usage).length > 0 ? usage : void 0;
 }
-function parseAgentTurnSessionCheckpoint(value) {
+function parseStoredRecord(value) {
+  if (isRecord(value)) {
+    return value;
+  }
   if (typeof value !== "string") {
     return void 0;
   }
   try {
     const parsed = JSON.parse(value);
-    if (!isRecord(parsed)) {
-      return void 0;
-    }
-    const status = parsed.state;
-    if (status !== "running" && status !== "awaiting_resume" && status !== "completed" && status !== "failed" && status !== "superseded") {
-      return void 0;
-    }
-    const conversationId = parsed.conversationId;
-    const sessionId = parsed.sessionId;
-    const sliceId = parsed.sliceId;
-    const checkpointVersion = parsed.checkpointVersion;
-    const updatedAtMs = parsed.updatedAtMs;
-    const cumulativeDurationMs = toFiniteNonNegativeNumber(
-      parsed.cumulativeDurationMs
-    );
-    const cumulativeUsage = parseAgentTurnUsage(parsed.cumulativeUsage);
-    if (typeof conversationId !== "string" || typeof sessionId !== "string" || typeof sliceId !== "number" || typeof checkpointVersion !== "number" || typeof updatedAtMs !== "number") {
-      return void 0;
-    }
-    return {
+    return isRecord(parsed) ? parsed : void 0;
+  } catch {
+    return void 0;
+  }
+}
+function parseAgentTurnSessionRecord(value) {
+  const parsed = parseStoredRecord(value);
+  if (!parsed) {
+    return void 0;
+  }
+  const status = parsed.state;
+  if (status !== "running" && status !== "awaiting_resume" && status !== "completed" && status !== "failed" && status !== "superseded") {
+    return void 0;
+  }
+  const conversationId = parsed.conversationId;
+  const sessionId = parsed.sessionId;
+  const sliceId = parsed.sliceId;
+  const checkpointVersion = parsed.checkpointVersion;
+  const updatedAtMs = parsed.updatedAtMs;
+  const cumulativeDurationMs = toFiniteNonNegativeNumber(
+    parsed.cumulativeDurationMs
+  );
+  const cumulativeUsage = parseAgentTurnUsage(parsed.cumulativeUsage);
+  if (typeof conversationId !== "string" || typeof sessionId !== "string" || typeof sliceId !== "number" || typeof checkpointVersion !== "number" || typeof updatedAtMs !== "number") {
+    return void 0;
+  }
+  const legacyPiMessages = Array.isArray(parsed.piMessages) ? parsed.piMessages : [];
+  const messageCount = toFiniteNonNegativeNumber(parsed.messageCount) ?? legacyPiMessages.length;
+  return {
+    legacyPiMessages,
+    record: {
       checkpointVersion,
       conversationId,
       sessionId,
       sliceId,
       state: status,
       updatedAtMs,
+      messageCount,
       ...cumulativeDurationMs !== void 0 ? { cumulativeDurationMs } : {},
       ...cumulativeUsage ? { cumulativeUsage } : {},
-      piMessages: Array.isArray(parsed.piMessages) ? parsed.piMessages : [],
       ...Array.isArray(parsed.loadedSkillNames) ? {
         loadedSkillNames: parsed.loadedSkillNames.filter(
           (value2) => typeof value2 === "string"
@@ -11397,10 +11580,20 @@ function parseAgentTurnSessionCheckpoint(value) {
       ...parsed.resumeReason === "timeout" || parsed.resumeReason === "auth" ? { resumeReason: parsed.resumeReason } : {},
       ...typeof parsed.errorMessage === "string" ? { errorMessage: parsed.errorMessage } : {},
       ...typeof parsed.resumedFromSliceId === "number" ? { resumedFromSliceId: parsed.resumedFromSliceId } : {}
-    };
-  } catch {
-    return void 0;
+    }
+  };
+}
+function materializePiMessages(legacyPiMessages, messageCount, sessionMessages) {
+  if (messageCount === 0) {
+    return [];
+  }
+  if (sessionMessages) {
+    return sessionMessages;
+  }
+  if (legacyPiMessages.length >= messageCount) {
+    return legacyPiMessages.slice(0, messageCount);
   }
+  return void 0;
 }
 async function getAgentTurnSessionCheckpoint(conversationId, sessionId) {
   const stateAdapter = getStateAdapter();
@@ -11408,23 +11601,51 @@ async function getAgentTurnSessionCheckpoint(conversationId, sessionId) {
   const value = await stateAdapter.get(
     agentTurnSessionKey(conversationId, sessionId)
   );
-  return parseAgentTurnSessionCheckpoint(value);
+  const parsed = parseAgentTurnSessionRecord(value);
+  if (!parsed) {
+    return void 0;
+  }
+  const sessionMessages = await loadPiSessionMessages({
+    conversationId,
+    sessionId,
+    messageCount: parsed.record.messageCount
+  });
+  const piMessages = materializePiMessages(
+    parsed.legacyPiMessages,
+    parsed.record.messageCount,
+    sessionMessages
+  );
+  if (!piMessages) {
+    return void 0;
+  }
+  return {
+    ...parsed.record,
+    piMessages
+  };
 }
 async function upsertAgentTurnSessionCheckpoint(args) {
   const stateAdapter = getStateAdapter();
   await stateAdapter.connect();
-  const existing = await getAgentTurnSessionCheckpoint(
-    args.conversationId,
-    args.sessionId
+  const existingValue = await stateAdapter.get(
+    agentTurnSessionKey(args.conversationId, args.sessionId)
   );
+  const existingRecord = parseAgentTurnSessionRecord(existingValue);
+  const ttlMs = Math.max(1, args.ttlMs ?? AGENT_TURN_SESSION_TTL_MS);
+  await commitPiSessionMessages({
+    conversationId: args.conversationId,
+    sessionId: args.sessionId,
+    messages: args.piMessages,
+    ttlMs
+  });
+  const storedMessageCount = args.piMessages.length;
   const checkpoint = {
-    checkpointVersion: (existing?.checkpointVersion ?? 0) + 1,
+    checkpointVersion: (existingRecord?.record.checkpointVersion ?? 0) + 1,
     conversationId: args.conversationId,
     sessionId: args.sessionId,
     sliceId: args.sliceId,
     state: args.state,
     updatedAtMs: Date.now(),
-    piMessages: Array.isArray(args.piMessages) ? args.piMessages : [],
+    messageCount: storedMessageCount,
     ...typeof args.cumulativeDurationMs === "number" && Number.isFinite(args.cumulativeDurationMs) ? {
       cumulativeDurationMs: Math.max(
         0,
@@ -11441,13 +11662,15 @@ async function upsertAgentTurnSessionCheckpoint(args) {
     ...args.errorMessage ? { errorMessage: args.errorMessage } : {},
     ...typeof args.resumedFromSliceId === "number" ? { resumedFromSliceId: args.resumedFromSliceId } : {}
   };
-  const ttlMs = Math.max(1, args.ttlMs ?? AGENT_TURN_SESSION_TTL_MS);
   await stateAdapter.set(
     agentTurnSessionKey(args.conversationId, args.sessionId),
-    JSON.stringify(checkpoint),
+    checkpoint,
     ttlMs
   );
-  return checkpoint;
+  return {
+    ...checkpoint,
+    piMessages: [...args.piMessages]
+  };
 }
 async function supersedeAgentTurnSessionCheckpoint(args) {
   const existing = await getAgentTurnSessionCheckpoint(
@@ -12000,7 +12223,6 @@ function buildBriefPostCanvasReply(artifactStatePatch) {
 function buildTurnResult(input) {
   const {
     newMessages,
-    piMessages,
     userInput,
     replyFiles,
     artifactStatePatch,
@@ -12104,7 +12326,6 @@ function buildTurnResult(input) {
     text: resolvedText,
     files: replyFiles.length > 0 ? replyFiles : void 0,
     artifactStatePatch: Object.keys(artifactStatePatch).length > 0 ? artifactStatePatch : void 0,
-    piMessages,
     deliveryPlan,
     deliveryMode,
     sandboxId,
@@ -12787,69 +13008,6 @@ function buildUserTurnInput(args) {
   }
   return { routerBlocks, userContentParts };
 }
-function refreshCheckpointTurnContext(messages, turnContextPrompt) {
-  const marker = getTurnContextMarker(turnContextPrompt);
-  for (let index = 0; index < messages.length; index += 1) {
-    const content = getUserMessageContent(messages[index]);
-    if (!content) {
-      continue;
-    }
-    const contextIndex = content.findIndex(
-      (part) => isTurnContextPart(part, marker)
-    );
-    if (contextIndex < 0) {
-      continue;
-    }
-    const updatedMessages = [...messages];
-    const updatedContent = [...content];
-    updatedContent[contextIndex] = {
-      ...updatedContent[contextIndex],
-      text: turnContextPrompt
-    };
-    updatedMessages[index] = {
-      ...messages[index],
-      content: updatedContent
-    };
-    return updatedMessages;
-  }
-  return [
-    ...messages,
-    {
-      role: "user",
-      content: [{ type: "text", text: turnContextPrompt }],
-      timestamp: Date.now()
-    }
-  ];
-}
-function stripTurnContextFromMessages(messages, turnContextPrompt) {
-  const marker = getTurnContextMarker(turnContextPrompt);
-  return messages.flatMap((message) => {
-    const content = getUserMessageContent(message);
-    if (!content) {
-      return [message];
-    }
-    const strippedContent = content.filter(
-      (part) => !isTurnContextPart(part, marker)
-    );
-    if (strippedContent.length === content.length) {
-      return [message];
-    }
-    if (strippedContent.length === 0) {
-      return [];
-    }
-    return [{ ...message, content: strippedContent }];
-  });
-}
-function getTurnContextMarker(turnContextPrompt) {
-  return turnContextPrompt.split("\n", 1)[0];
-}
-function getUserMessageContent(message) {
-  const record = message;
-  return record.role === "user" && Array.isArray(record.content) ? record.content : void 0;
-}
-function isTurnContextPart(part, marker) {
-  return part !== null && typeof part === "object" && part.type === "text" && typeof part.text === "string" && part.text.startsWith(marker);
-}
 async function generateAssistantReply(messageText, context = {}) {
   const replyStartedAtMs = Date.now();
   let timeoutResumeConversationId;
@@ -13362,7 +13520,7 @@ async function generateAssistantReply(messageText, context = {}) {
     beforeMessageCount = agent.state.messages.length;
     try {
       if (resumedFromCheckpoint) {
-        agent.state.messages = refreshCheckpointTurnContext(
+        agent.state.messages = refreshRuntimeTurnContext(
           existingCheckpoint.piMessages,
           turnContextPrompt
         );
@@ -13475,10 +13633,6 @@ async function generateAssistantReply(messageText, context = {}) {
     }
     return buildTurnResult({
       newMessages,
-      piMessages: stripTurnContextFromMessages(
-        agent.state.messages,
-        turnContextPrompt
-      ),
       userInput,
       replyFiles,
       artifactStatePatch,
@@ -14940,7 +15094,7 @@ function completeAuthPauseTurn(args) {
       skippedReason: void 0
     }
   );
-  markTurnCompleted({
+  markTurnClosed({
     conversation: args.conversation,
     nowMs: Date.now(),
     sessionId: args.sessionId,
@@ -15142,9 +15296,9 @@ async function persistCompletedReplyState(channelId, threadTs, sessionId, reply)
   const conversation = coerceThreadConversationState(currentState);
   const artifacts = coerceThreadArtifactsState(currentState);
   const nextArtifacts = reply.artifactStatePatch ? mergeArtifactsState(artifacts, reply.artifactStatePatch) : void 0;
-  const userMessageId = getTurnUserMessageId(conversation, sessionId);
+  const userMessage = getTurnUserMessage(conversation, sessionId);
   clearPendingAuth(conversation, sessionId);
-  markConversationMessage(conversation, userMessageId, {
+  markConversationMessage(conversation, userMessage?.id, {
     replied: true,
     skippedReason: void 0
   });
@@ -15161,9 +15315,6 @@ async function persistCompletedReplyState(channelId, threadTs, sessionId, reply)
       replied: true
     }
   });
-  if (reply.piMessages) {
-    conversation.piMessages = reply.piMessages;
-  }
   markTurnCompleted({
     conversation,
     nowMs: Date.now(),
@@ -15585,9 +15736,6 @@ async function persistCompletedOAuthReplyState(args) {
       replied: true
     }
   });
-  if (args.reply.piMessages) {
-    conversation.piMessages = args.reply.piMessages;
-  }
   markTurnCompleted({
     conversation,
     nowMs: Date.now(),
@@ -16554,9 +16702,6 @@ async function persistCompletedReplyState2(args) {
       replied: true
     }
   });
-  if (args.reply.piMessages) {
-    conversation.piMessages = args.reply.piMessages;
-  }
   markTurnCompleted({
     conversation,
     nowMs: Date.now(),
@@ -18260,6 +18405,31 @@ function getCurrentTurnCanvasUrl(args) {
 function buildCanvasRecoveryReply(canvasUrl) {
   return `I created the canvas, but the turn was interrupted before I could finish the thread reply: ${canvasUrl}`;
 }
+async function loadPiMessagesForTurn(args) {
+  const fallback = args.fallback.length > 0 ? [...args.fallback] : void 0;
+  if (!args.conversationId) {
+    return fallback;
+  }
+  if (args.activeTurnId) {
+    const checkpoint2 = await getAgentTurnSessionCheckpoint(
+      args.conversationId,
+      args.activeTurnId
+    );
+    if (checkpoint2?.piMessages.length) {
+      return stripRuntimeTurnContext(
+        trimTrailingAssistantMessages(checkpoint2.piMessages)
+      );
+    }
+  }
+  if (!args.lastSessionId) {
+    return fallback;
+  }
+  const checkpoint = await getAgentTurnSessionCheckpoint(
+    args.conversationId,
+    args.lastSessionId
+  );
+  return checkpoint?.state === "completed" && checkpoint.piMessages.length > 0 ? stripRuntimeTurnContext(checkpoint.piMessages) : fallback;
+}
 function createReplyToThread(deps) {
   return async function replyToThread(thread, message, options = {}) {
     if (message.author.isMe) {
@@ -18412,6 +18582,7 @@ function createReplyToThread(deps) {
             return;
           }
         }
+        const lastSessionIdForHistory = preparedState.conversation.processing.lastSessionId;
         const configReply = await maybeApplyProviderDefaultConfigRequest({
           channelConfiguration: preparedState.channelConfiguration,
           requesterId: message.author.userId,
@@ -18487,6 +18658,12 @@ function createReplyToThread(deps) {
           }
         );
         const omittedImageAttachmentCount = !isVisionEnabled() && hasPotentialImageAttachment(message.attachments) ? countPotentialImageAttachments(message.attachments) : 0;
+        const piMessages = await loadPiMessagesForTurn({
+          conversationId,
+          activeTurnId,
+          lastSessionId: lastSessionIdForHistory,
+          fallback: preparedState.conversation.piMessages
+        });
         const status = createSlackAdapterAssistantStatusSession({
           channelId: assistantThreadContext?.channelId,
           threadTs: assistantThreadContext?.threadTs,
@@ -18538,7 +18715,7 @@ function createReplyToThread(deps) {
             },
             conversationContext: preparedState.routingContext ?? preparedState.conversationContext,
             artifactState: preparedState.artifacts,
-            piMessages: preparedState.conversation.piMessages,
+            piMessages,
             pendingAuth: preparedState.conversation.processing.pendingAuth,
             configuration: preparedState.configuration,
             channelConfiguration: preparedState.channelConfiguration,
@@ -18621,9 +18798,6 @@ function createReplyToThread(deps) {
               replied: true
             }
           });
-          if (reply.piMessages) {
-            preparedState.conversation.piMessages = reply.piMessages;
-          }
           const artifactStatePatch = reply.artifactStatePatch ? { ...reply.artifactStatePatch } : {};
           const reactionPerformed = reply.diagnostics.toolCalls.includes(
             "slackMessageAddReaction"
@@ -18701,6 +18875,7 @@ function createReplyToThread(deps) {
           markTurnCompleted({
             conversation: preparedState.conversation,
             nowMs: Date.now(),
+            sessionId: turnId,
             updateConversationStats
           });
           await persistThreadState(thread, {
@@ -18826,9 +19001,10 @@ function createReplyToThread(deps) {
                 replied: true
               }
             });
-            markTurnCompleted({
+            markTurnClosed({
               conversation: preparedState.conversation,
               nowMs: Date.now(),
+              sessionId: turnId,
               updateConversationStats
             });
             await persistThreadState(thread, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sentry/junior",
-  "version": "0.52.0",
+  "version": "0.53.0",
   "private": false,
   "publishConfig": {
     "access": "public"