@machina.ai/cell-cli-core 1.18.4-rc1 → 1.19.4-rc2

package/dist/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@machina.ai/cell-cli-core",
-  "version": "1.18.4-rc1",
+  "version": "1.19.4-rc2",
   "description": "Cell CLI Core",
   "repository": {
     "type": "git",
@@ -18,7 +18,7 @@
     "@google/genai": "1.30.0",
     "@iarna/toml": "^2.2.5",
     "@joshua.litt/get-ripgrep": "^0.0.3",
-    "@machina.ai/auth-keycloak": "1.11.0-rc0",
+    "@machina.ai/auth-keycloak": "1.19.4-rc0",
     "@machina.ai/config-yaml": "1.11.0-rc0",
     "@modelcontextprotocol/sdk": "^1.23.0",
     "@opentelemetry/api": "^1.9.0",

package/dist/src/agents/codebase-investigator.js

@@ -74,7 +74,7 @@ You are a sub-agent in a larger system. Your only responsibility is to provide d
 - **DO:** Find the key modules, classes, and functions that are part of the problem and its solution.
 - **DO:** Understand *why* the code is written the way it is. Question everything.
 - **DO:** Foresee the ripple effects of a change. If \`function A\` is modified, you must check its callers. If a data structure is altered, you must identify where its type definitions need to be updated.
-- **DO:** provide a conclusion and insights to the main agent that invoked you. If the agent is trying to solve a bug, you should provide the root cause of the bug, its impacts, how to fix it etc. If it's a new feature, you should provide insights on where to implement it, what chagnes are necessary etc. 
+- **DO:** provide a conclusion and insights to the main agent that invoked you. If the agent is trying to solve a bug, you should provide the root cause of the bug, its impacts, how to fix it etc. If it's a new feature, you should provide insights on where to implement it, what changes are necessary etc.
 - **DO NOT:** Write the final implementation code yourself.
 - **DO NOT:** Stop at the first relevant file. Your goal is a comprehensive understanding of the entire relevant subsystem.
 You operate in a non-interactive loop and must reason based on the information provided and the output of your tools.

package/dist/src/availability/modelAvailabilityService.d.ts

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export type ModelId = string;
+export type TerminalUnavailabilityReason = 'quota' | 'capacity';
+export type TurnUnavailabilityReason = 'retry_once_per_turn';
+export type UnavailabilityReason = TerminalUnavailabilityReason | TurnUnavailabilityReason | 'unknown';
+export type ModelHealthStatus = 'terminal' | 'sticky_retry';
+export interface ModelAvailabilitySnapshot {
+    available: boolean;
+    reason?: UnavailabilityReason;
+}
+export interface ModelSelectionResult {
+    selected: ModelId | null;
+    attempts?: number;
+    skipped: Array<{
+        model: ModelId;
+        reason: UnavailabilityReason;
+    }>;
+}
+export declare class ModelAvailabilityService {
+    private readonly health;
+    markTerminal(model: ModelId, reason: TerminalUnavailabilityReason): void;
+    markHealthy(model: ModelId): void;
+    markRetryOncePerTurn(model: ModelId): void;
+    consumeStickyAttempt(model: ModelId): void;
+    snapshot(model: ModelId): ModelAvailabilitySnapshot;
+    selectFirstAvailable(models: ModelId[]): ModelSelectionResult;
+    resetTurn(): void;
+    private setState;
+    private clearState;
+}

package/dist/src/availability/modelAvailabilityService.js

@@ -0,0 +1,84 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export class ModelAvailabilityService {
+    health = new Map();
+    markTerminal(model, reason) {
+        this.setState(model, {
+            status: 'terminal',
+            reason,
+        });
+    }
+    markHealthy(model) {
+        this.clearState(model);
+    }
+    markRetryOncePerTurn(model) {
+        const currentState = this.health.get(model);
+        // Do not override a terminal failure with a transient one.
+        if (currentState?.status === 'terminal') {
+            return;
+        }
+        // Only reset consumption if we are not already in the sticky_retry state.
+        // This prevents infinite loops if the model fails repeatedly in the same turn.
+        let consumed = false;
+        if (currentState?.status === 'sticky_retry') {
+            consumed = currentState.consumed;
+        }
+        this.setState(model, {
+            status: 'sticky_retry',
+            reason: 'retry_once_per_turn',
+            consumed,
+        });
+    }
+    consumeStickyAttempt(model) {
+        const state = this.health.get(model);
+        if (state?.status === 'sticky_retry') {
+            this.setState(model, { ...state, consumed: true });
+        }
+    }
+    snapshot(model) {
+        const state = this.health.get(model);
+        if (!state) {
+            return { available: true };
+        }
+        if (state.status === 'terminal') {
+            return { available: false, reason: state.reason };
+        }
+        if (state.status === 'sticky_retry' && state.consumed) {
+            return { available: false, reason: state.reason };
+        }
+        return { available: true };
+    }
+    selectFirstAvailable(models) {
+        const skipped = [];
+        for (const model of models) {
+            const snapshot = this.snapshot(model);
+            if (snapshot.available) {
+                const state = this.health.get(model);
+                // A sticky model is being attempted, so note that.
+                const attempts = state?.status === 'sticky_retry' ? 1 : undefined;
+                return { selected: model, skipped, attempts };
+            }
+            else {
+                skipped.push({ model, reason: snapshot.reason ?? 'unknown' });
+            }
+        }
+        return { selected: null, skipped };
+    }
+    resetTurn() {
+        for (const [model, state] of this.health.entries()) {
+            if (state.status === 'sticky_retry') {
+                this.setState(model, { ...state, consumed: false });
+            }
+        }
+    }
+    setState(model, nextState) {
+        this.health.set(model, nextState);
+    }
+    clearState(model) {
+        this.health.delete(model);
+    }
+}
+//# sourceMappingURL=modelAvailabilityService.js.map

package/dist/src/availability/modelAvailabilityService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelAvailabilityService.js","sourceRoot":"","sources":["../../../src/availability/modelAvailabilityService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAoCH,MAAM,OAAO,wBAAwB;IAClB,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IAE1D,YAAY,CAAC,KAAc,EAAE,MAAoC;QAC/D,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE;YACnB,MAAM,EAAE,UAAU;YAClB,MAAM;SACP,CAAC,CAAC;IACL,CAAC;IAED,WAAW,CAAC,KAAc;QACxB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,oBAAoB,CAAC,KAAc;QACjC,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5C,2DAA2D;QAC3D,IAAI,YAAY,EAAE,MAAM,KAAK,UAAU,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QAED,0EAA0E;QAC1E,+EAA+E;QAC/E,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,YAAY,EAAE,MAAM,KAAK,cAAc,EAAE,CAAC;YAC5C,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE;YACnB,MAAM,EAAE,cAAc;YACtB,MAAM,EAAE,qBAAqB;YAC7B,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAAC,KAAc;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,KAAK,EAAE,MAAM,KAAK,cAAc,EAAE,CAAC;YACrC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAAc;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAErC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAChC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;QACpD,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,cAAc,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACtD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;QACpD,CAAC;QAED,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED,oBAAoB,CAAC,MAAiB;QACpC,MAAM,OAAO,GAAoC,EAAE,CAAC;QAEpD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACtC,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACvB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACrC,mDAAmD;gBACnD,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAM,KAAK,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAClE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;YAChD,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IACrC,CAAC;IAED,SAAS;QACP,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,IAAI,KAAK,CAAC,MAAM,KAAK,cAAc,EAAE,CAAC;gBACpC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC;IAEO,QAAQ,CAAC,KAAc,EAAE,SAAsB;QACrD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC;IAEO,UAAU,CAAC,KAAc;QAC/B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;CACF"}

package/dist/src/availability/modelAvailabilityService.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/availability/modelAvailabilityService.test.js

@@ -0,0 +1,140 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, expect, it, vi, beforeEach } from 'vitest';
+import { ModelAvailabilityService } from './modelAvailabilityService.js';
+describe('ModelAvailabilityService', () => {
+    let service;
+    const model = 'test-model';
+    beforeEach(() => {
+        service = new ModelAvailabilityService();
+        vi.useRealTimers();
+    });
+    it('returns available snapshot when no state recorded', () => {
+        expect(service.snapshot(model)).toEqual({ available: true });
+    });
+    it('tracks retry-once-per-turn failures', () => {
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model)).toEqual({ available: true });
+        service.consumeStickyAttempt(model);
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'retry_once_per_turn',
+        });
+        service.resetTurn();
+        expect(service.snapshot(model)).toEqual({ available: true });
+    });
+    it('tracks terminal failures', () => {
+        service.markTerminal(model, 'quota');
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'quota',
+        });
+    });
+    it('does not override terminal failure with sticky failure', () => {
+        service.markTerminal(model, 'quota');
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'quota',
+        });
+    });
+    it('selects models respecting terminal and sticky states', () => {
+        const stickyModel = 'stick-model';
+        const healthyModel = 'healthy-model';
+        service.markTerminal(model, 'capacity');
+        service.markRetryOncePerTurn(stickyModel);
+        const first = service.selectFirstAvailable([
+            model,
+            stickyModel,
+            healthyModel,
+        ]);
+        expect(first).toEqual({
+            selected: stickyModel,
+            attempts: 1,
+            skipped: [
+                {
+                    model,
+                    reason: 'capacity',
+                },
+            ],
+        });
+        service.consumeStickyAttempt(stickyModel);
+        const second = service.selectFirstAvailable([
+            model,
+            stickyModel,
+            healthyModel,
+        ]);
+        expect(second).toEqual({
+            selected: healthyModel,
+            skipped: [
+                {
+                    model,
+                    reason: 'capacity',
+                },
+                {
+                    model: stickyModel,
+                    reason: 'retry_once_per_turn',
+                },
+            ],
+        });
+        service.resetTurn();
+        const third = service.selectFirstAvailable([
+            model,
+            stickyModel,
+            healthyModel,
+        ]);
+        expect(third).toEqual({
+            selected: stickyModel,
+            attempts: 1,
+            skipped: [
+                {
+                    model,
+                    reason: 'capacity',
+                },
+            ],
+        });
+    });
+    it('preserves consumed state when marking retry-once-per-turn again', () => {
+        service.markRetryOncePerTurn(model);
+        service.consumeStickyAttempt(model);
+        // It is currently consumed
+        expect(service.snapshot(model).available).toBe(false);
+        // Marking it again should not reset the consumed flag
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model).available).toBe(false);
+    });
+    it('clears consumed state when marked healthy', () => {
+        service.markRetryOncePerTurn(model);
+        service.consumeStickyAttempt(model);
+        expect(service.snapshot(model).available).toBe(false);
+        service.markHealthy(model);
+        expect(service.snapshot(model).available).toBe(true);
+        // If we mark it sticky again, it should be fresh (not consumed)
+        service.markRetryOncePerTurn(model);
+        expect(service.snapshot(model).available).toBe(true);
+    });
+    it('resetTurn resets consumed state for multiple sticky models', () => {
+        const model2 = 'model-2';
+        service.markRetryOncePerTurn(model);
+        service.markRetryOncePerTurn(model2);
+        service.consumeStickyAttempt(model);
+        service.consumeStickyAttempt(model2);
+        expect(service.snapshot(model).available).toBe(false);
+        expect(service.snapshot(model2).available).toBe(false);
+        service.resetTurn();
+        expect(service.snapshot(model).available).toBe(true);
+        expect(service.snapshot(model2).available).toBe(true);
+    });
+    it('resetTurn does not affect terminal models', () => {
+        service.markTerminal(model, 'quota');
+        service.resetTurn();
+        expect(service.snapshot(model)).toEqual({
+            available: false,
+            reason: 'quota',
+        });
+    });
+});
+//# sourceMappingURL=modelAvailabilityService.test.js.map

package/dist/src/availability/modelAvailabilityService.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelAvailabilityService.test.js","sourceRoot":"","sources":["../../../src/availability/modelAvailabilityService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AAEzE,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,OAAiC,CAAC;IACtC,MAAM,KAAK,GAAG,YAAY,CAAC;IAE3B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,wBAAwB,EAAE,CAAC;QACzC,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE7D,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,qBAAqB;SAC9B,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACrC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,WAAW,GAAG,aAAa,CAAC;QAClC,MAAM,YAAY,GAAG,eAAe,CAAC;QAErC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACxC,OAAO,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAE1C,MAAM,KAAK,GAAG,OAAO,CAAC,oBAAoB,CAAC;YACzC,KAAK;YACL,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,QAAQ,EAAE,WAAW;YACrB,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE;gBACP;oBACE,KAAK;oBACL,MAAM,EAAE,UAAU;iBACnB;aACF;SACF,CAAC,CAAC;QAEH,OAAO,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC;YAC1C,KAAK;YACL,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE;gBACP;oBACE,KAAK;oBACL,MAAM,EAAE,UAAU;iBACnB;gBACD;oBACE,KAAK,EAAE,WAAW;oBAClB,MAAM,EAAE,qBAAqB;iBAC9B;aACF;SACF,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,OAAO,CAAC,oBAAoB,CAAC;YACzC,KAAK;YACL,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,QAAQ,EAAE,WAAW;YACrB,QAAQ,EAAE,CAAC;YACX,OAAO,EAAE;gBACP;oBACE,KAAK;oBACL,MAAM,EAAE,UAAU;iBACnB;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEpC,2BAA2B;QAC3B,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEtD,sDAAsD;QACtD,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEtD,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC3B,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErD,gEAAgE;QAChE,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,MAAM,GAAG,SAAS,CAAC;QACzB,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAErC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAErC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEvD,OAAO,CAAC,SAAS,EAAE,CAAC;QAEpB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACrC,OAAO,CAAC,SAAS,EAAE,CAAC;QACpB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;YACtC,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/availability/modelPolicy.d.ts

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ModelHealthStatus, ModelId } from './modelAvailabilityService.js';
+/**
+ * Whether to prompt the user or fallback silently on a model API failure.
+ */
+export type FallbackAction = 'silent' | 'prompt';
+/**
+ * Type of possible errors from model API failures.
+ */
+export type FailureKind = 'terminal' | 'transient' | 'not_found' | 'unknown';
+/**
+ * Map from model API failure reason to user interaction.
+ */
+export type ModelPolicyActionMap = Partial<Record<FailureKind, FallbackAction>>;
+/**
+ * What state (e.g. Terminal, Sticky Retry) to set a model after failed API call.
+ */
+export type ModelPolicyStateMap = Partial<Record<FailureKind, ModelHealthStatus>>;
+/**
+ * Defines the policy for a single model in the availability chain.
+ *
+ * This includes:
+ * - Which model this policy applies to.
+ * - What actions to take (prompt vs silent fallback) for different failure kinds.
+ * - How the model's health status should transition upon failure.
+ * - Whether this model is considered a "last resort" (i.e. use if all models are unavailable).
+ */
+export interface ModelPolicy {
+    model: ModelId;
+    actions: ModelPolicyActionMap;
+    stateTransitions: ModelPolicyStateMap;
+    isLastResort?: boolean;
+}
+/**
+ * A chain of model policies defining the priority and fallback behavior.
+ * The first model in the chain is the primary model.
+ */
+export type ModelPolicyChain = ModelPolicy[];

package/dist/src/availability/modelPolicy.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};
+//# sourceMappingURL=modelPolicy.js.map

package/dist/src/availability/modelPolicy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelPolicy.js","sourceRoot":"","sources":["../../../src/availability/modelPolicy.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

package/dist/src/availability/policyCatalog.d.ts

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ModelPolicy, ModelPolicyChain } from './modelPolicy.js';
+import type { UserTierId } from '../code_assist/types.js';
+export interface ModelPolicyOptions {
+    previewEnabled: boolean;
+    userTier?: UserTierId;
+}
+/**
+ * Returns the default ordered model policy chain for the user.
+ */
+export declare function getModelPolicyChain(options: ModelPolicyOptions): ModelPolicyChain;
+/**
+ * Provides a default policy scaffold for models not present in the catalog.
+ */
+export declare function createDefaultPolicy(model: string): ModelPolicy;
+export declare function validateModelPolicyChain(chain: ModelPolicyChain): void;

package/dist/src/availability/policyCatalog.js

@@ -0,0 +1,83 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, } from '../config/models.js';
+const DEFAULT_ACTIONS = {
+    terminal: 'prompt',
+    transient: 'prompt',
+    not_found: 'prompt',
+    unknown: 'prompt',
+};
+const DEFAULT_STATE = {
+    terminal: 'terminal',
+    transient: 'terminal',
+    not_found: 'terminal',
+    unknown: 'terminal',
+};
+const DEFAULT_CHAIN = [
+    definePolicy({ model: DEFAULT_GEMINI_MODEL }),
+    definePolicy({ model: DEFAULT_GEMINI_FLASH_MODEL, isLastResort: true }),
+];
+const PREVIEW_CHAIN = [
+    definePolicy({
+        model: PREVIEW_GEMINI_MODEL,
+        stateTransitions: { transient: 'sticky_retry' },
+    }),
+    definePolicy({ model: DEFAULT_GEMINI_MODEL }),
+    definePolicy({ model: DEFAULT_GEMINI_FLASH_MODEL, isLastResort: true }),
+];
+/**
+ * Returns the default ordered model policy chain for the user.
+ */
+export function getModelPolicyChain(options) {
+    if (options.previewEnabled) {
+        return cloneChain(PREVIEW_CHAIN);
+    }
+    return cloneChain(DEFAULT_CHAIN);
+}
+/**
+ * Provides a default policy scaffold for models not present in the catalog.
+ */
+export function createDefaultPolicy(model) {
+    return definePolicy({ model });
+}
+export function validateModelPolicyChain(chain) {
+    if (chain.length === 0) {
+        throw new Error('Model policy chain must include at least one model.');
+    }
+    const lastResortCount = chain.filter((policy) => policy.isLastResort).length;
+    if (lastResortCount === 0) {
+        throw new Error('Model policy chain must include an `isLastResort` model.');
+    }
+    if (lastResortCount > 1) {
+        throw new Error('Model policy chain must only have one `isLastResort`.');
+    }
+}
+/**
+ * Helper to define a ModelPolicy with default actions and state transitions.
+ * Ensures every policy is a fresh instance to avoid shared state.
+ */
+function definePolicy(config) {
+    return {
+        model: config.model,
+        isLastResort: config.isLastResort,
+        actions: { ...DEFAULT_ACTIONS, ...(config.actions ?? {}) },
+        stateTransitions: {
+            ...DEFAULT_STATE,
+            ...(config.stateTransitions ?? {}),
+        },
+    };
+}
+function clonePolicy(policy) {
+    return {
+        ...policy,
+        actions: { ...policy.actions },
+        stateTransitions: { ...policy.stateTransitions },
+    };
+}
+function cloneChain(chain) {
+    return chain.map(clonePolicy);
+}
+//# sourceMappingURL=policyCatalog.js.map

package/dist/src/availability/policyCatalog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policyCatalog.js","sourceRoot":"","sources":["../../../src/availability/policyCatalog.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,OAAO,EACL,0BAA0B,EAC1B,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAc7B,MAAM,eAAe,GAAyB;IAC5C,QAAQ,EAAE,QAAQ;IAClB,SAAS,EAAE,QAAQ;IACnB,SAAS,EAAE,QAAQ;IACnB,OAAO,EAAE,QAAQ;CAClB,CAAC;AAEF,MAAM,aAAa,GAAwB;IACzC,QAAQ,EAAE,UAAU;IACpB,SAAS,EAAE,UAAU;IACrB,SAAS,EAAE,UAAU;IACrB,OAAO,EAAE,UAAU;CACpB,CAAC;AAEF,MAAM,aAAa,GAAqB;IACtC,YAAY,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC;IAC7C,YAAY,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;CACxE,CAAC;AAEF,MAAM,aAAa,GAAqB;IACtC,YAAY,CAAC;QACX,KAAK,EAAE,oBAAoB;QAC3B,gBAAgB,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE;KAChD,CAAC;IACF,YAAY,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC;IAC7C,YAAY,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;CACxE,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,OAA2B;IAE3B,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;QAC3B,OAAO,UAAU,CAAC,aAAa,CAAC,CAAC;IACnC,CAAC;IAED,OAAO,UAAU,CAAC,aAAa,CAAC,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,KAAa;IAC/C,OAAO,YAAY,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,KAAuB;IAC9D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IACD,MAAM,eAAe,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC;IAC7E,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC9E,CAAC;IACD,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;IAC3E,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,YAAY,CAAC,MAAoB;IACxC,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,OAAO,EAAE,EAAE,GAAG,eAAe,EAAE,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE;QAC1D,gBAAgB,EAAE;YAChB,GAAG,aAAa;YAChB,GAAG,CAAC,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC;SACnC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,MAAmB;IACtC,OAAO;QACL,GAAG,MAAM;QACT,OAAO,EAAE,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE;QAC9B,gBAAgB,EAAE,EAAE,GAAG,MAAM,CAAC,gBAAgB,EAAE;KACjD,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,KAAuB;IACzC,OAAO,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AAChC,CAAC"}

package/dist/src/availability/policyCatalog.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/availability/policyCatalog.test.js

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { createDefaultPolicy, getModelPolicyChain, validateModelPolicyChain, } from './policyCatalog.js';
+import { DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, } from '../config/models.js';
+describe('policyCatalog', () => {
+    it('returns preview chain when preview enabled', () => {
+        const chain = getModelPolicyChain({ previewEnabled: true });
+        expect(chain[0]?.model).toBe(PREVIEW_GEMINI_MODEL);
+        expect(chain).toHaveLength(3);
+    });
+    it('returns default chain when preview disabled', () => {
+        const chain = getModelPolicyChain({ previewEnabled: false });
+        expect(chain[0]?.model).toBe(DEFAULT_GEMINI_MODEL);
+        expect(chain).toHaveLength(2);
+    });
+    it('marks preview transients as sticky retries', () => {
+        const [previewPolicy] = getModelPolicyChain({ previewEnabled: true });
+        expect(previewPolicy.model).toBe(PREVIEW_GEMINI_MODEL);
+        expect(previewPolicy.stateTransitions.transient).toBe('sticky_retry');
+    });
+    it('applies default actions and state transitions for unspecified kinds', () => {
+        const [previewPolicy] = getModelPolicyChain({ previewEnabled: true });
+        expect(previewPolicy.stateTransitions.not_found).toBe('terminal');
+        expect(previewPolicy.stateTransitions.unknown).toBe('terminal');
+        expect(previewPolicy.actions.unknown).toBe('prompt');
+    });
+    it('clones policy maps so edits do not leak between calls', () => {
+        const firstCall = getModelPolicyChain({ previewEnabled: false });
+        firstCall[0].actions.terminal = 'silent';
+        const secondCall = getModelPolicyChain({ previewEnabled: false });
+        expect(secondCall[0].actions.terminal).toBe('prompt');
+    });
+    it('passes when there is exactly one last-resort policy', () => {
+        const validChain = [
+            createDefaultPolicy('test-model'),
+            { ...createDefaultPolicy('last-resort'), isLastResort: true },
+        ];
+        expect(() => validateModelPolicyChain(validChain)).not.toThrow();
+    });
+    it('fails when no policies are marked last-resort', () => {
+        const chain = [
+            createDefaultPolicy('model-a'),
+            createDefaultPolicy('model-b'),
+        ];
+        expect(() => validateModelPolicyChain(chain)).toThrow('must include an `isLastResort`');
+    });
+    it('fails when a single-model chain is not last-resort', () => {
+        const chain = [createDefaultPolicy('lonely-model')];
+        expect(() => validateModelPolicyChain(chain)).toThrow('must include an `isLastResort`');
+    });
+    it('fails when multiple policies are marked last-resort', () => {
+        const chain = [
+            { ...createDefaultPolicy('model-a'), isLastResort: true },
+            { ...createDefaultPolicy('model-b'), isLastResort: true },
+        ];
+        expect(() => validateModelPolicyChain(chain)).toThrow('must only have one `isLastResort`');
+    });
+    it('createDefaultPolicy seeds default actions and states', () => {
+        const policy = createDefaultPolicy('custom');
+        expect(policy.actions.terminal).toBe('prompt');
+        expect(policy.actions.unknown).toBe('prompt');
+        expect(policy.stateTransitions.terminal).toBe('terminal');
+        expect(policy.stateTransitions.unknown).toBe('terminal');
+    });
+});
+//# sourceMappingURL=policyCatalog.test.js.map

package/dist/src/availability/policyCatalog.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policyCatalog.test.js","sourceRoot":"","sources":["../../../src/availability/policyCatalog.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,mBAAmB,EACnB,mBAAmB,EACnB,wBAAwB,GACzB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAE7B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,KAAK,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,aAAa,CAAC,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACvD,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,CAAC,aAAa,CAAC,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClE,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChE,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,SAAS,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,SAAS,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC1C,MAAM,UAAU,GAAG,mBAAmB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,UAAU,GAAG;YACjB,mBAAmB,CAAC,YAAY,CAAC;YACjC,EAAE,GAAG,mBAAmB,CAAC,aAAa,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE;SAC9D,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,KAAK,GAAG;YACZ,mBAAmB,CAAC,SAAS,CAAC;YAC9B,mBAAmB,CAAC,SAAS,CAAC;SAC/B,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACnD,gCAAgC,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,KAAK,GAAG,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACnD,gCAAgC,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG;YACZ,EAAE,GAAG,mBAAmB,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE;YACzD,EAAE,GAAG,mBAAmB,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE;SAC1D,CAAC;QACF,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CACnD,mCAAmC,CACpC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/config/config.d.ts

@@ -31,6 +31,7 @@ import type { EventEmitter } from 'node:events';
 import { MessageBus } from '../confirmation-bus/message-bus.js';
 import { PolicyEngine } from '../policy/policy-engine.js';
 import type { PolicyEngineConfig } from '../policy/types.js';
+import { HookSystem } from '../hooks/index.js';
 import type { UserTierId } from '../code_assist/types.js';
 import type { Experiments } from '../code_assist/experiments/experiments.js';
 import { AgentRegistry } from '../agents/registry.js';
@@ -206,7 +207,6 @@ export interface ConfigParameters {
     useWriteTodos?: boolean;
     policyEngineConfig?: PolicyEngineConfig;
     output?: OutputSettings;
-    useModelRouter?: boolean;
     enableMessageBusIntegration?: boolean;
     disableModelRouterForAuth?: AuthType[];
     codebaseInvestigatorSettings?: CodebaseInvestigatorSettings;
@@ -224,6 +224,7 @@ export interface ConfigParameters {
         [K in HookEventName]?: HookDefinition[];
     };
     previewFeatures?: boolean;
+    enableModelAvailabilityService?: boolean;
 }
 export declare class Config {
     private toolRegistry;
@@ -314,9 +315,6 @@ export declare class Config {
     private readonly messageBus;
     private readonly policyEngine;
     private readonly outputSettings;
-    private useModelRouter;
-    private readonly initialUseModelRouter;
-    private readonly disableModelRouterForAuth?;
     private readonly enableMessageBusIntegration;
     private readonly codebaseInvestigatorSettings;
     private readonly continueOnFailedApiCall;
@@ -330,8 +328,10 @@ export declare class Config {
     private readonly hooks;
     private experiments;
     private experimentsPromise;
+    private hookSystem?;
     private previewModelFallbackMode;
     private previewModelBypassMode;
+    private readonly enableModelAvailabilityService;
     constructor(params: ConfigParameters);
     /**
      * Must only be called once, throws if called again.
@@ -339,6 +339,7 @@ export declare class Config {
     initialize(): Promise<void>;
     getContentGenerator(): ContentGenerator;
     refreshAuth(authMethod: AuthType, displayMessage?: (message: string) => void): Promise<void>;
+    getExperimentsAsync(): Promise<Experiments | undefined>;
     getUserTier(): UserTierId | undefined;
     /**
      * Provides access to the BaseLlmClient for stateless LLM operations.
@@ -455,6 +456,7 @@ export declare class Config {
     getExtensionLoader(): ExtensionLoader;
     getEnabledExtensions(): string[];
     getEnableExtensionReloading(): boolean;
+    isModelAvailabilityServiceEnabled(): boolean;
     getNoBrowser(): boolean;
     isBrowserLaunchSuppressed(): boolean;
     getSummarizeToolOutputConfig(): Record<string, SummarizeToolOutputSettings> | undefined;
@@ -500,7 +502,6 @@ export declare class Config {
     getUseSmartEdit(): boolean;
     getUseWriteTodos(): boolean;
     getOutputFormat(): OutputFormat;
-    getUseModelRouter(): boolean;
     getGitService(): Promise<GitService>;
     getFileExclusions(): FileExclusions;
     getMessageBus(): MessageBus;
@@ -509,6 +510,10 @@ export declare class Config {
     getEnableHooks(): boolean;
     getCodebaseInvestigatorSettings(): CodebaseInvestigatorSettings;
     createToolRegistry(): Promise<ToolRegistry>;
+    /**
+     * Get the hook system instance
+     */
+    getHookSystem(): HookSystem | undefined;
     /**
      * Get hooks configuration
      */