@office-ai/aioncli-core 0.2.3 → 0.18.4

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

@@ -0,0 +1,531 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { ModelConfigService } from './modelConfigService.js';
+describe('ModelConfigService', () => {
+    it('should resolve a basic alias to its model and settings', () => {
+        const config = {
+            aliases: {
+                classifier: {
+                    modelConfig: {
+                        model: 'gemini-1.5-flash-latest',
+                        generateContentConfig: {
+                            temperature: 0,
+                            topP: 0.9,
+                        },
+                    },
+                },
+            },
+            overrides: [],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({ model: 'classifier' });
+        expect(resolved.model).toBe('gemini-1.5-flash-latest');
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0,
+            topP: 0.9,
+        });
+    });
+    it('should apply a simple override on top of an alias', () => {
+        const config = {
+            aliases: {
+                classifier: {
+                    modelConfig: {
+                        model: 'gemini-1.5-flash-latest',
+                        generateContentConfig: {
+                            temperature: 0,
+                            topP: 0.9,
+                        },
+                    },
+                },
+            },
+            overrides: [
+                {
+                    match: { model: 'classifier' },
+                    modelConfig: {
+                        generateContentConfig: {
+                            temperature: 0.5,
+                            maxOutputTokens: 1000,
+                        },
+                    },
+                },
+            ],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({ model: 'classifier' });
+        expect(resolved.model).toBe('gemini-1.5-flash-latest');
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.5,
+            topP: 0.9,
+            maxOutputTokens: 1000,
+        });
+    });
+    it('should apply the most specific override rule', () => {
+        const config = {
+            aliases: {},
+            overrides: [
+                {
+                    match: { model: 'gemini-pro' },
+                    modelConfig: { generateContentConfig: { temperature: 0.5 } },
+                },
+                {
+                    match: { model: 'gemini-pro', overrideScope: 'my-agent' },
+                    modelConfig: { generateContentConfig: { temperature: 0.1 } },
+                },
+            ],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({
+            model: 'gemini-pro',
+            overrideScope: 'my-agent',
+        });
+        expect(resolved.model).toBe('gemini-pro');
+        expect(resolved.generateContentConfig).toEqual({ temperature: 0.1 });
+    });
+    it('should use the last override in case of a tie in specificity', () => {
+        const config = {
+            aliases: {},
+            overrides: [
+                {
+                    match: { model: 'gemini-pro' },
+                    modelConfig: {
+                        generateContentConfig: { temperature: 0.5, topP: 0.8 },
+                    },
+                },
+                {
+                    match: { model: 'gemini-pro' },
+                    modelConfig: { generateContentConfig: { temperature: 0.1 } },
+                },
+            ],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({ model: 'gemini-pro' });
+        expect(resolved.model).toBe('gemini-pro');
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.1,
+            topP: 0.8,
+        });
+    });
+    it('should correctly pass through generation config from an alias', () => {
+        const config = {
+            aliases: {
+                'thinking-alias': {
+                    modelConfig: {
+                        model: 'gemini-pro',
+                        generateContentConfig: {
+                            candidateCount: 500,
+                        },
+                    },
+                },
+            },
+            overrides: [],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({ model: 'thinking-alias' });
+        expect(resolved.generateContentConfig).toEqual({ candidateCount: 500 });
+    });
+    it('should let an override generation config win over an alias config', () => {
+        const config = {
+            aliases: {
+                'thinking-alias': {
+                    modelConfig: {
+                        model: 'gemini-pro',
+                        generateContentConfig: {
+                            candidateCount: 500,
+                        },
+                    },
+                },
+            },
+            overrides: [
+                {
+                    match: { model: 'thinking-alias' },
+                    modelConfig: {
+                        generateContentConfig: {
+                            candidateCount: 1000,
+                        },
+                    },
+                },
+            ],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({ model: 'thinking-alias' });
+        expect(resolved.generateContentConfig).toEqual({
+            candidateCount: 1000,
+        });
+    });
+    it('should merge settings from global, alias, and multiple matching overrides', () => {
+        const config = {
+            aliases: {
+                'test-alias': {
+                    modelConfig: {
+                        model: 'gemini-test-model',
+                        generateContentConfig: {
+                            topP: 0.9,
+                            topK: 50,
+                        },
+                    },
+                },
+            },
+            overrides: [
+                {
+                    match: { model: 'gemini-test-model' },
+                    modelConfig: {
+                        generateContentConfig: {
+                            topK: 40,
+                            maxOutputTokens: 2048,
+                        },
+                    },
+                },
+                {
+                    match: { overrideScope: 'test-agent' },
+                    modelConfig: {
+                        generateContentConfig: {
+                            maxOutputTokens: 4096,
+                        },
+                    },
+                },
+                {
+                    match: { model: 'gemini-test-model', overrideScope: 'test-agent' },
+                    modelConfig: {
+                        generateContentConfig: {
+                            temperature: 0.2,
+                        },
+                    },
+                },
+            ],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({
+            model: 'test-alias',
+            overrideScope: 'test-agent',
+        });
+        expect(resolved.model).toBe('gemini-test-model');
+        expect(resolved.generateContentConfig).toEqual({
+            // From global, overridden by most specific override
+            temperature: 0.2,
+            // From alias, not overridden
+            topP: 0.9,
+            // From alias, overridden by less specific override
+            topK: 40,
+            // From first matching override, overridden by second matching override
+            maxOutputTokens: 4096,
+        });
+    });
+    it('should match an agent:core override when agent is undefined', () => {
+        const config = {
+            aliases: {},
+            overrides: [
+                {
+                    match: { overrideScope: 'core' },
+                    modelConfig: {
+                        generateContentConfig: {
+                            temperature: 0.1,
+                        },
+                    },
+                },
+            ],
+        };
+        const service = new ModelConfigService(config);
+        const resolved = service.getResolvedConfig({
+            model: 'gemini-pro',
+            overrideScope: undefined, // Explicitly undefined
+        });
+        expect(resolved.model).toBe('gemini-pro');
+        expect(resolved.generateContentConfig).toEqual({
+            temperature: 0.1,
+        });
+    });
+    describe('alias inheritance', () => {
+        it('should resolve a simple "extends" chain', () => {
+            const config = {
+                aliases: {
+                    base: {
+                        modelConfig: {
+                            model: 'gemini-1.5-pro-latest',
+                            generateContentConfig: {
+                                temperature: 0.7,
+                                topP: 0.9,
+                            },
+                        },
+                    },
+                    'flash-variant': {
+                        extends: 'base',
+                        modelConfig: {
+                            model: 'gemini-1.5-flash-latest',
+                        },
+                    },
+                },
+            };
+            const service = new ModelConfigService(config);
+            const resolved = service.getResolvedConfig({ model: 'flash-variant' });
+            expect(resolved.model).toBe('gemini-1.5-flash-latest');
+            expect(resolved.generateContentConfig).toEqual({
+                temperature: 0.7,
+                topP: 0.9,
+            });
+        });
+        it('should override parent properties from child alias', () => {
+            const config = {
+                aliases: {
+                    base: {
+                        modelConfig: {
+                            model: 'gemini-1.5-pro-latest',
+                            generateContentConfig: {
+                                temperature: 0.7,
+                                topP: 0.9,
+                            },
+                        },
+                    },
+                    'flash-variant': {
+                        extends: 'base',
+                        modelConfig: {
+                            model: 'gemini-1.5-flash-latest',
+                            generateContentConfig: {
+                                temperature: 0.2,
+                            },
+                        },
+                    },
+                },
+            };
+            const service = new ModelConfigService(config);
+            const resolved = service.getResolvedConfig({ model: 'flash-variant' });
+            expect(resolved.model).toBe('gemini-1.5-flash-latest');
+            expect(resolved.generateContentConfig).toEqual({
+                temperature: 0.2,
+                topP: 0.9,
+            });
+        });
+        it('should resolve a multi-level "extends" chain', () => {
+            const config = {
+                aliases: {
+                    base: {
+                        modelConfig: {
+                            model: 'gemini-1.5-pro-latest',
+                            generateContentConfig: {
+                                temperature: 0.7,
+                                topP: 0.9,
+                            },
+                        },
+                    },
+                    'base-flash': {
+                        extends: 'base',
+                        modelConfig: {
+                            model: 'gemini-1.5-flash-latest',
+                        },
+                    },
+                    'classifier-flash': {
+                        extends: 'base-flash',
+                        modelConfig: {
+                            generateContentConfig: {
+                                temperature: 0,
+                            },
+                        },
+                    },
+                },
+            };
+            const service = new ModelConfigService(config);
+            const resolved = service.getResolvedConfig({
+                model: 'classifier-flash',
+            });
+            expect(resolved.model).toBe('gemini-1.5-flash-latest');
+            expect(resolved.generateContentConfig).toEqual({
+                temperature: 0,
+                topP: 0.9,
+            });
+        });
+        it('should throw an error for circular dependencies', () => {
+            const config = {
+                aliases: {
+                    a: { extends: 'b', modelConfig: {} },
+                    b: { extends: 'a', modelConfig: {} },
+                },
+            };
+            const service = new ModelConfigService(config);
+            expect(() => service.getResolvedConfig({ model: 'a' })).toThrow('Circular alias dependency: a -> b -> a');
+        });
+        describe('abstract aliases', () => {
+            it('should allow an alias to extend an abstract alias without a model', () => {
+                const config = {
+                    aliases: {
+                        'abstract-base': {
+                            modelConfig: {
+                                generateContentConfig: {
+                                    temperature: 0.1,
+                                },
+                            },
+                        },
+                        'concrete-child': {
+                            extends: 'abstract-base',
+                            modelConfig: {
+                                model: 'gemini-1.5-pro-latest',
+                                generateContentConfig: {
+                                    topP: 0.9,
+                                },
+                            },
+                        },
+                    },
+                };
+                const service = new ModelConfigService(config);
+                const resolved = service.getResolvedConfig({ model: 'concrete-child' });
+                expect(resolved.model).toBe('gemini-1.5-pro-latest');
+                expect(resolved.generateContentConfig).toEqual({
+                    temperature: 0.1,
+                    topP: 0.9,
+                });
+            });
+            it('should throw an error if a resolved alias chain has no model', () => {
+                const config = {
+                    aliases: {
+                        'abstract-base': {
+                            modelConfig: {
+                                generateContentConfig: { temperature: 0.7 },
+                            },
+                        },
+                    },
+                };
+                const service = new ModelConfigService(config);
+                expect(() => service.getResolvedConfig({ model: 'abstract-base' })).toThrow('Could not resolve a model name for alias "abstract-base". Please ensure the alias chain or a matching override specifies a model.');
+            });
+            it('should resolve an abstract alias if an override provides the model', () => {
+                const config = {
+                    aliases: {
+                        'abstract-base': {
+                            modelConfig: {
+                                generateContentConfig: {
+                                    temperature: 0.1,
+                                },
+                            },
+                        },
+                    },
+                    overrides: [
+                        {
+                            match: { model: 'abstract-base' },
+                            modelConfig: {
+                                model: 'gemini-1.5-flash-latest',
+                            },
+                        },
+                    ],
+                };
+                const service = new ModelConfigService(config);
+                const resolved = service.getResolvedConfig({ model: 'abstract-base' });
+                expect(resolved.model).toBe('gemini-1.5-flash-latest');
+                expect(resolved.generateContentConfig).toEqual({
+                    temperature: 0.1,
+                });
+            });
+        });
+        it('should throw an error if an extended alias does not exist', () => {
+            const config = {
+                aliases: {
+                    'bad-alias': {
+                        extends: 'non-existent',
+                        modelConfig: {},
+                    },
+                },
+            };
+            const service = new ModelConfigService(config);
+            expect(() => service.getResolvedConfig({ model: 'bad-alias' })).toThrow('Alias "non-existent" not found.');
+        });
+    });
+    describe('deep merging', () => {
+        it('should deep merge nested config objects from aliases and overrides', () => {
+            const config = {
+                aliases: {
+                    'base-safe': {
+                        modelConfig: {
+                            model: 'gemini-pro',
+                            generateContentConfig: {
+                                safetySettings: {
+                                    HARM_CATEGORY_HARASSMENT: 'BLOCK_ONLY_HIGH',
+                                    HARM_CATEGORY_HATE_SPEECH: 'BLOCK_ONLY_HIGH',
+                                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                },
+                            },
+                        },
+                    },
+                },
+                overrides: [
+                    {
+                        match: { model: 'base-safe' },
+                        modelConfig: {
+                            generateContentConfig: {
+                                safetySettings: {
+                                    HARM_CATEGORY_HATE_SPEECH: 'BLOCK_NONE',
+                                    HARM_CATEGORY_SEXUALLY_EXPLICIT: 'BLOCK_MEDIUM_AND_ABOVE',
+                                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                                },
+                            },
+                        },
+                    },
+                ],
+            };
+            const service = new ModelConfigService(config);
+            const resolved = service.getResolvedConfig({ model: 'base-safe' });
+            expect(resolved.model).toBe('gemini-pro');
+            expect(resolved.generateContentConfig.safetySettings).toEqual({
+                // From alias
+                HARM_CATEGORY_HARASSMENT: 'BLOCK_ONLY_HIGH',
+                // From alias, overridden by override
+                HARM_CATEGORY_HATE_SPEECH: 'BLOCK_NONE',
+                // From override
+                HARM_CATEGORY_SEXUALLY_EXPLICIT: 'BLOCK_MEDIUM_AND_ABOVE',
+            });
+        });
+        it('should not deeply merge merge arrays from aliases and overrides', () => {
+            const config = {
+                aliases: {
+                    base: {
+                        modelConfig: {
+                            model: 'gemini-pro',
+                            generateContentConfig: {
+                                stopSequences: ['foo'],
+                            },
+                        },
+                    },
+                },
+                overrides: [
+                    {
+                        match: { model: 'base' },
+                        modelConfig: {
+                            generateContentConfig: {
+                                stopSequences: ['overrideFoo'],
+                            },
+                        },
+                    },
+                ],
+            };
+            const service = new ModelConfigService(config);
+            const resolved = service.getResolvedConfig({ model: 'base' });
+            expect(resolved.model).toBe('gemini-pro');
+            expect(resolved.generateContentConfig.stopSequences).toEqual([
+                'overrideFoo',
+            ]);
+        });
+    });
+    describe('runtime aliases', () => {
+        it('should resolve a simple runtime-registered alias', () => {
+            const config = {
+                aliases: {},
+                overrides: [],
+            };
+            const service = new ModelConfigService(config);
+            service.registerRuntimeModelConfig('runtime-alias', {
+                modelConfig: {
+                    model: 'gemini-runtime-model',
+                    generateContentConfig: {
+                        temperature: 0.123,
+                    },
+                },
+            });
+            const resolved = service.getResolvedConfig({ model: 'runtime-alias' });
+            expect(resolved.model).toBe('gemini-runtime-model');
+            expect(resolved.generateContentConfig).toEqual({
+                temperature: 0.123,
+            });
+        });
+    });
+});
+//# sourceMappingURL=modelConfigService.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"modelConfigService.test.js","sourceRoot":"","sources":["../../../src/services/modelConfigService.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE;gBACP,UAAU,EAAE;oBACV,WAAW,EAAE;wBACX,KAAK,EAAE,yBAAyB;wBAChC,qBAAqB,EAAE;4BACrB,WAAW,EAAE,CAAC;4BACd,IAAI,EAAE,GAAG;yBACV;qBACF;iBACF;aACF;YACD,SAAS,EAAE,EAAE;SACd,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,GAAG;SACV,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE;gBACP,UAAU,EAAE;oBACV,WAAW,EAAE;wBACX,KAAK,EAAE,yBAAyB;wBAChC,qBAAqB,EAAE;4BACrB,WAAW,EAAE,CAAC;4BACd,IAAI,EAAE,GAAG;yBACV;qBACF;iBACF;aACF;YACD,SAAS,EAAE;gBACT;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC9B,WAAW,EAAE;wBACX,qBAAqB,EAAE;4BACrB,WAAW,EAAE,GAAG;4BAChB,eAAe,EAAE,IAAI;yBACtB;qBACF;iBACF;aACF;SACF,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG;YAChB,IAAI,EAAE,GAAG;YACT,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE,EAAE;YACX,SAAS,EAAE;gBACT;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC9B,WAAW,EAAE,EAAE,qBAAqB,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE;iBAC7D;gBACD;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE;oBACzD,WAAW,EAAE,EAAE,qBAAqB,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE;iBAC7D;aACF;SACF,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,YAAY;YACnB,aAAa,EAAE,UAAU;SAC1B,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE,EAAE;YACX,SAAS,EAAE;gBACT;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC9B,WAAW,EAAE;wBACX,qBAAqB,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE;qBACvD;iBACF;gBACD;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC9B,WAAW,EAAE,EAAE,qBAAqB,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE;iBAC7D;aACF;SACF,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG;YAChB,IAAI,EAAE,GAAG;SACV,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE;gBACP,gBAAgB,EAAE;oBAChB,WAAW,EAAE;wBACX,KAAK,EAAE,YAAY;wBACnB,qBAAqB,EAAE;4BACrB,cAAc,EAAE,GAAG;yBACpB;qBACF;iBACF;aACF;YACD,SAAS,EAAE,EAAE;SACd,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAExE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE;gBACP,gBAAgB,EAAE;oBAChB,WAAW,EAAE;wBACX,KAAK,EAAE,YAAY;wBACnB,qBAAqB,EAAE;4BACrB,cAAc,EAAE,GAAG;yBACpB;qBACF;iBACF;aACF;YACD,SAAS,EAAE;gBACT;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,gBAAgB,EAAE;oBAClC,WAAW,EAAE;wBACX,qBAAqB,EAAE;4BACrB,cAAc,EAAE,IAAI;yBACrB;qBACF;iBACF;aACF;SACF,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAExE,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,cAAc,EAAE,IAAI;SACrB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;QACnF,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE;gBACP,YAAY,EAAE;oBACZ,WAAW,EAAE;wBACX,KAAK,EAAE,mBAAmB;wBAC1B,qBAAqB,EAAE;4BACrB,IAAI,EAAE,GAAG;4BACT,IAAI,EAAE,EAAE;yBACT;qBACF;iBACF;aACF;YACD,SAAS,EAAE;gBACT;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE;oBACrC,WAAW,EAAE;wBACX,qBAAqB,EAAE;4BACrB,IAAI,EAAE,EAAE;4BACR,eAAe,EAAE,IAAI;yBACtB;qBACF;iBACF;gBACD;oBACE,KAAK,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE;oBACtC,WAAW,EAAE;wBACX,qBAAqB,EAAE;4BACrB,eAAe,EAAE,IAAI;yBACtB;qBACF;iBACF;gBACD;oBACE,KAAK,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE,aAAa,EAAE,YAAY,EAAE;oBAClE,WAAW,EAAE;wBACX,qBAAqB,EAAE;4BACrB,WAAW,EAAE,GAAG;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,YAAY;YACnB,aAAa,EAAE,YAAY;SAC5B,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,oDAAoD;YACpD,WAAW,EAAE,GAAG;YAChB,6BAA6B;YAC7B,IAAI,EAAE,GAAG;YACT,mDAAmD;YACnD,IAAI,EAAE,EAAE;YACR,uEAAuE;YACvE,eAAe,EAAE,IAAI;SACtB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,MAAM,GAA6B;YACvC,OAAO,EAAE,EAAE;YACX,SAAS,EAAE;gBACT;oBACE,KAAK,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;oBAChC,WAAW,EAAE;wBACX,qBAAqB,EAAE;4BACrB,WAAW,EAAE,GAAG;yBACjB;qBACF;iBACF;aACF;SACF,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;YACzC,KAAK,EAAE,YAAY;YACnB,aAAa,EAAE,SAAS,EAAE,uBAAuB;SAClD,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;YAC7C,WAAW,EAAE,GAAG;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,IAAI,EAAE;wBACJ,WAAW,EAAE;4BACX,KAAK,EAAE,uBAAuB;4BAC9B,qBAAqB,EAAE;gCACrB,WAAW,EAAE,GAAG;gCAChB,IAAI,EAAE,GAAG;6BACV;yBACF;qBACF;oBACD,eAAe,EAAE;wBACf,OAAO,EAAE,MAAM;wBACf,WAAW,EAAE;4BACX,KAAK,EAAE,yBAAyB;yBACjC;qBACF;iBACF;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;gBAC7C,WAAW,EAAE,GAAG;gBAChB,IAAI,EAAE,GAAG;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,IAAI,EAAE;wBACJ,WAAW,EAAE;4BACX,KAAK,EAAE,uBAAuB;4BAC9B,qBAAqB,EAAE;gCACrB,WAAW,EAAE,GAAG;gCAChB,IAAI,EAAE,GAAG;6BACV;yBACF;qBACF;oBACD,eAAe,EAAE;wBACf,OAAO,EAAE,MAAM;wBACf,WAAW,EAAE;4BACX,KAAK,EAAE,yBAAyB;4BAChC,qBAAqB,EAAE;gCACrB,WAAW,EAAE,GAAG;6BACjB;yBACF;qBACF;iBACF;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;gBAC7C,WAAW,EAAE,GAAG;gBAChB,IAAI,EAAE,GAAG;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,IAAI,EAAE;wBACJ,WAAW,EAAE;4BACX,KAAK,EAAE,uBAAuB;4BAC9B,qBAAqB,EAAE;gCACrB,WAAW,EAAE,GAAG;gCAChB,IAAI,EAAE,GAAG;6BACV;yBACF;qBACF;oBACD,YAAY,EAAE;wBACZ,OAAO,EAAE,MAAM;wBACf,WAAW,EAAE;4BACX,KAAK,EAAE,yBAAyB;yBACjC;qBACF;oBACD,kBAAkB,EAAE;wBAClB,OAAO,EAAE,YAAY;wBACrB,WAAW,EAAE;4BACX,qBAAqB,EAAE;gCACrB,WAAW,EAAE,CAAC;6BACf;yBACF;qBACF;iBACF;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC;gBACzC,KAAK,EAAE,kBAAkB;aAC1B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;gBAC7C,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,GAAG;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE;oBACpC,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE;iBACrC;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAC7D,wCAAwC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;YAChC,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;gBAC3E,MAAM,MAAM,GAA6B;oBACvC,OAAO,EAAE;wBACP,eAAe,EAAE;4BACf,WAAW,EAAE;gCACX,qBAAqB,EAAE;oCACrB,WAAW,EAAE,GAAG;iCACjB;6BACF;yBACF;wBACD,gBAAgB,EAAE;4BAChB,OAAO,EAAE,eAAe;4BACxB,WAAW,EAAE;gCACX,KAAK,EAAE,uBAAuB;gCAC9B,qBAAqB,EAAE;oCACrB,IAAI,EAAE,GAAG;iCACV;6BACF;yBACF;qBACF;iBACF,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;gBAExE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBACrD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;oBAC7C,WAAW,EAAE,GAAG;oBAChB,IAAI,EAAE,GAAG;iBACV,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;gBACtE,MAAM,MAAM,GAA6B;oBACvC,OAAO,EAAE;wBACP,eAAe,EAAE;4BACf,WAAW,EAAE;gCACX,qBAAqB,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;6BAC5C;yBACF;qBACF;iBACF,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,CAAC,GAAG,EAAE,CACV,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CACtD,CAAC,OAAO,CACP,mIAAmI,CACpI,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;gBAC5E,MAAM,MAAM,GAA6B;oBACvC,OAAO,EAAE;wBACP,eAAe,EAAE;4BACf,WAAW,EAAE;gCACX,qBAAqB,EAAE;oCACrB,WAAW,EAAE,GAAG;iCACjB;6BACF;yBACF;qBACF;oBACD,SAAS,EAAE;wBACT;4BACE,KAAK,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;4BACjC,WAAW,EAAE;gCACX,KAAK,EAAE,yBAAyB;6BACjC;yBACF;qBACF;iBACF,CAAC;gBACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;gBAEvE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;gBACvD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;oBAC7C,WAAW,EAAE,GAAG;iBACjB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,OAAO,EAAE,cAAc;wBACvB,WAAW,EAAE,EAAE;qBAChB;iBACF;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,OAAO,CACrE,iCAAiC,CAClC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,WAAW,EAAE;4BACX,KAAK,EAAE,YAAY;4BACnB,qBAAqB,EAAE;gCACrB,cAAc,EAAE;oCACd,wBAAwB,EAAE,iBAAiB;oCAC3C,yBAAyB,EAAE,iBAAiB;oCAC5C,8DAA8D;iCACxD;6BACT;yBACF;qBACF;iBACF;gBACD,SAAS,EAAE;oBACT;wBACE,KAAK,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;wBAC7B,WAAW,EAAE;4BACX,qBAAqB,EAAE;gCACrB,cAAc,EAAE;oCACd,yBAAyB,EAAE,YAAY;oCACvC,+BAA+B,EAAE,wBAAwB;oCACzD,8DAA8D;iCACxD;6BACT;yBACF;qBACF;iBACF;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;YAEnE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC;gBAC5D,aAAa;gBACb,wBAAwB,EAAE,iBAAiB;gBAC3C,qCAAqC;gBACrC,yBAAyB,EAAE,YAAY;gBACvC,gBAAgB;gBAChB,+BAA+B,EAAE,wBAAwB;aAC1D,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE;oBACP,IAAI,EAAE;wBACJ,WAAW,EAAE;4BACX,KAAK,EAAE,YAAY;4BACnB,qBAAqB,EAAE;gCACrB,aAAa,EAAE,CAAC,KAAK,CAAC;6BACvB;yBACF;qBACF;iBACF;gBACD,SAAS,EAAE;oBACT;wBACE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE;wBACxB,WAAW,EAAE;4BACX,qBAAqB,EAAE;gCACrB,aAAa,EAAE,CAAC,aAAa,CAAC;6BAC/B;yBACF;qBACF;iBACF;aACF,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAE9D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC;gBAC3D,aAAa;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,MAAM,GAA6B;gBACvC,OAAO,EAAE,EAAE;gBACX,SAAS,EAAE,EAAE;aACd,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAE/C,OAAO,CAAC,0BAA0B,CAAC,eAAe,EAAE;gBAClD,WAAW,EAAE;oBACX,KAAK,EAAE,sBAAsB;oBAC7B,qBAAqB,EAAE;wBACrB,WAAW,EAAE,KAAK;qBACnB;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;YAEvE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACpD,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC;gBAC7C,WAAW,EAAE,KAAK;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { type AnsiOutput } from '../utils/terminalSerializer.js';
 /** A structured result from a shell command execution. */
 export interface ShellExecutionResult {
     /** The raw, unprocessed output buffer. */
@@ -29,6 +30,15 @@ export interface ShellExecutionHandle {
     /** A promise that resolves with the complete execution result. */
     result: Promise<ShellExecutionResult>;
 }
+export interface ShellExecutionConfig {
+    terminalWidth?: number;
+    terminalHeight?: number;
+    pager?: string;
+    showColor?: boolean;
+    defaultFg?: string;
+    defaultBg?: string;
+    disableDynamicLineTrimming?: boolean;
+}
 /**
  * Describes a structured event emitted during shell command execution.
  */
@@ -36,7 +46,7 @@ export type ShellOutputEvent = {
     /** The event contains a chunk of output data. */
     type: 'data';
     /** The decoded string chunk. */
-    chunk: string;
+    chunk: string | AnsiOutput;
 } | {
     /** Signals that the output stream has been identified as binary. */
     type: 'binary_detected';
@@ -52,6 +62,7 @@ export type ShellOutputEvent = {
  *
  */
 export declare class ShellExecutionService {
+    private static activePtys;
     /**
      * Executes a shell command using `node-pty`, capturing all output and lifecycle events.
      *
@@ -62,7 +73,31 @@ export declare class ShellExecutionService {
      * @returns An object containing the process ID (pid) and a promise that
      *          resolves with the complete execution result.
      */
-    static execute(commandToExecute: string, cwd: string, onOutputEvent: (event: ShellOutputEvent) => void, abortSignal: AbortSignal, shouldUseNodePty: boolean, terminalColumns?: number, terminalRows?: number): Promise<ShellExecutionHandle>;
+    static execute(commandToExecute: string, cwd: string, onOutputEvent: (event: ShellOutputEvent) => void, abortSignal: AbortSignal, shouldUseNodePty: boolean, shellExecutionConfig: ShellExecutionConfig): Promise<ShellExecutionHandle>;
+    private static appendAndTruncate;
     private static childProcessFallback;
     private static executeWithPty;
+    /**
+     * Writes a string to the pseudo-terminal (PTY) of a running process.
+     *
+     * @param pid The process ID of the target PTY.
+     * @param input The string to write to the terminal.
+     */
+    static writeToPty(pid: number, input: string): void;
+    static isPtyActive(pid: number): boolean;
+    /**
+     * Resizes the pseudo-terminal (PTY) of a running process.
+     *
+     * @param pid The process ID of the target PTY.
+     * @param cols The new number of columns.
+     * @param rows The new number of rows.
+     */
+    static resizePty(pid: number, cols: number, rows: number): void;
+    /**
+     * Scrolls the pseudo-terminal (PTY) of a running process.
+     *
+     * @param pid The process ID of the target PTY.
+     * @param lines The number of lines to scroll.
+     */
+    static scrollPty(pid: number, lines: number): void;
 }