@tekmidian/pai 0.37.0 → 0.39.0

package/dist/daemon-mcp/index.mjs

@@ -1,7 +1,8 @@
 #!/usr/bin/env node
 import { t as PaiClient } from "../ipc-client-BmypMNYk.mjs";
 import { o as loadConfig } from "../config-B64vFg14.mjs";
-import { C as loadStatus, D as workersLogDir, E as ledgerPath, a as setRole, c as useProvider, d as replayOutput, i as setProviderEnabled, k as readWorkersSection, m as testProvider, n as describeProviders, o as setWorkersEnabled, p as runWorker, r as removeProvider, s as unsetRole, t as addProvider, u as psOutput, v as sayToWorker, w as ledgerSummary } from "../providers-sXcK5bDZ.mjs";
+import { B as workersLogDir, F as ledgerSummary, J as readWorkersSection, M as loadStatuses, a as testProvider, g as sayToWorker, h as readInbox, i as runWorker, j as loadStatus, k as alive, m as handoffFromInside, t as runChain, z as ledgerPath } from "../chain-CTtHligI.mjs";
+import { a as setClass, c as unsetClass, f as psOutput, i as removeProvider, l as updateProvider, n as classTargetText, o as setProviderEnabled, p as replayOutput, r as describeProviders, s as setWorkersEnabled, t as addProvider, u as useProvider } from "../providers-FYkZjn_C.mjs";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 
@@ -1220,6 +1221,10 @@ const $ZodBoolean = /* @__PURE__ */ $constructor("$ZodBoolean", (inst, def) => {
 		return payload;
 	};
 });
+const $ZodUnknown = /* @__PURE__ */ $constructor("$ZodUnknown", (inst, def) => {
+	$ZodType.init(inst, def);
+	inst._zod.parse = (payload) => payload;
+});
 function handleArrayResult(result, final, index) {
 	if (result.issues.length) final.issues.push(...prefixIssues(index, result.issues));
 	final.value[index] = result.value;
@@ -2084,6 +2089,10 @@ function _boolean(Class, params) {
 	});
 }
 /* @__NO_SIDE_EFFECTS__ */
