@tekmidian/pai 0.37.0 → 0.39.0

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

@@ -131,6 +131,14 @@ var CONFIG_DIR = join(homedir(), ".config", "pai");
 var CONFIG_FILE = join(CONFIG_DIR, "config.json");
 
 // src/workers/config.ts
+var PROVIDER_TAGS = [
+  "code",
+  "vision",
+  "image-gen",
+  "long-context",
+  "fast",
+  "reasoning"
+];
 var DEFAULT_LOG_DIR = "~/.claude/logs/workers";
 var DEFAULT_PANE = {
   enabled: true,
@@ -142,16 +150,24 @@ var DEFAULT_ROUTING2 = {
   cooldownMinutes: 30,
   retryOnQuota: true
 };
+var DEFAULT_TREE = {
+  maxDepth: 2,
+  maxChildren: 4
+};
+var DEFAULT_MCP_SETS = {
+  desktop: ["clickr"]
+};
 function defaultWorkersConfig() {
   return {
     enabled: false,
     active: null,
     providers: {},
-    roles: {},
-    mcpSets: {},
+    classes: {},
+    mcpSets: { ...DEFAULT_MCP_SETS },
     pane: { ...DEFAULT_PANE },
     logDir: DEFAULT_LOG_DIR,
-    routing: { ...DEFAULT_ROUTING2, order: [] }
+    routing: { ...DEFAULT_ROUTING2, order: [] },
+    tree: { ...DEFAULT_TREE }
   };
 }
 var WorkersConfigError = class extends Error {
@@ -210,6 +226,24 @@ function parseProvider(name, raw) {
   if (protocol === "openai" && !upstreamUrl) {
     bad(`.providers.${name}.upstreamUrl`, `is required for protocol "openai" (the Chat Completions base, e.g. "https://api.openai.com/v1")`);
   }
+  const costTier = p.costTier === void 0 ? void 0 : p.costTier;
+  if (costTier !== void 0) {
+    if (typeof costTier !== "number" || !Number.isInteger(costTier) || costTier < 1 || costTier > 5) {
+      bad(`.providers.${name}.costTier`, "must be an integer 1 (cheapest) \u2026 5 (most expensive)");
+    }
+  }
+  let tags;
+  if (p.tags !== void 0) {
+    if (!Array.isArray(p.tags) || p.tags.some((x) => typeof x !== "string")) {
+      bad(`.providers.${name}.tags`, `must be an array of tags from: ${PROVIDER_TAGS.join(", ")}`);
+    }
+    for (const t of p.tags) {
+      if (!PROVIDER_TAGS.includes(t)) {
+        bad(`.providers.${name}.tags`, `"${t}" is not a tag (from: ${PROVIDER_TAGS.join(", ")})`);
+      }
+    }
+    tags = p.tags;
+  }
   return {
     enabled: p.enabled === void 0 ? true : p.enabled === true,
     protocol,
@@ -222,7 +256,9 @@ function parseProvider(name, raw) {
     ...engine !== "claude" ? { engine } : {},
     ...str(p.quotaProbe) ? { quotaProbe: str(p.quotaProbe) } : {},
     ...quotaSkipAt !== void 0 ? { quotaSkipAt } : {},
-    ...contextWindow !== void 0 ? { contextWindow } : {}
+    ...contextWindow !== void 0 ? { contextWindow } : {},
+    ...costTier !== void 0 ? { costTier } : {},
+    ...tags ? { tags } : {}
   };
 }
 function parseWorkersConfig(raw) {
@@ -241,34 +277,68 @@ function parseWorkersConfig(raw) {
       providers[name] = parseProvider(name, p);
     }
   }
-  const roles = {};
-  if (w.roles !== void 0) {
-    if (typeof w.roles !== "object" || w.roles === null || Array.isArray(w.roles)) {
-      bad(".roles", "must be an object of role \u2192 provider[/alias] or {provider, mcp}");
+  const classes = {};
+  const classesRaw = w.classes !== void 0 ? w.classes : w.roles;
+  if (classesRaw !== void 0) {
+    if (typeof classesRaw !== "object" || classesRaw === null || Array.isArray(classesRaw)) {
+      bad(w.classes !== void 0 ? ".classes" : ".roles", "must be an object of class \u2192 provider[/alias] or {provider, mcp, maxCostTier, requireTags, order}");
     }
-    for (const [role, target] of Object.entries(w.roles)) {
+    for (const [cls, target] of Object.entries(classesRaw)) {
       if (typeof target === "object" && target !== null && !Array.isArray(target)) {
         const o = target;
         const provider = str(o.provider);
-        if (!provider || provider.includes(" ")) {
-          bad(`.roles.${role}.provider`, `invalid provider "${provider}"`);
+        if (o.provider !== void 0 && (!provider || provider.includes(" "))) {
+          bad(`.classes.${cls}.provider`, `invalid provider "${provider}"`);
         }
         let mcp;
         if (o.mcp !== void 0) {
           if (!Array.isArray(o.mcp) || o.mcp.some((x) => typeof x !== "string")) {
-            bad(`.roles.${role}.mcp`, "must be an array of MCP server or set names");
+            bad(`.classes.${cls}.mcp`, "must be an array of MCP server or set names");
           }
           mcp = o.mcp;
         }
-        roles[role] = mcp ? { provider, mcp } : { provider };
+        let maxCostTier;
+        if (o.maxCostTier !== void 0) {
+          if (typeof o.maxCostTier !== "number" || !Number.isInteger(o.maxCostTier) || o.maxCostTier < 1 || o.maxCostTier > 5) {
+            bad(`.classes.${cls}.maxCostTier`, "must be an integer 1 \u2026 5");
+          }
+          maxCostTier = o.maxCostTier;
+        }
+        let requireTags;
+        if (o.requireTags !== void 0) {
+          if (!Array.isArray(o.requireTags) || o.requireTags.some((x) => typeof x !== "string")) {
+            bad(`.classes.${cls}.requireTags`, `must be an array of tags from: ${PROVIDER_TAGS.join(", ")}`);
+          }
+          for (const t of o.requireTags) {
+            if (!PROVIDER_TAGS.includes(t)) {
+              bad(`.classes.${cls}.requireTags`, `"${t}" is not a tag (from: ${PROVIDER_TAGS.join(", ")})`);
+            }
+          }
+          requireTags = o.requireTags;
+        }
+        let order;
+        if (o.order !== void 0) {
+          if (!Array.isArray(o.order) || o.order.some((x) => typeof x !== "string")) {
+            bad(`.classes.${cls}.order`, "must be an array of provider names");
+          }
+          order = o.order;
+        }
+        const obj = {
+          ...provider ? { provider } : {},
+          ...mcp ? { mcp } : {},
+          ...maxCostTier !== void 0 ? { maxCostTier } : {},
+          ...requireTags ? { requireTags } : {},
+          ...order ? { order } : {}
+        };
+        classes[cls] = Object.keys(obj).length ? obj : {};
       } else {
         const t = str(target);
-        if (!t || t.includes(" ")) bad(`.roles.${role}`, `invalid target "${t}"`);
-        roles[role] = t;
+        if (!t || t.includes(" ")) bad(`.classes.${cls}`, `invalid target "${t}"`);
+        classes[cls] = t;
       }
     }
   }
-  const mcpSets = {};
+  const mcpSets = { ...DEFAULT_MCP_SETS };
   if (w.mcpSets !== void 0) {
     if (typeof w.mcpSets !== "object" || w.mcpSets === null || Array.isArray(w.mcpSets)) {
       bad(".mcpSets", "must be an object of set name \u2192 [server names]");
@@ -319,6 +389,27 @@ function parseWorkersConfig(raw) {
       retryOnQuota: r.retryOnQuota === void 0 ? DEFAULT_ROUTING2.retryOnQuota : r.retryOnQuota === true
     };
   }
+  let tree = { ...DEFAULT_TREE };
+  if (w.tree !== void 0) {
+    if (typeof w.tree !== "object" || w.tree === null || Array.isArray(w.tree)) {
+      bad(".tree", "must be an object");
+    }
+    const t = w.tree;
+    if (t.maxDepth !== void 0) {
+      if (typeof t.maxDepth !== "number" || !Number.isInteger(t.maxDepth) || t.maxDepth < 0) {
+        bad(".tree.maxDepth", "must be a non-negative integer");
+      }
+    }
+    if (t.maxChildren !== void 0) {
+      if (typeof t.maxChildren !== "number" || !Number.isInteger(t.maxChildren) || t.maxChildren < 1) {
+        bad(".tree.maxChildren", "must be a positive integer");
+      }
+    }
+    tree = {
+      maxDepth: t.maxDepth === void 0 ? DEFAULT_TREE.maxDepth : t.maxDepth,
+      maxChildren: t.maxChildren === void 0 ? DEFAULT_TREE.maxChildren : t.maxChildren
+    };
+  }
   const active = w.active === void 0 || w.active === null ? null : str(w.active);
   if (active !== null && active !== "auto" && !(active in providers)) {
   }
@@ -326,11 +417,12 @@ function parseWorkersConfig(raw) {
     enabled: w.enabled === void 0 ? d.enabled : w.enabled === true,
     active,
     providers,
-    roles,
+    classes,
     mcpSets,
     pane,
     logDir: str(w.logDir) || d.logDir,
-    routing
+    routing,
+    tree
   };
 }
 function readWorkersSection() {
@@ -406,7 +498,44 @@ function workerInScope(worker, term2) {
   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`);
+}
+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 {
+    }
+  }
+  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
+var OPERATOR_MARK = "[operator]";
 var WORKER_CONTRACT_PROMPT = [
   "You are a headless implementation worker, run non-interactively by an orchestrating session.",
   "No narration, no timestamps, no greetings, no summaries of what you read; act, verify, then stop.",
@@ -414,7 +543,20 @@ var WORKER_CONTRACT_PROMPT = [
   '{"changed":[{"path":"\u2026","summary":"\u2026"}],"commands":["\u2026"],"checks":[{"name":"\u2026","ok":true,"detail":"\u2026"}],"open":["\u2026"],"notes":"one line"}',
   "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."
+  "notes: one line, the headline a reviewer reads first.",
+  "",
+  `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",
+  "a question gets its answer, an instruction gets one line naming what will change \u2014 then continue the",
+  "task you were on.",
+  "",
+  "Sub-workers: you may start your own workers with `pai worker run --class <class> -p '<prompt>'`",
+  "(Bash tool). Your children run on their own provider and report back to you automatically when",
+  "they finish. Send a handoff UP instead of doing work yourself when a cheaper provider would",
+  "suffice, the task is out of your scope, or a decision is needed:",
+  '`pai worker handoff \'{"kind":"proposal","text":"\u2026","data":{\u2026}}\'` (kinds: proposal,',
+  "question, blocker; results are sent for you when you finish). Sibling workers are not a",
+  "coordination path \u2014 there is no sideways channel; everything goes up to your parent."
 ].join("\n");
 
 // src/workers/render.ts
@@ -442,9 +584,9 @@ function contextMeter(c, s, minPct = 0) {
   if (pct > 70) return c("yellow", label);
   return label;
 }
-function renderStatusLine(mine, now = /* @__PURE__ */ new Date(), c = makeColor(true)) {
+function renderStatusLine(mine, now = /* @__PURE__ */ new Date(), c = makeColor(true), all = mine, inbox = {}) {
   if (!mine.length) return "";
-  const running = mine.filter((s) => s.state === "running" && alive(s.pid));
+  const running = mine.filter((s) => s.state === "running" && alive(s.pid)).sort((a, b) => workerDepth(all, a.id) - workerDepth(all, b.id));
   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;
@@ -454,7 +596,10 @@ function renderStatusLine(mine, now = /* @__PURE__ */ new Date(), c = makeColor(
   let head = `${providerTag} \u25B6${running.length}`;
   const parts = running.slice(0, 3).map((s) => {
     const meter = contextMeter(c, s, 60);
-    return `${s.id.slice(-4)} ${s.label.slice(0, 26)} ${ageOf(s.started, now)} \xB7 ${s.last.slice(0, 30)}` + (meter ? ` ${meter}` : "");
+    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}` : "");
   });
   if (parts.length) head += "  " + parts.join(" | ");
   if (doneToday.length) head += `   \u2713${ok} \u2717${bad2} today`;
@@ -462,6 +607,14 @@ function renderStatusLine(mine, now = /* @__PURE__ */ new Date(), c = makeColor(
 }
 
 // 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()) {
   const statuses = loadStatuses(logDir);
   const mine = statuses.filter((s) => {
@@ -469,7 +622,7 @@ function statusLineOutput(logDir, term2, cwd2, now = /* @__PURE__ */ new Date())
     const sameDir = cwd2 && s.cwd.startsWith(cwd2);
     return sameScope || !term2 && sameDir;
   });
-  return renderStatusLine(mine, now);
+  return renderStatusLine(mine, now, makeColor(false), statuses, inboxCounts(logDir, mine));
 }
 
 // src/workers/standalone/status-line.ts