@office-ai/aioncli-core 0.8.1 → 0.18.5

package/dist/src/services/modelConfig.integration.test.js

@@ -0,0 +1,247 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { ModelConfigService } from './modelConfigService.js';
+// This test suite is designed to validate the end-to-end logic of the
+// ModelConfigService with a complex, realistic configuration.
+// It tests the interplay of global settings, alias inheritance, and overrides
+// of varying specificities.
+describe('ModelConfigService Integration', () => {
+    const complexConfig = {
+        aliases: {
+            // Abstract base with no model
+            base: {
+                modelConfig: {
+                    generateContentConfig: {
+                        topP: 0.95,
+                        topK: 64,
+                    },
+                },
+            },
+            'default-text-model': {
+                extends: 'base',
+                modelConfig: {
+                    model: 'gemini-1.5-pro-latest',
+                    generateContentConfig: {
+                        topK: 40, // Override base
+                    },
+                },
+            },
+            'creative-writer': {
+                extends: 'default-text-model',
+                modelConfig: {
+                    generateContentConfig: {
+                        temperature: 0.9, // Override global
+                        topK: 50, // Override parent
+                    },
+                },
+            },
+            'fast-classifier': {
+                extends: 'base',
+                modelConfig: {
+                    model: 'gemini-1.5-flash-latest',
+                    generateContentConfig: {
+                        temperature: 0.1,
+                        candidateCount: 4,
+                    },
+                },
+            },
+        },
+        overrides: [
+            // Broad override for all flash models
+            {
+                match: { model: 'gemini-1.5-flash-latest' },
+                modelConfig: {
+                    generateContentConfig: {
+                        maxOutputTokens: 2048,
+                    },
+                },
+            },
+            // Specific override for the 'core' agent
+            {
+                match: { overrideScope: 'core' },
+                modelConfig: {
+                    generateContentConfig: {
+                        temperature: 0.5,
+                        stopSequences: ['AGENT_STOP'],
+                    },
+                },
+            },
+            // Highly specific override for the 'fast-classifier' when used by the 'core' agent
+            {
+                match: { model: 'fast-classifier', overrideScope: 'core' },
+                modelConfig: {
+                    generateContentConfig: {
+                        temperature: 0.0,
+                        maxOutputTokens: 4096,
+                    },
+                },
+            },
+            // Override to provide a model for the abstract alias
+            {
+                match: { model: 'base', overrideScope: 'core' },
+                modelConfig: {
+                    model: 'gemini-1.5-pro-latest',
+                },
+            },
+        ],
+    };
+    const service = new ModelConfigService(complexConfig);
+    it('should resolve a simple model, applying core agent defaults', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'gemini-test-model',
+        });
+        expect(resolved.model).toBe('gemini-test-model');
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5, // from agent override
+            stopSequences: ['AGENT_STOP'], // from agent override
+        });
+    });
+    it('should correctly apply a simple inherited alias and merge with global defaults', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'default-text-model',
+        });
+        expect(resolved.model).toBe('gemini-1.5-pro-latest'); // from alias
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5, // from agent override
+            topP: 0.95, // from base
+            topK: 40, // from alias
+            stopSequences: ['AGENT_STOP'], // from agent override
+        });
+    });
+    it('should resolve a multi-level inherited alias', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'creative-writer',
+        });
+        expect(resolved.model).toBe('gemini-1.5-pro-latest'); // from default-text-model
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5, // from agent override
+            topP: 0.95, // from base
+            topK: 50, // from alias
+            stopSequences: ['AGENT_STOP'], // from agent override
+        });
+    });
+    it('should apply an inherited alias and a broad model-based override', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'fast-classifier',
+            // No agent specified, so it should match core agent-specific rules
+        });
+        expect(resolved.model).toBe('gemini-1.5-flash-latest'); // from alias
+        expect(resolved.generateContentConfig).toEqual({
+            topP: 0.95, // from base
+            topK: 64, // from base
+            candidateCount: 4, // from alias
+            stopSequences: ['AGENT_STOP'], // from agent override
+            maxOutputTokens: 4096, // from most specific override
+            temperature: 0.0, // from most specific override
+        });
+    });
+    it('should apply settings for an unknown model but a known agent', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'gemini-test-model',
+            overrideScope: 'core',
+        });
+        expect(resolved.model).toBe('gemini-test-model');
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5, // from agent override
+            stopSequences: ['AGENT_STOP'], // from agent override
+        });
+    });
+    it('should apply the most specific override for a known inherited alias and agent', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'fast-classifier',
+            overrideScope: 'core',
+        });
+        expect(resolved.model).toBe('gemini-1.5-flash-latest');
+        expect(resolved.generateContentConfig).toEqual({
+            // Inherited from 'base'
+            topP: 0.95,
+            topK: 64,
+            // From 'fast-classifier' alias
+            candidateCount: 4,
+            // From 'core' agent override
+            stopSequences: ['AGENT_STOP'],
+            // From most specific override (model+agent)
+            temperature: 0.0,
+            maxOutputTokens: 4096,
+        });
+    });
+    it('should correctly apply agent override on top of a multi-level inherited alias', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'creative-writer',
+            overrideScope: 'core',
+        });
+        expect(resolved.model).toBe('gemini-1.5-pro-latest'); // from default-text-model
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5, // from agent override (wins over alias)
+            topP: 0.95, // from base
+            topK: 50, // from creative-writer alias
+            stopSequences: ['AGENT_STOP'], // from agent override
+        });
+    });
+    it('should resolve an abstract alias if a specific override provides the model', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'base',
+            overrideScope: 'core',
+        });
+        expect(resolved.model).toBe('gemini-1.5-pro-latest'); // from override
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5, // from agent override
+            topP: 0.95, // from base alias
+            topK: 64, // from base alias
+            stopSequences: ['AGENT_STOP'], // from agent override
+        });
+    });
+    it('should not apply core agent overrides when a different agent is specified', () => {
+        const resolved = service.getResolvedConfig({
+            model: 'fast-classifier',
+            overrideScope: 'non-core-agent',
+        });
+        expect(resolved.model).toBe('gemini-1.5-flash-latest');
+        expect(resolved.generateContentConfig).toEqual({
+            candidateCount: 4, // from alias
+            maxOutputTokens: 2048, // from override of model
+            temperature: 0.1, // from alias
+            topK: 64, // from base
+            topP: 0.95, // from base
+        });
+    });
+    it('should correctly merge static aliases, runtime aliases, and overrides', () => {
+        // Re-instantiate service for this isolated test to not pollute other tests
+        const service = new ModelConfigService(complexConfig);
+        // Register a runtime alias, simulating what AgentExecutor does.
+        // This alias extends a static base and provides its own settings.
+        service.registerRuntimeModelConfig('agent-runtime:my-agent', {
+            extends: 'creative-writer', // extends a multi-level alias
+            modelConfig: {
+                generateContentConfig: {
+                    temperature: 0.1, // Overrides parent
+                    maxOutputTokens: 8192, // Adds a new property
+                },
+            },
+        });
+        // Resolve the configuration for the runtime alias, with a matching agent scope
+        const resolved = service.getResolvedConfig({
+            model: 'agent-runtime:my-agent',
+            overrideScope: 'core',
+        });
+        // Assert the final merged configuration.
+        expect(resolved.model).toBe('gemini-1.5-pro-latest'); // from 'default-text-model'
+        expect(resolved.generateContentConfig).toEqual({
+            // from 'core' agent override, wins over runtime alias's 0.1 and creative-writer's 0.9
+            temperature: 0.5,
+            // from 'base' alias
+            topP: 0.95,
+            // from 'creative-writer' alias
+            topK: 50,
+            // from runtime alias
+            maxOutputTokens: 8192,
+            // from 'core' agent override
+            stopSequences: ['AGENT_STOP'],
+        });
+    });
+});
+//# sourceMappingURL=modelConfig.integration.test.js.map

