whale-code 6.5.6 → 6.5.7

package/dist/cli/services/cli-agent-loop.d.ts

@@ -0,0 +1,41 @@
+/**
+ * CLIAgentLoop — AgentLoopBase subclass for CLI interactive agent execution.
+ *
+ * Handles streaming via callServerProxy + parseSSEStream, routes tools through
+ * interactive → local → server → MCP pipeline, and supports shell output summarization.
+ *
+ * Note: This is the v2 loop. The existing runAgentLoop() can delegate to this class
+ * for the core iteration logic while preserving its existing configuration wiring.
+ */
+import type Anthropic from "@anthropic-ai/sdk";
+import { AgentLoopBase, type AgentLoopConfig, type AgentLoopEvents, type TurnResult } from "../../shared/agent-loop-base.js";
+import type { CompactableMessage } from "../../shared/compaction.js";
+import type { ToolExecutor } from "../../shared/tool-dispatch.js";
+import type { ToolUseBlock, ToolResultMessage, StreamResult } from "../../shared/types.js";
+export interface CLILoopConfig extends AgentLoopConfig {
+    /** Function that calls API and returns a stream result. Injected by runAgentLoop. */
+    callAPI: (messages: CompactableMessage[], systemPrompt: string) => Promise<StreamResult>;
+    /** Tool executor injected by runAgentLoop (routes interactive → local → server → MCP). */
+    toolExecutor: ToolExecutor;
+    /** Abort signal for cancellation (ESC). */
+    abortSignal?: AbortSignal;
+    /** Shell output summarization function (injected). */
+    summarizeShellOutput?: (output: string) => Promise<string>;
+    /** Callbacks. */
+    onDone?: (finalMessages: Anthropic.MessageParam[]) => void;
+    onError?: (error: string, partialMessages?: Anthropic.MessageParam[]) => void;
+    onUsage?: (inputTokens: number, outputTokens: number, thinkingTokens?: number, model?: string, costUsd?: number, cacheRead?: number, cacheCreation?: number) => void;
+    onAutoCompact?: (before: number, after: number, tokensSaved: number) => void;
+}
+export declare class CLIAgentLoop extends AgentLoopBase {
+    private cliConfig;
+    constructor(config: CLILoopConfig, events?: AgentLoopEvents);
+    protected executeTurn(messages: CompactableMessage[], _iteration: number): Promise<TurnResult>;
+    protected createToolExecutor(): ToolExecutor;
+    protected checkAbort(): boolean;
+    protected onAborted(): void;
+    protected onToolResultsReady(_toolBlocks: ToolUseBlock[], results: ToolResultMessage[]): Promise<void>;
+    protected handleTurnError(err: unknown, iteration: number): Promise<boolean>;
+    /** Override run to call onDone at the end */
+    run(): Promise<import("../../shared/agent-loop-base.js").AgentLoopResult>;
+}

package/dist/cli/services/cli-agent-loop.js

