@wrongstack/cli 0.119.1 → 0.141.0

package/dist/index.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { color, writeErr, renderProgress, SpecStore, TaskGraphStore, analyzeCriticalPath, getTemplate, listTemplates, templateToMarkdown, SpecParser, renderSpecAnalysis, AISpecBuilder, expectDefined, DefaultTaskStore, TaskTracker, renderTaskGraph, DefaultSecretScrubber, DefaultPathResolver, EventBus, TOKENS, mergeCustomModelDefs, DefaultSystemPromptBuilder, makeAutonomyPromptContributor, ToolRegistry, createContextManagerTool, resolveSessionLoggingConfig, createSessionEventBridge, HookRegistry, HookRunner, SlashCommandRegistry, SessionMemoryConsolidator, BrainDecisionQueue, ObservableBrainArbiter, HumanEscalatingBrainArbiter, DefaultBrainArbiter, createDelegateTool, FLEET_ROSTER, createMcpControlTool, SpecVersioning, atomicWrite, DefaultLogger, DefaultModelsRegistry, isStdinTTY, writeOut, runProviderWithRetry, ReplayLogStore, ReplayProviderRunner, ProviderRegistry, InMemoryMetricsSink, wireMetricsToEvents, DefaultHealthRegistry, startMetricsServer, DEFAULT_SESSION_PRUNE_DAYS, RecoveryLock, DefaultAttachmentStore, QueueStore, Context, loadTodosCheckpoint, attachTodosCheckpoint, loadDirectorState, loadPlan, createDefaultPipelines, resolveContextWindowPolicy, resolveAuditLevel, AutoCompactionMiddleware, estimateRequestTokensCalibrated, Agent, loadPlugins, FleetManager, makeDirectorSessionFactory, Director, makeFleetEmitTool, makeFleetStatusTool, resolveModelMatrix, DEFAULT_SUBAGENT_BASELINE, AutoApprovePermissionPolicy, PhaseStore, AutoPhasePlanner, PhaseGraphBuilder, WorktreeManager, PhaseOrchestrator, makeLLMClassifier, ParallelEternalEngine, EternalAutonomyEngine, allServers as allServers$1, decryptConfigSecrets as decryptConfigSecrets$1, encryptConfigSecrets as encryptConfigSecrets$1, bootConfig as bootConfig$1, setOutputLineGuard, setRawMode, DefaultSessionReader, resolveWstackPaths, ToolAuditLog, DefaultSessionRewinder, DefaultSessionStore, DefaultPluginAPI, StreamHangError, ProviderError, makeAgentSubagentRunner, NULL_FLEET_BUS, buildChildEnv, formatContextWindowModeList, repairToolUseAdjacency, getContextWindowMode, AGENTS_BY_PHASE, dispatchAgent, formatTodosList, loadTasks, emptyTaskFile, saveTasks, formatTaskProgress, formatTaskList, SessionRecovery, loadGoal, goalFilePath, summarizeUsage, saveGoal, formatGoal, emptyGoal, buildGoalPreamble, pendingBtwCount, setBtwNote, MATRIX_PHASE_KEYS, AGENT_CATALOG, matrixKeyKind, phaseForRole, onResize, ERROR_CODES, InputBuilder, FsError } from '@wrongstack/core';
+import { color, writeErr, renderProgress, SpecStore, TaskGraphStore, analyzeCriticalPath, getTemplate, listTemplates, templateToMarkdown, SpecParser, renderSpecAnalysis, AISpecBuilder, expectDefined, DefaultTaskStore, TaskTracker, renderTaskGraph, DefaultSecretScrubber, DefaultPathResolver, EventBus, TOKENS, mergeCustomModelDefs, DefaultSystemPromptBuilder, makeAutonomyPromptContributor, ToolRegistry, createContextManagerTool, resolveSessionLoggingConfig, createSessionEventBridge, HookRegistry, HookRunner, SlashCommandRegistry, SessionMemoryConsolidator, BrainDecisionQueue, ObservableBrainArbiter, HumanEscalatingBrainArbiter, DefaultBrainArbiter, createDelegateTool, FLEET_ROSTER, createMcpControlTool, SpecVersioning, atomicWrite, DefaultLogger, DefaultModelsRegistry, isStdinTTY, writeOut, runProviderWithRetry, ReplayLogStore, ReplayProviderRunner, ProviderRegistry, InMemoryMetricsSink, wireMetricsToEvents, DefaultHealthRegistry, startMetricsServer, DEFAULT_SESSION_PRUNE_DAYS, RecoveryLock, DefaultAttachmentStore, QueueStore, Context, loadTodosCheckpoint, attachTodosCheckpoint, loadDirectorState, loadPlan, createDefaultPipelines, resolveContextWindowPolicy, resolveAuditLevel, AutoCompactionMiddleware, estimateRequestTokensCalibrated, Agent, loadPlugins, FleetManager, makeDirectorSessionFactory, Director, makeFleetEmitTool, makeFleetStatusTool, resolveModelMatrix, DEFAULT_SUBAGENT_BASELINE, AutoApprovePermissionPolicy, PhaseStore, AutoPhasePlanner, PhaseGraphBuilder, WorktreeManager, PhaseOrchestrator, makeLLMClassifier, ParallelEternalEngine, EternalAutonomyEngine, allServers as allServers$1, decryptConfigSecrets as decryptConfigSecrets$1, encryptConfigSecrets as encryptConfigSecrets$1, bootConfig as bootConfig$1, setOutputLineGuard, setRawMode, DefaultSessionReader, resolveWstackPaths, ToolAuditLog, DefaultSessionRewinder, DefaultSessionStore, DefaultPluginAPI, StreamHangError, ProviderError, makeAgentSubagentRunner, NULL_FLEET_BUS, buildChildEnv, formatContextWindowModeList, repairToolUseAdjacency, getContextWindowMode, AGENTS_BY_PHASE, dispatchAgent, formatTodosList, loadTasks, emptyTaskFile, saveTasks, formatTaskProgress, formatTaskList, SessionRecovery, loadGoal, goalFilePath, summarizeUsage, saveGoal, formatGoal, emptyGoal, buildGoalPreamble, pendingBtwCount, setBtwNote, MATRIX_PHASE_KEYS, AGENT_CATALOG, matrixKeyKind, phaseForRole, onResize, ERROR_CODES, InputBuilder, FsError, estimateMessageTokens } from '@wrongstack/core';
 import * as fsp4 from 'fs/promises';
 import { DefaultSecretVault, decryptConfigSecrets, encryptConfigSecrets, isSecretField } from '@wrongstack/core/security';
 import * as path9 from 'path';