package/dist/src/services/modelConfig.integration.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelConfig.integration.test.js","sourceRoot":"","sources":["../../../src/services/modelConfig.integration.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAG7D,sEAAsE;AACtE,8DAA8D;AAC9D,8EAA8E;AAC9E,4BAA4B;AAC5B,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,MAAM,aAAa,GAA6B;QAC9C,OAAO,EAAE;YACP,8BAA8B;YAC9B,IAAI,EAAE;gBACJ,WAAW,EAAE;oBACX,qBAAqB,EAAE;wBACrB,IAAI,EAAE,IAAI;wBACV,IAAI,EAAE,EAAE;qBACT;iBACF;aACF;YACD,oBAAoB,EAAE;gBACpB,OAAO,EAAE,MAAM;gBACf,WAAW,EAAE;oBACX,KAAK,EAAE,uBAAuB;oBAC9B,qBAAqB,EAAE;wBACrB,IAAI,EAAE,EAAE,EAAE,gBAAgB;qBAC3B;iBACF;aACF;YACD,iBAAiB,EAAE;gBACjB,OAAO,EAAE,oBAAoB;gBAC7B,WAAW,EAAE;oBACX,qBAAqB,EAAE;wBACrB,WAAW,EAAE,GAAG,EAAE,kBAAkB;wBACpC,IAAI,EAAE,EAAE,EAAE,kBAAkB;qBAC7B;iBACF;aACF;YACD,iBAAiB,EAAE;gBACjB,OAAO,EAAE,MAAM;gBACf,WAAW,EAAE;oBACX,KAAK,EAAE,yBAAyB;oBAChC,qBAAqB,EAAE;wBACrB,WAAW,EAAE,GAAG;wBAChB,cAAc,EAAE,CAAC;qBAClB;iBACF;aACF;SACF;QACD,SAAS,EAAE;YACT,sCAAsC;YACtC;gBACE,KAAK,EAAE,EAAE,KAAK,EAAE,yBAAyB,EAAE;gBAC3C,WAAW,EAAE;oBACX,qBAAqB,EAAE;wBACrB,eAAe,EAAE,IAAI;qBACtB;iBACF;aACF;YACD,yCAAyC;YACzC;gBACE,KAAK,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;gBAChC,WAAW,EAAE;oBACX,qBAAqB,EAAE;wBACrB,WAAW,EAAE,GAAG;wBAChB,aAAa,EAAE,CAAC,YAAY,CAAC;qBAC9B;iBACF;aACF;YACD,mFAAmF;YACnF;gBACE,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,EAAE;gBAC1D,WAAW,EAAE;oBACX,qBAAqB,EAAE;wBACrB,WAAW,EAAE,GAAG;wBAChB,eAAe,EAAE,IAAI;qBACtB;iBACF;aACF;YACD,qDAAqD;YACrD;gBACE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE;gBAC/C,WAAW,EAAE;oBACX,KAAK,EAAE,uBAAuB;iBAC/B;aACF;SACF;KACF,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,aAAa,CAAC,CAAC;IAEtD,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,mBAAmB;SAC3B,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG,EAAE,sBAAsB;YACxC,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;SACtD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,oBAAoB;SAC5B,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,aAAa;QACnE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG,EAAE,sBAAsB;YACxC,IAAI,EAAE,IAAI,EAAE,YAAY;YACxB,IAAI,EAAE,EAAE,EAAE,aAAa;YACvB,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;SACtD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,iBAAiB;SACzB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,0BAA0B;QAChF,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG,EAAE,sBAAsB;YACxC,IAAI,EAAE,IAAI,EAAE,YAAY;YACxB,IAAI,EAAE,EAAE,EAAE,aAAa;YACvB,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;SACtD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,iBAAiB;YACxB,mEAAmE;SACpE,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,aAAa;QACrE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,IAAI,EAAE,IAAI,EAAE,YAAY;YACxB,IAAI,EAAE,EAAE,EAAE,YAAY;YACtB,cAAc,EAAE,CAAC,EAAE,aAAa;YAChC,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;YACrD,eAAe,EAAE,IAAI,EAAE,8BAA8B;YACrD,WAAW,EAAE,GAAG,EAAE,8BAA8B;SACjD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,mBAAmB;YAC1B,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG,EAAE,sBAAsB;YACxC,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;SACtD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;QACvF,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,iBAAiB;YACxB,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,wBAAwB;YACxB,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,EAAE;YACR,+BAA+B;YAC/B,cAAc,EAAE,CAAC;YACjB,6BAA6B;YAC7B,aAAa,EAAE,CAAC,YAAY,CAAC;YAC7B,4CAA4C;YAC5C,WAAW,EAAE,GAAG;YAChB,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;QACvF,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,iBAAiB;YACxB,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,0BAA0B;QAChF,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG,EAAE,wCAAwC;YAC1D,IAAI,EAAE,IAAI,EAAE,YAAY;YACxB,IAAI,EAAE,EAAE,EAAE,6BAA6B;YACvC,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;SACtD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;QACpF,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,MAAM;YACb,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,gBAAgB;QACtE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG,EAAE,sBAAsB;YACxC,IAAI,EAAE,IAAI,EAAE,kBAAkB;YAC9B,IAAI,EAAE,EAAE,EAAE,kBAAkB;YAC5B,aAAa,EAAE,CAAC,YAAY,CAAC,EAAE,sBAAsB;SACtD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;QACnF,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,iBAAiB;YACxB,aAAa,EAAE,gBAAgB;SAChC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,cAAc,EAAE,CAAC,EAAE,aAAa;YAChC,eAAe,EAAE,IAAI,EAAE,yBAAyB;YAChD,WAAW,EAAE,GAAG,EAAE,aAAa;YAC/B,IAAI,EAAE,EAAE,EAAE,YAAY;YACtB,IAAI,EAAE,IAAI,EAAE,YAAY;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,2EAA2E;QAC3E,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,aAAa,CAAC,CAAC;QAEtD,gEAAgE;QAChE,kEAAkE;QAClE,OAAO,CAAC,0BAA0B,CAAC,wBAAwB,EAAE;YAC3D,OAAO,EAAE,iBAAiB,EAAE,8BAA8B;YAC1D,WAAW,EAAE;gBACX,qBAAqB,EAAE;oBACrB,WAAW,EAAE,GAAG,EAAE,mBAAmB;oBACrC,eAAe,EAAE,IAAI,EAAE,sBAAsB;iBAC9C;aACF;SACF,CAAC,CAAC;QAEH,+EAA+E;QAC/E,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,wBAAwB;YAC/B,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,yCAAyC;QACzC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,4BAA4B;QAClF,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,sFAAsF;YACtF,WAAW,EAAE,GAAG;YAChB,oBAAoB;YACpB,IAAI,EAAE,IAAI;YACV,+BAA+B;YAC/B,IAAI,EAAE,EAAE;YACR,qBAAqB;YACrB,eAAe,EAAE,IAAI;YACrB,6BAA6B;YAC7B,aAAa,EAAE,CAAC,YAAY,CAAC;SAC9B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/services/modelConfigService.d.ts

