@wrongstack/cli 0.51.3 → 0.54.1

package/dist/index.js

@@ -2,7 +2,7 @@
 import * as path8 from 'path';
 import { join } from 'path';
 import * as fsp3 from 'fs/promises';
-import { color, writeErr, DefaultTaskStore, TaskTracker, renderProgress, SpecStore, TaskGraphStore, analyzeCriticalPath, getTemplate, listTemplates, templateToMarkdown, SpecParser, renderSpecAnalysis, AISpecBuilder, renderTaskGraph, SpecVersioning, atomicWrite, DefaultPathResolver, TOKENS, DefaultSystemPromptBuilder, makeAutonomyPromptContributor, ToolRegistry, createContextManagerTool, EventBus, SlashCommandRegistry, BrainDecisionQueue, ObservableBrainArbiter, HumanEscalatingBrainArbiter, DefaultBrainArbiter, createDelegateTool, FLEET_ROSTER, createMcpControlTool, DefaultLogger, DefaultModelsRegistry, isStdinTTY, writeOut, runProviderWithRetry, ReplayLogStore, ReplayProviderRunner, ProviderRegistry, InMemoryMetricsSink, wireMetricsToEvents, DefaultHealthRegistry, startMetricsServer, RecoveryLock, DefaultAttachmentStore, QueueStore, Context, loadTodosCheckpoint, attachTodosCheckpoint, loadDirectorState, loadPlan, createDefaultPipelines, AutoCompactionMiddleware, estimateRequestTokensCalibrated, Agent, loadPlugins, FleetManager, makeDirectorSessionFactory, Director, makeFleetEmitTool, makeFleetStatusTool, resolveModelMatrix, AutoApprovePermissionPolicy, PhaseStore, AutoPhasePlanner, PhaseGraphBuilder, WorktreeManager, PhaseOrchestrator, makeLLMClassifier, ParallelEternalEngine, EternalAutonomyEngine, allServers as allServers$1, bootConfig as bootConfig$1, setRawMode, DefaultSessionReader, resolveWstackPaths, ToolAuditLog, DefaultSessionRewinder, DefaultSessionStore, DefaultPluginAPI, makeAgentSubagentRunner, NULL_FLEET_BUS, buildChildEnv, formatContextWindowModeList, repairToolUseAdjacency, getContextWindowMode, resolveContextWindowPolicy, AGENTS_BY_PHASE, dispatchAgent, formatTodosList, SessionRecovery, loadGoal, goalFilePath, summarizeUsage, saveGoal, emptyGoal, buildGoalPreamble, formatGoal, pendingBtwCount, setBtwNote, MATRIX_PHASE_KEYS, AGENT_CATALOG, matrixKeyKind, onResize, decryptConfigSecrets as decryptConfigSecrets$1, encryptConfigSecrets as encryptConfigSecrets$1, InputBuilder, FsError, ERROR_CODES } from '@wrongstack/core';
+import { color, writeErr, DefaultTaskStore, TaskTracker, renderProgress, SpecStore, TaskGraphStore, analyzeCriticalPath, getTemplate, listTemplates, templateToMarkdown, SpecParser, renderSpecAnalysis, AISpecBuilder, renderTaskGraph, SpecVersioning, DefaultSecretScrubber, atomicWrite, DefaultPathResolver, TOKENS, mergeCustomModelDefs, DefaultSystemPromptBuilder, makeAutonomyPromptContributor, ToolRegistry, createContextManagerTool, EventBus, SlashCommandRegistry, BrainDecisionQueue, ObservableBrainArbiter, HumanEscalatingBrainArbiter, DefaultBrainArbiter, createDelegateTool, FLEET_ROSTER, createMcpControlTool, DefaultLogger, DefaultModelsRegistry, isStdinTTY, writeOut, runProviderWithRetry, ReplayLogStore, ReplayProviderRunner, ProviderRegistry, InMemoryMetricsSink, wireMetricsToEvents, DefaultHealthRegistry, startMetricsServer, RecoveryLock, DefaultAttachmentStore, QueueStore, Context, loadTodosCheckpoint, attachTodosCheckpoint, loadDirectorState, loadPlan, createDefaultPipelines, AutoCompactionMiddleware, estimateRequestTokensCalibrated, Agent, loadPlugins, FleetManager, makeDirectorSessionFactory, Director, makeFleetEmitTool, makeFleetStatusTool, resolveModelMatrix, AutoApprovePermissionPolicy, PhaseStore, AutoPhasePlanner, PhaseGraphBuilder, WorktreeManager, PhaseOrchestrator, makeLLMClassifier, ParallelEternalEngine, EternalAutonomyEngine, allServers as allServers$1, bootConfig as bootConfig$1, setRawMode, DefaultSessionReader, resolveWstackPaths, ToolAuditLog, DefaultSessionRewinder, DefaultSessionStore, DefaultPluginAPI, makeAgentSubagentRunner, NULL_FLEET_BUS, buildChildEnv, formatContextWindowModeList, repairToolUseAdjacency, getContextWindowMode, resolveContextWindowPolicy, AGENTS_BY_PHASE, dispatchAgent, formatTodosList, SessionRecovery, loadGoal, goalFilePath, summarizeUsage, saveGoal, emptyGoal, buildGoalPreamble, formatGoal, pendingBtwCount, setBtwNote, MATRIX_PHASE_KEYS, AGENT_CATALOG, matrixKeyKind, onResize, decryptConfigSecrets as decryptConfigSecrets$1, encryptConfigSecrets as encryptConfigSecrets$1, InputBuilder, FsError, ERROR_CODES } from '@wrongstack/core';
 import { createRequire } from 'module';
 import * as os2 from 'os';
 import os2__default from 'os';