@@ -0,0 +1,104 @@
+/**
+ * CLIAgentLoop — AgentLoopBase subclass for CLI interactive agent execution.
+ *
+ * Handles streaming via callServerProxy + parseSSEStream, routes tools through
+ * interactive → local → server → MCP pipeline, and supports shell output summarization.
+ *
+ * Note: This is the v2 loop. The existing runAgentLoop() can delegate to this class
+ * for the core iteration logic while preserving its existing configuration wiring.
+ */
+
+import { AgentLoopBase } from "../../shared/agent-loop-base.js";
+import { estimateCostUsd, categorizeError } from "../../shared/agent-core.js";
+
+// ============================================================================
+// TYPES
+// ============================================================================
+
+// ============================================================================
+// CLI AGENT LOOP
+// ============================================================================
+
+export class CLIAgentLoop extends AgentLoopBase {
+  constructor(config, events = {}) {
+    super(config, events);
+    this.cliConfig = config;
+  }
+
+  // ============================================================================
+  // ABSTRACT IMPLEMENTATIONS
+  // ============================================================================
+
+  async executeTurn(messages, _iteration) {
+    const result = await this.cliConfig.callAPI(messages, this.config.systemPrompt);
+
+    // Emit usage callback
+    const costUsd = estimateCostUsd(result.usage.inputTokens, result.usage.outputTokens, this.config.model, result.thinkingTokens || 0, result.usage.cacheReadTokens || 0, result.usage.cacheCreationTokens || 0);
+    this.cliConfig.onUsage?.(result.usage.inputTokens, result.usage.outputTokens, result.thinkingTokens, this.config.model, costUsd, result.usage.cacheReadTokens, result.usage.cacheCreationTokens);
+    return {
+      text: result.text || "",
+      toolUseBlocks: result.toolUseBlocks || [],
+      thinkingBlocks: result.thinkingBlocks || [],
+      compactionContent: result.compactionContent || null,
+      stopReason: result.stopReason || "end_turn",
+      usage: {
+        inputTokens: result.usage.inputTokens || 0,
+        outputTokens: result.usage.outputTokens || 0,
+        cacheRead: result.usage.cacheReadTokens || 0,
+        cacheCreation: result.usage.cacheCreationTokens || 0,
+        thinkingTokens: result.thinkingTokens || 0
+      }
+    };
+  }
+  createToolExecutor() {
+    return this.cliConfig.toolExecutor;
+  }
+
+  // ============================================================================
+  // VIRTUAL OVERRIDES
+  // ============================================================================
+
+  checkAbort() {
+    return this.cliConfig.abortSignal?.aborted === true;
+  }
+  onAborted() {
+    this.cliConfig.onError?.("Cancelled", this.messages);
+  }
+  async onToolResultsReady(_toolBlocks, results) {
+    // Shell output summarization for large outputs
+    if (this.cliConfig.summarizeShellOutput) {
+      for (const result of results) {
+        if (typeof result.content === "string" && result.content.length > 200_000) {
+          try {
+            result.content = await this.cliConfig.summarizeShellOutput(result.content);
+          } catch {
+            // Summarization failure is non-critical
+          }
+        }
+      }
+    }
+  }
+  async handleTurnError(err, iteration) {
+    const {
+      retryable
+    } = categorizeError(err);
+    if (retryable && iteration < 2) {
+      await new Promise(r => setTimeout(r, 1000 * Math.pow(2, iteration)));
+      return true;
+    }
+    this.cliConfig.onError?.(`API error: ${err.message}`, this.messages);
+    return false;
+  }
+
+  // ============================================================================
+  // RESULT
+  // ============================================================================
+
+  /** Override run to call onDone at the end */
+  async run() {
+    const result = await super.run();
+    this.cliConfig.onDone?.(this.messages);
+    return result;
+  }
+}
+//# sourceMappingURL=cli-agent-loop.js.map

