@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/agents/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../../src/agents/registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AACvE,OAAO,EAAU,MAAM,KAAK,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAGtD;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,UAAoC;IAEpC,OAAO,GAAG,UAAU,CAAC,IAAI,SAAS,CAAC;AACrC,CAAC;AAED;;;GAGG;AACH,MAAM,OAAO,aAAa;IAIK;IAH7B,8DAA8D;IAC7C,MAAM,GAAG,IAAI,GAAG,EAAgC,CAAC;IAElE,YAA6B,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAG,CAAC;IAE/C;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC;YAC/B,WAAW,CAAC,GAAG,CACb,oCAAoC,IAAI,CAAC,MAAM,CAAC,IAAI,UAAU,CAC/D,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC,+BAA+B,EAAE,CAAC;QAE3E,2DAA2D;QAC3D,IAAI,oBAAoB,EAAE,OAAO,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG;gBACf,GAAG,yBAAyB;gBAC5B,WAAW,EAAE;oBACX,GAAG,yBAAyB,CAAC,WAAW;oBACxC,KAAK,EACH,oBAAoB,CAAC,KAAK;wBAC1B,yBAAyB,CAAC,WAAW,CAAC,KAAK;oBAC7C,cAAc,EACZ,oBAAoB,CAAC,cAAc;wBACnC,yBAAyB,CAAC,WAAW,CAAC,cAAc;iBACvD;gBACD,SAAS,EAAE;oBACT,GAAG,yBAAyB,CAAC,SAAS;oBACtC,gBAAgB,EACd,oBAAoB,CAAC,cAAc;wBACnC,yBAAyB,CAAC,SAAS,CAAC,gBAAgB;oBACtD,SAAS,EACP,oBAAoB,CAAC,WAAW;wBAChC,yBAAyB,CAAC,SAAS,CAAC,SAAS;iBAChD;aACF,CAAC;YACF,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED;;;;OAIG;IACO,aAAa,CACrB,UAAoC;QAEpC,mBAAmB;QACnB,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;YAChD,WAAW,CAAC,IAAI,CACd,iFAAiF,CAClF,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC;YACnE,WAAW,CAAC,GAAG,CAAC,qCAAqC,UAAU,CAAC,IAAI,GAAG,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAE7C,yBAAyB;QACzB,oEAAoE;QACpE,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAE3C,MAAM,YAAY,GAAqB;YACrC,WAAW,EAAE;gBACX,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,qBAAqB,EAAE;oBACrB,WAAW,EAAE,WAAW,CAAC,IAAI;oBAC7B,IAAI,EAAE,WAAW,CAAC,KAAK;oBACvB,cAAc,EAAE;wBACd,eAAe,EAAE,IAAI;wBACrB,cAAc,EAAE,WAAW,CAAC,cAAc,IAAI,CAAC,CAAC;qBACjD;iBACF;aACF;SACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,0BAA0B,CACvD,mBAAmB,CAAC,UAAU,CAAC,EAC/B,YAAY,CACb,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,CAAC;CACF"}

package/dist/src/agents/registry.test.d.ts

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

package/dist/src/agents/registry.test.js

