@tekmidian/pai 0.37.0 → 0.38.0

package/dist/hooks/worker-proxy.mjs

@@ -132,6 +132,14 @@ var CONFIG_DIR = join(homedir(), ".config", "pai");
 var CONFIG_FILE = join(CONFIG_DIR, "config.json");
 
 // src/workers/config.ts
+var PROVIDER_TAGS = [
+  "code",
+  "vision",
+  "image-gen",
+  "long-context",
+  "fast",
+  "reasoning"
+];
 var DEFAULT_LOG_DIR = "~/.claude/logs/workers";
 var DEFAULT_PANE = {
   enabled: true,
@@ -148,7 +156,7 @@ function defaultWorkersConfig() {
     enabled: false,
     active: null,
     providers: {},
-    roles: {},
+    classes: {},
     mcpSets: {},
     pane: { ...DEFAULT_PANE },
     logDir: DEFAULT_LOG_DIR,
@@ -211,6 +219,24 @@ function parseProvider(name, raw) {
   if (protocol === "openai" && !upstreamUrl) {
     bad(`.providers.${name}.upstreamUrl`, `is required for protocol "openai" (the Chat Completions base, e.g. "https://api.openai.com/v1")`);
   }
+  const costTier = p.costTier === void 0 ? void 0 : p.costTier;
+  if (costTier !== void 0) {
+    if (typeof costTier !== "number" || !Number.isInteger(costTier) || costTier < 1 || costTier > 5) {
+      bad(`.providers.${name}.costTier`, "must be an integer 1 (cheapest) \u2026 5 (most expensive)");
+    }
+  }
+  let tags;
+  if (p.tags !== void 0) {
+    if (!Array.isArray(p.tags) || p.tags.some((x) => typeof x !== "string")) {
+      bad(`.providers.${name}.tags`, `must be an array of tags from: ${PROVIDER_TAGS.join(", ")}`);
+    }
+    for (const t of p.tags) {
+      if (!PROVIDER_TAGS.includes(t)) {
+        bad(`.providers.${name}.tags`, `"${t}" is not a tag (from: ${PROVIDER_TAGS.join(", ")})`);
+      }
+    }
+    tags = p.tags;
+  }
   return {
     enabled: p.enabled === void 0 ? true : p.enabled === true,
     protocol,
@@ -223,7 +249,9 @@ function parseProvider(name, raw) {
     ...engine !== "claude" ? { engine } : {},
     ...str(p.quotaProbe) ? { quotaProbe: str(p.quotaProbe) } : {},
     ...quotaSkipAt !== void 0 ? { quotaSkipAt } : {},
-    ...contextWindow !== void 0 ? { contextWindow } : {}
+    ...contextWindow !== void 0 ? { contextWindow } : {},
+    ...costTier !== void 0 ? { costTier } : {},
+    ...tags ? { tags } : {}
   };
 }
 function parseWorkersConfig(raw) {
@@ -242,30 +270,64 @@ function parseWorkersConfig(raw) {
       providers[name] = parseProvider(name, p);
     }
   }
-  const roles = {};
-  if (w.roles !== void 0) {
-    if (typeof w.roles !== "object" || w.roles === null || Array.isArray(w.roles)) {
-      bad(".roles", "must be an object of role \u2192 provider[/alias] or {provider, mcp}");
+  const classes = {};
+  const classesRaw = w.classes !== void 0 ? w.classes : w.roles;
+  if (classesRaw !== void 0) {
+    if (typeof classesRaw !== "object" || classesRaw === null || Array.isArray(classesRaw)) {
+      bad(w.classes !== void 0 ? ".classes" : ".roles", "must be an object of class \u2192 provider[/alias] or {provider, mcp, maxCostTier, requireTags, order}");
     }
-    for (const [role, target] of Object.entries(w.roles)) {
+    for (const [cls, target] of Object.entries(classesRaw)) {
       if (typeof target === "object" && target !== null && !Array.isArray(target)) {
         const o = target;
         const provider = str(o.provider);
-        if (!provider || provider.includes(" ")) {
-          bad(`.roles.${role}.provider`, `invalid provider "${provider}"`);
+        if (o.provider !== void 0 && (!provider || provider.includes(" "))) {
+          bad(`.classes.${cls}.provider`, `invalid provider "${provider}"`);
         }
         let mcp;
         if (o.mcp !== void 0) {
           if (!Array.isArray(o.mcp) || o.mcp.some((x) => typeof x !== "string")) {
-            bad(`.roles.${role}.mcp`, "must be an array of MCP server or set names");
+            bad(`.classes.${cls}.mcp`, "must be an array of MCP server or set names");
           }
           mcp = o.mcp;
         }
-        roles[role] = mcp ? { provider, mcp } : { provider };
+        let maxCostTier;
+        if (o.maxCostTier !== void 0) {
+          if (typeof o.maxCostTier !== "number" || !Number.isInteger(o.maxCostTier) || o.maxCostTier < 1 || o.maxCostTier > 5) {
+            bad(`.classes.${cls}.maxCostTier`, "must be an integer 1 \u2026 5");
+          }
+          maxCostTier = o.maxCostTier;
+        }
+        let requireTags;
+        if (o.requireTags !== void 0) {
+          if (!Array.isArray(o.requireTags) || o.requireTags.some((x) => typeof x !== "string")) {
+            bad(`.classes.${cls}.requireTags`, `must be an array of tags from: ${PROVIDER_TAGS.join(", ")}`);
+          }
+          for (const t of o.requireTags) {
+            if (!PROVIDER_TAGS.includes(t)) {
+              bad(`.classes.${cls}.requireTags`, `"${t}" is not a tag (from: ${PROVIDER_TAGS.join(", ")})`);
+            }
+          }
+          requireTags = o.requireTags;
+        }
+        let order;
+        if (o.order !== void 0) {
+          if (!Array.isArray(o.order) || o.order.some((x) => typeof x !== "string")) {
+            bad(`.classes.${cls}.order`, "must be an array of provider names");
+          }
+          order = o.order;
+        }
+        const obj = {
+          ...provider ? { provider } : {},
+          ...mcp ? { mcp } : {},
+          ...maxCostTier !== void 0 ? { maxCostTier } : {},
+          ...requireTags ? { requireTags } : {},
+          ...order ? { order } : {}
+        };
+        classes[cls] = Object.keys(obj).length ? obj : {};
       } else {
         const t = str(target);
-        if (!t || t.includes(" ")) bad(`.roles.${role}`, `invalid target "${t}"`);
-        roles[role] = t;
+        if (!t || t.includes(" ")) bad(`.classes.${cls}`, `invalid target "${t}"`);
+        classes[cls] = t;
       }
     }
   }
@@ -327,7 +389,7 @@ function parseWorkersConfig(raw) {
     enabled: w.enabled === void 0 ? d.enabled : w.enabled === true,
     active,
     providers,
-    roles,
+    classes,
     mcpSets,
     pane,
     logDir: str(w.logDir) || d.logDir,