package/dist/cli/services/cli-agent-loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli-agent-loop.js","names":["AgentLoopBase","estimateCostUsd","categorizeError","CLIAgentLoop","constructor","config","events","cliConfig","executeTurn","messages","_iteration","result","callAPI","systemPrompt","costUsd","usage","inputTokens","outputTokens","model","thinkingTokens","cacheReadTokens","cacheCreationTokens","onUsage","text","toolUseBlocks","thinkingBlocks","compactionContent","stopReason","cacheRead","cacheCreation","createToolExecutor","toolExecutor","checkAbort","abortSignal","aborted","onAborted","onError","onToolResultsReady","_toolBlocks","results","summarizeShellOutput","content","length","handleTurnError","err","iteration","retryable","Promise","r","setTimeout","Math","pow","message","run","onDone"],"sources":["../../../src/cli/services/cli-agent-loop.ts"],"sourcesContent":["/**\n * CLIAgentLoop — AgentLoopBase subclass for CLI interactive agent execution.\n *\n * Handles streaming via callServerProxy + parseSSEStream, routes tools through\n * interactive → local → server → MCP pipeline, and supports shell output summarization.\n *\n * Note: This is the v2 loop. The existing runAgentLoop() can delegate to this class\n * for the core iteration logic while preserving its existing configuration wiring.\n */\n\nimport type Anthropic from \"@anthropic-ai/sdk\";\nimport { AgentLoopBase, type AgentLoopConfig, type AgentLoopEvents, type TurnResult } from \"../../shared/agent-loop-base.js\";\nimport type { CompactableMessage } from \"../../shared/compaction.js\";\nimport { estimateCostUsd, categorizeError } from \"../../shared/agent-core.js\";\nimport type { ToolExecutor } from \"../../shared/tool-dispatch.js\";\nimport type { ToolUseBlock, ToolResultMessage, StreamResult } from \"../../shared/types.js\";\nimport { collectStreamResult } from \"../../shared/sse-parser.js\";\n\n// ============================================================================\n// TYPES\n// ============================================================================\n\nexport interface CLILoopConfig extends AgentLoopConfig {\n  /** Function that calls API and returns a stream result. Injected by runAgentLoop. */\n  callAPI: (messages: CompactableMessage[], systemPrompt: string) => Promise<StreamResult>;\n\n  /** Tool executor injected by runAgentLoop (routes interactive → local → server → MCP). */\n  toolExecutor: ToolExecutor;\n\n  /** Abort signal for cancellation (ESC). */\n  abortSignal?: AbortSignal;\n\n  /** Shell output summarization function (injected). */\n  summarizeShellOutput?: (output: string) => Promise<string>;\n\n  /** Callbacks. */\n  onDone?: (finalMessages: Anthropic.MessageParam[]) => void;\n  onError?: (error: string, partialMessages?: Anthropic.MessageParam[]) => void;\n  onUsage?: (inputTokens: number, outputTokens: number, thinkingTokens?: number, model?: string, costUsd?: number, cacheRead?: number, cacheCreation?: number) => void;\n  onAutoCompact?: (before: number, after: number, tokensSaved: number) => void;\n}\n\n// ============================================================================\n// CLI AGENT LOOP\n// ============================================================================\n\nexport class CLIAgentLoop extends AgentLoopBase {\n  private cliConfig: CLILoopConfig;\n\n  constructor(config: CLILoopConfig, events: AgentLoopEvents = {}) {\n    super(config, events);\n    this.cliConfig = config;\n  }\n\n  // ============================================================================\n  // ABSTRACT IMPLEMENTATIONS\n  // ============================================================================\n\n  protected async executeTurn(messages: CompactableMessage[], _iteration: number): Promise<TurnResult> {\n    const result = await this.cliConfig.callAPI(messages, this.config.systemPrompt);\n\n    // Emit usage callback\n    const costUsd = estimateCostUsd(\n      result.usage.inputTokens, result.usage.outputTokens, this.config.model,\n      result.thinkingTokens || 0, result.usage.cacheReadTokens || 0, result.usage.cacheCreationTokens || 0,\n    );\n    this.cliConfig.onUsage?.(\n      result.usage.inputTokens, result.usage.outputTokens,\n      result.thinkingTokens, this.config.model, costUsd,\n      result.usage.cacheReadTokens, result.usage.cacheCreationTokens,\n    );\n\n    return {\n      text: result.text || \"\",\n      toolUseBlocks: result.toolUseBlocks || [],\n      thinkingBlocks: result.thinkingBlocks || [],\n      compactionContent: result.compactionContent || null,\n      stopReason: result.stopReason || \"end_turn\",\n      usage: {\n        inputTokens: result.usage.inputTokens || 0,\n        outputTokens: result.usage.outputTokens || 0,\n        cacheRead: result.usage.cacheReadTokens || 0,\n        cacheCreation: result.usage.cacheCreationTokens || 0,\n        thinkingTokens: result.thinkingTokens || 0,\n      },\n    };\n  }\n\n  protected createToolExecutor(): ToolExecutor {\n    return this.cliConfig.toolExecutor;\n  }\n\n  // ============================================================================\n  // VIRTUAL OVERRIDES\n  // ============================================================================\n\n  protected checkAbort(): boolean {\n    return this.cliConfig.abortSignal?.aborted === true;\n  }\n\n  protected onAborted(): void {\n    this.cliConfig.onError?.(\"Cancelled\", this.messages as Anthropic.MessageParam[]);\n  }\n\n  protected async onToolResultsReady(_toolBlocks: ToolUseBlock[], results: ToolResultMessage[]): Promise<void> {\n    // Shell output summarization for large outputs\n    if (this.cliConfig.summarizeShellOutput) {\n      for (const result of results) {\n        if (typeof result.content === \"string\" && result.content.length > 200_000) {\n          try {\n            result.content = await this.cliConfig.summarizeShellOutput(result.content);\n          } catch {\n            // Summarization failure is non-critical\n          }\n        }\n      }\n    }\n  }\n\n  protected async handleTurnError(err: unknown, iteration: number): Promise<boolean> {\n    const { retryable } = categorizeError(err);\n    if (retryable && iteration < 2) {\n      await new Promise(r => setTimeout(r, 1000 * Math.pow(2, iteration)));\n      return true;\n    }\n    this.cliConfig.onError?.(\n      `API error: ${(err as Error).message}`,\n      this.messages as Anthropic.MessageParam[],\n    );\n    return false;\n  }\n\n  // ============================================================================\n  // RESULT\n  // ============================================================================\n\n  /** Override run to call onDone at the end */\n  async run() {\n    const result = await super.run();\n    this.cliConfig.onDone?.(this.messages as Anthropic.MessageParam[]);\n    return result;\n  }\n}\n"],"mappings":"AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;AAGA,SAASA,aAAa,QAAqE,iCAAiC;AAE5H,SAASC,eAAe,EAAEC,eAAe,QAAQ,4BAA4B;;AAK7E;AACA;AACA;;AAsBA;AACA;AACA;;AAEA,OAAO,MAAMC,YAAY,SAASH,aAAa,CAAC;EAG9CI,WAAWA,CAACC,MAAqB,EAAEC,MAAuB,GAAG,CAAC,CAAC,EAAE;IAC/D,KAAK,CAACD,MAAM,EAAEC,MAAM,CAAC;IACrB,IAAI,CAACC,SAAS,GAAGF,MAAM;EACzB;;EAEA;EACA;EACA;;EAEA,MAAgBG,WAAWA,CAACC,QAA8B,EAAEC,UAAkB,EAAuB;IACnG,MAAMC,MAAM,GAAG,MAAM,IAAI,CAACJ,SAAS,CAACK,OAAO,CAACH,QAAQ,EAAE,IAAI,CAACJ,MAAM,CAACQ,YAAY,CAAC;;IAE/E;IACA,MAAMC,OAAO,GAAGb,eAAe,CAC7BU,MAAM,CAACI,KAAK,CAACC,WAAW,EAAEL,MAAM,CAACI,KAAK,CAACE,YAAY,EAAE,IAAI,CAACZ,MAAM,CAACa,KAAK,EACtEP,MAAM,CAACQ,cAAc,IAAI,CAAC,EAAER,MAAM,CAACI,KAAK,CAACK,eAAe,IAAI,CAAC,EAAET,MAAM,CAACI,KAAK,CAACM,mBAAmB,IAAI,CACrG,CAAC;IACD,IAAI,CAACd,SAAS,CAACe,OAAO,GACpBX,MAAM,CAACI,KAAK,CAACC,WAAW,EAAEL,MAAM,CAACI,KAAK,CAACE,YAAY,EACnDN,MAAM,CAACQ,cAAc,EAAE,IAAI,CAACd,MAAM,CAACa,KAAK,EAAEJ,OAAO,EACjDH,MAAM,CAACI,KAAK,CAACK,eAAe,EAAET,MAAM,CAACI,KAAK,CAACM,mBAC7C,CAAC;IAED,OAAO;MACLE,IAAI,EAAEZ,MAAM,CAACY,IAAI,IAAI,EAAE;MACvBC,aAAa,EAAEb,MAAM,CAACa,aAAa,IAAI,EAAE;MACzCC,cAAc,EAAEd,MAAM,CAACc,cAAc,IAAI,EAAE;MAC3CC,iBAAiB,EAAEf,MAAM,CAACe,iBAAiB,IAAI,IAAI;MACnDC,UAAU,EAAEhB,MAAM,CAACgB,UAAU,IAAI,UAAU;MAC3CZ,KAAK,EAAE;QACLC,WAAW,EAAEL,MAAM,CAACI,KAAK,CAACC,WAAW,IAAI,CAAC;QAC1CC,YAAY,EAAEN,MAAM,CAACI,KAAK,CAACE,YAAY,IAAI,CAAC;QAC5CW,SAAS,EAAEjB,MAAM,CAACI,KAAK,CAACK,eAAe,IAAI,CAAC;QAC5CS,aAAa,EAAElB,MAAM,CAACI,KAAK,CAACM,mBAAmB,IAAI,CAAC;QACpDF,cAAc,EAAER,MAAM,CAACQ,cAAc,IAAI;MAC3C;IACF,CAAC;EACH;EAEUW,kBAAkBA,CAAA,EAAiB;IAC3C,OAAO,IAAI,CAACvB,SAAS,CAACwB,YAAY;EACpC;;EAEA;EACA;EACA;;EAEUC,UAAUA,CAAA,EAAY;IAC9B,OAAO,IAAI,CAACzB,SAAS,CAAC0B,WAAW,EAAEC,OAAO,KAAK,IAAI;EACrD;EAEUC,SAASA,CAAA,EAAS;IAC1B,IAAI,CAAC5B,SAAS,CAAC6B,OAAO,GAAG,WAAW,EAAE,IAAI,CAAC3B,QAAoC,CAAC;EAClF;EAEA,MAAgB4B,kBAAkBA,CAACC,WAA2B,EAAEC,OAA4B,EAAiB;IAC3G;IACA,IAAI,IAAI,CAAChC,SAAS,CAACiC,oBAAoB,EAAE;MACvC,KAAK,MAAM7B,MAAM,IAAI4B,OAAO,EAAE;QAC5B,IAAI,OAAO5B,MAAM,CAAC8B,OAAO,KAAK,QAAQ,IAAI9B,MAAM,CAAC8B,OAAO,CAACC,MAAM,GAAG,OAAO,EAAE;UACzE,IAAI;YACF/B,MAAM,CAAC8B,OAAO,GAAG,MAAM,IAAI,CAAClC,SAAS,CAACiC,oBAAoB,CAAC7B,MAAM,CAAC8B,OAAO,CAAC;UAC5E,CAAC,CAAC,MAAM;YACN;UAAA;QAEJ;MACF;IACF;EACF;EAEA,MAAgBE,eAAeA,CAACC,GAAY,EAAEC,SAAiB,EAAoB;IACjF,MAAM;MAAEC;IAAU,CAAC,GAAG5C,eAAe,CAAC0C,GAAG,CAAC;IAC1C,IAAIE,SAAS,IAAID,SAAS,GAAG,CAAC,EAAE;MAC9B,MAAM,IAAIE,OAAO,CAACC,CAAC,IAAIC,UAAU,CAACD,CAAC,EAAE,IAAI,GAAGE,IAAI,CAACC,GAAG,CAAC,CAAC,EAAEN,SAAS,CAAC,CAAC,CAAC;MACpE,OAAO,IAAI;IACb;IACA,IAAI,CAACtC,SAAS,CAAC6B,OAAO,GACpB,cAAeQ,GAAG,CAAWQ,OAAO,EAAE,EACtC,IAAI,CAAC3C,QACP,CAAC;IACD,OAAO,KAAK;EACd;;EAEA;EACA;EACA;;EAEA;EACA,MAAM4C,GAAGA,CAAA,EAAG;IACV,MAAM1C,MAAM,GAAG,MAAM,KAAK,CAAC0C,GAAG,CAAC,CAAC;IAChC,IAAI,CAAC9C,SAAS,CAAC+C,MAAM,GAAG,IAAI,CAAC7C,QAAoC,CAAC;IAClE,OAAOE,MAAM;EACf;AACF","ignoreList":[]}