@@ -0,0 +1,160 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { AgentRegistry, getModelConfigAlias } from './registry.js';
+import { makeFakeConfig } from '../test-utils/config.js';
+import { debugLogger } from '../utils/debugLogger.js';
+// A test-only subclass to expose the protected `registerAgent` method.
+class TestableAgentRegistry extends AgentRegistry {
+    testRegisterAgent(definition) {
+        this.registerAgent(definition);
+    }
+}
+// Define mock agent structures for testing registration logic
+const MOCK_AGENT_V1 = {
+    name: 'MockAgent',
+    description: 'Mock Description V1',
+    inputConfig: { inputs: {} },
+    modelConfig: { model: 'test', temp: 0, top_p: 1 },
+    runConfig: { max_time_minutes: 1 },
+    promptConfig: { systemPrompt: 'test' },
+};
+const MOCK_AGENT_V2 = {
+    ...MOCK_AGENT_V1,
+    description: 'Mock Description V2 (Updated)',
+};
+describe('AgentRegistry', () => {
+    let mockConfig;
+    let registry;
+    beforeEach(() => {
+        // Default configuration (debugMode: false)
+        mockConfig = makeFakeConfig();
+        registry = new TestableAgentRegistry(mockConfig);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks(); // Restore spies after each test
+    });
+    describe('initialize', () => {
+        // TODO: Add this test once we actually have a built-in agent configured.
+        // it('should load built-in agents upon initialization', async () => {
+        //   expect(registry.getAllDefinitions()).toHaveLength(0);
+        //   await registry.initialize();
+        //   // There are currently no built-in agents.
+        //   expect(registry.getAllDefinitions()).toEqual([]);
+        // });
+        it('should log the count of loaded agents in debug mode', async () => {
+            const debugConfig = makeFakeConfig({ debugMode: true });
+            const debugRegistry = new TestableAgentRegistry(debugConfig);
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            await debugRegistry.initialize();
+            const agentCount = debugRegistry.getAllDefinitions().length;
+            expect(debugLogSpy).toHaveBeenCalledWith(`[AgentRegistry] Initialized with ${agentCount} agents.`);
+        });
+    });
+    describe('registration logic', () => {
+        it('should register a valid agent definition', () => {
+            registry.testRegisterAgent(MOCK_AGENT_V1);
+            expect(registry.getDefinition('MockAgent')).toEqual(MOCK_AGENT_V1);
+            expect(mockConfig.modelConfigService.getResolvedConfig({
+                model: getModelConfigAlias(MOCK_AGENT_V1),
+            })).toStrictEqual({
+                model: MOCK_AGENT_V1.modelConfig.model,
+                generateContentConfig: {
+                    temperature: MOCK_AGENT_V1.modelConfig.temp,
+                    topP: MOCK_AGENT_V1.modelConfig.top_p,
+                    thinkingConfig: {
+                        includeThoughts: true,
+                        thinkingBudget: -1,
+                    },
+                },
+            });
+        });
+        it('should handle special characters in agent names', () => {
+            const specialAgent = {
+                ...MOCK_AGENT_V1,
+                name: 'Agent-123_$pecial.v2',
+            };
+            registry.testRegisterAgent(specialAgent);
+            expect(registry.getDefinition('Agent-123_$pecial.v2')).toEqual(specialAgent);
+        });
+        it('should reject an agent definition missing a name', () => {
+            const invalidAgent = { ...MOCK_AGENT_V1, name: '' };
+            const debugWarnSpy = vi
+                .spyOn(debugLogger, 'warn')
+                .mockImplementation(() => { });
+            registry.testRegisterAgent(invalidAgent);
+            expect(registry.getDefinition('MockAgent')).toBeUndefined();
+            expect(debugWarnSpy).toHaveBeenCalledWith('[AgentRegistry] Skipping invalid agent definition. Missing name or description.');
+        });
+        it('should reject an agent definition missing a description', () => {
+            const invalidAgent = { ...MOCK_AGENT_V1, description: '' };
+            const debugWarnSpy = vi
+                .spyOn(debugLogger, 'warn')
+                .mockImplementation(() => { });
+            registry.testRegisterAgent(invalidAgent);
+            expect(registry.getDefinition('MockAgent')).toBeUndefined();
+            expect(debugWarnSpy).toHaveBeenCalledWith('[AgentRegistry] Skipping invalid agent definition. Missing name or description.');
+        });
+        it('should overwrite an existing agent definition', () => {
+            registry.testRegisterAgent(MOCK_AGENT_V1);
+            expect(registry.getDefinition('MockAgent')?.description).toBe('Mock Description V1');
+            registry.testRegisterAgent(MOCK_AGENT_V2);
+            expect(registry.getDefinition('MockAgent')?.description).toBe('Mock Description V2 (Updated)');
+            expect(registry.getAllDefinitions()).toHaveLength(1);
+        });
+        it('should log overwrites when in debug mode', () => {
+            const debugConfig = makeFakeConfig({ debugMode: true });
+            const debugRegistry = new TestableAgentRegistry(debugConfig);
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            debugRegistry.testRegisterAgent(MOCK_AGENT_V1);
+            debugRegistry.testRegisterAgent(MOCK_AGENT_V2);
+            expect(debugLogSpy).toHaveBeenCalledWith(`[AgentRegistry] Overriding agent 'MockAgent'`);
+        });
+        it('should not log overwrites when not in debug mode', () => {
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            registry.testRegisterAgent(MOCK_AGENT_V1);
+            registry.testRegisterAgent(MOCK_AGENT_V2);
+            expect(debugLogSpy).not.toHaveBeenCalledWith(`[AgentRegistry] Overriding agent 'MockAgent'`);
+        });
+        it('should handle bulk registrations correctly', async () => {
+            const promises = Array.from({ length: 100 }, (_, i) => Promise.resolve(registry.testRegisterAgent({
+                ...MOCK_AGENT_V1,
+                name: `Agent${i}`,
+            })));
+            await Promise.all(promises);
+            expect(registry.getAllDefinitions()).toHaveLength(100);
+        });
+    });
+    describe('accessors', () => {
+        const ANOTHER_AGENT = {
+            ...MOCK_AGENT_V1,
+            name: 'AnotherAgent',
+        };
+        beforeEach(() => {
+            registry.testRegisterAgent(MOCK_AGENT_V1);
+            registry.testRegisterAgent(ANOTHER_AGENT);
+        });
+        it('getDefinition should return the correct definition', () => {
+            expect(registry.getDefinition('MockAgent')).toEqual(MOCK_AGENT_V1);
+            expect(registry.getDefinition('AnotherAgent')).toEqual(ANOTHER_AGENT);
+        });
+        it('getDefinition should return undefined for unknown agents', () => {
+            expect(registry.getDefinition('NonExistentAgent')).toBeUndefined();
+        });
+        it('getAllDefinitions should return all registered definitions', () => {
+            const all = registry.getAllDefinitions();
+            expect(all).toHaveLength(2);
+            expect(all).toEqual(expect.arrayContaining([MOCK_AGENT_V1, ANOTHER_AGENT]));
+        });
+    });
+});
+//# sourceMappingURL=registry.test.js.map

