@didim365/agent-cli-core 0.1.0

package/dist/src/routing/strategies/numericalClassifierStrategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"numericalClassifierStrategy.js","sourceRoot":"","sources":["../../../../src/routing/strategies/numericalClassifierStrategy.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,uBAAuB,EAAE,MAAM,gCAAgC,CAAC;AAMzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAEhE,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,2EAA2E;AAC3E,MAAM,yBAAyB,GAAG,CAAC,CAAC;AAEpC,MAAM,WAAW,GAAG,OAAO,CAAC;AAC5B,MAAM,SAAS,GAAG,KAAK,CAAC;AAExB,MAAM,eAAe,GAAG;IACtB,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,oBAAoB,EAAE;YACpB,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,kCAAkC;SAChD;QACD,gBAAgB,EAAE;YAChB,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,8BAA8B;SAC5C;KACF;IACD,QAAQ,EAAE,CAAC,sBAAsB,EAAE,kBAAkB,CAAC;CACvD,CAAC;AAEF,MAAM,wBAAwB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA+B/B,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;;;;;;;;;;;;;;;CAezC,CAAC;AAEF,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,oBAAoB,EAAE,CAAC,CAAC,MAAM,EAAE;IAChC,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;CAC7C,CAAC,CAAC;AAEH;;;;;;;;;GASG;AACH,SAAS,sBAAsB,CAAC,SAAiB;IAC/C,MAAM,mBAAmB,GAAG,UAAU,CAAC;IACvC,MAAM,YAAY,GAAG,UAAU,CAAC;IAEhC,IAAI,IAAI,GAAG,mBAAmB,CAAC;IAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,IAAI,IAAI,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChC,kEAAkE;QAClE,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IACvC,CAAC;IAED,0BAA0B;IAC1B,IAAI,GAAG,IAAI,KAAK,CAAC,CAAC;IAElB,oBAAoB;IACpB,MAAM,UAAU,GAAG,IAAI,GAAG,GAAG,CAAC;IAC9B,aAAa;IACb,oBAAoB;IACpB,sBAAsB;IACtB,OAAO,UAAU,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;AACnC,CAAC;AAED,MAAM,OAAO,2BAA2B;IAC7B,IAAI,GAAG,sBAAsB,CAAC;IAEvC,KAAK,CAAC,KAAK,CACT,OAAuB,EACvB,MAAc,EACd,aAA4B;QAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,IAAI,CAAC,CAAC,MAAM,MAAM,CAAC,0BAA0B,EAAE,CAAC,EAAE,CAAC;gBACjD,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,QAAQ,GAAG,uBAAuB,CAAC,mBAAmB,CAAC,CAAC;YAE9D,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,yBAAyB,CAAC,CAAC;YAEvE,0EAA0E;YAC1E,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,WAAW,GAAG;gBAClB,IAAI,EAAE,MAAiB;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;aAC1D,CAAC;YAEF,MAAM,YAAY,GAAG,MAAM,aAAa,CAAC,YAAY,CAAC;gBACpD,cAAc,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;gBACvC,QAAQ,EAAE,CAAC,GAAG,YAAY,EAAE,WAAW,CAAC;gBACxC,MAAM,EAAE,eAAe;gBACvB,iBAAiB,EAAE,wBAAwB;gBAC3C,WAAW,EAAE,OAAO,CAAC,MAAM;gBAC3B,QAAQ;aACT,CAAC,CAAC;YAEH,MAAM,cAAc,GAAG,wBAAwB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,cAAc,CAAC,gBAAgB,CAAC;YAE9C,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,GACzC,MAAM,IAAI,CAAC,kBAAkB,CAC3B,KAAK,EACL,MAAM,EACN,MAAM,CAAC,YAAY,EAAE,IAAI,iBAAiB,CAC3C,CAAC;YAEJ,MAAM,aAAa,GAAG,sBAAsB,CAC1C,MAAM,CAAC,QAAQ,EAAE,EACjB,UAAU,EACV,MAAM,CAAC,kBAAkB,EAAE,CAC5B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAEzC,OAAO;gBACL,KAAK,EAAE,aAAa;gBACpB,QAAQ,EAAE;oBACR,MAAM,EAAE,eAAe,UAAU,GAAG;oBACpC,SAAS;oBACT,SAAS,EAAE,WAAW,KAAK,iBAAiB,SAAS,KAAK,cAAc,CAAC,oBAAoB,EAAE;iBAChG;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,WAAW,CAAC,IAAI,CAAC,+CAA+C,EAAE,KAAK,CAAC,CAAC;YACzE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,KAAa,EACb,MAAc,EACd,SAAiB;QAMjB,IAAI,SAAiB,CAAC;QACtB,IAAI,UAAkB,CAAC;QAEvB,MAAM,oBAAoB,GAAG,MAAM,MAAM,CAAC,sBAAsB,EAAE,CAAC;QAEnE,IACE,oBAAoB,KAAK,SAAS;YAClC,CAAC,KAAK,CAAC,oBAAoB,CAAC;YAC5B,oBAAoB,IAAI,CAAC;YACzB,oBAAoB,IAAI,GAAG,EAC3B,CAAC;YACD,SAAS,GAAG,oBAAoB,CAAC;YACjC,UAAU,GAAG,QAAQ,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,qCAAqC;YACrC,SAAS,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC;YAC9C,UAAU,GAAG,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;QACvD,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC;QAEhE,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;IAC/C,CAAC;CACF"}

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

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

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