package/dist/cli/services/config-modules-model.test.js

@@ -0,0 +1,133 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { setModel, setModelById, getModel, getModelShortName, estimateCostUsd } from "./model-manager.js";
+describe("model-manager", () => {
+  beforeEach(() => {
+    setModelById("claude-opus-4-6");
+  });
+  describe("getModel / setModelById", () => {
+    it("defaults to claude-opus-4-6", () => {
+      expect(getModel()).toBe("claude-opus-4-6");
+    });
+    it("sets model by full ID", () => {
+      setModelById("claude-sonnet-4-6");
+      expect(getModel()).toBe("claude-sonnet-4-6");
+    });
+    it("accepts arbitrary model IDs (no validation)", () => {
+      setModelById("some-custom-model-id");
+      expect(getModel()).toBe("some-custom-model-id");
+    });
+  });
+  describe("setModel — alias resolution", () => {
+    it("resolves 'opus' to claude-opus-4-6", () => {
+      const result = setModel("opus");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("claude-opus-4-6");
+      expect(getModel()).toBe("claude-opus-4-6");
+    });
+    it("resolves 'sonnet' to claude-sonnet-4-6", () => {
+      const result = setModel("sonnet");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("claude-sonnet-4-6");
+    });
+    it("resolves 'haiku' to claude-haiku-4-5-20251001", () => {
+      const result = setModel("haiku");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("claude-haiku-4-5-20251001");
+    });
+    it("resolves bedrock aliases", () => {
+      const result = setModel("bedrock-sonnet");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("anthropic.claude-sonnet-4-6");
+    });
+    it("resolves gemini aliases", () => {
+      const result = setModel("gemini-pro");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("gemini-2.5-pro");
+    });
+    it("resolves gemini-flash alias", () => {
+      const result = setModel("gemini-flash");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("gemini-2.5-flash");
+    });
+    it("is case-insensitive for aliases", () => {
+      const result = setModel("OPUS");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("claude-opus-4-6");
+    });
+    it("handles mixed case", () => {
+      const result = setModel("Sonnet");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("claude-sonnet-4-6");
+    });
+    it("handles legacy 'claude-opus' prefix stripping", () => {
+      const result = setModel("claude-opus");
+      expect(result.success).toBe(true);
+      expect(result.model).toBe("claude-opus-4-6");
+    });
+    it("returns error for unknown model name", () => {
+      const result = setModel("gpt-4");
+      expect(result.success).toBe(false);
+      expect(result.error).toContain("Unknown model");
+      expect(result.error).toContain("gpt-4");
+      expect(result.model).toBe("claude-opus-4-6");
+    });
+    it("returns error for empty string", () => {
+      const result = setModel("");
+      expect(result.success).toBe(false);
+      expect(result.error).toContain("Unknown model");
+    });
+    it("lists valid model keys in error message", () => {
+      const result = setModel("nonexistent");
+      expect(result.error).toContain("opus");
+      expect(result.error).toContain("sonnet");
+      expect(result.error).toContain("haiku");
+    });
+  });
+  describe("getModelShortName", () => {
+    it("returns 'opus' for claude-opus-4-6", () => {
+      setModelById("claude-opus-4-6");
+      expect(getModelShortName()).toBe("opus");
+    });
+    it("returns 'sonnet-4.6' for claude-sonnet-4-6", () => {
+      setModelById("claude-sonnet-4-6");
+      expect(getModelShortName()).toBe("sonnet-4.6");
+    });
+    it("returns 'haiku' for claude-haiku-4-5-20251001", () => {
+      setModelById("claude-haiku-4-5-20251001");
+      expect(getModelShortName()).toBe("haiku");
+    });
+    it("falls back to 'sonnet' for unknown model IDs", () => {
+      setModelById("some-unknown-model");
+      expect(getModelShortName()).toBe("sonnet");
+    });
+    it("returns correct short name for gemini models", () => {
+      setModelById("gemini-2.5-pro");
+      expect(getModelShortName()).toBe("gemini-pro");
+    });
+  });
+  describe("estimateCostUsd", () => {
+    it("uses the active model when none specified", () => {
+      setModelById("claude-opus-4-6");
+      const cost = estimateCostUsd(1_000_000, 0);
+      expect(cost).toBeCloseTo(5.0);
+    });
+    it("uses the specified model when provided", () => {
+      setModelById("claude-opus-4-6");
+      const cost = estimateCostUsd(1_000_000, 0, "claude-sonnet-4-6");
+      expect(cost).toBeCloseTo(3.0);
+    });
+    it("includes thinking tokens", () => {
+      setModelById("claude-opus-4-6");
+      const cost = estimateCostUsd(0, 0, undefined, 1_000_000);
+      expect(cost).toBeCloseTo(25.0);
+    });
+    it("returns 0 for zero tokens", () => {
+      expect(estimateCostUsd(0, 0)).toBe(0);
+    });
+    it("calculates combined input + output cost", () => {
+      setModelById("claude-opus-4-6");
+      const cost = estimateCostUsd(1_000_000, 1_000_000);
+      expect(cost).toBeCloseTo(30.0);
+    });
+  });
+});

