@didim365/agent-cli-core 0.1.1 → 0.1.2

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

@@ -1,106 +0,0 @@
-/**
- * @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 { 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 { NumericalClassifierStrategy } from './strategies/numericalClassifierStrategy.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('./strategies/numericalClassifierStrategy.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);
-        vi.spyOn(mockConfig, 'getNumericalRoutingEnabled').mockResolvedValue(false);
-        vi.spyOn(mockConfig, 'getClassifierThreshold').mockResolvedValue(undefined);
-        mockCompositeStrategy = new CompositeStrategy([
-            new FallbackStrategy(),
-            new OverrideStrategy(),
-            new ClassifierStrategy(),
-            new NumericalClassifierStrategy(),
-            new DefaultStrategy(),
-        ], 'agent-router');
-        vi.mocked(CompositeStrategy).mockImplementation(() => mockCompositeStrategy);
-        service = new ModelRouterService(mockConfig);
-        mockContext = {
-            history: [],
-            request: [{ type: 'text', 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(5);
-        expect(childStrategies[0]).toBeInstanceOf(FallbackStrategy);
-        expect(childStrategies[1]).toBeInstanceOf(OverrideStrategy);
-        expect(childStrategies[2]).toBeInstanceOf(ClassifierStrategy);
-        expect(childStrategies[3]).toBeInstanceOf(NumericalClassifierStrategy);
-        expect(childStrategies[4]).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, false, undefined);
-            expect(logModelRouting).toHaveBeenCalledWith(mockConfig, expect.any(ModelRoutingEvent));
-        });
-        it('should log a telemetry event and return fallback on a failed decision', async () => {
-            const testError = new Error('Strategy failed');
-            vi.spyOn(mockCompositeStrategy, 'route').mockRejectedValue(testError);
-            vi.spyOn(mockConfig, 'getModel').mockReturnValue('default-model');
-            const decision = await service.route(mockContext);
-            expect(decision.model).toBe('default-model');
-            expect(decision.metadata.source).toBe('router-exception');
-            expect(ModelRoutingEvent).toHaveBeenCalledWith('default-model', 'router-exception', expect.any(Number), 'An exception occurred during routing.', true, 'Strategy failed', false, undefined);
-            expect(logModelRouting).toHaveBeenCalledWith(mockConfig, expect.any(ModelRoutingEvent));
-        });
-    });
-});
-//# sourceMappingURL=modelRouterService.test.js.map

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

@@ -1 +0,0 @@
-{"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;AAI7C,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,2BAA2B,EAAE,MAAM,6CAA6C,CAAC;AAC1F,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,6CAA6C,CAAC,CAAC;AACvD,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;QAC5E,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,4BAA4B,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC5E,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,wBAAwB,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAE5E,qBAAqB,GAAG,IAAI,iBAAiB,CAC3C;YACE,IAAI,gBAAgB,EAAE;YACtB,IAAI,gBAAgB,EAAE;YACtB,IAAI,kBAAkB,EAAE;YACxB,IAAI,2BAA2B,EAAE;YACjC,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,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAChD,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,2BAA2B,CAAC,CAAC;QACvE,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,EACT,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,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACrF,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,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC7C,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,eAAe,EACf,kBAAkB,EAClB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,uCAAuC,EACvC,IAAI,EACJ,iBAAiB,EACjB,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;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

package/dist/src/routing/strategies/classifierStrategy.test.js

@@ -1,249 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { ClassifierStrategy } from './classifierStrategy.js';
-import { isToolCallMessage, isToolResultMessage, } from '../../utils/llmUtils.js';
-import { DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_MODEL_AUTO, } from '../../config/models.js';
-import { promptIdContext } from '../../utils/promptIdContext.js';
-import { debugLogger } from '../../utils/debugLogger.js';
-vi.mock('../../core/baseLlmClient.js');
-describe('ClassifierStrategy', () => {
-    let strategy;
-    let mockContext;
-    let mockConfig;
-    let mockBaseLlmClient;
-    let mockResolvedConfig;
-    beforeEach(() => {
-        vi.clearAllMocks();
-        strategy = new ClassifierStrategy();
-        mockContext = {
-            history: [],
-            request: [{ type: 'text', text: 'simple task' }],
-            signal: new AbortController().signal,
-        };
-        mockResolvedConfig = {
-            model: 'classifier',
-            generateContentConfig: {},
-        };
-        mockConfig = {
-            modelConfigService: {
-                getResolvedConfig: vi.fn().mockReturnValue(mockResolvedConfig),
-            },
-            getModel: () => DEFAULT_GEMINI_MODEL_AUTO,
-            getPreviewFeatures: () => false,
-            getNumericalRoutingEnabled: vi.fn().mockResolvedValue(false),
-        };
-        mockBaseLlmClient = {
-            generateJson: vi.fn(),
-        };
-        vi.spyOn(promptIdContext, 'getStore').mockReturnValue('test-prompt-id');
-    });
-    it('should return null if numerical routing is enabled', async () => {
-        vi.mocked(mockConfig.getNumericalRoutingEnabled).mockResolvedValue(true);
-        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(decision).toBeNull();
-        expect(mockBaseLlmClient.generateJson).not.toHaveBeenCalled();
-    });
-    it('should call generateJson with the correct parameters', async () => {
-        const mockApiResponse = {
-            reasoning: 'Simple task',
-            model_choice: 'flash',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockBaseLlmClient.generateJson).toHaveBeenCalledWith(expect.objectContaining({
-            modelConfigKey: { model: mockResolvedConfig.model },
-            promptId: 'test-prompt-id',
-        }));
-    });
-    it('should route to FLASH model for a simple task', async () => {
-        const mockApiResponse = {
-            reasoning: 'This is a simple task.',
-            model_choice: 'flash',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockBaseLlmClient.generateJson).toHaveBeenCalledOnce();
-        expect(decision).toEqual({
-            model: DEFAULT_GEMINI_FLASH_MODEL,
-            metadata: {
-                source: 'Classifier',
-                latencyMs: expect.any(Number),
-                reasoning: mockApiResponse.reasoning,
-            },
-        });
-    });
-    it('should route to PRO model for a complex task', async () => {
-        const mockApiResponse = {
-            reasoning: 'This is a complex task.',
-            model_choice: 'pro',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        mockContext.request = [
-            { type: 'text', text: 'how do I build a spaceship?' },
-        ];
-        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockBaseLlmClient.generateJson).toHaveBeenCalledOnce();
-        expect(decision).toEqual({
-            model: DEFAULT_GEMINI_MODEL,
-            metadata: {
-                source: 'Classifier',
-                latencyMs: expect.any(Number),
-                reasoning: mockApiResponse.reasoning,
-            },
-        });
-    });
-    it('should return null if the classifier API call fails', async () => {
-        const consoleWarnSpy = vi
-            .spyOn(debugLogger, 'warn')
-            .mockImplementation(() => { });
-        const testError = new Error('API Failure');
-        vi.mocked(mockBaseLlmClient.generateJson).mockRejectedValue(testError);
-        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(decision).toBeNull();
-        expect(consoleWarnSpy).toHaveBeenCalled();
-        consoleWarnSpy.mockRestore();
-    });
-    it('should return null if the classifier returns a malformed JSON object', async () => {
-        const consoleWarnSpy = vi
-            .spyOn(debugLogger, 'warn')
-            .mockImplementation(() => { });
-        const malformedApiResponse = {
-            reasoning: 'This is a simple task.',
-            // model_choice is missing, which will cause a Zod parsing error.
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(malformedApiResponse);
-        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(decision).toBeNull();
-        expect(consoleWarnSpy).toHaveBeenCalled();
-        consoleWarnSpy.mockRestore();
-    });
-    it('should filter out tool-related history before sending to classifier', async () => {
-        mockContext.history = [
-            { role: 'user', content: [{ type: 'text', text: 'call a tool' }] },
-            {
-                role: 'assistant',
-                content: [
-                    {
-                        type: 'tool_call',
-                        id: 'call-1',
-                        name: 'test_tool',
-                        arguments: {},
-                    },
-                ],
-            },
-            {
-                role: 'user',
-                content: [
-                    {
-                        type: 'tool_result',
-                        toolCallId: 'call-1',
-                        name: 'test_tool',
-                        content: { ok: true },
-                    },
-                ],
-            },
-            { role: 'user', content: [{ type: 'text', text: 'another user turn' }] },
-        ];
-        const mockApiResponse = {
-            reasoning: 'Simple.',
-            model_choice: 'flash',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        const generateJsonCall = vi.mocked(mockBaseLlmClient.generateJson).mock
-            .calls[0][0];
-        const messages = generateJsonCall.messages;
-        const expectedMessages = [
-            { role: 'user', content: [{ type: 'text', text: 'call a tool' }] },
-            {
-                role: 'user',
-                content: [{ type: 'text', text: 'another user turn' }],
-            },
-            { role: 'user', content: [{ type: 'text', text: 'simple task' }] },
-        ];
-        expect(messages).toEqual(expectedMessages);
-    });
-    it('should respect HISTORY_SEARCH_WINDOW and HISTORY_TURNS_FOR_CONTEXT', async () => {
-        const longHistory = [];
-        for (let i = 0; i < 30; i++) {
-            longHistory.push({
-                role: 'user',
-                content: [{ type: 'text', text: `Message ${i}` }],
-            });
-            // Add noise that should be filtered
-            if (i % 2 === 0) {
-                longHistory.push({
-                    role: 'assistant',
-                    content: [
-                        {
-                            type: 'tool_call',
-                            id: `call-${i}`,
-                            name: 'noise',
-                            arguments: {},
-                        },
-                    ],
-                });
-            }
-        }
-        mockContext.history = longHistory;
-        const mockApiResponse = {
-            reasoning: 'Simple.',
-            model_choice: 'flash',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        const generateJsonCall = vi.mocked(mockBaseLlmClient.generateJson).mock
-            .calls[0][0];
-        const messages = generateJsonCall.messages;
-        // Manually calculate what the history should be
-        const HISTORY_SEARCH_WINDOW = 20;
-        const HISTORY_TURNS_FOR_CONTEXT = 4;
-        const historySlice = longHistory.slice(-HISTORY_SEARCH_WINDOW);
-        const cleanHistory = historySlice.filter((message) => !isToolCallMessage(message) && !isToolResultMessage(message));
-        const finalHistory = cleanHistory.slice(-HISTORY_TURNS_FOR_CONTEXT);
-        expect(messages).toEqual([
-            ...finalHistory,
-            { role: 'user', content: [{ type: 'text', text: 'simple task' }] },
-        ]);
-        // There should be 4 history items + the current request
-        expect(messages).toHaveLength(5);
-    });
-    it('should use a fallback promptId if not found in context', async () => {
-        const consoleWarnSpy = vi
-            .spyOn(debugLogger, 'warn')
-            .mockImplementation(() => { });
-        vi.spyOn(promptIdContext, 'getStore').mockReturnValue(undefined);
-        const mockApiResponse = {
-            reasoning: 'Simple.',
-            model_choice: 'flash',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
-        const generateJsonCall = vi.mocked(mockBaseLlmClient.generateJson).mock
-            .calls[0][0];
-        expect(generateJsonCall.promptId).toMatch(/^classifier-router-fallback-\d+-\w+$/);
-        expect(consoleWarnSpy).toHaveBeenCalledWith(expect.stringContaining('Could not find promptId in context for classifier-router. This is unexpected. Using a fallback ID:'));
-        consoleWarnSpy.mockRestore();
-    });
-    it('should respect requestedModel from context in resolveClassifierModel', async () => {
-        const requestedModel = DEFAULT_GEMINI_MODEL; // Pro model
-        const mockApiResponse = {
-            reasoning: 'Choice is flash',
-            model_choice: 'flash',
-        };
-        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
-        const contextWithRequestedModel = {
-            ...mockContext,
-            requestedModel,
-        };
-        const decision = await strategy.route(contextWithRequestedModel, mockConfig, mockBaseLlmClient);
-        expect(decision).not.toBeNull();
-        // Since requestedModel is Pro, and choice is flash, it should resolve to Flash
-        expect(decision?.model).toBe(DEFAULT_GEMINI_FLASH_MODEL);
-    });
-});
-//# sourceMappingURL=classifierStrategy.test.js.map

package/dist/src/routing/strategies/classifierStrategy.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"classifierStrategy.test.js","sourceRoot":"","sources":["../../../../src/routing/strategies/classifierStrategy.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;AAO7D,OAAO,EACL,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAC1B,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AAGjE,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,EAAE,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;AAEvC,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,QAA4B,CAAC;IACjC,IAAI,WAA2B,CAAC;IAChC,IAAI,UAAkB,CAAC;IACvB,IAAI,iBAAgC,CAAC;IACrC,IAAI,kBAAuC,CAAC;IAE5C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,QAAQ,GAAG,IAAI,kBAAkB,EAAE,CAAC;QACpC,WAAW,GAAG;YACZ,OAAO,EAAE,EAAE;YACX,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAChD,MAAM,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACrC,CAAC;QAEF,kBAAkB,GAAG;YACnB,KAAK,EAAE,YAAY;YACnB,qBAAqB,EAAE,EAAE;SACQ,CAAC;QACpC,UAAU,GAAG;YACX,kBAAkB,EAAE;gBAClB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC;aAC/D;YACD,QAAQ,EAAE,GAAG,EAAE,CAAC,yBAAyB;YACzC,kBAAkB,EAAE,GAAG,EAAE,CAAC,KAAK;YAC/B,0BAA0B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC;SACxC,CAAC;QACvB,iBAAiB,GAAG;YAClB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;SACM,CAAC;QAE9B,EAAE,CAAC,KAAK,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEzE,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,aAAa;YACxB,YAAY,EAAE,OAAO;SACtB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QAEF,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACzD,MAAM,CAAC,gBAAgB,CAAC;YACtB,cAAc,EAAE,EAAE,KAAK,EAAE,kBAAkB,CAAC,KAAK,EAAE;YACnD,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,wBAAwB;YACnC,YAAY,EAAE,OAAO;SACtB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,KAAK,EAAE,0BAA0B;YACjC,QAAQ,EAAE;gBACR,MAAM,EAAE,YAAY;gBACpB,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC7B,SAAS,EAAE,eAAe,CAAC,SAAS;aACrC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,yBAAyB;YACpC,YAAY,EAAE,KAAK;SACpB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QACF,WAAW,CAAC,OAAO,GAAG;YACpB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,6BAA6B,EAAE;SACtD,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,KAAK,EAAE,oBAAoB;YAC3B,QAAQ,EAAE;gBACR,MAAM,EAAE,YAAY;gBACpB,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC7B,SAAS,EAAE,eAAe,CAAC,SAAS;aACrC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,cAAc,GAAG,EAAE;aACtB,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC;aAC1B,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;QAC3C,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAEvE,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,MAAM,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,cAAc,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,cAAc,GAAG,EAAE;aACtB,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC;aAC1B,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChC,MAAM,oBAAoB,GAAG;YAC3B,SAAS,EAAE,wBAAwB;YACnC,iEAAiE;SAClE,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,oBAAoB,CACrB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,MAAM,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,cAAc,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,WAAW,CAAC,OAAO,GAAG;YACpB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;YAClE;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,WAAW;wBACjB,EAAE,EAAE,QAAQ;wBACZ,IAAI,EAAE,WAAW;wBACjB,SAAS,EAAE,EAAE;qBACd;iBACF;aACF;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,aAAa;wBACnB,UAAU,EAAE,QAAQ;wBACpB,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE;qBACtB;iBACF;aACF;YACD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,EAAE;SACzE,CAAC;QACF,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,SAAS;YACpB,YAAY,EAAE,OAAO;SACtB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QAEF,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,IAAI;aACpE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAA2B,CAAC;QACzC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,QAAQ,CAAC;QAE3C,MAAM,gBAAgB,GAAG;YACvB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;YAClE;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;aACvD;YACD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;SACnE,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,WAAW,GAAiB,EAAE,CAAC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC;aAClD,CAAC,CAAC;YACH,oCAAoC;YACpC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChB,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,WAAW;4BACjB,EAAE,EAAE,QAAQ,CAAC,EAAE;4BACf,IAAI,EAAE,OAAO;4BACb,SAAS,EAAE,EAAE;yBACd;qBACF;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC;QAClC,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,SAAS;YACpB,YAAY,EAAE,OAAO;SACtB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QAEF,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,IAAI;aACpE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAA2B,CAAC;QACzC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,QAAQ,CAAC;QAE3C,gDAAgD;QAChD,MAAM,qBAAqB,GAAG,EAAE,CAAC;QACjC,MAAM,yBAAyB,GAAG,CAAC,CAAC;QACpC,MAAM,YAAY,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC;QAC/D,MAAM,YAAY,GAAG,YAAY,CAAC,MAAM,CACtC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAC1E,CAAC;QACF,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,yBAAyB,CAAC,CAAC;QAEpE,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,GAAG,YAAY;YACf,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE;SACnE,CAAC,CAAC;QACH,wDAAwD;QACxD,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,cAAc,GAAG,EAAE;aACtB,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC;aAC1B,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChC,EAAE,CAAC,KAAK,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACjE,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,SAAS;YACpB,YAAY,EAAE,OAAO;SACtB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QAEF,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,IAAI;aACpE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEf,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,OAAO,CACvC,sCAAsC,CACvC,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CACzC,MAAM,CAAC,gBAAgB,CACrB,oGAAoG,CACrG,CACF,CAAC;QACF,cAAc,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,cAAc,GAAG,oBAAoB,CAAC,CAAC,YAAY;QACzD,MAAM,eAAe,GAAG;YACtB,SAAS,EAAE,iBAAiB;YAC5B,YAAY,EAAE,OAAO;SACtB,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;QAEF,MAAM,yBAAyB,GAAG;YAChC,GAAG,WAAW;YACd,cAAc;SACG,CAAC;QAEpB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,yBAAyB,EACzB,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAChC,+EAA+E;QAC/E,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/routing/strategies/compositeStrategy.test.d.ts

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

package/dist/src/routing/strategies/compositeStrategy.test.js

@@ -1,124 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { CompositeStrategy } from './compositeStrategy.js';
-import { debugLogger } from '../../utils/debugLogger.js';
-import { coreEvents } from '../../utils/events.js';
-vi.mock('../../utils/debugLogger.js', () => ({
-    debugLogger: {
-        warn: vi.fn(),
-    },
-}));
-describe('CompositeStrategy', () => {
-    let mockContext;
-    let mockConfig;
-    let mockBaseLlmClient;
-    let mockStrategy1;
-    let mockStrategy2;
-    let mockTerminalStrategy;
-    let emitFeedbackSpy;
-    beforeEach(() => {
-        vi.clearAllMocks();
-        mockContext = {};
-        mockConfig = {};
-        mockBaseLlmClient = {};
-        emitFeedbackSpy = vi.spyOn(coreEvents, 'emitFeedback');
-        mockStrategy1 = {
-            name: 'strategy1',
-            route: vi.fn().mockResolvedValue(null),
-        };
-        mockStrategy2 = {
-            name: 'strategy2',
-            route: vi.fn().mockResolvedValue(null),
-        };
-        mockTerminalStrategy = {
-            name: 'terminal',
-            route: vi.fn().mockResolvedValue({
-                model: 'terminal-model',
-                metadata: {
-                    source: 'terminal',
-                    latencyMs: 10,
-                    reasoning: 'Terminal decision',
-                },
-            }),
-        };
-    });
-    it('should try strategies in order and return the first successful decision', async () => {
-        const decision = {
-            model: 'strategy2-model',
-            metadata: {
-                source: 'strategy2',
-                latencyMs: 20,
-                reasoning: 'Strategy 2 decided',
-            },
-        };
-        vi.spyOn(mockStrategy2, 'route').mockResolvedValue(decision);
-        const composite = new CompositeStrategy([mockStrategy1, mockStrategy2, mockTerminalStrategy], 'test-router');
-        const result = await composite.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockStrategy1.route).toHaveBeenCalledWith(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockStrategy2.route).toHaveBeenCalledWith(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockTerminalStrategy.route).not.toHaveBeenCalled();
-        expect(result.model).toBe('strategy2-model');
-        expect(result.metadata.source).toBe('test-router/strategy2');
-    });
-    it('should fall back to the terminal strategy if no other strategy provides a decision', async () => {
-        const composite = new CompositeStrategy([mockStrategy1, mockStrategy2, mockTerminalStrategy], 'test-router');
-        const result = await composite.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(mockStrategy1.route).toHaveBeenCalledTimes(1);
-        expect(mockStrategy2.route).toHaveBeenCalledTimes(1);
-        expect(mockTerminalStrategy.route).toHaveBeenCalledTimes(1);
-        expect(result.model).toBe('terminal-model');
-        expect(result.metadata.source).toBe('test-router/terminal');
-    });
-    it('should handle errors in non-terminal strategies and continue', async () => {
-        vi.spyOn(mockStrategy1, 'route').mockRejectedValue(new Error('Strategy 1 failed'));
-        const composite = new CompositeStrategy([mockStrategy1, mockTerminalStrategy], 'test-router');
-        const result = await composite.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(debugLogger.warn).toHaveBeenCalledWith("[Routing] Strategy 'strategy1' failed. Continuing to next strategy. Error:", expect.any(Error));
-        expect(result.model).toBe('terminal-model');
-    });
-    it('should re-throw an error from the terminal strategy', async () => {
-        const terminalError = new Error('Terminal strategy failed');
-        vi.spyOn(mockTerminalStrategy, 'route').mockRejectedValue(terminalError);
-        const composite = new CompositeStrategy([mockTerminalStrategy]);
-        await expect(composite.route(mockContext, mockConfig, mockBaseLlmClient)).rejects.toThrow(terminalError);
-        expect(emitFeedbackSpy).toHaveBeenCalledWith('error', "[Routing] Critical Error: Terminal strategy 'terminal' failed. Routing cannot proceed. Error:", terminalError);
-    });
-    it('should correctly finalize the decision metadata', async () => {
-        const decision = {
-            model: 'some-model',
-            metadata: {
-                source: 'child-source',
-                latencyMs: 50,
-                reasoning: 'Child reasoning',
-            },
-        };
-        vi.spyOn(mockStrategy1, 'route').mockResolvedValue(decision);
-        const composite = new CompositeStrategy([mockStrategy1, mockTerminalStrategy], 'my-composite');
-        const result = await composite.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(result.model).toBe('some-model');
-        expect(result.metadata.source).toBe('my-composite/child-source');
-        expect(result.metadata.reasoning).toBe('Child reasoning');
-        // It should keep the child's latency
-        expect(result.metadata.latencyMs).toBe(50);
-    });
-    it('should calculate total latency if child latency is not provided', async () => {
-        const decision = {
-            model: 'some-model',
-            metadata: {
-                source: 'child-source',
-                // No latencyMs here
-                latencyMs: 0,
-                reasoning: 'Child reasoning',
-            },
-        };
-        vi.spyOn(mockStrategy1, 'route').mockResolvedValue(decision);
-        const composite = new CompositeStrategy([mockStrategy1, mockTerminalStrategy], 'my-composite');
-        const result = await composite.route(mockContext, mockConfig, mockBaseLlmClient);
-        expect(result.metadata.latencyMs).toBeGreaterThanOrEqual(0);
-    });
-});
-//# sourceMappingURL=compositeStrategy.test.js.map

package/dist/src/routing/strategies/compositeStrategy.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"compositeStrategy.test.js","sourceRoot":"","sources":["../../../../src/routing/strategies/compositeStrategy.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,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAS3D,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnD,EAAE,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3C,WAAW,EAAE;QACX,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;KACd;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,WAA2B,CAAC;IAChC,IAAI,UAAkB,CAAC;IACvB,IAAI,iBAAgC,CAAC;IACrC,IAAI,aAA8B,CAAC;IACnC,IAAI,aAA8B,CAAC;IACnC,IAAI,oBAAsC,CAAC;IAC3C,IAAI,eAA4C,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,WAAW,GAAG,EAAoB,CAAC;QACnC,UAAU,GAAG,EAAY,CAAC;QAC1B,iBAAiB,GAAG,EAAmB,CAAC;QAExC,eAAe,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;QAEvD,aAAa,GAAG;YACd,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SACvC,CAAC;QAEF,aAAa,GAAG;YACd,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SACvC,CAAC;QAEF,oBAAoB,GAAG;YACrB,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC/B,KAAK,EAAE,gBAAgB;gBACvB,QAAQ,EAAE;oBACR,MAAM,EAAE,UAAU;oBAClB,SAAS,EAAE,EAAE;oBACb,SAAS,EAAE,mBAAmB;iBAC/B;aACF,CAAC;SACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,QAAQ,GAAoB;YAChC,KAAK,EAAE,iBAAiB;YACxB,QAAQ,EAAE;gBACR,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,EAAE;gBACb,SAAS,EAAE,oBAAoB;aAChC;SACF,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAE7D,MAAM,SAAS,GAAG,IAAI,iBAAiB,CACrC,CAAC,aAAa,EAAE,aAAa,EAAE,oBAAoB,CAAC,EACpD,aAAa,CACd,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAClC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC9C,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC9C,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QACF,MAAM,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QAClG,MAAM,SAAS,GAAG,IAAI,iBAAiB,CACrC,CAAC,aAAa,EAAE,aAAa,EAAE,oBAAoB,CAAC,EACpD,aAAa,CACd,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAClC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE5D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAChD,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAC/B,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CACrC,CAAC,aAAa,EAAE,oBAAoB,CAAC,EACrC,aAAa,CACd,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAClC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC3C,4EAA4E,EAC5E,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC5D,EAAE,CAAC,KAAK,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAEzE,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,CACV,SAAS,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAC5D,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAEjC,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,OAAO,EACP,+FAA+F,EAC/F,aAAa,CACd,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,QAAQ,GAAoB;YAChC,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE;gBACR,MAAM,EAAE,cAAc;gBACtB,SAAS,EAAE,EAAE;gBACb,SAAS,EAAE,iBAAiB;aAC7B;SACF,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAE7D,MAAM,SAAS,GAAG,IAAI,iBAAiB,CACrC,CAAC,aAAa,EAAE,oBAAoB,CAAC,EACrC,cAAc,CACf,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAClC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC1D,qCAAqC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,QAAQ,GAAoB;YAChC,KAAK,EAAE,YAAY;YACnB,QAAQ,EAAE;gBACR,MAAM,EAAE,cAAc;gBACtB,oBAAoB;gBACpB,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,iBAAiB;aAC7B;SACF,CAAC;QACF,EAAE,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAE7D,MAAM,SAAS,GAAG,IAAI,iBAAiB,CACrC,CAAC,aAAa,EAAE,oBAAoB,CAAC,EACrC,cAAc,CACf,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAClC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/routing/strategies/defaultStrategy.test.d.ts

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

package/dist/src/routing/strategies/defaultStrategy.test.js

@@ -1,102 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { describe, it, expect, vi } from 'vitest';
-import { DefaultStrategy } from './defaultStrategy.js';
-import { DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_MODEL, PREVIEW_GEMINI_MODEL_AUTO, DEFAULT_GEMINI_MODEL_AUTO, GEMINI_MODEL_ALIAS_AUTO, PREVIEW_GEMINI_FLASH_MODEL, } from '../../config/models.js';
-describe('DefaultStrategy', () => {
-    it('should route to the default model when requested model is default auto', async () => {
-        const strategy = new DefaultStrategy();
-        const mockContext = {};
-        const mockConfig = {
-            getModel: vi.fn().mockReturnValue(DEFAULT_GEMINI_MODEL_AUTO),
-            getPreviewFeatures: vi.fn().mockReturnValue(false),
-        };
-        const mockClient = {};
-        const decision = await strategy.route(mockContext, mockConfig, mockClient);
-        expect(decision).toEqual({
-            model: DEFAULT_GEMINI_MODEL,
-            metadata: {
-                source: 'default',
-                latencyMs: 0,
-                reasoning: `Routing to default model: ${DEFAULT_GEMINI_MODEL}`,
-            },
-        });
-    });
-    it('should route to the preview model when requested model is preview auto', async () => {
-        const strategy = new DefaultStrategy();
-        const mockContext = {};
-        const mockConfig = {
-            getModel: vi.fn().mockReturnValue(PREVIEW_GEMINI_MODEL_AUTO),
-            getPreviewFeatures: vi.fn().mockReturnValue(false),
-        };
-        const mockClient = {};
-        const decision = await strategy.route(mockContext, mockConfig, mockClient);
-        expect(decision).toEqual({
-            model: PREVIEW_GEMINI_MODEL,
-            metadata: {
-                source: 'default',
-                latencyMs: 0,
-                reasoning: `Routing to default model: ${PREVIEW_GEMINI_MODEL}`,
-            },
-        });
-    });
-    it('should route to the preview model when requested model is auto and previewfeature is on', async () => {
-        const strategy = new DefaultStrategy();
-        const mockContext = {};
-        const mockConfig = {
-            getModel: vi.fn().mockReturnValue(GEMINI_MODEL_ALIAS_AUTO),
-            getPreviewFeatures: vi.fn().mockReturnValue(true),
-        };
-        const mockClient = {};
-        const decision = await strategy.route(mockContext, mockConfig, mockClient);
-        expect(decision).toEqual({
-            model: PREVIEW_GEMINI_MODEL,
-            metadata: {
-                source: 'default',
-                latencyMs: 0,
-                reasoning: `Routing to default model: ${PREVIEW_GEMINI_MODEL}`,
-            },
-        });
-    });
-    it('should route to the default model when requested model is auto and previewfeature is off', async () => {
-        const strategy = new DefaultStrategy();
-        const mockContext = {};
-        const mockConfig = {
-            getModel: vi.fn().mockReturnValue(GEMINI_MODEL_ALIAS_AUTO),
-            getPreviewFeatures: vi.fn().mockReturnValue(false),
-        };
-        const mockClient = {};
-        const decision = await strategy.route(mockContext, mockConfig, mockClient);
-        expect(decision).toEqual({
-            model: DEFAULT_GEMINI_MODEL,
-            metadata: {
-                source: 'default',
-                latencyMs: 0,
-                reasoning: `Routing to default model: ${DEFAULT_GEMINI_MODEL}`,
-            },
-        });
-    });
-    // this should not happen, adding the test just in case it happens.
-    it('should route to the same model if it is not an auto mode', async () => {
-        const strategy = new DefaultStrategy();
-        const mockContext = {};
-        const mockConfig = {
-            getModel: vi.fn().mockReturnValue(PREVIEW_GEMINI_FLASH_MODEL),
-            getPreviewFeatures: vi.fn().mockReturnValue(false),
-        };
-        const mockClient = {};
-        const decision = await strategy.route(mockContext, mockConfig, mockClient);
-        expect(decision).toEqual({
-            model: PREVIEW_GEMINI_FLASH_MODEL,
-            metadata: {
-                source: 'default',
-                latencyMs: 0,
-                reasoning: `Routing to default model: ${PREVIEW_GEMINI_FLASH_MODEL}`,
-            },
-        });
-    });
-});
-//# sourceMappingURL=defaultStrategy.test.js.map