@tekmidian/pai 0.66.48 → 0.66.49

package/dist/hooks/route-agents-to-worker.mjs

@@ -7816,7 +7816,10 @@ var DEFAULT_PANE = {
 var DEFAULT_ROUTING2 = {
   order: [],
   cooldownMinutes: 30,
-  retryOnQuota: true
+  retryOnQuota: true,
+  policy: "budget",
+  priority: [],
+  budgetWalls: {}
 };
 var DEFAULT_TREE = {
   maxDepth: 2,
@@ -8177,10 +8180,31 @@ function parseWorkersConfig(raw) {
     if (r.retryOnQuota !== void 0 && typeof r.retryOnQuota !== "boolean") {
       bad(".routing.retryOnQuota", "must be boolean");
     }
+    if (r.policy !== void 0 && r.policy !== "budget" && r.policy !== "order") {
+      bad(".routing.policy", 'must be "budget" or "order"');
+    }
+    if (r.priority !== void 0) {
+      if (!Array.isArray(r.priority) || r.priority.some((x) => typeof x !== "string")) {
+        bad(".routing.priority", "must be an array of provider names");
+      }
+    }
+    if (r.budgetWalls !== void 0) {
+      if (typeof r.budgetWalls !== "object" || r.budgetWalls === null || Array.isArray(r.budgetWalls)) {
+        bad(".routing.budgetWalls", "must be an object of provider name \u2192 percent");
+      }
+      for (const [k, v] of Object.entries(r.budgetWalls)) {
+        if (typeof v !== "number" || !Number.isFinite(v) || v < 0 || v > 100) {
+          bad(`.routing.budgetWalls.${k}`, "must be a percent 0\u2013100");
+        }
+      }
+    }
     routing = {
       order: routing.order,
       cooldownMinutes: r.cooldownMinutes === void 0 ? DEFAULT_ROUTING2.cooldownMinutes : r.cooldownMinutes,
-      retryOnQuota: r.retryOnQuota === void 0 ? DEFAULT_ROUTING2.retryOnQuota : r.retryOnQuota === true
+      retryOnQuota: r.retryOnQuota === void 0 ? DEFAULT_ROUTING2.retryOnQuota : r.retryOnQuota === true,
+      policy: r.policy ?? DEFAULT_ROUTING2.policy,
+      priority: r.priority === void 0 ? [] : [...r.priority],
+      budgetWalls: r.budgetWalls === void 0 ? {} : { ...r.budgetWalls }
     };
   }
   let tree = { ...DEFAULT_TREE };