package/dist/src/agents/registry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.test.js","sourceRoot":"","sources":["../../../src/agents/registry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGzD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,uEAAuE;AACvE,MAAM,qBAAsB,SAAQ,aAAa;IAC/C,iBAAiB,CAAC,UAA2B;QAC3C,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IACjC,CAAC;CACF;AAED,8DAA8D;AAC9D,MAAM,aAAa,GAAoB;IACrC,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,qBAAqB;IAClC,WAAW,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IAC3B,WAAW,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;IACjD,SAAS,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE;IAClC,YAAY,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE;CACvC,CAAC;AAEF,MAAM,aAAa,GAAoB;IACrC,GAAG,aAAa;IAChB,WAAW,EAAE,+BAA+B;CAC7C,CAAC;AAEF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,UAAkB,CAAC;IACvB,IAAI,QAA+B,CAAC;IAEpC,UAAU,CAAC,GAAG,EAAE;QACd,2CAA2C;QAC3C,UAAU,GAAG,cAAc,EAAE,CAAC;QAC9B,QAAQ,GAAG,IAAI,qBAAqB,CAAC,UAAU,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC,CAAC,gCAAgC;IACxD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,yEAAyE;QACzE,sEAAsE;QACtE,0DAA0D;QAE1D,iCAAiC;QAEjC,+CAA+C;QAC/C,sDAAsD;QACtD,MAAM;QAEN,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,WAAW,GAAG,cAAc,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACxD,MAAM,aAAa,GAAG,IAAI,qBAAqB,CAAC,WAAW,CAAC,CAAC;YAC7D,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,MAAM,aAAa,CAAC,UAAU,EAAE,CAAC;YAEjC,MAAM,UAAU,GAAG,aAAa,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC;YAC5D,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,oCAAoC,UAAU,UAAU,CACzD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YACnE,MAAM,CACJ,UAAU,CAAC,kBAAkB,CAAC,iBAAiB,CAAC;gBAC9C,KAAK,EAAE,mBAAmB,CAAC,aAAa,CAAC;aAC1C,CAAC,CACH,CAAC,aAAa,CAAC;gBACd,KAAK,EAAE,aAAa,CAAC,WAAW,CAAC,KAAK;gBACtC,qBAAqB,EAAE;oBACrB,WAAW,EAAE,aAAa,CAAC,WAAW,CAAC,IAAI;oBAC3C,IAAI,EAAE,aAAa,CAAC,WAAW,CAAC,KAAK;oBACrC,cAAc,EAAE;wBACd,eAAe,EAAE,IAAI;wBACrB,cAAc,EAAE,CAAC,CAAC;qBACnB;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,YAAY,GAAG;gBACnB,GAAG,aAAa;gBAChB,IAAI,EAAE,sBAAsB;aAC7B,CAAC;YACF,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACzC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAC,CAAC,OAAO,CAC5D,YAAY,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,YAAY,GAAG,EAAE,GAAG,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YACpD,MAAM,YAAY,GAAG,EAAE;iBACpB,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAEzC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YAC5D,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,iFAAiF,CAClF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,YAAY,GAAG,EAAE,GAAG,aAAa,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;YAC3D,MAAM,YAAY,GAAG,EAAE;iBACpB,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC;iBAC1B,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,QAAQ,CAAC,iBAAiB,CAAC,YAA+B,CAAC,CAAC;YAE5D,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YAC5D,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,iFAAiF,CAClF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,CAC3D,qBAAqB,CACtB,CAAC;YAEF,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,CAC3D,+BAA+B,CAChC,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,WAAW,GAAG,cAAc,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACxD,MAAM,aAAa,GAAG,IAAI,qBAAqB,CAAC,WAAW,CAAC,CAAC;YAC7D,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,aAAa,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC/C,aAAa,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAE/C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,8CAA8C,CAC/C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEhC,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1C,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAE1C,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAC1C,8CAA8C,CAC/C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACpD,OAAO,CAAC,OAAO,CACb,QAAQ,CAAC,iBAAiB,CAAC;gBACzB,GAAG,aAAa;gBAChB,IAAI,EAAE,QAAQ,CAAC,EAAE;aAClB,CAAC,CACH,CACF,CAAC;YAEF,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC5B,MAAM,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,MAAM,aAAa,GAAoB;YACrC,GAAG,aAAa;YAChB,IAAI,EAAE,cAAc;SACrB,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE;YACd,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC1C,QAAQ,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YACnE,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,GAAG,GAAG,QAAQ,CAAC,iBAAiB,EAAE,CAAC;YACzC,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CACjB,MAAM,CAAC,eAAe,CAAC,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/schema-utils.d.ts

