@office-ai/aioncli-core 0.2.2 → 0.8.1

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

@@ -0,0 +1,206 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { SubagentInvocation } from './invocation.js';
+import { AgentExecutor } from './executor.js';
+import { AgentTerminateMode } from './types.js';
+import { makeFakeConfig } from '../test-utils/config.js';
+import { ToolErrorType } from '../tools/tool-error.js';
+vi.mock('./executor.js');
+const MockAgentExecutor = vi.mocked(AgentExecutor);
+let mockConfig;
+const testDefinition = {
+    name: 'MockAgent',
+    description: 'A mock agent.',
+    inputConfig: {
+        inputs: {
+            task: { type: 'string', required: true, description: 'task' },
+            priority: { type: 'number', required: false, description: 'prio' },
+        },
+    },
+    modelConfig: { model: 'test', temp: 0, top_p: 1 },
+    runConfig: { max_time_minutes: 1 },
+    promptConfig: { systemPrompt: 'test' },
+};
+describe('SubagentInvocation', () => {
+    let mockExecutorInstance;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockConfig = makeFakeConfig();
+        mockExecutorInstance = {
+            run: vi.fn(),
+            definition: testDefinition,
+        };
+        MockAgentExecutor.create.mockResolvedValue(mockExecutorInstance);
+    });
+    describe('getDescription', () => {
+        it('should format the description with inputs', () => {
+            const params = { task: 'Analyze data', priority: 5 };
+            const invocation = new SubagentInvocation(params, testDefinition, mockConfig);
+            const description = invocation.getDescription();
+            expect(description).toBe("Running subagent 'MockAgent' with inputs: { task: Analyze data, priority: 5 }");
+        });
+        it('should truncate long input values', () => {
+            const longTask = 'A'.repeat(100);
+            const params = { task: longTask };
+            const invocation = new SubagentInvocation(params, testDefinition, mockConfig);
+            const description = invocation.getDescription();
+            // Default INPUT_PREVIEW_MAX_LENGTH is 50
+            expect(description).toBe(`Running subagent 'MockAgent' with inputs: { task: ${'A'.repeat(50)} }`);
+        });
+        it('should truncate the overall description if it exceeds the limit', () => {
+            // Create a definition and inputs that result in a very long description
+            const longNameDef = {
+                ...testDefinition,
+                name: 'VeryLongAgentNameThatTakesUpSpace',
+            };
+            const params = {};
+            for (let i = 0; i < 20; i++) {
+                params[`input${i}`] = `value${i}`;
+            }
+            const invocation = new SubagentInvocation(params, longNameDef, mockConfig);
+            const description = invocation.getDescription();
+            // Default DESCRIPTION_MAX_LENGTH is 200
+            expect(description.length).toBe(200);
+            expect(description.startsWith("Running subagent 'VeryLongAgentNameThatTakesUpSpace'")).toBe(true);
+        });
+    });
+    describe('execute', () => {
+        let signal;
+        let updateOutput;
+        const params = { task: 'Execute task' };
+        let invocation;
+        beforeEach(() => {
+            signal = new AbortController().signal;
+            updateOutput = vi.fn();
+            invocation = new SubagentInvocation(params, testDefinition, mockConfig);
+        });
+        it('should initialize and run the executor successfully', async () => {
+            const mockOutput = {
+                result: 'Analysis complete.',
+                terminate_reason: AgentTerminateMode.GOAL,
+            };
+            mockExecutorInstance.run.mockResolvedValue(mockOutput);
+            const result = await invocation.execute(signal, updateOutput);
+            expect(MockAgentExecutor.create).toHaveBeenCalledWith(testDefinition, mockConfig, expect.any(Function));
+            expect(updateOutput).toHaveBeenCalledWith('Subagent starting...\n');
+            expect(mockExecutorInstance.run).toHaveBeenCalledWith(params, signal);
+            expect(result.llmContent).toEqual([
+                {
+                    text: expect.stringContaining("Subagent 'MockAgent' finished.\nTermination Reason: GOAL\nResult:\nAnalysis complete."),
+                },
+            ]);
+            expect(result.returnDisplay).toContain('Result:\nAnalysis complete.');
+            expect(result.returnDisplay).toContain('Termination Reason:\n GOAL');
+        });
+        it('should stream THOUGHT_CHUNK activities from the executor', async () => {
+            mockExecutorInstance.run.mockImplementation(async () => {
+                const onActivity = MockAgentExecutor.create.mock.calls[0][2];
+                if (onActivity) {
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'THOUGHT_CHUNK',
+                        data: { text: 'Analyzing...' },
+                    });
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'THOUGHT_CHUNK',
+                        data: { text: ' Still thinking.' },
+                    });
+                }
+                return { result: 'Done', terminate_reason: AgentTerminateMode.GOAL };
+            });
+            await invocation.execute(signal, updateOutput);
+            expect(updateOutput).toHaveBeenCalledWith('Subagent starting...\n');
+            expect(updateOutput).toHaveBeenCalledWith('🤖💭 Analyzing...');
+            expect(updateOutput).toHaveBeenCalledWith('🤖💭  Still thinking.');
+            expect(updateOutput).toHaveBeenCalledTimes(3); // Initial message + 2 thoughts
+        });
+        it('should NOT stream other activities (e.g., TOOL_CALL_START, ERROR)', async () => {
+            mockExecutorInstance.run.mockImplementation(async () => {
+                const onActivity = MockAgentExecutor.create.mock.calls[0][2];
+                if (onActivity) {
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'TOOL_CALL_START',
+                        data: { name: 'ls' },
+                    });
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'ERROR',
+                        data: { error: 'Failed' },
+                    });
+                }
+                return { result: 'Done', terminate_reason: AgentTerminateMode.GOAL };
+            });
+            await invocation.execute(signal, updateOutput);
+            // Should only contain the initial "Subagent starting..." message
+            expect(updateOutput).toHaveBeenCalledTimes(1);
+            expect(updateOutput).toHaveBeenCalledWith('Subagent starting...\n');
+        });
+        it('should run successfully without an updateOutput callback', async () => {
+            mockExecutorInstance.run.mockImplementation(async () => {
+                const onActivity = MockAgentExecutor.create.mock.calls[0][2];
+                if (onActivity) {
+                    // Ensure calling activity doesn't crash when updateOutput is undefined
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'testAgent',
+                        type: 'THOUGHT_CHUNK',
+                        data: { text: 'Thinking silently.' },
+                    });
+                }
+                return { result: 'Done', terminate_reason: AgentTerminateMode.GOAL };
+            });
+            // Execute without the optional callback
+            const result = await invocation.execute(signal);
+            expect(result.error).toBeUndefined();
+            expect(result.returnDisplay).toContain('Result:\nDone');
+        });
+        it('should handle executor run failure', async () => {
+            const error = new Error('Model failed during execution.');
+            mockExecutorInstance.run.mockRejectedValue(error);
+            const result = await invocation.execute(signal, updateOutput);
+            expect(result.error).toEqual({
+                message: error.message,
+                type: ToolErrorType.EXECUTION_FAILED,
+            });
+            expect(result.returnDisplay).toBe(`Subagent Failed: MockAgent\nError: ${error.message}`);
+            expect(result.llmContent).toBe(`Subagent 'MockAgent' failed. Error: ${error.message}`);
+        });
+        it('should handle executor creation failure', async () => {
+            const creationError = new Error('Failed to initialize tools.');
+            MockAgentExecutor.create.mockRejectedValue(creationError);
+            const result = await invocation.execute(signal, updateOutput);
+            expect(mockExecutorInstance.run).not.toHaveBeenCalled();
+            expect(result.error).toEqual({
+                message: creationError.message,
+                type: ToolErrorType.EXECUTION_FAILED,
+            });
+            expect(result.returnDisplay).toContain(`Error: ${creationError.message}`);
+        });
+        /**
+         * This test verifies that the AbortSignal is correctly propagated and
+         * that a rejection from the executor due to abortion is handled gracefully.
+         */
+        it('should handle abortion signal during execution', async () => {
+            const abortError = new Error('Aborted');
+            mockExecutorInstance.run.mockRejectedValue(abortError);
+            const controller = new AbortController();
+            const executePromise = invocation.execute(controller.signal, updateOutput);
+            controller.abort();
+            const result = await executePromise;
+            expect(mockExecutorInstance.run).toHaveBeenCalledWith(params, controller.signal);
+            expect(result.error?.message).toBe('Aborted');
+            expect(result.error?.type).toBe(ToolErrorType.EXECUTION_FAILED);
+        });
+    });
+});
+//# sourceMappingURL=invocation.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"invocation.test.js","sourceRoot":"","sources":["../../../src/agents/invocation.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAe,MAAM,QAAQ,CAAC;AAC3E,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAM9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAGvD,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAEzB,MAAM,iBAAiB,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;AAEnD,IAAI,UAAkB,CAAC;AAEvB,MAAM,cAAc,GAAoB;IACtC,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,eAAe;IAC5B,WAAW,EAAE;QACX,MAAM,EAAE;YACN,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE;YAC7D,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE;SACnE;KACF;IACD,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,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,oBAA2C,CAAC;IAEhD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,GAAG,cAAc,EAAE,CAAC;QAE9B,oBAAoB,GAAG;YACrB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YACZ,UAAU,EAAE,cAAc;SACS,CAAC;QAEtC,iBAAiB,CAAC,MAAM,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,IAAI,kBAAkB,CACvC,MAAM,EACN,cAAc,EACd,UAAU,CACX,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CACtB,+EAA+E,CAChF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,IAAI,kBAAkB,CACvC,MAAM,EACN,cAAc,EACd,UAAU,CACX,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,yCAAyC;YACzC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CACtB,qDAAqD,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CACxE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,wEAAwE;YACxE,MAAM,WAAW,GAAG;gBAClB,GAAG,cAAc;gBACjB,IAAI,EAAE,mCAAmC;aAC1C,CAAC;YACF,MAAM,MAAM,GAAgB,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC;YACpC,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,kBAAkB,CACvC,MAAM,EACN,WAAW,EACX,UAAU,CACX,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,wCAAwC;YACxC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,CACJ,WAAW,CAAC,UAAU,CACpB,sDAAsD,CACvD,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,IAAI,MAAmB,CAAC;QACxB,IAAI,YAAsC,CAAC;QAC3C,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACxC,IAAI,UAA8B,CAAC;QAEnC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YACtC,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACvB,UAAU,GAAG,IAAI,kBAAkB,CAAC,MAAM,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,oBAAoB;gBAC5B,gBAAgB,EAAE,kBAAkB,CAAC,IAAI;aAC1C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACnD,cAAc,EACd,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;YAEpE,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBAChC;oBACE,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAC3B,uFAAuF,CACxF;iBACF;aACF,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE7D,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;qBACN,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;qBACV,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;YACpE,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,CAAC;YAC/D,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,uBAAuB,CAAC,CAAC;YACnE,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,+BAA+B;QAChF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE7D,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,iBAAiB;wBACvB,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE;qBACI,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;qBACD,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,iEAAiE;YACjE,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC7D,IAAI,UAAU,EAAE,CAAC;oBACf,uEAAuE;oBACvE,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE;qBACZ,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,wCAAwC;YACxC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAC1D,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBAC3B,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,IAAI,EAAE,aAAa,CAAC,gBAAgB;aACrC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAC/B,sCAAsC,KAAK,CAAC,OAAO,EAAE,CACtD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAC5B,uCAAuC,KAAK,CAAC,OAAO,EAAE,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAC/D,iBAAiB,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBAC3B,OAAO,EAAE,aAAa,CAAC,OAAO;gBAC9B,IAAI,EAAE,aAAa,CAAC,gBAAgB;aACrC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,UAAU,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH;;;WAGG;QACH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;YACxC,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,cAAc,GAAG,UAAU,CAAC,OAAO,CACvC,UAAU,CAAC,MAAM,EACjB,YAAY,CACb,CAAC;YACF,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;YAEpC,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACnD,MAAM,EACN,UAAU,CAAC,MAAM,CAClB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { AgentDefinition } from './types.js';
+/**
+ * Manages the discovery, loading, validation, and registration of
+ * AgentDefinitions.
+ */
+export declare class AgentRegistry {
+    private readonly config;
+    private readonly agents;
+    constructor(config: Config);
+    /**
+     * Discovers and loads agents.
+     */
+    initialize(): Promise<void>;
+    private loadBuiltInAgents;
+    /**
+     * Registers an agent definition. If an agent with the same name exists,
+     * it will be overwritten, respecting the precedence established by the
+     * initialization order.
+     */
+    protected registerAgent(definition: AgentDefinition): void;
+    /**
+     * Retrieves an agent definition by name.
+     */
+    getDefinition(name: string): AgentDefinition | undefined;
+    /**
+     * Returns all active agent definitions.
+     */
+    getAllDefinitions(): AgentDefinition[];
+}

