replicas-cli 0.2.323 → 0.2.324

package/dist/index.mjs

@@ -9536,7 +9536,7 @@ var HOOK_EXEC_MAX_BUFFER_BYTES = 10 * 1024 * 1024;
 var REPLICAS_CONFIG_FILENAMES = ["replicas.json", "replicas.yaml", "replicas.yml"];
 
 // ../shared/src/cli-version.ts
-var CLI_VERSION = "0.2.323";
+var CLI_VERSION = "0.2.324";
 
 // ../shared/src/engine/environment.ts
 var DESKTOP_NOVNC_PORT = 6080;
@@ -10486,6 +10486,11 @@ function parseClaudeEvents(events, parentToolUseId) {
   const toolCallMap = /* @__PURE__ */ new Map();
   const taskMessageMap = /* @__PURE__ */ new Map();
   const partialIndexes = /* @__PURE__ */ new Map();
+  const completedStreamIds = /* @__PURE__ */ new Set();
+  const supersededStreamIds = /* @__PURE__ */ new Set();
+  let liveStreamId = null;
+  const assistantThinking = /* @__PURE__ */ new Map();
+  const assistantTextCounts = /* @__PURE__ */ new Map();
   const taskAccumulator = new TaskAccumulator();
   const taskSnapshot = () => taskAccumulator.getTasks().map((task) => ({
     text: task.subject,
@@ -10496,6 +10501,10 @@ function parseClaudeEvents(events, parentToolUseId) {
     if (event.type === CLAUDE_PARTIAL_MESSAGE_EVENT_TYPE) {
       const payload = coerceClaudePartialMessagePayload(event.payload);
       if (!payload) return;
+      if (liveStreamId !== null && liveStreamId !== payload.streamId) {
+        supersededStreamIds.add(liveStreamId);
+      }
+      liveStreamId = payload.streamId;
       const existing = partialIndexes.get(payload.streamId) ?? {};
       if (payload.thinking) {
         const reasoningMessage = {
@@ -10558,24 +10567,51 @@ function parseClaudeEvents(events, parentToolUseId) {
     if (event.type === "claude-assistant") {
       const messageId = event.payload.message?.id;
       const contentBlocks = normalizeContentBlocks(event.payload.message?.content);
-      const thinkingText = contentBlocks.filter((block) => block.type === "thinking" && block.thinking).map((block) => block.thinking).join("\n\n");
-      if (thinkingText) {
-        upsertDisplayMessage(messages, {
-          id: `reasoning-${messageId || event.timestamp}-thinking`,
+      const messageKey = messageId || event.timestamp;
+      const streamRefs = messageId ? partialIndexes.get(messageId) : void 0;
+      if (messageId) {
+        completedStreamIds.add(messageId);
+        if (liveStreamId !== null && liveStreamId !== messageId) {
+          supersededStreamIds.add(liveStreamId);
+        }
+        liveStreamId = null;
+      }
+      const thinkingBlocks = contentBlocks.flatMap(
+        (block) => block.type === "thinking" && typeof block.thinking === "string" ? [block.thinking] : []
+      );
+      if (thinkingBlocks.length > 0) {
+        const allThinking = [...assistantThinking.get(messageKey) ?? [], ...thinkingBlocks];
+        assistantThinking.set(messageKey, allThinking);
+        const reasoningMessage = {
+          id: `reasoning-${messageKey}-thinking`,
           type: "reasoning",
-          content: thinkingText,
+          content: allThinking.join("\n\n"),
           status: "completed",
           timestamp: event.timestamp
-        });
+        };
+        if (streamRefs?.reasoning !== void 0) {
+          messages[streamRefs.reasoning] = reasoningMessage;
+          streamRefs.reasoning = void 0;
+        } else {
+          upsertDisplayMessage(messages, reasoningMessage);
+        }
       }
-      contentBlocks.forEach((block, blockIndex) => {
+      contentBlocks.forEach((block) => {
         if (block.type === "text" && block.text) {
-          upsertDisplayMessage(messages, {
-            id: `agent-${messageId || event.timestamp}-${blockIndex}`,
+          const textIndex = assistantTextCounts.get(messageKey) ?? 0;
+          assistantTextCounts.set(messageKey, textIndex + 1);
+          const agentMessage = {
+            id: `agent-${messageKey}-${textIndex}`,
             type: "agent",
             content: block.text,
             timestamp: event.timestamp
-          });
+          };
+          if (streamRefs?.agent !== void 0) {
+            messages[streamRefs.agent] = agentMessage;
+            streamRefs.agent = void 0;
+          } else {
+            upsertDisplayMessage(messages, agentMessage);
+          }
         }
         if (block.type === "tool_use" && block.id) {
           const toolName = block.name || "unknown";
@@ -10847,7 +10883,14 @@ function parseClaudeEvents(events, parentToolUseId) {
       }
     }
   });
-  return messages;
+  const staleIndexes = /* @__PURE__ */ new Set();
+  for (const streamId of [...completedStreamIds, ...supersededStreamIds]) {
+    const refs = partialIndexes.get(streamId);
+    if (refs?.reasoning !== void 0) staleIndexes.add(refs.reasoning);
+    if (refs?.agent !== void 0) staleIndexes.add(refs.agent);
+  }
+  if (staleIndexes.size === 0) return messages;
+  return messages.filter((_, index) => !staleIndexes.has(index));
 }
 
 // ../shared/src/agent-event-utils.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "replicas-cli",
-  "version": "0.2.323",
+  "version": "0.2.324",
   "description": "CLI for managing Replicas workspaces - SSH into cloud dev environments with automatic port forwarding",
   "main": "dist/index.mjs",
   "bin": {