@@ -2083,8 +2083,7 @@ async function runWebUI(opts) {
         ws.close();
       }
       clients.clear();
-      void unregisterInstance(process.pid, registryBaseDir).catch(() => {
-      });
+      void unregisterInstance(process.pid, registryBaseDir).catch((err) => console.debug(`[webui-server] unregister failed: ${err}`));
       httpServer?.close();
       wss.close(() => {
         console.log("[WebUI] Server stopped");
@@ -2160,6 +2159,15 @@ async function runWebUI(opts) {
         await handleProviderRemove(ws, m.payload.providerId);
         break;
       }
+      default:
+        send(ws, {
+          type: "error",
+          payload: {
+            phase: "ws.dispatch",
+            message: `Unknown message type: ${String(msg.type)}`
+          }
+        });
+        break;
     }
   }
   async function handleUserMessage(ws, _client, content) {
@@ -2873,20 +2881,7 @@ function countToolResults(messages) {
   return count;
 }
 function estimateTokens(messages) {
-  let total = 0;
-  for (const m of messages) {
-    const content = m.content;
-    if (typeof content === "string") {
-      total += Math.ceil(content.length / 4);
-    } else if (Array.isArray(content)) {
-      for (const b of content) {
-        if (b.type === "text") total += Math.ceil(b.text.length / 4);
-        else if (b.type === "tool_use" || b.type === "tool_result")
-          total += Math.ceil(JSON.stringify(b).length / 4);
-      }
-    }
-  }
-  return total;
+  return estimateMessageTokens(messages);
 }
 
 // src/slash-commands/auth.ts
@@ -3187,6 +3182,7 @@ function buildAutoPhaseCommand(opts) {
           };
         }
       }
+      return { message: `Unknown subcommand "${sub}". Run \`/autophase\` for usage.` };
     }
   };
 }
@@ -3874,7 +3870,7 @@ ${formatContextWindowModeList(active)}`;
       const lines = [
         `${color.bold("Context Window")}`,
         `  messages:    ${messages.length} total (${countTurnPairs(messages)} user+assistant pairs)`,
-        `  tokens (est): ${estimateTokens(messages).toLocaleString()} (chars / 4 estimate)`,
+        `  tokens (est): ${estimateTokens(messages).toLocaleString()} (\u2248 chars/3.5)`,
         `  mode:        ${policy ? `${policy.id} (${policy.name})` : "balanced"}`,
         `  limit:       ${formatLimit(readEffectiveLimit(ctx, opts))}`,
         `  system prompt: ${ctx.systemPrompt.length} block${ctx.systemPrompt.length !== 1 ? "s" : ""}`,
@@ -4137,7 +4133,7 @@ async function persistTelegramConfig(deps, mutator) {
   decrypted.extensions = extensions;
   const encrypted = encryptConfigSecrets$1(decrypted, deps.vault);
   await atomicWrite(deps.globalConfigPath, JSON.stringify(encrypted, null, 2), { mode: 384 });
-  deps.configStore.update({ extensions: decrypted.extensions });
+  deps.configStore.update({ extensions });
 }
 
 // src/slash-commands/enhance.ts
@@ -5605,6 +5601,7 @@ ${formatGoal(updated)}` };
           } catch {
           }
           if (opts.onEternalStop) opts.onEternalStop();