+function _unknown(Class) {
+	return new Class({ type: "unknown" });
+}
+/* @__NO_SIDE_EFFECTS__ */
 function _lt(value, params) {
 	return new $ZodCheckLessThan({
 		check: "less_than",
@@ -2646,6 +2655,7 @@ const numberProcessor = (schema, ctx, _json, _params) => {
 const booleanProcessor = (_schema, _ctx, json, _params) => {
 	json.type = "boolean";
 };
+const unknownProcessor = (_schema, _ctx, _json, _params) => {};
 const enumProcessor = (schema, _ctx, json, _params) => {
 	const def = schema._zod.def;
 	const values = getEnumValues(def.entries);
@@ -3133,6 +3143,14 @@ const ZodBoolean = /* @__PURE__ */ $constructor("ZodBoolean", (inst, def) => {
 function boolean(params) {
 	return _boolean(ZodBoolean, params);
 }
+const ZodUnknown = /* @__PURE__ */ $constructor("ZodUnknown", (inst, def) => {
+	$ZodUnknown.init(inst, def);
+	ZodType.init(inst, def);
+	inst._zod.processJSONSchema = (ctx, json, params) => unknownProcessor(inst, ctx, json, params);
+});
+function unknown() {
+	return _unknown(ZodUnknown);
+}
 const ZodArray = /* @__PURE__ */ $constructor("ZodArray", (inst, def) => {
 	$ZodArray.init(inst, def);
 	ZodType.init(inst, def);
@@ -4524,40 +4542,62 @@ const worker = {
 	description: "Run, watch and route subagent workers on configured providers",
 	content: `## Worker Skill
 
-USE WHEN delegating subagent work, checking on running workers, or managing worker providers.
+USE WHEN delegating subagent work, checking on running workers, or managing worker providers/classes from chat.
 
 ### Delegating
 
 The Agent tool is denied when workers are on — delegate with Bash instead:
 
 \`\`\`
-pai worker run --label "short task label" --role research \\
+pai worker run --label "short task label" --class research \\
   -p '<full, self-contained task spec>' \\
   --allowedTools 'Read,Edit,Write,Bash,Grep,Glob' --output-format json
 \`\`\`
 
 - Always run in background (run_in_background: true) and with a timeout.
-- Roles: implement (default quality), research (web: add WebSearch,WebFetch), spotcheck (cheap/fast model).
+- Classes: draft (spec writing), plan, implement (default quality), review,
+  research (web: add WebSearch,WebFetch), spotcheck (cheap/fast model), simple,
+  complex, image. \`--role\` still works as an alias of \`--class\`.
+- Anything larger than a one-file change goes through a chain:
+  \`pai worker run --chain draft,implement …\` — the draft stage turns the brief
+  into a spec file, implement runs with it, \`--chain draft,implement,review\`
+  adds a review pass. \`--class spotcheck\` for verification runs.
 - The answer is in the \`result\` field of the JSON it prints. Review the diff yourself.
 - \`--no-pane\` suppresses the iTerm follow pane; panes open automatically otherwise.
-- Panes run under the \`pai-worker\` dynamic profile (default profile's font
-  family at \`workers.pane.fontSize\`, default 13); \`pai worker pane <id> --check\`
-  reports the pane plus that profile's path, existence and font.
+- \`--agent <name>\` runs a definition from ~/.claude/agents/<name>.md (the agent
+  library runs on workers: body becomes the system prompt, tools the allowlist,
+  model the class).
 
 ### Watching
 
-- \`worker_ps\` — running + last finished workers.
+- \`worker_ps\` — running + last finished workers (chains show as trees).
 - \`worker_replay\` with id — the transcript of one worker.
 - CLI equivalents: \`pai worker ps\`, \`pai worker follow <id>\`.
 
 ### Routing
 
 - \`worker_status\` — on/off, active provider, providers, run tally.
-- \`worker_providers\` — list/add/remove/use/enable/disable/test. Adding needs
-  name, base_url, model and a key (key file path, or a raw key which is
-  parked in ~/.config/pai/keys/<name>, mode 0600).
-- \`worker_roles\` — list/set/unset role → provider[/fast].
-- \`worker_toggle\` — off makes Agent subagents run on Anthropic again.`
+- \`worker_providers\` — list/add/update/remove/use/enable/disable/test. Adding
+  needs name, base_url, model and a key (key file path, or a raw key which is
+  parked in ~/.config/pai/keys/<name>, mode 0600). update changes cost_tier
+  (1 cheapest … 5 most expensive) and tags (code, vision, image-gen,
+  long-context, fast, reasoning).
+- \`worker_classes\` — list/set/unset class → provider[/fast], or constraints
+  only (max_cost_tier, require_tags) so auto-routing picks a qualifying provider.
+- \`worker_run\` — start a worker or chain from chat; returns the id immediately.
+- \`worker_toggle\` — off makes Agent subagents run on Anthropic again.
+
+### Preferences from chat (map the phrase, use the tool, never mention files)
+
+- "use <provider> for image generation" / "route research to <provider>" →
+  worker_classes set <class>=<provider>. One line back.
+- "prefer the flash model for simple tasks" / "cheap only for drafts" →
+  worker_classes set simple|draft=<provider>/fast (or a max_cost_tier). One line.
+- "reviews should use a reasoning model" → worker_classes set review with
+  require_tags ["reasoning"] (drop the pin if one exists). One line.
+- "what handles reviews" / "show the routing table" → worker_classes list
+  (plus worker_status for providers). One or two lines.
+- Never tell the user to edit a config file — the tools do it.`
 };
 
 //#endregion
@@ -5718,7 +5758,8 @@ async function startShim() {
 	});
 	server.tool("worker_status", [
 		"Show the worker system state: on/off, active provider, configured providers,",
-		"roles, and today's run tally from the ledger.",
+		"roles, today's run tally from the ledger, and the running workers with",
+		"their inbox counts (◆N = handoffs waiting).",
 		"",
 		"Use this before delegating with `pai worker run` to see what routing will choose."
 	].join("\n"), {}, async () => {
@@ -5726,32 +5767,42 @@ async function startShim() {
 			const { workers } = readWorkersSection();
 			const s = ledgerSummary(ledgerPath(workersLogDir(workers)), "all", 10);
 			const tally = s ? [`runs ${s.scope}: ${s.started} started, ${s.endedOk} ok, ${s.endedFailed} failed`, `agent hook: ${s.denied} denied, ${s.allowed} allowed, ${s.reroutes} reroutes`] : ["no runs recorded yet"];
+			const logDir = workersLogDir(workers);
+			const running = loadStatuses(logDir).filter((w) => w.state === "running" && alive(w.pid));
+			const runningLines = running.length ? [
+				"",
+				`running (${running.length}):`,
+				...running.map((w) => `  ${w.id} ${w.label}${w.parent ? ` (sub-worker of ${w.parent})` : ""} · ${w.last}` + (readInbox(logDir, w.id).length ? ` · ◆${readInbox(logDir, w.id).length}` : ""))
+			] : [];
 			return workerText([
 				`workers: ${workers.enabled ? "on" : "off"}`,
 				`active provider: ${workers.active ?? "(none)"}`,
 				...describeProviders(workers),
 				"",
-				...tally
+				...tally,
+				...runningLines
 			].join("\n"));
 		} catch (e) {
 			return workerError(e);
 		}
 	});
 	server.tool("worker_providers", [
-		"Manage worker providers (list, add, remove, use, enable, disable, test).",
+		"Manage worker providers (list, add, update, remove, use, enable, disable, test).",
 		"",
-		"action=list (default) shows providers, roles and routing.",
+		"action=list (default) shows providers (with cost tier and tags), classes and routing.",
 		"action=add needs name, model — and either key_file or key, plus:",
 		"  base_url (anthropic protocol) or upstream_url (protocol=openai, runs",
 		"  through the local PAI proxy). A raw key is written to",
 		"  ~/.config/pai/keys/<name> (mode 0600); only the path lands in the config.",
 		"  engine=codex runs the Codex CLI instead of Claude Code.",
+		"action=update changes cost_tier / tags of an existing provider.",
 		"action=test runs a one-word pong probe through the provider",
 		"  (reports 'codex not installed' when that engine's CLI is missing)."
 	].join("\n"), {
 		action: _enum([
 			"list",
 			"add",
+			"update",
 			"remove",
 			"use",
 			"enable",
@@ -5770,12 +5821,29 @@ async function startShim() {
 		fast_model: string().optional().describe("Cheaper model for spotchecks (add, optional)."),
 		env: record(string(), string()).optional().describe("Extra env for runs (add, optional)."),
 		note: string().optional().describe("Human note shown in listings (add, optional)."),
-		quota_probe: string().optional().describe("URL whose JSON first number is the quota percent (add, optional).")
+		quota_probe: string().optional().describe("URL whose JSON first number is the quota percent (add, optional)."),
+		cost_tier: number().int().min(1).max(5).optional().describe("Cost tier 1 (cheapest) … 5 (most expensive). Default 3. (add, update)"),
+		tags: array(_enum([
+			"code",
+			"vision",
+			"image-gen",
+			"long-context",
+			"fast",
+			"reasoning"
+		])).optional().describe("Capability tags (add, update).")
 	}, async (args) => {
 		try {
 			const action = args.action ?? "list";
 			if (action === "list") return workerText(describeProviders(readWorkersSection().workers).join("\n"));
 			if (!args.name) return workerError(/* @__PURE__ */ new Error("name is required for this action"));
+			if (action === "update") {
+				if (args.cost_tier === void 0 && args.tags === void 0) return workerError(/* @__PURE__ */ new Error("update needs cost_tier and/or tags"));
+				updateProvider(args.name, {
+					...args.cost_tier !== void 0 ? { costTier: args.cost_tier } : {},
+					...args.tags !== void 0 ? { tags: args.tags } : {}
+				});
+				return workerText([`provider ${args.name} updated`, ...describeProviders(readWorkersSection().workers)].join("\n"));
+			}
 			if (action === "add") {
 				if (!args.model) return workerError(/* @__PURE__ */ new Error("add needs model"));
 				if (args.protocol !== "openai" && !args.base_url) return workerError(/* @__PURE__ */ new Error("add needs base_url (only protocol=openai goes without it)"));
@@ -5792,7 +5860,9 @@ async function startShim() {
 					...args.upstream_url ? { upstreamUrl: args.upstream_url } : {},
 					...args.engine ? { engine: args.engine } : {},
 					...args.context_window ? { contextWindow: args.context_window } : {},
-					...args.quota_probe ? { quotaProbe: args.quota_probe } : {}
+					...args.quota_probe ? { quotaProbe: args.quota_probe } : {},
+					...args.cost_tier !== void 0 ? { costTier: args.cost_tier } : {},
+					...args.tags?.length ? { tags: args.tags } : {}
 				});
 				return workerText([`provider ${args.name} added; active: ${workers.active ?? "(none)"}; workers ${workers.enabled ? "on" : "off"}`, ...describeProviders(workers)].join("\n"));
 			}
@@ -5818,37 +5888,140 @@ async function startShim() {
 			return workerError(e);
 		}
 	});
-	server.tool("worker_roles", [
-		"Manage worker roles (list, set, unset).",
+	server.tool("worker_classes", [
+		"Manage worker classes (list, set, unset).",
 		"",
-		"A role maps a task class to a provider, optionally its fast model:",
+		"A class maps a task class to a provider, optionally its fast model:",
 		"implement=glm, research=glm, spotcheck=glm/fast.",
-		"Runs pick the provider via --role first, then the active provider."
+		"set can instead give routing constraints only (max_cost_tier,",
+		"require_tags) — auto-routing then picks a qualifying provider.",
+		"Runs pick the provider via --class first, then the active provider."
 	].join("\n"), {
 		action: _enum([
 			"list",
 			"set",
 			"unset"
 		]).optional().describe("Default: list."),
-		role: string().optional().describe("Role name (set/unset)."),
-		target: string().optional().describe("Provider or provider/fast (set).")
+		class: string().optional().describe("Class name (set/unset), e.g. implement, spotcheck."),
+		target: string().optional().describe("Provider or provider/fast (set)."),
+		max_cost_tier: number().int().min(1).max(5).optional().describe("Auto-routing considers only providers up to this cost tier (set)."),
+		require_tags: array(_enum([
+			"code",
+			"vision",
+			"image-gen",
+			"long-context",
+			"fast",
+			"reasoning"
+		])).optional().describe("Auto-routing needs these tags (set).")
 	}, async (args) => {
 		try {
 			const action = args.action ?? "list";
 			if (action === "list") {
 				const { workers } = readWorkersSection();
-				const entries = Object.entries(workers.roles);
-				const text = (t) => typeof t === "string" ? t : typeof t === "object" && t !== null ? `${t.provider ?? "?"}${t.mcp?.length ? ` +mcp(${t.mcp.join(",")})` : ""}` : String(t);
-				return workerText(entries.length ? entries.map(([r, t]) => `${r}: ${text(t)}`).join("\n") : "no roles set");
+				const entries = Object.entries(workers.classes);
+				return workerText(entries.length ? entries.map(([cls, t]) => `${cls}: ${classTargetText(t)}`).join("\n") : "no classes set");
 			}
-			if (!args.role) return workerError(/* @__PURE__ */ new Error("role is required for this action"));
+			if (!args.class) return workerError(/* @__PURE__ */ new Error("class is required for this action"));
 			if (action === "set") {
-				if (!args.target) return workerError(/* @__PURE__ */ new Error("set needs target provider[/fast]"));
-				setRole(args.role, args.target);
-				return workerText(`role ${args.role} → ${args.target}`);
+				const hasConstraints = args.max_cost_tier !== void 0 || args.require_tags !== void 0;
+				if (!args.target && !hasConstraints) return workerError(/* @__PURE__ */ new Error("set needs target provider[/fast] and/or max_cost_tier/require_tags"));
+				const target = args.target !== void 0 && !hasConstraints ? args.target : {
+					...args.target ? { provider: args.target.split("/")[0] } : {},
+					...args.max_cost_tier !== void 0 ? { maxCostTier: args.max_cost_tier } : {},
+					...args.require_tags !== void 0 ? { requireTags: args.require_tags } : {}
+				};
+				setClass(args.class, target);
+				return workerText(`class ${args.class} → ${classTargetText(target)}`);
 			}
-			unsetRole(args.role);
-			return workerText(`role ${args.role} removed`);
+			unsetClass(args.class);
+			return workerText(`class ${args.class} removed`);
+		} catch (e) {
+			return workerError(e);
+		}
+	});
+	server.tool("worker_run", [
+		"Start a worker (or a chain) and return its id immediately.",
+		"",
+		"This is the chat-side face of `pai worker run`: prompt is required,",
+		"chain (e.g. \"draft,implement\") runs spec-first stages, class picks the",
+		"provider (default: active). Check on it with worker_ps / worker_replay,",
+		"talk to it with worker_say."
+	].join("\n"), {
+		prompt: string().min(1).describe("The task (the -p value)."),
+		chain: string().optional().describe("Comma-separated stages, e.g. draft,implement or draft,implement,review."),
+		class: string().optional().describe("Task class (draft, implement, review, research, spotcheck, simple, complex, image)."),
+		label: string().optional().describe("Short task label shown in worker_ps."),
+		cwd: string().optional().describe("Working directory (default: here)."),
+		allowed_tools: string().optional().describe("Comma-separated tool allowlist passed to the worker."),
+		mcp: string().optional().describe("MCP servers/sets the worker may use (comma-separated).")
+	}, async (args) => {
+		try {
+			const claudeArgs = [
+				"-p",
+				args.prompt,
+				...args.allowed_tools ? ["--allowedTools", args.allowed_tools] : []
+			];
+			let startedIdResolve;
+			let startupFailure = null;
+			const idArrived = new Promise((resolve) => {
+				startedIdResolve = resolve;
+			});
+			const background = async () => {
+				if (args.chain) return runChain({
+					stages: args.chain.split(","),
+					className: args.class,
+					label: args.label,
+					noPane: false,
+					mcpFlag: args.mcp,
+					brief: args.prompt,
+					claudeArgs,
+					...args.cwd ? { cwd: args.cwd } : {},
+					onChainStart: (id) => startedIdResolve(id),
+					quiet: true
+				});
+				return runWorker({
+					className: args.class,
+					label: args.label,
+					noPane: false,
+					mcpFlag: args.mcp,
+					claudeArgs,
+					...args.cwd ? { cwd: args.cwd } : {},
+					onWorkerStart: (wid) => startedIdResolve(wid),
+					quiet: true
+				});
+			};
+			background().catch((e) => {
+				startupFailure = e;
+				startedIdResolve("");
+			});
+			const id = await Promise.race([idArrived, new Promise((_, reject) => setTimeout(() => reject(/* @__PURE__ */ new Error("worker did not start within 15s")), 15e3))]);
+			if (!id) return workerError(startupFailure ?? /* @__PURE__ */ new Error("worker failed to start"));
+			return workerText(`${id} started${args.chain ? ` (chain: ${args.chain})` : ""} — check on it with worker_ps, worker_replay, worker_say`);
+		} catch (e) {
+			return workerError(e);
+		}
+	});
+	server.tool("worker_handoff", [
+		"From inside a worker: send a handoff up to the parent worker.",
+		"",
+		"kind: proposal (do this instead of me), question, blocker, or result",
+		"(results are sent for you when you finish — only send one yourself for",
+		"mid-run findings). The handoff lands in the parent's inbox and is said to",
+		"it when it is still running. Outside a worker this fails — there is no",
+		"sideways or downward path."
+	].join("\n"), {
+		kind: _enum([
+			"proposal",
+			"result",
+			"question",
+			"blocker"
+		]).describe("What the handoff carries."),
+		text: string().min(1).describe("The message body, one paragraph."),
+		data: record(string(), unknown()).optional().describe("Structured payload (optional).")
+	}, async (args) => {
+		try {
+			const h = await handoffFromInside(workersLogDir(readWorkersSection().workers), process.env, args);
+			return workerText(`ok → ${h.to} (${h.kind}); the parent sees it in its inbox`);
 		} catch (e) {
 			return workerError(e);
 		}