@@ -16,7 +16,7 @@ import { createDefaultContainer, routeImagesForModel, readClipboardImage } from
 import { builtinToolsPack, rememberTool, forgetTool } from '@wrongstack/tools';
 import { fileURLToPath } from 'url';
 import * as readline from 'readline';
-import * as fs10 from 'fs';
+import * as fs11 from 'fs';
 import { writeFileSync, existsSync, readFileSync } from 'fs';
 import { WrongStackACPServer } from '@wrongstack/acp/agent';
 import { ACP_AGENT_COMMANDS, makeACPSubagentRunner, makeACPSubagentRunnerWithStop } from '@wrongstack/acp';
@@ -1696,6 +1696,7 @@ __export(webui_server_exports, {
 async function runWebUI(opts) {
   const port = opts.port ?? 3457;
   const clients = /* @__PURE__ */ new Map();
+  const secretScrubber = new DefaultSecretScrubber();
   let abortController = null;
   const authToken = crypto2.randomBytes(16).toString("hex");
   const wss = new WebSocketServer({ port, host: "127.0.0.1", maxPayload: 1 * 1024 * 1024 });
@@ -1735,7 +1736,7 @@ async function runWebUI(opts) {
           payload: {
             id: e.id,
             name: e.name,
-            input: e.input,
+            input: secretScrubber.scrubObject(e.input),
             messageId: `tool_${e.id}`
           }
         });
@@ -1764,8 +1765,8 @@ async function runWebUI(opts) {
             name: e.name,
             durationMs: e.durationMs,
             ok: e.ok,
-            input: e.input,
-            output: e.output
+            input: secretScrubber.scrubObject(e.input),
+            output: secretScrubber.scrubObject(e.output)
           }
         });
       })
@@ -2349,11 +2350,21 @@ async function resolveRuntimeMaxContext(input) {
     providerConfig?.baseUrl || topLevelBaseUrlApplies && input.config.baseUrl
   );
   if (input.modelsRegistry && !hasCustomBaseUrl) {
-    const caps = await capabilitiesFor(input.modelsRegistry, input.providerId, input.modelId).catch(
-      () => void 0
+    const mergedModels = mergeCustomModelDefs(
+      providerConfig?.customModels,
+      input.config.models
     );
+    const caps = await capabilitiesFor(
+      input.modelsRegistry,
+      input.providerId,
+      input.modelId,
+      mergedModels
+    ).catch(() => void 0);
     const catalogMax = positiveNumber(caps?.maxContext);
     if (catalogMax) return catalogMax;
+    const directModel = await input.modelsRegistry.getModel(input.providerId, input.modelId).catch(() => void 0);
+    const directMax = positiveNumber(directModel?.capabilities.maxContext);
+    if (directMax) return directMax;
   }
   return positiveNumber(input.provider.capabilities.maxContext) ?? 0;
 }
