replicas-engine 0.1.303 → 0.1.304

package/dist/src/index.js

@@ -41,6 +41,7 @@ function codexReasoningEffortForThinkingLevel(thinkingLevel) {
 }
 
 // ../shared/src/event.ts
+var CLAUDE_PARTIAL_MESSAGE_EVENT_TYPE = "claude-partial-message";
 var ACCEPTED_USER_MESSAGE_SOURCE = "replicas-chat-turn-accepted";
 var USER_MESSAGE_ID_PAYLOAD_KEY = "replicasMessageId";
 var CODEX_ASP_ITEM_ID_PAYLOAD_KEY = "codexAspItemId";
@@ -301,7 +302,7 @@ var WORKSPACE_SIZES = ["small", "large"];
 var INVALID_WORKSPACE_SIZE_ERROR = `Invalid size: must be one of ${WORKSPACE_SIZES.join(", ")}`;
 
 // ../shared/src/e2b.ts
-var E2B_TEMPLATE_NAME = "replicas-sandbox-2026-06-12-v6";
+var E2B_TEMPLATE_NAME = "replicas-sandbox-2026-06-12-v7";
 
 // ../shared/src/runtime-env.ts
 function parsePosixEnvFile(content) {
@@ -5825,6 +5826,11 @@ var COMMAND_PROTECTION_SAFE_TOOLS = /* @__PURE__ */ new Set([
   "WebFetch",
   "LS"
 ]);
+var CLAUDE_PARTIAL_MESSAGE_FLUSH_MS = 500;
+function supportsClaudeThinkingDisplay(model) {
+  const normalized = model.toLowerCase();
+  return AGENT_MODELS.claude.includes(normalized) || normalized === "opus" || /^claude-(?:opus|sonnet|haiku)-4/.test(normalized);
+}
 var TOOL_INPUT_HANDLERS = {
   ExitPlanMode: {
     getRequest: () => ({
@@ -5919,6 +5925,8 @@ var ClaudeManager = class _ClaudeManager extends CodingAgentManager {
   pendingTurn = null;
   pendingInterrupt = false;
   activeBackgroundTasks = /* @__PURE__ */ new Set();
+  partialMessageStreams = /* @__PURE__ */ new Map();
+  partialMessageFlushTimers = /* @__PURE__ */ new Map();
   systemPromptOverride;
   toolsOverride;
   mcpServersConfig;
@@ -6300,6 +6308,8 @@ var ClaudeManager = class _ClaudeManager extends CodingAgentManager {
         ...this.mcpServersConfig ? { mcpServers: this.mcpServersConfig } : {},
         env: queryEnv,
         model: resolvedModel,
+        includePartialMessages: true,
+        ...supportsClaudeThinkingDisplay(resolvedModel) ? { thinking: { type: "adaptive", display: "summarized" } } : {},
         ...thinkingLevel ? { effort: thinkingLevel } : {},
         canUseTool: this.buildCanUseTool(),
         ...ENGINE_ENV.REPLICAS_DISABLE_GH_PR_MERGE ? {
@@ -6372,6 +6382,11 @@ var ClaudeManager = class _ClaudeManager extends CodingAgentManager {
     this.sessionLinearForwarder = null;
     this.sessionLoop = null;
     this.activeBackgroundTasks.clear();
+    for (const timer of this.partialMessageFlushTimers.values()) {
+      clearTimeout(timer);
+    }
+    this.partialMessageFlushTimers.clear();
+    this.partialMessageStreams.clear();
   }
   resolvePendingTurn() {
     const pending = this.pendingTurn;
@@ -6498,6 +6513,80 @@ var ClaudeManager = class _ClaudeManager extends CodingAgentManager {
       events
     };
   }
+  partialMessageStreamKey(message) {
+    return `${message.session_id}:${message.parent_tool_use_id ?? "root"}`;
+  }
+  clearPartialMessageFlush(key) {
+    const timer = this.partialMessageFlushTimers.get(key);
+    if (!timer) return;
+    clearTimeout(timer);
+    this.partialMessageFlushTimers.delete(key);
+  }
+  flushPartialMessageStream(key, status = "in_progress") {
+    this.clearPartialMessageFlush(key);
+    const stream = this.partialMessageStreams.get(key);
+    if (!stream || !stream.text && !stream.thinking) return;
+    this.onEvent({
+      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+      type: CLAUDE_PARTIAL_MESSAGE_EVENT_TYPE,
+      payload: {
+        streamId: stream.streamId,
+        parent_tool_use_id: stream.parent_tool_use_id,
+        ...stream.text ? { text: stream.text } : {},
+        ...stream.thinking ? { thinking: stream.thinking } : {},
+        status
+      }
+    });
+  }
+  schedulePartialMessageFlush(key) {
+    if (this.partialMessageFlushTimers.has(key)) return;
+    this.partialMessageFlushTimers.set(key, setTimeout(() => {
+      this.partialMessageFlushTimers.delete(key);
+      this.flushPartialMessageStream(key);
+    }, CLAUDE_PARTIAL_MESSAGE_FLUSH_MS));
+  }
+  handlePartialAssistantMessage(message) {
+    const key = this.partialMessageStreamKey(message);
+    const streamEvent = message.event;
+    if (streamEvent.type === "message_start") {
+      this.clearPartialMessageFlush(key);
+      this.partialMessageStreams.set(key, {
+        streamId: streamEvent.message.id || key,
+        parent_tool_use_id: message.parent_tool_use_id,
+        text: "",
+        thinking: ""
+      });
+      return;
+    }
+    const stream = this.partialMessageStreams.get(key);
+    if (!stream) return;
+    if (streamEvent.type === "content_block_start") {
+      const block = streamEvent.content_block;
+      if (block.type === "text" && block.text) {
+        stream.text += block.text;
+        this.schedulePartialMessageFlush(key);
+      } else if (block.type === "thinking" && block.thinking) {
+        stream.thinking += block.thinking;
+        this.schedulePartialMessageFlush(key);
+      }
+      return;
+    }
+    if (streamEvent.type === "content_block_delta") {
+      const delta = streamEvent.delta;
+      if (delta.type === "text_delta") {
+        stream.text += delta.text;
+        this.schedulePartialMessageFlush(key);
+      } else if (delta.type === "thinking_delta") {
+        stream.thinking += delta.thinking;
+        this.schedulePartialMessageFlush(key);
+      }
+      return;
+    }
+    if (streamEvent.type === "message_stop") {
+      this.flushPartialMessageStream(key, "completed");
+      this.partialMessageStreams.delete(key);
+    }
+  }
   async initialize() {
     const historyDir = join13(homedir11(), ".replicas", "claude");
     await mkdir10(historyDir, { recursive: true });
@@ -6512,6 +6601,10 @@ var ClaudeManager = class _ClaudeManager extends CodingAgentManager {
       await this.onSaveSessionId(this.sessionId);
       console.log(`[ClaudeManager] Captured and persisted session ID: ${this.sessionId}`);
     }
+    if (message.type === "stream_event") {
+      this.handlePartialAssistantMessage(message);
+      return;
+    }
     this.trackNativeCompaction(message);
     await this.recordEvent(message);
   }
@@ -6734,7 +6827,7 @@ var AspClient = class {
 // src/managers/codex-asp/app-server-process.ts
 var DEFAULT_CODEX_BINARY = "codex";
 var DEFAULT_CODEX_ARGS = ["app-server", "--listen", "stdio://"];
-var ENGINE_PACKAGE_VERSION = "0.1.303";
+var ENGINE_PACKAGE_VERSION = "0.1.304";
 var INITIALIZE_METHOD = "initialize";
 var INITIALIZED_NOTIFICATION = "initialized";
 var ACCOUNT_LOGIN_START_METHOD = "account/login/start";
@@ -9660,9 +9753,11 @@ var ChatService = class {
         }
       }
     }
-    this.touch(chat);
-    this.observeCurrentBranches(chat).catch(() => {
-    });
+    if (event.type !== CLAUDE_PARTIAL_MESSAGE_EVENT_TYPE) {
+      this.touch(chat);
+      this.observeCurrentBranches(chat).catch(() => {
+      });
+    }
     this.publish({
       type: "chat.turn.delta",
       payload: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "replicas-engine",
-  "version": "0.1.303",
+  "version": "0.1.304",
   "description": "Lightweight API server for Replicas workspaces",
   "type": "module",
   "main": "dist/src/index.js",