@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/prompts/prompt-registry.test.js

@@ -0,0 +1,96 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { PromptRegistry } from './prompt-registry.js';
+import { debugLogger } from '../utils/debugLogger.js';
+vi.mock('../utils/debugLogger.js', () => ({
+    debugLogger: {
+        warn: vi.fn(),
+    },
+}));
+describe('PromptRegistry', () => {
+    let registry;
+    const prompt1 = {
+        name: 'prompt1',
+        serverName: 'server1',
+        invoke: async () => ({
+            messages: [
+                { role: 'assistant', content: { type: 'text', text: 'response1' } },
+            ],
+        }),
+    };
+    const prompt2 = {
+        name: 'prompt2',
+        serverName: 'server1',
+        invoke: async () => ({
+            messages: [
+                { role: 'assistant', content: { type: 'text', text: 'response2' } },
+            ],
+        }),
+    };
+    const prompt3 = {
+        name: 'prompt1',
+        serverName: 'server2',
+        invoke: async () => ({
+            messages: [
+                { role: 'assistant', content: { type: 'text', text: 'response3' } },
+            ],
+        }),
+    };
+    beforeEach(() => {
+        registry = new PromptRegistry();
+        vi.clearAllMocks();
+    });
+    it('should register a prompt', () => {
+        registry.registerPrompt(prompt1);
+        expect(registry.getPrompt('prompt1')).toEqual(prompt1);
+    });
+    it('should get all prompts, sorted by name', () => {
+        registry.registerPrompt(prompt2);
+        registry.registerPrompt(prompt1);
+        expect(registry.getAllPrompts()).toEqual([prompt1, prompt2]);
+    });
+    it('should get a specific prompt by name', () => {
+        registry.registerPrompt(prompt1);
+        expect(registry.getPrompt('prompt1')).toEqual(prompt1);
+        expect(registry.getPrompt('non-existent')).toBeUndefined();
+    });
+    it('should get prompts by server, sorted by name', () => {
+        registry.registerPrompt(prompt1);
+        registry.registerPrompt(prompt2);
+        registry.registerPrompt(prompt3); // different server
+        expect(registry.getPromptsByServer('server1')).toEqual([prompt1, prompt2]);
+        expect(registry.getPromptsByServer('server2')).toEqual([
+            { ...prompt3, name: 'server2_prompt1' },
+        ]);
+    });
+    it('should handle prompt name collision by renaming', () => {
+        registry.registerPrompt(prompt1);
+        registry.registerPrompt(prompt3);
+        expect(registry.getPrompt('prompt1')).toEqual(prompt1);
+        const renamedPrompt = { ...prompt3, name: 'server2_prompt1' };
+        expect(registry.getPrompt('server2_prompt1')).toEqual(renamedPrompt);
+        expect(debugLogger.warn).toHaveBeenCalledWith('Prompt with name "prompt1" is already registered. Renaming to "server2_prompt1".');
+    });
+    it('should clear all prompts', () => {
+        registry.registerPrompt(prompt1);
+        registry.registerPrompt(prompt2);
+        registry.clear();
+        expect(registry.getAllPrompts()).toEqual([]);
+    });
+    it('should remove prompts by server', () => {
+        registry.registerPrompt(prompt1);
+        registry.registerPrompt(prompt2);
+        registry.registerPrompt(prompt3);
+        registry.removePromptsByServer('server1');
+        const renamedPrompt = { ...prompt3, name: 'server2_prompt1' };
+        expect(registry.getAllPrompts()).toEqual([renamedPrompt]);
+        expect(registry.getPrompt('prompt1')).toBeUndefined();
+        expect(registry.getPrompt('prompt2')).toBeUndefined();
+        expect(registry.getPrompt('server2_prompt1')).toEqual(renamedPrompt);
+    });
+});
+//# sourceMappingURL=prompt-registry.test.js.map