@@ -0,0 +1,367 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { NumericalClassifierStrategy } from './numericalClassifierStrategy.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('NumericalClassifierStrategy', () => {
+    let strategy;
+    let mockContext;
+    let mockConfig;
+    let mockBaseLlmClient;
+    let mockResolvedConfig;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        strategy = new NumericalClassifierStrategy();
+        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,
+            getSessionId: vi.fn().mockReturnValue('control-group-id'), // Default to Control Group (Hash 71 >= 50)
+            getNumericalRoutingEnabled: vi.fn().mockResolvedValue(true),
+            getClassifierThreshold: vi.fn().mockResolvedValue(undefined),
+        };
+        mockBaseLlmClient = {
+            generateJson: vi.fn(),
+        };
+        vi.spyOn(promptIdContext, 'getStore').mockReturnValue('test-prompt-id');
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    it('should return null if numerical routing is disabled', async () => {
+        vi.mocked(mockConfig.getNumericalRoutingEnabled).mockResolvedValue(false);
+        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+        expect(decision).toBeNull();
+        expect(mockBaseLlmClient.generateJson).not.toHaveBeenCalled();
+    });
+    it('should call generateJson with the correct parameters and wrapped user content', async () => {
+        const mockApiResponse = {
+            complexity_reasoning: 'Simple task',
+            complexity_score: 10,
+        };
+        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+        await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+        const generateJsonCall = vi.mocked(mockBaseLlmClient.generateJson).mock
+            .calls[0][0];
+        expect(generateJsonCall).toMatchObject({
+            modelConfigKey: { model: mockResolvedConfig.model },
+            promptId: 'test-prompt-id',
+        });
+        // Verify user content
+        const userMessage = generateJsonCall.messages[generateJsonCall.messages.length - 1];
+        expect(userMessage.content[0]).toEqual({
+            type: 'text',
+            text: 'simple task',
+        });
+    });
+    describe('A/B Testing Logic (Deterministic)', () => {
+        it('Control Group (SessionID "control-group-id" -> Threshold 50): Score 40 -> FLASH', async () => {
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('control-group-id'); // Hash 71 -> Control
+            const mockApiResponse = {
+                complexity_reasoning: 'Standard task',
+                complexity_score: 40,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_FLASH_MODEL,
+                metadata: {
+                    source: 'Classifier (Control)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 40 / Threshold: 50'),
+                },
+            });
+        });
+        it('Control Group (SessionID "control-group-id" -> Threshold 50): Score 60 -> PRO', async () => {
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('control-group-id');
+            const mockApiResponse = {
+                complexity_reasoning: 'Complex task',
+                complexity_score: 60,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: {
+                    source: 'Classifier (Control)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 60 / Threshold: 50'),
+                },
+            });
+        });
+        it('Strict Group (SessionID "test-session-1" -> Threshold 80): Score 60 -> FLASH', async () => {
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('test-session-1'); // FNV Normalized 18 < 50 -> Strict
+            const mockApiResponse = {
+                complexity_reasoning: 'Complex task',
+                complexity_score: 60,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_FLASH_MODEL, // Routed to Flash because 60 < 80
+                metadata: {
+                    source: 'Classifier (Strict)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 60 / Threshold: 80'),
+                },
+            });
+        });
+        it('Strict Group (SessionID "test-session-1" -> Threshold 80): Score 90 -> PRO', async () => {
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('test-session-1');
+            const mockApiResponse = {
+                complexity_reasoning: 'Extreme task',
+                complexity_score: 90,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: {
+                    source: 'Classifier (Strict)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 90 / Threshold: 80'),
+                },
+            });
+        });
+    });
+    describe('Remote Threshold Logic', () => {
+        it('should use the remote CLASSIFIER_THRESHOLD if provided (int value)', async () => {
+            vi.mocked(mockConfig.getClassifierThreshold).mockResolvedValue(70);
+            const mockApiResponse = {
+                complexity_reasoning: 'Test task',
+                complexity_score: 60,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_FLASH_MODEL, // Score 60 < Threshold 70
+                metadata: {
+                    source: 'Classifier (Remote)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 60 / Threshold: 70'),
+                },
+            });
+        });
+        it('should use the remote CLASSIFIER_THRESHOLD if provided (float value)', async () => {
+            vi.mocked(mockConfig.getClassifierThreshold).mockResolvedValue(45.5);
+            const mockApiResponse = {
+                complexity_reasoning: 'Test task',
+                complexity_score: 40,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_FLASH_MODEL, // Score 40 < Threshold 45.5
+                metadata: {
+                    source: 'Classifier (Remote)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 40 / Threshold: 45.5'),
+                },
+            });
+        });
+        it('should use PRO model if score >= remote CLASSIFIER_THRESHOLD', async () => {
+            vi.mocked(mockConfig.getClassifierThreshold).mockResolvedValue(30);
+            const mockApiResponse = {
+                complexity_reasoning: 'Test task',
+                complexity_score: 35,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_MODEL, // Score 35 >= Threshold 30
+                metadata: {
+                    source: 'Classifier (Remote)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 35 / Threshold: 30'),
+                },
+            });
+        });
+        it('should fall back to A/B testing if CLASSIFIER_THRESHOLD is not present in experiments', async () => {
+            // Mock getClassifierThreshold to return undefined
+            vi.mocked(mockConfig.getClassifierThreshold).mockResolvedValue(undefined);
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('control-group-id'); // Should resolve to Control (50)
+            const mockApiResponse = {
+                complexity_reasoning: 'Test task',
+                complexity_score: 40,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_FLASH_MODEL, // Score 40 < Default A/B Threshold 50
+                metadata: {
+                    source: 'Classifier (Control)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 40 / Threshold: 50'),
+                },
+            });
+        });
+        it('should fall back to A/B testing if CLASSIFIER_THRESHOLD is out of range (less than 0)', async () => {
+            vi.mocked(mockConfig.getClassifierThreshold).mockResolvedValue(-10);
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('control-group-id');
+            const mockApiResponse = {
+                complexity_reasoning: 'Test task',
+                complexity_score: 40,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_FLASH_MODEL,
+                metadata: {
+                    source: 'Classifier (Control)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 40 / Threshold: 50'),
+                },
+            });
+        });
+        it('should fall back to A/B testing if CLASSIFIER_THRESHOLD is out of range (greater than 100)', async () => {
+            vi.mocked(mockConfig.getClassifierThreshold).mockResolvedValue(110);
+            vi.mocked(mockConfig.getSessionId).mockReturnValue('control-group-id');
+            const mockApiResponse = {
+                complexity_reasoning: 'Test task',
+                complexity_score: 60,
+            };
+            vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(mockApiResponse);
+            const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+            expect(decision).toEqual({
+                model: DEFAULT_GEMINI_MODEL,
+                metadata: {
+                    source: 'Classifier (Control)',
+                    latencyMs: expect.any(Number),
+                    reasoning: expect.stringContaining('Score: 60 / Threshold: 50'),
+                },
+            });
+        });
+    });
+    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();
+    });
+    it('should return null if the classifier returns a malformed JSON object', async () => {
+        const consoleWarnSpy = vi
+            .spyOn(debugLogger, 'warn')
+            .mockImplementation(() => { });
+        const malformedApiResponse = {
+            complexity_reasoning: 'This is a simple task.',
+            // complexity_score is missing
+        };
+        vi.mocked(mockBaseLlmClient.generateJson).mockResolvedValue(malformedApiResponse);
+        const decision = await strategy.route(mockContext, mockConfig, mockBaseLlmClient);
+        expect(decision).toBeNull();
+        expect(consoleWarnSpy).toHaveBeenCalled();
+    });
+    it('should include tool-related history when 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 = {
+            complexity_reasoning: 'Simple.',
+            complexity_score: 10,
+        };
+        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 = [
+            ...mockContext.history,
+            // The last user turn is the request
+            {
+                role: 'user',
+                content: [{ type: 'text', text: 'simple task' }],
+            },
+        ];
+        expect(messages).toEqual(expectedMessages);
+    });
+    it('should respect HISTORY_TURNS_FOR_CONTEXT', async () => {
+        const longHistory = [];
+        for (let i = 0; i < 30; i++) {
+            longHistory.push({
+                role: 'user',
+                content: [{ type: 'text', text: `Message ${i}` }],
+            });
+        }
+        mockContext.history = longHistory;
+        const mockApiResponse = {
+            complexity_reasoning: 'Simple.',
+            complexity_score: 10,
+        };
+        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_TURNS_FOR_CONTEXT = 8;
+        const finalHistory = longHistory.slice(-HISTORY_TURNS_FOR_CONTEXT);
+        // Last message is the request
+        const requestMessage = {
+            role: 'user',
+            content: [{ type: 'text', text: 'simple task' }],
+        };
+        expect(messages).toEqual([...finalHistory, requestMessage]);
+        expect(messages).toHaveLength(9);
+    });
+    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 = {
+            complexity_reasoning: 'Simple.',
+            complexity_score: 10,
+        };
+        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:'));
+    });
+});
+//# sourceMappingURL=numericalClassifierStrategy.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"numericalClassifierStrategy.test.js","sourceRoot":"","sources":["../../../../src/routing/strategies/numericalClassifierStrategy.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAO/E,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,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,QAAqC,CAAC;IAC1C,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,2BAA2B,EAAE,CAAC;QAC7C,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,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,EAAE,2CAA2C;YACtG,0BAA0B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;YAC3D,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,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,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAE1E,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,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,eAAe,GAAG;YACtB,oBAAoB,EAAE,aAAa;YACnC,gBAAgB,EAAE,EAAE;SACrB,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;QAEzC,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,CAAC;YACrC,cAAc,EAAE,EAAE,KAAK,EAAE,kBAAkB,CAAC,KAAK,EAAE;YACnD,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;QAEH,sBAAsB;QACtB,MAAM,WAAW,GACf,gBAAgB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAClE,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,aAAa;SACpB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,qBAAqB;YAC7F,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,eAAe;gBACrC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,0BAA0B;gBACjC,QAAQ,EAAE;oBACR,MAAM,EAAE,sBAAsB;oBAC9B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YACvE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,cAAc;gBACpC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE;oBACR,MAAM,EAAE,sBAAsB;oBAC9B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,CAAC,mCAAmC;YACzG,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,cAAc;gBACpC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,0BAA0B,EAAE,kCAAkC;gBACrE,QAAQ,EAAE;oBACR,MAAM,EAAE,qBAAqB;oBAC7B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;YAC1F,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YACrE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,cAAc;gBACpC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE;oBACR,MAAM,EAAE,qBAAqB;oBAC7B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACnE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,WAAW;gBACjC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,0BAA0B,EAAE,0BAA0B;gBAC7D,QAAQ,EAAE;oBACR,MAAM,EAAE,qBAAqB;oBAC7B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACrE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,WAAW;gBACjC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,0BAA0B,EAAE,4BAA4B;gBAC/D,QAAQ,EAAE;oBACR,MAAM,EAAE,qBAAqB;oBAC7B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,6BAA6B,CAAC;iBAClE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACnE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,WAAW;gBACjC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,oBAAoB,EAAE,2BAA2B;gBACxD,QAAQ,EAAE;oBACR,MAAM,EAAE,qBAAqB;oBAC7B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;YACrG,kDAAkD;YAClD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAC1E,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,iCAAiC;YACzG,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,WAAW;gBACjC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,0BAA0B,EAAE,sCAAsC;gBACzE,QAAQ,EAAE;oBACR,MAAM,EAAE,sBAAsB;oBAC9B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;YACrG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC;YACpE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YACvE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,WAAW;gBACjC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,0BAA0B;gBACjC,QAAQ,EAAE;oBACR,MAAM,EAAE,sBAAsB;oBAC9B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;YAC1G,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YACpE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YACvE,MAAM,eAAe,GAAG;gBACtB,oBAAoB,EAAE,WAAW;gBACjC,gBAAgB,EAAE,EAAE;aACrB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,iBAAiB,CACzD,eAAe,CAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,WAAW,EACX,UAAU,EACV,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,oBAAoB;gBAC3B,QAAQ,EAAE;oBACR,MAAM,EAAE,sBAAsB;oBAC9B,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;oBAC7B,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;iBAChE;aACF,CAAC,CAAC;QACL,CAAC,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;IAC5C,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,oBAAoB,EAAE,wBAAwB;YAC9C,8BAA8B;SAC/B,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;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,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,oBAAoB,EAAE,SAAS;YAC/B,gBAAgB,EAAE,EAAE;SACrB,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,GAAG,WAAW,CAAC,OAAO;YACtB,oCAAoC;YACpC;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;aACjD;SACF,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,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;QACL,CAAC;QACD,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC;QAClC,MAAM,eAAe,GAAG;YACtB,oBAAoB,EAAE,SAAS;YAC/B,gBAAgB,EAAE,EAAE;SACrB,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,yBAAyB,GAAG,CAAC,CAAC;QACpC,MAAM,YAAY,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,yBAAyB,CAAC,CAAC;QAEnE,8BAA8B;QAC9B,MAAM,cAAc,GAAG;YACrB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;SACjD,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC;QAC5D,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,oBAAoB,EAAE,SAAS;YAC/B,gBAAgB,EAAE,EAAE;SACrB,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;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,15 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../../config/config.js';
+import type { BaseLlmClient } from '../../core/baseLlmClient.js';
+import type { RoutingContext, RoutingDecision, RoutingStrategy } from '../routingStrategy.js';
+/**
+ * Handles cases where the user explicitly specifies a model (override).
+ */
+export declare class OverrideStrategy implements RoutingStrategy {
+    readonly name = "override";
+    route(context: RoutingContext, config: Config, _baseLlmClient: BaseLlmClient): Promise<RoutingDecision | null>;
+}

