@wrongstack/cli 0.24.0 → 0.31.1

package/dist/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import * as path24 from 'path';
 import * as fsp3 from 'fs/promises';
-import { color, DefaultPathResolver, TOKENS, DefaultSystemPromptBuilder, makeAutonomyPromptContributor, ToolRegistry, createContextManagerTool, EventBus, SlashCommandRegistry, createDelegateTool, FLEET_ROSTER, createMcpControlTool, EternalAutonomyEngine, DefaultLogger, DefaultModelsRegistry, runProviderWithRetry, ReplayLogStore, ReplayProviderRunner, ProviderRegistry, InMemoryMetricsSink, wireMetricsToEvents, DefaultHealthRegistry, startMetricsServer, RecoveryLock, DefaultAttachmentStore, QueueStore, Context, loadTodosCheckpoint, attachTodosCheckpoint, loadDirectorState, loadPlan, createDefaultPipelines, AutoCompactionMiddleware, estimateRequestTokens, Agent, loadPlugins, FleetManager, makeDirectorSessionFactory, Director, makeFleetEmitTool, makeFleetStatusTool, AutoApprovePermissionPolicy, PhaseStore, AutoPhasePlanner, PhaseGraphBuilder, WorktreeManager, PhaseOrchestrator, makeLLMClassifier, resolveWstackPaths, DefaultSecretVault, migratePlaintextSecrets, DefaultConfigLoader, DefaultSessionReader, ToolAuditLog, DefaultSessionRewinder, DefaultSessionStore, atomicWrite, DefaultPluginAPI, makeAgentSubagentRunner, NULL_FLEET_BUS, buildChildEnv, formatContextWindowModeList, repairToolUseAdjacency, getContextWindowMode, resolveContextWindowPolicy, AGENTS_BY_PHASE, dispatchAgent, formatTodosList, SpecStore, TaskGraphStore, analyzeCriticalPath, getTemplate, listTemplates, templateToMarkdown, SpecParser, renderSpecAnalysis, AISpecBuilder, DefaultTaskStore, TaskTracker, renderProgress, renderTaskGraph, SessionRecovery, loadGoal, goalFilePath, summarizeUsage, saveGoal, emptyGoal, buildGoalPreamble, formatGoal, pendingBtwCount, setBtwNote, decryptConfigSecrets, encryptConfigSecrets as encryptConfigSecrets$1, InputBuilder, FsError, ERROR_CODES, SpecVersioning, ParallelEternalEngine, allServers as allServers$1 } from '@wrongstack/core';
+import { color, DefaultPathResolver, TOKENS, DefaultSystemPromptBuilder, makeAutonomyPromptContributor, ToolRegistry, createContextManagerTool, EventBus, SlashCommandRegistry, createDelegateTool, FLEET_ROSTER, createMcpControlTool, EternalAutonomyEngine, DefaultLogger, DefaultModelsRegistry, 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, AutoApprovePermissionPolicy, PhaseStore, AutoPhasePlanner, PhaseGraphBuilder, WorktreeManager, PhaseOrchestrator, makeLLMClassifier, resolveWstackPaths, DefaultSecretVault, migratePlaintextSecrets, DefaultConfigLoader, DefaultSessionReader, ToolAuditLog, DefaultSessionRewinder, DefaultSessionStore, atomicWrite, DefaultPluginAPI, makeAgentSubagentRunner, NULL_FLEET_BUS, buildChildEnv, formatContextWindowModeList, repairToolUseAdjacency, getContextWindowMode, resolveContextWindowPolicy, AGENTS_BY_PHASE, dispatchAgent, formatTodosList, SpecStore, TaskGraphStore, analyzeCriticalPath, getTemplate, listTemplates, templateToMarkdown, SpecParser, renderSpecAnalysis, AISpecBuilder, DefaultTaskStore, TaskTracker, renderProgress, renderTaskGraph, SessionRecovery, loadGoal, goalFilePath, summarizeUsage, saveGoal, emptyGoal, buildGoalPreamble, formatGoal, pendingBtwCount, setBtwNote, decryptConfigSecrets, encryptConfigSecrets as encryptConfigSecrets$1, InputBuilder, FsError, ERROR_CODES, SpecVersioning, ParallelEternalEngine, allServers as allServers$1 } from '@wrongstack/core';
 import { createRequire } from 'module';
 import * as os3 from 'os';
 import os3__default from 'os';
