@machina.ai/cell-cli-core 1.22.5-rc1 → 1.25.0-rc1

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

@@ -0,0 +1,281 @@
+/**
+ * @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");
+        });
+        it('should clear the cache', async () => {
+            await manager.loadAgent('TestAgent', 'http://test.agent/card');
+            expect(manager.getAgentCard('TestAgent')).toBeDefined();
+            expect(manager.getClient('TestAgent')).toBeDefined();
+            manager.clearCache();
+            expect(manager.getAgentCard('TestAgent')).toBeUndefined();
+            expect(manager.getClient('TestAgent')).toBeUndefined();
+            expect(debugLogger.debug).toHaveBeenCalledWith('[A2AClientManager] Cache cleared.');
+        });
+    });
+    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');
+        });
+        it('bypasses adapter for JSON-RPC requests', async () => {
+            const baseFetch = vi.fn().mockResolvedValue(new Response('{}'));
+            const adapter = createAdapterFetch(baseFetch);
+            const rpcBody = JSON.stringify({ jsonrpc: '2.0', method: 'foo' });
+            await adapter('http://example.com', {
+                method: 'POST',
+                body: rpcBody,
+            });
+            // Verify baseFetch was called with original body, not modified
+            expect(baseFetch).toHaveBeenCalledWith('http://example.com', expect.objectContaining({ body: rpcBody }));
+        });
+        it('applies dialect translation for remote REST requests', async () => {
+            const baseFetch = vi.fn().mockResolvedValue(new Response('{}'));
+            const adapter = createAdapterFetch(baseFetch);
+            const originalBody = JSON.stringify({
+                message: {
+                    role: 'user',
+                    parts: [{ kind: 'text', text: 'hi' }],
+                },
+            });
+            await adapter('https://remote-agent.com/v1/message:send', {
+                method: 'POST',
+                body: originalBody,
+            });
+            // Verify body WAS modified:
+            // 1. role: 'user' -> 'ROLE_USER'
+            // 2. parts mapped to content, kind stripped
+            const expectedBody = JSON.stringify({
+                message: {
+                    role: 'ROLE_USER',
+                    content: [{ text: 'hi' }],
+                },
+            });
+            expect(baseFetch).toHaveBeenCalledWith(expect.any(String), expect.objectContaining({ body: expectedBody }));
+        });
+    });
+});
+//# 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;QAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,wBAAwB,CAAC,CAAC;YAC/D,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACxD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAErD,OAAO,CAAC,UAAU,EAAE,CAAC;YAErB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACvD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAC5C,mCAAmC,CACpC,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;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAyB,CAAC,CAAC;YAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAElE,MAAM,OAAO,CAAC,oBAAoB,EAAE;gBAClC,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,OAAO;aACd,CAAC,CAAC;YAEH,+DAA+D;YAC/D,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,oBAAoB,EACpB,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAyB,CAAC,CAAC;YAC9D,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBACtC;aACF,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,0CAA0C,EAAE;gBACxD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,YAAY;aACnB,CAAC,CAAC;YAEH,4BAA4B;YAC5B,iCAAiC;YACjC,4CAA4C;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;iBAC1B;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAChD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,29 @@
+/**
+ * @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 clean, human-readable text summary from a Task object.
+ * Includes the status message and any artifact content with context headers.
+ * Technical metadata like ID and State are omitted for better clarity and token efficiency.
+ */
+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,113 @@
+/**
+ * @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) {
+        return '';
+    }
+    return extractPartsText(message.parts);
+}
+/**
+ * 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 clean, human-readable text summary from a Task object.
+ * Includes the status message and any artifact content with context headers.
+ * Technical metadata like ID and State are omitted for better clarity and token efficiency.
+ */
+export function extractTaskText(task) {
+    const parts = [];
+    // Status Message
+    const statusMessageText = extractMessageText(task.status?.message);
+    if (statusMessageText) {
+        parts.push(statusMessageText);
+    }
+    // Artifacts
+    if (task.artifacts) {
+        for (const artifact of task.artifacts) {
+            const artifactContent = extractPartsText(artifact.parts);
+            if (artifactContent) {
+                const header = artifact.name
+                    ? `Artifact (${artifact.name}):`
+                    : 'Artifact:';
+                parts.push(`${header}\n${artifactContent}`);
+            }
+        }
+    }
+    return parts.join('\n\n');
+}
+/**
+ * Extracts text from an array of parts.
+ */
+function extractPartsText(parts) {
+    if (!parts || parts.length === 0) {
+        return '';
+    }
+    return parts
+        .map((p) => extractPartText(p))
+        .filter(Boolean)
+        .join('\n');
+}
+// 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,EAAE,CAAC;QACb,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AACzC,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;;;;GAIG;AACH,MAAM,UAAU,eAAe,CAAC,IAAU;IACxC,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,iBAAiB;IACjB,MAAM,iBAAiB,GAAG,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnE,IAAI,iBAAiB,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAChC,CAAC;IAED,YAAY;IACZ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,eAAe,GAAG,gBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAEzD,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI;oBAC1B,CAAC,CAAC,aAAa,QAAQ,CAAC,IAAI,IAAI;oBAChC,CAAC,CAAC,WAAW,CAAC;gBAChB,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,KAAK,eAAe,EAAE,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5B,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,KAAyB;IACjD,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,KAAK;SACT,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;SAC9B,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,IAAI,CAAC,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.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 (clean)', () => {
+            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).not.toContain('ID: task-1');
+            expect(result).not.toContain('State: working');
+            expect(result).toBe('Processing...');
+        });
+        it('should extract artifacts with headers', () => {
+            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('Artifact (Report):');
+            expect(result).toContain('This is the report.');
+            expect(result).not.toContain('Artifacts:');
+            expect(result).not.toContain('  - Name: 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,wCAAwC,EAAE,GAAG,EAAE;YAChD,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,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,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,oBAAoB,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}