@didim365/agent-cli-core 0.1.0

package/dist/src/providers/types.test.js

@@ -0,0 +1,253 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+describe('LlmMessage types', () => {
+    describe('LlmRole', () => {
+        it('should accept valid roles', () => {
+            const roles = ['user', 'assistant', 'system', 'tool'];
+            expect(roles).toHaveLength(4);
+        });
+    });
+    describe('LlmTextContent', () => {
+        it('should have correct structure', () => {
+            const textContent = {
+                type: 'text',
+                text: 'Hello, world!',
+            };
+            expect(textContent.type).toBe('text');
+            expect(textContent.text).toBe('Hello, world!');
+        });
+    });
+    describe('LlmImageContent', () => {
+        it('should support base64 images', () => {
+            const imageContent = {
+                type: 'image',
+                source: {
+                    type: 'base64',
+                    mediaType: 'image/png',
+                    data: 'base64encodeddata...',
+                },
+            };
+            expect(imageContent.source.type).toBe('base64');
+            if (imageContent.source.type === 'base64') {
+                expect(imageContent.source.data).toBe('base64encodeddata...');
+            }
+        });
+        it('should support URL images with url field', () => {
+            const imageContent = {
+                type: 'image',
+                source: {
+                    type: 'url',
+                    mediaType: 'image/png',
+                    url: 'https://example.com/image.png',
+                },
+            };
+            expect(imageContent.source.type).toBe('url');
+            if (imageContent.source.type === 'url') {
+                expect(imageContent.source.url).toBe('https://example.com/image.png');
+            }
+        });
+    });
+    describe('LlmToolCallContent', () => {
+        it('should have correct structure', () => {
+            const toolCall = {
+                type: 'tool_call',
+                id: 'call-123',
+                name: 'read_file',
+                arguments: { path: '/tmp/test.txt' },
+            };
+            expect(toolCall.type).toBe('tool_call');
+            expect(toolCall.name).toBe('read_file');
+        });
+    });
+    describe('LlmToolResultContent', () => {
+        it('should have correct structure', () => {
+            const toolResult = {
+                type: 'tool_result',
+                toolCallId: 'call-123',
+                content: 'File contents here',
+            };
+            expect(toolResult.type).toBe('tool_result');
+            expect(toolResult.toolCallId).toBe('call-123');
+        });
+        it('should support optional error flag', () => {
+            const toolResult = {
+                type: 'tool_result',
+                toolCallId: 'call-123',
+                content: 'Error occurred',
+                isError: true,
+            };
+            expect(toolResult.isError).toBe(true);
+        });
+    });
+    describe('LlmThoughtContent', () => {
+        it('should have correct structure', () => {
+            const thought = {
+                type: 'thought',
+                thought: 'Let me think about this...',
+            };
+            expect(thought.type).toBe('thought');
+        });
+        it('should support optional metadata', () => {
+            const thought = {
+                type: 'thought',
+                thought: 'Planning step',
+                metadata: {
+                    step: 1,
+                    phase: 'planning',
+                    provider: 'gemini',
+                },
+            };
+            expect(thought.metadata?.phase).toBe('planning');
+        });
+    });
+    describe('LlmContent union type', () => {
+        it('should accept all content types', () => {
+            const contents = [
+                { type: 'text', text: 'Hello' },
+                {
+                    type: 'image',
+                    source: { type: 'base64', mediaType: 'image/png', data: 'xyz' },
+                },
+                { type: 'tool_call', id: '1', name: 'test', arguments: {} },
+                { type: 'tool_result', toolCallId: '1', content: 'result' },
+                {
+                    type: 'tool_result',
+                    toolCallId: '2',
+                    content: { structured: 'data' },
+                }, // Extended type
+                { type: 'thought', thought: 'thinking...' },
+            ];
+            expect(contents).toHaveLength(6);
+        });
+    });
+    describe('LlmMessage', () => {
+        it('should have correct structure', () => {
+            const message = {
+                role: 'user',
+                content: [{ type: 'text', text: 'Hello' }],
+            };
+            expect(message.role).toBe('user');
+            expect(message.content).toHaveLength(1);
+        });
+        it('should support optional fields', () => {
+            const toolMessage = {
+                role: 'tool',
+                content: [
+                    { type: 'tool_result', toolCallId: 'call-1', content: 'result' },
+                ],
+                name: 'read_file',
+                toolCallId: 'call-1',
+            };
+            expect(toolMessage.name).toBe('read_file');
+        });
+    });
+});
+describe('LlmGenerateRequest types', () => {
+    it('should have required fields', () => {
+        const request = {
+            model: 'gemini-2.0-flash',
+            messages: [{ role: 'user', content: [{ type: 'text', text: 'Hello' }] }],
+        };
+        expect(request.model).toBeDefined();
+        expect(request.messages).toBeDefined();
+    });
+    it('should support all optional fields', () => {
+        const request = {
+            model: 'gemini-2.0-flash',
+            messages: [],
+            systemInstruction: 'You are a helpful assistant',
+            tools: [],
+            toolChoice: 'auto',
+            temperature: 0.7,
+            maxTokens: 1024,
+            stopSequences: ['STOP'],
+            topP: 0.9,
+            topK: 40,
+            responseFormat: 'json',
+        };
+        expect(request.temperature).toBe(0.7);
+    });
+});
+describe('LlmGenerateResponse types', () => {
+    it('should have correct structure', () => {
+        const response = {
+            id: 'resp-123',
+            content: [{ type: 'text', text: 'Response' }],
+            model: 'gemini-2.0-flash',
+            stopReason: 'end_turn',
+            usage: {
+                promptTokens: 10,
+                completionTokens: 20,
+                totalTokens: 30,
+            },
+        };
+        expect(response.stopReason).toBe('end_turn');
+    });
+    it('should allow optional usage for streaming scenarios', () => {
+        const response = {
+            id: 'resp-123',
+            content: [{ type: 'text', text: 'Response' }],
+            model: 'gemini-2.0-flash',
+            stopReason: 'end_turn',
+            // usage is optional now
+        };
+        expect(response.usage).toBeUndefined();
+    });
+});
+describe('LlmStopReason', () => {
+    it('should accept all valid stop reasons', () => {
+        const reasons = [
+            'end_turn',
+            'max_tokens',
+            'stop_sequence',
+            'tool_use',
+            'content_filter',
+            'error',
+        ];
+        expect(reasons).toHaveLength(6);
+    });
+});
+describe('LlmTokenUsage', () => {
+    it('should have required fields', () => {
+        const usage = {
+            promptTokens: 100,
+            completionTokens: 200,
+            totalTokens: 300,
+        };
+        expect(usage.totalTokens).toBe(300);
+    });
+    it('should support optional cachedTokens', () => {
+        const usage = {
+            promptTokens: 100,
+            completionTokens: 200,
+            totalTokens: 300,
+            cachedTokens: 50,
+        };
+        expect(usage.cachedTokens).toBe(50);
+    });
+});
+describe('LlmToolDefinition', () => {
+    it('should have correct structure', () => {
+        const tool = {
+            name: 'read_file',
+            description: 'Reads a file from disk',
+            parameters: {
+                type: 'object',
+                properties: {
+                    path: {
+                        type: 'string',
+                        description: 'File path to read',
+                    },
+                },
+                required: ['path'],
+            },
+        };
+        expect(tool.name).toBe('read_file');
+        expect(tool.parameters.properties['path'].type).toBe('string');
+    });
+});
+//# sourceMappingURL=types.test.js.map