@@ -0,0 +1,48 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { GenerateContentConfig } from '@google/genai';
+export interface ModelConfigKey {
+    model: string;
+    overrideScope?: string;
+}
+export interface ModelConfig {
+    model?: string;
+    generateContentConfig?: GenerateContentConfig;
+}
+export interface ModelConfigOverride {
+    match: {
+        model?: string;
+        overrideScope?: string;
+    };
+    modelConfig: ModelConfig;
+}
+export interface ModelConfigAlias {
+    extends?: string;
+    modelConfig: ModelConfig;
+}
+export interface ModelConfigServiceConfig {
+    aliases?: Record<string, ModelConfigAlias>;
+    overrides?: ModelConfigOverride[];
+}
+export type ResolvedModelConfig = _ResolvedModelConfig & {
+    readonly _brand: unique symbol;
+};
+export interface _ResolvedModelConfig {
+    model: string;
+    generateContentConfig: GenerateContentConfig;
+}
+export declare class ModelConfigService {
+    private readonly config;
+    private readonly runtimeAliases;
+    constructor(config: ModelConfigServiceConfig);
+    registerRuntimeModelConfig(aliasName: string, alias: ModelConfigAlias): void;
+    private resolveAlias;
+    private internalGetResolvedConfig;
+    getResolvedConfig(context: ModelConfigKey): ResolvedModelConfig;
+    private isObject;
+    private deepMerge;
+    private genericDeepMerge;
+}