package/dist/src/routing/strategies/overrideStrategy.js

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isAutoModel, resolveModel } from '../../config/models.js';
+/**
+ * Handles cases where the user explicitly specifies a model (override).
+ */
+export class OverrideStrategy {
+    name = 'override';
+    async route(context, config, _baseLlmClient) {
+        const overrideModel = context.requestedModel ?? config.getModel();
+        // If the model is 'auto' we should pass to the next strategy.
+        if (isAutoModel(overrideModel)) {
+            return null;
+        }
+        // Return the overridden model name.
+        return {
+            model: resolveModel(overrideModel, config.getPreviewFeatures()),
+            metadata: {
+                source: this.name,
+                latencyMs: 0,
+                reasoning: `Routing bypassed by forced model directive. Using: ${overrideModel}`,
+            },
+        };
+    }
+}
+//# sourceMappingURL=overrideStrategy.js.map

package/dist/src/routing/strategies/overrideStrategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"overrideStrategy.js","sourceRoot":"","sources":["../../../../src/routing/strategies/overrideStrategy.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAQnE;;GAEG;AACH,MAAM,OAAO,gBAAgB;IAClB,IAAI,GAAG,UAAU,CAAC;IAE3B,KAAK,CAAC,KAAK,CACT,OAAuB,EACvB,MAAc,EACd,cAA6B;QAE7B,MAAM,aAAa,GAAG,OAAO,CAAC,cAAc,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAElE,8DAA8D;QAC9D,IAAI,WAAW,CAAC,aAAa,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,oCAAoC;QACpC,OAAO;YACL,KAAK,EAAE,YAAY,CAAC,aAAa,EAAE,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC/D,QAAQ,EAAE;gBACR,MAAM,EAAE,IAAI,CAAC,IAAI;gBACjB,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,sDAAsD,aAAa,EAAE;aACjF;SACF,CAAC;IACJ,CAAC;CACF"}

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

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

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