package/dist/src/agents/registry.js

@@ -0,0 +1,58 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CodebaseInvestigatorAgent } from './codebase-investigator.js';
+/**
+ * Manages the discovery, loading, validation, and registration of
+ * AgentDefinitions.
+ */
+export class AgentRegistry {
+    config;
+    agents = new Map();
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Discovers and loads agents.
+     */
+    async initialize() {
+        this.loadBuiltInAgents();
+        if (this.config.getDebugMode()) {
+            console.log(`[AgentRegistry] Initialized with ${this.agents.size} agents.`);
+        }
+    }
+    loadBuiltInAgents() {
+        this.registerAgent(CodebaseInvestigatorAgent);
+    }
+    /**
+     * Registers an agent definition. If an agent with the same name exists,
+     * it will be overwritten, respecting the precedence established by the
+     * initialization order.
+     */
+    registerAgent(definition) {
+        // Basic validation
+        if (!definition.name || !definition.description) {
+            console.warn(`[AgentRegistry] Skipping invalid agent definition. Missing name or description.`);
+            return;
+        }
+        if (this.agents.has(definition.name) && this.config.getDebugMode()) {
+            console.log(`[AgentRegistry] Overriding agent '${definition.name}'`);
+        }
+        this.agents.set(definition.name, definition);
+    }
+    /**
+     * Retrieves an agent definition by name.
+     */
+    getDefinition(name) {
+        return this.agents.get(name);
+    }
+    /**
+     * Returns all active agent definitions.
+     */
+    getAllDefinitions() {
+        return Array.from(this.agents.values());
+    }
+}
+//# sourceMappingURL=registry.js.map

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;AAEvE;;;GAGG;AACH,MAAM,OAAO,aAAa;IAGK;IAFZ,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;IAE7D,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,OAAO,CAAC,GAAG,CACT,oCAAoC,IAAI,CAAC,MAAM,CAAC,IAAI,UAAU,CAC/D,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,aAAa,CAAC,yBAAyB,CAAC,CAAC;IAChD,CAAC;IAED;;;;OAIG;IACO,aAAa,CAAC,UAA2B;QACjD,mBAAmB;QACnB,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;YAChD,OAAO,CAAC,IAAI,CACV,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,OAAO,CAAC,GAAG,CAAC,qCAAqC,UAAU,CAAC,IAAI,GAAG,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAC/C,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.js

