whale-code 6.5.6 → 6.5.7

package/dist/shared/agent-loop-base.js

@@ -0,0 +1,347 @@
+/**
+ * AgentLoopBase — Template Method base class for all agent loops.
+ *
+ * Owns the turn iteration skeleton. Subclasses implement the abstract methods
+ * to customize API calling, tool routing, abort checking, and lifecycle hooks.
+ *
+ * Shared concrete logic:
+ * - Turn loop with maxTurns enforcement
+ * - Cost budget checking and warnings
+ * - Compaction handling (native + circuit breaker)
+ * - max_tokens continuation injection
+ * - Token/cost accumulation
+ * - Tool dispatch orchestration
+ *
+ * Subclasses:
+ * - CLIAgentLoop (src/cli/services/cli-agent-loop.ts)
+ * - ServerAgentLoop (src/server/lib/server-agent-loop-v2.ts)
+ * - SubagentLoop (src/cli/services/subagent-loop-v2.ts)
+ */
+
+import { LoopDetector, estimateCostUsd, emitCostWarningIfNeeded, DEFAULT_SESSION_COST_BUDGET_USD, AGENT_DEFAULTS } from "./agent-core.js";
+import { dispatchTools } from "./tool-dispatch.js";
+
+// ============================================================================
+// TYPES
+// ============================================================================
+
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+
+const MAX_CONSECUTIVE_COMPACTIONS = 3;
+const CONTINUATION_PROMPT = "Continue from where you stopped. Do NOT repeat yourself.";
+const WRAP_UP_PROMPT = "You've used significant context. Wrap up your current work and provide a final summary.";
+
+// ============================================================================
+// ABSTRACT BASE CLASS
+// ============================================================================
+
+export class AgentLoopBase {
+  // Accumulated state
+
+  turnCount = 0;
+  totalInputTokens = 0;
+  totalOutputTokens = 0;
+  totalCacheRead = 0;
+  totalCacheCreation = 0;
+  totalThinkingTokens = 0;
+  sessionCostUsd = 0;
+  toolsUsed = new Set();
+  toolCallCount = 0;
+  lastStopReason = "end_turn";
+  lastText = "";
+
+  // Compaction state
+  compactionCount = 0;
+  consecutiveCompactions = 0;
+  continuationCount = 0;
+  costWarningsEmitted = new Set();
+  constructor(config, events = {}) {
+    this.config = config;
+    this.events = events;
+    this.messages = [...config.messages];
+    this.loopDetector = new LoopDetector(config.loopDetectorConfig);
+  }
+
+  // ============================================================================
+  // TEMPLATE METHOD — the turn loop skeleton
+  // ============================================================================
+
+  async run() {
+    const maxTurns = this.config.maxTurns;
+    const maxCostUsd = this.config.maxCostUsd ?? DEFAULT_SESSION_COST_BUDGET_USD;
+    const maxContinuations = this.config.maxContinuations ?? AGENT_DEFAULTS.maxContinuations;
+    for (let iteration = 0; iteration < maxTurns; iteration++) {
+      // 1. Check abort (subclass: signal, disconnect, timeout)
+      if (this.checkAbort()) {
+        this.onAborted();
+        break;
+      }
+
+      // 2. Check cost budget
+      if (isFinite(maxCostUsd) && this.sessionCostUsd >= maxCostUsd) {
+        this.events.onText?.(`\n[Cost budget exhausted ($${this.sessionCostUsd.toFixed(2)}/$${maxCostUsd.toFixed(2)}). Stopping.]`);
+        break;
+      }
+
+      // 3. Pre-compact messages (subclass can override for aggressive compaction)
+      this.messages = this.preCompactMessages(this.messages);
+
+      // 4. Execute turn — ABSTRACT (API call, streaming, response parsing)
+      let turnResult;
+      try {
+        turnResult = await this.executeTurn(this.messages, iteration);
+      } catch (err) {
+        const handled = await this.handleTurnError(err, iteration);
+        if (!handled) break;
+        continue;
+      }
+
+      // 5. Accumulate tokens
+      this.accumulateTokens(turnResult);
+
+      // 6. Emit cost warnings
+      emitCostWarningIfNeeded(this.sessionCostUsd, maxCostUsd, this.costWarningsEmitted, this.events.onText);
+
+      // 7. Subclass lifecycle hook
+      await this.onTurnComplete(turnResult, iteration);
+
+      // 8. Handle native compaction
+      if (turnResult.stopReason === "compaction" && turnResult.compactionContent) {
+        this.compactionCount++;
+        this.consecutiveCompactions++;
+
+        // Circuit breaker — too many compactions or budget exceeded
+        if (this.consecutiveCompactions >= MAX_CONSECUTIVE_COMPACTIONS) {
+          this.messages = [{
+            role: "assistant",
+            content: turnResult.compactionContent
+          }, {
+            role: "user",
+            content: WRAP_UP_PROMPT
+          }];
+          this.consecutiveCompactions = 0;
+          iteration--; // Don't count compaction as a turn
+          continue;
+        }
+
+        // Normal compaction — replace messages with compacted summary
+        this.messages = [{
+          role: "assistant",
+          content: turnResult.compactionContent
+        }, {
+          role: "user",
+          content: CONTINUATION_PROMPT
+        }];
+        await this.onMessagesUpdated("compaction");
+        iteration--; // Don't count compaction as a turn
+        continue;
+      }
+      this.consecutiveCompactions = 0;
+
+      // 9. Handle max_tokens continuation
+      if (turnResult.stopReason === "max_tokens" && turnResult.toolUseBlocks.length === 0) {
+        this.continuationCount++;
+        if (this.continuationCount <= maxContinuations) {
+          // Build assistant message, then inject continuation
+          const assistantContent = this.buildAssistantContent(turnResult);
+          this.messages.push({
+            role: "assistant",
+            content: assistantContent
+          });
+          this.messages.push({
+            role: "user",
+            content: CONTINUATION_PROMPT
+          });
+          this.lastText = turnResult.text;
+          await this.onMessagesUpdated("continuation");
+          continue;
+        }
+        // Exceeded max continuations — stop
+        this.lastText = turnResult.text;
+        break;
+      }
+
+      // 10. Record text
+      if (turnResult.text) {
+        this.lastText = turnResult.text;
+      }
+      this.lastStopReason = turnResult.stopReason;
+
+      // 11. No tools → done
+      if (turnResult.toolUseBlocks.length === 0) {
+        break;
+      }
+
+      // 12. Dispatch tools and append results
+      //     Subclass can override dispatchToolsAndAppend() entirely for custom dispatch.
+      const shouldContinue = await this.dispatchToolsAndAppend(turnResult);
+      if (!shouldContinue) break;
+    }
+    return this.buildResult();
+  }
+
+  // ============================================================================
+  // CONCRETE METHODS — shared behavior
+  // ============================================================================
+
+  accumulateTokens(turn) {
+    this.totalInputTokens += turn.usage.inputTokens;
+    this.totalOutputTokens += turn.usage.outputTokens;
+    this.totalCacheRead += turn.usage.cacheRead || 0;
+    this.totalCacheCreation += turn.usage.cacheCreation || 0;
+    this.totalThinkingTokens += turn.usage.thinkingTokens || 0;
+    this.turnCount++;
+    this.sessionCostUsd = estimateCostUsd(this.totalInputTokens, this.totalOutputTokens, this.config.model, this.totalThinkingTokens, this.totalCacheRead, this.totalCacheCreation);
+  }
+  getToolDispatchOptions() {
+    return {
+      loopDetector: this.loopDetector,
+      maxConcurrent: this.config.maxConcurrentTools ?? AGENT_DEFAULTS.maxConcurrentTools,
+      batchErrorLimit: this.config.batchErrorLimit ?? AGENT_DEFAULTS.loopBatchErrorLimit,
+      onStart: this.events.onToolStart,
+      onResult: this.events.onToolResult
+    };
+  }
+  buildAssistantContent(turn) {
+    const blocks = [];
+
+    // Thinking blocks first
+    if (turn.thinkingBlocks?.length) {
+      for (const tb of turn.thinkingBlocks) {
+        blocks.push({
+          type: "thinking",
+          thinking: tb.thinking,
+          signature: tb.signature
+        });
+      }
+    }
+
+    // Text
+    if (turn.text) {
+      blocks.push({
+        type: "text",
+        text: turn.text
+      });
+    }
+
+    // Tool use blocks
+    for (const tu of turn.toolUseBlocks) {
+      blocks.push({
+        type: "tool_use",
+        id: tu.id,
+        name: tu.name,
+        input: tu.input
+      });
+    }
+    return blocks.length > 0 ? blocks : turn.text;
+  }
+  buildResult() {
+    return {
+      finalText: this.lastText,
+      turnCount: this.turnCount,
+      tokens: {
+        input: this.totalInputTokens,
+        output: this.totalOutputTokens,
+        cacheRead: this.totalCacheRead,
+        cacheCreation: this.totalCacheCreation
+      },
+      costUsd: this.sessionCostUsd,
+      toolsUsed: Array.from(this.toolsUsed),
+      toolCallCount: this.toolCallCount,
+      stopReason: this.lastStopReason,
+      loopDetectorStats: this.loopDetector.getSessionStats()
+    };
+  }
+
+  // ============================================================================
+  // ABSTRACT METHODS — subclass must implement
+  // ============================================================================
+
+  /** Execute a single API turn and return normalized result. */
+
+  /** Create a tool executor appropriate for this loop type. */
+
+  // ============================================================================
+  // VIRTUAL METHODS — subclass may override
+  // ============================================================================
+
+  /** Check if the loop should abort (signal, disconnect, timeout). */
+  checkAbort() {
+    return false;
+  }
+
+  /** Called when abort is detected. */
+  onAborted() {}
+
+  /** Pre-compact messages before each turn. Subagent can be more aggressive. */
+  preCompactMessages(msgs) {
+    return msgs;
+  }
+
+  /** Called after each completed turn. Subclass: telemetry, usage events. */
+  async onTurnComplete(_turn, _iteration) {}
+
+  /** Called after tool results are ready. CLI: shell summarization. */
+  async onToolResultsReady(_toolBlocks, _results) {}
+
+  /** Called after messages are updated. Server: checkpointing, persistence. */
+  async onMessagesUpdated(_reason) {}
+
+  /** Handle a turn-level error. Return true if the loop should continue, false to break. */
+  async handleTurnError(_err, _iteration) {
+    return false;
+  }
+
+  /**
+   * Dispatch tools and append results to messages. Returns true to continue loop, false to break.
+   * Default implementation uses shared dispatchTools(). Subclass can override entirely for
+   * custom dispatch (e.g. SubagentLoop uses executeToolBlocks instead).
+   */
+  async dispatchToolsAndAppend(turnResult) {
+    const toolExecutor = this.createToolExecutor();
+    const dispatchOpts = this.getToolDispatchOptions();
+
+    // Record tool names
+    for (const tb of turnResult.toolUseBlocks) {
+      this.toolsUsed.add(tb.name);
+      this.toolCallCount++;
+    }
+    const {
+      results: toolResults,
+      bailOut,
+      bailMessage
+    } = await dispatchTools(turnResult.toolUseBlocks, toolExecutor, dispatchOpts);
+
+    // Bail out from loop detector
+    if (bailOut) {
+      this.events.onText?.(bailMessage || "\n[Tool loop detected. Stopping.]");
+      return false;
+    }
+
+    // End-of-turn loop detector check
+    const bailCheck = this.loopDetector.endTurn();
+    if (bailCheck.shouldBail) {
+      this.events.onText?.(bailCheck.message || "\n[Consecutive failures detected. Stopping.]");
+      return false;
+    }
+
+    // Subclass hook for tool results
+    await this.onToolResultsReady(turnResult.toolUseBlocks, toolResults);
+
+    // Append messages
+    const assistantContent = this.buildAssistantContent(turnResult);
+    this.messages.push({
+      role: "assistant",
+      content: assistantContent
+    });
+    this.messages.push({
+      role: "user",
+      content: toolResults
+    });
+    await this.onMessagesUpdated("tool_results");
+    return true;
+  }
+}
+//# sourceMappingURL=agent-loop-base.js.map