@@ -0,0 +1,39 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { InputConfig } from './types.js';
+/**
+ * Defines the structure for a JSON Schema object, used for tool function
+ * declarations.
+ */
+interface JsonSchemaObject {
+    type: 'object';
+    properties: Record<string, JsonSchemaProperty>;
+    required?: string[];
+}
+/**
+ * Defines the structure for a property within a {@link JsonSchemaObject}.
+ */
+interface JsonSchemaProperty {
+    type: 'string' | 'number' | 'integer' | 'boolean' | 'array';
+    description: string;
+    items?: {
+        type: 'string' | 'number';
+    };
+}
+/**
+ * Converts an internal `InputConfig` definition into a standard JSON Schema
+ * object suitable for a tool's `FunctionDeclaration`.
+ *
+ * This utility ensures that the configuration for a subagent's inputs is
+ * correctly translated into the format expected by the generative model.
+ *
+ * @param inputConfig The internal `InputConfig` to convert.
+ * @returns A JSON Schema object representing the inputs.
+ * @throws An `Error` if an unsupported input type is encountered, ensuring
+ * configuration errors are caught early.
+ */
+export declare function convertInputConfigToJsonSchema(inputConfig: InputConfig): JsonSchemaObject;
+export {};

package/dist/src/agents/schema-utils.js

@@ -0,0 +1,57 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Converts an internal `InputConfig` definition into a standard JSON Schema
+ * object suitable for a tool's `FunctionDeclaration`.
+ *
+ * This utility ensures that the configuration for a subagent's inputs is
+ * correctly translated into the format expected by the generative model.
+ *
+ * @param inputConfig The internal `InputConfig` to convert.
+ * @returns A JSON Schema object representing the inputs.
+ * @throws An `Error` if an unsupported input type is encountered, ensuring
+ * configuration errors are caught early.
+ */
+export function convertInputConfigToJsonSchema(inputConfig) {
+    const properties = {};
+    const required = [];
+    for (const [name, definition] of Object.entries(inputConfig.inputs)) {
+        const schemaProperty = {
+            description: definition.description,
+        };
+        switch (definition.type) {
+            case 'string':
+            case 'number':
+            case 'integer':
+            case 'boolean':
+                schemaProperty.type = definition.type;
+                break;
+            case 'string[]':
+                schemaProperty.type = 'array';
+                schemaProperty.items = { type: 'string' };
+                break;
+            case 'number[]':
+                schemaProperty.type = 'array';
+                schemaProperty.items = { type: 'number' };
+                break;
+            default: {
+                const exhaustiveCheck = definition.type;
+                throw new Error(`Unsupported input type '${exhaustiveCheck}' for parameter '${name}'. ` +
+                    'Supported types: string, number, integer, boolean, string[], number[]');
+            }
+        }
+        properties[name] = schemaProperty;
+        if (definition.required) {
+            required.push(name);
+        }
+    }
+    return {
+        type: 'object',
+        properties,
+        required: required.length > 0 ? required : undefined,
+    };
+}
+//# sourceMappingURL=schema-utils.js.map

package/dist/src/agents/schema-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema-utils.js","sourceRoot":"","sources":["../../../src/agents/schema-utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAuBH;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,8BAA8B,CAC5C,WAAwB;IAExB,MAAM,UAAU,GAAuC,EAAE,CAAC;IAC1D,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;QACpE,MAAM,cAAc,GAAgC;YAClD,WAAW,EAAE,UAAU,CAAC,WAAW;SACpC,CAAC;QAEF,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;YACxB,KAAK,QAAQ,CAAC;YACd,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS,CAAC;YACf,KAAK,SAAS;gBACZ,cAAc,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;gBACtC,MAAM;YAER,KAAK,UAAU;gBACb,cAAc,CAAC,IAAI,GAAG,OAAO,CAAC;gBAC9B,cAAc,CAAC,KAAK,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAC1C,MAAM;YAER,KAAK,UAAU;gBACb,cAAc,CAAC,IAAI,GAAG,OAAO,CAAC;gBAC9B,cAAc,CAAC,KAAK,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAC1C,MAAM;YAER,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,eAAe,GAAU,UAAU,CAAC,IAAI,CAAC;gBAC/C,MAAM,IAAI,KAAK,CACb,2BAA2B,eAAe,oBAAoB,IAAI,KAAK;oBACrE,uEAAuE,CAC1E,CAAC;YACJ,CAAC;QACH,CAAC;QAED,UAAU,CAAC,IAAI,CAAC,GAAG,cAAoC,CAAC;QAExD,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;YACxB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,UAAU;QACV,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KACrD,CAAC;AACJ,CAAC"}

