@ai-sdk/harness-claude-code 1.0.168 → 1.0.170

package/src/bridge/claude-code-system-prompt.ts

@@ -1,11 +1,18 @@
-export function createClaudeCodeSystemPrompt(instructions?: string): {
+export function createClaudeCodeSystemPrompt(
+  instructions?: string,
+  options?: { excludeDynamicSections?: boolean },
+): {
   type: 'preset';
   preset: 'claude_code';
   append?: string;
+  excludeDynamicSections?: boolean;
 } {
   return {
     type: 'preset',
     preset: 'claude_code',
     ...(instructions ? { append: instructions } : {}),
+    ...(options?.excludeDynamicSections !== undefined
+      ? { excludeDynamicSections: options.excludeDynamicSections }
+      : {}),
   };
 }

package/src/bridge/create-emit-stream-event.ts

@@ -194,7 +194,14 @@ export function createEmitStreamEvent({
     // Emit `stream-start` once, on the first message, carrying the model the
     // CLI resolved to (the `system`/`init` message reports it — this is the
     // default model when none was configured).
-    if (!streamStarted) {
+    /*
+     * `command_lifecycle` frames are skipped here. The CLI's stdin reader
+     * forwards them (e.g. `queued` and `started` for the prompt) straight to
+     * the output stream, independently of the engine that yields
+     * `system`/`init` at the start of each turn, so they can arrive before
+     * `init`. They produce no stream parts.
+     */
+    if (!streamStarted && type !== 'command_lifecycle') {
       const initModel =
         type === 'system' &&
         msg.subtype === 'init' &&

package/src/bridge/index.ts

@@ -570,6 +570,12 @@ async function runTurn(start: StartMessage, turn: BridgeTurn): Promise<void> {
     options: {
       ...(start.model ? { model: start.model } : {}),
       ...(start.maxTurns !== undefined ? { maxTurns: start.maxTurns } : {}),
+      ...(start.maxBudgetUsd !== undefined
+        ? { maxBudgetUsd: start.maxBudgetUsd }
+        : {}),
+      ...(start.strictMcpConfig !== undefined
+        ? { strictMcpConfig: start.strictMcpConfig }
+        : {}),
       ...(start.agentProgressSummaries !== undefined
         ? { agentProgressSummaries: start.agentProgressSummaries }
         : {}),
@@ -582,7 +588,9 @@ async function runTurn(start: StartMessage, turn: BridgeTurn): Promise<void> {
       ...(inactiveNativeTools.length > 0
         ? { disallowedTools: inactiveNativeTools }
         : {}),
-      systemPrompt: createClaudeCodeSystemPrompt(start.instructions),
+      systemPrompt: createClaudeCodeSystemPrompt(start.instructions, {
+        excludeDynamicSections: start.excludeDynamicSections,
+      }),
       thinking: start.thinking,
       ...(start.effort !== undefined ? { effort: start.effort } : {}),
       ...(start.responseFormat?.type === 'json' &&
@@ -603,6 +611,9 @@ async function runTurn(start: StartMessage, turn: BridgeTurn): Promise<void> {
           {
             matcher: 'AskUserQuestion',
             hooks: [questionPreToolUseHook],
+            ...(start.questionTimeoutMs !== undefined
+              ? { timeout: Math.ceil(start.questionTimeoutMs / 1000) }
+              : {}),
           },
         ],
         PostCompact: [

package/src/claude-code-bridge-protocol.ts

@@ -37,6 +37,10 @@ export const startMessageSchema = harnessV1BridgeStartBaseSchema.extend({
   thinking: thinkingSchema,
   effort: z.enum(['low', 'medium', 'high', 'xhigh', 'max']).optional(),
   maxTurns: z.number().optional(),
+  maxBudgetUsd: z.number().optional(),
+  strictMcpConfig: z.boolean().optional(),
+  questionTimeoutMs: z.number().optional(),
+  excludeDynamicSections: z.boolean().optional(),
   agentProgressSummaries: z.boolean().optional(),
   forwardSubagentText: z.boolean().optional(),
   env: z.record(z.string(), z.string()).optional(),

package/src/claude-code-harness.ts

@@ -46,6 +46,7 @@ import {
   waitForBridgeReady,
   writeSkills,
   type BridgeHello,
+  type SandboxChannelDebugEvent,
   type SandboxChannelReconnectOptions,
 } from '@ai-sdk/harness/utils';
 import {
@@ -103,6 +104,27 @@ export type ClaudeCodeHarnessSettings = {
    * back to the caller. Unset means the CLI's default.
    */
   readonly maxTurns?: number;
+  /**
+   * Maximum estimated spend in USD for each Claude query. When it is exceeded,
+   * the query stops and the turn ends with an error. Unset means no limit.
+   */
+  readonly maxBudgetUsd?: number;
+  /**
+   * Only use the MCP servers configured with `mcpServers`, and ignore any other
+   * MCP configuration such as a project `.mcp.json` or user settings.
+   */
+  readonly strictMcpConfig?: boolean;
+  /**
+   * Maximum milliseconds Claude waits for an answer to an `AskUserQuestion`
+   * question before the question fails. Unset means the Claude Code default.
+   */
+  readonly questionTimeoutMs?: number;
+  /**
+   * Moves per-session sections (working directory, memory, git status) out of
+   * the Claude Code system prompt and into the first user message, so the
+   * system prompt can be cached across sessions.
+   */
+  readonly excludeDynamicSections?: boolean;
   /**
    * Enables periodic AI-generated progress summaries for running subagents.
    * The summaries are forwarded in raw `task_progress` stream parts.
@@ -149,6 +171,14 @@ export type ClaudeCodeHarnessSettings = {
    * milliseconds up to 2 seconds.
    */
   readonly reconnect?: SandboxChannelReconnectOptions;
+  /**
+   * Called with bridge connection lifecycle events: reconnect attempts,
+   * successful reconnects, and `reconnect-failed` once the connection is
+   * closed for good. Hosts that cache idle sessions can use it to reattach
+   * before sending the next prompt. The `cause` of a failed reconnect can
+   * contain connection details, so avoid logging it as is.
+   */
+  readonly onDebug?: (event: SandboxChannelDebugEvent) => void;
   /**
    * Creates the authentication token used by the sandbox bridge. Defaults to
    * a random 32-byte hexadecimal token.
@@ -1028,6 +1058,7 @@ export function createClaudeCode(
             onDiagnostic,
             onBridgeError,
             reconnect: settings.reconnect,
+            onDebug: settings.onDebug,
           });
           const finishListenerAttachment = isContinue
             ? attachChannel.beginListenerAttachment()
@@ -1048,6 +1079,10 @@ export function createClaudeCode(
             // sandbox is left running, stopped, or destroyed.
             proc: undefined,
             maxTurns: settings.maxTurns,
+            maxBudgetUsd: settings.maxBudgetUsd,
+            strictMcpConfig: settings.strictMcpConfig,
+            questionTimeoutMs: settings.questionTimeoutMs,
+            excludeDynamicSections: settings.excludeDynamicSections,
             agentProgressSummaries: settings.agentProgressSummaries,
             forwardSubagentText: settings.forwardSubagentText,
             env: sandboxClaudeEnvironment,
@@ -1191,6 +1226,7 @@ export function createClaudeCode(
         onDiagnostic,
         onBridgeError,
         reconnect: settings.reconnect,
+        onDebug: settings.onDebug,
         // In replay mode the respawned bridge reloaded the finished turn from
         // disk; seed the cursor and resume so it streams the tail (incl.
         // `finish`) rather than starting empty.
@@ -1217,6 +1253,10 @@ export function createClaudeCode(
         finishListenerAttachment,
         proc,
         maxTurns: settings.maxTurns,
+        maxBudgetUsd: settings.maxBudgetUsd,
+        strictMcpConfig: settings.strictMcpConfig,
+        questionTimeoutMs: settings.questionTimeoutMs,
+        excludeDynamicSections: settings.excludeDynamicSections,
         agentProgressSummaries: settings.agentProgressSummaries,
         forwardSubagentText: settings.forwardSubagentText,
         env: sandboxClaudeEnvironment,
@@ -1255,6 +1295,10 @@ function createSession({
   finishListenerAttachment,
   proc,
   maxTurns,
+  maxBudgetUsd,
+  strictMcpConfig,
+  questionTimeoutMs,
+  excludeDynamicSections,
   agentProgressSummaries,
   forwardSubagentText,
   env,
@@ -1282,6 +1326,10 @@ function createSession({
   /** Undefined on `attach` — the live bridge was spawned by another process. */
   proc: Experimental_SandboxProcess | undefined;
   maxTurns: number | undefined;
+  maxBudgetUsd: number | undefined;
+  strictMcpConfig: boolean | undefined;
+  questionTimeoutMs: number | undefined;
+  excludeDynamicSections: boolean | undefined;
   agentProgressSummaries: boolean | undefined;
   forwardSubagentText: boolean | undefined;
   env: Readonly<Record<string, string>> | undefined;
@@ -1559,6 +1607,12 @@ function createSession({
           : {}),
         model: promptOpts.model,
         maxTurns,
+        ...(maxBudgetUsd !== undefined ? { maxBudgetUsd } : {}),
+        ...(strictMcpConfig !== undefined ? { strictMcpConfig } : {}),
+        ...(questionTimeoutMs !== undefined ? { questionTimeoutMs } : {}),
+        ...(excludeDynamicSections !== undefined
+          ? { excludeDynamicSections }
+          : {}),
         ...(agentProgressSummaries !== undefined
           ? { agentProgressSummaries }
           : {}),
@@ -1630,6 +1684,12 @@ function createSession({
             : {}),
           model: continueOpts.model,
           maxTurns,
+          ...(maxBudgetUsd !== undefined ? { maxBudgetUsd } : {}),
+          ...(strictMcpConfig !== undefined ? { strictMcpConfig } : {}),
+          ...(questionTimeoutMs !== undefined ? { questionTimeoutMs } : {}),
+          ...(excludeDynamicSections !== undefined
+            ? { excludeDynamicSections }
+            : {}),
           ...(agentProgressSummaries !== undefined
             ? { agentProgressSummaries }
             : {}),