@stackmemoryai/stackmemory 1.5.3 → 1.5.5

package/dist/src/cli/commands/orchestrate.js

@@ -60,6 +60,55 @@ function formatElapsed(ms) {
   const days = Math.floor(hours / 24);
   return `${days}d ago`;
 }
+function fmtTokens(n) {
+  if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
+  if (n >= 1e3) return `${(n / 1e3).toFixed(1)}K`;
+  return String(n);
+}
+function budgetBar(pct, width = 30) {
+  const filled = Math.min(Math.round(pct / 100 * width), width);
+  const empty = width - filled;
+  const color = pct >= 75 ? "\x1B[31m" : pct >= 50 ? "\x1B[33m" : "\x1B[32m";
+  const dim = "\x1B[2m";
+  const rst = "\x1B[0m";
+  return `${color}${"\u2588".repeat(filled)}${dim}${"\u2591".repeat(empty)}${rst} ${String(pct).padStart(3)}%`;
+}
+function fmtMinutes(m) {
+  if (m < 0) return "N/A";
+  if (m >= 60) return `${Math.floor(m / 60)}h ${m % 60}m`;
+  return `${m}m`;
+}
+function printUsageSummary(u) {
+  const totalTokens = u.totalTokens || 0;
+  const inputTokens = u.inputTokens || 0;
+  const outputTokens = u.outputTokens || 0;
+  const estMessages = u.estimatedMessages || 0;
+  const tokensPerMin = u.tokensPerMin || 0;
+  const budgetPct5x = u.budgetPct5x || 0;
+  const budgetPct20x = u.budgetPct20x || 0;
+  const mins5x = u.minutesRemaining5x ?? -1;
+  const mins20x = u.minutesRemaining20x ?? -1;
+  const cacheHitRate = u.cacheHitRate || 0;
+  const b = "\x1B[1m";
+  const d = "\x1B[2m";
+  const w = "\x1B[37m";
+  const r = "\x1B[0m";
+  console.log(`${b}Token Usage${r}`);
+  console.log(
+    `  Input  ${w}${fmtTokens(inputTokens)}${r}  ${d}|${r}  Output  ${w}${fmtTokens(outputTokens)}${r}  ${d}|${r}  Total  ${w}${fmtTokens(totalTokens)}${r}`
+  );
+  console.log(
+    `  Rate   ${w}${fmtTokens(tokensPerMin)}/min${r}  ${d}|${r}  Messages  ${w}${estMessages}${r}  ${d}|${r}  Cache hit  ${w}${cacheHitRate}%${r}`
+  );
+  console.log("");
+  console.log(`${b}Budget (Max plan, 5h window)${r}`);
+  console.log(
+    `  5x  (225 msgs)  ${budgetBar(budgetPct5x)}  ${d}~${fmtMinutes(mins5x)} left${r}`
+  );
+  console.log(
+    `  20x (900 msgs)  ${budgetBar(budgetPct20x)}  ${d}~${fmtMinutes(mins20x)} left${r}`
+  );
+}
 function createConductorCommands() {
   const cmd = new Command("conductor");
   cmd.description("Conductor \u2014 autonomous agent orchestration via Linear");
@@ -334,6 +383,54 @@ function createConductorCommands() {
       process.on("SIGTERM", forward);
     });
   });
+  cmd.command("usage").description("Show token usage, budget, and time-to-exhaustion").option("--json", "Output as JSON", false).option("--scan", "Scan Claude Code JSONL logs for historical data", false).action(async (options) => {
+    const statusPath = join(
+      process.cwd(),
+      ".stackmemory",
+      "conductor-status.json"
+    );
+    if (options.scan) {
+      const conductor = new Conductor({ repoRoot: process.cwd() });
+      await conductor.scanUsageLogs();
+      const summary = conductor.getUsageSummary();
+      if (options.json) {
+        console.log(JSON.stringify(summary, null, 2));
+        return;
+      }
+      printUsageSummary(summary);
+      return;
+    }
+    if (!existsSync(statusPath)) {
+      console.log(
+        "No conductor-status.json found. Run with --scan to check JSONL logs, or start the conductor first."
+      );
+      return;
+    }
+    try {
+      const data = JSON.parse(readFileSync(statusPath, "utf-8"));
+      const usage = data.usage || {};
+      if (options.json) {
+        console.log(JSON.stringify(usage, null, 2));
+        return;
+      }
+      printUsageSummary(usage);
+      const rl = data.rateLimit;
+      if (rl) {
+        console.log("");
+        if (rl.inBackoff) {
+          console.log(
+            `Rate limit: \x1B[31mBACKOFF\x1B[0m (${rl.backoffRemainingSec}s remaining, ${rl.totalHits} total hits)`
+          );
+        } else {
+          console.log(
+            `Rate limit: \x1B[32mOK\x1B[0m (${rl.totalHits} total hits)`
+          );
+        }
+      }
+    } catch (err) {
+      console.error("Failed to read status file:", err.message);
+    }
+  });
   cmd.command("start").description("Start the orchestrator daemon").option("--team <id>", "Linear team ID").option(
     "--states <states>",
     "Comma-separated issue states to pick up",
@@ -346,7 +443,11 @@ function createConductorCommands() {
     "--in-review <state>",
     "State name for completed review",
     "In Review"
-  ).option("--poll <ms>", "Polling interval in milliseconds", "30000").option("--concurrency <n>", "Max concurrent agents", "3").option("--workspace-root <path>", "Workspace root directory").option("--repo <path>", "Git repo root for worktrees", process.cwd()).option("--branch <name>", "Base branch for worktrees", "main").option("--retries <n>", "Max retries per issue", "1").option("--turn-timeout <ms>", "Agent turn timeout in ms", "3600000").action(async (options) => {
+  ).option("--poll <ms>", "Polling interval in milliseconds", "30000").option("--concurrency <n>", "Max concurrent agents", "3").option("--workspace-root <path>", "Workspace root directory").option("--repo <path>", "Git repo root for worktrees", process.cwd()).option("--branch <name>", "Base branch for worktrees", "main").option("--retries <n>", "Max retries per issue", "1").option("--turn-timeout <ms>", "Agent turn timeout in ms", "3600000").option(
+    "--mode <mode>",
+    'Agent mode: "cli" (claude -p, session auth) or "adapter" (JSON-RPC, API key)',
+    "cli"
+  ).action(async (options) => {
     const conductor = new Conductor({
       teamId: options.team,
       activeStates: options.states.split(",").map((s) => s.trim()),
@@ -358,7 +459,8 @@ function createConductorCommands() {
       repoRoot: options.repo,
       baseBranch: options.branch,
       maxRetries: parseInt(options.retries, 10),
-      turnTimeoutMs: parseInt(options.turnTimeout, 10)
+      turnTimeoutMs: parseInt(options.turnTimeout, 10),
+      agentMode: options.mode === "adapter" ? "adapter" : "cli"
     });
     await conductor.start();
   });