@@ -0,0 +1,146 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { AgentRegistry } from './registry.js';
+import { makeFakeConfig } from '../test-utils/config.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 consoleLogSpy = vi
+                .spyOn(console, 'log')
+                .mockImplementation(() => { });
+            await debugRegistry.initialize();
+            const agentCount = debugRegistry.getAllDefinitions().length;
+            expect(consoleLogSpy).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);
+        });
+        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 consoleWarnSpy = vi
+                .spyOn(console, 'warn')
+                .mockImplementation(() => { });
+            registry.testRegisterAgent(invalidAgent);
+            expect(registry.getDefinition('MockAgent')).toBeUndefined();
+            expect(consoleWarnSpy).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 consoleWarnSpy = vi
+                .spyOn(console, 'warn')
+                .mockImplementation(() => { });
+            registry.testRegisterAgent(invalidAgent);
+            expect(registry.getDefinition('MockAgent')).toBeUndefined();
+            expect(consoleWarnSpy).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 consoleLogSpy = vi
+                .spyOn(console, 'log')
+                .mockImplementation(() => { });
+            debugRegistry.testRegisterAgent(MOCK_AGENT_V1);
+            debugRegistry.testRegisterAgent(MOCK_AGENT_V2);
+            expect(consoleLogSpy).toHaveBeenCalledWith(`[AgentRegistry] Overriding agent 'MockAgent'`);
+        });
+        it('should not log overwrites when not in debug mode', () => {
+            const consoleLogSpy = vi
+                .spyOn(console, 'log')
+                .mockImplementation(() => { });
+            registry.testRegisterAgent(MOCK_AGENT_V1);
+            registry.testRegisterAgent(MOCK_AGENT_V2);
+            expect(consoleLogSpy).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,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAIzD,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,aAAa,GAAG,EAAE;iBACrB,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;iBACrB,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,aAAa,CAAC,CAAC,oBAAoB,CACxC,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;QACrE,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,cAAc,GAAG,EAAE;iBACtB,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC;iBACtB,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,cAAc,CAAC,CAAC,oBAAoB,CACzC,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,cAAc,GAAG,EAAE;iBACtB,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC;iBACtB,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,cAAc,CAAC,CAAC,oBAAoB,CACzC,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,aAAa,GAAG,EAAE;iBACrB,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;iBACrB,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,aAAa,CAAC,CAAC,oBAAoB,CACxC,8CAA8C,CAC/C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,aAAa,GAAG,EAAE;iBACrB,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;iBACrB,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,aAAa,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAC5C,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 {};