package/dist/src/agents/schema-utils.test.d.ts

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

package/dist/src/agents/schema-utils.test.js

@@ -0,0 +1,144 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { convertInputConfigToJsonSchema } from './schema-utils.js';
+const PRIMITIVE_TYPES_CONFIG = {
+    inputs: {
+        goal: {
+            type: 'string',
+            description: 'The primary objective',
+            required: true,
+        },
+        max_retries: {
+            type: 'integer',
+            description: 'Maximum number of retries',
+            required: false,
+        },
+        temperature: {
+            type: 'number',
+            description: 'The model temperature',
+            required: true,
+        },
+        verbose: {
+            type: 'boolean',
+            description: 'Enable verbose logging',
+            required: false,
+        },
+    },
+};
+const ARRAY_TYPES_CONFIG = {
+    inputs: {
+        filenames: {
+            type: 'string[]',
+            description: 'A list of file paths',
+            required: true,
+        },
+        scores: {
+            type: 'number[]',
+            description: 'A list of scores',
+            required: false,
+        },
+    },
+};
+const NO_REQUIRED_FIELDS_CONFIG = {
+    inputs: {
+        optional_param: {
+            type: 'string',
+            description: 'An optional parameter',
+            required: false,
+        },
+    },
+};
+const ALL_REQUIRED_FIELDS_CONFIG = {
+    inputs: {
+        paramA: { type: 'string', description: 'Parameter A', required: true },
+        paramB: { type: 'boolean', description: 'Parameter B', required: true },
+    },
+};
+const EMPTY_CONFIG = {
+    inputs: {},
+};
+const UNSUPPORTED_TYPE_CONFIG = {
+    inputs: {
+        invalid_param: {
+            // @ts-expect-error - Intentionally testing an invalid type
+            type: 'date',
+            description: 'This type is not supported',
+            required: true,
+        },
+    },
+};
+describe('convertInputConfigToJsonSchema', () => {
+    describe('type conversion', () => {
+        it('should correctly convert an InputConfig with various primitive types', () => {
+            const result = convertInputConfigToJsonSchema(PRIMITIVE_TYPES_CONFIG);
+            expect(result).toEqual({
+                type: 'object',
+                properties: {
+                    goal: { type: 'string', description: 'The primary objective' },
+                    max_retries: {
+                        type: 'integer',
+                        description: 'Maximum number of retries',
+                    },
+                    temperature: { type: 'number', description: 'The model temperature' },
+                    verbose: { type: 'boolean', description: 'Enable verbose logging' },
+                },
+                required: ['goal', 'temperature'],
+            });
+        });
+        it('should correctly handle array types for strings and numbers', () => {
+            const result = convertInputConfigToJsonSchema(ARRAY_TYPES_CONFIG);
+            expect(result).toEqual({
+                type: 'object',
+                properties: {
+                    filenames: {
+                        type: 'array',
+                        description: 'A list of file paths',
+                        items: { type: 'string' },
+                    },
+                    scores: {
+                        type: 'array',
+                        description: 'A list of scores',
+                        items: { type: 'number' },
+                    },
+                },
+                required: ['filenames'],
+            });
+        });
+    });
+    describe('required field handling', () => {
+        it('should produce an undefined `required` field when no inputs are required', () => {
+            const result = convertInputConfigToJsonSchema(NO_REQUIRED_FIELDS_CONFIG);
+            expect(result.properties['optional_param']).toBeDefined();
+            // Per the implementation and JSON Schema spec, the `required` field
+            // should be omitted if no properties are required.
+            expect(result.required).toBeUndefined();
+        });
+        it('should list all properties in `required` when all are marked as required', () => {
+            const result = convertInputConfigToJsonSchema(ALL_REQUIRED_FIELDS_CONFIG);
+            expect(result.required).toHaveLength(2);
+            expect(result.required).toEqual(expect.arrayContaining(['paramA', 'paramB']));
+        });
+    });
+    describe('edge cases', () => {
+        it('should return a valid, empty schema for an empty input config', () => {
+            const result = convertInputConfigToJsonSchema(EMPTY_CONFIG);
+            expect(result).toEqual({
+                type: 'object',
+                properties: {},
+                required: undefined,
+            });
+        });
+    });
+    describe('error handling', () => {
+        it('should throw an informative error for an unsupported input type', () => {
+            const action = () => convertInputConfigToJsonSchema(UNSUPPORTED_TYPE_CONFIG);
+            expect(action).toThrow(/Unsupported input type 'date'/);
+            expect(action).toThrow(/parameter 'invalid_param'/);
+        });
+    });
+});
+//# sourceMappingURL=schema-utils.test.js.map

