@oisin-ee/pi-tiers 0.9.0 → 0.9.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Momokaya
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@oisin-ee/pi-tiers",
-  "version": "0.9.0",
+  "version": "0.9.1",
   "description": "Sticky virtual model tiers with quota-aware recovery for pi.",
   "keywords": [
     "pi-extension",
@@ -20,8 +20,8 @@
   ],
   "type": "module",
   "devDependencies": {
-    "@earendil-works/pi-ai": "catalog:",
-    "@earendil-works/pi-coding-agent": "catalog:"
+    "@earendil-works/pi-ai": "1.3.0",
+    "@earendil-works/pi-coding-agent": "1.3.0"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": "*",
@@ -32,4 +32,4 @@
       "./src/index.ts"
     ]
   }
-}
+}

package/src/recovery.ts

@@ -9,6 +9,7 @@ import type { Context } from "./ports.ts";
 import type { Router } from "./router.ts";
 import type { Tiers } from "./config.ts";
 import { classify } from "./policy.ts";
+import { listed } from "./router.ts";
 
 const ZERO = 0,
   ONE = 1;
@@ -76,7 +77,8 @@ class Recovery {
     }
     const tier = ctx.model.id,
       state = branchState(ctx.sessionManager.getBranch(), tier),
-      targets = this.#tiers.get(tier),
+      tierTargets = this.#tiers.get(tier),
+      targets = tierTargets === undefined ? undefined : listed(tierTargets, ctx),
       message = event.context.contextMessages.findLast((item) => item.role === "assistant"),
       entry = event.context.contextEntries.findLast(
         (item: Readonly<{ sourceEntry: SessionEntry; messages: readonly unknown[] }>) =>

package/src/router.ts

@@ -10,7 +10,23 @@ import type { Window } from "./quota.ts";
 
 const ZERO = 0,
   ONE = 1,
-  CACHE_MS = 60_000;
+  CACHE_MS = 60_000,
+  /**
+   * Keeps only targets the model registry lists, so a provider model that is not
+   * configured yet behaves like an unavailable target instead of a terminal error.
+   * @param {readonly string[]} targets Ordered tier.
+   * @param {Context} ctx Pi authority.
+   * @returns {readonly string[]} Listed targets, in tier order.
+   */
+  listed = (targets: readonly string[], ctx: Context): readonly string[] => {
+    const available = new Set(
+      ctx.modelRegistry.getAvailable().map((candidate) => `${candidate.provider}/${candidate.id}`),
+    );
+    return targets.filter((reference) => {
+      const parsed = target(reference);
+      return available.has(`${parsed.provider}/${parsed.id}`);
+    });
+  };
 
 /** Sticky routing with host-wide quota and gateway health. */
 class Router {
@@ -139,19 +155,20 @@ class Router {
 
   /**
    * Routes a tier without fabricating conversation messages.
-   * @param {readonly string[]} targets Ordered physical models.
+   * @param {readonly string[]} tier Ordered physical models, listed or not.
    * @param {Request} request Pi request.
    * @param {Context} ctx Pi authority.
    * @returns {Promise<ModelRoute<State>>} Physical dispatch.
    */
   public async route(
-    targets: readonly string[],
+    tier: readonly string[],
     request: Request,
     ctx: Context,
   ): Promise<ModelRoute<State>> {
     request.signal?.throwIfAborted();
     await this.#refresh();
-    const previous =
+    const targets = listed(tier, ctx),
+      previous =
         request.reason === "direct"
           ? undefined
           : (branchState(ctx.sessionManager.getBranch(), request.model.id) ?? request.state),
@@ -163,7 +180,9 @@ class Router {
       state = await this.#retry(request, current),
       health = await this.#effects.read(),
       chosen =
-        !state.attempted.includes(state.chosen) && this.#available(state.chosen, health)
+        targets.includes(state.chosen) &&
+        !state.attempted.includes(state.chosen) &&
+        this.#available(state.chosen, health)
           ? state.chosen
           : await this.next(targets, state);
     if (chosen === undefined) {
@@ -173,4 +192,4 @@ class Router {
   }
 }
 
-export { Router };
+export { Router, listed };