@bitbaum/ai-kit 0.10.0 → 0.12.0

package/dist/liveness.d.ts

@@ -39,6 +39,7 @@
  * is "does the pipe carry water", not "is the model any good".
  */
 import { type CompleteOptions } from "./complete.js";
+import type { Link, Env } from "./chain.js";
 import type { HealthTracker } from "./health.js";
 export interface LivenessResult {
     /** Did a model answer? */
@@ -64,6 +65,35 @@ export interface LivenessOptions extends Omit<CompleteOptions, "messages" | "max
      * Default 10 minutes. Set 0 to disable caching — only for a test.
      */
     minIntervalMs?: number;
+    /**
+     * Resolve the chain when a probe actually runs, rather than once at
+     * construction.
+     *
+     * For an app whose provider list lives in a DATABASE — an admin screen with
+     * enabled/default rows and per-provider keys — a chain fixed at construction
+     * is a chain frozen at process start. The probe would then keep reporting on
+     * a configuration the operator changed twenty minutes ago, which is the
+     * opposite of "the truth about right now".
+     *
+     * It is called only on a real probe, never on a cache hit, so a monitor
+     * polling this route does not also poll the database.
+     *
+     * Return `{ chain, env }` when the KEYS move with the chain — an admin
+     * screen that stores a per-provider key alongside the model is the ordinary
+     * case, and resolving links from the database while reading credentials from
+     * a `process.env` captured at construction would have the probe report "no
+     * key" for a provider that is configured and working. Returning a bare
+     * `Link[]` keeps whatever `env` the probe was built with.
+     *
+     * Takes precedence over `chain` when both are given.
+     */
+    resolveChain?: () => Link[] | {
+        chain: Link[];
+        env?: Env;
+    } | Promise<Link[] | {
+        chain: Link[];
+        env?: Env;
+    }>;
     /** Injected for tests. Defaults to `Date.now`. */
     now?: () => number;
 }

package/dist/liveness.js

@@ -90,9 +90,19 @@ export function createLivenessProbe(options = {}) {
             }
             const started = now();
             try {
+                // Resolved here, not at construction, and only on a real probe — so a
+                // monitor polling this route does not also poll whatever backs it.
+                const resolved = options.resolveChain ? await options.resolveChain() : options.chain;
+                const chain = Array.isArray(resolved) ? resolved : resolved?.chain;
+                // Only when the resolver supplied one. Otherwise the probe keeps the
+                // env it was built with, so a bare `Link[]` resolver behaves exactly
+                // as it did before this option grew a second shape.
+                const env = !Array.isArray(resolved) && resolved?.env !== undefined ? resolved.env : options.env;
                 const result = await complete({
                     timeoutMs: PROBE_TIMEOUT_MS,
                     ...options,
+                    chain,
+                    env,
                     messages: PROBE_MESSAGES,
                     maxTokens: PROBE_MAX_TOKENS,
                     temperature: 0,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bitbaum/ai-kit",
-  "version": "0.10.0",
+  "version": "0.12.0",
   "description": "One install for the AI layer of an app: which model to call, what to do when the vendor retires it, how to walk the fallback chain and know when none of it worked, how to read the three kinds of 429, a fair daily budget across users, headless AI form filling — and now the model registry (one SSOT for every callable id, with the paid/free boundary as a field) and the grounding harness (facts, contract, deterministic fabrication check).",
   "license": "MIT",
   "author": "Mao Nakamoto",

package/src/liveness.ts

@@ -40,6 +40,7 @@
  */
 
 import { complete, type CompleteOptions } from "./complete.js";
+import type { Link, Env } from "./chain.js";
 import { ChainExhaustedError } from "./attempt.js";
 import type { HealthTracker } from "./health.js";
 
@@ -71,6 +72,30 @@ export interface LivenessOptions extends Omit<
    * Default 10 minutes. Set 0 to disable caching — only for a test.
    */
   minIntervalMs?: number;
+  /**
+   * Resolve the chain when a probe actually runs, rather than once at
+   * construction.
+   *
+   * For an app whose provider list lives in a DATABASE — an admin screen with
+   * enabled/default rows and per-provider keys — a chain fixed at construction
+   * is a chain frozen at process start. The probe would then keep reporting on
+   * a configuration the operator changed twenty minutes ago, which is the
+   * opposite of "the truth about right now".
+   *
+   * It is called only on a real probe, never on a cache hit, so a monitor
+   * polling this route does not also poll the database.
+   *
+   * Return `{ chain, env }` when the KEYS move with the chain — an admin
+   * screen that stores a per-provider key alongside the model is the ordinary
+   * case, and resolving links from the database while reading credentials from
+   * a `process.env` captured at construction would have the probe report "no
+   * key" for a provider that is configured and working. Returning a bare
+   * `Link[]` keeps whatever `env` the probe was built with.
+   *
+   * Takes precedence over `chain` when both are given.
+   */
+  resolveChain?: () =>
+    Link[] | { chain: Link[]; env?: Env } | Promise<Link[] | { chain: Link[]; env?: Env }>;
   /** Injected for tests. Defaults to `Date.now`. */
   now?: () => number;
 }
@@ -140,9 +165,21 @@ export function createLivenessProbe(options: LivenessOptions = {}): LivenessProb
 
       const started = now();
       try {
+        // Resolved here, not at construction, and only on a real probe — so a
+        // monitor polling this route does not also poll whatever backs it.
+        const resolved = options.resolveChain ? await options.resolveChain() : options.chain;
+        const chain = Array.isArray(resolved) ? resolved : resolved?.chain;
+        // Only when the resolver supplied one. Otherwise the probe keeps the
+        // env it was built with, so a bare `Link[]` resolver behaves exactly
+        // as it did before this option grew a second shape.
+        const env =
+          !Array.isArray(resolved) && resolved?.env !== undefined ? resolved.env : options.env;
+
         const result = await complete({
           timeoutMs: PROBE_TIMEOUT_MS,
           ...options,
+          chain,
+          env,
           messages: PROBE_MESSAGES,
           maxTokens: PROBE_MAX_TOKENS,
           temperature: 0,