package/dist/shared/agent-loop-base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-loop-base.js","names":["LoopDetector","estimateCostUsd","emitCostWarningIfNeeded","DEFAULT_SESSION_COST_BUDGET_USD","AGENT_DEFAULTS","dispatchTools","MAX_CONSECUTIVE_COMPACTIONS","CONTINUATION_PROMPT","WRAP_UP_PROMPT","AgentLoopBase","turnCount","totalInputTokens","totalOutputTokens","totalCacheRead","totalCacheCreation","totalThinkingTokens","sessionCostUsd","toolsUsed","Set","toolCallCount","lastStopReason","lastText","compactionCount","consecutiveCompactions","continuationCount","costWarningsEmitted","constructor","config","events","messages","loopDetector","loopDetectorConfig","run","maxTurns","maxCostUsd","maxContinuations","iteration","checkAbort","onAborted","isFinite","onText","toFixed","preCompactMessages","turnResult","executeTurn","err","handled","handleTurnError","accumulateTokens","onTurnComplete","stopReason","compactionContent","role","content","onMessagesUpdated","toolUseBlocks","length","assistantContent","buildAssistantContent","push","text","shouldContinue","dispatchToolsAndAppend","buildResult","turn","usage","inputTokens","outputTokens","cacheRead","cacheCreation","thinkingTokens","model","getToolDispatchOptions","maxConcurrent","maxConcurrentTools","batchErrorLimit","loopBatchErrorLimit","onStart","onToolStart","onResult","onToolResult","blocks","thinkingBlocks","tb","type","thinking","signature","tu","id","name","input","finalText","tokens","output","costUsd","Array","from","loopDetectorStats","getSessionStats","msgs","_turn","_iteration","onToolResultsReady","_toolBlocks","_results","_reason","_err","toolExecutor","createToolExecutor","dispatchOpts","add","results","toolResults","bailOut","bailMessage","bailCheck","endTurn","shouldBail","message"],"sources":["../../src/shared/agent-loop-base.ts"],"sourcesContent":["/**\n * AgentLoopBase — Template Method base class for all agent loops.\n *\n * Owns the turn iteration skeleton. Subclasses implement the abstract methods\n * to customize API calling, tool routing, abort checking, and lifecycle hooks.\n *\n * Shared concrete logic:\n * - Turn loop with maxTurns enforcement\n * - Cost budget checking and warnings\n * - Compaction handling (native + circuit breaker)\n * - max_tokens continuation injection\n * - Token/cost accumulation\n * - Tool dispatch orchestration\n *\n * Subclasses:\n * - CLIAgentLoop (src/cli/services/cli-agent-loop.ts)\n * - ServerAgentLoop (src/server/lib/server-agent-loop-v2.ts)\n * - SubagentLoop (src/cli/services/subagent-loop-v2.ts)\n */\n\nimport { LoopDetector, estimateCostUsd, emitCostWarningIfNeeded, DEFAULT_SESSION_COST_BUDGET_USD, AGENT_DEFAULTS } from \"./agent-core.js\";\nimport type { BailCheckResult, LoopDetectorConfig } from \"./agent-core-types.js\";\nimport type { CompactableMessage } from \"./compaction.js\";\nimport { dispatchTools, type ToolExecutor, type ToolDispatchOptions } from \"./tool-dispatch.js\";\nimport type { ToolUseBlock, ToolResultMessage } from \"./types.js\";\n\n// ============================================================================\n// TYPES\n// ============================================================================\n\nexport interface AgentLoopConfig {\n  model: string;\n  systemPrompt: string;\n  messages: CompactableMessage[];\n  maxTurns: number;\n  maxCostUsd?: number;\n  maxConcurrentTools?: number;\n  maxToolResultChars?: number;\n  maxContinuations?: number;\n  loopDetectorConfig?: Partial<LoopDetectorConfig>;\n  batchErrorLimit?: number;\n}\n\nexport interface AgentLoopEvents {\n  onText?: (text: string) => void;\n  onToolStart?: (name: string, input: Record<string, unknown>) => void;\n  onToolResult?: (name: string, success: boolean, output: string, durationMs: number) => void;\n  onCostWarning?: (pct: number, costUsd: number, budgetUsd: number) => void;\n}\n\nexport interface TurnResult {\n  text: string;\n  toolUseBlocks: ToolUseBlock[];\n  thinkingBlocks?: Array<{ type: \"thinking\"; thinking: string; signature: string }>;\n  compactionContent?: string | null;\n  stopReason: string;\n  usage: {\n    inputTokens: number;\n    outputTokens: number;\n    cacheRead?: number;\n    cacheCreation?: number;\n    thinkingTokens?: number;\n  };\n}\n\nexport interface AgentLoopResult {\n  finalText: string;\n  turnCount: number;\n  tokens: {\n    input: number;\n    output: number;\n    cacheRead: number;\n    cacheCreation: number;\n  };\n  costUsd: number;\n  toolsUsed: string[];\n  toolCallCount: number;\n  stopReason: string;\n  loopDetectorStats?: { totalErrors: number; failedStrategies: number; consecutiveFailedTurns: number };\n}\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nconst MAX_CONSECUTIVE_COMPACTIONS = 3;\nconst CONTINUATION_PROMPT = \"Continue from where you stopped. Do NOT repeat yourself.\";\nconst WRAP_UP_PROMPT = \"You've used significant context. Wrap up your current work and provide a final summary.\";\n\n// ============================================================================\n// ABSTRACT BASE CLASS\n// ============================================================================\n\nexport abstract class AgentLoopBase {\n  protected config: AgentLoopConfig;\n  protected events: AgentLoopEvents;\n  protected loopDetector: LoopDetector;\n\n  // Accumulated state\n  protected messages: CompactableMessage[];\n  protected turnCount = 0;\n  protected totalInputTokens = 0;\n  protected totalOutputTokens = 0;\n  protected totalCacheRead = 0;\n  protected totalCacheCreation = 0;\n  protected totalThinkingTokens = 0;\n  protected sessionCostUsd = 0;\n  protected toolsUsed = new Set<string>();\n  protected toolCallCount = 0;\n  protected lastStopReason = \"end_turn\";\n  protected lastText = \"\";\n\n  // Compaction state\n  protected compactionCount = 0;\n  protected consecutiveCompactions = 0;\n  protected continuationCount = 0;\n  protected costWarningsEmitted = new Set<number>();\n\n  constructor(config: AgentLoopConfig, events: AgentLoopEvents = {}) {\n    this.config = config;\n    this.events = events;\n    this.messages = [...config.messages];\n    this.loopDetector = new LoopDetector(config.loopDetectorConfig);\n  }\n\n  // ============================================================================\n  // TEMPLATE METHOD — the turn loop skeleton\n  // ============================================================================\n\n  async run(): Promise<AgentLoopResult> {\n    const maxTurns = this.config.maxTurns;\n    const maxCostUsd = this.config.maxCostUsd ?? DEFAULT_SESSION_COST_BUDGET_USD;\n    const maxContinuations = this.config.maxContinuations ?? AGENT_DEFAULTS.maxContinuations;\n\n    for (let iteration = 0; iteration < maxTurns; iteration++) {\n      // 1. Check abort (subclass: signal, disconnect, timeout)\n      if (this.checkAbort()) {\n        this.onAborted();\n        break;\n      }\n\n      // 2. Check cost budget\n      if (isFinite(maxCostUsd) && this.sessionCostUsd >= maxCostUsd) {\n        this.events.onText?.(`\\n[Cost budget exhausted ($${this.sessionCostUsd.toFixed(2)}/$${maxCostUsd.toFixed(2)}). Stopping.]`);\n        break;\n      }\n\n      // 3. Pre-compact messages (subclass can override for aggressive compaction)\n      this.messages = this.preCompactMessages(this.messages);\n\n      // 4. Execute turn — ABSTRACT (API call, streaming, response parsing)\n      let turnResult: TurnResult;\n      try {\n        turnResult = await this.executeTurn(this.messages, iteration);\n      } catch (err) {\n        const handled = await this.handleTurnError(err, iteration);\n        if (!handled) break;\n        continue;\n      }\n\n      // 5. Accumulate tokens\n      this.accumulateTokens(turnResult);\n\n      // 6. Emit cost warnings\n      emitCostWarningIfNeeded(\n        this.sessionCostUsd, maxCostUsd,\n        this.costWarningsEmitted, this.events.onText,\n      );\n\n      // 7. Subclass lifecycle hook\n      await this.onTurnComplete(turnResult, iteration);\n\n      // 8. Handle native compaction\n      if (turnResult.stopReason === \"compaction\" && turnResult.compactionContent) {\n        this.compactionCount++;\n        this.consecutiveCompactions++;\n\n        // Circuit breaker — too many compactions or budget exceeded\n        if (this.consecutiveCompactions >= MAX_CONSECUTIVE_COMPACTIONS) {\n          this.messages = [\n            { role: \"assistant\", content: turnResult.compactionContent },\n            { role: \"user\", content: WRAP_UP_PROMPT },\n          ];\n          this.consecutiveCompactions = 0;\n          iteration--; // Don't count compaction as a turn\n          continue;\n        }\n\n        // Normal compaction — replace messages with compacted summary\n        this.messages = [\n          { role: \"assistant\", content: turnResult.compactionContent },\n          { role: \"user\", content: CONTINUATION_PROMPT },\n        ];\n        await this.onMessagesUpdated(\"compaction\");\n        iteration--; // Don't count compaction as a turn\n        continue;\n      }\n\n      this.consecutiveCompactions = 0;\n\n      // 9. Handle max_tokens continuation\n      if (turnResult.stopReason === \"max_tokens\" && turnResult.toolUseBlocks.length === 0) {\n        this.continuationCount++;\n        if (this.continuationCount <= maxContinuations) {\n          // Build assistant message, then inject continuation\n          const assistantContent = this.buildAssistantContent(turnResult);\n          this.messages.push({ role: \"assistant\", content: assistantContent });\n          this.messages.push({ role: \"user\", content: CONTINUATION_PROMPT });\n          this.lastText = turnResult.text;\n          await this.onMessagesUpdated(\"continuation\");\n          continue;\n        }\n        // Exceeded max continuations — stop\n        this.lastText = turnResult.text;\n        break;\n      }\n\n      // 10. Record text\n      if (turnResult.text) {\n        this.lastText = turnResult.text;\n      }\n      this.lastStopReason = turnResult.stopReason;\n\n      // 11. No tools → done\n      if (turnResult.toolUseBlocks.length === 0) {\n        break;\n      }\n\n      // 12. Dispatch tools and append results\n      //     Subclass can override dispatchToolsAndAppend() entirely for custom dispatch.\n      const shouldContinue = await this.dispatchToolsAndAppend(turnResult);\n      if (!shouldContinue) break;\n    }\n\n    return this.buildResult();\n  }\n\n  // ============================================================================\n  // CONCRETE METHODS — shared behavior\n  // ============================================================================\n\n  protected accumulateTokens(turn: TurnResult): void {\n    this.totalInputTokens += turn.usage.inputTokens;\n    this.totalOutputTokens += turn.usage.outputTokens;\n    this.totalCacheRead += turn.usage.cacheRead || 0;\n    this.totalCacheCreation += turn.usage.cacheCreation || 0;\n    this.totalThinkingTokens += turn.usage.thinkingTokens || 0;\n    this.turnCount++;\n\n    this.sessionCostUsd = estimateCostUsd(\n      this.totalInputTokens, this.totalOutputTokens, this.config.model,\n      this.totalThinkingTokens, this.totalCacheRead, this.totalCacheCreation,\n    );\n  }\n\n  protected getToolDispatchOptions(): ToolDispatchOptions {\n    return {\n      loopDetector: this.loopDetector,\n      maxConcurrent: this.config.maxConcurrentTools ?? AGENT_DEFAULTS.maxConcurrentTools,\n      batchErrorLimit: this.config.batchErrorLimit ?? AGENT_DEFAULTS.loopBatchErrorLimit,\n      onStart: this.events.onToolStart,\n      onResult: this.events.onToolResult,\n    };\n  }\n\n  protected buildAssistantContent(turn: TurnResult): unknown {\n    const blocks: Array<Record<string, unknown>> = [];\n\n    // Thinking blocks first\n    if (turn.thinkingBlocks?.length) {\n      for (const tb of turn.thinkingBlocks) {\n        blocks.push({ type: \"thinking\", thinking: tb.thinking, signature: tb.signature });\n      }\n    }\n\n    // Text\n    if (turn.text) {\n      blocks.push({ type: \"text\", text: turn.text });\n    }\n\n    // Tool use blocks\n    for (const tu of turn.toolUseBlocks) {\n      blocks.push({ type: \"tool_use\", id: tu.id, name: tu.name, input: tu.input });\n    }\n\n    return blocks.length > 0 ? blocks : turn.text;\n  }\n\n  protected buildResult(): AgentLoopResult {\n    return {\n      finalText: this.lastText,\n      turnCount: this.turnCount,\n      tokens: {\n        input: this.totalInputTokens,\n        output: this.totalOutputTokens,\n        cacheRead: this.totalCacheRead,\n        cacheCreation: this.totalCacheCreation,\n      },\n      costUsd: this.sessionCostUsd,\n      toolsUsed: Array.from(this.toolsUsed),\n      toolCallCount: this.toolCallCount,\n      stopReason: this.lastStopReason,\n      loopDetectorStats: this.loopDetector.getSessionStats(),\n    };\n  }\n\n  // ============================================================================\n  // ABSTRACT METHODS — subclass must implement\n  // ============================================================================\n\n  /** Execute a single API turn and return normalized result. */\n  protected abstract executeTurn(messages: CompactableMessage[], iteration: number): Promise<TurnResult>;\n\n  /** Create a tool executor appropriate for this loop type. */\n  protected abstract createToolExecutor(): ToolExecutor;\n\n  // ============================================================================\n  // VIRTUAL METHODS — subclass may override\n  // ============================================================================\n\n  /** Check if the loop should abort (signal, disconnect, timeout). */\n  protected checkAbort(): boolean { return false; }\n\n  /** Called when abort is detected. */\n  protected onAborted(): void {}\n\n  /** Pre-compact messages before each turn. Subagent can be more aggressive. */\n  protected preCompactMessages(msgs: CompactableMessage[]): CompactableMessage[] {\n    return msgs;\n  }\n\n  /** Called after each completed turn. Subclass: telemetry, usage events. */\n  protected async onTurnComplete(_turn: TurnResult, _iteration: number): Promise<void> {}\n\n  /** Called after tool results are ready. CLI: shell summarization. */\n  protected async onToolResultsReady(\n    _toolBlocks: ToolUseBlock[], _results: ToolResultMessage[],\n  ): Promise<void> {}\n\n  /** Called after messages are updated. Server: checkpointing, persistence. */\n  protected async onMessagesUpdated(_reason: \"compaction\" | \"continuation\" | \"tool_results\"): Promise<void> {}\n\n  /** Handle a turn-level error. Return true if the loop should continue, false to break. */\n  protected async handleTurnError(_err: unknown, _iteration: number): Promise<boolean> { return false; }\n\n  /**\n   * Dispatch tools and append results to messages. Returns true to continue loop, false to break.\n   * Default implementation uses shared dispatchTools(). Subclass can override entirely for\n   * custom dispatch (e.g. SubagentLoop uses executeToolBlocks instead).\n   */\n  protected async dispatchToolsAndAppend(turnResult: TurnResult): Promise<boolean> {\n    const toolExecutor = this.createToolExecutor();\n    const dispatchOpts = this.getToolDispatchOptions();\n\n    // Record tool names\n    for (const tb of turnResult.toolUseBlocks) {\n      this.toolsUsed.add(tb.name);\n      this.toolCallCount++;\n    }\n\n    const { results: toolResults, bailOut, bailMessage } = await dispatchTools(\n      turnResult.toolUseBlocks, toolExecutor, dispatchOpts,\n    );\n\n    // Bail out from loop detector\n    if (bailOut) {\n      this.events.onText?.(bailMessage || \"\\n[Tool loop detected. Stopping.]\");\n      return false;\n    }\n\n    // End-of-turn loop detector check\n    const bailCheck = this.loopDetector.endTurn();\n    if (bailCheck.shouldBail) {\n      this.events.onText?.(bailCheck.message || \"\\n[Consecutive failures detected. Stopping.]\");\n      return false;\n    }\n\n    // Subclass hook for tool results\n    await this.onToolResultsReady(turnResult.toolUseBlocks, toolResults);\n\n    // Append messages\n    const assistantContent = this.buildAssistantContent(turnResult);\n    this.messages.push({ role: \"assistant\", content: assistantContent });\n    this.messages.push({\n      role: \"user\",\n      content: toolResults as unknown as CompactableMessage[\"content\"],\n    });\n    await this.onMessagesUpdated(\"tool_results\");\n    return true;\n  }\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAEA,SAASA,YAAY,EAAEC,eAAe,EAAEC,uBAAuB,EAAEC,+BAA+B,EAAEC,cAAc,QAAQ,iBAAiB;AAGzI,SAASC,aAAa,QAAqD,oBAAoB;;AAG/F;AACA;AACA;;AAqDA;AACA;AACA;;AAEA,MAAMC,2BAA2B,GAAG,CAAC;AACrC,MAAMC,mBAAmB,GAAG,0DAA0D;AACtF,MAAMC,cAAc,GAAG,yFAAyF;;AAEhH;AACA;AACA;;AAEA,OAAO,MAAeC,aAAa,CAAC;EAKlC;;EAEUC,SAAS,GAAG,CAAC;EACbC,gBAAgB,GAAG,CAAC;EACpBC,iBAAiB,GAAG,CAAC;EACrBC,cAAc,GAAG,CAAC;EAClBC,kBAAkB,GAAG,CAAC;EACtBC,mBAAmB,GAAG,CAAC;EACvBC,cAAc,GAAG,CAAC;EAClBC,SAAS,GAAG,IAAIC,GAAG,CAAS,CAAC;EAC7BC,aAAa,GAAG,CAAC;EACjBC,cAAc,GAAG,UAAU;EAC3BC,QAAQ,GAAG,EAAE;;EAEvB;EACUC,eAAe,GAAG,CAAC;EACnBC,sBAAsB,GAAG,CAAC;EAC1BC,iBAAiB,GAAG,CAAC;EACrBC,mBAAmB,GAAG,IAAIP,GAAG,CAAS,CAAC;EAEjDQ,WAAWA,CAACC,MAAuB,EAAEC,MAAuB,GAAG,CAAC,CAAC,EAAE;IACjE,IAAI,CAACD,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,QAAQ,GAAG,CAAC,GAAGF,MAAM,CAACE,QAAQ,CAAC;IACpC,IAAI,CAACC,YAAY,GAAG,IAAI9B,YAAY,CAAC2B,MAAM,CAACI,kBAAkB,CAAC;EACjE;;EAEA;EACA;EACA;;EAEA,MAAMC,GAAGA,CAAA,EAA6B;IACpC,MAAMC,QAAQ,GAAG,IAAI,CAACN,MAAM,CAACM,QAAQ;IACrC,MAAMC,UAAU,GAAG,IAAI,CAACP,MAAM,CAACO,UAAU,IAAI/B,+BAA+B;IAC5E,MAAMgC,gBAAgB,GAAG,IAAI,CAACR,MAAM,CAACQ,gBAAgB,IAAI/B,cAAc,CAAC+B,gBAAgB;IAExF,KAAK,IAAIC,SAAS,GAAG,CAAC,EAAEA,SAAS,GAAGH,QAAQ,EAAEG,SAAS,EAAE,EAAE;MACzD;MACA,IAAI,IAAI,CAACC,UAAU,CAAC,CAAC,EAAE;QACrB,IAAI,CAACC,SAAS,CAAC,CAAC;QAChB;MACF;;MAEA;MACA,IAAIC,QAAQ,CAACL,UAAU,CAAC,IAAI,IAAI,CAAClB,cAAc,IAAIkB,UAAU,EAAE;QAC7D,IAAI,CAACN,MAAM,CAACY,MAAM,GAAG,8BAA8B,IAAI,CAACxB,cAAc,CAACyB,OAAO,CAAC,CAAC,CAAC,KAAKP,UAAU,CAACO,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;QAC3H;MACF;;MAEA;MACA,IAAI,CAACZ,QAAQ,GAAG,IAAI,CAACa,kBAAkB,CAAC,IAAI,CAACb,QAAQ,CAAC;;MAEtD;MACA,IAAIc,UAAsB;MAC1B,IAAI;QACFA,UAAU,GAAG,MAAM,IAAI,CAACC,WAAW,CAAC,IAAI,CAACf,QAAQ,EAAEO,SAAS,CAAC;MAC/D,CAAC,CAAC,OAAOS,GAAG,EAAE;QACZ,MAAMC,OAAO,GAAG,MAAM,IAAI,CAACC,eAAe,CAACF,GAAG,EAAET,SAAS,CAAC;QAC1D,IAAI,CAACU,OAAO,EAAE;QACd;MACF;;MAEA;MACA,IAAI,CAACE,gBAAgB,CAACL,UAAU,CAAC;;MAEjC;MACAzC,uBAAuB,CACrB,IAAI,CAACc,cAAc,EAAEkB,UAAU,EAC/B,IAAI,CAACT,mBAAmB,EAAE,IAAI,CAACG,MAAM,CAACY,MACxC,CAAC;;MAED;MACA,MAAM,IAAI,CAACS,cAAc,CAACN,UAAU,EAAEP,SAAS,CAAC;;MAEhD;MACA,IAAIO,UAAU,CAACO,UAAU,KAAK,YAAY,IAAIP,UAAU,CAACQ,iBAAiB,EAAE;QAC1E,IAAI,CAAC7B,eAAe,EAAE;QACtB,IAAI,CAACC,sBAAsB,EAAE;;QAE7B;QACA,IAAI,IAAI,CAACA,sBAAsB,IAAIjB,2BAA2B,EAAE;UAC9D,IAAI,CAACuB,QAAQ,GAAG,CACd;YAAEuB,IAAI,EAAE,WAAW;YAAEC,OAAO,EAAEV,UAAU,CAACQ;UAAkB,CAAC,EAC5D;YAAEC,IAAI,EAAE,MAAM;YAAEC,OAAO,EAAE7C;UAAe,CAAC,CAC1C;UACD,IAAI,CAACe,sBAAsB,GAAG,CAAC;UAC/Ba,SAAS,EAAE,CAAC,CAAC;UACb;QACF;;QAEA;QACA,IAAI,CAACP,QAAQ,GAAG,CACd;UAAEuB,IAAI,EAAE,WAAW;UAAEC,OAAO,EAAEV,UAAU,CAACQ;QAAkB,CAAC,EAC5D;UAAEC,IAAI,EAAE,MAAM;UAAEC,OAAO,EAAE9C;QAAoB,CAAC,CAC/C;QACD,MAAM,IAAI,CAAC+C,iBAAiB,CAAC,YAAY,CAAC;QAC1ClB,SAAS,EAAE,CAAC,CAAC;QACb;MACF;MAEA,IAAI,CAACb,sBAAsB,GAAG,CAAC;;MAE/B;MACA,IAAIoB,UAAU,CAACO,UAAU,KAAK,YAAY,IAAIP,UAAU,CAACY,aAAa,CAACC,MAAM,KAAK,CAAC,EAAE;QACnF,IAAI,CAAChC,iBAAiB,EAAE;QACxB,IAAI,IAAI,CAACA,iBAAiB,IAAIW,gBAAgB,EAAE;UAC9C;UACA,MAAMsB,gBAAgB,GAAG,IAAI,CAACC,qBAAqB,CAACf,UAAU,CAAC;UAC/D,IAAI,CAACd,QAAQ,CAAC8B,IAAI,CAAC;YAAEP,IAAI,EAAE,WAAW;YAAEC,OAAO,EAAEI;UAAiB,CAAC,CAAC;UACpE,IAAI,CAAC5B,QAAQ,CAAC8B,IAAI,CAAC;YAAEP,IAAI,EAAE,MAAM;YAAEC,OAAO,EAAE9C;UAAoB,CAAC,CAAC;UAClE,IAAI,CAACc,QAAQ,GAAGsB,UAAU,CAACiB,IAAI;UAC/B,MAAM,IAAI,CAACN,iBAAiB,CAAC,cAAc,CAAC;UAC5C;QACF;QACA;QACA,IAAI,CAACjC,QAAQ,GAAGsB,UAAU,CAACiB,IAAI;QAC/B;MACF;;MAEA;MACA,IAAIjB,UAAU,CAACiB,IAAI,EAAE;QACnB,IAAI,CAACvC,QAAQ,GAAGsB,UAAU,CAACiB,IAAI;MACjC;MACA,IAAI,CAACxC,cAAc,GAAGuB,UAAU,CAACO,UAAU;;MAE3C;MACA,IAAIP,UAAU,CAACY,aAAa,CAACC,MAAM,KAAK,CAAC,EAAE;QACzC;MACF;;MAEA;MACA;MACA,MAAMK,cAAc,GAAG,MAAM,IAAI,CAACC,sBAAsB,CAACnB,UAAU,CAAC;MACpE,IAAI,CAACkB,cAAc,EAAE;IACvB;IAEA,OAAO,IAAI,CAACE,WAAW,CAAC,CAAC;EAC3B;;EAEA;EACA;EACA;;EAEUf,gBAAgBA,CAACgB,IAAgB,EAAQ;IACjD,IAAI,CAACrD,gBAAgB,IAAIqD,IAAI,CAACC,KAAK,CAACC,WAAW;IAC/C,IAAI,CAACtD,iBAAiB,IAAIoD,IAAI,CAACC,KAAK,CAACE,YAAY;IACjD,IAAI,CAACtD,cAAc,IAAImD,IAAI,CAACC,KAAK,CAACG,SAAS,IAAI,CAAC;IAChD,IAAI,CAACtD,kBAAkB,IAAIkD,IAAI,CAACC,KAAK,CAACI,aAAa,IAAI,CAAC;IACxD,IAAI,CAACtD,mBAAmB,IAAIiD,IAAI,CAACC,KAAK,CAACK,cAAc,IAAI,CAAC;IAC1D,IAAI,CAAC5D,SAAS,EAAE;IAEhB,IAAI,CAACM,cAAc,GAAGf,eAAe,CACnC,IAAI,CAACU,gBAAgB,EAAE,IAAI,CAACC,iBAAiB,EAAE,IAAI,CAACe,MAAM,CAAC4C,KAAK,EAChE,IAAI,CAACxD,mBAAmB,EAAE,IAAI,CAACF,cAAc,EAAE,IAAI,CAACC,kBACtD,CAAC;EACH;EAEU0D,sBAAsBA,CAAA,EAAwB;IACtD,OAAO;MACL1C,YAAY,EAAE,IAAI,CAACA,YAAY;MAC/B2C,aAAa,EAAE,IAAI,CAAC9C,MAAM,CAAC+C,kBAAkB,IAAItE,cAAc,CAACsE,kBAAkB;MAClFC,eAAe,EAAE,IAAI,CAAChD,MAAM,CAACgD,eAAe,IAAIvE,cAAc,CAACwE,mBAAmB;MAClFC,OAAO,EAAE,IAAI,CAACjD,MAAM,CAACkD,WAAW;MAChCC,QAAQ,EAAE,IAAI,CAACnD,MAAM,CAACoD;IACxB,CAAC;EACH;EAEUtB,qBAAqBA,CAACM,IAAgB,EAAW;IACzD,MAAMiB,MAAsC,GAAG,EAAE;;IAEjD;IACA,IAAIjB,IAAI,CAACkB,cAAc,EAAE1B,MAAM,EAAE;MAC/B,KAAK,MAAM2B,EAAE,IAAInB,IAAI,CAACkB,cAAc,EAAE;QACpCD,MAAM,CAACtB,IAAI,CAAC;UAAEyB,IAAI,EAAE,UAAU;UAAEC,QAAQ,EAAEF,EAAE,CAACE,QAAQ;UAAEC,SAAS,EAAEH,EAAE,CAACG;QAAU,CAAC,CAAC;MACnF;IACF;;IAEA;IACA,IAAItB,IAAI,CAACJ,IAAI,EAAE;MACbqB,MAAM,CAACtB,IAAI,CAAC;QAAEyB,IAAI,EAAE,MAAM;QAAExB,IAAI,EAAEI,IAAI,CAACJ;MAAK,CAAC,CAAC;IAChD;;IAEA;IACA,KAAK,MAAM2B,EAAE,IAAIvB,IAAI,CAACT,aAAa,EAAE;MACnC0B,MAAM,CAACtB,IAAI,CAAC;QAAEyB,IAAI,EAAE,UAAU;QAAEI,EAAE,EAAED,EAAE,CAACC,EAAE;QAAEC,IAAI,EAAEF,EAAE,CAACE,IAAI;QAAEC,KAAK,EAAEH,EAAE,CAACG;MAAM,CAAC,CAAC;IAC9E;IAEA,OAAOT,MAAM,CAACzB,MAAM,GAAG,CAAC,GAAGyB,MAAM,GAAGjB,IAAI,CAACJ,IAAI;EAC/C;EAEUG,WAAWA,CAAA,EAAoB;IACvC,OAAO;MACL4B,SAAS,EAAE,IAAI,CAACtE,QAAQ;MACxBX,SAAS,EAAE,IAAI,CAACA,SAAS;MACzBkF,MAAM,EAAE;QACNF,KAAK,EAAE,IAAI,CAAC/E,gBAAgB;QAC5BkF,MAAM,EAAE,IAAI,CAACjF,iBAAiB;QAC9BwD,SAAS,EAAE,IAAI,CAACvD,cAAc;QAC9BwD,aAAa,EAAE,IAAI,CAACvD;MACtB,CAAC;MACDgF,OAAO,EAAE,IAAI,CAAC9E,cAAc;MAC5BC,SAAS,EAAE8E,KAAK,CAACC,IAAI,CAAC,IAAI,CAAC/E,SAAS,CAAC;MACrCE,aAAa,EAAE,IAAI,CAACA,aAAa;MACjC+B,UAAU,EAAE,IAAI,CAAC9B,cAAc;MAC/B6E,iBAAiB,EAAE,IAAI,CAACnE,YAAY,CAACoE,eAAe,CAAC;IACvD,CAAC;EACH;;EAEA;EACA;EACA;;EAEA;;EAGA;;EAGA;EACA;EACA;;EAEA;EACU7D,UAAUA,CAAA,EAAY;IAAE,OAAO,KAAK;EAAE;;EAEhD;EACUC,SAASA,CAAA,EAAS,CAAC;;EAE7B;EACUI,kBAAkBA,CAACyD,IAA0B,EAAwB;IAC7E,OAAOA,IAAI;EACb;;EAEA;EACA,MAAgBlD,cAAcA,CAACmD,KAAiB,EAAEC,UAAkB,EAAiB,CAAC;;EAEtF;EACA,MAAgBC,kBAAkBA,CAChCC,WAA2B,EAAEC,QAA6B,EAC3C,CAAC;;EAElB;EACA,MAAgBlD,iBAAiBA,CAACmD,OAAuD,EAAiB,CAAC;;EAE3G;EACA,MAAgB1D,eAAeA,CAAC2D,IAAa,EAAEL,UAAkB,EAAoB;IAAE,OAAO,KAAK;EAAE;;EAErG;AACF;AACA;AACA;AACA;EACE,MAAgBvC,sBAAsBA,CAACnB,UAAsB,EAAoB;IAC/E,MAAMgE,YAAY,GAAG,IAAI,CAACC,kBAAkB,CAAC,CAAC;IAC9C,MAAMC,YAAY,GAAG,IAAI,CAACrC,sBAAsB,CAAC,CAAC;;IAElD;IACA,KAAK,MAAMW,EAAE,IAAIxC,UAAU,CAACY,aAAa,EAAE;MACzC,IAAI,CAACtC,SAAS,CAAC6F,GAAG,CAAC3B,EAAE,CAACM,IAAI,CAAC;MAC3B,IAAI,CAACtE,aAAa,EAAE;IACtB;IAEA,MAAM;MAAE4F,OAAO,EAAEC,WAAW;MAAEC,OAAO;MAAEC;IAAY,CAAC,GAAG,MAAM7G,aAAa,CACxEsC,UAAU,CAACY,aAAa,EAAEoD,YAAY,EAAEE,YAC1C,CAAC;;IAED;IACA,IAAII,OAAO,EAAE;MACX,IAAI,CAACrF,MAAM,CAACY,MAAM,GAAG0E,WAAW,IAAI,mCAAmC,CAAC;MACxE,OAAO,KAAK;IACd;;IAEA;IACA,MAAMC,SAAS,GAAG,IAAI,CAACrF,YAAY,CAACsF,OAAO,CAAC,CAAC;IAC7C,IAAID,SAAS,CAACE,UAAU,EAAE;MACxB,IAAI,CAACzF,MAAM,CAACY,MAAM,GAAG2E,SAAS,CAACG,OAAO,IAAI,8CAA8C,CAAC;MACzF,OAAO,KAAK;IACd;;IAEA;IACA,MAAM,IAAI,CAAChB,kBAAkB,CAAC3D,UAAU,CAACY,aAAa,EAAEyD,WAAW,CAAC;;IAEpE;IACA,MAAMvD,gBAAgB,GAAG,IAAI,CAACC,qBAAqB,CAACf,UAAU,CAAC;IAC/D,IAAI,CAACd,QAAQ,CAAC8B,IAAI,CAAC;MAAEP,IAAI,EAAE,WAAW;MAAEC,OAAO,EAAEI;IAAiB,CAAC,CAAC;IACpE,IAAI,CAAC5B,QAAQ,CAAC8B,IAAI,CAAC;MACjBP,IAAI,EAAE,MAAM;MACZC,OAAO,EAAE2D;IACX,CAAC,CAAC;IACF,MAAM,IAAI,CAAC1D,iBAAiB,CAAC,cAAc,CAAC;IAC5C,OAAO,IAAI;EACb;AACF","ignoreList":[]}

