@autohq/cli 0.1.326 → 0.1.327

package/dist/agent-bridge.js

@@ -23434,7 +23434,7 @@ Object.assign(lookup, {
 // package.json
 var package_default = {
   name: "@autohq/cli",
-  version: "0.1.326",
+  version: "0.1.327",
   license: "SEE LICENSE IN README.md",
   publishConfig: {
     access: "public"
@@ -42432,6 +42432,7 @@ function providerMetadataField(providerMetadata) {
 
 // src/commands/agent-bridge/harness/output-buffer.ts
 var AGENT_BRIDGE_OUTPUT_DELTA_FLUSH_MS = 50;
+var AGENT_BRIDGE_OUTPUT_UI_DELTA_MAX_CHARS = 32768;
 var AGENT_BRIDGE_OUTPUT_ACK_RETRY_BACKOFF_MS = [250, 1e3];
 var AgentBridgeOutputBuffer = class {
   constructor(input) {
@@ -42443,6 +42444,8 @@ var AgentBridgeOutputBuffer = class {
   pendingOutputs = /* @__PURE__ */ new Map();
   pendingDelta = null;
   deltaFlushTimer = null;
+  pendingUiDelta = null;
+  uiDeltaFlushTimer = null;
   activeUiMessageAssembler = null;
   uiMessagePartTracker = new UiMessagePartTracker();
   drainBlocked = false;
@@ -42452,6 +42455,12 @@ var AgentBridgeOutputBuffer = class {
   // ---------------------------------------------------------------------------
   async emitProjection(context, projection) {
     if (projection.type === "ui_message_chunk") {
+      const coalescible = coalescibleUiDeltaChunk(projection);
+      if (coalescible) {
+        await this.bufferUiDeltaChunk(context, coalescible);
+        return;
+      }
+      await this.flushPendingUiDelta();
       await this.emitUiMessageChunk(context, projection);
       return;
     }
@@ -42459,10 +42468,12 @@ var AgentBridgeOutputBuffer = class {
       await this.bufferDelta(context, projection.delta);
       return;
     }
+    await this.flushPendingUiDelta();
     await this.flushPendingDelta();
     await this.enqueueProjectionAndDrain(context, projection);
   }
   async replayPendingOutputs() {
+    await this.materializePendingUiDelta();
     await this.flushPendingDelta({ force: true });
     await this.drainPendingOutputs({ force: true });
   }
@@ -42586,10 +42597,15 @@ var AgentBridgeOutputBuffer = class {
     this.drainPromise ??= (async () => {
       while (true) {
         const output = this.nextPendingOutput();
-        if (!output) {
-          return;
+        if (output) {
+          await this.emitUntilAcked(output);
+          continue;
         }
-        await this.emitUntilAcked(output);
+        if (this.pendingUiDelta) {
+          await this.materializePendingUiDelta();
+          continue;
+        }
+        return;
       }
     })().catch((error51) => {
       this.drainBlocked = true;
@@ -42600,6 +42616,85 @@ var AgentBridgeOutputBuffer = class {
     await this.drainPromise;
   }
   // ---------------------------------------------------------------------------
+  // UI chunk delta coalescing
+  // ---------------------------------------------------------------------------
+  async bufferUiDeltaChunk(context, chunk) {
+    const pending = this.pendingUiDelta;
+    if (pending && canCoalesceUiDelta(pending, context, chunk)) {
+      pending.chunk = mergeUiDeltaChunks(pending.chunk, chunk);
+      if (uiDeltaTextLength(pending.chunk) >= AGENT_BRIDGE_OUTPUT_UI_DELTA_MAX_CHARS) {
+        await this.flushPendingUiDelta();
+      }
+      return;
+    }
+    await this.flushPendingUiDelta();
+    this.pendingUiDelta = { context, chunk: { ...chunk }, createdAt: now2() };
+    this.scheduleUiDeltaFlush();
+  }
+  scheduleUiDeltaFlush() {
+    if (this.uiDeltaFlushTimer !== null) {
+      return;
+    }
+    this.uiDeltaFlushTimer = setTimeout(() => {
+      this.uiDeltaFlushTimer = null;
+      if (!this.pendingUiDelta) {
+        return;
+      }
+      if (this.drainPromise) {
+        this.scheduleUiDeltaFlush();
+        return;
+      }
+      void this.flushPendingUiDelta().catch((error51) => {
+        this.drainBlocked = true;
+        this.input.runtimeLogger?.warn(
+          "agent_bridge_output_buffer_flush_failed",
+          {
+            error: error51 instanceof Error ? error51.message : String(error51),
+            pending_count: this.pendingOutputs.size
+          }
+        );
+      });
+    }, AGENT_BRIDGE_OUTPUT_DELTA_FLUSH_MS);
+  }
+  clearUiDeltaFlushTimer() {
+    if (this.uiDeltaFlushTimer === null) {
+      return;
+    }
+    clearTimeout(this.uiDeltaFlushTimer);
+    this.uiDeltaFlushTimer = null;
+  }
+  async flushPendingUiDelta() {
+    if (!this.pendingUiDelta) {
+      return;
+    }
+    await this.materializePendingUiDelta();
+    await this.drainPendingOutputs();
+  }
+  // Turns the parked coalesced chunk into a queued envelope and applies it to
+  // the part tracker and message assembler, exactly as its source chunks would
+  // have applied one by one (delta chunks never settle a part or complete a
+  // message, so tracker/assembler side effects reduce to text accumulation).
+  // Does not drain: the drain loop calls this mid-loop and keeps pulling.
+  async materializePendingUiDelta() {
+    const pending = this.pendingUiDelta;
+    if (!pending) {
+      return;
+    }
+    this.pendingUiDelta = null;
+    this.clearUiDeltaFlushTimer();
+    this.outputSeq += 1;
+    this.enqueueOutput(
+      buildUiMessageChunkOutputEnvelope({
+        context: pending.context,
+        outputSeq: this.outputSeq,
+        chunk: pending.chunk,
+        createdAt: pending.createdAt
+      })
+    );
+    this.uiMessagePartTracker.append(pending.chunk);
+    await this.appendUiChunkToAssembler(pending.chunk);
+  }
+  // ---------------------------------------------------------------------------
   // Delta coalescing
   // ---------------------------------------------------------------------------
   async bufferDelta(context, delta) {
@@ -42905,6 +43000,50 @@ function terminalEntryStatus(projection) {
   }
   return "completed";
 }
+function coalescibleUiDeltaChunk(projection) {
+  if (projection.statusText !== void 0 || projection.turnStatus !== void 0 || projection.usage !== void 0 || projection.consumedCommandIds !== void 0) {
+    return null;
+  }
+  const chunk = projection.chunk;
+  switch (chunk.type) {
+    case "text-delta":
+    case "reasoning-delta":
+      return chunk.providerMetadata === void 0 ? chunk : null;
+    case "tool-input-delta":
+      return chunk;
+    default:
+      return null;
+  }
+}
+function canCoalesceUiDelta(pending, context, chunk) {
+  if (pending.context.sessionId !== context.sessionId || pending.context.runtimeId !== context.runtimeId || pending.context.bridgeLeaseId !== context.bridgeLeaseId || pending.chunk.type !== chunk.type) {
+    return false;
+  }
+  if (pending.chunk.type === "tool-input-delta" && chunk.type === "tool-input-delta") {
+    return pending.chunk.toolCallId === chunk.toolCallId;
+  }
+  if (pending.chunk.type !== "tool-input-delta" && chunk.type !== "tool-input-delta") {
+    return pending.chunk.id === chunk.id;
+  }
+  return false;
+}
+function mergeUiDeltaChunks(pending, chunk) {
+  if (pending.type === "tool-input-delta" && chunk.type === "tool-input-delta") {
+    return {
+      ...pending,
+      inputTextDelta: pending.inputTextDelta + chunk.inputTextDelta
+    };
+  }
+  if (pending.type !== "tool-input-delta" && chunk.type !== "tool-input-delta") {
+    return { ...pending, delta: pending.delta + chunk.delta };
+  }
+  throw new Error(
+    `Cannot merge ui delta chunks of kinds ${pending.type} and ${chunk.type}`
+  );
+}
+function uiDeltaTextLength(chunk) {
+  return chunk.type === "tool-input-delta" ? chunk.inputTextDelta.length : chunk.delta.length;
+}
 function canCoalesceDelta(pending, context, delta) {
   return pending.context.sessionId === context.sessionId && pending.context.runtimeId === context.runtimeId && pending.context.bridgeLeaseId === context.bridgeLeaseId && pending.delta.messageId === delta.messageId && pending.delta.partId === delta.partId && pending.delta.role === delta.role && pending.delta.kind === delta.kind && pending.delta.delta.type === delta.delta.type;
 }

package/dist/index.js

@@ -28558,7 +28558,7 @@ var init_package = __esm({
   "package.json"() {
     package_default = {
       name: "@autohq/cli",
-      version: "0.1.326",
+      version: "0.1.327",
       license: "SEE LICENSE IN README.md",
       publishConfig: {
         access: "public"
@@ -39562,6 +39562,7 @@ function providerMetadataField(providerMetadata) {
 
 // src/commands/agent-bridge/harness/output-buffer.ts
 var AGENT_BRIDGE_OUTPUT_DELTA_FLUSH_MS = 50;
+var AGENT_BRIDGE_OUTPUT_UI_DELTA_MAX_CHARS = 32768;
 var AGENT_BRIDGE_OUTPUT_ACK_RETRY_BACKOFF_MS = [250, 1e3];
 var AgentBridgeOutputBuffer = class {
   constructor(input) {
@@ -39573,6 +39574,8 @@ var AgentBridgeOutputBuffer = class {
   pendingOutputs = /* @__PURE__ */ new Map();
   pendingDelta = null;
   deltaFlushTimer = null;
+  pendingUiDelta = null;
+  uiDeltaFlushTimer = null;
   activeUiMessageAssembler = null;
   uiMessagePartTracker = new UiMessagePartTracker();
   drainBlocked = false;
@@ -39582,6 +39585,12 @@ var AgentBridgeOutputBuffer = class {
   // ---------------------------------------------------------------------------
   async emitProjection(context, projection) {
     if (projection.type === "ui_message_chunk") {
+      const coalescible = coalescibleUiDeltaChunk(projection);
+      if (coalescible) {
+        await this.bufferUiDeltaChunk(context, coalescible);
+        return;
+      }
+      await this.flushPendingUiDelta();
       await this.emitUiMessageChunk(context, projection);
       return;
     }
@@ -39589,10 +39598,12 @@ var AgentBridgeOutputBuffer = class {
       await this.bufferDelta(context, projection.delta);
       return;
     }
+    await this.flushPendingUiDelta();
     await this.flushPendingDelta();
     await this.enqueueProjectionAndDrain(context, projection);
   }
   async replayPendingOutputs() {
+    await this.materializePendingUiDelta();
     await this.flushPendingDelta({ force: true });
     await this.drainPendingOutputs({ force: true });
   }
@@ -39716,10 +39727,15 @@ var AgentBridgeOutputBuffer = class {
     this.drainPromise ??= (async () => {
       while (true) {
         const output = this.nextPendingOutput();
-        if (!output) {
-          return;
+        if (output) {
+          await this.emitUntilAcked(output);
+          continue;
         }
-        await this.emitUntilAcked(output);
+        if (this.pendingUiDelta) {
+          await this.materializePendingUiDelta();
+          continue;
+        }
+        return;
       }
     })().catch((error51) => {
       this.drainBlocked = true;
@@ -39730,6 +39746,85 @@ var AgentBridgeOutputBuffer = class {
     await this.drainPromise;
   }
   // ---------------------------------------------------------------------------
+  // UI chunk delta coalescing
+  // ---------------------------------------------------------------------------
+  async bufferUiDeltaChunk(context, chunk) {
+    const pending = this.pendingUiDelta;
+    if (pending && canCoalesceUiDelta(pending, context, chunk)) {
+      pending.chunk = mergeUiDeltaChunks(pending.chunk, chunk);
+      if (uiDeltaTextLength(pending.chunk) >= AGENT_BRIDGE_OUTPUT_UI_DELTA_MAX_CHARS) {
+        await this.flushPendingUiDelta();
+      }
+      return;
+    }
+    await this.flushPendingUiDelta();
+    this.pendingUiDelta = { context, chunk: { ...chunk }, createdAt: now2() };
+    this.scheduleUiDeltaFlush();
+  }
+  scheduleUiDeltaFlush() {
+    if (this.uiDeltaFlushTimer !== null) {
+      return;
+    }
+    this.uiDeltaFlushTimer = setTimeout(() => {
+      this.uiDeltaFlushTimer = null;
+      if (!this.pendingUiDelta) {
+        return;
+      }
+      if (this.drainPromise) {
+        this.scheduleUiDeltaFlush();
+        return;
+      }
+      void this.flushPendingUiDelta().catch((error51) => {
+        this.drainBlocked = true;
+        this.input.runtimeLogger?.warn(
+          "agent_bridge_output_buffer_flush_failed",
+          {
+            error: error51 instanceof Error ? error51.message : String(error51),
+            pending_count: this.pendingOutputs.size
+          }
+        );
+      });
+    }, AGENT_BRIDGE_OUTPUT_DELTA_FLUSH_MS);
+  }
+  clearUiDeltaFlushTimer() {
+    if (this.uiDeltaFlushTimer === null) {
+      return;
+    }
+    clearTimeout(this.uiDeltaFlushTimer);
+    this.uiDeltaFlushTimer = null;
+  }
+  async flushPendingUiDelta() {
+    if (!this.pendingUiDelta) {
+      return;
+    }
+    await this.materializePendingUiDelta();
+    await this.drainPendingOutputs();
+  }
+  // Turns the parked coalesced chunk into a queued envelope and applies it to
+  // the part tracker and message assembler, exactly as its source chunks would
+  // have applied one by one (delta chunks never settle a part or complete a
+  // message, so tracker/assembler side effects reduce to text accumulation).
+  // Does not drain: the drain loop calls this mid-loop and keeps pulling.
+  async materializePendingUiDelta() {
+    const pending = this.pendingUiDelta;
+    if (!pending) {
+      return;
+    }
+    this.pendingUiDelta = null;
+    this.clearUiDeltaFlushTimer();
+    this.outputSeq += 1;
+    this.enqueueOutput(
+      buildUiMessageChunkOutputEnvelope({
+        context: pending.context,
+        outputSeq: this.outputSeq,
+        chunk: pending.chunk,
+        createdAt: pending.createdAt
+      })
+    );
+    this.uiMessagePartTracker.append(pending.chunk);
+    await this.appendUiChunkToAssembler(pending.chunk);
+  }
+  // ---------------------------------------------------------------------------
   // Delta coalescing
   // ---------------------------------------------------------------------------
   async bufferDelta(context, delta) {
@@ -40035,6 +40130,50 @@ function terminalEntryStatus(projection) {
   }
   return "completed";
 }
+function coalescibleUiDeltaChunk(projection) {
+  if (projection.statusText !== void 0 || projection.turnStatus !== void 0 || projection.usage !== void 0 || projection.consumedCommandIds !== void 0) {
+    return null;
+  }
+  const chunk = projection.chunk;
+  switch (chunk.type) {
+    case "text-delta":
+    case "reasoning-delta":
+      return chunk.providerMetadata === void 0 ? chunk : null;
+    case "tool-input-delta":
+      return chunk;
+    default:
+      return null;
+  }
+}
+function canCoalesceUiDelta(pending, context, chunk) {
+  if (pending.context.sessionId !== context.sessionId || pending.context.runtimeId !== context.runtimeId || pending.context.bridgeLeaseId !== context.bridgeLeaseId || pending.chunk.type !== chunk.type) {
+    return false;
+  }
+  if (pending.chunk.type === "tool-input-delta" && chunk.type === "tool-input-delta") {
+    return pending.chunk.toolCallId === chunk.toolCallId;
+  }
+  if (pending.chunk.type !== "tool-input-delta" && chunk.type !== "tool-input-delta") {
+    return pending.chunk.id === chunk.id;
+  }
+  return false;
+}
+function mergeUiDeltaChunks(pending, chunk) {
+  if (pending.type === "tool-input-delta" && chunk.type === "tool-input-delta") {
+    return {
+      ...pending,
+      inputTextDelta: pending.inputTextDelta + chunk.inputTextDelta
+    };
+  }
+  if (pending.type !== "tool-input-delta" && chunk.type !== "tool-input-delta") {
+    return { ...pending, delta: pending.delta + chunk.delta };
+  }
+  throw new Error(
+    `Cannot merge ui delta chunks of kinds ${pending.type} and ${chunk.type}`
+  );
+}
+function uiDeltaTextLength(chunk) {
+  return chunk.type === "tool-input-delta" ? chunk.inputTextDelta.length : chunk.delta.length;
+}
 function canCoalesceDelta(pending, context, delta) {
   return pending.context.sessionId === context.sessionId && pending.context.runtimeId === context.runtimeId && pending.context.bridgeLeaseId === context.bridgeLeaseId && pending.delta.messageId === delta.messageId && pending.delta.partId === delta.partId && pending.delta.role === delta.role && pending.delta.kind === delta.kind && pending.delta.delta.type === delta.delta.type;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autohq/cli",
-  "version": "0.1.326",
+  "version": "0.1.327",
   "license": "SEE LICENSE IN README.md",
   "publishConfig": {
     "access": "public"