package/dist/src/agents/schema-utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema-utils.test.js","sourceRoot":"","sources":["../../../src/agents/schema-utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,8BAA8B,EAAE,MAAM,mBAAmB,CAAC;AAGnE,MAAM,sBAAsB,GAAgB;IAC1C,MAAM,EAAE;QACN,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,uBAAuB;YACpC,QAAQ,EAAE,IAAI;SACf;QACD,WAAW,EAAE;YACX,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,2BAA2B;YACxC,QAAQ,EAAE,KAAK;SAChB;QACD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,uBAAuB;YACpC,QAAQ,EAAE,IAAI;SACf;QACD,OAAO,EAAE;YACP,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,wBAAwB;YACrC,QAAQ,EAAE,KAAK;SAChB;KACF;CACF,CAAC;AAEF,MAAM,kBAAkB,GAAgB;IACtC,MAAM,EAAE;QACN,SAAS,EAAE;YACT,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,sBAAsB;YACnC,QAAQ,EAAE,IAAI;SACf;QACD,MAAM,EAAE;YACN,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,kBAAkB;YAC/B,QAAQ,EAAE,KAAK;SAChB;KACF;CACF,CAAC;AAEF,MAAM,yBAAyB,GAAgB;IAC7C,MAAM,EAAE;QACN,cAAc,EAAE;YACd,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,uBAAuB;YACpC,QAAQ,EAAE,KAAK;SAChB;KACF;CACF,CAAC;AAEF,MAAM,0BAA0B,GAAgB;IAC9C,MAAM,EAAE;QACN,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE;QACtE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE;KACxE;CACF,CAAC;AAEF,MAAM,YAAY,GAAgB;IAChC,MAAM,EAAE,EAAE;CACX,CAAC;AAEF,MAAM,uBAAuB,GAAgB;IAC3C,MAAM,EAAE;QACN,aAAa,EAAE;YACb,2DAA2D;YAC3D,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,4BAA4B;YACzC,QAAQ,EAAE,IAAI;SACf;KACF;CACF,CAAC;AAEF,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,MAAM,MAAM,GAAG,8BAA8B,CAAC,sBAAsB,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,uBAAuB,EAAE;oBAC9D,WAAW,EAAE;wBACX,IAAI,EAAE,SAAS;wBACf,WAAW,EAAE,2BAA2B;qBACzC;oBACD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,uBAAuB,EAAE;oBACrE,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,wBAAwB,EAAE;iBACpE;gBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,aAAa,CAAC;aAClC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,MAAM,GAAG,8BAA8B,CAAC,kBAAkB,CAAC,CAAC;YAElE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,SAAS,EAAE;wBACT,IAAI,EAAE,OAAO;wBACb,WAAW,EAAE,sBAAsB;wBACnC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC1B;oBACD,MAAM,EAAE;wBACN,IAAI,EAAE,OAAO;wBACb,WAAW,EAAE,kBAAkB;wBAC/B,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC1B;iBACF;gBACD,QAAQ,EAAE,CAAC,WAAW,CAAC;aACxB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,MAAM,GAAG,8BAA8B,CAAC,yBAAyB,CAAC,CAAC;YAEzE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAC1D,oEAAoE;YACpE,mDAAmD;YACnD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,MAAM,GAAG,8BAA8B,CAAC,0BAA0B,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAC7B,MAAM,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAC7C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,MAAM,GAAG,8BAA8B,CAAC,YAAY,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;gBACd,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,MAAM,GAAG,GAAG,EAAE,CAClB,8BAA8B,CAAC,uBAAuB,CAAC,CAAC;YAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/subagent-tool-wrapper.d.ts

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseDeclarativeTool, type ToolInvocation, type ToolResult } from '../tools/tools.js';
+import type { Config } from '../config/config.js';
+import type { AgentDefinition, AgentInputs } from './types.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+/**
+ * A tool wrapper that dynamically exposes a subagent as a standard,
+ * strongly-typed `DeclarativeTool`.
+ */
+export declare class SubagentToolWrapper extends BaseDeclarativeTool<AgentInputs, ToolResult> {
+    private readonly definition;
+    private readonly config;
+    /**
+     * Constructs the tool wrapper.
+     *
+     * The constructor dynamically generates the JSON schema for the tool's
+     * parameters based on the subagent's input configuration.
+     *
+     * @param definition The `AgentDefinition` of the subagent to wrap.
+     * @param config The runtime configuration, passed down to the subagent.
+     * @param messageBus Optional message bus for policy enforcement.
+     */
+    constructor(definition: AgentDefinition, config: Config, messageBus?: MessageBus);
+    /**
+     * Creates an invocation instance for executing the subagent.
+     *
+     * This method is called by the tool framework when the parent agent decides
+     * to use this tool.
+     *
+     * @param params The validated input parameters from the parent agent's call.
+     * @returns A `ToolInvocation` instance ready for execution.
+     */
+    protected createInvocation(params: AgentInputs): ToolInvocation<AgentInputs, ToolResult>;
+}