package/dist/src/prompts/prompt-registry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-registry.test.js","sourceRoot":"","sources":["../../../src/prompts/prompt-registry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE;QACX,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;KACd;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,QAAwB,CAAC;IAE7B,MAAM,OAAO,GAAwB;QACnC,IAAI,EAAE,SAAS;QACf,UAAU,EAAE,SAAS;QACrB,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACnB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;aACpE;SACF,CAAC;KACH,CAAC;IAEF,MAAM,OAAO,GAAwB;QACnC,IAAI,EAAE,SAAS;QACf,UAAU,EAAE,SAAS;QACrB,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACnB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;aACpE;SACF,CAAC;KACH,CAAC;IAEF,MAAM,OAAO,GAAwB;QACnC,IAAI,EAAE,SAAS;QACf,UAAU,EAAE,SAAS;QACrB,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;YACnB,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;aACpE;SACF,CAAC;KACH,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,cAAc,EAAE,CAAC;QAChC,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB;QACrD,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC3E,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;YACrD,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE;SACxC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvD,MAAM,aAAa,GAAG,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACrE,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC3C,kFAAkF,CACnF,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACjB,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjC,QAAQ,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;QAE1C,MAAM,aAAa,GAAG,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACtD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACtD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/routing/modelRouterService.d.ts

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { RoutingContext, RoutingDecision } from './routingStrategy.js';
+/**
+ * A centralized service for making model routing decisions.
+ */
+export declare class ModelRouterService {
+    private config;
+    private strategy;
+    constructor(config: Config);
+    private initializeDefaultStrategy;
+    /**
+     * Determines which model to use for a given request context.
+     *
+     * @param context The full context of the request.
+     * @returns A promise that resolves to a RoutingDecision.
+     */
+    route(context: RoutingContext): Promise<RoutingDecision>;
+}

package/dist/src/routing/modelRouterService.js

@@ -0,0 +1,85 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { PREVIEW_GEMINI_MODEL, DEFAULT_GEMINI_MODEL, } from '../config/models.js';
+import { DefaultStrategy } from './strategies/defaultStrategy.js';
+import { ClassifierStrategy } from './strategies/classifierStrategy.js';
+import { CompositeStrategy } from './strategies/compositeStrategy.js';
+import { FallbackStrategy } from './strategies/fallbackStrategy.js';
+import { OverrideStrategy } from './strategies/overrideStrategy.js';
+import { logModelRouting } from '../telemetry/loggers.js';
+import { ModelRoutingEvent } from '../telemetry/types.js';
+/**
+ * A centralized service for making model routing decisions.
+ */
+export class ModelRouterService {
+    config;
+    strategy;
+    constructor(config) {
+        this.config = config;
+        this.strategy = this.initializeDefaultStrategy();
+    }
+    initializeDefaultStrategy() {
+        // Initialize the composite strategy with the desired priority order.
+        // The strategies are ordered in order of highest priority.
+        return new CompositeStrategy([
+            new FallbackStrategy(),
+            new OverrideStrategy(),
+            new ClassifierStrategy(),
+            new DefaultStrategy(),
+        ], 'agent-router');
+    }
+    /**
+     * Determines which model to use for a given request context.
+     *
+     * @param context The full context of the request.
+     * @returns A promise that resolves to a RoutingDecision.
+     */
+    async route(context) {
+        const startTime = Date.now();
+        let decision;
+        try {
+            decision = await this.strategy.route(context, this.config, this.config.getBaseLlmClient());
+            // Unified Preview Model Logic:
+            // If the decision is to use 'gemini-2.5-pro' and preview features are enabled,
+            // we attempt to upgrade to 'gemini-3.0-pro' (Preview Model).
+            if (decision.model === DEFAULT_GEMINI_MODEL &&
+                this.config.getPreviewFeatures() &&
+                !decision.metadata.source.includes('override')) {
+                // We ALWAYS attempt to upgrade to Preview Model here.
+                // If we are in fallback mode, the 'previewModelBypassMode' flag (handled in handler.ts/geminiChat.ts)
+                // will ensure we downgrade to 2.5 Pro for the actual API call if needed.
+                // This allows us to "probe" Preview Model periodically (i.e., every new request tries Preview Model first).
+                decision.model = PREVIEW_GEMINI_MODEL;
+                decision.metadata.source += ' (Preview Model)';
+                decision.metadata.reasoning += ' (Upgraded to Preview Model)';
+            }
+            const event = new ModelRoutingEvent(decision.model, decision.metadata.source, decision.metadata.latencyMs, decision.metadata.reasoning, false, // failed
+            undefined);
+            logModelRouting(this.config, event);
+            return decision;
+        }
+        catch (e) {
+            const failed = true;
+            const error_message = e instanceof Error ? e.message : String(e);
+            // Create a fallback decision for logging purposes
+            // We do not actually route here. This should never happen so we should
+            // fail loudly to catch any issues where this happens.
+            decision = {
+                model: this.config.getModel(),
+                metadata: {
+                    source: 'router-exception',
+                    latencyMs: Date.now() - startTime,
+                    reasoning: 'An exception occurred during routing.',
+                    error: error_message,
+                },
+            };
+            const event = new ModelRoutingEvent(decision.model, decision.metadata.source, decision.metadata.latencyMs, decision.metadata.reasoning, failed, error_message);
+            logModelRouting(this.config, event);
+            throw e;
+        }
+    }
+}
+//# sourceMappingURL=modelRouterService.js.map