package/dist/cli/services/config-modules-permission.test.js

@@ -0,0 +1,85 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { setPermissionMode, getPermissionMode, isToolAllowedByPermission } from "./permission-modes.js";
+describe("permission-modes", () => {
+  beforeEach(() => {
+    setPermissionMode("default");
+  });
+  describe("getPermissionMode / setPermissionMode", () => {
+    it("defaults to 'default' mode", () => {
+      expect(getPermissionMode()).toBe("default");
+    });
+    it("sets mode to 'plan'", () => {
+      const result = setPermissionMode("plan");
+      expect(result.success).toBe(true);
+      expect(result.message).toContain("plan");
+      expect(getPermissionMode()).toBe("plan");
+    });
+    it("sets mode to 'yolo'", () => {
+      const result = setPermissionMode("yolo");
+      expect(result.success).toBe(true);
+      expect(result.message).toContain("yolo");
+      expect(getPermissionMode()).toBe("yolo");
+    });
+    it("sets mode back to 'default'", () => {
+      setPermissionMode("yolo");
+      const result = setPermissionMode("default");
+      expect(result.success).toBe(true);
+      expect(getPermissionMode()).toBe("default");
+    });
+    it("returns success:true for all valid modes", () => {
+      const modes = ["default", "plan", "yolo"];
+      for (const mode of modes) {
+        const result = setPermissionMode(mode);
+        expect(result.success).toBe(true);
+      }
+    });
+    it("includes the mode name in the result message", () => {
+      const result = setPermissionMode("plan");
+      expect(result.message).toBe("Permission mode: plan");
+    });
+  });
+  describe("isToolAllowedByPermission", () => {
+    it("allows all tools in default mode", () => {
+      setPermissionMode("default");
+      expect(isToolAllowedByPermission("bash")).toBe(true);
+      expect(isToolAllowedByPermission("write_file")).toBe(true);
+      expect(isToolAllowedByPermission("edit_file")).toBe(true);
+      expect(isToolAllowedByPermission("read_file")).toBe(true);
+    });
+    it("allows all tools in yolo mode", () => {
+      setPermissionMode("yolo");
+      expect(isToolAllowedByPermission("bash")).toBe(true);
+      expect(isToolAllowedByPermission("write_file")).toBe(true);
+      expect(isToolAllowedByPermission("delete_file")).toBe(true);
+      expect(isToolAllowedByPermission("anything_at_all")).toBe(true);
+    });
+    it("allows read-only tools in plan mode", () => {
+      setPermissionMode("plan");
+      const allowedTools = ["read_file", "list_directory", "search_files", "search_content", "glob", "grep", "web_fetch", "web_search", "task", "task_output", "bash_output", "list_shells", "tasks", "config", "ask_user", "ask_user_question", "enter_plan_mode", "exit_plan_mode"];
+      for (const tool of allowedTools) {
+        expect(isToolAllowedByPermission(tool)).toBe(true);
+      }
+    });
+    it("blocks write/mutation tools in plan mode", () => {
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("bash")).toBe(false);
+      expect(isToolAllowedByPermission("write_file")).toBe(false);
+      expect(isToolAllowedByPermission("edit_file")).toBe(false);
+      expect(isToolAllowedByPermission("multi_edit")).toBe(false);
+      expect(isToolAllowedByPermission("delete_file")).toBe(false);
+    });
+    it("blocks unknown tools in plan mode", () => {
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("some_random_tool")).toBe(false);
+      expect(isToolAllowedByPermission("")).toBe(false);
+    });
+    it("reflects mode changes immediately", () => {
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("bash")).toBe(false);
+      setPermissionMode("yolo");
+      expect(isToolAllowedByPermission("bash")).toBe(true);
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("bash")).toBe(false);
+    });
+  });
+});