package/dist/src/services/modelConfigService.js

@@ -0,0 +1,151 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export class ModelConfigService {
+    config;
+    runtimeAliases = {};
+    // TODO(12597): Process config to build a typed alias hierarchy.
+    constructor(config) {
+        this.config = config;
+    }
+    registerRuntimeModelConfig(aliasName, alias) {
+        this.runtimeAliases[aliasName] = alias;
+    }
+    resolveAlias(aliasName, aliases, visited = new Set()) {
+        if (visited.has(aliasName)) {
+            throw new Error(`Circular alias dependency: ${[...visited, aliasName].join(' -> ')}`);
+        }
+        visited.add(aliasName);
+        const alias = aliases[aliasName];
+        if (!alias) {
+            throw new Error(`Alias "${aliasName}" not found.`);
+        }
+        if (!alias.extends) {
+            return alias;
+        }
+        const baseAlias = this.resolveAlias(alias.extends, aliases, visited);
+        return {
+            modelConfig: {
+                model: alias.modelConfig.model ?? baseAlias.modelConfig.model,
+                generateContentConfig: this.deepMerge(baseAlias.modelConfig.generateContentConfig, alias.modelConfig.generateContentConfig),
+            },
+        };
+    }
+    internalGetResolvedConfig(context) {
+        const config = this.config || {};
+        const { aliases = {}, overrides = [] } = config;
+        const allAliases = { ...aliases, ...this.runtimeAliases };
+        let baseModel = context.model;
+        let resolvedConfig = {};
+        // Step 1: Alias Resolution
+        if (allAliases[context.model]) {
+            const resolvedAlias = this.resolveAlias(context.model, allAliases);
+            baseModel = resolvedAlias.modelConfig.model; // This can now be undefined
+            resolvedConfig = this.deepMerge(resolvedConfig, resolvedAlias.modelConfig.generateContentConfig);
+        }
+        // If an alias was used but didn't resolve to a model, `baseModel` is undefined.
+        // We still need a model for matching overrides. We'll use the original alias name
+        // for matching if no model is resolved yet.
+        const modelForMatching = baseModel ?? context.model;
+        const finalContext = {
+            ...context,
+            model: modelForMatching,
+        };
+        // Step 2: Override Application
+        const matches = overrides
+            .map((override, index) => {
+            const matchEntries = Object.entries(override.match);
+            if (matchEntries.length === 0) {
+                return null;
+            }
+            const isMatch = matchEntries.every(([key, value]) => {
+                if (key === 'model') {
+                    return value === context.model || value === finalContext.model;
+                }
+                if (key === 'overrideScope' && value === 'core') {
+                    // The 'core' overrideScope is special. It should match if the
+                    // overrideScope is explicitly 'core' or if the overrideScope
+                    // is not specified.
+                    return context.overrideScope === 'core' || !context.overrideScope;
+                }
+                return finalContext[key] === value;
+            });
+            if (isMatch) {
+                return {
+                    specificity: matchEntries.length,
+                    modelConfig: override.modelConfig,
+                    index,
+                };
+            }
+            return null;
+        })
+            .filter((match) => match !== null);
+        // The override application logic is designed to be both simple and powerful.
+        // By first sorting all matching overrides by specificity (and then by their
+        // original order as a tie-breaker), we ensure that as we merge the `config`
+        // objects, the settings from the most specific rules are applied last,
+        // correctly overwriting any values from broader, less-specific rules.
+        // This achieves a per-property override effect without complex per-property logic.
+        matches.sort((a, b) => {
+            if (a.specificity !== b.specificity) {
+                return a.specificity - b.specificity;
+            }
+            return a.index - b.index;
+        });
+        // Apply matching overrides
+        for (const match of matches) {
+            if (match.modelConfig.model) {
+                baseModel = match.modelConfig.model;
+            }
+            if (match.modelConfig.generateContentConfig) {
+                resolvedConfig = this.deepMerge(resolvedConfig, match.modelConfig.generateContentConfig);
+            }
+        }
+        return {
+            model: baseModel,
+            generateContentConfig: resolvedConfig,
+        };
+    }
+    getResolvedConfig(context) {
+        const resolved = this.internalGetResolvedConfig(context);
+        if (!resolved.model) {
+            throw new Error(`Could not resolve a model name for alias "${context.model}". Please ensure the alias chain or a matching override specifies a model.`);
+        }
+        return {
+            model: resolved.model,
+            generateContentConfig: resolved.generateContentConfig,
+        };
+    }
+    isObject(item) {
+        return !!item && typeof item === 'object' && !Array.isArray(item);
+    }
+    deepMerge(config1, config2) {
+        return this.genericDeepMerge(config1, config2);
+    }
+    genericDeepMerge(...objects) {
+        return objects.reduce((acc, obj) => {
+            if (!obj) {
+                return acc;
+            }
+            Object.keys(obj).forEach((key) => {
+                const accValue = acc[key];
+                const objValue = obj[key];
+                // For now, we only deep merge objects, and not arrays. This is because
+                // If we deep merge arrays, there is no way for the user to completely
+                // override the base array.
+                // TODO(joshualitt): Consider knobs here, i.e. opt-in to deep merging
+                // arrays on a case-by-case basis.
+                if (this.isObject(accValue) && this.isObject(objValue)) {
+                    acc[key] = this.deepMerge(accValue, objValue);
+                }
+                else {
+                    acc[key] = objValue;
+                }
+            });
+            return acc;
+        }, {});
+    }
+}
+//# sourceMappingURL=modelConfigService.js.map

