@office-ai/aioncli-core 0.18.7 → 0.24.0-preview.1

package/dist/src/agents/a2a-client-manager.test.js

@@ -0,0 +1,237 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { A2AClientManager, createAdapterFetch, } from './a2a-client-manager.js';
+import { ClientFactory, DefaultAgentCardResolver } from '@a2a-js/sdk/client';
+import { debugLogger } from '../utils/debugLogger.js';
+import { createAuthenticatingFetchWithRetry, ClientFactoryOptions, } from '@a2a-js/sdk/client';
+vi.mock('../utils/debugLogger.js', () => ({
+    debugLogger: {
+        debug: vi.fn(),
+    },
+}));
+vi.mock('@a2a-js/sdk/client', () => {
+    const ClientFactory = vi.fn();
+    const DefaultAgentCardResolver = vi.fn();
+    const RestTransportFactory = vi.fn();
+    const JsonRpcTransportFactory = vi.fn();
+    const ClientFactoryOptions = {
+        default: {},
+        createFrom: vi.fn(),
+    };
+    const createAuthenticatingFetchWithRetry = vi.fn();
+    DefaultAgentCardResolver.prototype.resolve = vi.fn();
+    ClientFactory.prototype.createFromUrl = vi.fn();
+    return {
+        ClientFactory,
+        ClientFactoryOptions,
+        DefaultAgentCardResolver,
+        RestTransportFactory,
+        JsonRpcTransportFactory,
+        createAuthenticatingFetchWithRetry,
+    };
+});
+describe('A2AClientManager', () => {
+    let manager;
+    // Stable mocks initialized once
+    const sendMessageMock = vi.fn();
+    const getTaskMock = vi.fn();
+    const cancelTaskMock = vi.fn();
+    const getAgentCardMock = vi.fn();
+    const authFetchMock = vi.fn();
+    const mockClient = {
+        sendMessage: sendMessageMock,
+        getTask: getTaskMock,
+        cancelTask: cancelTaskMock,
+        getAgentCard: getAgentCardMock,
+    };
+    const mockAgentCard = { name: 'TestAgent' };
+    beforeEach(() => {
+        vi.clearAllMocks();
+        A2AClientManager.resetInstanceForTesting();
+        manager = A2AClientManager.getInstance();
+        // Default mock implementations
+        getAgentCardMock.mockResolvedValue({
+            ...mockAgentCard,
+            url: 'http://test.agent/real/endpoint',
+        });
+        vi.mocked(ClientFactory.prototype.createFromUrl).mockResolvedValue(mockClient);
+        vi.mocked(DefaultAgentCardResolver.prototype.resolve).mockResolvedValue({
+            ...mockAgentCard,
+            url: 'http://test.agent/real/endpoint',
+        });
+        vi.mocked(ClientFactoryOptions.createFrom).mockImplementation((_defaults, overrides) => overrides);
+        vi.mocked(createAuthenticatingFetchWithRetry).mockReturnValue(authFetchMock);
+        vi.stubGlobal('fetch', vi.fn().mockResolvedValue({
+            ok: true,
+            json: async () => ({}),
+        }));
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+        vi.unstubAllGlobals();
+    });
+    it('should enforce the singleton pattern', () => {
+        const instance1 = A2AClientManager.getInstance();
+        const instance2 = A2AClientManager.getInstance();
+        expect(instance1).toBe(instance2);
+    });
+    describe('loadAgent', () => {
+        it('should create and cache an A2AClient', async () => {
+            const agentCard = await manager.loadAgent('TestAgent', 'http://test.agent/card');
+            expect(agentCard).toMatchObject(mockAgentCard);
+            expect(manager.getAgentCard('TestAgent')).toBe(agentCard);
+            expect(manager.getClient('TestAgent')).toBeDefined();
+        });
+        it('should throw an error if an agent with the same name is already loaded', async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent/card');
+            await expect(manager.loadAgent('TestAgent', 'http://another.agent/card')).rejects.toThrow("Agent with name 'TestAgent' is already loaded.");
+        });
+        it('should use native fetch by default', async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent/card');
+            expect(createAuthenticatingFetchWithRetry).not.toHaveBeenCalled();
+        });
+        it('should use provided custom authentication handler', async () => {
+            const customAuthHandler = {
+                headers: vi.fn(),
+                shouldRetryWithHeaders: vi.fn(),
+            };
+            await manager.loadAgent('CustomAuthAgent', 'http://custom.agent/card', customAuthHandler);
+            expect(createAuthenticatingFetchWithRetry).toHaveBeenCalledWith(expect.anything(), customAuthHandler);
+        });
+        it('should log a debug message upon loading an agent', async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent/card');
+            expect(debugLogger.debug).toHaveBeenCalledWith("[A2AClientManager] Loaded agent 'TestAgent' from http://test.agent/card");
+        });
+    });
+    describe('sendMessage', () => {
+        beforeEach(async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent');
+        });
+        it('should send a message to the correct agent', async () => {
+            sendMessageMock.mockResolvedValue({
+                kind: 'message',
+                messageId: 'a',
+                parts: [],
+                role: 'agent',
+            });
+            await manager.sendMessage('TestAgent', 'Hello');
+            expect(sendMessageMock).toHaveBeenCalledWith(expect.objectContaining({
+                message: expect.anything(),
+            }));
+        });
+        it('should use contextId and taskId when provided', async () => {
+            sendMessageMock.mockResolvedValue({
+                kind: 'message',
+                messageId: 'a',
+                parts: [],
+                role: 'agent',
+            });
+            const expectedContextId = 'user-context-id';
+            const expectedTaskId = 'user-task-id';
+            await manager.sendMessage('TestAgent', 'Hello', {
+                contextId: expectedContextId,
+                taskId: expectedTaskId,
+            });
+            const call = sendMessageMock.mock.calls[0][0];
+            expect(call.message.contextId).toBe(expectedContextId);
+            expect(call.message.taskId).toBe(expectedTaskId);
+        });
+        it('should return result from client', async () => {
+            const mockResult = {
+                contextId: 'server-context-id',
+                id: 'ctx-1',
+                kind: 'task',
+                status: { state: 'working' },
+            };
+            sendMessageMock.mockResolvedValueOnce(mockResult);
+            const response = await manager.sendMessage('TestAgent', 'Hello');
+            expect(response).toEqual(mockResult);
+        });
+        it('should throw prefixed error on failure', async () => {
+            sendMessageMock.mockRejectedValueOnce(new Error('Network error'));
+            await expect(manager.sendMessage('TestAgent', 'Hello')).rejects.toThrow('A2AClient SendMessage Error [TestAgent]: Network error');
+        });
+        it('should throw an error if the agent is not found', async () => {
+            await expect(manager.sendMessage('NonExistentAgent', 'Hello')).rejects.toThrow("Agent 'NonExistentAgent' not found.");
+        });
+    });
+    describe('getTask', () => {
+        beforeEach(async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent');
+        });
+        it('should get a task from the correct agent', async () => {
+            getTaskMock.mockResolvedValue({
+                id: 'task123',
+                contextId: 'a',
+                kind: 'task',
+                status: { state: 'completed' },
+            });
+            await manager.getTask('TestAgent', 'task123');
+            expect(getTaskMock).toHaveBeenCalledWith({
+                id: 'task123',
+            });
+        });
+        it('should throw prefixed error on failure', async () => {
+            getTaskMock.mockRejectedValueOnce(new Error('Network error'));
+            await expect(manager.getTask('TestAgent', 'task123')).rejects.toThrow('A2AClient getTask Error [TestAgent]: Network error');
+        });
+        it('should throw an error if the agent is not found', async () => {
+            await expect(manager.getTask('NonExistentAgent', 'task123')).rejects.toThrow("Agent 'NonExistentAgent' not found.");
+        });
+    });
+    describe('cancelTask', () => {
+        beforeEach(async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent');
+        });
+        it('should cancel a task on the correct agent', async () => {
+            cancelTaskMock.mockResolvedValue({
+                id: 'task123',
+                contextId: 'a',
+                kind: 'task',
+                status: { state: 'canceled' },
+            });
+            await manager.cancelTask('TestAgent', 'task123');
+            expect(cancelTaskMock).toHaveBeenCalledWith({
+                id: 'task123',
+            });
+        });
+        it('should throw prefixed error on failure', async () => {
+            cancelTaskMock.mockRejectedValueOnce(new Error('Network error'));
+            await expect(manager.cancelTask('TestAgent', 'task123')).rejects.toThrow('A2AClient cancelTask Error [TestAgent]: Network error');
+        });
+        it('should throw an error if the agent is not found', async () => {
+            await expect(manager.cancelTask('NonExistentAgent', 'task123')).rejects.toThrow("Agent 'NonExistentAgent' not found.");
+        });
+    });
+    describe('createAdapterFetch', () => {
+        it('normalizes TASK_STATE_ enums to lower-case', async () => {
+            const baseFetch = vi
+                .fn()
+                .mockResolvedValue(new Response(JSON.stringify({ status: { state: 'TASK_STATE_WORKING' } })));
+            const adapter = createAdapterFetch(baseFetch);
+            const response = await adapter('http://example.com', {
+                method: 'POST',
+                body: '{}',
+            });
+            const data = await response.json();
+            expect(data.status.state).toBe('working');
+        });
+        it('lowercases non-prefixed task states', async () => {
+            const baseFetch = vi
+                .fn()
+                .mockResolvedValue(new Response(JSON.stringify({ status: { state: 'WORKING' } })));
+            const adapter = createAdapterFetch(baseFetch);
+            const response = await adapter('http://example.com', {
+                method: 'POST',
+                body: '{}',
+            });
+            const data = await response.json();
+            expect(data.status.state).toBe('working');
+        });
+    });
+});
+//# sourceMappingURL=a2a-client-manager.test.js.map