package/dist/src/agents/subagent-tool-wrapper.js

@@ -0,0 +1,48 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseDeclarativeTool, Kind, } from '../tools/tools.js';
+import { convertInputConfigToJsonSchema } from './schema-utils.js';
+import { SubagentInvocation } from './invocation.js';
+/**
+ * A tool wrapper that dynamically exposes a subagent as a standard,
+ * strongly-typed `DeclarativeTool`.
+ */
+export class SubagentToolWrapper extends BaseDeclarativeTool {
+    definition;
+    config;
+    /**
+     * Constructs the tool wrapper.
+     *
+     * The constructor dynamically generates the JSON schema for the tool's
+     * parameters based on the subagent's input configuration.
+     *
+     * @param definition The `AgentDefinition` of the subagent to wrap.
+     * @param config The runtime configuration, passed down to the subagent.
+     * @param messageBus Optional message bus for policy enforcement.
+     */
+    constructor(definition, config, messageBus) {
+        // Dynamically generate the JSON schema required for the tool definition.
+        const parameterSchema = convertInputConfigToJsonSchema(definition.inputConfig);
+        super(definition.name, definition.displayName ?? definition.name, definition.description, Kind.Think, parameterSchema, 
+        /* isOutputMarkdown */ true, 
+        /* canUpdateOutput */ true, messageBus);
+        this.definition = definition;
+        this.config = config;
+    }
+    /**
+     * Creates an invocation instance for executing the subagent.
+     *
+     * This method is called by the tool framework when the parent agent decides
+     * to use this tool.
+     *
+     * @param params The validated input parameters from the parent agent's call.
+     * @returns A `ToolInvocation` instance ready for execution.
+     */
+    createInvocation(params) {
+        return new SubagentInvocation(params, this.definition, this.config, this.messageBus);
+    }
+}
+//# sourceMappingURL=subagent-tool-wrapper.js.map

package/dist/src/agents/subagent-tool-wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"subagent-tool-wrapper.js","sourceRoot":"","sources":["../../../src/agents/subagent-tool-wrapper.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,mBAAmB,EACnB,IAAI,GAGL,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EAAE,8BAA8B,EAAE,MAAM,mBAAmB,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAGrD;;;GAGG;AACH,MAAM,OAAO,mBAAoB,SAAQ,mBAGxC;IAYoB;IACA;IAZnB;;;;;;;;;OASG;IACH,YACmB,UAA2B,EAC3B,MAAc,EAC/B,UAAuB;QAEvB,yEAAyE;QACzE,MAAM,eAAe,GAAG,8BAA8B,CACpD,UAAU,CAAC,WAAW,CACvB,CAAC;QAEF,KAAK,CACH,UAAU,CAAC,IAAI,EACf,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,IAAI,EACzC,UAAU,CAAC,WAAW,EACtB,IAAI,CAAC,KAAK,EACV,eAAe;QACf,sBAAsB,CAAC,IAAI;QAC3B,qBAAqB,CAAC,IAAI,EAC1B,UAAU,CACX,CAAC;QAlBe,eAAU,GAAV,UAAU,CAAiB;QAC3B,WAAM,GAAN,MAAM,CAAQ;IAkBjC,CAAC;IAED;;;;;;;;OAQG;IACO,gBAAgB,CACxB,MAAmB;QAEnB,OAAO,IAAI,kBAAkB,CAC3B,MAAM,EACN,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,UAAU,CAChB,CAAC;IACJ,CAAC;CACF"}

package/dist/src/agents/subagent-tool-wrapper.test.d.ts

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

package/dist/src/agents/subagent-tool-wrapper.test.js

