@tekmidian/pai 0.39.0 → 0.40.0

package/dist/hooks/worker-proxy.mjs

@@ -132,6 +132,7 @@ var CONFIG_DIR = join(homedir(), ".config", "pai");
 var CONFIG_FILE = join(CONFIG_DIR, "config.json");
 
 // src/workers/config.ts
+var MODEL_TIERS = ["haiku", "sonnet", "opus"];
 var PROVIDER_TAGS = [
   "code",
   "vision",
@@ -140,6 +141,10 @@ var PROVIDER_TAGS = [
   "fast",
   "reasoning"
 ];
+var MODEL_CAPABILITIES = ["default", "fast", "image"];
+function isModelCapability(v) {
+  return MODEL_CAPABILITIES.includes(v);
+}
 var DEFAULT_LOG_DIR = "~/.claude/logs/workers";
 var DEFAULT_PANE = {
   enabled: true,
@@ -168,7 +173,8 @@ function defaultWorkersConfig() {
     pane: { ...DEFAULT_PANE },
     logDir: DEFAULT_LOG_DIR,
     routing: { ...DEFAULT_ROUTING2, order: [] },
-    tree: { ...DEFAULT_TREE }
+    tree: { ...DEFAULT_TREE },
+    fallback: null
   };
 }
 var WorkersConfigError = class extends Error {
@@ -200,7 +206,35 @@ function parseProvider(name, raw) {
   const m = modelsRaw;
   const defaultModel = str(m.default);
   if (!defaultModel) bad(`.providers.${name}.models.default`, "is required");
-  const fast = m.fast === void 0 ? void 0 : str(m.fast);
+  const models = { default: defaultModel };
+  for (const key of Object.keys(m)) {
+    if (key === "default") continue;
+    if (!isModelCapability(key)) {
+      bad(
+        `.providers.${name}.models.${key}`,
+        `"${key}" is not a model capability (from: ${MODEL_CAPABILITIES.join(", ")})`
+      );
+    }
+    const id = str(m[key]);
+    if (!id) bad(`.providers.${name}.models.${key}`, "must be a non-empty model id");
+    models[key] = id;
+  }
+  let modelTiers;
+  if (p.modelTiers !== void 0) {
+    if (typeof p.modelTiers !== "object" || p.modelTiers === null || Array.isArray(p.modelTiers)) {
+      bad(`.providers.${name}.modelTiers`, "must be an object of model id \u2192 haiku | sonnet | opus");
+    }
+    modelTiers = {};
+    for (const [id, tier] of Object.entries(p.modelTiers)) {
+      if (typeof tier !== "string" || !MODEL_TIERS.includes(tier)) {
+        bad(
+          `.providers.${name}.modelTiers.${id}`,
+          `must be "haiku", "sonnet" or "opus" (got ${JSON.stringify(tier)})`
+        );
+      }
+      modelTiers[id] = tier;
+    }
+  }
   const env = {};
   if (p.env !== void 0) {
     if (typeof p.env !== "object" || p.env === null || Array.isArray(p.env)) {
@@ -250,7 +284,8 @@ function parseProvider(name, raw) {
     protocol,
     baseUrl,
     keyFile,
-    models: fast ? { default: defaultModel, fast } : { default: defaultModel },
+    models,
+    ...modelTiers ? { modelTiers } : {},
     env,
     ...str(p.note) ? { note: str(p.note) } : {},
     ...upstreamUrl ? { upstreamUrl } : {},
@@ -414,6 +449,39 @@ function parseWorkersConfig(raw) {
   const active = w.active === void 0 || w.active === null ? null : str(w.active);
   if (active !== null && active !== "auto" && !(active in providers)) {
   }
+  let fallback = null;
+  if (w.fallback !== void 0 && w.fallback !== null) {
+    if (typeof w.fallback !== "object" || Array.isArray(w.fallback)) {
+      bad(".fallback", "must be an object (written by `pai worker fallback on`)");
+    }
+    const f = w.fallback;
+    const provider = str(f.provider);
+    if (!provider) bad(".fallback.provider", "is required");
+    const savedRaw = f.saved;
+    if (typeof savedRaw !== "object" || savedRaw === null || Array.isArray(savedRaw)) {
+      bad(".fallback.saved", "must be an object");
+    }
+    const s = savedRaw;
+    if (typeof s.env !== "object" || s.env === null || Array.isArray(s.env)) {
+      bad(".fallback.saved.env", "must be an object of env key \u2192 previous value or null");
+    }
+    const env = {};
+    for (const [k, v] of Object.entries(s.env)) {
+      env[k] = v === null || typeof v === "string" ? v : null;
+    }
+    if (s.model !== null && s.model !== void 0 && typeof s.model !== "string") {
+      bad(".fallback.saved.model", "must be a string or null");
+    }
+    fallback = {
+      provider,
+      saved: {
+        env,
+        model: typeof s.model === "string" ? s.model : null,
+        envExisted: s.envExisted === true
+      },
+      on: str(f.on) || (/* @__PURE__ */ new Date(0)).toISOString()
+    };
+  }
   return {
     enabled: w.enabled === void 0 ? d.enabled : w.enabled === true,
     active,
@@ -423,11 +491,12 @@ function parseWorkersConfig(raw) {
     pane,
     logDir: str(w.logDir) || d.logDir,
     routing,
-    tree
+    tree,
+    fallback
   };
 }
-function readWorkersSection() {
-  const raw = readJsonStrict(CONFIG_FILE, "~/.config/pai/config.json");
+function readWorkersSection(path = CONFIG_FILE) {
+  const raw = readJsonStrict(path, "~/.config/pai/config.json");
   return { raw, workers: parseWorkersConfig(raw.workers) };
 }
 function expandHome(p) {