@dylanrussell/agent-router 1.0.7 → 1.1.0

package/dist/tui.js

@@ -14615,14 +14615,26 @@ function date4(params) {
 config(en_default());
 
 // src/core/schema.ts
+var FallbackSchema = external_exports.object({
+  model: external_exports.string().regex(/^[^/\s]+\/\S+$/, "Expected provider/model"),
+  variant: external_exports.string().min(1).optional()
+}).strict();
+var RoutingEntrySchema = external_exports.object({
+  model: external_exports.string().min(1),
+  variant: external_exports.string().min(1).nullable().optional(),
+  fallbacks: external_exports.array(FallbackSchema).max(8).optional()
+}).strict();
 var StateFileSchema = external_exports.object({
   version: external_exports.literal(1),
   active: external_exports.string().min(1),
   previousActive: external_exports.string().min(1).nullable(),
-  lastSwitchedAt: external_exports.string().min(1)
+  lastSwitchedAt: external_exports.string().min(1),
+  fallbackAgents: external_exports.record(external_exports.string(), RoutingEntrySchema).optional()
 }).strict();
 var AgentEntrySchema = external_exports.object({
-  model: external_exports.string().min(1)
+  model: external_exports.string().min(1),
+  variant: external_exports.string().min(1).nullable().optional(),
+  fallbacks: external_exports.array(FallbackSchema).max(8).optional()
 }).passthrough();
 var StackFileSchema = external_exports.object({
   agents: external_exports.record(external_exports.string(), AgentEntrySchema)
@@ -15303,6 +15315,9 @@ function collectModelRefs(stack) {
   const refs = [];
   for (const [k, v] of Object.entries(stack.agents)) {
     refs.push({ path: `agents.${k}.model`, modelId: v.model });
+    for (const [index, candidate] of (v.fallbacks ?? []).entries()) {
+      refs.push({ path: `agents.${k}.fallbacks.${index}.model`, modelId: candidate.model });
+    }
   }
   return refs;
 }
@@ -15382,7 +15397,7 @@ async function applyStack(paths, name, options = {}) {
   }
   const prevState = await readState(paths.statePath);
   const prevActive = prevState?.active ?? null;
-  const displaced = { agents: entriesToStackAgents(await readAgentEntries(paths.agentsDir)) };
+  const displaced = { agents: await captureAgents(paths) };
   const historyId = await appendHistory(
     paths.historyDir,
     prevActive ?? "(none)",
@@ -15400,7 +15415,20 @@ async function applyStack(paths, name, options = {}) {
     version: 1,
     active: name,
     previousActive: prevActive,
-    lastSwitchedAt: (/* @__PURE__ */ new Date()).toISOString()
+    lastSwitchedAt: (/* @__PURE__ */ new Date()).toISOString(),
+    fallbackAgents: Object.fromEntries(
+      Object.entries(target.agents).filter(([, entry]) => entry.fallbacks?.length).map(([agent, entry]) => {
+        const variant = "variant" in entry ? entry.variant : displaced.agents[agent]?.variant;
+        return [
+          agent,
+          {
+            model: entry.model,
+            variant: typeof variant === "string" ? variant : void 0,
+            fallbacks: entry.fallbacks
+          }
+        ];
+      })
+    )
   });
   await trimHistory(paths.historyDir).catch(() => {
   });
@@ -15416,7 +15444,7 @@ function entryOptions(entry) {
   const rec = entry;
   const out = {};
   for (const [k, v] of Object.entries(rec)) {
-    if (k === "model" || RESERVED_AGENT_KEYS.has(k)) continue;
+    if (k === "model" || k === "fallbacks" || RESERVED_AGENT_KEYS.has(k)) continue;
     out[k] = v;
   }
   return out;
@@ -15428,11 +15456,24 @@ function entriesToStackAgents(entries) {
     if (!entry) continue;
     const { model, options } = entry;
     const stackEntry = { model };
-    for (const [k, v] of Object.entries(options)) stackEntry[k] = v;
+    for (const [k, v] of Object.entries(options)) {
+      if (k !== "fallbacks") stackEntry[k] = v;
+    }
     out[name] = stackEntry;
   }
   return out;
 }
+async function captureAgents(paths) {
+  const agents = entriesToStackAgents(await readAgentEntries(paths.agentsDir));
+  const state = await readState(paths.statePath);
+  for (const [name, entry] of Object.entries(agents)) {
+    const routing = state?.fallbackAgents?.[name];
+    if (routing && routing.model === entry.model && (routing.variant ?? void 0) === (entry.variant ?? void 0)) {
+      entry.fallbacks = routing.fallbacks;
+    }
+  }
+  return agents;
+}
 async function back(paths, n = 1, options = {}) {
   if (n < 1) throw new UserError("`back -n` must be at least 1.");
   const state = await readState(paths.statePath);