@tekmidian/pai 0.49.0 → 0.51.0

package/dist/daemon-mcp/index.mjs

@@ -1,12 +1,12 @@
 #!/usr/bin/env node
 import "../db-Ddns80yf.mjs";
-import { o as loadConfig } from "../config-DCu8THKi.mjs";
-import { r as MODEL_CAPABILITIES, v as readWorkersSection } from "../config-DxMZjpvY.mjs";
-import { C as ledgerSummary, E as alive, L as setWorkerLabel, M as loadStatuses, d as readInbox, f as sayToWorker, j as loadStatus, n as runWorker, r as testProvider, u as handoffFromInside } from "../run-Cls0YGgV.mjs";
+import { p as MainConfigError, s as loadConfig } from "../config-Bn9jpblO.mjs";
+import { C as readWorkersSection, c as MODEL_CAPABILITIES, n as CAPABILITY_NAME_RE } from "../config-DYXARgk8.mjs";
+import { B as setWorkerLabel, E as ledgerSummary, F as loadStatuses, P as loadStatus, c as handoffFromInside, k as alive, l as readInbox, n as runWorker, r as testProvider, u as sayToWorker } from "../run-TChOnMO_.mjs";
 import { t as PaiClient } from "../ipc-client-C6NryMMb.mjs";
-import { _ as workersLogDir, m as ledgerPath } from "../server-Ch99xt5K.mjs";
-import { t as runChain } from "../chain-CNSa0A3G.mjs";
-import { C as readSpecPrompt, _ as updateProvider, a as addProvider, b as psOutput, c as describeProviders, d as resolveProviderName, f as setClass, g as unsetClass, h as setWorkersEnabled, i as fallbackStatusText, l as modelPrefsText, m as setProviderModel, n as fallbackOn, o as classTargetText, p as setProviderEnabled, r as fallbackStatus, s as describeModels, t as fallbackOff, u as removeProvider, v as useProvider, w as resolveSpecPath, x as replayOutput } from "../fallback-IeR2FrFn.mjs";
+import { A as getConfigValueOp, C as psOutput, D as resolveSpecPath, E as readSpecPrompt, M as setConfigValueOp, N as unsetConfigValueOp, O as MainConfigOpsError, _ as setWorkersEnabled, a as addProvider, b as updateProvider, c as describeModels, d as removeProvider, f as resolveProviderName, g as setProviderModel, h as setProviderEnabled, i as fallbackStatusText, j as listConfigOp, l as describeProviders, m as setClass, n as fallbackOn, o as classTargetText, p as setCapabilityPreference, r as fallbackStatus, s as describeCapabilities, t as fallbackOff, u as modelPrefsText, v as unsetCapabilityPreference, w as replayOutput, x as useProvider, y as unsetClass } from "../fallback-CIGspCSi.mjs";
+import { g as ledgerPath, y as workersLogDir } from "../server-R9UkCn3v.mjs";
+import { t as runChain } from "../chain-BBbVbWVv.mjs";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 
@@ -3839,7 +3839,7 @@ USE WHEN user says 'load vault context', 'brief me from Obsidian', 'morning brie
 
 Reads: daily note → open TODOs → PAI index (active projects) → HOME.md (focus areas) → recent insights. Synthesizes into morning briefing with Suggested First Action.
 
-All vault skills work with the vault configured in \`~/.claude/pai/config.json\` (\`vaultPath\` key).`
+All vault skills work with the vault configured in \`~/.claude/pai/config.yaml\` (\`vaultPath\` key, or \`config.json\` until you run \`pai config yaml\`).`
 };
 
 //#endregion
@@ -4709,7 +4709,7 @@ User
 | Memory | \${PAI_DIR}/projects/*/Notes/ | Session notes, todos |
 | Skills | ~/.claude/Skills/ | Workflow instructions |
 | History | \${PAI_HOME}/History/ | Automated capture |
-| Config | ~/.claude/pai/ | Agent preferences (config.json, workers.yaml) |
+| Config | ~/.claude/pai/ | Agent preferences (config.yaml, workers.yaml, etc.) |
 
 ## Two-Tier MCP Strategy
 
@@ -5441,6 +5441,7 @@ const workerRunShape = {
 	specPath: string().min(1).optional().describe("Path to a file (or \"-\" for stdin) whose contents become the prompt. Mutually exclusive with prompt."),
 	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)."),
+	capability: string().optional().describe("Run on whichever provider serves this capability instead of the active one — e.g. \"image\" runs an engine=image provider directly (POSTs the prompt, writes a PNG under the worker's log dir) instead of spawning claude. Wins over class."),
 	label: string().min(1).describe("Short task label shown in worker_ps / the status line and worker pane."),
 	cwd: string().optional().describe("Working directory (default: here)."),
 	allowed_tools: string().optional().describe("Comma-separated tool allowlist passed to the worker."),
@@ -5471,11 +5472,11 @@ function resolveWorkerRunPrompt(args, cwd) {
 * handler, index.ts registers it. The mutations themselves live in
 * src/workers/providers.ts, shared with the `pai worker model` CLI.
 */