package/dist/src/services/modelConfigService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelConfigService.js","sourceRoot":"","sources":["../../../src/services/modelConfigService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAqDH,MAAM,OAAO,kBAAkB;IAIA;IAHZ,cAAc,GAAqC,EAAE,CAAC;IAEvE,gEAAgE;IAChE,YAA6B,MAAgC;QAAhC,WAAM,GAAN,MAAM,CAA0B;IAAG,CAAC;IAEjE,0BAA0B,CAAC,SAAiB,EAAE,KAAuB;QACnE,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC;IACzC,CAAC;IAEO,YAAY,CAClB,SAAiB,EACjB,OAAyC,EACzC,UAAU,IAAI,GAAG,EAAU;QAE3B,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CACb,8BAA8B,CAAC,GAAG,OAAO,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CACrE,CAAC;QACJ,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEvB,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,UAAU,SAAS,cAAc,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAErE,OAAO;YACL,WAAW,EAAE;gBACX,KAAK,EAAE,KAAK,CAAC,WAAW,CAAC,KAAK,IAAI,SAAS,CAAC,WAAW,CAAC,KAAK;gBAC7D,qBAAqB,EAAE,IAAI,CAAC,SAAS,CACnC,SAAS,CAAC,WAAW,CAAC,qBAAqB,EAC3C,KAAK,CAAC,WAAW,CAAC,qBAAqB,CACxC;aACF;SACF,CAAC;IACJ,CAAC;IAEO,yBAAyB,CAAC,OAAuB;QAIvD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;QACjC,MAAM,EAAE,OAAO,GAAG,EAAE,EAAE,SAAS,GAAG,EAAE,EAAE,GAAG,MAAM,CAAC;QAChD,MAAM,UAAU,GAAG,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1D,IAAI,SAAS,GAAuB,OAAO,CAAC,KAAK,CAAC;QAClD,IAAI,cAAc,GAA0B,EAAE,CAAC;QAE/C,2BAA2B;QAC3B,IAAI,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YACnE,SAAS,GAAG,aAAa,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,4BAA4B;YACzE,cAAc,GAAG,IAAI,CAAC,SAAS,CAC7B,cAAc,EACd,aAAa,CAAC,WAAW,CAAC,qBAAqB,CAChD,CAAC;QACJ,CAAC;QAED,gFAAgF;QAChF,kFAAkF;QAClF,4CAA4C;QAC5C,MAAM,gBAAgB,GAAG,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC;QAEpD,MAAM,YAAY,GAAG;YACnB,GAAG,OAAO;YACV,KAAK,EAAE,gBAAgB;SACxB,CAAC;QAEF,+BAA+B;QAC/B,MAAM,OAAO,GAAG,SAAS;aACtB,GAAG,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YACpD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAClD,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;oBACpB,OAAO,KAAK,KAAK,OAAO,CAAC,KAAK,IAAI,KAAK,KAAK,YAAY,CAAC,KAAK,CAAC;gBACjE,CAAC;gBACD,IAAI,GAAG,KAAK,eAAe,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;oBAChD,8DAA8D;oBAC9D,6DAA6D;oBAC7D,oBAAoB;oBACpB,OAAO,OAAO,CAAC,aAAa,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;gBACpE,CAAC;gBACD,OAAO,YAAY,CAAC,GAA2B,CAAC,KAAK,KAAK,CAAC;YAC7D,CAAC,CAAC,CAAC;YAEH,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO;oBACL,WAAW,EAAE,YAAY,CAAC,MAAM;oBAChC,WAAW,EAAE,QAAQ,CAAC,WAAW;oBACjC,KAAK;iBACN,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,KAAK,EAAsC,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC;QAEzE,6EAA6E;QAC7E,4EAA4E;QAC5E,4EAA4E;QAC5E,uEAAuE;QACvE,sEAAsE;QACtE,mFAAmF;QACnF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACpB,IAAI,CAAC,CAAC,WAAW,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;gBACpC,OAAO,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YACvC,CAAC;YACD,OAAO,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,2BAA2B;QAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;gBAC5B,SAAS,GAAG,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC;YACtC,CAAC;YACD,IAAI,KAAK,CAAC,WAAW,CAAC,qBAAqB,EAAE,CAAC;gBAC5C,cAAc,GAAG,IAAI,CAAC,SAAS,CAC7B,cAAc,EACd,KAAK,CAAC,WAAW,CAAC,qBAAqB,CACxC,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,KAAK,EAAE,SAAS;YAChB,qBAAqB,EAAE,cAAc;SACtC,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,OAAuB;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;QAEzD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CACb,6CAA6C,OAAO,CAAC,KAAK,4EAA4E,CACvI,CAAC;QACJ,CAAC;QAED,OAAO;YACL,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,qBAAqB,EAAE,QAAQ,CAAC,qBAAqB;SAC/B,CAAC;IAC3B,CAAC;IAEO,QAAQ,CAAC,IAAa;QAC5B,OAAO,CAAC,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;IAEO,SAAS,CACf,OAA0C,EAC1C,OAA0C;QAE1C,OAAO,IAAI,CAAC,gBAAgB,CAC1B,OAA8C,EAC9C,OAA8C,CAC/C,CAAC;IACJ,CAAC;IAEO,gBAAgB,CACtB,GAAG,OAAmD;QAEtD,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,GAA4B,EAAE,GAAG,EAAE,EAAE;YAC1D,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,OAAO,GAAG,CAAC;YACb,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC/B,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;gBAC1B,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;gBAE1B,uEAAuE;gBACvE,sEAAsE;gBACtE,2BAA2B;gBAC3B,qEAAqE;gBACrE,kCAAkC;gBAClC,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACvD,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CACvB,QAAmC,EACnC,QAAmC,CACpC,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC;gBACtB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;IACT,CAAC;CACF"}

package/dist/src/services/modelConfigService.test.d.ts

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