@@ -10420,6 +10420,13 @@ async function runRepl(opts) {
             activeCtrl = runCtrl2;
             try {
               const runResult = await opts.agent.run(runBlocks, { signal: runCtrl2.signal });
+              opts.onAgentIterationComplete?.(
+                estimateRequestTokensCalibrated(
+                  opts.agent.ctx.messages,
+                  opts.agent.ctx.systemPrompt,
+                  opts.agent.ctx.tools ?? []
+                ).total
+              );
               if (runResult.status === "done" && runResult.finalText) {
                 const specSaved = await trySaveSpecFromAIOutput(runResult.finalText);
                 if (specSaved) {
@@ -10549,6 +10556,13 @@ ${taskList}`;
           );
         }
         const result = await opts.agent.run(routed.blocks, { signal: runCtrl.signal });
+        opts.onAgentIterationComplete?.(
+          estimateRequestTokensCalibrated(
+            opts.agent.ctx.messages,
+            opts.agent.ctx.systemPrompt,
+            opts.agent.ctx.tools ?? []
+          ).total
+        );
         if (result.status === "aborted") {
           opts.renderer.writeWarning("Aborted.");
         } else if (result.status === "failed") {
@@ -10647,6 +10661,13 @@ ${color.dim(
             activeCtrl = nextCtrl;
             try {
               const nextResult = await opts.agent.run(nextBlocks, { signal: nextCtrl.signal });
+              opts.onAgentIterationComplete?.(
+                estimateRequestTokensCalibrated(
+                  opts.agent.ctx.messages,
+                  opts.agent.ctx.systemPrompt,
+                  opts.agent.ctx.tools ?? []
+                ).total
+              );
               if (nextResult.status === "done" && nextResult.finalText?.trim() === "DONE") {
                 opts.renderer.write(color.dim("\n  \u21B3 [autonomy] agent reports task complete.\n"));
               }
@@ -10666,6 +10687,13 @@ ${color.dim(
             activeCtrl = suggestCtrl;
             try {
               const suggestResult = await opts.agent.run(suggestBlocks, { signal: suggestCtrl.signal });
+              opts.onAgentIterationComplete?.(
+                estimateRequestTokensCalibrated(
+                  opts.agent.ctx.messages,
+                  opts.agent.ctx.systemPrompt,
+                  opts.agent.ctx.tools ?? []
+                ).total
+              );
               if (suggestResult.status === "done" && suggestResult.finalText) {
                 opts.renderer.write(
                   `
@@ -11196,7 +11224,10 @@ async function execute(deps) {
           getParallelEngine,
           skillLoader,
           agentsMonitorController,
-          fleetStreamController
+          fleetStreamController,
+          // Report context pressure to the Director after each iteration so
+          // the spawn pre-check (maxLeaderContextLoad) stays accurate.
+          onAgentIterationComplete: director ? (tokens) => director.setLeaderContextPressure(tokens) : void 0
         });
       } finally {
         await webuiPromise.catch(() => void 0);
@@ -11220,7 +11251,8 @@ async function execute(deps) {
         getParallelEngine,
         skillLoader,
         agentsMonitorController,
-        fleetStreamController
+        fleetStreamController,
+        onAgentIterationComplete: director ? (tokens) => director.setLeaderContextPressure(tokens) : void 0
       });
     }
   } finally {
@@ -12398,9 +12430,9 @@ async function setupCompaction(params) {
     autoCompactor = new AutoCompactionMiddleware(
       compactor,
       effectiveMaxContext,
-      // Use the full API request estimator: messages + system prompt + tool definitions.
-      // This matches what the provider actually counts as input tokens.
-      (ctx) => estimateRequestTokens(ctx.messages, ctx.systemPrompt, ctx.tools ?? []).total,
+      // 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,
       {
         warn: config.context.warnThreshold,
         soft: config.context.softThreshold,