-const text = (s) => ({ content: [{
+const text$2 = (s) => ({ content: [{
 	type: "text",
 	text: s
 }] });
-const error = (e) => ({
+const error$1 = (e) => ({
 	content: [{
 		type: "text",
 		text: e instanceof Error ? e.message : String(e)
@@ -5490,22 +5491,128 @@ const error = (e) => ({
 function workerModel(args, configPath) {
 	try {
 		const action = args.action ?? "get";
-		if (action !== "get" && action !== "set") return error(/* @__PURE__ */ new Error(`unknown action "${String(action)}" (expected: get, set)`));
-		if (args.capability !== void 0 && args.slot !== void 0 && args.capability !== args.slot) return error(/* @__PURE__ */ new Error(`capability "${args.capability}" and slot "${args.slot}" disagree — pass one`));
+		if (action !== "get" && action !== "set") return error$1(/* @__PURE__ */ new Error(`unknown action "${String(action)}" (expected: get, set)`));
+		if (args.capability !== void 0 && args.slot !== void 0 && args.capability !== args.slot) return error$1(/* @__PURE__ */ new Error(`capability "${args.capability}" and slot "${args.slot}" disagree — pass one`));
 		const capability = args.capability ?? args.slot ?? "default";
 		const { workers } = readWorkersSection(configPath);
 		if (action === "get") {
 			if (args.provider) {
 				const name = resolveProviderName(workers, args.provider);
 				const p = workers.providers[name];
-				return p ? text(`${name}  ${modelPrefsText(p)}`) : error(/* @__PURE__ */ new Error(`no provider named "${args.provider}"`));
+				return p ? text$2(`${name}  ${modelPrefsText(p)}`) : error$1(/* @__PURE__ */ new Error(`no provider named "${args.provider}"`));
 			}
-			return text(describeModels(workers).join("\n"));
+			return text$2(describeModels(workers).join("\n"));
 		}
-		if (!args.model || !args.model.trim()) return error(/* @__PURE__ */ new Error("set needs a non-empty model id"));
+		if (!args.model || !args.model.trim()) return error$1(/* @__PURE__ */ new Error("set needs a non-empty model id"));
 		const name = resolveProviderName(workers, args.provider);
 		const p = setProviderModel(name, capability, args.model, configPath).providers[name];
-		return text(`${name} ${capability} model: ${p?.models[capability] ?? "?"}`);
+		return text$2(`${name} ${capability} model: ${p?.models[capability] ?? "?"}`);
+	} catch (e) {
+		return error$1(e);
+	}
+}
+
+//#endregion
+//#region src/daemon-mcp/tools/config.ts
+/**
+* MCP handlers for config_list/config_get/config_set/config_unset — thin
+* wrappers around src/config/main-config-ops.ts, the same functions
+* src/cli/commands/config.ts calls, so masking and validation cannot drift
+* between the CLI and MCP surfaces.
+*/
+function text$1(s) {
+	return { content: [{
+		type: "text",
+		text: s
+	}] };
+}
+function errorText(e) {
+	return {
+		content: [{
+			type: "text",
+			text: e instanceof MainConfigOpsError || e instanceof MainConfigError ? e.message : e instanceof Error ? e.message : String(e)
+		}],
+		isError: true
+	};
+}
+function configList(args) {
+	try {
+		const r = listConfigOp({ all: args.all });
+		return text$1(args.json ? JSON.stringify(r.data, null, 2) : r.yaml.trimEnd());
+	} catch (e) {
+		return errorText(e);
+	}
+}
+function configGet(args) {
+	try {
+		const r = getConfigValueOp(args.path);
+		if (!r.found) return errorText(/* @__PURE__ */ new Error(`Not set: ${args.path}`));
+		return text$1(typeof r.value === "object" ? JSON.stringify(r.value, null, 2) : String(r.value));
+	} catch (e) {
+		return errorText(e);
+	}
+}
+function configSet(args) {
+	try {
+		const r = setConfigValueOp(args.path, args.value, { force: args.force });
+		const lines = [];
+		if (r.yamlCreated) lines.push(`created ${r.yamlPath}`);
+		lines.push(`Set ${args.path} = ${typeof r.value === "object" ? JSON.stringify(r.value) : String(r.value)}`);
+		return text$1(lines.join("\n"));
+	} catch (e) {
+		return errorText(e);
+	}
+}
+function configUnset(args) {
+	try {
+		return text$1(unsetConfigValueOp(args.path).existed ? `Unset ${args.path}` : `${args.path} was not set`);
+	} catch (e) {
+		return errorText(e);
+	}
+}
+
+//#endregion
+//#region src/daemon-mcp/tools/worker-capability.ts
+/**
+* worker_capability — the MCP tool behind cross-provider capability
+* preference (the config behind `pai worker capability`).
+*
+* Extracted like worker-model.ts: index.ts starts its transport on import,
+* so its inline handlers cannot be imported by tests. The mutations
+* themselves live in src/workers/providers.ts, shared with the CLI.
+*/
+const text = (s) => ({ content: [{
+	type: "text",
+	text: s
+}] });
+const error = (e) => ({
+	content: [{
+		type: "text",
+		text: e instanceof Error ? e.message : String(e)
+	}],
+	isError: true
+});
+/**
+* list (default): every capability preference and what it resolves to right
+* now. set: which provider(s) serve a capability, first usable one wins.
+* unset: remove a preference.
+*/
+function workerCapability(args, configPath) {
+	try {
+		const action = args.action ?? "list";
+		if (action === "list") {
+			const { workers } = readWorkersSection(configPath);
+			return text(describeCapabilities(workers).join("\n"));
+		}
+		if (!args.capability) return error(/* @__PURE__ */ new Error(`capability is required for action "${action}"`));
+		if (action === "unset") {
+			unsetCapabilityPreference(args.capability, configPath);
+			return text(`removed the capability preference for "${args.capability}"`);
+		}
+		if (action !== "set") return error(/* @__PURE__ */ new Error(`unknown action "${String(action)}" (expected: list, set, unset)`));
+		if (!args.providers || !args.providers.length) return error(/* @__PURE__ */ new Error("set needs a non-empty providers list"));
+		const workers = setCapabilityPreference(args.capability, args.providers, configPath);
+		return text(describeCapabilities(workers).find((l) => l.startsWith(`${args.capability}:`)) ?? `${args.capability}: [${workers.capabilities[args.capability]?.join(", ")}]`);
 	} catch (e) {
 		return error(e);
 	}
@@ -5636,7 +5743,7 @@ async function startShim() {
 		"",
 		"Recency boost optionally down-weights older results (recency_boost=90 means scores halve every 90 days).",
 		"",
-		"Defaults come from ~/.claude/pai/config.json (search section). Per-call parameters override config defaults.",
+		"Defaults come from ~/.claude/pai/config.yaml (search section). Per-call parameters override config defaults.",
 		"",
 		"Returns ranked snippets with project slug, file path, line range, and score.",
 		"Higher score = more relevant.",
@@ -5898,7 +6005,9 @@ async function startShim() {
 		"  base_url (anthropic protocol) or upstream_url (protocol=openai, runs",
 		"  through the local PAI proxy). A raw key is written to",
 		"  ~/.claude/pai/keys/<name> (mode 0600); only the path lands in the config.",
-		"  engine=codex runs the Codex CLI instead of Claude Code.",
+		"  engine=codex runs the Codex CLI instead of Claude Code; engine=image POSTs",
+		"  straight to {base_url}/images/generations instead of spawning a process —",
+		"  pair it with a worker_capability image preference to run it.",
 		"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)."
@@ -5917,7 +6026,11 @@ async function startShim() {
 		base_url: string().optional().describe("Anthropic-compatible API base URL (add)."),
 		upstream_url: string().optional().describe("Chat Completions base URL (add, protocol=openai)."),
 		protocol: _enum(["anthropic", "openai"]).optional().describe("Protocol (add). Default: anthropic."),
-		engine: _enum(["claude", "codex"]).optional().describe("Runner engine (add). Default: claude."),
+		engine: _enum([
+			"claude",
+			"codex",
+			"image"
+		]).optional().describe("Runner engine (add). Default: claude."),
 		context_window: number().int().positive().optional().describe("Context window for the meter (add). No default: unset hides the meter unless the init event announces one."),
 		key_file: string().optional().describe("File holding the API token, 0600 (add)."),
 		key: string().optional().describe("Raw API token (add) — parked in ~/.claude/pai/keys/<name>."),
@@ -6086,14 +6199,34 @@ async function startShim() {
 		"",
 		"action=get (default): the model ids of one provider (provider given,",
 		"default: the active one) or of every provider when provider is omitted.",
-		"action=set needs model; capability picks which one (default, fast, image)."
+		"action=set needs model; capability picks which one — the set is open",
+		`(well-known: ${MODEL_CAPABILITIES.join(", ")}; any other ^[a-z][a-z0-9-]*$ name works too).`
 	].join("\n"), {
 		action: _enum(["get", "set"]).optional().describe("Default: get."),
 		provider: string().optional().describe("Provider to read or change (default: the active one)."),
-		capability: _enum(MODEL_CAPABILITIES).optional().describe("Which model capability to set (default, fast, image). Default: default."),
+		capability: string().regex(CAPABILITY_NAME_RE).optional().describe(`Which model capability to set (open set; well-known: ${MODEL_CAPABILITIES.join(", ")}). Default: default.`),
 		slot: _enum(["default", "fast"]).optional().describe("Deprecated pre-capability alias of `capability`."),
 		model: string().optional().describe("The new model id (required for set).")
 	}, async (args) => workerModel(args));
+	server.tool("worker_capability", [
+		"Which provider(s) serve a capability across the whole config (the",
+		"config behind `pai worker capability`) — distinct from worker_model,",
+		"which sets a model id on one provider's own table.",
+		"",
+		"action=list (default): every preference and what it resolves to now.",
+		"action=set needs capability, providers (first usable one wins).",
+		"action=unset needs capability. E.g. set capability=image,",
+		"providers=[\"pictures\"] to run worker_run with capability=image against",
+		"an engine=image provider named \"pictures\"."
+	].join("\n"), {
+		action: _enum([
+			"list",
+			"set",
+			"unset"
+		]).optional().describe("Default: list."),
+		capability: string().optional().describe("Capability name, e.g. image (required for set, unset)."),
+		providers: array(string()).optional().describe("Preference list, first usable one wins (required for set).")
+	}, async (args) => workerCapability(args));
 	server.tool("worker_run", [
 		"Start a worker (or a chain) and return its id immediately.",
 		"",
@@ -6101,8 +6234,9 @@ async function startShim() {
 		"exactly one of prompt/specPath (the -p value, or a file/stdin '-' to read",
 		"it from — avoids quoting long prompts through JSON-RPC). 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."
+		"(default: active); capability=image runs an engine=image provider",
+		"directly instead, writing a PNG rather than spawning claude.",
+		"Check on it with worker_ps / worker_replay, talk to it with worker_say."
 	].join("\n"), workerRunShape, async (args) => {
 		try {
 			const cwd = args.cwd ?? process.cwd();
@@ -6139,6 +6273,7 @@ async function startShim() {
 				});
 				return runWorker({
 					className: args.class,
+					capabilityFlag: args.capability,
 					label: args.label,
 					noPane: false,
 					mcpFlag: args.mcp,
@@ -6283,6 +6418,34 @@ async function startShim() {
 			return workerError(e);
 		}
 	});
+	server.tool("config_list", [
+		"Print the main PAI config as YAML (or JSON). Secrets (key/token/secret/",
+		"password fields, and postgres.connectionString) are always masked to",
+		"****<last4>. Defaults to only what the file explicitly sets; all=true",
+		"includes every built-in default value too."
+	].join("\n"), {
+		all: boolean().optional().describe("Include every default value, not just what the file sets. Default: false."),
+		json: boolean().optional().describe("Return JSON instead of YAML. Default: false.")
+	}, async (args) => configList(args));
+	server.tool("config_get", [
+		"Read one config value by dotted path (e.g. search.recencyBoostDays).",
+		"Resolves against the defaults-merged effective config, so a value never",
+		"set in the file still answers with its built-in default. Masked",
+		"(****<last4>) if the path looks like a secret."
+	].join("\n"), { path: string().min(1).describe("Dotted config path, e.g. search.recencyBoostDays.") }, async (args) => configGet(args));
+	server.tool("config_set", [
+		"Write one config value by dotted path. WRITES THE FILE: creates",
+		"config.yaml on first use (converting config.json if it exists), then",
+		"writes comment-preserving. Value parsing: true/false, null, numbers,",
+		"[...] / {...} as JSON, else a plain string. Refuses an unknown",
+		"top-level key or a value whose type disagrees with the built-in",
+		"default unless force=true."
+	].join("\n"), {
+		path: string().min(1).describe("Dotted config path, e.g. search.recencyBoostDays."),
+		value: string().describe("Value to set, as a string (parsed per the rules above)."),
+		force: boolean().optional().describe("Allow an unknown top-level key or a type mismatch. Default: false.")
+	}, async (args) => configSet(args));
+	server.tool("config_unset", ["Remove one config value by dotted path, WRITING THE FILE — reverting it", "to the built-in default. No-ops when the path was never explicitly set."].join("\n"), { path: string().min(1).describe("Dotted config path, e.g. search.recencyBoostDays.") }, async (args) => configUnset(args));
 	const transport = new StdioServerTransport();
 	await server.connect(transport);
 }