@@ -0,0 +1,59 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { OverrideStrategy } from './overrideStrategy.js';
+import { DEFAULT_GEMINI_MODEL_AUTO } from '../../config/models.js';
+describe('OverrideStrategy', () => {
+    const strategy = new OverrideStrategy();
+    const mockContext = {};
+    const mockClient = {};
+    it('should return null when the override model is auto', async () => {
+        const mockConfig = {
+            getModel: () => DEFAULT_GEMINI_MODEL_AUTO,
+            getPreviewFeatures: () => false,
+        };
+        const decision = await strategy.route(mockContext, mockConfig, mockClient);
+        expect(decision).toBeNull();
+    });
+    it('should return a decision with the override model when one is specified', async () => {
+        const overrideModel = 'gemini-2.5-pro-custom';
+        const mockConfig = {
+            getModel: () => overrideModel,
+            getPreviewFeatures: () => false,
+        };
+        const decision = await strategy.route(mockContext, mockConfig, mockClient);
+        expect(decision).not.toBeNull();
+        expect(decision?.model).toBe(overrideModel);
+        expect(decision?.metadata.source).toBe('override');
+        expect(decision?.metadata.reasoning).toContain('Routing bypassed by forced model directive');
+        expect(decision?.metadata.reasoning).toContain(overrideModel);
+    });
+    it('should handle different override model names', async () => {
+        const overrideModel = 'gemini-2.5-flash-experimental';
+        const mockConfig = {
+            getModel: () => overrideModel,
+            getPreviewFeatures: () => false,
+        };
+        const decision = await strategy.route(mockContext, mockConfig, mockClient);
+        expect(decision).not.toBeNull();
+        expect(decision?.model).toBe(overrideModel);
+    });
+    it('should respect requestedModel from context', async () => {
+        const requestedModel = 'requested-model';
+        const configModel = 'config-model';
+        const mockConfig = {
+            getModel: () => configModel,
+            getPreviewFeatures: () => false,
+        };
+        const contextWithRequestedModel = {
+            requestedModel,
+        };
+        const decision = await strategy.route(contextWithRequestedModel, mockConfig, mockClient);
+        expect(decision).not.toBeNull();
+        expect(decision?.model).toBe(requestedModel);
+    });
+});
+//# sourceMappingURL=overrideStrategy.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"overrideStrategy.test.js","sourceRoot":"","sources":["../../../../src/routing/strategies/overrideStrategy.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAIzD,OAAO,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AAEnE,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;IACxC,MAAM,WAAW,GAAG,EAAoB,CAAC;IACzC,MAAM,UAAU,GAAG,EAAmB,CAAC;IAEvC,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE,GAAG,EAAE,CAAC,yBAAyB;YACzC,kBAAkB,EAAE,GAAG,EAAE,CAAC,KAAK;SACtB,CAAC;QAEZ,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAC3E,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,aAAa,GAAG,uBAAuB,CAAC;QAC9C,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE,GAAG,EAAE,CAAC,aAAa;YAC7B,kBAAkB,EAAE,GAAG,EAAE,CAAC,KAAK;SACtB,CAAC;QAEZ,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAE3E,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,SAAS,CAC5C,4CAA4C,CAC7C,CAAC;QACF,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,aAAa,GAAG,+BAA+B,CAAC;QACtD,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE,GAAG,EAAE,CAAC,aAAa;YAC7B,kBAAkB,EAAE,GAAG,EAAE,CAAC,KAAK;SACtB,CAAC;QAEZ,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAE3E,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,cAAc,GAAG,iBAAiB,CAAC;QACzC,MAAM,WAAW,GAAG,cAAc,CAAC;QACnC,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE,GAAG,EAAE,CAAC,WAAW;YAC3B,kBAAkB,EAAE,GAAG,EAAE,CAAC,KAAK;SACtB,CAAC;QACZ,MAAM,yBAAyB,GAAG;YAChC,cAAc;SACG,CAAC;QAEpB,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CACnC,yBAAyB,EACzB,UAAU,EACV,UAAU,CACX,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/safety/built-in.d.ts

@@ -0,0 +1,21 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { SafetyCheckInput, SafetyCheckResult } from './protocol.js';
+/**
+ * Interface for all in-process safety checkers.
+ */
+export interface InProcessChecker {
+    check(input: SafetyCheckInput): Promise<SafetyCheckResult>;
+}
+/**
+ * An in-process checker to validate file paths.
+ */
+export declare class AllowedPathChecker implements InProcessChecker {
+    check(input: SafetyCheckInput): Promise<SafetyCheckResult>;
+    private safelyResolvePath;
+    private isPathAllowed;
+    private collectPathsToCheck;
+}