package/dist/src/routing/modelRouterService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelRouterService.js","sourceRoot":"","sources":["../../../src/routing/modelRouterService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EACL,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAM7B,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AAEpE,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1D;;GAEG;AACH,MAAM,OAAO,kBAAkB;IACrB,MAAM,CAAS;IACf,QAAQ,CAAmB;IAEnC,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACnD,CAAC;IAEO,yBAAyB;QAC/B,qEAAqE;QACrE,2DAA2D;QAC3D,OAAO,IAAI,iBAAiB,CAC1B;YACE,IAAI,gBAAgB,EAAE;YACtB,IAAI,gBAAgB,EAAE;YACtB,IAAI,kBAAkB,EAAE;YACxB,IAAI,eAAe,EAAE;SACtB,EACD,cAAc,CACf,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,OAAuB;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,QAAyB,CAAC;QAE9B,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAClC,OAAO,EACP,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAC/B,CAAC;YAEF,+BAA+B;YAC/B,+EAA+E;YAC/E,6DAA6D;YAC7D,IACE,QAAQ,CAAC,KAAK,KAAK,oBAAoB;gBACvC,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE;gBAChC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAC9C,CAAC;gBACD,sDAAsD;gBACtD,sGAAsG;gBACtG,yEAAyE;gBACzE,4GAA4G;gBAC5G,QAAQ,CAAC,KAAK,GAAG,oBAAoB,CAAC;gBACtC,QAAQ,CAAC,QAAQ,CAAC,MAAM,IAAI,kBAAkB,CAAC;gBAC/C,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,8BAA8B,CAAC;YAChE,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,iBAAiB,CACjC,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,QAAQ,CAAC,MAAM,EACxB,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAC3B,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAC3B,KAAK,EAAE,SAAS;YAChB,SAAS,CACV,CAAC;YACF,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAEpC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,IAAI,CAAC;YACpB,MAAM,aAAa,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACjE,kDAAkD;YAClD,uEAAuE;YACvE,sDAAsD;YACtD,QAAQ,GAAG;gBACT,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;gBAC7B,QAAQ,EAAE;oBACR,MAAM,EAAE,kBAAkB;oBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBACjC,SAAS,EAAE,uCAAuC;oBAClD,KAAK,EAAE,aAAa;iBACrB;aACF,CAAC;YAEF,MAAM,KAAK,GAAG,IAAI,iBAAiB,CACjC,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,QAAQ,CAAC,MAAM,EACxB,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAC3B,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAC3B,MAAM,EACN,aAAa,CACd,CAAC;YAEF,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAEpC,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;CACF"}