@@ -0,0 +1,110 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { SubagentToolWrapper } from './subagent-tool-wrapper.js';
+import { SubagentInvocation } from './invocation.js';
+import { convertInputConfigToJsonSchema } from './schema-utils.js';
+import { makeFakeConfig } from '../test-utils/config.js';
+import { Kind } from '../tools/tools.js';
+// Mock dependencies to isolate the SubagentToolWrapper class
+vi.mock('./invocation.js');
+vi.mock('./schema-utils.js');
+const MockedSubagentInvocation = vi.mocked(SubagentInvocation);
+const mockConvertInputConfigToJsonSchema = vi.mocked(convertInputConfigToJsonSchema);
+// Define reusable test data
+let mockConfig;
+const mockDefinition = {
+    name: 'TestAgent',
+    displayName: 'Test Agent Display Name',
+    description: 'An agent for testing.',
+    inputConfig: {
+        inputs: {
+            goal: { type: 'string', required: true, description: 'The goal.' },
+            priority: {
+                type: 'number',
+                required: false,
+                description: 'The priority.',
+            },
+        },
+    },
+    modelConfig: { model: 'gemini-test-model', temp: 0, top_p: 1 },
+    runConfig: { max_time_minutes: 5 },
+    promptConfig: { systemPrompt: 'You are a test agent.' },
+};
+const mockSchema = {
+    type: 'object',
+    properties: {
+        goal: { type: 'string', description: 'The goal.' },
+        priority: { type: 'number', description: 'The priority.' },
+    },
+    required: ['goal'],
+};
+describe('SubagentToolWrapper', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockConfig = makeFakeConfig();
+        // Provide a mock implementation for the schema conversion utility
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        mockConvertInputConfigToJsonSchema.mockReturnValue(mockSchema);
+    });
+    describe('constructor', () => {
+        it('should call convertInputConfigToJsonSchema with the correct agent inputConfig', () => {
+            new SubagentToolWrapper(mockDefinition, mockConfig);
+            expect(convertInputConfigToJsonSchema).toHaveBeenCalledExactlyOnceWith(mockDefinition.inputConfig);
+        });
+        it('should correctly configure the tool properties from the agent definition', () => {
+            const wrapper = new SubagentToolWrapper(mockDefinition, mockConfig);
+            expect(wrapper.name).toBe(mockDefinition.name);
+            expect(wrapper.displayName).toBe(mockDefinition.displayName);
+            expect(wrapper.description).toBe(mockDefinition.description);
+            expect(wrapper.kind).toBe(Kind.Think);
+            expect(wrapper.isOutputMarkdown).toBe(true);
+            expect(wrapper.canUpdateOutput).toBe(true);
+        });
+        it('should fall back to the agent name for displayName if it is not provided', () => {
+            const definitionWithoutDisplayName = {
+                ...mockDefinition,
+                displayName: undefined,
+            };
+            const wrapper = new SubagentToolWrapper(definitionWithoutDisplayName, mockConfig);
+            expect(wrapper.displayName).toBe(definitionWithoutDisplayName.name);
+        });
+        it('should generate a valid tool schema using the definition and converted schema', () => {
+            const wrapper = new SubagentToolWrapper(mockDefinition, mockConfig);
+            const schema = wrapper.schema;
+            expect(schema.name).toBe(mockDefinition.name);
+            expect(schema.description).toBe(mockDefinition.description);
+            expect(schema.parametersJsonSchema).toEqual(mockSchema);
+        });
+    });
+    describe('createInvocation', () => {
+        it('should create a SubagentInvocation with the correct parameters', () => {
+            const wrapper = new SubagentToolWrapper(mockDefinition, mockConfig);
+            const params = { goal: 'Test the invocation', priority: 1 };
+            // The public `build` method calls the protected `createInvocation` after validation
+            const invocation = wrapper.build(params);
+            expect(invocation).toBeInstanceOf(SubagentInvocation);
+            expect(MockedSubagentInvocation).toHaveBeenCalledExactlyOnceWith(params, mockDefinition, mockConfig, undefined);
+        });
+        it('should pass the messageBus to the SubagentInvocation constructor', () => {
+            const mockMessageBus = {};
+            const wrapper = new SubagentToolWrapper(mockDefinition, mockConfig, mockMessageBus);
+            const params = { goal: 'Test the invocation', priority: 1 };
+            wrapper.build(params);
+            expect(MockedSubagentInvocation).toHaveBeenCalledWith(params, mockDefinition, mockConfig, mockMessageBus);
+        });
+        it('should throw a validation error for invalid parameters before creating an invocation', () => {
+            const wrapper = new SubagentToolWrapper(mockDefinition, mockConfig);
+            // Missing the required 'goal' parameter
+            const invalidParams = { priority: 1 };
+            // The `build` method in the base class performs JSON schema validation
+            // before calling the protected `createInvocation` method.
+            expect(() => wrapper.build(invalidParams)).toThrow("params must have required property 'goal'");
+            expect(MockedSubagentInvocation).not.toHaveBeenCalled();
+        });
+    });
+});
+//# sourceMappingURL=subagent-tool-wrapper.test.js.map