+          if (opts.onAutonomy) opts.onAutonomy("off");
           const msg = `${color.amber("Goal cleared.")} Previous goal marked abandoned; eternal mode will stop.`;
           opts.renderer.write(msg);
           return { message: msg };
@@ -8823,7 +8820,10 @@ var ReadlineInputReader = class {
       const fresh = this.ensure();
       this.installPromptGuard(fresh);
       return new Promise((resolve5) => {
+        let settled = false;
         const settle = (line) => {
+          if (settled) return;
+          settled = true;
           setOutputLineGuard(null);
           resolve5(line);
         };
@@ -8835,6 +8835,7 @@ var ReadlineInputReader = class {
           settle(line);
         });
         fresh.once("close", () => settle(""));
+        fresh.on && fresh.on("error", (_e) => settle(""));
       }).then((result) => {
         this.rl?.close();
         return result;
@@ -13040,10 +13041,10 @@ function roleCat(role) {
 function createProviderForId(providerId, cfg) {
   const savedCfg = cfg.providers?.[providerId];
   const resolvedProviderId = savedCfg?.type ?? providerId;
-  const cfgWithType = {
-    ...savedCfg ?? { type: providerId, apiKey: cfg.apiKey, baseUrl: cfg.baseUrl },
-    type: resolvedProviderId
-  };
+  const cfgWithType = Object.assign(
+    { type: resolvedProviderId },
+    savedCfg ?? { apiKey: cfg.apiKey, baseUrl: cfg.baseUrl }
+  );
   try {
     return makeProviderFromConfig(resolvedProviderId, cfgWithType);
   } catch {
@@ -13288,10 +13289,17 @@ var modeldiagCmd = async (args, deps) => {
     const providersEqIdx = benchArgs.findIndex((a) => a.startsWith("--providers="));
     let providerFilter;
     if (providersEqIdx >= 0) {
-      providerFilter = benchArgs[providersEqIdx].replace("--providers=", "").split(",").map((s) => s.trim()).filter(Boolean);
-      benchArgs.splice(providersEqIdx, 1);
+      const rawFilter = benchArgs[providersEqIdx]?.replace("--providers=", "").split(",");
+      if (rawFilter) {
+        providerFilter = rawFilter.map((s) => s.trim()).filter(Boolean);
+        benchArgs.splice(providersEqIdx, 1);
+      }
     }
     const benchRole = benchArgs[0];
+    if (!benchRole) {
+      writeLine(`${color.amber("No benchmark role specified")}. Usage: wstack diag bench <role> [prompt]`);
+      return 1;
+    }
     const benchPrompt = benchArgs.slice(1).join(" ");
     const cat = roleCat(benchRole);
     const candidates = rankModels(providers, hasKey, cat, 5);
@@ -14431,6 +14439,28 @@ async function runRepl(opts) {
 `
               );
             }
+            if (engine.currentState === "stopped") {
+              const goal = await loadGoalSafe(opts);
+              if (goal?.goalState === "completed") {
+                const u = goal.journal.length > 0 ? summarizeUsage(goal) : null;
+                const costLine = u && u.iterationsWithUsage > 0 ? color.dim(` \u2014 ${u.totalCostUsd.toFixed(4)} \xB7 ${u.totalInputTokens} in / ${u.totalOutputTokens} out \xB7 ${goal.iterations} iterations`) : goal.iterations > 0 ? color.dim(` \u2014 ${goal.iterations} iterations`) : "";
+                opts.renderer.write(
+                  color.green(`
+  \u{1F3AF} Goal completed!${costLine}
+
+`) + color.dim("  Goal cleared. Use /goal set <mission> to create a new goal.\n")
+                );
+                if (opts.projectRoot) {
+                  try {
+                    const { unlink: unlink4 } = await import('fs/promises');
+                    await unlink4(goalFilePath(opts.projectRoot));
+                  } catch {
+                  }
+                }
+              }
+              opts.onAutonomy?.("off");
+              continue;
+            }
           } catch (err) {
             opts.renderer.writeError(
               `[eternal] ${err instanceof Error ? err.message : String(err)}`
@@ -14477,6 +14507,28 @@ async function runRepl(opts) {
 `
               );
             }
+            if (engine.currentState === "stopped") {
+              const goal = await loadGoalSafe(opts);
+              if (goal?.goalState === "completed") {
+                const u = goal.journal.length > 0 ? summarizeUsage(goal) : null;
+                const costLine = u && u.iterationsWithUsage > 0 ? color.dim(` \u2014 ${u.totalCostUsd.toFixed(4)} \xB7 ${u.totalInputTokens} in / ${u.totalOutputTokens} out \xB7 ${goal.iterations} iterations`) : goal.iterations > 0 ? color.dim(` \u2014 ${goal.iterations} iterations`) : "";
+                opts.renderer.write(
+                  color.green(`
+  \u{1F3AF} Goal completed!${costLine}
+
+`) + color.dim("  Goal cleared. Use /goal set <mission> to create a new goal.\n")
+                );
+                if (opts.projectRoot) {
+                  try {
+                    const { unlink: unlink4 } = await import('fs/promises');
+                    await unlink4(goalFilePath(opts.projectRoot));
+                  } catch {
+                  }
+                }
+              }
+              opts.onAutonomy?.("off");
+              continue;
+            }
           } catch (err) {
             opts.renderer.writeError(
               `[parallel] ${err instanceof Error ? err.message : String(err)}`
@@ -15337,8 +15389,7 @@ async function execute(deps) {
                 const toWrite = targetPath === wpaths.globalConfig ? decrypted : filterSafeForProject(decrypted);
                 const encrypted = encryptConfigSecrets$1(toWrite, vault);
                 if (targetPath !== wpaths.globalConfig) {
-                  await fsp4.mkdir(path9.dirname(targetPath), { recursive: true }).catch(() => {
-                  });
+                  await fsp4.mkdir(path9.dirname(targetPath), { recursive: true }).catch((err) => console.debug(`[execution] mkdir failed: ${err}`));
                 }
                 await atomicWrite(targetPath, JSON.stringify(encrypted, null, 2), { mode: 384 });
                 configStore.update({
@@ -15428,13 +15479,15 @@ async function execute(deps) {
             return typeof metaMode === "string" ? metaMode : modeId ?? "default";
           },
           registerDebugStreamCallback: (cb) => {
-            void import('@wrongstack/providers').then(
-              ({ setDebugStreamCallback }) => setDebugStreamCallback(cb)
+            void import('@wrongstack/providers').then(({ setDebugStreamCallback }) => setDebugStreamCallback(cb)).catch(
+              (err) => console.error("[execution] failed to register debug stream callback:", err)
             );
           },
           restoreDebugStreamCallback: () => {
             void import('@wrongstack/providers').then(
               ({ setDebugStreamCallback, defaultDebugStreamCallback }) => setDebugStreamCallback(defaultDebugStreamCallback)
+            ).catch(
+              (err) => console.error("[execution] failed to restore debug stream callback:", err)
             );
           }
         });
@@ -15481,7 +15534,7 @@ async function execute(deps) {
           onAgentIterationComplete: director ? (tokens) => director.setLeaderContextPressure(tokens) : void 0
         });
       } finally {
-        await webuiPromise.catch(() => void 0);
+        await webuiPromise.catch((err) => console.debug(`[execution] webui shutdown failed: ${err}`));
       }
     } else {
       code = await runRepl({
@@ -16885,7 +16938,12 @@ async function setupCompaction(params) {
       effectiveMaxContext,
       // Calibrated estimator: recordActualUsage() is called after each API
       // response so this converges on real token counts for compaction decisions.
-      (ctx) => estimateRequestTokensCalibrated(ctx.messages, ctx.systemPrompt, ctx.tools ?? []).total,
+      (ctx) => estimateRequestTokensCalibrated(
+        ctx.messages,
+        ctx.systemPrompt,
+        ctx.tools ?? [],
+        `${ctx.provider?.id ?? "unknown"}/${ctx.model}`
+      ).total,
       initialPolicy.thresholds,
       {
         aggressiveOn: initialPolicy.aggressiveOn,
@@ -17443,7 +17501,7 @@ async function setupSession(params) {
   } = params;
   sessionStore.prune(DEFAULT_SESSION_PRUNE_DAYS).then((count) => {
     if (count > 0) renderer.writeInfo(`Pruned ${count} old session${count === 1 ? "" : "s"}.`);
-  }).catch(() => void 0);
+  }).catch((err) => console.debug(`[session] prune failed: ${err}`));
   let resumeId = typeof flags["resume"] === "string" ? flags["resume"] : void 0;
   const recoveryLock = new RecoveryLock({ dir: wpaths.projectSessions, sessionStore });
   if (!resumeId && !flags["no-recovery"]) {
@@ -18866,6 +18924,7 @@ Restart WrongStack to load or unload plugin code in this session.`;
           message: `\u26A0 ${color.yellow(`${lines.length} uncommitted change${lines.length > 1 ? "s" : ""}`)} \u2014 session ended without commit`
         };
       }
+      return void 0;
     },
     onClear: () => {
       if (flags.tui && !flags["no-tui"]) return;