package/dist/src/routing/modelRouterService.test.d.ts

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

package/dist/src/routing/modelRouterService.test.js

@@ -0,0 +1,160 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { ModelRouterService } from './modelRouterService.js';
+import { Config } from '../config/config.js';
+import { PREVIEW_GEMINI_MODEL, DEFAULT_GEMINI_MODEL, } from '../config/models.js';
+import { DefaultStrategy } from './strategies/defaultStrategy.js';
+import { CompositeStrategy } from './strategies/compositeStrategy.js';
+import { FallbackStrategy } from './strategies/fallbackStrategy.js';
+import { OverrideStrategy } from './strategies/overrideStrategy.js';
+import { ClassifierStrategy } from './strategies/classifierStrategy.js';
+import { logModelRouting } from '../telemetry/loggers.js';
+import { ModelRoutingEvent } from '../telemetry/types.js';
+vi.mock('../config/config.js');
+vi.mock('../core/baseLlmClient.js');
+vi.mock('./strategies/defaultStrategy.js');
+vi.mock('./strategies/compositeStrategy.js');
+vi.mock('./strategies/fallbackStrategy.js');
+vi.mock('./strategies/overrideStrategy.js');
+vi.mock('./strategies/classifierStrategy.js');
+vi.mock('../telemetry/loggers.js');
+vi.mock('../telemetry/types.js');
+describe('ModelRouterService', () => {
+    let service;
+    let mockConfig;
+    let mockBaseLlmClient;
+    let mockContext;
+    let mockCompositeStrategy;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockConfig = new Config({});
+        mockBaseLlmClient = {};
+        vi.spyOn(mockConfig, 'getBaseLlmClient').mockReturnValue(mockBaseLlmClient);
+        mockCompositeStrategy = new CompositeStrategy([
+            new FallbackStrategy(),
+            new OverrideStrategy(),
+            new ClassifierStrategy(),
+            new DefaultStrategy(),
+        ], 'agent-router');
+        vi.mocked(CompositeStrategy).mockImplementation(() => mockCompositeStrategy);
+        service = new ModelRouterService(mockConfig);
+        mockContext = {
+            history: [],
+            request: [{ text: 'test prompt' }],
+            signal: new AbortController().signal,
+        };
+    });
+    it('should initialize with a CompositeStrategy', () => {
+        expect(CompositeStrategy).toHaveBeenCalled();
+        expect(service['strategy']).toBeInstanceOf(CompositeStrategy);
+    });
+    it('should initialize the CompositeStrategy with the correct child strategies in order', () => {
+        // This test relies on the mock implementation detail of the constructor
+        const compositeStrategyArgs = vi.mocked(CompositeStrategy).mock.calls[0];
+        const childStrategies = compositeStrategyArgs[0];
+        expect(childStrategies.length).toBe(4);
+        expect(childStrategies[0]).toBeInstanceOf(FallbackStrategy);
+        expect(childStrategies[1]).toBeInstanceOf(OverrideStrategy);
+        expect(childStrategies[2]).toBeInstanceOf(ClassifierStrategy);
+        expect(childStrategies[3]).toBeInstanceOf(DefaultStrategy);
+        expect(compositeStrategyArgs[1]).toBe('agent-router');
+    });
+    describe('route()', () => {
+        const strategyDecision = {
+            model: 'strategy-chosen-model',
+            metadata: {
+                source: 'test-router/fallback',
+                latencyMs: 10,
+                reasoning: 'Strategy reasoning',
+            },
+        };
+        it('should delegate routing to the composite strategy', async () => {
+            const strategySpy = vi
+                .spyOn(mockCompositeStrategy, 'route')
+                .mockResolvedValue(strategyDecision);
+            const decision = await service.route(mockContext);
+            expect(strategySpy).toHaveBeenCalledWith(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual(strategyDecision);
+        });
+        it('should log a telemetry event on a successful decision', async () => {
+            vi.spyOn(mockCompositeStrategy, 'route').mockResolvedValue(strategyDecision);
+            await service.route(mockContext);
+            expect(ModelRoutingEvent).toHaveBeenCalledWith('strategy-chosen-model', 'test-router/fallback', 10, 'Strategy reasoning', false, undefined);
+            expect(logModelRouting).toHaveBeenCalledWith(mockConfig, expect.any(ModelRoutingEvent));
+        });
+        it('should log a telemetry event and re-throw on a failed decision', async () => {
+            const testError = new Error('Strategy failed');
+            vi.spyOn(mockCompositeStrategy, 'route').mockRejectedValue(testError);
+            vi.spyOn(mockConfig, 'getModel').mockReturnValue('default-model');
+            await expect(service.route(mockContext)).rejects.toThrow(testError);
+            expect(ModelRoutingEvent).toHaveBeenCalledWith('default-model', 'router-exception', expect.any(Number), 'An exception occurred during routing.', true, 'Strategy failed');
+            expect(logModelRouting).toHaveBeenCalledWith(mockConfig, expect.any(ModelRoutingEvent));
+        });
+        it('should upgrade to Preview Model when preview features are enabled and model is 2.5 Pro', async () => {
+            vi.spyOn(mockCompositeStrategy, 'route').mockResolvedValue({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: { source: 'test', latencyMs: 0, reasoning: 'test' },
+            });
+            vi.spyOn(mockConfig, 'getPreviewFeatures').mockReturnValue(true);
+            vi.spyOn(mockConfig, 'isPreviewModelFallbackMode').mockReturnValue(false);
+            const decision = await service.route(mockContext);
+            expect(decision.model).toBe(PREVIEW_GEMINI_MODEL);
+        });
+        it('should NOT upgrade to Preview Model when preview features are disabled', async () => {
+            vi.spyOn(mockCompositeStrategy, 'route').mockResolvedValue({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: { source: 'test', latencyMs: 0, reasoning: 'test' },
+            });
+            vi.spyOn(mockConfig, 'getPreviewFeatures').mockReturnValue(false);
+            const decision = await service.route(mockContext);
+            expect(decision.model).toBe(DEFAULT_GEMINI_MODEL);
+        });
+        it('should upgrade to Preview Model when preview features are enabled and model is explicitly set to Pro', async () => {
+            // Simulate OverrideStrategy returning Preview Model (as resolveModel would do for "pro")
+            vi.spyOn(mockCompositeStrategy, 'route').mockResolvedValue({
+                model: PREVIEW_GEMINI_MODEL,
+                metadata: {
+                    source: 'override',
+                    latencyMs: 0,
+                    reasoning: 'User selected',
+                },
+            });
+            vi.spyOn(mockConfig, 'getPreviewFeatures').mockReturnValue(true);
+            vi.spyOn(mockConfig, 'isPreviewModelFallbackMode').mockReturnValue(false);
+            const decision = await service.route(mockContext);
+            expect(decision.model).toBe(PREVIEW_GEMINI_MODEL);
+        });
+        it('should NOT upgrade to Preview Model when preview features are enabled and model is explicitly set to a specific string', async () => {
+            // Simulate OverrideStrategy returning a specific model (e.g. "gemini-2.5-pro")
+            // This happens when user explicitly sets model to "gemini-2.5-pro" instead of "pro"
+            vi.spyOn(mockCompositeStrategy, 'route').mockResolvedValue({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: {
+                    source: 'override',
+                    latencyMs: 0,
+                    reasoning: 'User selected',
+                },
+            });
+            vi.spyOn(mockConfig, 'getPreviewFeatures').mockReturnValue(true);
+            vi.spyOn(mockConfig, 'isPreviewModelFallbackMode').mockReturnValue(false);
+            const decision = await service.route(mockContext);
+            // Should NOT upgrade to Preview Model because source is 'override' and model is specific
+            expect(decision.model).toBe(DEFAULT_GEMINI_MODEL);
+        });
+        it('should upgrade to Preview Model even if fallback mode is active (probing behavior)', async () => {
+            vi.spyOn(mockCompositeStrategy, 'route').mockResolvedValue({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: { source: 'default', latencyMs: 0, reasoning: 'Default' },
+            });
+            vi.spyOn(mockConfig, 'getPreviewFeatures').mockReturnValue(true);
+            vi.spyOn(mockConfig, 'isPreviewModelFallbackMode').mockReturnValue(true);
+            const decision = await service.route(mockContext);
+            expect(decision.model).toBe(PREVIEW_GEMINI_MODEL);
+        });
+    });
+});
+//# sourceMappingURL=modelRouterService.test.js.map

