@autohq/cli 0.1.551 → 0.1.553

package/dist/agent-bridge.js

@@ -30870,7 +30870,7 @@ Object.assign(lookup, {
 // package.json
 var package_default = {
   name: "@autohq/cli",
-  version: "0.1.551",
+  version: "0.1.553",
   license: "SEE LICENSE IN README.md",
   publishConfig: {
     access: "public"
@@ -31880,6 +31880,18 @@ var AuthActorSchema = external_exports.object({
   principal: PrincipalSchema,
   client: AuthClientSchema
 });
+var PersistedPrincipalSchema = external_exports.preprocess(
+  normalizePersistedPrincipal,
+  PrincipalSchema
+);
+var PersistedAuthClientSchema = external_exports.preprocess(
+  normalizePersistedAuthClient,
+  AuthClientSchema
+);
+var PersistedAuthActorSchema = external_exports.object({
+  principal: PersistedPrincipalSchema,
+  client: PersistedAuthClientSchema
+});
 var AuthContextSchema = external_exports.object({
   actor: AuthActorSchema,
   organizationId: OrganizationIdSchema,
@@ -31909,6 +31921,23 @@ var AuthWhoamiResponseSchema = external_exports.object({
     })
   }).optional()
 });
+function normalizePersistedPrincipal(value2) {
+  if (typeof value2 !== "object" || value2 === null) {
+    return value2;
+  }
+  const principal = value2;
+  if (principal.kind !== "run" || typeof principal.runId !== "string" || principal.runId.trim().length === 0) {
+    return value2;
+  }
+  return { kind: "session", sessionId: principal.runId };
+}
+function normalizePersistedAuthClient(value2) {
+  if (typeof value2 !== "object" || value2 === null) {
+    return value2;
+  }
+  const client = value2;
+  return client.kind === "run_token" ? { kind: "session_token" } : value2;
+}
 var SERVICE_ACCOUNT_READ_ONLY_SCOPES = [
   "environments:read",
   "tools:read",
@@ -71964,7 +71993,9 @@ var AgentBridgeOutputBuffer = class {
   uiDeltaFlushTimer = null;
   activeUiMessageAssembler = null;
   uiMessagePartTracker = new UiMessagePartTracker();
+  discardOnFailureOutputSeqs = /* @__PURE__ */ new Set();
   drainBlocked = false;
+  drainBlockedError = null;
   drainPromise = null;
   // Sequence of the most recent liveness beat, for the skip-while-pending
   // gate in emitLivenessBeat.
@@ -71975,7 +72006,7 @@ var AgentBridgeOutputBuffer = class {
   // ---------------------------------------------------------------------------
   // Public API
   // ---------------------------------------------------------------------------
-  async emitProjection(context, projection) {
+  async emitProjection(context, projection, options = {}) {
     if (projection.type === "ui_message_chunk") {
       const coalescible = coalescibleUiDeltaChunk(projection);
       if (coalescible) {
@@ -71992,7 +72023,7 @@ var AgentBridgeOutputBuffer = class {
     }
     await this.flushPendingUiDelta();
     await this.flushPendingDelta();
-    await this.enqueueProjectionAndDrain(context, projection);
+    await this.enqueueProjectionAndDrain(context, projection, options);
   }
   async replayPendingOutputs() {
     await this.materializePendingUiDelta();
@@ -72131,15 +72162,22 @@ var AgentBridgeOutputBuffer = class {
   // Transport emit
   // ---------------------------------------------------------------------------
   async drainPendingOutputs(options = {}) {
+    let previousDrainFailed = false;
     if (this.drainPromise && options.force) {
       try {
         await this.drainPromise;
       } catch {
+        previousDrainFailed = true;
       }
       this.drainBlocked = false;
+      this.drainBlockedError = null;
+    }
+    if (previousDrainFailed) {
+      this.discardFailureSensitiveOutputs();
     }
     if (options.force) {
       this.drainBlocked = false;
+      this.drainBlockedError = null;
       this.input.runtimeLogger?.info(
         "agent_bridge_output_buffer_replay_started",
         { pending_count: this.pendingOutputs.size }
@@ -72150,6 +72188,9 @@ var AgentBridgeOutputBuffer = class {
         "agent_bridge_output_buffer_drain_blocked",
         { pending_count: this.pendingOutputs.size }
       );
+      if (options.failIfBlocked) {
+        throw outputDrainBlockedError(this.drainBlockedError);
+      }
       return;
     }
     this.drainPromise ??= (async () => {
@@ -72167,6 +72208,7 @@ var AgentBridgeOutputBuffer = class {
       }
     })().catch((error51) => {
       this.drainBlocked = true;
+      this.drainBlockedError = error51;
       throw error51;
     }).finally(() => {
       this.drainPromise = null;
@@ -72363,6 +72405,7 @@ var AgentBridgeOutputBuffer = class {
     const ack = await this.input.emitOutput(output);
     if (isSuccessfulOutputAck(ack)) {
       this.pendingOutputs.delete(output.outputSeq);
+      this.discardOnFailureOutputSeqs.delete(output.outputSeq);
       this.input.runtimeLogger?.info(
         "agent_bridge_output_buffer_emit_ready",
         this.outputLogContext(output, {
@@ -72390,12 +72433,33 @@ var AgentBridgeOutputBuffer = class {
       (left, right) => left.outputSeq - right.outputSeq
     )[0] ?? null;
   }
-  async enqueueProjectionAndDrain(context, projection) {
+  async enqueueProjectionAndDrain(context, projection, options = {}) {
     this.outputSeq += 1;
-    this.enqueueOutput(
-      buildOutputEnvelope(context, this.outputSeq, projection)
-    );
-    await this.drainPendingOutputs();
+    const outputSeq = this.outputSeq;
+    this.enqueueOutput(buildOutputEnvelope(context, outputSeq, projection));
+    if (options.discardOnFailure) {
+      this.discardOnFailureOutputSeqs.add(outputSeq);
+    }
+    try {
+      await this.drainPendingOutputs({
+        failIfBlocked: options.requireDrain
+      });
+      this.discardOnFailureOutputSeqs.delete(outputSeq);
+    } catch (error51) {
+      if (options.discardOnFailure) {
+        this.pendingOutputs.delete(outputSeq);
+        this.discardOnFailureOutputSeqs.delete(outputSeq);
+        this.ackExhaustsBySeq.delete(outputSeq);
+      }
+      throw error51;
+    }
+  }
+  discardFailureSensitiveOutputs() {
+    for (const outputSeq of this.discardOnFailureOutputSeqs) {
+      this.pendingOutputs.delete(outputSeq);
+      this.discardOnFailureOutputSeqs.delete(outputSeq);
+      this.ackExhaustsBySeq.delete(outputSeq);
+    }
   }
   enqueueOutput(output) {
     const sanitized = isLiveOnlyRuntimeOutputEnvelope(output) ? output : sanitizeUnstorableStrings(output);
@@ -72821,6 +72885,10 @@ function buildUiMessageCompletedOutputEnvelope(input) {
 function isSuccessfulOutputAck(ack) {
   return ack.status === "persisted" || ack.status === "duplicate" || ack.status === "published";
 }
+function outputDrainBlockedError(cause) {
+  const detail = cause instanceof Error ? `: ${cause.message}` : "";
+  return new Error(`Bridge output drain is blocked${detail}`);
+}
 function now2() {
   return (/* @__PURE__ */ new Date()).toISOString();
 }
@@ -75178,7 +75246,7 @@ var ClaudeCodeCommandHandler = class {
         "agent_bridge_claude_command_user_entry_emit_started",
         commandLogContext(delivery, { socket_id: socketId })
       );
-      await this.emitBridgeOutput(activeContext, {
+      await this.emitRequiredBridgeOutput(activeContext, {
         type: "entry",
         entry: {
           messageId: delivery.commandId,
@@ -75279,7 +75347,7 @@ var ClaudeCodeCommandHandler = class {
       return;
     }
     const message = answerFallbackMessage(answer);
-    await this.emitBridgeOutput(activeContext, {
+    await this.emitRequiredBridgeOutput(activeContext, {
       type: "entry",
       entry: {
         messageId: delivery.commandId,
@@ -75359,6 +75427,24 @@ var ClaudeCodeCommandHandler = class {
       );
     }
   }
+  async emitRequiredBridgeOutput(activeContext, projection) {
+    const options = {
+      discardOnFailure: true,
+      requireDrain: true
+    };
+    try {
+      await this.outputBuffer.emitProjection(
+        activeContext,
+        projection,
+        options
+      );
+    } catch (error51) {
+      this.input.writeOutput?.(
+        `agent_bridge_output_emit_failed error=${error51 instanceof Error ? error51.message : String(error51)}`
+      );
+      throw error51;
+    }
+  }
   async handleAgentMessage(message, meta3) {
     const activeContext = this.context;
     if (!activeContext) {

package/dist/index.js

@@ -15442,7 +15442,24 @@ var init_account = __esm({
 });
 
 // ../../packages/schemas/src/auth.ts
-var AuthRoleSchema, AuthIdentityProviderSchema, AuthScopeSchema, AUTH_SCOPES, PrincipalSchema, AuthClientSchema, AuthActorSchema, AuthContextSchema, AuthWhoamiResponseSchema, SERVICE_ACCOUNT_READ_ONLY_SCOPES, SERVICE_ACCOUNT_SCOPE_PRESETS, ServiceAccountScopePresetSchema;
+function normalizePersistedPrincipal(value) {
+  if (typeof value !== "object" || value === null) {
+    return value;
+  }
+  const principal = value;
+  if (principal.kind !== "run" || typeof principal.runId !== "string" || principal.runId.trim().length === 0) {
+    return value;
+  }
+  return { kind: "session", sessionId: principal.runId };
+}
+function normalizePersistedAuthClient(value) {
+  if (typeof value !== "object" || value === null) {
+    return value;
+  }
+  const client = value;
+  return client.kind === "run_token" ? { kind: "session_token" } : value;
+}
+var AuthRoleSchema, AuthIdentityProviderSchema, AuthScopeSchema, AUTH_SCOPES, PrincipalSchema, AuthClientSchema, AuthActorSchema, PersistedPrincipalSchema, PersistedAuthClientSchema, PersistedAuthActorSchema, AuthContextSchema, AuthWhoamiResponseSchema, SERVICE_ACCOUNT_READ_ONLY_SCOPES, SERVICE_ACCOUNT_SCOPE_PRESETS, ServiceAccountScopePresetSchema;
 var init_auth = __esm({
   "../../packages/schemas/src/auth.ts"() {
     "use strict";
@@ -15529,6 +15546,18 @@ var init_auth = __esm({
       principal: PrincipalSchema,
       client: AuthClientSchema
     });
+    PersistedPrincipalSchema = external_exports.preprocess(
+      normalizePersistedPrincipal,
+      PrincipalSchema
+    );
+    PersistedAuthClientSchema = external_exports.preprocess(
+      normalizePersistedAuthClient,
+      AuthClientSchema
+    );
+    PersistedAuthActorSchema = external_exports.object({
+      principal: PersistedPrincipalSchema,
+      client: PersistedAuthClientSchema
+    });
     AuthContextSchema = external_exports.object({
       actor: AuthActorSchema,
       organizationId: OrganizationIdSchema,
@@ -82927,7 +82956,7 @@ var init_package = __esm({
   "package.json"() {
     package_default = {
       name: "@autohq/cli",
-      version: "0.1.551",
+      version: "0.1.553",
       license: "SEE LICENSE IN README.md",
       publishConfig: {
         access: "public"
@@ -96105,7 +96134,9 @@ var AgentBridgeOutputBuffer = class {
   uiDeltaFlushTimer = null;
   activeUiMessageAssembler = null;
   uiMessagePartTracker = new UiMessagePartTracker();
+  discardOnFailureOutputSeqs = /* @__PURE__ */ new Set();
   drainBlocked = false;
+  drainBlockedError = null;
   drainPromise = null;
   // Sequence of the most recent liveness beat, for the skip-while-pending
   // gate in emitLivenessBeat.
@@ -96116,7 +96147,7 @@ var AgentBridgeOutputBuffer = class {
   // ---------------------------------------------------------------------------
   // Public API
   // ---------------------------------------------------------------------------
-  async emitProjection(context, projection) {
+  async emitProjection(context, projection, options = {}) {
     if (projection.type === "ui_message_chunk") {
       const coalescible = coalescibleUiDeltaChunk(projection);
       if (coalescible) {
@@ -96133,7 +96164,7 @@ var AgentBridgeOutputBuffer = class {
     }
     await this.flushPendingUiDelta();
     await this.flushPendingDelta();
-    await this.enqueueProjectionAndDrain(context, projection);
+    await this.enqueueProjectionAndDrain(context, projection, options);
   }
   async replayPendingOutputs() {
     await this.materializePendingUiDelta();
@@ -96272,15 +96303,22 @@ var AgentBridgeOutputBuffer = class {
   // Transport emit
   // ---------------------------------------------------------------------------
   async drainPendingOutputs(options = {}) {
+    let previousDrainFailed = false;
     if (this.drainPromise && options.force) {
       try {
         await this.drainPromise;
       } catch {
+        previousDrainFailed = true;
       }
       this.drainBlocked = false;
+      this.drainBlockedError = null;
+    }
+    if (previousDrainFailed) {
+      this.discardFailureSensitiveOutputs();
     }
     if (options.force) {
       this.drainBlocked = false;
+      this.drainBlockedError = null;
       this.input.runtimeLogger?.info(
         "agent_bridge_output_buffer_replay_started",
         { pending_count: this.pendingOutputs.size }
@@ -96291,6 +96329,9 @@ var AgentBridgeOutputBuffer = class {
         "agent_bridge_output_buffer_drain_blocked",
         { pending_count: this.pendingOutputs.size }
       );
+      if (options.failIfBlocked) {
+        throw outputDrainBlockedError(this.drainBlockedError);
+      }
       return;
     }
     this.drainPromise ??= (async () => {
@@ -96308,6 +96349,7 @@ var AgentBridgeOutputBuffer = class {
       }
     })().catch((error51) => {
       this.drainBlocked = true;
+      this.drainBlockedError = error51;
       throw error51;
     }).finally(() => {
       this.drainPromise = null;
@@ -96504,6 +96546,7 @@ var AgentBridgeOutputBuffer = class {
     const ack = await this.input.emitOutput(output);
     if (isSuccessfulOutputAck(ack)) {
       this.pendingOutputs.delete(output.outputSeq);
+      this.discardOnFailureOutputSeqs.delete(output.outputSeq);
       this.input.runtimeLogger?.info(
         "agent_bridge_output_buffer_emit_ready",
         this.outputLogContext(output, {
@@ -96531,12 +96574,33 @@ var AgentBridgeOutputBuffer = class {
       (left, right) => left.outputSeq - right.outputSeq
     )[0] ?? null;
   }
-  async enqueueProjectionAndDrain(context, projection) {
+  async enqueueProjectionAndDrain(context, projection, options = {}) {
     this.outputSeq += 1;
-    this.enqueueOutput(
-      buildOutputEnvelope(context, this.outputSeq, projection)
-    );
-    await this.drainPendingOutputs();
+    const outputSeq = this.outputSeq;
+    this.enqueueOutput(buildOutputEnvelope(context, outputSeq, projection));
+    if (options.discardOnFailure) {
+      this.discardOnFailureOutputSeqs.add(outputSeq);
+    }
+    try {
+      await this.drainPendingOutputs({
+        failIfBlocked: options.requireDrain
+      });
+      this.discardOnFailureOutputSeqs.delete(outputSeq);
+    } catch (error51) {
+      if (options.discardOnFailure) {
+        this.pendingOutputs.delete(outputSeq);
+        this.discardOnFailureOutputSeqs.delete(outputSeq);
+        this.ackExhaustsBySeq.delete(outputSeq);
+      }
+      throw error51;
+    }
+  }
+  discardFailureSensitiveOutputs() {
+    for (const outputSeq of this.discardOnFailureOutputSeqs) {
+      this.pendingOutputs.delete(outputSeq);
+      this.discardOnFailureOutputSeqs.delete(outputSeq);
+      this.ackExhaustsBySeq.delete(outputSeq);
+    }
   }
   enqueueOutput(output) {
     const sanitized = isLiveOnlyRuntimeOutputEnvelope(output) ? output : sanitizeUnstorableStrings(output);
@@ -96962,6 +97026,10 @@ function buildUiMessageCompletedOutputEnvelope(input) {
 function isSuccessfulOutputAck(ack) {
   return ack.status === "persisted" || ack.status === "duplicate" || ack.status === "published";
 }
+function outputDrainBlockedError(cause) {
+  const detail = cause instanceof Error ? `: ${cause.message}` : "";
+  return new Error(`Bridge output drain is blocked${detail}`);
+}
 function now2() {
   return (/* @__PURE__ */ new Date()).toISOString();
 }
@@ -99326,7 +99394,7 @@ var ClaudeCodeCommandHandler = class {
         "agent_bridge_claude_command_user_entry_emit_started",
         commandLogContext(delivery, { socket_id: socketId })
       );
-      await this.emitBridgeOutput(activeContext, {
+      await this.emitRequiredBridgeOutput(activeContext, {
         type: "entry",
         entry: {
           messageId: delivery.commandId,
@@ -99427,7 +99495,7 @@ var ClaudeCodeCommandHandler = class {
       return;
     }
     const message = answerFallbackMessage(answer);
-    await this.emitBridgeOutput(activeContext, {
+    await this.emitRequiredBridgeOutput(activeContext, {
       type: "entry",
       entry: {
         messageId: delivery.commandId,
@@ -99507,6 +99575,24 @@ var ClaudeCodeCommandHandler = class {
       );
     }
   }
+  async emitRequiredBridgeOutput(activeContext, projection) {
+    const options = {
+      discardOnFailure: true,
+      requireDrain: true
+    };
+    try {
+      await this.outputBuffer.emitProjection(
+        activeContext,
+        projection,
+        options
+      );
+    } catch (error51) {
+      this.input.writeOutput?.(
+        `agent_bridge_output_emit_failed error=${error51 instanceof Error ? error51.message : String(error51)}`
+      );
+      throw error51;
+    }
+  }
   async handleAgentMessage(message, meta3) {
     const activeContext = this.context;
     if (!activeContext) {

package/package.json

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