package/dist/cli/services/config-modules-permissions.test.js

@@ -0,0 +1,85 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { setPermissionMode, getPermissionMode, isToolAllowedByPermission } from "./permission-modes.js";
+describe("permission-modes", () => {
+  beforeEach(() => {
+    setPermissionMode("default");
+  });
+  describe("getPermissionMode / setPermissionMode", () => {
+    it("defaults to 'default' mode", () => {
+      expect(getPermissionMode()).toBe("default");
+    });
+    it("sets mode to 'plan'", () => {
+      const result = setPermissionMode("plan");
+      expect(result.success).toBe(true);
+      expect(result.message).toContain("plan");
+      expect(getPermissionMode()).toBe("plan");
+    });
+    it("sets mode to 'yolo'", () => {
+      const result = setPermissionMode("yolo");
+      expect(result.success).toBe(true);
+      expect(result.message).toContain("yolo");
+      expect(getPermissionMode()).toBe("yolo");
+    });
+    it("sets mode back to 'default'", () => {
+      setPermissionMode("yolo");
+      const result = setPermissionMode("default");
+      expect(result.success).toBe(true);
+      expect(getPermissionMode()).toBe("default");
+    });
+    it("returns success:true for all valid modes", () => {
+      const modes = ["default", "plan", "yolo"];
+      for (const mode of modes) {
+        const result = setPermissionMode(mode);
+        expect(result.success).toBe(true);
+      }
+    });
+    it("includes the mode name in the result message", () => {
+      const result = setPermissionMode("plan");
+      expect(result.message).toBe("Permission mode: plan");
+    });
+  });
+  describe("isToolAllowedByPermission", () => {
+    it("allows all tools in default mode", () => {
+      setPermissionMode("default");
+      expect(isToolAllowedByPermission("bash")).toBe(true);
+      expect(isToolAllowedByPermission("write_file")).toBe(true);
+      expect(isToolAllowedByPermission("edit_file")).toBe(true);
+      expect(isToolAllowedByPermission("read_file")).toBe(true);
+    });
+    it("allows all tools in yolo mode", () => {
+      setPermissionMode("yolo");
+      expect(isToolAllowedByPermission("bash")).toBe(true);
+      expect(isToolAllowedByPermission("write_file")).toBe(true);
+      expect(isToolAllowedByPermission("delete_file")).toBe(true);
+      expect(isToolAllowedByPermission("anything_at_all")).toBe(true);
+    });
+    it("allows read-only tools in plan mode", () => {
+      setPermissionMode("plan");
+      const allowedTools = ["read_file", "list_directory", "search_files", "search_content", "glob", "grep", "web_fetch", "web_search", "task", "task_output", "bash_output", "list_shells", "tasks", "config", "ask_user", "ask_user_question", "enter_plan_mode", "exit_plan_mode"];
+      for (const tool of allowedTools) {
+        expect(isToolAllowedByPermission(tool)).toBe(true);
+      }
+    });
+    it("blocks write/mutation tools in plan mode", () => {
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("bash")).toBe(false);
+      expect(isToolAllowedByPermission("write_file")).toBe(false);
+      expect(isToolAllowedByPermission("edit_file")).toBe(false);
+      expect(isToolAllowedByPermission("multi_edit")).toBe(false);
+      expect(isToolAllowedByPermission("delete_file")).toBe(false);
+    });
+    it("blocks unknown tools in plan mode", () => {
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("some_random_tool")).toBe(false);
+      expect(isToolAllowedByPermission("")).toBe(false);
+    });
+    it("reflects mode changes immediately", () => {
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("bash")).toBe(false);
+      setPermissionMode("yolo");
+      expect(isToolAllowedByPermission("bash")).toBe(true);
+      setPermissionMode("plan");
+      expect(isToolAllowedByPermission("bash")).toBe(false);
+    });
+  });
+});