package/dist/src/routing/modelRouterService.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelRouterService.test.js","sourceRoot":"","sources":["../../../src/routing/modelRouterService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EACL,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1D,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;AAC3C,EAAE,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;AAC7C,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;AAC5C,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;AAC5C,EAAE,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;AAC9C,EAAE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;AACnC,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AAEjC,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,OAA2B,CAAC;IAChC,IAAI,UAAkB,CAAC;IACvB,IAAI,iBAAgC,CAAC;IACrC,IAAI,WAA2B,CAAC;IAChC,IAAI,qBAAwC,CAAC;IAE7C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG,IAAI,MAAM,CAAC,EAAW,CAAC,CAAC;QACrC,iBAAiB,GAAG,EAAmB,CAAC;QACxC,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC;QAE5E,qBAAqB,GAAG,IAAI,iBAAiB,CAC3C;YACE,IAAI,gBAAgB,EAAE;YACtB,IAAI,gBAAgB,EAAE;YACtB,IAAI,kBAAkB,EAAE;YACxB,IAAI,eAAe,EAAE;SACtB,EACD,cAAc,CACf,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,kBAAkB,CAC7C,GAAG,EAAE,CAAC,qBAAqB,CAC5B,CAAC;QAEF,OAAO,GAAG,IAAI,kBAAkB,CAAC,UAAU,CAAC,CAAC;QAE7C,WAAW,GAAG;YACZ,OAAO,EAAE,EAAE;YACX,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAClC,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACrC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oFAAoF,EAAE,GAAG,EAAE;QAC5F,wEAAwE;QACxE,MAAM,qBAAqB,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzE,MAAM,eAAe,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEjD,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QAC5D,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QAC5D,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;QAC9D,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAC3D,MAAM,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,MAAM,gBAAgB,GAAoB;YACxC,KAAK,EAAE,uBAAuB;YAC9B,QAAQ,EAAE;gBACR,MAAM,EAAE,sBAAsB;gBAC9B,SAAS,EAAE,EAAE;gBACb,SAAS,EAAE,oBAAoB;aAChC;SACF,CAAC;QAEF,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC;iBACrC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAEvC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CACxD,gBAAgB,CACjB,CAAC;YAEF,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEjC,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,uBAAuB,EACvB,sBAAsB,EACtB,EAAE,EACF,oBAAoB,EACpB,KAAK,EACL,SAAS,CACV,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC/C,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YACtE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAElE,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAEpE,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,eAAe,EACf,kBAAkB,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,uCAAuC,EACvC,IAAI,EACJ,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wFAAwF,EAAE,KAAK,IAAI,EAAE;YACtG,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBACzD,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE;aAC9D,CAAC,CAAC;YACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACjE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,4BAA4B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAE1E,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBACzD,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE;aAC9D,CAAC,CAAC;YACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAElE,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sGAAsG,EAAE,KAAK,IAAI,EAAE;YACpH,yFAAyF;YACzF,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBACzD,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE;oBACR,MAAM,EAAE,UAAU;oBAClB,SAAS,EAAE,CAAC;oBACZ,SAAS,EAAE,eAAe;iBAC3B;aACF,CAAC,CAAC;YACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACjE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,4BAA4B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAE1E,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wHAAwH,EAAE,KAAK,IAAI,EAAE;YACtI,+EAA+E;YAC/E,oFAAoF;YACpF,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBACzD,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE;oBACR,MAAM,EAAE,UAAU;oBAClB,SAAS,EAAE,CAAC;oBACZ,SAAS,EAAE,eAAe;iBAC3B;aACF,CAAC,CAAC;YACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACjE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,4BAA4B,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAE1E,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,yFAAyF;YACzF,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAClG,EAAE,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC;gBACzD,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE;aACpE,CAAC,CAAC;YACH,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACjE,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,4BAA4B,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAEzE,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/routing/routingStrategy.d.ts