package/dist/src/providers/types.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.test.js","sourceRoot":"","sources":["../../../src/providers/types.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAiB9C,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,KAAK,GAAc,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,WAAW,GAAmB;gBAClC,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,eAAe;aACtB,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,YAAY,GAAoB;gBACpC,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,SAAS,EAAE,WAAW;oBACtB,IAAI,EAAE,sBAAsB;iBAC7B;aACF,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,YAAY,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1C,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAChE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,YAAY,GAAoB;gBACpC,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE;oBACN,IAAI,EAAE,KAAK;oBACX,SAAS,EAAE,WAAW;oBACtB,GAAG,EAAE,+BAA+B;iBACrC;aACF,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC7C,IAAI,YAAY,CAAC,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;gBACvC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YACxE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,QAAQ,GAAuB;gBACnC,IAAI,EAAE,WAAW;gBACjB,EAAE,EAAE,UAAU;gBACd,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE;aACrC,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,UAAU,GAAyB;gBACvC,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,UAAU;gBACtB,OAAO,EAAE,oBAAoB;aAC9B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC5C,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,UAAU,GAAyB;gBACvC,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,UAAU;gBACtB,OAAO,EAAE,gBAAgB;gBACzB,OAAO,EAAE,IAAI;aACd,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,OAAO,GAAsB;gBACjC,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,4BAA4B;aACtC,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,OAAO,GAAsB;gBACjC,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,eAAe;gBACxB,QAAQ,EAAE;oBACR,IAAI,EAAE,CAAC;oBACP,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,QAAQ;iBACnB;aACF,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAiB;gBAC7B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;gBAC/B;oBACE,IAAI,EAAE,OAAO;oBACb,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,EAAE;iBAChE;gBACD,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE;gBAC3D,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE;gBAC3D;oBACE,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,GAAG;oBACf,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE;iBAChC,EAAE,gBAAgB;gBACnB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,EAAE;aAC5C,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,OAAO,GAAe;gBAC1B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3C,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,WAAW,GAAe;gBAC9B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE;iBACjE;gBACD,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,QAAQ;aACrB,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,OAAO,GAAuB;YAClC,KAAK,EAAE,kBAAkB;YACzB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;SACzE,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,OAAO,GAAuB;YAClC,KAAK,EAAE,kBAAkB;YACzB,QAAQ,EAAE,EAAE;YACZ,iBAAiB,EAAE,6BAA6B;YAChD,KAAK,EAAE,EAAE;YACT,UAAU,EAAE,MAAM;YAClB,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;YACf,aAAa,EAAE,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,EAAE;YACR,cAAc,EAAE,MAAM;SACvB,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,QAAQ,GAAwB;YACpC,EAAE,EAAE,UAAU;YACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAC7C,KAAK,EAAE,kBAAkB;YACzB,UAAU,EAAE,UAAU;YACtB,KAAK,EAAE;gBACL,YAAY,EAAE,EAAE;gBAChB,gBAAgB,EAAE,EAAE;gBACpB,WAAW,EAAE,EAAE;aAChB;SACF,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,QAAQ,GAAwB;YACpC,EAAE,EAAE,UAAU;YACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAC7C,KAAK,EAAE,kBAAkB;YACzB,UAAU,EAAE,UAAU;YACtB,wBAAwB;SACzB,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,OAAO,GAAoB;YAC/B,UAAU;YACV,YAAY;YACZ,eAAe;YACf,UAAU;YACV,gBAAgB;YAChB,OAAO;SACR,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,KAAK,GAAkB;YAC3B,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,GAAG;YACrB,WAAW,EAAE,GAAG;SACjB,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,KAAK,GAAkB;YAC3B,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,GAAG;YACrB,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,IAAI,GAAsB;YAC9B,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,wBAAwB;YACrC,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE,QAAQ;wBACd,WAAW,EAAE,mBAAmB;qBACjC;iBACF;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB;SACF,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/resources/resource-registry.d.ts