package/dist/src/agents/a2a-client-manager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"a2a-client-manager.test.js","sourceRoot":"","sources":["../../../src/agents/a2a-client-manager.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,gBAAgB,EAEhB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EACL,kCAAkC,EAClC,oBAAoB,GACrB,MAAM,oBAAoB,CAAC;AAE5B,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,WAAW,EAAE;QACX,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf;CACF,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE;IACjC,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC9B,MAAM,wBAAwB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACzC,MAAM,oBAAoB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACrC,MAAM,uBAAuB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACxC,MAAM,oBAAoB,GAAG;QAC3B,OAAO,EAAE,EAAE;QACX,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;KACpB,CAAC;IACF,MAAM,kCAAkC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAEnD,wBAAwB,CAAC,SAAS,CAAC,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACrD,aAAa,CAAC,SAAS,CAAC,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAEhD,OAAO;QACL,aAAa;QACb,oBAAoB;QACpB,wBAAwB;QACxB,oBAAoB;QACpB,uBAAuB;QACvB,kCAAkC;KACnC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,OAAyB,CAAC;IAE9B,gCAAgC;IAChC,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAChC,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC5B,MAAM,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC/B,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACjC,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE9B,MAAM,UAAU,GAAG;QACjB,WAAW,EAAE,eAAe;QAC5B,OAAO,EAAE,WAAW;QACpB,UAAU,EAAE,cAAc;QAC1B,YAAY,EAAE,gBAAgB;KACV,CAAC;IAEvB,MAAM,aAAa,GAAuB,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IAEhE,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,gBAAgB,CAAC,uBAAuB,EAAE,CAAC;QAC3C,OAAO,GAAG,gBAAgB,CAAC,WAAW,EAAE,CAAC;QAEzC,+BAA+B;QAC/B,gBAAgB,CAAC,iBAAiB,CAAC;YACjC,GAAG,aAAa;YAChB,GAAG,EAAE,iCAAiC;SAC1B,CAAC,CAAC;QAEhB,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,iBAAiB,CAChE,UAAU,CACX,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,wBAAwB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,iBAAiB,CAAC;YACtE,GAAG,aAAa;YAChB,GAAG,EAAE,iCAAiC;SAC1B,CAAC,CAAC;QAEhB,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,kBAAkB,CAC3D,CAAC,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC,SAAiC,CAC5D,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,kCAAkC,CAAC,CAAC,eAAe,CAC3D,aAAa,CACd,CAAC;QAEF,EAAE,CAAC,UAAU,CACX,OAAO,EACP,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACxB,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;SACX,CAAC,CACf,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,SAAS,GAAG,gBAAgB,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,SAAS,GAAG,gBAAgB,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,SAAS,CACvC,WAAW,EACX,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,wBAAwB,CAAC,CAAC;YAC/D,MAAM,MAAM,CACV,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAC5D,CAAC,OAAO,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,wBAAwB,CAAC,CAAC;YAC/D,MAAM,CAAC,kCAAkC,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,iBAAiB,GAAG;gBACxB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;gBAChB,sBAAsB,EAAE,EAAE,CAAC,EAAE,EAAE;aAChC,CAAC;YACF,MAAM,OAAO,CAAC,SAAS,CACrB,iBAAiB,EACjB,0BAA0B,EAC1B,iBAAqD,CACtD,CAAC;YAEF,MAAM,CAAC,kCAAkC,CAAC,CAAC,oBAAoB,CAC7D,MAAM,CAAC,QAAQ,EAAE,EACjB,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,wBAAwB,CAAC,CAAC;YAC/D,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC5C,yEAAyE,CAC1E,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,eAAe,CAAC,iBAAiB,CAAC;gBAChC,IAAI,EAAE,SAAS;gBACf,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE,EAAE;gBACT,IAAI,EAAE,OAAO;aACO,CAAC,CAAC;YAExB,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAChD,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE;aAC3B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,eAAe,CAAC,iBAAiB,CAAC;gBAChC,IAAI,EAAE,SAAS;gBACf,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE,EAAE;gBACT,IAAI,EAAE,OAAO;aACO,CAAC,CAAC;YAExB,MAAM,iBAAiB,GAAG,iBAAiB,CAAC;YAC5C,MAAM,cAAc,GAAG,cAAc,CAAC;YAEtC,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE;gBAC9C,SAAS,EAAE,iBAAiB;gBAC5B,MAAM,EAAE,cAAc;aACvB,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,UAAU,GAAG;gBACjB,SAAS,EAAE,mBAAmB;gBAC9B,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;aAC7B,CAAC;YAEF,eAAe,CAAC,qBAAqB,CAAC,UAA+B,CAAC,CAAC;YAEvE,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAEjE,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,eAAe,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;YAElE,MAAM,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,wDAAwD,CACzD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,CACV,OAAO,CAAC,WAAW,CAAC,kBAAkB,EAAE,OAAO,CAAC,CACjD,CAAC,OAAO,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,WAAW,CAAC,iBAAiB,CAAC;gBAC5B,EAAE,EAAE,SAAS;gBACb,SAAS,EAAE,GAAG;gBACd,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;aACvB,CAAC,CAAC;YAEX,MAAM,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACvC,EAAE,EAAE,SAAS;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,WAAW,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;YAE9D,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACnE,oDAAoD,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,CACV,OAAO,CAAC,OAAO,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAC/C,CAAC,OAAO,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,cAAc,CAAC,iBAAiB,CAAC;gBAC/B,EAAE,EAAE,SAAS;gBACb,SAAS,EAAE,GAAG;gBACd,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE;aACtB,CAAC,CAAC;YAEX,MAAM,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC;gBAC1C,EAAE,EAAE,SAAS;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,cAAc,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;YAEjE,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACtE,uDAAuD,CACxD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,CACV,OAAO,CAAC,UAAU,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAClD,CAAC,OAAO,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,iBAAiB,CAChB,IAAI,QAAQ,CACV,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,oBAAoB,EAAE,EAAE,CAAC,CAC5D,CACF,CAAC;YAEJ,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAyB,CAAC,CAAC;YAC9D,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,oBAAoB,EAAE;gBACnD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAEnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,SAAS,GAAG,EAAE;iBACjB,EAAE,EAAE;iBACJ,iBAAiB,CAChB,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,CAC/D,CAAC;YAEJ,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAyB,CAAC,CAAC;YAC9D,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,oBAAoB,EAAE;gBACnD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAEnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Message, Task, Part } from '@a2a-js/sdk';
+/**
+ * Extracts a human-readable text representation from a Message object.
+ * Handles Text, Data (JSON), and File parts.
+ */
+export declare function extractMessageText(message: Message | undefined): string;
+/**
+ * Extracts text from a single Part.
+ */
+export declare function extractPartText(part: Part): string;
+/**
+ * Extracts a human-readable text summary from a Task object.
+ * Includes status, ID, and any artifact content.
+ */
+export declare function extractTaskText(task: Task): string;
+/**
+ * Extracts contextId and taskId from a Message or Task response.
+ * Follows the pattern from the A2A CLI sample to maintain conversational continuity.
+ */
+export declare function extractIdsFromResponse(result: Message | Task): {
+    contextId?: string;
+    taskId?: string;
+};

package/dist/src/agents/a2aUtils.js

@@ -0,0 +1,111 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Extracts a human-readable text representation from a Message object.
+ * Handles Text, Data (JSON), and File parts.
+ */
+export function extractMessageText(message) {
+    if (!message || !message.parts) {
+        return '';
+    }
+    const parts = message.parts
+        .map((part) => extractPartText(part))
+        .filter(Boolean);
+    return parts.join('\n');
+}
+/**
+ * Extracts text from a single Part.
+ */
+export function extractPartText(part) {
+    if (isTextPart(part)) {
+        return part.text;
+    }
+    if (isDataPart(part)) {
+        // Attempt to format known data types if metadata exists, otherwise JSON stringify
+        return `Data: ${JSON.stringify(part.data)}`;
+    }
+    if (isFilePart(part)) {
+        const fileData = part.file;
+        if (fileData.name) {
+            return `File: ${fileData.name}`;
+        }
+        if ('uri' in fileData && fileData.uri) {
+            return `File: ${fileData.uri}`;
+        }
+        return `File: [binary/unnamed]`;
+    }
+    return '';
+}
+/**
+ * Extracts a human-readable text summary from a Task object.
+ * Includes status, ID, and any artifact content.
+ */
+export function extractTaskText(task) {
+    let output = `ID: ${task.id}\n`;
+    output += `State: ${task.status.state}\n`;
+    // Status Message
+    const statusMessageText = extractMessageText(task.status.message);
+    if (statusMessageText) {
+        output += `Status Message: ${statusMessageText}\n`;
+    }
+    // Artifacts
+    if (task.artifacts && task.artifacts.length > 0) {
+        output += `Artifacts:\n`;
+        for (const artifact of task.artifacts) {
+            output += `  - Name: ${artifact.name}\n`;
+            if (artifact.parts && artifact.parts.length > 0) {
+                // Treat artifact parts as a message for extraction
+                const artifactContent = artifact.parts
+                    .map((p) => extractPartText(p))
+                    .filter(Boolean)
+                    .join('\n');
+                if (artifactContent) {
+                    // Indent content for readability
+                    const indentedContent = artifactContent.replace(/^/gm, '    ');
+                    output += `    Content:\n${indentedContent}\n`;
+                }
+            }
+        }
+    }
+    return output;
+}
+// Type Guards
+function isTextPart(part) {
+    return part.kind === 'text';
+}
+function isDataPart(part) {
+    return part.kind === 'data';
+}
+function isFilePart(part) {
+    return part.kind === 'file';
+}
+/**
+ * Extracts contextId and taskId from a Message or Task response.
+ * Follows the pattern from the A2A CLI sample to maintain conversational continuity.
+ */
+export function extractIdsFromResponse(result) {
+    let contextId;
+    let taskId;
+    if (result.kind === 'message') {
+        taskId = result.taskId;
+        contextId = result.contextId;
+    }
+    else if (result.kind === 'task') {
+        taskId = result.id;
+        contextId = result.contextId;
+        // If the task is in a final state (and not input-required), we clear the taskId
+        // so that the next interaction starts a fresh task (or keeps context without being bound to the old task).
+        if (result.status &&
+            result.status.state !== 'input-required' &&
+            (result.status.state === 'completed' ||
+                result.status.state === 'failed' ||
+                result.status.state === 'canceled')) {
+            taskId = undefined;
+        }
+    }
+    return { contextId, taskId };
+}
+//# sourceMappingURL=a2aUtils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"a2aUtils.js","sourceRoot":"","sources":["../../../src/agents/a2aUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAA4B;IAC7D,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC/B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK;SACxB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;SACpC,MAAM,CAAC,OAAO,CAAC,CAAC;IACnB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,IAAU;IACxC,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,kFAAkF;QAClF,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IAC9C,CAAC;IAED,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;QAC3B,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAClB,OAAO,SAAS,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClC,CAAC;QACD,IAAI,KAAK,IAAI,QAAQ,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;YACtC,OAAO,SAAS,QAAQ,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;QACD,OAAO,wBAAwB,CAAC;IAClC,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,IAAU;IACxC,IAAI,MAAM,GAAG,OAAO,IAAI,CAAC,EAAE,IAAI,CAAC;IAChC,MAAM,IAAI,UAAU,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC;IAE1C,iBAAiB;IACjB,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,iBAAiB,EAAE,CAAC;QACtB,MAAM,IAAI,mBAAmB,iBAAiB,IAAI,CAAC;IACrD,CAAC;IAED,YAAY;IACZ,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,cAAc,CAAC;QACzB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,IAAI,aAAa,QAAQ,CAAC,IAAI,IAAI,CAAC;YACzC,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChD,mDAAmD;gBACnD,MAAM,eAAe,GAAG,QAAQ,CAAC,KAAK;qBACnC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;qBAC9B,MAAM,CAAC,OAAO,CAAC;qBACf,IAAI,CAAC,IAAI,CAAC,CAAC;gBAEd,IAAI,eAAe,EAAE,CAAC;oBACpB,iCAAiC;oBACjC,MAAM,eAAe,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;oBAC/D,MAAM,IAAI,iBAAiB,eAAe,IAAI,CAAC;gBACjD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,cAAc;AAEd,SAAS,UAAU,CAAC,IAAU;IAC5B,OAAO,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;AAC9B,CAAC;AAED,SAAS,UAAU,CAAC,IAAU;IAC5B,OAAO,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;AAC9B,CAAC;AAED,SAAS,UAAU,CAAC,IAAU;IAC5B,OAAO,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAsB;IAI3D,IAAI,SAA6B,CAAC;IAClC,IAAI,MAA0B,CAAC;IAE/B,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAC9B,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QACvB,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IAC/B,CAAC;SAAM,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAClC,MAAM,GAAG,MAAM,CAAC,EAAE,CAAC;QACnB,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAE7B,gFAAgF;QAChF,2GAA2G;QAC3G,IACE,MAAM,CAAC,MAAM;YACb,MAAM,CAAC,MAAM,CAAC,KAAK,KAAK,gBAAgB;YACxC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,KAAK,WAAW;gBAClC,MAAM,CAAC,MAAM,CAAC,KAAK,KAAK,QAAQ;gBAChC,MAAM,CAAC,MAAM,CAAC,KAAK,KAAK,UAAU,CAAC,EACrC,CAAC;YACD,MAAM,GAAG,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;AAC/B,CAAC"}

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

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

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

@@ -0,0 +1,147 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { extractMessageText, extractTaskText, extractIdsFromResponse, } from './a2aUtils.js';
+describe('a2aUtils', () => {
+    describe('extractIdsFromResponse', () => {
+        it('should extract IDs from a message response', () => {
+            const message = {
+                kind: 'message',
+                role: 'agent',
+                messageId: 'm1',
+                contextId: 'ctx-1',
+                taskId: 'task-1',
+                parts: [],
+            };
+            const result = extractIdsFromResponse(message);
+            expect(result).toEqual({ contextId: 'ctx-1', taskId: 'task-1' });
+        });
+        it('should extract IDs from an in-progress task response', () => {
+            const task = {
+                id: 'task-2',
+                contextId: 'ctx-2',
+                kind: 'task',
+                status: { state: 'working' },
+            };
+            const result = extractIdsFromResponse(task);
+            expect(result).toEqual({ contextId: 'ctx-2', taskId: 'task-2' });
+        });
+    });
+    describe('extractMessageText', () => {
+        it('should extract text from simple text parts', () => {
+            const message = {
+                kind: 'message',
+                role: 'user',
+                messageId: '1',
+                parts: [
+                    { kind: 'text', text: 'Hello' },
+                    { kind: 'text', text: 'World' },
+                ],
+            };
+            expect(extractMessageText(message)).toBe('Hello\nWorld');
+        });
+        it('should extract data from data parts', () => {
+            const message = {
+                kind: 'message',
+                role: 'user',
+                messageId: '1',
+                parts: [{ kind: 'data', data: { foo: 'bar' } }],
+            };
+            expect(extractMessageText(message)).toBe('Data: {"foo":"bar"}');
+        });
+        it('should extract file info from file parts', () => {
+            const message = {
+                kind: 'message',
+                role: 'user',
+                messageId: '1',
+                parts: [
+                    {
+                        kind: 'file',
+                        file: {
+                            name: 'test.txt',
+                            uri: 'file://test.txt',
+                            mimeType: 'text/plain',
+                        },
+                    },
+                    {
+                        kind: 'file',
+                        file: {
+                            uri: 'http://example.com/doc',
+                            mimeType: 'application/pdf',
+                        },
+                    },
+                ],
+            };
+            // The formatting logic in a2aUtils prefers name over uri
+            expect(extractMessageText(message)).toContain('File: test.txt');
+            expect(extractMessageText(message)).toContain('File: http://example.com/doc');
+        });
+        it('should handle mixed parts', () => {
+            const message = {
+                kind: 'message',
+                role: 'user',
+                messageId: '1',
+                parts: [
+                    { kind: 'text', text: 'Here is data:' },
+                    { kind: 'data', data: { value: 123 } },
+                ],
+            };
+            expect(extractMessageText(message)).toBe('Here is data:\nData: {"value":123}');
+        });
+        it('should return empty string for undefined or empty message', () => {
+            expect(extractMessageText(undefined)).toBe('');
+            expect(extractMessageText({
+                kind: 'message',
+                role: 'user',
+                messageId: '1',
+                parts: [],
+            })).toBe('');
+        });
+    });
+    describe('extractTaskText', () => {
+        it('should extract basic task info', () => {
+            const task = {
+                id: 'task-1',
+                contextId: 'ctx-1',
+                kind: 'task',
+                status: {
+                    state: 'working',
+                    message: {
+                        kind: 'message',
+                        role: 'agent',
+                        messageId: 'm1',
+                        parts: [{ kind: 'text', text: 'Processing...' }],
+                    },
+                },
+            };
+            const result = extractTaskText(task);
+            expect(result).toContain('ID: task-1');
+            expect(result).toContain('State: working');
+            expect(result).toContain('Status Message: Processing...');
+        });
+        it('should extract artifacts', () => {
+            const task = {
+                id: 'task-1',
+                contextId: 'ctx-1',
+                kind: 'task',
+                status: { state: 'completed' },
+                artifacts: [
+                    {
+                        artifactId: 'art-1',
+                        name: 'Report',
+                        parts: [{ kind: 'text', text: 'This is the report.' }],
+                    },
+                ],
+            };
+            const result = extractTaskText(task);
+            expect(result).toContain('Artifacts:');
+            expect(result).toContain('  - Name: Report');
+            expect(result).toContain('    Content:');
+            expect(result).toContain('    This is the report.');
+        });
+    });
+});
+//# sourceMappingURL=a2aUtils.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"a2aUtils.test.js","sourceRoot":"","sources":["../../../src/agents/a2aUtils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,kBAAkB,EAClB,eAAe,EACf,sBAAsB,GACvB,MAAM,eAAe,CAAC;AAGvB,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,EAAE;aACV,CAAC;YAEF,MAAM,MAAM,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,IAAI,GAAS;gBACjB,EAAE,EAAE,QAAQ;gBACZ,SAAS,EAAE,OAAO;gBAClB,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;aAC7B,CAAC;YAEF,MAAM,MAAM,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE;oBACL,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAc;oBAC3C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAc;iBAC5C;aACF,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAc,CAAC;aAC5D,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE;4BACJ,IAAI,EAAE,UAAU;4BAChB,GAAG,EAAE,iBAAiB;4BACtB,QAAQ,EAAE,YAAY;yBACvB;qBACU;oBACb;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE;4BACJ,GAAG,EAAE,wBAAwB;4BAC7B,QAAQ,EAAE,iBAAiB;yBAC5B;qBACU;iBACd;aACF,CAAC;YACF,yDAAyD;YACzD,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAC3C,8BAA8B,CAC/B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,OAAO,GAAY;gBACvB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE;oBACL,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAc;oBACnD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,EAAc;iBACnD;aACF,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACtC,oCAAoC,CACrC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CACJ,kBAAkB,CAAC;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,GAAG;gBACd,KAAK,EAAE,EAAE;aACC,CAAC,CACd,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,IAAI,GAAS;gBACjB,EAAE,EAAE,QAAQ;gBACZ,SAAS,EAAE,OAAO;gBAClB,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE;oBACN,KAAK,EAAE,SAAS;oBAChB,OAAO,EAAE;wBACP,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,OAAO;wBACb,SAAS,EAAE,IAAI;wBACf,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAc,CAAC;qBAC7D;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,+BAA+B,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,IAAI,GAAS;gBACjB,EAAE,EAAE,QAAQ;gBACZ,SAAS,EAAE,OAAO;gBAClB,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;gBAC9B,SAAS,EAAE;oBACT;wBACE,UAAU,EAAE,OAAO;wBACnB,IAAI,EAAE,QAAQ;wBACd,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,EAAc,CAAC;qBACnE;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/codebase-investigator.d.ts

@@ -3,7 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import type { AgentDefinition } from './types.js';
+import type { LocalAgentDefinition } from './types.js';
 import { z } from 'zod';
 declare const CodebaseInvestigationReportSchema: z.ZodObject<{
     SummaryOfFindings: z.ZodString;
@@ -42,5 +42,5 @@ declare const CodebaseInvestigationReportSchema: z.ZodObject<{
  * A Proof-of-Concept subagent specialized in analyzing codebase structure,
  * dependencies, and technologies.
  */
-export declare const CodebaseInvestigatorAgent: AgentDefinition<typeof CodebaseInvestigationReportSchema>;
+export declare const CodebaseInvestigatorAgent: LocalAgentDefinition<typeof CodebaseInvestigationReportSchema>;
 export {};

package/dist/src/agents/codebase-investigator.js

@@ -28,14 +28,15 @@ const CodebaseInvestigationReportSchema = z.object({
  */
 export const CodebaseInvestigatorAgent = {
     name: 'codebase_investigator',
+    kind: 'local',
     displayName: 'Codebase Investigator Agent',
-    description: `The specialized tool for codebase analysis, architectural mapping, and understanding system-wide dependencies. 
-    Invoke this tool for tasks like vague requests, bug root-cause analysis, system refactoring, comprehensive feature implementation or to answer questions about the codebase that require investigation. 
+    description: `The specialized tool for codebase analysis, architectural mapping, and understanding system-wide dependencies.
+    Invoke this tool for tasks like vague requests, bug root-cause analysis, system refactoring, comprehensive feature implementation or to answer questions about the codebase that require investigation.
     It returns a structured report with key file paths, symbols, and actionable architectural insights.`,
     inputConfig: {
         inputs: {
             objective: {
-                description: `A comprehensive and detailed description of the user's ultimate goal. 
+                description: `A comprehensive and detailed description of the user's ultimate goal.
           You must include original user's objective as well as questions and any extra context and questions you may have.`,
                 type: 'string',
                 required: true,
@@ -74,7 +75,7 @@ You are a sub-agent in a larger system. Your only responsibility is to provide d
 - **DO:** Find the key modules, classes, and functions that are part of the problem and its solution.
 - **DO:** Understand *why* the code is written the way it is. Question everything.
 - **DO:** Foresee the ripple effects of a change. If \`function A\` is modified, you must check its callers. If a data structure is altered, you must identify where its type definitions need to be updated.
-- **DO:** provide a conclusion and insights to the main agent that invoked you. If the agent is trying to solve a bug, you should provide the root cause of the bug, its impacts, how to fix it etc. If it's a new feature, you should provide insights on where to implement it, what chagnes are necessary etc. 
+- **DO:** provide a conclusion and insights to the main agent that invoked you. If the agent is trying to solve a bug, you should provide the root cause of the bug, its impacts, how to fix it etc. If it's a new feature, you should provide insights on where to implement it, what changes are necessary etc.
 - **DO NOT:** Write the final implementation code yourself.
 - **DO NOT:** Stop at the first relevant file. Your goal is a comprehensive understanding of the entire relevant subsystem.
 You operate in a non-interactive loop and must reason based on the information provided and the output of your tools.

package/dist/src/agents/codebase-investigator.js.map

@@ -1 +1 @@
-{"version":3,"file":"codebase-investigator.js","sourceRoot":"","sources":["../../../src/agents/codebase-investigator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EACL,cAAc,EACd,cAAc,EACd,YAAY,EACZ,mBAAmB,GACpB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,sEAAsE;AACtE,MAAM,iCAAiC,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,iBAAiB,EAAE,CAAC;SACjB,MAAM,EAAE;SACR,QAAQ,CACP,+EAA+E,CAChF;IACH,gBAAgB,EAAE,CAAC;SAChB,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,CACP,yEAAyE,CAC1E;IACH,iBAAiB,EAAE,CAAC;SACjB,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;QACP,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;QACpB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAChC,CAAC,CACH;SACA,QAAQ,CAAC,2DAA2D,CAAC;CACzE,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAElC;IACF,IAAI,EAAE,uBAAuB;IAC7B,WAAW,EAAE,6BAA6B;IAC1C,WAAW,EAAE;;wGAEyF;IACtG,WAAW,EAAE;QACX,MAAM,EAAE;YACN,SAAS,EAAE;gBACT,WAAW,EAAE;4HACuG;gBACpH,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,IAAI;aACf;SACF;KACF;IACD,YAAY,EAAE;QACZ,UAAU,EAAE,QAAQ;QACpB,WAAW,EAAE,kDAAkD;QAC/D,MAAM,EAAE,iCAAiC;KAC1C;IAED,oFAAoF;IACpF,aAAa,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAE1D,WAAW,EAAE;QACX,KAAK,EAAE,oBAAoB;QAC3B,IAAI,EAAE,GAAG;QACT,KAAK,EAAE,IAAI;QACX,cAAc,EAAE,CAAC,CAAC;KACnB;IAED,SAAS,EAAE;QACT,gBAAgB,EAAE,CAAC;QACnB,SAAS,EAAE,EAAE;KACd;IAED,UAAU,EAAE;QACV,wCAAwC;QACxC,KAAK,EAAE,CAAC,YAAY,EAAE,mBAAmB,EAAE,cAAc,EAAE,cAAc,CAAC;KAC3E;IAED,YAAY,EAAE;QACZ,KAAK,EAAE;;;aAGE;QACT,YAAY,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0DjB;KACE;CACF,CAAC"}
+{"version":3,"file":"codebase-investigator.js","sourceRoot":"","sources":["../../../src/agents/codebase-investigator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EACL,cAAc,EACd,cAAc,EACd,YAAY,EACZ,mBAAmB,GACpB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,sEAAsE;AACtE,MAAM,iCAAiC,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,iBAAiB,EAAE,CAAC;SACjB,MAAM,EAAE;SACR,QAAQ,CACP,+EAA+E,CAChF;IACH,gBAAgB,EAAE,CAAC;SAChB,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,CACP,yEAAyE,CAC1E;IACH,iBAAiB,EAAE,CAAC;SACjB,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;QACP,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;QACpB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAChC,CAAC,CACH;SACA,QAAQ,CAAC,2DAA2D,CAAC;CACzE,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAElC;IACF,IAAI,EAAE,uBAAuB;IAC7B,IAAI,EAAE,OAAO;IACb,WAAW,EAAE,6BAA6B;IAC1C,WAAW,EAAE;;wGAEyF;IACtG,WAAW,EAAE;QACX,MAAM,EAAE;YACN,SAAS,EAAE;gBACT,WAAW,EAAE;4HACuG;gBACpH,IAAI,EAAE,QAAQ;gBACd,QAAQ,EAAE,IAAI;aACf;SACF;KACF;IACD,YAAY,EAAE;QACZ,UAAU,EAAE,QAAQ;QACpB,WAAW,EAAE,kDAAkD;QAC/D,MAAM,EAAE,iCAAiC;KAC1C;IAED,oFAAoF;IACpF,aAAa,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAE1D,WAAW,EAAE;QACX,KAAK,EAAE,oBAAoB;QAC3B,IAAI,EAAE,GAAG;QACT,KAAK,EAAE,IAAI;QACX,cAAc,EAAE,CAAC,CAAC;KACnB;IAED,SAAS,EAAE;QACT,gBAAgB,EAAE,CAAC;QACnB,SAAS,EAAE,EAAE;KACd;IAED,UAAU,EAAE;QACV,wCAAwC;QACxC,KAAK,EAAE,CAAC,YAAY,EAAE,mBAAmB,EAAE,cAAc,EAAE,cAAc,CAAC;KAC3E;IAED,YAAY,EAAE;QACZ,KAAK,EAAE;;;aAGE;QACT,YAAY,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0DjB;KACE;CACF,CAAC"}

package/dist/src/agents/delegate-to-agent-tool.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseDeclarativeTool, type ToolInvocation, type ToolResult } from '../tools/tools.js';
+import type { AgentRegistry } from './registry.js';
+import type { Config } from '../config/config.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+type DelegateParams = {
+    agent_name: string;
+} & Record<string, unknown>;
+export declare class DelegateToAgentTool extends BaseDeclarativeTool<DelegateParams, ToolResult> {
+    private readonly registry;
+    private readonly config;
+    constructor(registry: AgentRegistry, config: Config, messageBus: MessageBus);
+    protected createInvocation(params: DelegateParams, messageBus: MessageBus, _toolName?: string, _toolDisplayName?: string): ToolInvocation<DelegateParams, ToolResult>;
+}
+export {};