@tekmidian/pai 0.39.0 → 0.40.0

package/dist/hooks/worker-status-line.mjs

@@ -131,6 +131,7 @@ var CONFIG_DIR = join(homedir(), ".config", "pai");
 var CONFIG_FILE = join(CONFIG_DIR, "config.json");
 
 // src/workers/config.ts
+var MODEL_TIERS = ["haiku", "sonnet", "opus"];
 var PROVIDER_TAGS = [
   "code",
   "vision",
@@ -139,6 +140,10 @@ var PROVIDER_TAGS = [
   "fast",
   "reasoning"
 ];
+var MODEL_CAPABILITIES = ["default", "fast", "image"];
+function isModelCapability(v) {
+  return MODEL_CAPABILITIES.includes(v);
+}
 var DEFAULT_LOG_DIR = "~/.claude/logs/workers";
 var DEFAULT_PANE = {
   enabled: true,
@@ -167,7 +172,8 @@ function defaultWorkersConfig() {
     pane: { ...DEFAULT_PANE },
     logDir: DEFAULT_LOG_DIR,
     routing: { ...DEFAULT_ROUTING2, order: [] },
-    tree: { ...DEFAULT_TREE }
+    tree: { ...DEFAULT_TREE },
+    fallback: null
   };
 }
 var WorkersConfigError = class extends Error {
@@ -199,7 +205,35 @@ function parseProvider(name, raw) {
   const m = modelsRaw;
   const defaultModel = str(m.default);
   if (!defaultModel) bad(`.providers.${name}.models.default`, "is required");
-  const fast = m.fast === void 0 ? void 0 : str(m.fast);
+  const models = { default: defaultModel };
+  for (const key of Object.keys(m)) {
+    if (key === "default") continue;
+    if (!isModelCapability(key)) {
+      bad(
+        `.providers.${name}.models.${key}`,
+        `"${key}" is not a model capability (from: ${MODEL_CAPABILITIES.join(", ")})`
+      );
+    }
+    const id = str(m[key]);
+    if (!id) bad(`.providers.${name}.models.${key}`, "must be a non-empty model id");
+    models[key] = id;
+  }
+  let modelTiers;
+  if (p.modelTiers !== void 0) {
+    if (typeof p.modelTiers !== "object" || p.modelTiers === null || Array.isArray(p.modelTiers)) {
+      bad(`.providers.${name}.modelTiers`, "must be an object of model id \u2192 haiku | sonnet | opus");
+    }
+    modelTiers = {};
+    for (const [id, tier] of Object.entries(p.modelTiers)) {
+      if (typeof tier !== "string" || !MODEL_TIERS.includes(tier)) {
+        bad(
+          `.providers.${name}.modelTiers.${id}`,
+          `must be "haiku", "sonnet" or "opus" (got ${JSON.stringify(tier)})`
+        );
+      }
+      modelTiers[id] = tier;
+    }
+  }
   const env = {};
   if (p.env !== void 0) {
     if (typeof p.env !== "object" || p.env === null || Array.isArray(p.env)) {
@@ -249,7 +283,8 @@ function parseProvider(name, raw) {
     protocol,
     baseUrl,
     keyFile,
-    models: fast ? { default: defaultModel, fast } : { default: defaultModel },
+    models,
+    ...modelTiers ? { modelTiers } : {},
     env,
     ...str(p.note) ? { note: str(p.note) } : {},
     ...upstreamUrl ? { upstreamUrl } : {},
@@ -413,6 +448,39 @@ function parseWorkersConfig(raw) {
   const active = w.active === void 0 || w.active === null ? null : str(w.active);
   if (active !== null && active !== "auto" && !(active in providers)) {
   }
+  let fallback = null;
+  if (w.fallback !== void 0 && w.fallback !== null) {
+    if (typeof w.fallback !== "object" || Array.isArray(w.fallback)) {
+      bad(".fallback", "must be an object (written by `pai worker fallback on`)");
+    }
+    const f = w.fallback;
+    const provider = str(f.provider);
+    if (!provider) bad(".fallback.provider", "is required");
+    const savedRaw = f.saved;
+    if (typeof savedRaw !== "object" || savedRaw === null || Array.isArray(savedRaw)) {
+      bad(".fallback.saved", "must be an object");
+    }
+    const s = savedRaw;
+    if (typeof s.env !== "object" || s.env === null || Array.isArray(s.env)) {
+      bad(".fallback.saved.env", "must be an object of env key \u2192 previous value or null");
+    }
+    const env = {};
+    for (const [k, v] of Object.entries(s.env)) {
+      env[k] = v === null || typeof v === "string" ? v : null;
+    }
+    if (s.model !== null && s.model !== void 0 && typeof s.model !== "string") {
+      bad(".fallback.saved.model", "must be a string or null");
+    }
+    fallback = {
+      provider,
+      saved: {
+        env,
+        model: typeof s.model === "string" ? s.model : null,
+        envExisted: s.envExisted === true
+      },
+      on: str(f.on) || (/* @__PURE__ */ new Date(0)).toISOString()
+    };
+  }
   return {
     enabled: w.enabled === void 0 ? d.enabled : w.enabled === true,
     active,
@@ -422,11 +490,12 @@ function parseWorkersConfig(raw) {
     pane,
     logDir: str(w.logDir) || d.logDir,
     routing,
-    tree
+    tree,
+    fallback
   };
 }
-function readWorkersSection() {
-  const raw = readJsonStrict(CONFIG_FILE, "~/.config/pai/config.json");
+function readWorkersSection(path = CONFIG_FILE) {
+  const raw = readJsonStrict(path, "~/.config/pai/config.json");
   return { raw, workers: parseWorkersConfig(raw.workers) };
 }
 function expandHome(p) {
@@ -439,13 +508,23 @@ function workersLogDir(config) {
   return expandHome(config.logDir);
 }
 
+// src/workers/scope.ts
+import { existsSync as existsSync3, readFileSync as readFileSync3, renameSync as renameSync3, writeFileSync as writeFileSync3, mkdirSync as mkdirSync2 } from "node:fs";
+import { homedir as homedir3 } from "node:os";
+import { join as join4 } from "node:path";
+
 // src/workers/status.ts
 import { existsSync as existsSync2, readFileSync as readFileSync2, renameSync as renameSync2, writeFileSync as writeFileSync2, readdirSync } from "node:fs";
 import { join as join3 } from "node:path";
-function contextPercent(s) {
-  if (!s.contextTokens || !s.contextWindow) return null;
-  return Math.round(s.contextTokens / s.contextWindow * 100);
+
+// src/workers/args.ts
+function shortText(s, n) {
+  const t = String(s ?? "").split(/\s+/).filter(Boolean).join(" ");
+  return t.length <= n ? t : t.slice(0, n - 1) + "\u2026";
 }
+
+// src/workers/status.ts
+var UNLABELED = "unlabeled";
 function loadStatuses(logDir) {
   if (!existsSync2(logDir)) return [];
   const out = [];
@@ -475,8 +554,6 @@ function ageOf(ts, now = /* @__PURE__ */ new Date()) {
 }
 
 // src/workers/scope.ts
-import { homedir as homedir3 } from "node:os";
-import { join as join4 } from "node:path";
 var AIBROKER_REGISTRY = join4(homedir3(), ".aibroker", "session-names.json");
 function tabKey(term2) {
   if (!term2) return "";
@@ -497,41 +574,36 @@ function workerInScope(worker, term2) {
   if (worker.session?.id && uuid) return worker.session.id === uuid;
   return tabKey(worker.term) === tabKey(term2) && tabKey(term2) !== "";
 }
-
-// src/workers/handoff.ts
-import { appendFileSync, existsSync as existsSync3, mkdirSync as mkdirSync2, readFileSync as readFileSync3 } from "node:fs";
-import { dirname, join as join5 } from "node:path";
-function inboxPath(logDir, id) {
-  return join5(logDir, `${id}.inbox.jsonl`);
+var SPAWNER_SESSION_TTL_MS = 10 * 6e4;
+var SESSION_MAP_PRUNE_MS = 60 * 6e4;
+function sessionMapPath(logDir) {
+  return join4(logDir, "claude-session-map.json");
 }
-function readInbox(logDir, id) {
-  const path = inboxPath(logDir, id);
-  if (!existsSync3(path)) return [];
-  const out = [];
-  for (const line of readFileSync3(path, "utf8").split("\n")) {
-    if (!line.trim()) continue;
-    try {
-      out.push(JSON.parse(line));
-    } catch {
+function recordSessionMapEntry(logDir, cwd2, session, now = Date.now()) {
+  if (!cwd2 || !session) return;
+  const path = sessionMapPath(logDir);
+  let map = {};
+  try {
+    if (existsSync3(path)) {
+      map = JSON.parse(readFileSync3(path, "utf8"));
     }
+  } catch {
+    map = {};
+  }
+  const prev = map[cwd2];
+  if (prev && prev.session === session && now - prev.ts < 6e4) return;
+  const pruned = {};
+  for (const [dir, e] of Object.entries(map)) {
+    if (now - e.ts < SESSION_MAP_PRUNE_MS) pruned[dir] = e;
+  }
+  pruned[cwd2] = { session, ts: now };
+  try {
+    mkdirSync2(logDir, { recursive: true });
+    const tmp = `${path}.tmp`;
+    writeFileSync3(tmp, JSON.stringify(pruned), "utf8");
+    renameSync3(tmp, path);
+  } catch {
   }
-  return out;
-}
-
-// src/workers/tree.ts
-function workerDepth(statuses, id) {
-  const byId = new Map(statuses.map((s) => [s.id, s]));
-  let depth = 0;
-  let cur = byId.get(id);
-  const seen = /* @__PURE__ */ new Set([id]);
-  while (cur?.parent && !seen.has(cur.parent)) {
-    seen.add(cur.parent);
-    const next = byId.get(cur.parent);
-    if (!next) break;
-    depth += 1;
-    cur = next;
-  }
-  return depth;
 }
 
 // src/workers/report.ts
@@ -544,6 +616,8 @@ var WORKER_CONTRACT_PROMPT = [
   "changed: files you touched (path + one-line summary). commands: the commands that verify the work.",
   "checks: each with ok true/false and the evidence in detail. open: what you could not finish, if anything.",
   "notes: one line, the headline a reviewer reads first.",
+  "Never write shell heredocs \u2014 write the script to a file with the Write tool, then run it",
+  "(heredocs fail silently or get swallowed).",
   "",
   `Operator messages: a user turn starting with ${OPERATOR_MARK} was typed by the operator while you`,
   "run (the prompt of a resumed run is the operator's too). Answer it FIRST, in one or two plain lines \u2014",
@@ -560,76 +634,61 @@ var WORKER_CONTRACT_PROMPT = [
 ].join("\n");
 
 // src/workers/render.ts
-var CODES = {
-  dim: "2",
-  bold: "1",
-  red: "31",
-  green: "32",
-  yellow: "33",
-  blue: "34",
-  mag: "35",
-  cyan: "36"
-};
-function makeColor(enabled) {
-  return (name, s) => enabled ? `\x1B[${CODES[name]}m${s}\x1B[0m` : s;
-}
-function fmtK(n) {
-  return n >= 1e3 ? `${Math.round(n / 1e3)}k` : String(n);
-}
-function contextMeter(c, s, minPct = 0) {
-  const pct = contextPercent(s);
-  if (pct === null || pct <= minPct) return null;
-  const label = `ctx ${fmtK(s.contextTokens ?? 0)}/${fmtK(s.contextWindow ?? 0)} (${pct}%)`;
-  if (pct > 85) return c("red", label);
-  if (pct > 70) return c("yellow", label);
-  return label;
+function stepOf(last) {
+  const s = String(last ?? "").trim();
+  if (s.startsWith("Bash: ")) {
+    const cmd = s.slice(6).trim();
+    const verb = cmd.split(/\s+/)[0] ?? "";
+    return shortText(cmd.length > verb.length ? `${verb} \u2026` : verb, 24);
+  }
+  return shortText(s, 24);
 }
-function renderStatusLine(mine, now = /* @__PURE__ */ new Date(), c = makeColor(true), all = mine, inbox = {}) {
+function renderStatusLine(mine, now = /* @__PURE__ */ new Date()) {
   if (!mine.length) return "";
-  const running = mine.filter((s) => s.state === "running" && alive(s.pid)).sort((a, b) => workerDepth(all, a.id) - workerDepth(all, b.id));
+  const isChat = (s) => s.origin === "chat" || !s.origin && (s.label === UNLABELED || s.label === "(no prompt)") && s.turns === 0;
+  const chat = mine.find((s) => isChat(s) && s.state === "running" && alive(s.pid));
+  const running = mine.filter((s) => !isChat(s) && s.state === "running" && alive(s.pid)).sort((a, b) => a.started.localeCompare(b.started));
   const today = `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, "0")}-${String(now.getDate()).padStart(2, "0")}`;
   const doneToday = mine.filter((s) => s.state !== "running" && s.started.startsWith(today));
   const ok = doneToday.filter((s) => s.state === "done").length;
   const bad2 = doneToday.length - ok;
-  const providers = new Set(running.map((s) => s.provider));
-  const providerTag = providers.size === 1 ? [...providers][0] : "workers";
-  let head = `${providerTag} \u25B6${running.length}`;
-  const parts = running.slice(0, 3).map((s) => {
-    const meter = contextMeter(c, s, 60);
-    const depth = workerDepth(all, s.id);
-    const lead = depth > 0 ? "  ".repeat(depth - 1) + "\u21B3 " : "";
-    const box = inbox[s.id] ? ` \u25C6${inbox[s.id]}` : "";
-    return `${lead}${s.id.slice(-4)} ${s.label.slice(0, 26)} ${ageOf(s.started, now)} \xB7 ${s.last.slice(0, 30)}${box}` + (meter ? ` ${meter}` : "");
+  const rows = running.slice(0, 3).map((s) => {
+    const twin = running.some((o) => o !== s && o.label === s.label);
+    const name = `${twin ? `#${s.id.slice(-4)} ` : ""}${s.label.slice(0, 26)}`;
+    return `${name} \xB7 ${ageOf(s.started, now)} \xB7 ${stepOf(s.last)}`;
   });
-  if (parts.length) head += "  " + parts.join(" | ");
+  let head;
+  if (chat) head = chat.provider;
+  else {
+    const providers = new Set(running.map((s) => s.provider));
+    head = providers.size === 1 ? [...providers][0] : "workers";
+  }
+  if (running.length) head += ` \u25B6${running.length}`;
+  if (rows.length) head += " | " + rows.join(" | ");
   if (doneToday.length) head += `   \u2713${ok} \u2717${bad2} today`;
   return head;
 }
 
 // src/workers/viewer.ts
-function inboxCounts(logDir, statuses) {
-  const out = {};
-  for (const s of statuses) {
-    const n = readInbox(logDir, s.id).length;
-    if (n) out[s.id] = n;
-  }
-  return out;
-}
-function statusLineOutput(logDir, term2, cwd2, now = /* @__PURE__ */ new Date()) {
+function statusLineOutput(logDir, term2, cwd2, claudeSession2 = "", now = /* @__PURE__ */ new Date()) {
   const statuses = loadStatuses(logDir);
   const mine = statuses.filter((s) => {
     const sameScope = term2 && workerInScope(s, term2);
+    const spawnedHere = claudeSession2 && s.spawnerSession === claudeSession2;
     const sameDir = cwd2 && s.cwd.startsWith(cwd2);
-    return sameScope || !term2 && sameDir;
+    return sameScope || spawnedHere || !term2 && sameDir;
   });
-  return renderStatusLine(mine, now, makeColor(false), statuses, inboxCounts(logDir, mine));
+  return renderStatusLine(mine, now);
 }
 
 // src/workers/standalone/status-line.ts
 var term = process.argv[2] ?? process.env.ITERM_SESSION_ID ?? "";
 var cwd = process.argv[3] ?? process.cwd();
+var claudeSession = process.argv[4] ?? "";
 try {
-  const out = statusLineOutput(workersLogDir(readWorkersSection().workers), term, cwd);
+  const logDir = workersLogDir(readWorkersSection().workers);
+  if (claudeSession) recordSessionMapEntry(logDir, cwd, claudeSession);
+  const out = statusLineOutput(logDir, term, cwd, claudeSession);
   if (out) process.stdout.write(out + "\n");
 } catch {
 }