@@ -0,0 +1,30 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Resource } from '@modelcontextprotocol/sdk/types.js';
+export interface MCPResource extends Resource {
+    serverName: string;
+    discoveredAt: number;
+}
+export type DiscoveredMCPResource = MCPResource;
+/**
+ * Tracks resources discovered from MCP servers so other
+ * components can query or include them in conversations.
+ */
+export declare class ResourceRegistry {
+    private resources;
+    /**
+     * Replace the resources for a specific server.
+     */
+    setResourcesForServer(serverName: string, resources: Resource[]): void;
+    getAllResources(): MCPResource[];
+    /**
+     * Find a resource by its identifier.
+     * Format: serverName:uri (e.g., "myserver:file:///data.txt")
+     */
+    findResourceByUri(identifier: string): MCPResource | undefined;
+    removeResourcesByServer(serverName: string): void;
+    clear(): void;
+}

package/dist/src/resources/resource-registry.js

@@ -0,0 +1,57 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const resourceKey = (serverName, uri) => `${serverName}::${uri}`;
+/**
+ * Tracks resources discovered from MCP servers so other
+ * components can query or include them in conversations.
+ */
+export class ResourceRegistry {
+    resources = new Map();
+    /**
+     * Replace the resources for a specific server.
+     */
+    setResourcesForServer(serverName, resources) {
+        this.removeResourcesByServer(serverName);
+        const discoveredAt = Date.now();
+        for (const resource of resources) {
+            if (!resource.uri) {
+                continue;
+            }
+            this.resources.set(resourceKey(serverName, resource.uri), {
+                serverName,
+                discoveredAt,
+                ...resource,
+            });
+        }
+    }
+    getAllResources() {
+        return Array.from(this.resources.values());
+    }
+    /**
+     * Find a resource by its identifier.
+     * Format: serverName:uri (e.g., "myserver:file:///data.txt")
+     */
+    findResourceByUri(identifier) {
+        const colonIndex = identifier.indexOf(':');
+        if (colonIndex <= 0) {
+            return undefined;
+        }
+        const serverName = identifier.substring(0, colonIndex);
+        const uri = identifier.substring(colonIndex + 1);
+        return this.resources.get(resourceKey(serverName, uri));
+    }
+    removeResourcesByServer(serverName) {
+        for (const key of Array.from(this.resources.keys())) {
+            if (key.startsWith(`${serverName}::`)) {
+                this.resources.delete(key);
+            }
+        }
+    }
+    clear() {
+        this.resources.clear();
+    }
+}
+//# sourceMappingURL=resource-registry.js.map

