@dylanrussell/agent-router 2.0.2 → 2.1.1

package/dist/tui.d.ts

@@ -26,6 +26,17 @@ interface SidebarTheme {
     readonly warning?: unknown;
     readonly success?: unknown;
 }
+/** Live model selection observed for one agent in a session. */
+interface LiveSelection extends ModelAssignment {
+    readonly agent: string;
+}
+/** Routing mode of the viewed session, as reported by the status service. */
+interface RoutingStatus {
+    readonly mode: "automatic" | "pinned";
+    readonly reason?: string | undefined;
+    /** Epoch ms of the last quota evaluation behind the current mode, if tracked. */
+    readonly checkedAt?: number | null | undefined;
+}
 interface SidebarContext {
     /** Active stack when the TUI booted — differing means a restart is due. */
     readonly bootActive: string | null;
@@ -34,6 +45,14 @@ interface SidebarContext {
     readonly current?: (ModelAssignment & {
         readonly agent: string;
     }) | undefined;
+    /**
+     * Selections of running direct children of the viewed session, at its location.
+     * Conflicting models are all retained. The viewed session takes precedence.
+     */
+    readonly live?: readonly LiveSelection[] | undefined;
+    readonly defaults?: readonly LiveSelection[] | undefined;
+    /** Routing mode of the viewed session; omitted when no status source answers. */
+    readonly routing?: RoutingStatus | undefined;
 }
 
 /**
@@ -115,6 +134,11 @@ interface RouterTuiApi {
         readonly provider?: unknown;
     };
     readonly currentModel?: (sessionID: string) => SidebarContext["current"];
+    readonly configuredModels?: () => readonly LiveSelection[];
+    /** Running direct-child selections at the viewed parent's location; retain conflicts. */
+    readonly liveSelections?: ((sessionID: string) => readonly LiveSelection[]) | undefined;
+    /** Read-only routing-status query; undefined means no status source is available. */
+    readonly routingStatus?: ((sessionID: string) => Promise<RoutingStatus | undefined>) | undefined;
 }
 
 /**

package/dist/tui.js

@@ -15872,6 +15872,9 @@ function normalizeVariant(variant) {
 function sameModel(a, b) {
   return a.model === b.model && normalizeVariant(a.variant) === normalizeVariant(b.variant);
 }
+function quotaAge(checkedAt, now = Date.now()) {
+  return `${Math.max(0, Math.round((now - checkedAt) / 1e3))}s ago`;
+}
 function buildSidebarNodes(snapshot, ctx) {
   const theme = ctx.theme ?? {};
   const nodes = [text("Agent Stacks", { fg: theme.text, attributes: TEXT_ATTR_BOLD })];
@@ -15894,16 +15897,32 @@ function buildSidebarNodes(snapshot, ctx) {
       marginTop: 1
     })
   );
+  if (ctx.routing) {
+    const parts = [`Routing \xB7 ${ctx.routing.mode === "pinned" ? "Pinned" : "Automatic"}`];
+    if (ctx.routing.reason) parts.push(ctx.routing.reason);
+    parts.push(
+      ctx.routing.checkedAt != null ? `quota ${quotaAge(ctx.routing.checkedAt)}` : "quota freshness unknown"
+    );
+    nodes.push(text(parts.join(" \xB7 "), { fg: theme.textMuted }));
+  } else {
+    nodes.push(text("Routing \xB7 unknown \xB7 freshness unknown", { fg: theme.textMuted }));
+  }
   if (snapshot.agents.length === 0) {
     nodes.push(text("\u2022 (none)", { fg: theme.textMuted }));
   } else {
-    nodes.push(text("Precedence \u2193 \xB7 \u25CF current", { fg: theme.textMuted }));
+    nodes.push(text("Precedence \u2193 \xB7 \u25CF selected", { fg: theme.textMuted }));
     for (const assignment of snapshot.agents) {
       const chain = [assignment, ...assignment.fallbacks ?? []];
       const current = ctx.current?.agent === assignment.agent ? ctx.current : void 0;
-      const selected = current ? chain.findIndex((model) => sameModel(model, current)) : -1;
-      const modelRow = (model, active, external = false) => text(`${active ? "\u25CF " : "  "}${external ? "Current \xB7 " : ""}${modelLabel(model)}`, {
-        fg: active ? theme.success : theme.textMuted,
+      const children = (ctx.live ?? []).filter((entry) => entry.agent === assignment.agent);
+      const configured = ctx.defaults?.find((entry) => entry.agent === assignment.agent);
+      const selections = current ? [current] : children.length ? children : [configured ?? assignment];
+      const unique = selections.filter(
+        (entry, index) => selections.findIndex((other) => sameModel(entry, other)) === index
+      );
+      const source = current ? "viewed session" : children.length ? `active children${unique.length > 1 ? " \xB7 multiple models" : ""}` : configured ? "native default" : "stack default";
+      const modelRow = (model, active) => text(`${active ? "\u25CF " : "  "}${modelLabel(model)}`, {
+        fg: active ? theme.warning : theme.textMuted,
         attributes: active ? TEXT_ATTR_BOLD : 0,
         wrapMode: "char",
         width: "100%"
@@ -15912,13 +15931,18 @@ function buildSidebarNodes(snapshot, ctx) {
         kind: "box",
         props: { flexDirection: "column", marginTop: 1 },
         children: [
-          text(assignment.agent, { fg: theme.text, attributes: TEXT_ATTR_BOLD }),
+          text(`${assignment.agent} \xB7 ${source}`, { fg: theme.text, attributes: TEXT_ATTR_BOLD }),
           {
             kind: "box",
             props: { flexDirection: "column", paddingLeft: 2 },
             children: [
-              ...chain.map((model, index) => modelRow(model, index === selected)),
-              ...current && selected === -1 ? [modelRow(current, true, true)] : []
+              ...chain.map(
+                (model) => modelRow(
+                  model,
+                  unique.some((selected) => sameModel(model, selected))
+                )
+              ),
+              ...unique.filter((selected) => !chain.some((model) => sameModel(model, selected))).map((selected) => modelRow(selected, true))
             ]
           }
         ]
@@ -15933,6 +15957,7 @@ function buildSidebarNodes(snapshot, ctx) {
 
 // src/tui/index.ts
 var POLL_INTERVAL_MS = 1500;
+var ROUTING_INTERVAL_MS = 3e3;
 var SIDEBAR_ORDER = 850;
 function debugLog(message) {
   const target = process.env.AGENT_ROUTER_TUI_DEBUG ?? process.env.OMO_TUI_DEBUG;
@@ -16018,16 +16043,25 @@ var tui = async (api) => {
   if (!paths) return;
   let snapshot = await readStackSnapshot(paths);
   const bootActive = snapshot.active;
+  let routing;
+  let routingSessionID;
+  let lastSessionID;
   const viewContext = (sessionID) => ({
     bootActive,
     theme: api.theme?.current,
-    current: sessionID ? api.currentModel?.(sessionID) : void 0
+    current: sessionID ? api.currentModel?.(sessionID) : void 0,
+    live: sessionID ? api.liveSelections?.(sessionID) : void 0,
+    defaults: api.configuredModels?.(),
+    routing: sessionID === routingSessionID ? routing : void 0
   });
   try {
     api.slots.register({
       order: SIDEBAR_ORDER,
       slots: {
-        sidebar_content: (sessionID) => materialize(buildSidebarNodes(snapshot, viewContext(sessionID)), solid)
+        sidebar_content: (sessionID) => {
+          if (sessionID) lastSessionID = sessionID;
+          return materialize(buildSidebarNodes(snapshot, viewContext(sessionID)), solid);
+        }
       }
     });
     debugLog("slots.register ok");
@@ -16060,6 +16094,20 @@ var tui = async (api) => {
       }
     }
   });
+  const stopRouting = api.routingStatus ? createSidebarPoller({
+    read: async () => {
+      const sessionID = lastSessionID;
+      const status2 = sessionID ? await api.routingStatus?.(sessionID).catch(() => void 0) : void 0;
+      return { routing: status2, sessionID, key: JSON.stringify([sessionID, status2 ?? null]) };
+    },
+    intervalMs: ROUTING_INTERVAL_MS,
+    initial: { routing: void 0, sessionID: void 0, key: "boot" },
+    onChange: (next) => {
+      routing = next.routing;
+      routingSessionID = next.sessionID;
+      api.renderer.requestRender();
+    }
+  }) : void 0;
   const deps = { api, paths, refresh };
   const status = () => api.ui.toast({
     title: "agent-router",
@@ -16073,6 +16121,7 @@ var tui = async (api) => {
     debugLog(`command.register FAILED: ${e.message}`);
   }
   api.lifecycle.onDispose(stopPolling);
+  if (stopRouting) api.lifecycle.onDispose(stopRouting);
   debugLog(`tui() init complete \u2014 active=${snapshot.active ?? "(none)"}`);
 };
 var agentRouterTui = {
@@ -16195,6 +16244,38 @@ var agentRouterTui = {
           variant: session.model.variant ?? null
         };
       },
+      liveSelections: (sessionID) => {
+        const sessions = ctx.data.session;
+        if (typeof sessions.list !== "function" || typeof sessions.status !== "function") return [];
+        try {
+          const parent = sessions.get(sessionID);
+          if (!parent?.location) return [];
+          const selections = [];
+          for (const session of sessions.list()) {
+            if (session.parentID !== sessionID || !session.agent || !session.model) continue;
+            if (session.location?.directory !== parent.location.directory) continue;
+            if (sessions.status(session.id) !== "running") continue;
+            const { agent, model } = session;
+            selections.push({
+              agent,
+              model: `${model.providerID}/${model.id}`,
+              variant: model.variant ?? null
+            });
+          }
+          return selections;
+        } catch {
+          return [];
+        }
+      },
+      configuredModels: () => (ctx.data.location.agent?.list(ctx.location) ?? []).flatMap(
+        (agent) => agent.model ? [
+          {
+            agent: agent.name,
+            model: `${agent.model.providerID}/${agent.model.id}`,
+            variant: agent.model.variant ?? null
+          }
+        ] : []
+      ),
       state: {
         get provider() {
           const models = ctx.data.location.model.list(ctx.location) ?? [];