@autohq/cli 0.1.291 → 0.1.293

package/dist/agent-bridge.js

@@ -19726,6 +19726,20 @@ var RuntimeBridgeOutputUiMessageChunkEnvelopeSchema = external_exports.object({
   chunk: JsonValueSchema,
   createdAt: external_exports.string().datetime()
 }).strict();
+var RuntimeBridgeOutputUiMessagePartEnvelopeSchema = external_exports.object({
+  type: external_exports.literal("ui.message.part"),
+  sessionId: SessionIdSchema,
+  runtimeId: RuntimeIdSchema,
+  bridgeLeaseId: RuntimeBridgeLeaseIdSchema,
+  outputSeq: external_exports.number().int().positive(),
+  messageId: external_exports.string().trim().min(1),
+  role: external_exports.enum(["system", "user", "assistant"]),
+  partIndex: external_exports.number().int().nonnegative(),
+  part: JsonValueSchema,
+  coveredSeq: external_exports.number().int().positive(),
+  messageMetadata: JsonValueSchema.optional(),
+  createdAt: external_exports.string().datetime()
+}).strict();
 var RuntimeBridgeOutputUiMessageCompletedEnvelopeSchema = external_exports.object({
   type: external_exports.literal("ui.message.completed"),
   sessionId: SessionIdSchema,
@@ -19745,6 +19759,7 @@ var RuntimeBridgeOutputEnvelopeSchema = external_exports.union([
   RuntimeBridgeOutputEntryEnvelopeSchema,
   RuntimeBridgeOutputDeltaEnvelopeSchema,
   RuntimeBridgeOutputUiMessageChunkEnvelopeSchema,
+  RuntimeBridgeOutputUiMessagePartEnvelopeSchema,
   RuntimeBridgeOutputUiMessageCompletedEnvelopeSchema
 ]);
 function legacyRuntimeTurnStatus(entry, sessionStatusAfter) {
@@ -23384,7 +23399,7 @@ Object.assign(lookup, {
 // package.json
 var package_default = {
   name: "@autohq/cli",
-  version: "0.1.291",
+  version: "0.1.293",
   license: "SEE LICENSE IN README.md",
   publishConfig: {
     access: "public"
@@ -36679,6 +36694,295 @@ var _a21;
 _a21 = symbol21;
 var defaultDownload2 = createDownload();
 
+// src/commands/agent-bridge/harness/ui-message-part-tracker.ts
+var UiMessagePartTracker = class {
+  messageId = null;
+  messageMetadata;
+  metadataEmitted = false;
+  parts = [];
+  textPartIndexById = /* @__PURE__ */ new Map();
+  reasoningPartIndexById = /* @__PURE__ */ new Map();
+  toolPartIndexByCallId = /* @__PURE__ */ new Map();
+  dataPartIndexById = /* @__PURE__ */ new Map();
+  append(chunk) {
+    switch (chunk.type) {
+      case "start": {
+        this.reset();
+        const messageId = chunk.messageId?.trim();
+        this.messageId = messageId && messageId !== UNKNOWN_MESSAGE_ID ? messageId : null;
+        this.messageMetadata = chunk.messageMetadata;
+        return [];
+      }
+      case "finish":
+      case "error":
+      case "abort": {
+        this.reset();
+        return [];
+      }
+      case "text-start": {
+        const index = this.pushPart({
+          type: "text",
+          text: "",
+          state: "streaming",
+          ...providerMetadataField(chunk.providerMetadata)
+        });
+        this.textPartIndexById.set(chunk.id, index);
+        return [];
+      }
+      case "text-delta": {
+        const part = this.partById(this.textPartIndexById, chunk.id);
+        if (part) {
+          part.text = String(part.text ?? "") + chunk.delta;
+        }
+        return [];
+      }
+      case "text-end": {
+        const index = this.textPartIndexById.get(chunk.id);
+        const part = this.partById(this.textPartIndexById, chunk.id);
+        this.textPartIndexById.delete(chunk.id);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "done";
+        if (chunk.providerMetadata !== void 0) {
+          part.providerMetadata = chunk.providerMetadata;
+        }
+        return this.persistable(index);
+      }
+      case "reasoning-start": {
+        const index = this.pushPart({
+          type: "reasoning",
+          text: "",
+          state: "streaming",
+          ...providerMetadataField(chunk.providerMetadata)
+        });
+        this.reasoningPartIndexById.set(chunk.id, index);
+        return [];
+      }
+      case "reasoning-delta": {
+        const part = this.partById(this.reasoningPartIndexById, chunk.id);
+        if (part) {
+          part.text = String(part.text ?? "") + chunk.delta;
+          if (chunk.providerMetadata !== void 0) {
+            part.providerMetadata = chunk.providerMetadata;
+          }
+        }
+        return [];
+      }
+      case "reasoning-end": {
+        const index = this.reasoningPartIndexById.get(chunk.id);
+        const part = this.partById(this.reasoningPartIndexById, chunk.id);
+        this.reasoningPartIndexById.delete(chunk.id);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "done";
+        if (chunk.providerMetadata !== void 0) {
+          part.providerMetadata = chunk.providerMetadata;
+        }
+        return this.persistable(index);
+      }
+      case "tool-input-start": {
+        const index = this.pushPart(
+          toolPartShape({
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            dynamic: chunk.dynamic,
+            state: "input-streaming"
+          })
+        );
+        this.toolPartIndexByCallId.set(chunk.toolCallId, index);
+        return [];
+      }
+      case "tool-input-available": {
+        const index = this.upsertToolPart(chunk.toolCallId, {
+          ...toolPartShape({
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            dynamic: chunk.dynamic,
+            state: "input-available"
+          }),
+          input: chunk.input,
+          ...chunk.providerExecuted !== void 0 ? { providerExecuted: chunk.providerExecuted } : {},
+          ...chunk.providerMetadata !== void 0 ? { callProviderMetadata: chunk.providerMetadata } : {}
+        });
+        return this.persistable(index);
+      }
+      case "tool-input-error": {
+        const index = this.upsertToolPart(chunk.toolCallId, {
+          ...toolPartShape({
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            dynamic: chunk.dynamic,
+            state: "output-error"
+          }),
+          input: chunk.input,
+          errorText: chunk.errorText
+        });
+        return this.persistable(index);
+      }
+      case "tool-output-available": {
+        const part = this.partById(
+          this.toolPartIndexByCallId,
+          chunk.toolCallId
+        );
+        const index = this.toolPartIndexByCallId.get(chunk.toolCallId);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "output-available";
+        part.output = chunk.output;
+        part.preliminary = chunk.preliminary;
+        return this.persistable(index);
+      }
+      case "tool-output-error": {
+        const part = this.partById(
+          this.toolPartIndexByCallId,
+          chunk.toolCallId
+        );
+        const index = this.toolPartIndexByCallId.get(chunk.toolCallId);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "output-error";
+        part.errorText = chunk.errorText;
+        return this.persistable(index);
+      }
+      case "start-step": {
+        const index = this.pushPart({ type: "step-start" });
+        return this.persistable(index);
+      }
+      case "finish-step": {
+        this.textPartIndexById.clear();
+        this.reasoningPartIndexById.clear();
+        return [];
+      }
+      case "file": {
+        const index = this.pushPart({
+          type: "file",
+          mediaType: chunk.mediaType,
+          url: chunk.url
+        });
+        return this.persistable(index);
+      }
+      case "source-url": {
+        const index = this.pushPart({
+          type: "source-url",
+          sourceId: chunk.sourceId,
+          url: chunk.url,
+          ...chunk.title !== void 0 ? { title: chunk.title } : {}
+        });
+        return this.persistable(index);
+      }
+      case "source-document": {
+        const index = this.pushPart({
+          type: "source-document",
+          sourceId: chunk.sourceId,
+          mediaType: chunk.mediaType,
+          title: chunk.title,
+          ...chunk.filename !== void 0 ? { filename: chunk.filename } : {}
+        });
+        return this.persistable(index);
+      }
+      default: {
+        return this.appendDataChunk(chunk);
+      }
+    }
+  }
+  appendDataChunk(chunk) {
+    if (!chunk.type.startsWith("data-") || chunk.type === "data-auto-question") {
+      return [];
+    }
+    const dataChunk = chunk;
+    if (dataChunk.transient) {
+      return [];
+    }
+    const part = {
+      type: dataChunk.type,
+      ...dataChunk.id !== void 0 ? { id: dataChunk.id } : {},
+      data: dataChunk.data
+    };
+    const dataKey = dataChunk.id ? `${dataChunk.type}:${dataChunk.id}` : null;
+    const existingIndex = dataKey ? this.dataPartIndexById.get(dataKey) : void 0;
+    if (existingIndex !== void 0) {
+      this.parts[existingIndex] = part;
+      return this.persistable(existingIndex);
+    }
+    const index = this.pushPart(part);
+    if (dataKey) {
+      this.dataPartIndexById.set(dataKey, index);
+    }
+    return this.persistable(index);
+  }
+  upsertToolPart(toolCallId, part) {
+    const existingIndex = this.toolPartIndexByCallId.get(toolCallId);
+    if (existingIndex !== void 0) {
+      this.parts[existingIndex] = part;
+      return existingIndex;
+    }
+    const index = this.pushPart(part);
+    this.toolPartIndexByCallId.set(toolCallId, index);
+    return index;
+  }
+  persistable(partIndex) {
+    const part = this.parts[partIndex];
+    if (!this.messageId || !part) {
+      return [];
+    }
+    const includeMetadata = !this.metadataEmitted && this.messageMetadata !== void 0;
+    if (includeMetadata) {
+      this.metadataEmitted = true;
+    }
+    return [
+      {
+        messageId: this.messageId,
+        partIndex,
+        part: toJsonValue(part),
+        ...includeMetadata ? { messageMetadata: toJsonValue(this.messageMetadata) } : {}
+      }
+    ];
+  }
+  pushPart(part) {
+    this.parts.push(part);
+    return this.parts.length - 1;
+  }
+  partById(indexById, id) {
+    const index = indexById.get(id);
+    if (index === void 0) {
+      return null;
+    }
+    return this.parts[index] ?? null;
+  }
+  reset() {
+    this.messageId = null;
+    this.messageMetadata = void 0;
+    this.metadataEmitted = false;
+    this.parts = [];
+    this.textPartIndexById = /* @__PURE__ */ new Map();
+    this.reasoningPartIndexById = /* @__PURE__ */ new Map();
+    this.toolPartIndexByCallId = /* @__PURE__ */ new Map();
+    this.dataPartIndexById = /* @__PURE__ */ new Map();
+  }
+};
+function toolPartShape(input) {
+  if (input.dynamic) {
+    return {
+      type: "dynamic-tool",
+      toolName: input.toolName,
+      toolCallId: input.toolCallId,
+      state: input.state
+    };
+  }
+  return {
+    type: `tool-${input.toolName}`,
+    toolCallId: input.toolCallId,
+    state: input.state
+  };
+}
+function providerMetadataField(providerMetadata) {
+  return providerMetadata !== void 0 ? { providerMetadata } : {};
+}
+
 // src/commands/agent-bridge/harness/output-buffer.ts
 var AGENT_BRIDGE_OUTPUT_DELTA_FLUSH_MS = 50;
 var AgentBridgeOutputBuffer = class {
@@ -36692,6 +36996,7 @@ var AgentBridgeOutputBuffer = class {
   pendingDelta = null;
   deltaFlushTimer = null;
   activeUiMessageAssembler = null;
+  uiMessagePartTracker = new UiMessagePartTracker();
   drainBlocked = false;
   drainPromise = null;
   // ---------------------------------------------------------------------------
@@ -36732,6 +37037,9 @@ var AgentBridgeOutputBuffer = class {
     }
     await this.flushPendingDelta();
     await this.emitLiveUiMessageChunk(context, projection.chunk);
+    await this.emitUiMessagePartSnapshots(context, projection.chunk, {
+      coveredSeq: this.outputSeq
+    });
     const completedMessage = await this.appendUiChunkToAssembler(
       projection.chunk
     );
@@ -36790,6 +37098,25 @@ var AgentBridgeOutputBuffer = class {
     this.enqueueOutput(output);
     await this.drainPendingOutputs();
   }
+  async emitUiMessagePartSnapshots(context, chunk, input) {
+    const snapshots = this.uiMessagePartTracker.append(chunk);
+    if (snapshots.length === 0) {
+      return;
+    }
+    for (const snapshot of snapshots) {
+      this.outputSeq += 1;
+      this.enqueueOutput(
+        buildUiMessagePartOutputEnvelope({
+          context,
+          outputSeq: this.outputSeq,
+          coveredSeq: input.coveredSeq,
+          snapshot,
+          createdAt: now2()
+        })
+      );
+    }
+    await this.drainPendingOutputs();
+  }
   // ---------------------------------------------------------------------------
   // Transport emit
   // ---------------------------------------------------------------------------
@@ -37102,6 +37429,23 @@ function buildUiMessageChunkOutputEnvelope(input) {
     createdAt: input.createdAt
   });
 }
+function buildUiMessagePartOutputEnvelope(input) {
+  const { context, snapshot } = input;
+  return RuntimeBridgeOutputEnvelopeSchema.parse({
+    type: "ui.message.part",
+    sessionId: context.sessionId,
+    runtimeId: context.runtimeId,
+    bridgeLeaseId: context.bridgeLeaseId,
+    outputSeq: input.outputSeq,
+    messageId: snapshot.messageId,
+    role: "assistant",
+    partIndex: snapshot.partIndex,
+    part: snapshot.part,
+    coveredSeq: input.coveredSeq,
+    ...snapshot.messageMetadata !== void 0 ? { messageMetadata: snapshot.messageMetadata } : {},
+    createdAt: input.createdAt
+  });
+}
 function buildUiMessageCompletedOutputEnvelope(input) {
   const { context } = input;
   return RuntimeBridgeOutputEnvelopeSchema.parse({

package/dist/index.js

@@ -23182,7 +23182,7 @@ var init_package = __esm({
   "package.json"() {
     package_default = {
       name: "@autohq/cli",
-      version: "0.1.291",
+      version: "0.1.293",
       license: "SEE LICENSE IN README.md",
       publishConfig: {
         access: "public"
@@ -33177,6 +33177,20 @@ var RuntimeBridgeOutputUiMessageChunkEnvelopeSchema = external_exports.object({
   chunk: JsonValueSchema2,
   createdAt: external_exports.string().datetime()
 }).strict();
+var RuntimeBridgeOutputUiMessagePartEnvelopeSchema = external_exports.object({
+  type: external_exports.literal("ui.message.part"),
+  sessionId: SessionIdSchema2,
+  runtimeId: RuntimeIdSchema2,
+  bridgeLeaseId: RuntimeBridgeLeaseIdSchema2,
+  outputSeq: external_exports.number().int().positive(),
+  messageId: external_exports.string().trim().min(1),
+  role: external_exports.enum(["system", "user", "assistant"]),
+  partIndex: external_exports.number().int().nonnegative(),
+  part: JsonValueSchema2,
+  coveredSeq: external_exports.number().int().positive(),
+  messageMetadata: JsonValueSchema2.optional(),
+  createdAt: external_exports.string().datetime()
+}).strict();
 var RuntimeBridgeOutputUiMessageCompletedEnvelopeSchema = external_exports.object({
   type: external_exports.literal("ui.message.completed"),
   sessionId: SessionIdSchema2,
@@ -33196,6 +33210,7 @@ var RuntimeBridgeOutputEnvelopeSchema = external_exports.union([
   RuntimeBridgeOutputEntryEnvelopeSchema,
   RuntimeBridgeOutputDeltaEnvelopeSchema,
   RuntimeBridgeOutputUiMessageChunkEnvelopeSchema,
+  RuntimeBridgeOutputUiMessagePartEnvelopeSchema,
   RuntimeBridgeOutputUiMessageCompletedEnvelopeSchema
 ]);
 function legacyRuntimeTurnStatus(entry, sessionStatusAfter) {
@@ -33697,6 +33712,298 @@ init_src();
 // src/commands/agent-bridge/harness/output-buffer.ts
 init_src();
 import { readUIMessageStream } from "ai";
+
+// src/commands/agent-bridge/harness/ui-message-part-tracker.ts
+init_src();
+var UiMessagePartTracker = class {
+  messageId = null;
+  messageMetadata;
+  metadataEmitted = false;
+  parts = [];
+  textPartIndexById = /* @__PURE__ */ new Map();
+  reasoningPartIndexById = /* @__PURE__ */ new Map();
+  toolPartIndexByCallId = /* @__PURE__ */ new Map();
+  dataPartIndexById = /* @__PURE__ */ new Map();
+  append(chunk) {
+    switch (chunk.type) {
+      case "start": {
+        this.reset();
+        const messageId = chunk.messageId?.trim();
+        this.messageId = messageId && messageId !== UNKNOWN_MESSAGE_ID ? messageId : null;
+        this.messageMetadata = chunk.messageMetadata;
+        return [];
+      }
+      case "finish":
+      case "error":
+      case "abort": {
+        this.reset();
+        return [];
+      }
+      case "text-start": {
+        const index = this.pushPart({
+          type: "text",
+          text: "",
+          state: "streaming",
+          ...providerMetadataField(chunk.providerMetadata)
+        });
+        this.textPartIndexById.set(chunk.id, index);
+        return [];
+      }
+      case "text-delta": {
+        const part = this.partById(this.textPartIndexById, chunk.id);
+        if (part) {
+          part.text = String(part.text ?? "") + chunk.delta;
+        }
+        return [];
+      }
+      case "text-end": {
+        const index = this.textPartIndexById.get(chunk.id);
+        const part = this.partById(this.textPartIndexById, chunk.id);
+        this.textPartIndexById.delete(chunk.id);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "done";
+        if (chunk.providerMetadata !== void 0) {
+          part.providerMetadata = chunk.providerMetadata;
+        }
+        return this.persistable(index);
+      }
+      case "reasoning-start": {
+        const index = this.pushPart({
+          type: "reasoning",
+          text: "",
+          state: "streaming",
+          ...providerMetadataField(chunk.providerMetadata)
+        });
+        this.reasoningPartIndexById.set(chunk.id, index);
+        return [];
+      }
+      case "reasoning-delta": {
+        const part = this.partById(this.reasoningPartIndexById, chunk.id);
+        if (part) {
+          part.text = String(part.text ?? "") + chunk.delta;
+          if (chunk.providerMetadata !== void 0) {
+            part.providerMetadata = chunk.providerMetadata;
+          }
+        }
+        return [];
+      }
+      case "reasoning-end": {
+        const index = this.reasoningPartIndexById.get(chunk.id);
+        const part = this.partById(this.reasoningPartIndexById, chunk.id);
+        this.reasoningPartIndexById.delete(chunk.id);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "done";
+        if (chunk.providerMetadata !== void 0) {
+          part.providerMetadata = chunk.providerMetadata;
+        }
+        return this.persistable(index);
+      }
+      case "tool-input-start": {
+        const index = this.pushPart(
+          toolPartShape({
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            dynamic: chunk.dynamic,
+            state: "input-streaming"
+          })
+        );
+        this.toolPartIndexByCallId.set(chunk.toolCallId, index);
+        return [];
+      }
+      case "tool-input-available": {
+        const index = this.upsertToolPart(chunk.toolCallId, {
+          ...toolPartShape({
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            dynamic: chunk.dynamic,
+            state: "input-available"
+          }),
+          input: chunk.input,
+          ...chunk.providerExecuted !== void 0 ? { providerExecuted: chunk.providerExecuted } : {},
+          ...chunk.providerMetadata !== void 0 ? { callProviderMetadata: chunk.providerMetadata } : {}
+        });
+        return this.persistable(index);
+      }
+      case "tool-input-error": {
+        const index = this.upsertToolPart(chunk.toolCallId, {
+          ...toolPartShape({
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            dynamic: chunk.dynamic,
+            state: "output-error"
+          }),
+          input: chunk.input,
+          errorText: chunk.errorText
+        });
+        return this.persistable(index);
+      }
+      case "tool-output-available": {
+        const part = this.partById(
+          this.toolPartIndexByCallId,
+          chunk.toolCallId
+        );
+        const index = this.toolPartIndexByCallId.get(chunk.toolCallId);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "output-available";
+        part.output = chunk.output;
+        part.preliminary = chunk.preliminary;
+        return this.persistable(index);
+      }
+      case "tool-output-error": {
+        const part = this.partById(
+          this.toolPartIndexByCallId,
+          chunk.toolCallId
+        );
+        const index = this.toolPartIndexByCallId.get(chunk.toolCallId);
+        if (!part || index === void 0) {
+          return [];
+        }
+        part.state = "output-error";
+        part.errorText = chunk.errorText;
+        return this.persistable(index);
+      }
+      case "start-step": {
+        const index = this.pushPart({ type: "step-start" });
+        return this.persistable(index);
+      }
+      case "finish-step": {
+        this.textPartIndexById.clear();
+        this.reasoningPartIndexById.clear();
+        return [];
+      }
+      case "file": {
+        const index = this.pushPart({
+          type: "file",
+          mediaType: chunk.mediaType,
+          url: chunk.url
+        });
+        return this.persistable(index);
+      }
+      case "source-url": {
+        const index = this.pushPart({
+          type: "source-url",
+          sourceId: chunk.sourceId,
+          url: chunk.url,
+          ...chunk.title !== void 0 ? { title: chunk.title } : {}
+        });
+        return this.persistable(index);
+      }
+      case "source-document": {
+        const index = this.pushPart({
+          type: "source-document",
+          sourceId: chunk.sourceId,
+          mediaType: chunk.mediaType,
+          title: chunk.title,
+          ...chunk.filename !== void 0 ? { filename: chunk.filename } : {}
+        });
+        return this.persistable(index);
+      }
+      default: {
+        return this.appendDataChunk(chunk);
+      }
+    }
+  }
+  appendDataChunk(chunk) {
+    if (!chunk.type.startsWith("data-") || chunk.type === "data-auto-question") {
+      return [];
+    }
+    const dataChunk = chunk;
+    if (dataChunk.transient) {
+      return [];
+    }
+    const part = {
+      type: dataChunk.type,
+      ...dataChunk.id !== void 0 ? { id: dataChunk.id } : {},
+      data: dataChunk.data
+    };
+    const dataKey = dataChunk.id ? `${dataChunk.type}:${dataChunk.id}` : null;
+    const existingIndex = dataKey ? this.dataPartIndexById.get(dataKey) : void 0;
+    if (existingIndex !== void 0) {
+      this.parts[existingIndex] = part;
+      return this.persistable(existingIndex);
+    }
+    const index = this.pushPart(part);
+    if (dataKey) {
+      this.dataPartIndexById.set(dataKey, index);
+    }
+    return this.persistable(index);
+  }
+  upsertToolPart(toolCallId, part) {
+    const existingIndex = this.toolPartIndexByCallId.get(toolCallId);
+    if (existingIndex !== void 0) {
+      this.parts[existingIndex] = part;
+      return existingIndex;
+    }
+    const index = this.pushPart(part);
+    this.toolPartIndexByCallId.set(toolCallId, index);
+    return index;
+  }
+  persistable(partIndex) {
+    const part = this.parts[partIndex];
+    if (!this.messageId || !part) {
+      return [];
+    }
+    const includeMetadata = !this.metadataEmitted && this.messageMetadata !== void 0;
+    if (includeMetadata) {
+      this.metadataEmitted = true;
+    }
+    return [
+      {
+        messageId: this.messageId,
+        partIndex,
+        part: toJsonValue(part),
+        ...includeMetadata ? { messageMetadata: toJsonValue(this.messageMetadata) } : {}
+      }
+    ];
+  }
+  pushPart(part) {
+    this.parts.push(part);
+    return this.parts.length - 1;
+  }
+  partById(indexById, id) {
+    const index = indexById.get(id);
+    if (index === void 0) {
+      return null;
+    }
+    return this.parts[index] ?? null;
+  }
+  reset() {
+    this.messageId = null;
+    this.messageMetadata = void 0;
+    this.metadataEmitted = false;
+    this.parts = [];
+    this.textPartIndexById = /* @__PURE__ */ new Map();
+    this.reasoningPartIndexById = /* @__PURE__ */ new Map();
+    this.toolPartIndexByCallId = /* @__PURE__ */ new Map();
+    this.dataPartIndexById = /* @__PURE__ */ new Map();
+  }
+};
+function toolPartShape(input) {
+  if (input.dynamic) {
+    return {
+      type: "dynamic-tool",
+      toolName: input.toolName,
+      toolCallId: input.toolCallId,
+      state: input.state
+    };
+  }
+  return {
+    type: `tool-${input.toolName}`,
+    toolCallId: input.toolCallId,
+    state: input.state
+  };
+}
+function providerMetadataField(providerMetadata) {
+  return providerMetadata !== void 0 ? { providerMetadata } : {};
+}
+
+// src/commands/agent-bridge/harness/output-buffer.ts
 var AGENT_BRIDGE_OUTPUT_DELTA_FLUSH_MS = 50;
 var AgentBridgeOutputBuffer = class {
   constructor(input) {
@@ -33709,6 +34016,7 @@ var AgentBridgeOutputBuffer = class {
   pendingDelta = null;
   deltaFlushTimer = null;
   activeUiMessageAssembler = null;
+  uiMessagePartTracker = new UiMessagePartTracker();
   drainBlocked = false;
   drainPromise = null;
   // ---------------------------------------------------------------------------
@@ -33749,6 +34057,9 @@ var AgentBridgeOutputBuffer = class {
     }
     await this.flushPendingDelta();
     await this.emitLiveUiMessageChunk(context, projection.chunk);
+    await this.emitUiMessagePartSnapshots(context, projection.chunk, {
+      coveredSeq: this.outputSeq
+    });
     const completedMessage = await this.appendUiChunkToAssembler(
       projection.chunk
     );
@@ -33807,6 +34118,25 @@ var AgentBridgeOutputBuffer = class {
     this.enqueueOutput(output);
     await this.drainPendingOutputs();
   }
+  async emitUiMessagePartSnapshots(context, chunk, input) {
+    const snapshots = this.uiMessagePartTracker.append(chunk);
+    if (snapshots.length === 0) {
+      return;
+    }
+    for (const snapshot of snapshots) {
+      this.outputSeq += 1;
+      this.enqueueOutput(
+        buildUiMessagePartOutputEnvelope({
+          context,
+          outputSeq: this.outputSeq,
+          coveredSeq: input.coveredSeq,
+          snapshot,
+          createdAt: now2()
+        })
+      );
+    }
+    await this.drainPendingOutputs();
+  }
   // ---------------------------------------------------------------------------
   // Transport emit
   // ---------------------------------------------------------------------------
@@ -34119,6 +34449,23 @@ function buildUiMessageChunkOutputEnvelope(input) {
     createdAt: input.createdAt
   });
 }
+function buildUiMessagePartOutputEnvelope(input) {
+  const { context, snapshot } = input;
+  return RuntimeBridgeOutputEnvelopeSchema.parse({
+    type: "ui.message.part",
+    sessionId: context.sessionId,
+    runtimeId: context.runtimeId,
+    bridgeLeaseId: context.bridgeLeaseId,
+    outputSeq: input.outputSeq,
+    messageId: snapshot.messageId,
+    role: "assistant",
+    partIndex: snapshot.partIndex,
+    part: snapshot.part,
+    coveredSeq: input.coveredSeq,
+    ...snapshot.messageMetadata !== void 0 ? { messageMetadata: snapshot.messageMetadata } : {},
+    createdAt: input.createdAt
+  });
+}
 function buildUiMessageCompletedOutputEnvelope(input) {
   const { context } = input;
   return RuntimeBridgeOutputEnvelopeSchema.parse({

package/package.json

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