@@ -0,0 +1,62 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Content, PartListUnion } from '@google/genai';
+import type { BaseLlmClient } from '../core/baseLlmClient.js';
+import type { Config } from '../config/config.js';
+/**
+ * The output of a routing decision. It specifies which model to use and why.
+ */
+export interface RoutingDecision {
+    /** The model identifier string to use for the next API call (e.g., 'gemini-2.5-pro'). */
+    model: string;
+    /**
+     * Metadata about the routing decision for logging purposes.
+     */
+    metadata: {
+        source: string;
+        latencyMs: number;
+        reasoning: string;
+        error?: string;
+    };
+}
+/**
+ * The context provided to the router for making a decision.
+ */
+export interface RoutingContext {
+    /** The full history of the conversation. */
+    history: Content[];
+    /** The immediate request parts to be processed. */
+    request: PartListUnion;
+    /** An abort signal to cancel an LLM call during routing. */
+    signal: AbortSignal;
+}
+/**
+ * The core interface that all routing strategies must implement.
+ * Strategies implementing this interface may decline a request by returning null.
+ */
+export interface RoutingStrategy {
+    /** The name of the strategy (e.g., 'fallback', 'override', 'composite'). */
+    readonly name: string;
+    /**
+     * Determines which model to use for a given request context.
+     * @param context The full context of the request.
+     * @param config The current configuration.
+     * @param client A reference to the GeminiClient, allowing the strategy to make its own API calls if needed.
+     * @returns A promise that resolves to a RoutingDecision, or null if the strategy is not applicable.
+     */
+    route(context: RoutingContext, config: Config, baseLlmClient: BaseLlmClient): Promise<RoutingDecision | null>;
+}
+/**
+ * A strategy that is guaranteed to return a decision. It must not return null.
+ * This is used to ensure that a composite chain always terminates.
+ */
+export interface TerminalStrategy extends RoutingStrategy {
+    /**
+     * Determines which model to use for a given request context.
+     * @returns A promise that resolves to a RoutingDecision.
+     */
+    route(context: RoutingContext, config: Config, baseLlmClient: BaseLlmClient): Promise<RoutingDecision>;
+}

package/dist/src/routing/routingStrategy.js

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

package/dist/src/routing/routingStrategy.js.map

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

package/dist/src/routing/strategies/classifierStrategy.d.ts

@@ -0,0 +1,12 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { BaseLlmClient } from '../../core/baseLlmClient.js';
+import type { RoutingContext, RoutingDecision, RoutingStrategy } from '../routingStrategy.js';
+import type { Config } from '../../config/config.js';
+export declare class ClassifierStrategy implements RoutingStrategy {
+    readonly name = "classifier";
+    route(context: RoutingContext, config: Config, baseLlmClient: BaseLlmClient): Promise<RoutingDecision | null>;
+}