@tekmidian/pai 0.37.0 → 0.38.0

package/dist/cli/index.mjs

@@ -7,13 +7,13 @@ import "../helpers-IjZkXBhj.mjs";
 import "../sync-BWbe8JTg.mjs";
 import "../embeddings-DOLZnT1X.mjs";
 import "../search-Rpk1cSBC.mjs";
-import { n as drainStdio, t as buildProgram } from "../program-CEIHn_Ma.mjs";
+import { n as drainStdio, t as buildProgram } from "../program-Y0hAiVy8.mjs";
 import "../context-handover-cache-BpUojjsi.mjs";
 import "../indexer-D7MvSQPY.mjs";
 import "../ipc-client-BmypMNYk.mjs";
 import "../config-B64vFg14.mjs";
 import "../factory-DD2T33C9.mjs";
-import "../providers-sXcK5bDZ.mjs";
+import "../providers-DUshcB-d.mjs";
 import "../main-resolver-DlaLOFBA.mjs";
 import { CommanderError } from "commander";
 

package/dist/cli/program.mjs

@@ -6,13 +6,13 @@ import "../helpers-IjZkXBhj.mjs";
 import "../sync-BWbe8JTg.mjs";
 import "../embeddings-DOLZnT1X.mjs";
 import "../search-Rpk1cSBC.mjs";
-import { t as buildProgram } from "../program-CEIHn_Ma.mjs";
+import { t as buildProgram } from "../program-Y0hAiVy8.mjs";
 import "../context-handover-cache-BpUojjsi.mjs";
 import "../indexer-D7MvSQPY.mjs";
 import "../ipc-client-BmypMNYk.mjs";
 import "../config-B64vFg14.mjs";
 import "../factory-DD2T33C9.mjs";
-import "../providers-sXcK5bDZ.mjs";
+import "../providers-DUshcB-d.mjs";
 import "../main-resolver-DlaLOFBA.mjs";
 
 export { buildProgram };

package/dist/daemon-mcp/index.mjs

@@ -1,7 +1,7 @@
 #!/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 { A as ledgerPath, D as ledgerSummary, E as loadStatus, P as readWorkersSection, _ as testProvider, a as setClass, c as unsetClass, f as psOutput, g as runWorker, h as runChain, i as removeProvider, j as workersLogDir, l as updateProvider, n as classTargetText, o as setProviderEnabled, p as replayOutput, r as describeProviders, s as setWorkersEnabled, t as addProvider, u as useProvider, x as sayToWorker } from "../providers-DUshcB-d.mjs";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 
@@ -4524,40 +4524,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
@@ -5738,20 +5760,22 @@ async function startShim() {
 		}
 	});
 	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 +5794,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 +5833,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 +5861,115 @@ 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);
 		}