@tekmidian/pai 0.49.0 → 0.51.0

package/dist/hooks/worker-supervision.mjs

@@ -2029,7 +2029,7 @@ var require_YAMLMap = __commonJS({
       }
       return void 0;
     }
-    var YAMLMap = class extends Collection.Collection {
+    var YAMLMap2 = class extends Collection.Collection {
       static get tagName() {
         return "tag:yaml.org,2002:map";
       }
@@ -2146,7 +2146,7 @@ var require_YAMLMap = __commonJS({
         });
       }
     };
-    exports.YAMLMap = YAMLMap;
+    exports.YAMLMap = YAMLMap2;
     exports.findPair = findPair;
   }
 });
@@ -2156,18 +2156,18 @@ var require_map = __commonJS({
   "node_modules/yaml/dist/schema/common/map.js"(exports) {
     "use strict";
     var identity = require_identity();
-    var YAMLMap = require_YAMLMap();
+    var YAMLMap2 = require_YAMLMap();
     var map = {
       collection: "map",
       default: true,
-      nodeClass: YAMLMap.YAMLMap,
+      nodeClass: YAMLMap2.YAMLMap,
       tag: "tag:yaml.org,2002:map",
       resolve(map2, onError) {
         if (!identity.isMap(map2))
           onError("Expected a mapping for this tag");
         return map2;
       },
-      createNode: (schema, obj, ctx) => YAMLMap.YAMLMap.from(schema, obj, ctx)
+      createNode: (schema, obj, ctx) => YAMLMap2.YAMLMap.from(schema, obj, ctx)
     };
     exports.map = map;
   }
@@ -2735,17 +2735,17 @@ var require_omap = __commonJS({
     "use strict";
     var identity = require_identity();
     var toJS = require_toJS();
-    var YAMLMap = require_YAMLMap();
+    var YAMLMap2 = require_YAMLMap();
     var YAMLSeq = require_YAMLSeq();
     var pairs = require_pairs();
     var YAMLOMap = class _YAMLOMap extends YAMLSeq.YAMLSeq {
       constructor() {
         super();
-        this.add = YAMLMap.YAMLMap.prototype.add.bind(this);
-        this.delete = YAMLMap.YAMLMap.prototype.delete.bind(this);
-        this.get = YAMLMap.YAMLMap.prototype.get.bind(this);
-        this.has = YAMLMap.YAMLMap.prototype.has.bind(this);
-        this.set = YAMLMap.YAMLMap.prototype.set.bind(this);
+        this.add = YAMLMap2.YAMLMap.prototype.add.bind(this);
+        this.delete = YAMLMap2.YAMLMap.prototype.delete.bind(this);
+        this.get = YAMLMap2.YAMLMap.prototype.get.bind(this);
+        this.has = YAMLMap2.YAMLMap.prototype.has.bind(this);
+        this.set = YAMLMap2.YAMLMap.prototype.set.bind(this);
         this.tag = _YAMLOMap.tag;
       }
       /**
@@ -2973,8 +2973,8 @@ var require_set = __commonJS({
     "use strict";
     var identity = require_identity();
     var Pair = require_Pair();
-    var YAMLMap = require_YAMLMap();
-    var YAMLSet = class _YAMLSet extends YAMLMap.YAMLMap {
+    var YAMLMap2 = require_YAMLMap();
+    var YAMLSet = class _YAMLSet extends YAMLMap2.YAMLMap {
       constructor(schema) {
         super(schema);
         this.tag = _YAMLSet.tag;
@@ -2987,7 +2987,7 @@ var require_set = __commonJS({
           pair = new Pair.Pair(key.key, null);
         else
           pair = new Pair.Pair(key, null);
-        const prev = YAMLMap.findPair(this.items, pair.key);
+        const prev = YAMLMap2.findPair(this.items, pair.key);
         if (!prev)
           this.items.push(pair);
       }
@@ -2996,13 +2996,13 @@ var require_set = __commonJS({
        * Otherwise, returns the value of that Pair's key.
        */
       get(key, keepPair) {
-        const pair = YAMLMap.findPair(this.items, key);
+        const pair = YAMLMap2.findPair(this.items, key);
         return !keepPair && identity.isPair(pair) ? identity.isScalar(pair.key) ? pair.key.value : pair.key : pair;
       }
       set(key, value) {
         if (typeof value !== "boolean")
           throw new Error(`Expected boolean value for set(key, value) in a YAML set, not ${typeof value}`);
-        const prev = YAMLMap.findPair(this.items, key);
+        const prev = YAMLMap2.findPair(this.items, key);
         if (prev && !value) {
           this.items.splice(this.items.indexOf(prev), 1);
         } else if (!prev && value) {
@@ -3409,7 +3409,7 @@ var require_Document = __commonJS({
     var applyReviver = require_applyReviver();
     var createNode = require_createNode();
     var directives = require_directives();
-    var Document2 = class _Document {
+    var Document4 = class _Document {
       constructor(value, replacer, options) {
         this.commentBefore = null;
         this.comment = null;
@@ -3699,7 +3699,7 @@ var require_Document = __commonJS({
         return true;
       throw new Error("Expected a YAML collection as document contents");
     }
-    exports.Document = Document2;
+    exports.Document = Document4;
   }
 });
 
@@ -3983,14 +3983,14 @@ var require_resolve_block_map = __commonJS({
   "node_modules/yaml/dist/compose/resolve-block-map.js"(exports) {
     "use strict";
     var Pair = require_Pair();
-    var YAMLMap = require_YAMLMap();
+    var YAMLMap2 = require_YAMLMap();
     var resolveProps = require_resolve_props();
     var utilContainsNewline = require_util_contains_newline();
     var utilFlowIndentCheck = require_util_flow_indent_check();
     var utilMapIncludes = require_util_map_includes();
     var startColMsg = "All mapping items must start at the same column";
     function resolveBlockMap({ composeNode, composeEmptyNode }, ctx, bm, onError, tag) {
-      const NodeClass = tag?.nodeClass ?? YAMLMap.YAMLMap;
+      const NodeClass = tag?.nodeClass ?? YAMLMap2.YAMLMap;
       const map = new NodeClass(ctx.schema);
       if (ctx.atRoot)
         ctx.atRoot = false;
@@ -4186,7 +4186,7 @@ var require_resolve_flow_collection = __commonJS({
     "use strict";
     var identity = require_identity();
     var Pair = require_Pair();
-    var YAMLMap = require_YAMLMap();
+    var YAMLMap2 = require_YAMLMap();
     var YAMLSeq = require_YAMLSeq();
     var resolveEnd = require_resolve_end();
     var resolveProps = require_resolve_props();
@@ -4197,7 +4197,7 @@ var require_resolve_flow_collection = __commonJS({
     function resolveFlowCollection({ composeNode, composeEmptyNode }, ctx, fc, onError, tag) {
       const isMap = fc.start.source === "{";
       const fcName = isMap ? "flow map" : "flow sequence";
-      const NodeClass = tag?.nodeClass ?? (isMap ? YAMLMap.YAMLMap : YAMLSeq.YAMLSeq);
+      const NodeClass = tag?.nodeClass ?? (isMap ? YAMLMap2.YAMLMap : YAMLSeq.YAMLSeq);
       const coll = new NodeClass(ctx.schema);
       coll.flow = true;
       const atRoot = ctx.atRoot;
@@ -4334,7 +4334,7 @@ var require_resolve_flow_collection = __commonJS({
               onError(keyStart, "DUPLICATE_KEY", "Map keys must be unique");
             map.items.push(pair);
           } else {
-            const map = new YAMLMap.YAMLMap(ctx.schema);
+            const map = new YAMLMap2.YAMLMap(ctx.schema);
             map.flow = true;
             map.items.push(pair);
             const endRange = (valueNode ?? keyNode).range;
@@ -4380,7 +4380,7 @@ var require_compose_collection = __commonJS({
     "use strict";
     var identity = require_identity();
     var Scalar2 = require_Scalar();
-    var YAMLMap = require_YAMLMap();
+    var YAMLMap2 = require_YAMLMap();
     var YAMLSeq = require_YAMLSeq();
     var resolveBlockMap = require_resolve_block_map();
     var resolveBlockSeq = require_resolve_block_seq();
@@ -4408,7 +4408,7 @@ var require_compose_collection = __commonJS({
         }
       }
       const expType = token.type === "block-map" ? "map" : token.type === "block-seq" ? "seq" : token.start.source === "{" ? "map" : "seq";
-      if (!tagToken || !tagName || tagName === "!" || tagName === YAMLMap.YAMLMap.tagName && expType === "map" || tagName === YAMLSeq.YAMLSeq.tagName && expType === "seq") {
+      if (!tagToken || !tagName || tagName === "!" || tagName === YAMLMap2.YAMLMap.tagName && expType === "map" || tagName === YAMLSeq.YAMLSeq.tagName && expType === "seq") {
         return resolveCollection(CN, ctx, token, onError, tagName);
       }
       let tag = ctx.schema.tags.find((t) => t.tag === tagName && t.collection === expType);
@@ -5064,13 +5064,13 @@ var require_compose_node = __commonJS({
 var require_compose_doc = __commonJS({
   "node_modules/yaml/dist/compose/compose-doc.js"(exports) {
     "use strict";
-    var Document2 = require_Document();
+    var Document4 = require_Document();
     var composeNode = require_compose_node();
     var resolveEnd = require_resolve_end();
     var resolveProps = require_resolve_props();
     function composeDoc(options, directives, { offset, start, value, end }, onError) {
       const opts = Object.assign({ _directives: directives }, options);
-      const doc = new Document2.Document(void 0, opts);
+      const doc = new Document4.Document(void 0, opts);
       const ctx = {
         atKey: false,
         atRoot: true,
@@ -5109,7 +5109,7 @@ var require_composer = __commonJS({
     "use strict";
     var node_process = __require("process");
     var directives = require_directives();
-    var Document2 = require_Document();
+    var Document4 = require_Document();
     var errors = require_errors();
     var identity = require_identity();
     var composeDoc = require_compose_doc();
@@ -5298,7 +5298,7 @@ ${end.comment}` : end.comment;
           this.doc = null;
         } else if (forceDoc) {
           const opts = Object.assign({ _directives: this.directives }, this.options);
-          const doc = new Document2.Document(void 0, opts);
+          const doc = new Document4.Document(void 0, opts);
           if (this.atDirectives)
             this.onError(endOffset, "MISSING_CHAR", "Missing directives-end indicator line");
           doc.range = [0, endOffset, endOffset];
@@ -7220,7 +7220,7 @@ var require_public_api = __commonJS({
   "node_modules/yaml/dist/public-api.js"(exports) {
     "use strict";
     var composer = require_composer();
-    var Document2 = require_Document();
+    var Document4 = require_Document();
     var errors = require_errors();
     var log = require_log();
     var identity = require_identity();
@@ -7245,7 +7245,7 @@ var require_public_api = __commonJS({
         return docs;
       return Object.assign([], { empty: true }, composer$1.streamInfo());
     }
-    function parseDocument2(source, options = {}) {
+    function parseDocument4(source, options = {}) {
       const { lineCounter: lineCounter2, prettyErrors } = parseOptions(options);
       const parser$1 = new parser.Parser(lineCounter2?.addNewLine);
       const composer$1 = new composer.Composer(options);
@@ -7271,7 +7271,7 @@ var require_public_api = __commonJS({
       } else if (options === void 0 && reviver && typeof reviver === "object") {
         options = reviver;
       }
-      const doc = parseDocument2(src, options);
+      const doc = parseDocument4(src, options);
       if (!doc)
         return null;
       doc.warnings.forEach((warning) => log.warn(doc.options.logLevel, warning));
@@ -7303,11 +7303,11 @@ var require_public_api = __commonJS({
       }
       if (identity.isDocument(value) && !_replacer)
         return value.toString(options);
-      return new Document2.Document(value, _replacer, options).toString(options);
+      return new Document4.Document(value, _replacer, options).toString(options);
     }
     exports.parse = parse;
     exports.parseAllDocuments = parseAllDocuments;
-    exports.parseDocument = parseDocument2;
+    exports.parseDocument = parseDocument4;
     exports.stringify = stringify;
   }
 });
@@ -7317,14 +7317,14 @@ var require_dist = __commonJS({
   "node_modules/yaml/dist/index.js"(exports) {
     "use strict";
     var composer = require_composer();
-    var Document2 = require_Document();
+    var Document4 = require_Document();
     var Schema = require_Schema();
     var errors = require_errors();
     var Alias = require_Alias();
     var identity = require_identity();
     var Pair = require_Pair();
     var Scalar2 = require_Scalar();
-    var YAMLMap = require_YAMLMap();
+    var YAMLMap2 = require_YAMLMap();
     var YAMLSeq = require_YAMLSeq();
     var cst = require_cst();
     var lexer = require_lexer();
@@ -7333,7 +7333,7 @@ var require_dist = __commonJS({
     var publicApi = require_public_api();
     var visit = require_visit();
     exports.Composer = composer.Composer;
-    exports.Document = Document2.Document;
+    exports.Document = Document4.Document;
     exports.Schema = Schema.Schema;
     exports.YAMLError = errors.YAMLError;
     exports.YAMLParseError = errors.YAMLParseError;
@@ -7349,7 +7349,7 @@ var require_dist = __commonJS({
     exports.isSeq = identity.isSeq;
     exports.Pair = Pair.Pair;
     exports.Scalar = Scalar2.Scalar;
-    exports.YAMLMap = YAMLMap.YAMLMap;
+    exports.YAMLMap = YAMLMap2.YAMLMap;
     exports.YAMLSeq = YAMLSeq.YAMLSeq;
     exports.CST = cst;
     exports.Lexer = lexer.Lexer;
@@ -7365,8 +7365,8 @@ var require_dist = __commonJS({
 });
 
 // src/hooks/ts/user-prompt/worker-supervision.ts
-import { existsSync as existsSync4, readFileSync as readFileSync4, writeFileSync as writeFileSync3 } from "node:fs";
-import { join as join5 } from "node:path";
+import { existsSync as existsSync5, readFileSync as readFileSync5, writeFileSync as writeFileSync3 } from "node:fs";
+import { join as join6 } from "node:path";
 
 // src/hooks/ts/lib/worker-session.ts
 function isWorkerSession(env = process.env) {
@@ -7375,43 +7375,11 @@ function isWorkerSession(env = process.env) {
 
 // src/workers/config.ts
 import { homedir as homedir4 } from "node:os";
-import { join as join4 } from "node:path";
-
-// src/config/json-store.ts
-import {
-  existsSync,
-  readFileSync,
-  writeFileSync,
-  copyFileSync,
-  renameSync,
-  unlinkSync,
-  mkdirSync
-} from "node:fs";
-function readJsonStrict(path, label = path) {
-  if (!existsSync(path)) return {};
-  let raw;
-  try {
-    raw = readFileSync(path, "utf8");
-  } catch (e) {
-    throw new Error(
-      `Could not read ${label}: ${e instanceof Error ? e.message : String(e)}
-Refusing to continue \u2014 writing now would replace its contents with ours alone.`
-    );
-  }
-  try {
-    return JSON.parse(raw);
-  } catch (e) {
-    throw new Error(
-      `${label} exists but is not valid JSON: ${e instanceof Error ? e.message : String(e)}
-Refusing to continue \u2014 overwriting it would destroy whatever it holds.
-Repair the file, or move it aside and re-run this command.`
-    );
-  }
-}
+import { join as join5 } from "node:path";
 
 // src/daemon/config.ts
 import { homedir as homedir2, userInfo } from "node:os";
-import { join as join2, dirname as dirname2 } from "node:path";
+import { join as join3, dirname as dirname3 } from "node:path";
 
 // src/notifications/types.ts
 var DEFAULT_ROUTING = {
@@ -7465,7 +7433,7 @@ function paiSocketPath() {
 }
 
 // src/config/pai-home.ts
-import { existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync as readFileSync2, copyFileSync as copyFileSync2, renameSync as renameSync2, readdirSync, statSync } from "node:fs";
+import { existsSync, mkdirSync, readFileSync, copyFileSync, renameSync, readdirSync, statSync } from "node:fs";
 import { homedir } from "node:os";
 import { join, dirname } from "node:path";
 function paiHomeDir() {
@@ -7476,9 +7444,9 @@ function paiHomePath(...segments) {
 }
 var noticesPrinted = /* @__PURE__ */ new Set();
 function resolvePaiFile(newPath, oldCandidates, migrateHint) {
-  if (existsSync2(newPath)) return newPath;
+  if (existsSync(newPath)) return newPath;
   for (const old of oldCandidates) {
-    if (existsSync2(old)) {
+    if (existsSync(old)) {
       if (!noticesPrinted.has(newPath)) {
         noticesPrinted.add(newPath);
         process.stderr.write(
@@ -7492,6 +7460,74 @@ function resolvePaiFile(newPath, oldCandidates, migrateHint) {
   return newPath;
 }
 
+// src/config/main-config.ts
+var import_yaml2 = __toESM(require_dist(), 1);
+import { existsSync as existsSync3, readFileSync as readFileSync3, renameSync as renameSync3 } from "node:fs";
+import { dirname as dirname2, join as join2, basename } from "node:path";
+
+// src/config/json-store.ts
+import {
+  existsSync as existsSync2,
+  readFileSync as readFileSync2,
+  writeFileSync,
+  copyFileSync as copyFileSync2,
+  renameSync as renameSync2,
+  unlinkSync,
+  mkdirSync as mkdirSync2
+} from "node:fs";
+function readJsonStrict(path, label = path) {
+  if (!existsSync2(path)) return {};
+  let raw;
+  try {
+    raw = readFileSync2(path, "utf8");
+  } catch (e) {
+    throw new Error(
+      `Could not read ${label}: ${e instanceof Error ? e.message : String(e)}
+Refusing to continue \u2014 writing now would replace its contents with ours alone.`
+    );
+  }
+  try {
+    return JSON.parse(raw);
+  } catch (e) {
+    throw new Error(
+      `${label} exists but is not valid JSON: ${e instanceof Error ? e.message : String(e)}
+Refusing to continue \u2014 overwriting it would destroy whatever it holds.
+Repair the file, or move it aside and re-run this command.`
+    );
+  }
+}
+
+// src/config/yaml-store.ts
+var import_yaml = __toESM(require_dist(), 1);
+
+// src/config/main-config.ts
+var MainConfigError = class extends Error {
+};
+function yamlSiblingPath(jsonPath) {
+  return join2(dirname2(jsonPath), "config.yaml");
+}
+function readDualFormatConfigRaw(jsonPath, yamlPath = yamlSiblingPath(jsonPath)) {
+  if (existsSync3(yamlPath)) {
+    let text;
+    try {
+      text = readFileSync3(yamlPath, "utf8");
+    } catch (e) {
+      throw new MainConfigError(`Could not read ${yamlPath}: ${e instanceof Error ? e.message : String(e)}`);
+    }
+    const doc = (0, import_yaml2.parseDocument)(text);
+    if (doc.errors.length) {
+      throw new MainConfigError(`${yamlPath}: ${doc.errors[0].message}`);
+    }
+    const parsed = doc.toJS();
+    if (parsed === null || parsed === void 0) return {};
+    if (typeof parsed !== "object" || Array.isArray(parsed)) {
+      throw new MainConfigError(`${yamlPath}: top level must be a mapping`);
+    }
+    return parsed;
+  }
+  return readJsonStrict(jsonPath, jsonPath);
+}
+
 // src/daemon/config.ts
 function perUserDbName() {
   const username = userInfo().username;
@@ -7502,6 +7538,14 @@ function perUserConnectionString() {
   const db = perUserDbName();
   return `postgresql://pai:pai@localhost:5432/${db}`;
 }
+var DEFAULT_SESSIONS_CACHE_KEEPALIVE = {
+  enabled: false,
+  idleMinutes: 50,
+  maxBeats: 6,
+  activeHours: "08:00-22:00",
+  minContextTokens: 2e4,
+  prompt: "keepalive"
+};
 var DEFAULTS = {
   socketPath: paiSocketPath(),
   indexIntervalSecs: 300,
@@ -7520,6 +7564,7 @@ var DEFAULTS = {
   // Deliberately empty. An install must not guess who the user is: a wrong
   // guess here is an address that can be mailed without review.
   identity: { selfEmails: [] },
+  sessions: { cacheKeepalive: { ...DEFAULT_SESSIONS_CACHE_KEEPALIVE } },
   search: {
     mode: "keyword",
     rerank: true,
@@ -7529,30 +7574,37 @@ var DEFAULTS = {
   }
 };
 var NEW_CONFIG_FILE = paiHomePath("config.json");
-var OLD_CONFIG_FILE = join2(homedir2(), ".claude", "pai.json");
-var LEGACY_CONFIG_FILE = join2(homedir2(), ".config", "pai", "config.json");
+var OLD_CONFIG_FILE = join3(homedir2(), ".claude", "pai.json");
+var LEGACY_CONFIG_FILE = join3(homedir2(), ".config", "pai", "config.json");
 function paiConfigFilePath() {
   const override = process.env.PAI_CONFIG_FILE;
   if (override) return override;
   return resolvePaiFile(NEW_CONFIG_FILE, [OLD_CONFIG_FILE, LEGACY_CONFIG_FILE], "pai config migrate");
 }
 var CONFIG_FILE = paiConfigFilePath();
-var CONFIG_DIR = dirname2(CONFIG_FILE);
+var CONFIG_DIR = dirname3(CONFIG_FILE);
+function readMainConfigRaw(path = CONFIG_FILE) {
+  try {
+    return readDualFormatConfigRaw(path);
+  } catch (e) {
+    throw e instanceof MainConfigError ? new Error(e.message) : e;
+  }
+}
 
 // src/workers/workers-config.ts
-var import_yaml = __toESM(require_dist(), 1);
-import { existsSync as existsSync3, readFileSync as readFileSync3, writeFileSync as writeFileSync2, copyFileSync as copyFileSync3, renameSync as renameSync3, unlinkSync as unlinkSync2, mkdirSync as mkdirSync3, chmodSync } from "node:fs";
+var import_yaml3 = __toESM(require_dist(), 1);
+import { existsSync as existsSync4, readFileSync as readFileSync4, writeFileSync as writeFileSync2, chmodSync } from "node:fs";
 import { homedir as homedir3 } from "node:os";
-import { dirname as dirname3, join as join3 } from "node:path";
+import { dirname as dirname4, join as join4 } from "node:path";
 var WORKERS_YAML_FILENAME = "workers.yaml";
 function defaultWorkersYamlPath() {
   return paiHomePath(WORKERS_YAML_FILENAME);
 }
 function oldWorkersYamlPath() {
-  return join3(homedir3(), ".claude", WORKERS_YAML_FILENAME);
+  return join4(homedir3(), ".claude", WORKERS_YAML_FILENAME);
 }
 function legacyWorkersYamlPath() {
-  return join3(homedir3(), ".config", "pai", WORKERS_YAML_FILENAME);
+  return join4(homedir3(), ".config", "pai", WORKERS_YAML_FILENAME);
 }
 function workersYamlPath() {
   const override = process.env.PAI_WORKERS_YAML;
@@ -7657,7 +7709,8 @@ function parseWorkersYamlDocument(doc, yamlPath) {
       const provider = typeof target === "string" ? target.split("/")[0] : target.provider;
       const alias = typeof target === "string" ? target.split("/")[1] : void 0;
       if (provider !== void 0) {
-        if (provider !== ANTHROPIC_NATIVE && !providers[provider]) {
+        const native = provider === ANTHROPIC_NATIVE;
+        if (!native && !providers[provider]) {
           withLine(
             doc,
             ["classes", cls],
@@ -7667,19 +7720,36 @@ function parseWorkersYamlDocument(doc, yamlPath) {
             )
           );
         }
-        if (alias !== void 0 && !isModelCapability(alias)) {
-          withLine(
-            doc,
-            ["classes", cls],
-            yamlPath,
-            new WorkersConfigError(
-              `classes.${cls}: unknown model role "${alias}" (from: ${MODEL_CAPABILITIES.join(", ")})`
-            )
-          );
+        if (alias !== void 0) {
+          if (!isModelCapability(alias)) {
+            withLine(
+              doc,
+              ["classes", cls],
+              yamlPath,
+              new WorkersConfigError(
+                `classes.${cls}: "${alias}" is not a valid capability name (must match ^[a-z][a-z0-9-]*$)`
+              )
+            );
+          } else if (!native && providers[provider] && alias !== "default" && !providers[provider].models[alias]) {
+            withLine(
+              doc,
+              ["classes", cls],
+              yamlPath,
+              new WorkersConfigError(`classes.${cls}: provider "${provider}" has no "${alias}" model configured`)
+            );
+          }
         }
       }
     }
   }
+  let capabilities = {};
+  if (r.capabilities !== void 0) {
+    try {
+      capabilities = parseCapabilitiesValue(r.capabilities, "capabilities");
+    } catch (e) {
+      withLine(doc, ["capabilities"], yamlPath, e);
+    }
+  }
   let mcpSets = {};
   if (r.mcp_sets !== void 0) {
     try {
@@ -7701,19 +7771,19 @@ function parseWorkersYamlDocument(doc, yamlPath) {
       )
     );
   }
-  return { active, providers, classes, mcpSets, nativeModels };
+  return { active, providers, classes, capabilities, mcpSets, nativeModels };
 }
 function readWorkersYaml(yamlPath) {
-  if (!existsSync3(yamlPath)) return null;
+  if (!existsSync4(yamlPath)) return null;
   let text;
   try {
-    text = readFileSync3(yamlPath, "utf8");
+    text = readFileSync4(yamlPath, "utf8");
   } catch (e) {
     throw new WorkersConfigError(
       `Could not read ${yamlPath}: ${e instanceof Error ? e.message : String(e)}`
     );
   }
-  const doc = (0, import_yaml.parseDocument)(text);
+  const doc = (0, import_yaml3.parseDocument)(text);
   if (doc.errors.length) {
     throw new WorkersConfigError(`${yamlPath}: ${doc.errors[0].message}`);
   }
@@ -7737,8 +7807,9 @@ var PROVIDER_TAGS = [
   "reasoning"
 ];
 var MODEL_CAPABILITIES = ["default", "fast", "image"];
+var CAPABILITY_NAME_RE = /^[a-z][a-z0-9-]*$/;
 function isModelCapability(v) {
-  return MODEL_CAPABILITIES.includes(v);
+  return CAPABILITY_NAME_RE.test(v);
 }
 var DEFAULT_LOG_DIR = "~/.claude/logs/workers";
 var DEFAULT_PANE = {
@@ -7766,6 +7837,7 @@ function defaultWorkersConfig() {
     providers: {},
     classes: {},
     mcpSets: { ...DEFAULT_MCP_SETS },
+    capabilities: {},
     pane: { ...DEFAULT_PANE },
     logDir: DEFAULT_LOG_DIR,
     routing: { ...DEFAULT_ROUTING2, order: [] },
@@ -7793,7 +7865,10 @@ function parseModelsBlock(pathPrefix, modelsRaw) {
   for (const key of Object.keys(m)) {
     if (key === "default") continue;
     if (!isModelCapability(key)) {
-      bad(`${pathPrefix}.${key}`, `"${key}" is not a model capability (from: ${MODEL_CAPABILITIES.join(", ")})`);
+      bad(
+        `${pathPrefix}.${key}`,
+        `"${key}" is not a valid capability name (must match ^[a-z][a-z0-9-]*$; well-known: ${MODEL_CAPABILITIES.join(", ")}, but any name in that shape is accepted)`
+      );
     }
     const id = str(m[key]);
     if (!id) bad(`${pathPrefix}.${key}`, "must be a non-empty model id");
@@ -7815,8 +7890,8 @@ function parseProvider(name, raw) {
     bad(`.providers.${name}.protocol`, `"${str(p.protocol)}" is neither "anthropic" nor "openai"`);
   }
   const engine = p.engine === void 0 ? "claude" : str(p.engine);
-  if (engine !== "claude" && engine !== "codex") {
-    bad(`.providers.${name}.engine`, `"${str(p.engine)}" is neither "claude" nor "codex"`);
+  if (engine !== "claude" && engine !== "codex" && engine !== "image") {
+    bad(`.providers.${name}.engine`, `"${str(p.engine)}" is none of "claude", "codex", "image"`);
   }
   const baseUrl = str(p.baseUrl);
   if (!baseUrl && protocol !== "openai") bad(`.providers.${name}.baseUrl`, "is required");
@@ -8024,6 +8099,23 @@ function parseMcpSetsValue(raw, badPathPrefix) {
   }
   return mcpSets;
 }
+function parseCapabilitiesValue(raw, badPathPrefix) {
+  const capabilities = {};
+  if (raw === void 0) return capabilities;
+  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) {
+    bad(badPathPrefix, "must be an object of capability name \u2192 [provider names]");
+  }
+  for (const [name, order] of Object.entries(raw)) {
+    if (!isModelCapability(name)) {
+      bad(`${badPathPrefix}.${name}`, `"${name}" is not a valid capability name (must match ^[a-z][a-z0-9-]*$)`);
+    }
+    if (!Array.isArray(order) || order.length === 0 || order.some((x) => typeof x !== "string" || !x)) {
+      bad(`${badPathPrefix}.${name}`, "must be a non-empty array of provider names");
+    }
+    capabilities[name] = order;
+  }
+  return capabilities;
+}
 function parseWorkersConfig(raw) {
   const d = defaultWorkersConfig();
   if (raw === void 0 || raw === null) return d;
@@ -8043,6 +8135,7 @@ function parseWorkersConfig(raw) {
   const classesRaw = w.classes !== void 0 ? w.classes : w.roles;
   const classes = parseClassesValue(classesRaw, w.classes !== void 0 ? ".classes" : ".roles");
   const mcpSets = parseMcpSetsValue(w.mcpSets, ".mcpSets");
+  const capabilities = parseCapabilitiesValue(w.capabilities, ".capabilities");
   let pane = { ...DEFAULT_PANE };
   if (w.pane !== void 0) {
     if (typeof w.pane !== "object" || w.pane === null) bad(".pane", "must be an object");
@@ -8152,6 +8245,7 @@ function parseWorkersConfig(raw) {
     providers,
     classes,
     mcpSets,
+    capabilities,
     pane,
     logDir: str(w.logDir) || d.logDir,
     routing,
@@ -8162,7 +8256,7 @@ function parseWorkersConfig(raw) {
   };
 }
 function readWorkersSection(path = CONFIG_FILE) {
-  const raw = readJsonStrict(path, path);
+  const raw = readMainConfigRaw(path);
   const workers = parseWorkersConfig(raw.workers);
   const yaml = readWorkersYaml(workersYamlPath());
   if (yaml) {
@@ -8170,12 +8264,13 @@ function readWorkersSection(path = CONFIG_FILE) {
     workers.providers = yaml.data.providers;
     workers.classes = yaml.data.classes;
     workers.mcpSets = yaml.data.mcpSets;
+    workers.capabilities = yaml.data.capabilities;
     workers.nativeModels = yaml.data.nativeModels;
   }
   return { raw, workers };
 }
 function expandHome(p) {
-  if (p === "~" || p.startsWith("~/")) return join4(homedir4(), p.slice(1));
+  if (p === "~" || p.startsWith("~/")) return join5(homedir4(), p.slice(1));
   return p;
 }
 
@@ -8189,7 +8284,7 @@ function workersLogDir(config) {
 // src/hooks/ts/user-prompt/worker-supervision.ts
 function readCursor(path) {
   try {
-    return Math.max(0, Number(readFileSync4(path, "utf8").trim()) || 0);
+    return Math.max(0, Number(readFileSync5(path, "utf8").trim()) || 0);
   } catch {
     return 0;
   }
@@ -8198,7 +8293,7 @@ function main() {
   if (isWorkerSession()) return;
   let raw = "";
   try {
-    raw = readFileSync4(0, "utf8");
+    raw = readFileSync5(0, "utf8");
   } catch {
     return;
   }
@@ -8217,13 +8312,13 @@ function main() {
   } catch {
     return;
   }
-  const dir = join5(logDir, "supervision");
-  const eventsPath = join5(dir, `${session}.events`);
-  const cursorPath = join5(dir, `${session}.cursor`);
-  if (!existsSync4(eventsPath)) return;
+  const dir = join6(logDir, "supervision");
+  const eventsPath = join6(dir, `${session}.events`);
+  const cursorPath = join6(dir, `${session}.cursor`);
+  if (!existsSync5(eventsPath)) return;
   let lines;
   try {
-    lines = readFileSync4(eventsPath, "utf8").split("\n").filter((l) => l.trim());
+    lines = readFileSync5(eventsPath, "utf8").split("\n").filter((l) => l.trim());
   } catch {
     return;
   }
@@ -8232,7 +8327,7 @@ function main() {
   let pushedIds = /* @__PURE__ */ new Set();
   try {
     pushedIds = new Set(
-      readFileSync4(join5(dir, `${session}.pushed`), "utf8").split("\n").map((l) => l.trim()).filter(Boolean)
+      readFileSync5(join6(dir, `${session}.pushed`), "utf8").split("\n").map((l) => l.trim()).filter(Boolean)
     );
   } catch {
   }