package/dist/src/resources/resource-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resource-registry.js","sourceRoot":"","sources":["../../../src/resources/resource-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,WAAW,GAAG,CAAC,UAAkB,EAAE,GAAW,EAAU,EAAE,CAC9D,GAAG,UAAU,KAAK,GAAG,EAAE,CAAC;AAQ1B;;;GAGG;AACH,MAAM,OAAO,gBAAgB;IACnB,SAAS,GAA6B,IAAI,GAAG,EAAE,CAAC;IAExD;;OAEG;IACH,qBAAqB,CAAC,UAAkB,EAAE,SAAqB;QAC7D,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAChC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;gBAClB,SAAS;YACX,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE;gBACxD,UAAU;gBACV,YAAY;gBACZ,GAAG,QAAQ;aACZ,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,eAAe;QACb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;;OAGG;IACH,iBAAiB,CAAC,UAAkB;QAClC,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,UAAU,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QACvD,MAAM,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,uBAAuB,CAAC,UAAkB;QACxC,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACpD,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,UAAU,IAAI,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/src/resources/resource-registry.test.d.ts

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

package/dist/src/resources/resource-registry.test.js

@@ -0,0 +1,54 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, expect, it, beforeEach } from 'vitest';
+import { ResourceRegistry } from './resource-registry.js';
+describe('ResourceRegistry', () => {
+    let registry;
+    beforeEach(() => {
+        registry = new ResourceRegistry();
+    });
+    const createResource = (overrides = {}) => ({
+        uri: 'file:///tmp/foo.txt',
+        name: 'foo',
+        description: 'example resource',
+        mimeType: 'text/plain',
+        ...overrides,
+    });
+    it('stores resources per server', () => {
+        registry.setResourcesForServer('a', [createResource()]);
+        registry.setResourcesForServer('b', [createResource({ uri: 'foo' })]);
+        expect(registry.getAllResources().filter((res) => res.serverName === 'a')).toHaveLength(1);
+        expect(registry.getAllResources().filter((res) => res.serverName === 'b')).toHaveLength(1);
+    });
+    it('clears resources for server before adding new ones', () => {
+        registry.setResourcesForServer('a', [
+            createResource(),
+            createResource({ uri: 'bar' }),
+        ]);
+        registry.setResourcesForServer('a', [createResource({ uri: 'baz' })]);
+        const resources = registry
+            .getAllResources()
+            .filter((res) => res.serverName === 'a');
+        expect(resources).toHaveLength(1);
+        expect(resources[0].uri).toBe('baz');
+    });
+    it('finds resources by serverName:uri identifier', () => {
+        registry.setResourcesForServer('a', [createResource()]);
+        registry.setResourcesForServer('b', [
+            createResource({ uri: 'file:///tmp/bar.txt' }),
+        ]);
+        expect(registry.findResourceByUri('b:file:///tmp/bar.txt')?.serverName).toBe('b');
+        expect(registry.findResourceByUri('a:file:///tmp/foo.txt')?.serverName).toBe('a');
+        expect(registry.findResourceByUri('a:file:///tmp/bar.txt')).toBeUndefined();
+        expect(registry.findResourceByUri('nonexistent')).toBeUndefined();
+    });
+    it('clears resources for a server', () => {
+        registry.setResourcesForServer('a', [createResource()]);
+        registry.removeResourcesByServer('a');
+        expect(registry.getAllResources().filter((res) => res.serverName === 'a')).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=resource-registry.test.js.map

package/dist/src/resources/resource-registry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resource-registry.test.js","sourceRoot":"","sources":["../../../src/resources/resource-registry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE1D,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,QAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,gBAAgB,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,CAAC,YAA+B,EAAE,EAAY,EAAE,CAAC,CAAC;QACvE,GAAG,EAAE,qBAAqB;QAC1B,IAAI,EAAE,KAAK;QACX,WAAW,EAAE,kBAAkB;QAC/B,QAAQ,EAAE,YAAY;QACtB,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACxD,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAEtE,MAAM,CACJ,QAAQ,CAAC,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,CACnE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClB,MAAM,CACJ,QAAQ,CAAC,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,CACnE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE;YAClC,cAAc,EAAE;YAChB,cAAc,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC;SAC/B,CAAC,CAAC;QACH,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAEtE,MAAM,SAAS,GAAG,QAAQ;aACvB,eAAe,EAAE;aACjB,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACxD,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE;YAClC,cAAc,CAAC,EAAE,GAAG,EAAE,qBAAqB,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,CACJ,QAAQ,CAAC,iBAAiB,CAAC,uBAAuB,CAAC,EAAE,UAAU,CAChE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,MAAM,CACJ,QAAQ,CAAC,iBAAiB,CAAC,uBAAuB,CAAC,EAAE,UAAU,CAChE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAC,uBAAuB,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5E,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,QAAQ,CAAC,qBAAqB,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACxD,QAAQ,CAAC,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAEtC,MAAM,CACJ,QAAQ,CAAC,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,CACnE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACpB,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,80 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DefaultStrategy } from './strategies/defaultStrategy.js';
+import { ClassifierStrategy } from './strategies/classifierStrategy.js';
+import { NumericalClassifierStrategy } from './strategies/numericalClassifierStrategy.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';
+import { debugLogger } from '../utils/debugLogger.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 NumericalClassifierStrategy(),
+            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;
+        const [enableNumericalRouting, thresholdValue] = await Promise.all([
+            this.config.getNumericalRoutingEnabled(),
+            this.config.getClassifierThreshold(),
+        ]);
+        const classifierThreshold = thresholdValue !== undefined ? String(thresholdValue) : undefined;
+        let failed = false;
+        let error_message;
+        try {
+            decision = await this.strategy.route(context, this.config, this.config.getBaseLlmClient());
+            debugLogger.debug(`[Routing] Selected model: ${decision.model} (Source: ${decision.metadata.source}, Latency: ${decision.metadata.latencyMs}ms)\n\t[Routing] Reasoning: ${decision.metadata.reasoning}`);
+        }
+        catch (e) {
+            failed = true;
+            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,
+                },
+            };
+            debugLogger.debug(`[Routing] Exception during routing: ${error_message}\n\tFallback model: ${decision.model} (Source: ${decision.metadata.source})`);
+        }
+        finally {
+            const event = new ModelRoutingEvent(decision.model, decision.metadata.source, decision.metadata.latencyMs, decision.metadata.reasoning, failed, error_message, enableNumericalRouting, classifierThreshold);
+            logModelRouting(this.config, event);
+        }
+        return decision;
+    }
+}
+//# 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;AAQH,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,2BAA2B,EAAE,MAAM,6CAA6C,CAAC;AAC1F,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;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD;;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,2BAA2B,EAAE;YACjC,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,MAAM,CAAC,sBAAsB,EAAE,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACjE,IAAI,CAAC,MAAM,CAAC,0BAA0B,EAAE;YACxC,IAAI,CAAC,MAAM,CAAC,sBAAsB,EAAE;SACrC,CAAC,CAAC;QACH,MAAM,mBAAmB,GACvB,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEpE,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,IAAI,aAAiC,CAAC;QAEtC,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,WAAW,CAAC,KAAK,CACf,6BAA6B,QAAQ,CAAC,KAAK,aAAa,QAAQ,CAAC,QAAQ,CAAC,MAAM,cAAc,QAAQ,CAAC,QAAQ,CAAC,SAAS,+BAA+B,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,CACtL,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,GAAG,IAAI,CAAC;YACd,aAAa,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC3D,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,WAAW,CAAC,KAAK,CACf,uCAAuC,aAAa,uBAAuB,QAAQ,CAAC,KAAK,aAAa,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAClI,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,MAAM,KAAK,GAAG,IAAI,iBAAiB,CACjC,QAAS,CAAC,KAAK,EACf,QAAS,CAAC,QAAQ,CAAC,MAAM,EACzB,QAAS,CAAC,QAAQ,CAAC,SAAS,EAC5B,QAAS,CAAC,QAAQ,CAAC,SAAS,EAC5B,MAAM,EACN,aAAa,EACb,sBAAsB,EACtB,mBAAmB,CACpB,CAAC;YACF,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,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,106 @@
+/**
+ * @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