tau-coding-agent 0.1.4 → 0.1.6

package/extensions/websearch/browser/firefox.ts

@@ -1,9 +1,10 @@
-import { existsSync, readFileSync } from "node:fs";
+import { readFile } from "node:fs/promises";
 import { DatabaseSync } from "node:sqlite";
 import os from "node:os";
 import path from "node:path";
 
 import type { BrowserCookie, BrowserProfile } from "../types.js";
+import { pathExists } from "./fs.js";
 import { withSqliteSnapshot } from "./sqlite.js";
 
 function firefoxBaseDir(): string | null {
@@ -45,7 +46,9 @@ function parseIni(text: string): Array<Record<string, string>> {
   return sections;
 }
 
-export function discoverFirefoxProfiles(preferredProfileName?: string): BrowserProfile[] {
+export async function discoverFirefoxProfiles(
+  preferredProfileName?: string,
+): Promise<BrowserProfile[]> {
   const baseDir = firefoxBaseDir();
   if (!baseDir) return [];
 
@@ -53,7 +56,7 @@ export function discoverFirefoxProfiles(preferredProfileName?: string): BrowserP
 
   let sections: Array<Record<string, string>> = [];
   try {
-    sections = parseIni(readFileSync(profilesIniPath, "utf8"));
+    sections = parseIni(await readFile(profilesIniPath, "utf8"));
   } catch {
     return [];
   }
@@ -67,7 +70,7 @@ export function discoverFirefoxProfiles(preferredProfileName?: string): BrowserP
     const profilePath =
       section.IsRelative === "1" ? path.join(baseDir, profilePathValue) : profilePathValue;
     const cookiesPath = path.join(profilePath, "cookies.sqlite");
-    if (!existsSync(cookiesPath)) continue;
+    if (!(await pathExists(cookiesPath))) continue;
 
     profiles.push({
       family: "firefox",
@@ -100,12 +103,12 @@ function isPreferredFirefoxProfile(
   );
 }
 
-export function loadFirefoxCookies(profile: BrowserProfile): BrowserCookie[] {
+export async function loadFirefoxCookies(profile: BrowserProfile): Promise<BrowserCookie[]> {
   const cookiesPath = path.join(profile.profilePath, "cookies.sqlite");
-  if (!existsSync(cookiesPath)) return [];
+  if (!(await pathExists(cookiesPath))) return [];
 
   try {
-    return withSqliteSnapshot(cookiesPath, "websearch-firefox", (tempDbPath) => {
+    return await withSqliteSnapshot(cookiesPath, "websearch-firefox", (tempDbPath) => {
       const db = new DatabaseSync(tempDbPath, { readOnly: true });
       try {
         const rows = db

package/extensions/websearch/browser/fs.ts

@@ -0,0 +1,10 @@
+import { access } from "node:fs/promises";
+
+export async function pathExists(filePath: string): Promise<boolean> {
+  try {
+    await access(filePath);
+    return true;
+  } catch {
+    return false;
+  }
+}

package/extensions/websearch/browser/sqlite.ts

@@ -1,33 +1,40 @@
-import { copyFileSync, existsSync, mkdirSync, rmSync } from "node:fs";
+import { copyFile, mkdtemp, rm } from "node:fs/promises";
 import os from "node:os";
 import path from "node:path";
 
-export function withSqliteSnapshot<T>(
+import { pathExists } from "./fs.js";
+
+export async function withSqliteSnapshot<T>(
   sourceDbPath: string,
   tempPrefix: string,
-  operation: (snapshotPath: string) => T,
-): T {
-  const tempDir = path.join(os.tmpdir(), `${tempPrefix}-${process.pid}-${Date.now()}`);
+  operation: (snapshotPath: string) => T | Promise<T>,
+): Promise<T> {
+  const tempDir = await mkdtemp(path.join(os.tmpdir(), `${tempPrefix}-${process.pid}-`));
   const snapshotPath = path.join(tempDir, path.basename(sourceDbPath));
 
   try {
-    mkdirSync(tempDir, { recursive: true });
-    copyFileSync(sourceDbPath, snapshotPath);
-    copySidecar(sourceDbPath, snapshotPath, "-wal");
-    copySidecar(sourceDbPath, snapshotPath, "-shm");
-    copySidecar(sourceDbPath, snapshotPath, "-journal");
-    return operation(snapshotPath);
+    await copyFile(sourceDbPath, snapshotPath);
+    await Promise.all([
+      copySidecar(sourceDbPath, snapshotPath, "-wal"),
+      copySidecar(sourceDbPath, snapshotPath, "-shm"),
+      copySidecar(sourceDbPath, snapshotPath, "-journal"),
+    ]);
+    return await operation(snapshotPath);
   } finally {
-    rmSync(tempDir, { recursive: true, force: true });
+    await rm(tempDir, { recursive: true, force: true });
   }
 }
 
-export function copySidecar(sourceDbPath: string, targetDbPath: string, suffix: string): void {
+export async function copySidecar(
+  sourceDbPath: string,
+  targetDbPath: string,
+  suffix: string,
+): Promise<void> {
   const source = `${sourceDbPath}${suffix}`;
-  if (!existsSync(source)) return;
+  if (!(await pathExists(source))) return;
 
   try {
-    copyFileSync(source, `${targetDbPath}${suffix}`);
+    await copyFile(source, `${targetDbPath}${suffix}`);
   } catch (error) {
     const code =
       error && typeof error === "object" && "code" in error

package/extensions/websearch/config.ts

@@ -1,10 +1,10 @@
 import { readFileSync } from "node:fs";
-import { homedir } from "node:os";
 import path from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
 
 import type { WebsearchConfig, WebsearchRouteId } from "./types.js";
 
-const WEBSEARCH_CONFIG_PATH = path.join(homedir(), ".pi", "websearch.json");
+const WEBSEARCH_CONFIG_PATH = path.join(getAgentDir(), "websearch.json");
 
 const DEFAULT_ROUTES: WebsearchRouteId[] = [
   "pi:openai-codex",

package/extensions/websearch/index.ts

@@ -128,15 +128,15 @@ function throwIfAborted(signal?: AbortSignal): void {
   throw signal.reason ?? new Error("The operation was aborted.");
 }
 
-function getBrowserProfiles(
+async function getBrowserProfiles(
   cache: Partial<Record<WebsearchBrowserFamily, BrowserProfile[]>>,
   family: WebsearchBrowserFamily,
   config: WebsearchConfig,
-): BrowserProfile[] {
+): Promise<BrowserProfile[]> {
   const cached = cache[family];
   if (cached) return cached;
 
-  const profiles = discoverProfiles([family], config.profiles);
+  const profiles = await discoverProfiles([family], config.profiles);
   cache[family] = profiles;
   return profiles;
 }
@@ -153,7 +153,7 @@ async function searchBrowserRoute(
     : "chromium";
   const browserProvider = route.endsWith(":gemini") ? browserGemini : browserOpenAICodex;
   const configuredProfileName = config.profiles[profileFamily];
-  const profiles = getBrowserProfiles(profileCache, profileFamily, config);
+  const profiles = await getBrowserProfiles(profileCache, profileFamily, config);
   let lastError: string | null = null;
 
   if (configuredProfileName && profiles.length === 0) {
@@ -164,7 +164,7 @@ async function searchBrowserRoute(
     throwIfAborted(signal);
 
     try {
-      const session = createBrowserSession(profile, browserProvider.domains);
+      const session = await createBrowserSession(profile, browserProvider.domains);
       if (!session) {
         if (configuredProfileName) {
           throw new Error(
@@ -265,7 +265,7 @@ export default function (pi: ExtensionAPI) {
       promptSnippet:
         "Search the web for current or external information unavailable in local files",
       promptGuidelines: [
-        "Use this for recent facts, live service behavior, or external documentation that is not already present in the repo.",
+        "Use websearch for recent facts, live service behavior, or external documentation that is not already present in the repo.",
         "Do not use websearch when repository files or supplied context already answer the question.",
       ],
       parameters: Type.Object({

package/extensions/websearch/providers/pi-model.shared.ts

@@ -5,6 +5,7 @@ export interface PiModelSelection {
   model: Model<Api>;
   apiKey?: string;
   headers?: Record<string, string>;
+  env?: Record<string, string>;
 }
 
 export async function selectCurrentPiModel(
@@ -42,7 +43,24 @@ async function resolvePiModelSelection(
   ctx: Pick<ExtensionContext, "modelRegistry">,
 ): Promise<PiModelSelection | null> {
   const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
-  return auth.ok ? { model, apiKey: auth.apiKey, headers: auth.headers } : null;
+  return auth.ok
+    ? {
+        model: resolveModelEnv(model, auth.env),
+        apiKey: auth.apiKey,
+        headers: auth.headers,
+        env: auth.env,
+      }
+    : null;
+}
+
+function resolveModelEnv(model: Model<Api>, env?: Record<string, string>): Model<Api> {
+  if (!model.baseUrl || !env) return model;
+
+  let baseUrl = model.baseUrl;
+  for (const [name, value] of Object.entries(env)) {
+    baseUrl = baseUrl.replaceAll(`{${name}}`, value);
+  }
+  return baseUrl === model.baseUrl ? model : { ...model, baseUrl };
 }
 
 function rankPiModels(models: Model<Api>[]): Model<Api>[] {

package/extensions/worktree.ts

@@ -40,9 +40,9 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 
 const STATUS_KEY = "0-worktree";
-const MANUAL_OPEN_MESSAGE_TYPE = "worktree-open-command";
-const RESTORE_STASH_MESSAGE_TYPE = "worktree-restore-command";
-const SCRIPT_RERUN_MESSAGE_TYPE = "worktree-script-rerun-command";
+const MANUAL_OPEN_ENTRY_TYPE = "worktree-open-command";
+const RESTORE_STASH_ENTRY_TYPE = "worktree-restore-command";
+const SCRIPT_RERUN_ENTRY_TYPE = "worktree-script-rerun-command";
 const MANUAL_OPEN_INTRO =
   "Worktree ready. Open it in a separate terminal or tmux pane with this command";
 
@@ -224,15 +224,10 @@ function showCommandMessage(
     return;
   }
 
-  pi.sendMessage({
-    customType,
-    content: intro,
-    display: true,
-    details: {
-      intro,
-      command,
-      copiedToClipboard,
-    } satisfies CommandMessageDetails,
+  pi.appendEntry<CommandMessageDetails>(customType, {
+    intro,
+    command,
+    copiedToClipboard,
   });
 }
 
@@ -241,7 +236,7 @@ function showManualOpenCommand(
   ctx: ExtensionCommandContext,
   command: string,
 ): void {
-  showCommandMessage(pi, ctx, MANUAL_OPEN_MESSAGE_TYPE, MANUAL_OPEN_INTRO, command);
+  showCommandMessage(pi, ctx, MANUAL_OPEN_ENTRY_TYPE, MANUAL_OPEN_INTRO, command);
 }
 
 function showRestoreStashCommand(
@@ -253,7 +248,7 @@ function showRestoreStashCommand(
   showCommandMessage(
     pi,
     ctx,
-    RESTORE_STASH_MESSAGE_TYPE,
+    RESTORE_STASH_ENTRY_TYPE,
     `Main-worktree changes were stashed as ${stashSpec}. Restore them in the new worktree with this command`,
     `cd ${shellQuoteCompact(targetPath)} && git stash apply ${shellQuote(stashSpec)}`,
   );
@@ -270,7 +265,7 @@ function showScriptRerunCommand(
   showCommandMessage(
     pi,
     ctx,
-    SCRIPT_RERUN_MESSAGE_TYPE,
+    SCRIPT_RERUN_ENTRY_TYPE,
     `${label} failed (exit ${exitCode}). Re-run it manually with this command`,
     `cd ${shellQuoteCompact(worktreeRoot)} && ${command}`,
   );
@@ -2008,18 +2003,23 @@ async function handleList(pi: ExtensionAPI, ctx: ExtensionCommandContext): Promi
     return { repo, items };
   });
 
-  if (!ctx.hasUI) {
-    for (const item of items) {
-      const dirty = item.status === "dirty" ? " *" : "";
-      const meta: string[] = [];
-      if (item.tracked) meta.push("↑");
-      if (item.isMain) meta.push("[primary]");
-      if (item.pathState !== "ok") meta.push(item.pathState);
-      if (item.wt.locked)
-        meta.push(item.wt.lockedReason ? `locked:${item.wt.lockedReason}` : "locked");
-      const suffix = meta.length > 0 ? `  ${meta.join("  ")}` : "";
-      console.log(`  ${item.branch}${dirty}${suffix}  ${collapsePath(item.wt.path)}`);
-    }
+  if (ctx.mode !== "tui") {
+    const output = items
+      .map((item) => {
+        const dirty = item.status === "dirty" ? " *" : "";
+        const meta: string[] = [];
+        if (item.tracked) meta.push("↑");
+        if (item.isMain) meta.push("[primary]");
+        if (item.pathState !== "ok") meta.push(item.pathState);
+        if (item.wt.locked)
+          meta.push(item.wt.lockedReason ? `locked:${item.wt.lockedReason}` : "locked");
+        const suffix = meta.length > 0 ? `  ${meta.join("  ")}` : "";
+        return `  ${item.branch}${dirty}${suffix}  ${collapsePath(item.wt.path)}`;
+      })
+      .join("\n");
+
+    if (ctx.mode === "print") console.log(output);
+    else ctx.ui.notify(output, "info");
     return;
   }
 
@@ -2114,14 +2114,14 @@ function parseSubcommand(args: string): { subcommand: Subcommand | null; rest: s
 
 export default function worktreeExtension(pi: ExtensionAPI) {
   for (const customType of [
-    MANUAL_OPEN_MESSAGE_TYPE,
-    RESTORE_STASH_MESSAGE_TYPE,
-    SCRIPT_RERUN_MESSAGE_TYPE,
+    MANUAL_OPEN_ENTRY_TYPE,
+    RESTORE_STASH_ENTRY_TYPE,
+    SCRIPT_RERUN_ENTRY_TYPE,
   ]) {
-    pi.registerMessageRenderer(customType, (message, _options, theme) => {
-      const details = message.details as CommandMessageDetails | undefined;
+    pi.registerEntryRenderer<CommandMessageDetails>(customType, (entry, _options, theme) => {
+      const details = entry.data;
       if (!details || typeof details.intro !== "string" || typeof details.command !== "string") {
-        return new Text(typeof message.content === "string" ? message.content : "", 0, 0);
+        return undefined;
       }
 
       const box = new Box(1, 1, (text) => theme.bg("customMessageBg", text));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tau-coding-agent",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "Opinionated Pi package with web search, review workflows, sandboxing, memory, usage reporting, skills, and a theme",
   "keywords": [
     "agent-tools",
@@ -27,7 +27,7 @@
   },
   "type": "module",
   "dependencies": {
-    "@anthropic-ai/sandbox-runtime": "^0.0.49",
+    "@anthropic-ai/sandbox-runtime": "^0.0.67",
     "@mozilla/readability": "^0.6.0",
     "jsdom": "^29.0.2",
     "puppeteer-core": "^24.40.0",
@@ -53,13 +53,13 @@
       "extensions/git-diff-stats.ts",
       "extensions/git-pr-status.ts",
       "extensions/insights.ts",
-      "extensions/interlude.ts",
+      "extensions/stash.ts",
       "extensions/loop.ts",
       "extensions/memory.ts",
       "extensions/notify.ts",
       "extensions/openai-fast.ts",
       "extensions/openai-verbosity.ts",
-      "extensions/review.ts",
+      "extensions/review/index.ts",
       "extensions/sandbox/index.ts",
       "extensions/tool-display-mode.ts",
       "extensions/usage/index.ts",

package/skills/git-clean-history/SKILL.md

@@ -12,6 +12,7 @@ Use this skill to reimplement the current branch on a new branch with a clean, n
 1. **Validate the source branch**
    - Ensure no uncommitted changes or merge conflicts
    - Confirm it is up to date with `main`
+   - Confirm `main` is up to date with `origin/main`
 
 2. **Analyze the diff**
    - Study all changes between source branch and `main`

package/skills/git-commit/SKILL.md

@@ -11,7 +11,7 @@ Use this skill as the playbook for producing reviewable commits and a clean, con
 
 1. Check what files have changed
 2. If there are no changes to commit, inform the user and stop
-3. If there are unstaged changes, stage them with `git add`
+3. If there are unstaged changes, stage relevant hunks or files with `git add -p` or `git add`
 
 Rules:
 
@@ -21,7 +21,7 @@ Rules:
 
 ## Phase 2: Generate commit message
 
-1. Remember commit messages best practices
+1. Remember commit message best practices
 2. Write a short subject line:
    - Up to 50 chars
    - Start subject with a capital letter, don’t end with a period

package/skills/oracle/SKILL.md

@@ -43,14 +43,14 @@ Run commands from this skill directory:
 ./scripts/oracle-bundle -p "<task>" --file "src/**" --file "!**/*.test.*"
 
 # Show which oracle model would be selected
-./scripts/oracle --list-models --current openai-codex/gpt-5.5
+./scripts/oracle --list-models --current openai-codex/gpt-5.6-sol
 
 # Ask the automatically selected oracle model
-./scripts/oracle --current openai-codex/gpt-5.5 \
+./scripts/oracle --current openai-codex/gpt-5.6-sol \
   -p "<task>" --file "src/**" --file "!**/*.test.*"
 
 # Override the oracle model when the automatic choice is wrong
-./scripts/oracle --model github-copilot/claude-opus-4.7 \
+./scripts/oracle --model anthropic/claude-fable-5 \
   -p "<task>" --file "src/**" --file "!**/*.test.*"
 ```
 

package/skills/oracle/scripts/oracle

@@ -1,12 +1,21 @@
 #!/usr/bin/env node
 import { spawnSync } from "node:child_process";
 import { existsSync, readFileSync } from "node:fs";
-import { homedir } from "node:os";
 import path from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
 import { fileURLToPath } from "node:url";
 
-const THINKING_LEVEL = "xhigh";
+const THINKING_LEVEL = "max";
 const SCRIPT_DIR = path.dirname(fileURLToPath(import.meta.url));
+const BARE_PI_ARGS = [
+  "-p",
+  "--no-session",
+  "--no-tools",
+  "--no-extensions",
+  "--no-skills",
+  "--no-prompt-templates",
+  "--no-themes",
+];
 
 const FAMILY_ORDER = {
   openai: ["anthropic", "google", "openai"],
@@ -47,11 +56,15 @@ if (bundle.status !== 0) {
   process.exit(bundle.status ?? 1);
 }
 
-const pi = spawnSync("pi", ["-p", "--no-tools", "--model", oracleModel, "--thinking", THINKING_LEVEL], {
-  cwd: repoRoot(),
-  input: bundle.stdout,
-  stdio: ["pipe", "inherit", "inherit"],
-});
+const pi = spawnSync(
+  "pi",
+  [...BARE_PI_ARGS, "--model", oracleModel, "--thinking", THINKING_LEVEL],
+  {
+    cwd: repoRoot(),
+    input: bundle.stdout,
+    stdio: ["pipe", "inherit", "inherit"],
+  },
+);
 
 if (pi.error) die(pi.error.message);
 process.exit(pi.status ?? 1);
@@ -75,7 +88,12 @@ function parseArgs(argv) {
     }
   }
 
-  return { bundleArgs, currentModel: currentModel ? normalizeModelSpec(currentModel) : undefined, model, listModels };
+  return {
+    bundleArgs,
+    currentModel: currentModel ? normalizeModelSpec(currentModel) : undefined,
+    model,
+    listModels,
+  };
 }
 
 function requireValue(argv, index, flag) {
@@ -85,8 +103,7 @@ function requireValue(argv, index, flag) {
 }
 
 function readSettings() {
-  const agentDir = process.env.PI_CODING_AGENT_DIR ?? path.join(homedir(), ".pi", "agent");
-  const settingsPath = path.join(agentDir, "settings.json");
+  const settingsPath = path.join(getAgentDir(), "settings.json");
   if (!existsSync(settingsPath)) return {};
 
   try {
@@ -127,82 +144,86 @@ function selectOracleModel(models, enabledModels, currentFamily) {
 
 function bestModelForFamily(models, enabledModels, family) {
   return models
-    .map((model) => ({ model, score: scoreModel(model, enabledModels, family) }))
-    .filter((entry) => entry.score >= 0)
-    .sort((a, b) => b.score - a.score || a.model.spec.localeCompare(b.model.spec))[0]?.model;
+    .filter((model) => modelFamily(model.spec) === family)
+    .sort((left, right) => compareModels(left, right, enabledModels, family))[0];
 }
 
-function scoreModel(entry, enabledModels, targetFamily) {
-  if (modelFamily(entry.spec) !== targetFamily) return -1;
-
-  let score = enabledModels.has(normalizeModelSpec(entry.spec)) ? 10_000 : 0;
-  score += providerScore(entry.provider, targetFamily);
-  score += nameScore(entry.model, targetFamily);
-  return score;
+function compareModels(left, right, enabledModels, family) {
+  return (
+    Number(enabledModels.has(normalizeModelSpec(right.spec))) -
+      Number(enabledModels.has(normalizeModelSpec(left.spec))) ||
+    modelRank(right.model, family) - modelRank(left.model, family) ||
+    providerRank(right.provider, family) - providerRank(left.provider, family) ||
+    left.spec.localeCompare(right.spec)
+  );
 }
 
-function providerScore(provider, family) {
+function providerRank(provider, family) {
   const normalized = provider.toLowerCase();
   if (family === "openai") {
-    if (normalized === "openai-codex") return 700;
-    if (normalized === "openai") return 600;
-    if (normalized === "github-copilot") return 500;
-    if (normalized.includes("azure")) return 450;
-    if (normalized === "openrouter") return 300;
-    return 100;
+    if (normalized === "openai-codex") return 5;
+    if (normalized === "openai") return 4;
+    if (normalized === "github-copilot") return 3;
+    if (normalized.includes("azure")) return 2;
+    if (normalized === "openrouter") return 1;
+    return 0;
   }
 
   if (family === "anthropic") {
-    if (normalized === "github-copilot") return 700;
-    if (normalized === "anthropic") return 600;
-    if (normalized === "amazon-bedrock") return 450;
-    if (normalized === "openrouter") return 300;
-    return 100;
+    if (normalized === "github-copilot") return 4;
+    if (normalized === "anthropic") return 3;
+    if (normalized === "amazon-bedrock") return 2;
+    if (normalized === "openrouter") return 1;
+    return 0;
   }
 
   if (family === "google") {
-    if (normalized === "google") return 600;
-    if (normalized === "github-copilot") return 500;
-    if (normalized.includes("vertex")) return 450;
-    if (normalized === "openrouter") return 300;
-    return 100;
+    if (normalized === "google") return 4;
+    if (normalized === "github-copilot") return 3;
+    if (normalized.includes("vertex")) return 2;
+    if (normalized === "openrouter") return 1;
+    return 0;
   }
 
   return 0;
 }
 
-function nameScore(model, family) {
+function modelRank(model, family) {
   const normalized = model.toLowerCase();
 
   if (family === "openai") {
-    if (/gpt-5\.5/.test(normalized)) return 1_000;
-    if (/gpt-5\.4/.test(normalized)) return 950;
-    if (/gpt-5\.3.*codex/.test(normalized)) return 925;
-    if (/gpt-5\.3/.test(normalized)) return 900;
-    if (/gpt-5\.2/.test(normalized)) return 850;
-    if (/gpt-5\.1/.test(normalized)) return 800;
-    if (/gpt-5/.test(normalized)) return 750;
-    if (/gpt-4\.1/.test(normalized)) return 500;
-    return 100;
+    if (/gpt-5\.6[-_.]sol/.test(normalized)) return 10;
+    if (/gpt-5\.6/.test(normalized)) return 9;
+    if (/gpt-5\.5/.test(normalized)) return 8;
+    if (/gpt-5\.4/.test(normalized)) return 7;
+    if (/gpt-5\.3.*codex/.test(normalized)) return 6;
+    if (/gpt-5\.3/.test(normalized)) return 5;
+    if (/gpt-5\.2/.test(normalized)) return 4;
+    if (/gpt-5\.1/.test(normalized)) return 3;
+    if (/gpt-5/.test(normalized)) return 2;
+    if (/gpt-4\.1/.test(normalized)) return 1;
+    return 0;
   }
 
   if (family === "anthropic") {
-    if (/claude[-_.]opus[-_.]4[-_.]7/.test(normalized)) return 1_000;
-    if (/claude[-_.]opus[-_.]4[-_.]6/.test(normalized)) return 950;
-    if (/claude[-_.]opus[-_.]4[-_.]5/.test(normalized)) return 900;
-    if (/claude[-_.]opus[-_.]4[-_.]1/.test(normalized)) return 850;
-    if (/claude[-_.]opus[-_.]4/.test(normalized)) return 800;
-    if (/claude[-_.]sonnet[-_.]4[-_.]6/.test(normalized)) return 700;
-    if (/claude[-_.]sonnet[-_.]4[-_.]5/.test(normalized)) return 650;
-    if (/claude[-_.]sonnet[-_.]4/.test(normalized)) return 600;
-    return 100;
+    if (/claude[-_.]fable[-_.]5/.test(normalized)) return 10;
+    if (/claude[-_.]opus[-_.]4[-_.]8/.test(normalized)) return 9;
+    if (/claude[-_.]opus[-_.]4[-_.]7/.test(normalized)) return 8;
+    if (/claude[-_.]opus[-_.]4[-_.]6/.test(normalized)) return 7;
+    if (/claude[-_.]opus[-_.]4[-_.]5/.test(normalized)) return 6;
+    if (/claude[-_.]opus[-_.]4[-_.]1/.test(normalized)) return 5;
+    if (/claude[-_.]opus[-_.]4/.test(normalized)) return 4;
+    if (/claude[-_.]sonnet[-_.]4[-_.]6/.test(normalized)) return 3;
+    if (/claude[-_.]sonnet[-_.]4[-_.]5/.test(normalized)) return 2;
+    if (/claude[-_.]sonnet[-_.]4/.test(normalized)) return 1;
+    return 0;
   }
 
   if (family === "google") {
-    if (/gemini[-_.]3\.1[-_.]pro/.test(normalized)) return 1_000;
-    if (/gemini[-_.]3[-_.]pro/.test(normalized)) return 950;
-    if (/gemini[-_.]2\.5[-_.]pro/.test(normalized)) return 800;
-    return 100;
+    if (/gemini[-_.]3\.1[-_.]pro/.test(normalized)) return 3;
+    if (/gemini[-_.]3[-_.]pro/.test(normalized)) return 2;
+    if (/gemini[-_.]2\.5[-_.]pro/.test(normalized)) return 1;
+    return 0;
   }
 
   return 0;
@@ -218,7 +239,7 @@ function modelFamily(spec) {
 }
 
 function normalizeModelSpec(spec) {
-  return String(spec).trim().replace(/:(off|minimal|low|medium|high|xhigh)$/i, "");
+  return String(spec).trim().replace(/:(off|minimal|low|medium|high|xhigh|max)$/i, "");
 }
 
 function printModelPlan(models, enabledModels, currentModel, currentFamily) {

package/themes/tau-dark.json

@@ -1,5 +1,5 @@
 {
-  "$schema": "https://raw.githubusercontent.com/badlogic/pi-mono/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
+  "$schema": "https://raw.githubusercontent.com/earendil-works/pi/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
   "name": "tau-dark",
   "vars": {
     "accent": "#8abeb7",
@@ -74,6 +74,7 @@
     "thinkingMedium": "#6f9bb3",
     "thinkingHigh": "#78a8a1",
     "thinkingXhigh": "accent",
+    "thinkingMax": "#a8d8d1",
 
     "bashMode": "green"
   },