package/dist/shared/api-client-build-request.test.js

@@ -0,0 +1,228 @@
+import { describe, it, expect } from "vitest";
+import { buildAPIRequest } from "./api-client.js";
+
+// ============================================================================
+// buildAPIRequest
+// ============================================================================
+
+describe("buildAPIRequest", () => {
+  // ---------- main profile ----------
+
+  describe("main profile", () => {
+    it("returns compact + clear betas for opus-4-6", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main"
+      });
+      expect(config.betas).toContain("context-management-2025-06-27");
+      expect(config.betas).toContain("compact-2026-01-12");
+    });
+    it("returns compact + clear betas for sonnet-4-6", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "main"
+      });
+      expect(config.betas).toContain("context-management-2025-06-27");
+      expect(config.betas).toContain("compact-2026-01-12");
+    });
+    it("includes compact edit at 167K and clear edit at 100K for opus-4-6", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main"
+      });
+      const compactEdit = config.contextManagement.edits.find(e => e.type === "compact_20260112");
+      const clearEdit = config.contextManagement.edits.find(e => e.type === "clear_tool_uses_20250919");
+      expect(compactEdit).toBeDefined();
+      expect(compactEdit.trigger.value).toBe(167_000);
+      expect(clearEdit).toBeDefined();
+      expect(clearEdit.trigger.value).toBe(100_000);
+      expect(clearEdit.keep.value).toBe(3);
+    });
+    it("includes compact + clear_tool_uses edits for sonnet-4-6 (thinking disabled)", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "main"
+      });
+      // With thinking disabled, clear_thinking is filtered out, leaving compact + clear_tool_uses
+      expect(config.contextManagement.edits).toHaveLength(2);
+      expect(config.contextManagement.edits[0].type).toBe("compact_20260112");
+      expect(config.contextManagement.edits[1].type).toBe("clear_tool_uses_20250919");
+    });
+    it("defaults maxTokens to 32000", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main"
+      });
+      expect(config.maxTokens).toBe(32000);
+    });
+    it("respects maxOutputTokens override", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main",
+        maxOutputTokens: 4096
+      });
+      expect(config.maxTokens).toBe(4096);
+    });
+    it("caps maxOutputTokens at model maximum", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main",
+        maxOutputTokens: 200000
+      });
+      // 200000 exceeds opus model max of 128000, so capped
+      expect(config.maxTokens).toBe(128000);
+    });
+  });
+
+  // ---------- subagent profile ----------
+
+  describe("subagent profile", () => {
+    it("uses clear at 60K/keep 2 for Anthropic models", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "subagent"
+      });
+      // Subagent profile now includes clear_thinking + clear_tool_uses (+ compact for supported models)
+      expect(config.contextManagement.edits.length).toBeGreaterThanOrEqual(2);
+      const edit = config.contextManagement.edits.find(e => e.type === "clear_tool_uses_20250919");
+      expect(edit).toBeDefined();
+      expect(edit.trigger.value).toBe(60_000);
+      expect(edit.keep.value).toBe(2);
+    });
+    it("defaults maxTokens to 32000 (subagent cap)", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "subagent"
+      });
+      expect(config.maxTokens).toBe(32000);
+    });
+    it("allows explicit maxOutputTokens to override subagent default", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "subagent",
+        maxOutputTokens: 4096
+      });
+      expect(config.maxTokens).toBe(4096);
+    });
+    it("has context-management beta for Anthropic models", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "subagent"
+      });
+      expect(config.betas).toContain("context-management-2025-06-27");
+    });
+  });
+
+  // ---------- teammate profile ----------
+
+  describe("teammate profile", () => {
+    it("uses same context management as main for opus-4-6", () => {
+      const main = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main"
+      });
+      const teammate = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "teammate"
+      });
+      expect(teammate.betas).toEqual(main.betas);
+      expect(teammate.contextManagement.edits).toEqual(main.contextManagement.edits);
+    });
+    it("defaults maxTokens to 32000 (not subagent cap)", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "teammate"
+      });
+      expect(config.maxTokens).toBe(32000);
+    });
+  });
+
+  // ---------- thinking config ----------
+
+  describe("thinking integration", () => {
+    it("does not include thinking when disabled", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main",
+        thinkingEnabled: false
+      });
+      expect(config.thinking).toBeUndefined();
+    });
+    it("defaults thinkingEnabled to false", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main"
+      });
+      expect(config.thinking).toBeUndefined();
+    });
+    it("includes adaptive thinking beta for opus-4-6 when enabled", () => {
+      const config = buildAPIRequest({
+        model: "claude-opus-4-6",
+        contextProfile: "main",
+        thinkingEnabled: true
+      });
+      expect(config.betas).toContain("adaptive-thinking-2026-01-28");
+      expect(config.thinking).toBeDefined();
+      expect(config.thinking.type).toBe("adaptive");
+      expect(config.thinking.budget_tokens).toBeUndefined();
+    });
+    it("includes adaptive thinking beta for sonnet-4-6 when enabled", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-6",
+        contextProfile: "main",
+        thinkingEnabled: true
+      });
+      expect(config.betas).toContain("adaptive-thinking-2026-01-28");
+      expect(config.thinking).toBeDefined();
+      expect(config.thinking.type).toBe("adaptive");
+      expect(config.thinking.budget_tokens).toBeUndefined();
+    });
+    it("uses fixed budget thinking for older sonnet models", () => {
+      const config = buildAPIRequest({
+        model: "claude-sonnet-4-20250514",
+        contextProfile: "main",
+        thinkingEnabled: true
+      });
+      // Older sonnet uses fixed budget, not adaptive
+      expect(config.thinking).toBeDefined();
+      expect(config.thinking.type).toBe("enabled");
+      expect(config.thinking.budget_tokens).toBeDefined();
+      expect(config.thinking.budget_tokens).toBeLessThan(config.maxTokens);
+    });
+  });
+
+  // ---------- Gemini (non-Anthropic) models ----------
+
+  describe("non-Anthropic models (Gemini)", () => {
+    it("returns empty betas for gemini models on main profile", () => {
+      const config = buildAPIRequest({
+        model: "gemini-3-pro-preview",
+        contextProfile: "main"
+      });
+      expect(config.betas).toEqual([]);
+    });
+    it("returns empty edits for gemini models on main profile", () => {
+      const config = buildAPIRequest({
+        model: "gemini-3-pro-preview",
+        contextProfile: "main"
+      });
+      expect(config.contextManagement.edits).toEqual([]);
+    });
+    it("returns empty betas for gemini models on subagent profile", () => {
+      const config = buildAPIRequest({
+        model: "gemini-2.5-flash",
+        contextProfile: "subagent"
+      });
+      expect(config.betas).toEqual([]);
+      expect(config.contextManagement.edits).toEqual([]);
+    });
+    it("disables thinking for gemini models even when enabled", () => {
+      const config = buildAPIRequest({
+        model: "gemini-3-pro-preview",
+        contextProfile: "main",
+        thinkingEnabled: true
+      });
+      expect(config.thinking).toBeUndefined();
+    });
+  });
+});