@@ -2389,6 +2400,7 @@ var BOOLEAN_FLAGS = /* @__PURE__ */ new Set([
   "metrics",
   "webui",
   "no-check",
+  "no-models-refresh",
   "director",
   "no-director",
   "no-autonomy",
@@ -5195,6 +5207,224 @@ ${targetMode.description}`
     }
   };
 }
+var noOpVault = {
+  encrypt: (v) => v,
+  decrypt: (v) => v,
+  isEncrypted: () => false
+};
+async function patchGlobalConfig(globalConfigPath, mutate) {
+  let raw = "{}";
+  let fileExists = true;
+  try {
+    raw = await fsp3.readFile(globalConfigPath, "utf8");
+  } catch (err) {
+    if (err.code !== "ENOENT") throw err;
+    fileExists = false;
+  }
+  let parsed;
+  try {
+    parsed = JSON.parse(raw);
+  } catch (err) {
+    if (fileExists) {
+      throw new Error(`Config at ${globalConfigPath} is not valid JSON: ${err.message}`);
+    }
+    parsed = {};
+  }
+  const decrypted = decryptConfigSecrets$1(parsed, noOpVault);
+  mutate(decrypted);
+  const encrypted = encryptConfigSecrets$1(decrypted, noOpVault);
+  await atomicWrite(globalConfigPath, JSON.stringify(encrypted, null, 2), { mode: 384 });
+  return decrypted;
+}
+function fmtModel(id, def) {
+  const parts = [];
+  if (def.provider) parts.push(`${color.dim("provider:")} ${color.cyan(def.provider)}`);
+  if (def.name) parts.push(`${color.dim("name:")} ${def.name}`);
+  const caps = def.capabilities;
+  if (caps) {
+    if (caps.maxContext) parts.push(`${color.dim("maxContext:")} ${color.yellow(String(caps.maxContext))}`);
+    const flags = [];
+    if (caps.tools) flags.push("tools");
+    if (caps.vision) flags.push("vision");
+    if (caps.reasoning) flags.push("reasoning");
+    if (caps.streaming) flags.push("streaming");
+    if (caps.jsonMode) flags.push("json");
+    if (flags.length) parts.push(`${color.dim("caps:")} ${flags.join(", ")}`);
+  }
+  if (def.maxOutput) parts.push(`${color.dim("maxOutput:")} ${color.yellow(String(def.maxOutput))}`);
+  return `    ${color.amber(id)}  ${parts.join("  ")}`;
+}
+function safeAt(arr, idx) {
+  const v = arr[idx];
+  if (v === void 0) throw new Error(`Missing value at position ${idx}`);
+  return v;
+}
+function parseFlags(tokens) {
+  let modelId = "";
+  const caps = {};
+  let provider;
+  let name;
+  let maxOutput;
+  let i = 0;
+  while (i < tokens.length) {
+    const t = tokens[i];
+    if (t.startsWith("--")) {
+      const key = t.slice(2);
+      switch (key) {
+        case "provider":
+          provider = safeAt(tokens, ++i);
+          break;
+        case "name":
+          name = safeAt(tokens, ++i);
+          break;
+        case "max-context":
+          caps.maxContext = Number(safeAt(tokens, ++i));
+          break;
+        case "max-output":
+          maxOutput = Number(safeAt(tokens, ++i));
+          break;
+        case "tools":
+          caps.tools = true;
+          break;
+        case "vision":
+          caps.vision = true;
+          break;
+        case "streaming":
+          caps.streaming = true;
+          break;
+        case "reasoning":
+          caps.reasoning = true;
+          break;
+        case "json-mode":
+          caps.jsonMode = true;
+          break;
+        default:
+          return { modelId: "", error: `Unknown flag: --${key}` };
+      }
+    } else if (!t.startsWith("-") && !modelId) {
+      modelId = t;
+    }
+    i++;
+  }
+  if (!modelId) return { modelId: "", error: "missing model id" };
+  const hasCaps = Object.keys(caps).length > 0;
+  const def = {};
+  if (provider !== void 0) def.provider = provider;
+  if (name !== void 0) def.name = name;
+  if (maxOutput !== void 0) def.maxOutput = maxOutput;
+  if (hasCaps) def.capabilities = caps;
+  return { modelId, def: Object.keys(def).length ? def : void 0 };
+}
+function buildModelsCommand(opts) {
+  const help = [
+    "Usage:",
+    "  /models                              List custom model definitions",
+    "  /models add <id> [flags]             Add or update a custom model",
+    "  /models remove <id>                  Remove a custom model",
+    "",
+    "Flags for add:",
+    "  --provider <id>      Owning provider",
+    '  --name "Display"     Display name',
+    "  --max-context <N>    Context window override",
+    "  --max-output <N>     Max output tokens",
+    "  --tools              Tool-capable",
+    "  --vision             Vision-capable",
+    "  --streaming          Streaming support",
+    "  --reasoning          Reasoning support",
+    "  --json-mode          JSON mode support",
+    "",
+    "Persisted to ~/.wrongstack/config.json."
+  ].join("\n");
+  return {
+    name: "models",
+    description: "Manage custom model definitions.",
+    help,
+    async run(args) {
+      const parts = args.trim().split(/\s+/).filter(Boolean);
+      const sub = (parts[0] ?? "").toLowerCase();
+      if (sub === "help" || sub === "--help") return { message: help };
+      if (!opts.configStore || !opts.paths) {
+        return { message: `${color.red("Error")} config store not available.` };
+      }
+      const config = opts.configStore.get();
+      const globalConfigPath = opts.paths.globalConfig;
+      if (!sub) {
+        const models = config.models ?? {};
+        const ids = Object.keys(models);
+        if (ids.length === 0) {
+          return {
+            message: [
+              `${color.bold("Custom Models")} ${color.dim("(none defined)")}`,
+              "",
+              color.dim("  Add one: /models add <id> --max-context 128000 --tools")
+            ].join("\n")
+          };
+        }
+        return {
+          message: [
+            `${color.bold("Custom Models")} ${color.dim(`(${ids.length})`)}`,
+            ...ids.sort().map((id) => fmtModel(id, models[id]))
+          ].join("\n")
+        };
+      }
+      try {
+        if (sub === "add") {
+          const { modelId, def, error } = parseFlags(parts.slice(1));
+          if (error) {
+            return { message: `${color.red("Error")}: ${error}. ${color.dim("/models help")}` };
+          }
+          if (!def && !error) {
+            return { message: `${color.amber("Usage:")} /models add <id> [--max-context N] [--tools] ... ${color.dim("/models help")}` };
+          }
+          const existingModels = config.models ?? {};
+          const existed = modelId in existingModels;
+          const decrypted = await patchGlobalConfig(globalConfigPath, (cfg) => {
+            const models = { ...cfg.models ?? {} };
+            models[modelId] = {
+              ...models[modelId],
+              ...def,
+              capabilities: {
+                ...models[modelId]?.capabilities,
+                ...def?.capabilities
+              }
+            };
+            cfg.models = models;
+          });
+          opts.configStore.update({
+            models: decrypted.models
+          });
+          return { message: `${color.green("\u2713")} ${color.amber(modelId)} ${existed ? "updated" : "added"}.` };
+        }
+        if (sub === "remove" || sub === "rm") {
+          const modelId = parts[1];
+          if (!modelId) {
+            return { message: `${color.amber("Usage:")} /models remove <id>` };
+          }
+          const existing = config.models ?? {};
+          if (!(modelId in existing)) {
+            return { message: `${color.amber("Not found")}: custom model "${modelId}" is not defined.` };
+          }
+          const decrypted = await patchGlobalConfig(globalConfigPath, (cfg) => {
+            const models = { ...cfg.models ?? {} };
+            delete models[modelId];
+            cfg.models = models;
+          });
+          opts.configStore.update({
+            models: decrypted.models
+          });
+          return { message: `${color.green("\u2713")} removed ${color.amber(modelId)}` };
+        }
+        return {
+          message: `${color.red("Unknown subcommand")} "${sub}". Try ${color.dim("/models")}, ${color.dim("/models add")}, or ${color.dim("/models help")}.`
+        };
+      } catch (err) {
+        return {
+          message: `${color.red("models error")}: ${err instanceof Error ? err.message : String(err)}`
+        };
+      }
+    }
+  };
+}
 function buildNextCommand(opts) {
   return {
     name: "next",
@@ -5452,7 +5682,7 @@ function summariseEvent(ev) {
       return color.dim("\u2026");
   }
 }
-var noOpVault = {
+var noOpVault2 = {
   encrypt: (v) => v,
   decrypt: (v) => v,
   isEncrypted: () => false
@@ -5487,7 +5717,7 @@ function parseTarget(tokens) {
 function fmtEntry(e) {
   return e.provider ? `${e.provider}/${e.model}` : `${e.model} ${color.dim("(leader provider)")}`;
 }
-async function patchGlobalConfig(globalConfigPath, mutate) {
+async function patchGlobalConfig2(globalConfigPath, mutate) {
   let raw = "{}";
   let fileExists = true;
   try {
@@ -5504,9 +5734,9 @@ async function patchGlobalConfig(globalConfigPath, mutate) {
       throw new Error(`Config at ${globalConfigPath} is not valid JSON: ${err.message}`);
     parsed = {};
   }
-  const decrypted = decryptConfigSecrets$1(parsed, noOpVault);
+  const decrypted = decryptConfigSecrets$1(parsed, noOpVault2);
   mutate(decrypted);
-  const encrypted = encryptConfigSecrets$1(decrypted, noOpVault);
+  const encrypted = encryptConfigSecrets$1(decrypted, noOpVault2);
   await atomicWrite(globalConfigPath, JSON.stringify(encrypted, null, 2), { mode: 384 });
   return decrypted;
 }
@@ -5597,7 +5827,7 @@ function buildSetModelCommand(opts) {
               message: `${color.red("Provider not available")}: "${provider}". Keyed: ${keyed.join(", ") || "(none)"}. ${color.dim("/setmodel list")}`
             };
           }
-          const decrypted = await patchGlobalConfig(globalConfigPath, (cfg) => {
+          const decrypted = await patchGlobalConfig2(globalConfigPath, (cfg) => {
             cfg.provider = provider;
             cfg.model = model;
           });
@@ -5628,7 +5858,7 @@ function buildSetModelCommand(opts) {
               message: `${color.red("Provider not available")}: "${parsed.provider}". Keyed: ${keyed.join(", ") || "(none)"}.`
             };
           }
-          const decrypted = await patchGlobalConfig(globalConfigPath, (cfg) => {
+          const decrypted = await patchGlobalConfig2(globalConfigPath, (cfg) => {
             const matrix = { ...cfg.modelMatrix ?? {} };
             matrix[key] = parsed.provider ? { provider: parsed.provider, model: parsed.model } : { model: parsed.model };
             cfg.modelMatrix = matrix;
@@ -5645,7 +5875,7 @@ function buildSetModelCommand(opts) {
           if (!(key in existing)) {
             return { message: `${color.amber("No matrix entry")} for "${key}".` };
           }
-          const decrypted = await patchGlobalConfig(globalConfigPath, (cfg) => {
+          const decrypted = await patchGlobalConfig2(globalConfigPath, (cfg) => {
             const matrix = { ...cfg.modelMatrix ?? {} };
             delete matrix[key];
             cfg.modelMatrix = matrix;
@@ -5697,7 +5927,7 @@ async function persistAutonomySetting(deps, mutator) {
 }
 
 // src/slash-commands/settings.ts
-var noOpVault2 = {
+var noOpVault3 = {
   encrypt: (v) => v,
   decrypt: (v) => v,
   isEncrypted: () => false
@@ -5762,7 +5992,7 @@ function buildSettingsCommand(opts) {
       const persistDeps = {
         configStore: opts.configStore,
         globalConfigPath: opts.paths.globalConfig,
-        vault: noOpVault2
+        vault: noOpVault3
       };
       try {
         if (sub === "delay") {
@@ -6186,6 +6416,7 @@ function buildBuiltinSlashCommands(opts) {
     buildWorktreeCommand(opts),
     buildSettingsCommand(opts),
     buildSetModelCommand(opts),
+    buildModelsCommand(opts),
     buildCollabCommand(opts),
     buildStatuslineCommand({
       cwd: opts.cwd,
@@ -6907,11 +7138,11 @@ async function restoreLast(homeFn = defaultHomeDir) {
 var theme = { primary: color.amber };
 async function saveToGlobalConfig(configPath2, provider, model, homeFn = () => process.env.HOME ?? __require("os").homedir()) {
   try {
-    const { atomicWrite: atomicWrite10 } = await import('@wrongstack/core');
-    const fs22 = await import('fs/promises');
+    const { atomicWrite: atomicWrite12 } = await import('@wrongstack/core');
+    const fs24 = await import('fs/promises');
     let existing = {};
     try {
-      const raw = await fs22.readFile(configPath2, "utf8");
+      const raw = await fs24.readFile(configPath2, "utf8");
       existing = JSON.parse(raw);
     } catch {
     }
@@ -6919,7 +7150,7 @@ async function saveToGlobalConfig(configPath2, provider, model, homeFn = () => p
     existing.provider = provider;
     existing.model = model;
     await backupCurrent(homeFn);
-    await atomicWrite10(configPath2, JSON.stringify(existing, null, 2), { mode: 384 });
+    await atomicWrite12(configPath2, JSON.stringify(existing, null, 2), { mode: 384 });
     try {
       await appendHistory(
         oldCfg,
@@ -7242,12 +7473,12 @@ function pickGroupIndex(opts) {
     try {
       let current = 0;
       try {
-        const parsed = Number.parseInt(fs10.readFileSync(opts.cursorFile, "utf8").trim(), 10);
+        const parsed = Number.parseInt(fs11.readFileSync(opts.cursorFile, "utf8").trim(), 10);
         if (Number.isFinite(parsed)) current = wrap(parsed);
       } catch {
       }
-      fs10.mkdirSync(path8.dirname(opts.cursorFile), { recursive: true });
-      fs10.writeFileSync(opts.cursorFile, String(wrap(current + 1)));
+      fs11.mkdirSync(path8.dirname(opts.cursorFile), { recursive: true });
+      fs11.writeFileSync(opts.cursorFile, String(wrap(current + 1)));
       return current;
     } catch {
     }
@@ -9154,8 +9385,49 @@ ${color.dim(`Current: ${deps.config.provider ?? "<unset>"} / ${deps.config.model
     return 1;
   }
 };
+function parseFlags2(args) {
+  const flags = {};
+  for (let i = 0; i < args.length; i++) {
+    const a = args[i];
+    if (a.startsWith("--")) {
+      const eq = a.indexOf("=");
+      if (eq !== -1) {
+        flags[a.slice(2, eq)] = a.slice(eq + 1);
+      } else {
+        const name = a.slice(2);
+        if (i + 1 < args.length && !args[i + 1].startsWith("--")) {
+          flags[name] = args[++i] ?? "";
+        } else {
+          flags[name] = true;
+        }
+      }
+    }
+  }
+  return flags;
+}
+function positionals(args) {
+  const out = [];
+  for (let i = 0; i < args.length; i++) {
+    const a = args[i];
+    if (a.startsWith("--")) {
+      const eq = a.indexOf("=");
+      if (eq === -1) {
+        if (i + 1 < args.length && !args[i + 1].startsWith("--")) {
+          i++;
+        }
+      }
+      continue;
+    }
+    out.push(a);
+  }
+  return out;
+}
+var DEFAULT_PER_PAGE = 15;
 var modelsCmd = async (args, deps) => {
   const sub = args[0];
+  if (sub === "add") return modelsAdd(args.slice(1), deps);
+  if (sub === "remove") return modelsRemove(args.slice(1), deps);
+  if (sub === "list") return modelsList(args.slice(1), deps);
   if (sub === "refresh") {
     deps.renderer.writeInfo("Refreshing models.dev cache\u2026");
     try {
@@ -9169,9 +9441,13 @@ var modelsCmd = async (args, deps) => {
       return 1;
     }
   }
-  const providerId = sub ?? deps.config.provider;
+  const flags = parseFlags2(args);
+  const search = typeof flags["search"] === "string" ? flags["search"].toLowerCase() : "";
+  const perPage = Number(flags["per-page"]) > 0 ? Number(flags["per-page"]) : DEFAULT_PER_PAGE;
+  const page = Math.max(1, Number(flags["page"]) || 1);
+  const providerId = sub ?? deps.config.provider ?? "";
   if (!providerId) {
-    deps.renderer.writeError("Usage: wstack models <provider> | refresh");
+    deps.renderer.writeError("Usage: wstack models <provider> [--search <term>] [--page N] [--per-page N]");
     return 1;
   }
   let lookupId = providerId;
@@ -9195,34 +9471,209 @@ var modelsCmd = async (args, deps) => {
 `));
   const userModels = deps.config.providers?.[providerId]?.models;
   const catalogById = new Map(provider.models.map((m) => [m.id, m]));
-  const sorted = userModels && userModels.length > 0 ? userModels.map((id) => catalogById.get(id) ?? { id, name: id }) : [...provider.models].sort(
+  const allSorted = userModels && userModels.length > 0 ? userModels.map((id) => catalogById.get(id) ?? { id, name: id }) : [...provider.models].sort(
     (a, b) => (b.release_date ?? "").localeCompare(a.release_date ?? "")
   );
   if (userModels && userModels.length > 0)
     deps.renderer.write(color.dim(`(${userModels.length} model(s) from your saved config)
 `));
-  for (const m of sorted) {
-    const caps = [];
-    if ("tool_call" in m && m.tool_call) caps.push("tools");
-    if ("reasoning" in m && m.reasoning) caps.push("reasoning");
-    if ("modalities" in m && m.modalities?.input?.includes("image")) caps.push("vision");
-    const ctx = "limit" in m && m.limit?.context ? `${(m.limit.context / 1e3).toFixed(0)}k` : "?";
-    const cost = "cost" in m && m.cost?.input !== void 0 ? `$${m.cost.input}/$${m.cost.output ?? "?"}` : "";
-    deps.renderer.write(
-      `  ${m.id.padEnd(40)} ${color.dim(ctx.padStart(6))}  ${color.dim(cost.padEnd(14))} ${color.dim(caps.join(","))}
+  const filtered = search ? allSorted.filter((m) => m.id.toLowerCase().includes(search)) : allSorted;
+  const total = filtered.length;
+  const totalPages = Math.max(1, Math.ceil(total / perPage));
+  const actualPage = Math.min(page, totalPages);
+  const start = (actualPage - 1) * perPage;
+  const pageItems = filtered.slice(start, start + perPage);
+  const end = Math.min(start + pageItems.length, total);
+  const pageHint = totalPages > 1 ? color.cyan(`[page ${actualPage}/${totalPages}]`) : "";
+  const searchHint = search ? color.yellow(` (filtered: "${search}" \u2014 ${total} match${total === 1 ? "" : "es"})`) : color.dim(` (${total} model${total === 1 ? "" : "s"})`);
+  deps.renderer.write(`${pageHint}${searchHint}
+`);
+  if (pageItems.length === 0) {
+    deps.renderer.write(color.dim("(no models match)\n"));
+  } else {
+    if (start > 0)
+      deps.renderer.write(color.dim(`  ${String.fromCharCode(8593)} ${start} above
+`));
+    for (const m of pageItems) {
+      const caps = [];
+      if ("tool_call" in m && m.tool_call) caps.push("tools");
+      if ("reasoning" in m && m.reasoning) caps.push("reasoning");
+      if ("modalities" in m && m.modalities?.input?.includes("image")) caps.push("vision");
+      const ctx = "limit" in m && m.limit?.context ? `${(m.limit.context / 1e3).toFixed(0)}k` : "?";
+      const cost = "cost" in m && m.cost?.input !== void 0 ? `${m.cost.input}/${m.cost.output ?? "?"}` : "";
+      deps.renderer.write(
+        `  ${m.id.padEnd(40)} ${color.dim(ctx.padStart(6))}  ${color.dim(cost.padEnd(14))} ${color.dim(caps.join(","))}
 `
-    );
+      );
+    }
+    if (end < total)
+      deps.renderer.write(color.dim(`  ${String.fromCharCode(8595)} ${total - end} below
+`));
+  }
+  const navLines = [];
+  if (totalPages > 1) {
+    if (actualPage > 1) navLines.push(`--page ${actualPage - 1} (prev)`);
+    if (actualPage < totalPages) navLines.push(`--page ${actualPage + 1} (next)`);
   }
+  navLines.push("--search <term> (filter)");
+  deps.renderer.write(color.dim(`
+${navLines.join(" \xB7 ")}
+`));
   const age = await deps.modelsRegistry.ageSeconds();
   deps.renderer.write(
     color.dim(
-      `
-Cache age: ${isFinite(age) ? `${Math.round(age / 60)}m` : "never fetched"}. Run \`wstack models refresh\` to update.
+      `Cache age: ${isFinite(age) ? `${Math.round(age / 60)}m` : "never fetched"}. Run \`wstack models refresh\` to update.
 `
     )
   );
   return 0;
 };
+async function mutateModelsConfig(deps, mutator) {
+  const vault = deps.vault;
+  const configPath2 = deps.paths.globalConfig;
+  let fileExists = true;
+  let raw;
+  try {
+    raw = await fsp3.readFile(configPath2, "utf8");
+  } catch (err) {
+    if (err.code !== "ENOENT") throw err;
+    fileExists = false;
+    raw = "{}";
+  }
+  let parsed;
+  try {
+    parsed = JSON.parse(raw);
+  } catch (err) {
+    if (fileExists) {
+      throw new Error(
+        `Refusing to overwrite corrupt config at ${configPath2} (${err.message}).`
+      );
+    }
+    parsed = {};
+  }
+  const decrypted = decryptConfigSecrets$1(parsed, vault);
+  const models = decrypted.models ?? {};
+  mutator(models);
+  decrypted.models = models;
+  const encrypted = encryptConfigSecrets$1(decrypted, vault);
+  await atomicWrite(configPath2, JSON.stringify(encrypted, null, 2), { mode: 384 });
+}
+function parseSizeFlag(raw) {
+  if (!raw) return void 0;
+  const s = raw.trim().toLowerCase();
+  const match = /^(\d+(?:\.\d+)?)\s*(k|m|b)?$/.exec(s);
+  if (!match) return void 0;
+  const num = Number.parseFloat(match[1]);
+  const unit = match[2];
+  if (unit === "b") return Math.round(num * 1e9);
+  if (unit === "m") return Math.round(num * 1e6);
+  if (unit === "k") return Math.round(num * 1e3);
+  return Math.round(num);
+}
+function parseBoolFlag(flags, key) {
+  if (flags[key] === true || flags[key] === "true") return true;
+  if (flags[`no-${key}`] !== void 0) return false;
+  return void 0;
+}
+async function modelsAdd(args, deps) {
+  const flags = parseFlags2(args);
+  const pos = positionals(args);
+  const modelId = pos[0];
+  if (!modelId) {
+    deps.renderer.writeError(
+      "Usage: wstack models add <modelId> [--provider <id>] [--name <name>] [--max-context <N>] [--max-output <N>] [--tools] [--no-tools] [--vision] [--no-vision] [--reasoning] [--streaming] [--no-streaming] [--json-mode]"
+    );
+    return 1;
+  }
+  const existing = deps.config.models?.[modelId];
+  if (existing) {
+    deps.renderer.writeWarning(
+      `Model "${modelId}" already defined. Overwriting.`
+    );
+  }
+  const capabilities = {};
+  const toolsVal = parseBoolFlag(flags, "tools");
+  if (toolsVal !== void 0) capabilities.tools = toolsVal;
+  const visionVal = parseBoolFlag(flags, "vision");
+  if (visionVal !== void 0) capabilities.vision = visionVal;
+  const streamingVal = parseBoolFlag(flags, "streaming");
+  if (streamingVal !== void 0) capabilities.streaming = streamingVal;
+  const reasoningVal = parseBoolFlag(flags, "reasoning");
+  if (reasoningVal !== void 0) capabilities.reasoning = reasoningVal;
+  const jsonModeVal = parseBoolFlag(flags, "json-mode");
+  if (jsonModeVal !== void 0) capabilities.jsonMode = jsonModeVal;
+  const maxContextRaw = typeof flags["max-context"] === "string" ? flags["max-context"] : void 0;
+  const maxContext = parseSizeFlag(maxContextRaw);
+  if (maxContext !== void 0) capabilities.maxContext = maxContext;
+  const def = {};
+  const nameFlag = typeof flags["name"] === "string" ? flags["name"] : void 0;
+  const providerFlag = typeof flags["provider"] === "string" ? flags["provider"] : void 0;
+  if (nameFlag) def.name = nameFlag;
+  if (providerFlag) def.provider = providerFlag;
+  if (Object.keys(capabilities).length > 0) def.capabilities = capabilities;
+  const maxOutputRaw = typeof flags["max-output"] === "string" ? flags["max-output"] : void 0;
+  const maxOutput = parseSizeFlag(maxOutputRaw);
+  if (maxOutput !== void 0) def.maxOutput = maxOutput;
+  await mutateModelsConfig(deps, (models) => {
+    models[modelId] = def;
+  });
+  deps.renderer.writeInfo(`Custom model "${modelId}" ${existing ? "updated" : "added"}.`);
+  const capLines = [];
+  if (def.capabilities) {
+    for (const [k, v] of Object.entries(def.capabilities)) {
+      capLines.push(`  ${k}: ${v}`);
+    }
+  }
+  if (def.maxOutput !== void 0) capLines.push(`  maxOutput: ${def.maxOutput}`);
+  if (capLines.length > 0) {
+    deps.renderer.write(color.dim(capLines.join("\n") + "\n"));
+  }
+  return 0;
+}
+async function modelsRemove(args, deps) {
+  const modelId = args[0];
+  if (!modelId) {
+    deps.renderer.writeError("Usage: wstack models remove <modelId>");
+    return 1;
+  }
+  const existing = deps.config.models?.[modelId];
+  if (!existing) {
+    deps.renderer.writeError(`No custom model "${modelId}" found.`);
+    return 1;
+  }
+  await mutateModelsConfig(deps, (models) => {
+    delete models[modelId];
+  });
+  deps.renderer.writeInfo(`Removed custom model "${modelId}".`);
+  return 0;
+}
+async function modelsList(_args, deps) {
+  const models = deps.config.models ?? {};
+  const entries = Object.entries(models);
+  if (entries.length === 0) {
+    deps.renderer.write(color.dim("No custom models defined.\n"));
+    deps.renderer.write(color.dim("Use `wstack models add <modelId> --max-context 128k --tools`\n"));
+    return 0;
+  }
+  deps.renderer.write(color.bold("Custom models\n"));
+  for (const [id, def] of entries.sort(([a], [b]) => a.localeCompare(b))) {
+    const label = def.name ?? id;
+    const provider = def.provider ? ` ${color.dim(`(${def.provider})`)}` : "";
+    deps.renderer.write(`  ${color.bold(label)}${provider}
+`);
+    if (def.capabilities) {
+      for (const [k, v] of Object.entries(def.capabilities)) {
+        deps.renderer.write(`    ${color.dim(`${k}:`)} ${v}
+`);
+      }
+    }
+    if (def.maxOutput !== void 0) {
+      deps.renderer.write(`    ${color.dim("maxOutput:")} ${def.maxOutput}
+`);
+    }
+  }
+  return 0;
+}
 function redactKeys(obj) {
   if (!obj || typeof obj !== "object") return obj;
   if (Array.isArray(obj)) return obj.map(redactKeys);
@@ -9815,10 +10266,10 @@ var auditCmd = async (args, deps) => {
   return verify.ok ? 0 : 1;
 };
 async function listAudits(log, dir, deps) {
-  const fs22 = await import('fs/promises');
+  const fs24 = await import('fs/promises');
   let entries;
   try {
-    entries = await fs22.readdir(dir);
+    entries = await fs24.readdir(dir);
   } catch {
     deps.renderer.write(
       color.dim(`No sessions dir found at ${dir}. Run a session first.`) + "\n"
@@ -9895,6 +10346,9 @@ var helpCmd = async (_args, deps) => {
     "  wstack providers [--all]     List providers from models.dev",
     "  wstack models [<provider>]   List models",
     "  wstack models refresh        Force-refresh cache",
+    "  wstack models add <mid>      Add/override custom model (--max-context, --tools, --vision, \u2026)",
+    "  wstack models remove <mid>   Remove a custom model",
+    "  wstack models list           List all custom models",
     "  wstack mcp [list]            List MCP servers",
     "  wstack plugin [list|status|official|install|add|remove|enable|disable]  Manage plugins",
     "  wstack projects              List tracked projects",
@@ -9955,22 +10409,22 @@ function fmtDuration(ms) {
   const remMin = m - h * 60;
   return `${h}h${remMin}m`;
 }
-function fmtTaskResultLine(r, color45) {
+function fmtTaskResultLine(r, color46) {
   const stats = `${r.iterations}it ${r.toolCalls}tc ${fmtDuration(r.durationMs)}`;
   const errMsg = typeof r.error === "string" ? r.error : r.error?.message;
   const errKind = typeof r.error === "object" ? r.error?.kind : void 0;
   const errTail = errMsg ? ` \u2014 ${errMsg.replace(/\s+/g, " ").slice(0, 80)}${errMsg.length > 80 ? "\u2026" : ""}` : "";
-  const errKindChip = errKind ? color45.dim(` [${errKind}]`) : "";
-  const errSnip = errMsg || errKind ? `${errKindChip}${color45.dim(errTail)}` : "";
+  const errKindChip = errKind ? color46.dim(` [${errKind}]`) : "";
+  const errSnip = errMsg || errKind ? `${errKindChip}${color46.dim(errTail)}` : "";
   switch (r.status) {
     case "success":
-      return { mark: color45.green("\u2713"), stats, tail: "" };
+      return { mark: color46.green("\u2713"), stats, tail: "" };
     case "timeout":
-      return { mark: color45.yellow("\u23F1"), stats: `${color45.yellow("timeout")} ${stats}`, tail: errSnip };
+      return { mark: color46.yellow("\u23F1"), stats: `${color46.yellow("timeout")} ${stats}`, tail: errSnip };
     case "stopped":
-      return { mark: color45.dim("\u2298"), stats: `${color45.dim("stopped")} ${stats}`, tail: errSnip };
+      return { mark: color46.dim("\u2298"), stats: `${color46.dim("stopped")} ${stats}`, tail: errSnip };
     case "failed":
-      return { mark: color45.red("\u2717"), stats: `${color45.red("failed")} ${stats}`, tail: errSnip };
+      return { mark: color46.red("\u2717"), stats: `${color46.red("failed")} ${stats}`, tail: errSnip };
   }
 }
 
@@ -10038,6 +10492,15 @@ async function boot(argv) {
     }).catch(() => {
     });
   }
+  if (!flags["no-models-refresh"]) {
+    try {
+      await modelsRegistry.refresh();
+      logger.info("models.dev catalog refreshed");
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      logger.warn(`models.dev refresh failed (${msg}); using cached catalog`);
+    }
+  }
   const first = positional[0];
   if (first && subcommands[first]) {
     const container = createDefaultContainer({
@@ -11139,7 +11602,17 @@ async function execute(deps) {
     const visionAdapters = () => createToolVisionAdapters(agent.tools);
     const supportsVision = async () => {
       try {
-        const caps = await capabilitiesFor(modelsRegistry, context.provider.id, context.model);
+        const providerConfig = config.providers?.[context.provider.id];
+        const mergedModels = mergeCustomModelDefs(
+          providerConfig?.customModels,
+          config.models
+        );
+        const caps = await capabilitiesFor(
+          modelsRegistry,
+          context.provider.id,
+          context.model,
+          mergedModels
+        );
         return caps.vision;
       } catch {
         return context.provider.capabilities.vision;
@@ -12833,7 +13306,7 @@ async function setupCompaction(params) {
     modelId: config.model ?? context.model
   });
   let autoCompactor;
-  if (config.context.autoCompact !== false) {
+  if (config.context.autoCompact !== false && effectiveMaxContext > 0) {
     const auditLevel = resolveAuditLevel(fullConfig ?? config);
     const sessionBridge = providedBridge ?? createSessionEventBridge(sessionWriter, auditLevel);
     autoCompactor = new AutoCompactionMiddleware(
@@ -13424,7 +13897,12 @@ async function main(argv) {
   const modeId = activeMode?.id ?? "default";
   const modePrompt = activeMode?.prompt ?? "";
   const [resolvedCaps, resolvedModel] = await Promise.all([
-    capabilitiesFor(modelsRegistry, provider.id, config.model).catch(() => void 0),
+    capabilitiesFor(
+      modelsRegistry,
+      provider.id,
+      config.model,
+      mergeCustomModelDefs(config.providers?.[provider.id]?.customModels, config.models)
+    ).catch(() => void 0),
     modelsRegistry.getModel(config.provider, config.model).catch(() => void 0)
   ]);
   const modelCapabilities = resolvedCaps ? {
@@ -13668,9 +14146,11 @@ async function main(argv) {
       providerId,
       modelId
     });
-    effectiveMaxContext = mc > 0 ? mc : 2e5;
+    effectiveMaxContext = mc;
     context.provider.capabilities.maxContext = effectiveMaxContext;
-    autoCompactor?.setMaxContext(effectiveMaxContext);
+    if (effectiveMaxContext > 0) {
+      autoCompactor?.setMaxContext(effectiveMaxContext);
+    }
     events.emit("ctx.max_context", { providerId, modelId, maxContext: effectiveMaxContext });
     updateSpinnerContext();
   };