@google/gemini-cli-a2a-server 0.10.0-nightly.20251014.0a3e492e

package/dist/src/persistence/gcs.test.js

@@ -0,0 +1,265 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Storage } from '@google-cloud/storage';
+import * as fse from 'fs-extra';
+import { promises as fsPromises, createReadStream } from 'node:fs';
+import * as tar from 'tar';
+import { gzipSync, gunzipSync } from 'node:zlib';
+import { v4 as uuidv4 } from 'uuid';
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { GCSTaskStore, NoOpTaskStore } from './gcs.js';
+import { logger } from '../utils/logger.js';
+import * as configModule from '../config/config.js';
+import { getPersistedState, METADATA_KEY } from '../types.js';
+// Mock dependencies
+vi.mock('@google-cloud/storage');
+vi.mock('fs-extra', () => ({
+    pathExists: vi.fn(),
+    readdir: vi.fn(),
+    remove: vi.fn(),
+    ensureDir: vi.fn(),
+}));
+vi.mock('node:fs', async () => {
+    const actual = await vi.importActual('node:fs');
+    return {
+        ...actual,
+        promises: {
+            ...actual.promises,
+            readdir: vi.fn(),
+        },
+        createReadStream: vi.fn(),
+    };
+});
+vi.mock('tar');
+vi.mock('zlib');
+vi.mock('uuid');
+vi.mock('../utils/logger.js', () => ({
+    logger: {
+        info: vi.fn(),
+        warn: vi.fn(),
+        error: vi.fn(),
+        debug: vi.fn(),
+    },
+}));
+vi.mock('../config/config.js', () => ({
+    setTargetDir: vi.fn(),
+}));
+vi.mock('node:stream/promises', () => ({
+    pipeline: vi.fn(),
+}));
+vi.mock('../types.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        getPersistedState: vi.fn(),
+    };
+});
+const mockStorage = Storage;
+const mockFse = fse;
+const mockCreateReadStream = createReadStream;
+const mockTar = tar;
+const mockGzipSync = gzipSync;
+const mockGunzipSync = gunzipSync;
+const mockUuidv4 = uuidv4;
+const mockSetTargetDir = configModule.setTargetDir;
+const mockGetPersistedState = getPersistedState;
+const TEST_METADATA_KEY = METADATA_KEY || '__persistedState';
+describe('GCSTaskStore', () => {
+    let bucketName;
+    let mockBucket;
+    let mockFile;
+    let mockWriteStream;
+    let mockStorageInstance;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        bucketName = 'test-bucket';
+        mockWriteStream = {
+            on: vi.fn((event, cb) => {
+                if (event === 'finish')
+                    setTimeout(cb, 0); // Simulate async finish
+                return mockWriteStream;
+            }),
+            destroy: vi.fn(),
+            destroyed: false,
+        };
+        mockFile = {
+            save: vi.fn().mockResolvedValue(undefined),
+            download: vi.fn().mockResolvedValue([Buffer.from('')]),
+            exists: vi.fn().mockResolvedValue([true]),
+            createWriteStream: vi.fn().mockReturnValue(mockWriteStream),
+        };
+        mockBucket = {
+            exists: vi.fn().mockResolvedValue([true]),
+            file: vi.fn().mockReturnValue(mockFile),
+            name: bucketName,
+        };
+        mockStorageInstance = {
+            bucket: vi.fn().mockReturnValue(mockBucket),
+            getBuckets: vi.fn().mockResolvedValue([[{ name: bucketName }]]),
+            createBucket: vi.fn().mockResolvedValue([mockBucket]),
+        };
+        mockStorage.mockReturnValue(mockStorageInstance);
+        mockUuidv4.mockReturnValue('test-uuid');
+        mockSetTargetDir.mockReturnValue('/tmp/workdir');
+        mockGetPersistedState.mockReturnValue({
+            _agentSettings: {},
+            _taskState: 'submitted',
+        });
+        fse.pathExists.mockResolvedValue(true);
+        fsPromises.readdir.mockResolvedValue(['file1.txt']);
+        mockTar.c.mockResolvedValue(undefined);
+        mockTar.x.mockResolvedValue(undefined);
+        mockFse.remove.mockResolvedValue(undefined);
+        mockFse.ensureDir.mockResolvedValue(undefined);
+        mockGzipSync.mockReturnValue(Buffer.from('compressed'));
+        mockGunzipSync.mockReturnValue(Buffer.from('{}'));
+        mockCreateReadStream.mockReturnValue({ on: vi.fn(), pipe: vi.fn() });
+    });
+    describe('Constructor & Initialization', () => {
+        it('should initialize and check bucket existence', async () => {
+            const store = new GCSTaskStore(bucketName);
+            await store['ensureBucketInitialized']();
+            expect(mockStorage).toHaveBeenCalledTimes(1);
+            expect(mockStorageInstance.getBuckets).toHaveBeenCalled();
+            expect(logger.info).toHaveBeenCalledWith(expect.stringContaining('Bucket test-bucket exists'));
+        });
+        it('should create bucket if it does not exist', async () => {
+            mockStorageInstance.getBuckets.mockResolvedValue([[]]);
+            const store = new GCSTaskStore(bucketName);
+            await store['ensureBucketInitialized']();
+            expect(mockStorageInstance.createBucket).toHaveBeenCalledWith(bucketName);
+            expect(logger.info).toHaveBeenCalledWith(expect.stringContaining('Bucket test-bucket created successfully'));
+        });
+        it('should throw if bucket creation fails', async () => {
+            mockStorageInstance.getBuckets.mockResolvedValue([[]]);
+            mockStorageInstance.createBucket.mockRejectedValue(new Error('Create failed'));
+            const store = new GCSTaskStore(bucketName);
+            await expect(store['ensureBucketInitialized']()).rejects.toThrow('Failed to create GCS bucket test-bucket: Error: Create failed');
+        });
+    });
+    describe('save', () => {
+        const mockTask = {
+            id: 'task1',
+            contextId: 'ctx1',
+            kind: 'task',
+            status: { state: 'working' },
+            metadata: {},
+        };
+        it.skip('should save metadata and workspace', async () => {
+            const store = new GCSTaskStore(bucketName);
+            await store.save(mockTask);
+            expect(mockFile.save).toHaveBeenCalledTimes(1);
+            expect(mockTar.c).toHaveBeenCalledTimes(1);
+            expect(mockCreateReadStream).toHaveBeenCalledTimes(1);
+            expect(mockFse.remove).toHaveBeenCalledTimes(1);
+            expect(logger.info).toHaveBeenCalledWith(expect.stringContaining('metadata saved to GCS'));
+            expect(logger.info).toHaveBeenCalledWith(expect.stringContaining('workspace saved to GCS'));
+        });
+        it('should handle tar creation failure', async () => {
+            mockFse.pathExists.mockImplementation(async (path) => !path.toString().includes('task-task1-workspace-test-uuid.tar.gz'));
+            const store = new GCSTaskStore(bucketName);
+            await expect(store.save(mockTask)).rejects.toThrow('tar.c command failed to create');
+        });
+    });
+    describe('load', () => {
+        it('should load task metadata and workspace', async () => {
+            mockGunzipSync.mockReturnValue(Buffer.from(JSON.stringify({
+                [TEST_METADATA_KEY]: {
+                    _agentSettings: {},
+                    _taskState: 'submitted',
+                },
+                _contextId: 'ctx1',
+            })));
+            mockFile.download.mockResolvedValue([Buffer.from('compressed metadata')]);
+            mockFile.download.mockResolvedValueOnce([
+                Buffer.from('compressed metadata'),
+            ]);
+            mockBucket.file = vi.fn((path) => {
+                const newMockFile = { ...mockFile };
+                if (path.includes('metadata')) {
+                    newMockFile.download = vi
+                        .fn()
+                        .mockResolvedValue([Buffer.from('compressed metadata')]);
+                    newMockFile.exists = vi.fn().mockResolvedValue([true]);
+                }
+                else {
+                    newMockFile.download = vi
+                        .fn()
+                        .mockResolvedValue([Buffer.from('compressed workspace')]);
+                    newMockFile.exists = vi.fn().mockResolvedValue([true]);
+                }
+                return newMockFile;
+            });
+            const store = new GCSTaskStore(bucketName);
+            const task = await store.load('task1');
+            expect(task).toBeDefined();
+            expect(task?.id).toBe('task1');
+            expect(mockBucket.file).toHaveBeenCalledWith('tasks/task1/metadata.tar.gz');
+            expect(mockBucket.file).toHaveBeenCalledWith('tasks/task1/workspace.tar.gz');
+            expect(mockTar.x).toHaveBeenCalledTimes(1);
+            expect(mockFse.remove).toHaveBeenCalledTimes(1);
+        });
+        it('should return undefined if metadata not found', async () => {
+            mockFile.exists.mockResolvedValue([false]);
+            const store = new GCSTaskStore(bucketName);
+            const task = await store.load('task1');
+            expect(task).toBeUndefined();
+            expect(mockBucket.file).toHaveBeenCalledWith('tasks/task1/metadata.tar.gz');
+        });
+        it('should load metadata even if workspace not found', async () => {
+            mockGunzipSync.mockReturnValue(Buffer.from(JSON.stringify({
+                [TEST_METADATA_KEY]: {
+                    _agentSettings: {},
+                    _taskState: 'submitted',
+                },
+                _contextId: 'ctx1',
+            })));
+            mockBucket.file = vi.fn((path) => {
+                const newMockFile = { ...mockFile };
+                if (path.includes('workspace.tar.gz')) {
+                    newMockFile.exists = vi.fn().mockResolvedValue([false]);
+                }
+                else {
+                    newMockFile.exists = vi.fn().mockResolvedValue([true]);
+                    newMockFile.download = vi
+                        .fn()
+                        .mockResolvedValue([Buffer.from('compressed metadata')]);
+                }
+                return newMockFile;
+            });
+            const store = new GCSTaskStore(bucketName);
+            const task = await store.load('task1');
+            expect(task).toBeDefined();
+            expect(mockTar.x).not.toHaveBeenCalled();
+            expect(logger.info).toHaveBeenCalledWith(expect.stringContaining('workspace archive not found'));
+        });
+    });
+});
+describe('NoOpTaskStore', () => {
+    let realStore;
+    let noOpStore;
+    beforeEach(() => {
+        // Create a mock of the real store to delegate to
+        realStore = {
+            save: vi.fn(),
+            load: vi.fn().mockResolvedValue({ id: 'task-123' }),
+        };
+        noOpStore = new NoOpTaskStore(realStore);
+    });
+    it("should not call the real store's save method", async () => {
+        const mockTask = { id: 'test-task' };
+        await noOpStore.save(mockTask);
+        expect(realStore.save).not.toHaveBeenCalled();
+    });
+    it('should delegate the load method to the real store', async () => {
+        const taskId = 'task-123';
+        const result = await noOpStore.load(taskId);
+        expect(realStore.load).toHaveBeenCalledWith(taskId);
+        expect(result).toBeDefined();
+        expect(result?.id).toBe(taskId);
+    });
+});
+//# sourceMappingURL=gcs.test.js.map

package/dist/src/persistence/gcs.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gcs.test.js","sourceRoot":"","sources":["../../../src/persistence/gcs.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AAChD,OAAO,KAAK,GAAG,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,QAAQ,IAAI,UAAU,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AACnE,OAAO,KAAK,GAAG,MAAM,KAAK,CAAC;AAC3B,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AAIpC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9D,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACvD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,KAAK,YAAY,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE9D,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AACjC,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IACzB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;IACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;IAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;IACf,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;IAC5B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,YAAY,CAA2B,SAAS,CAAC,CAAC;IAC1E,OAAO;QACL,GAAG,MAAM;QACT,QAAQ,EAAE;YACR,GAAG,MAAM,CAAC,QAAQ;YAClB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB;QACD,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC1B,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACf,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAChB,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,MAAM,EAAE;QACN,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf;CACF,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;CAClB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC9C,MAAM,MAAM,GAAG,MAAM,cAAc,EAAgC,CAAC;IACpE,OAAO;QACL,GAAG,MAAM;QACT,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC3B,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,OAAsC,CAAC;AAC3D,MAAM,OAAO,GAAG,GAAyB,CAAC;AAC1C,MAAM,oBAAoB,GAAG,gBAAwB,CAAC;AACtD,MAAM,OAAO,GAAG,GAAyB,CAAC;AAC1C,MAAM,YAAY,GAAG,QAAgB,CAAC;AACtC,MAAM,cAAc,GAAG,UAAkB,CAAC;AAC1C,MAAM,UAAU,GAAG,MAAc,CAAC;AAClC,MAAM,gBAAgB,GAAG,YAAY,CAAC,YAAoB,CAAC;AAC3D,MAAM,qBAAqB,GAAG,iBAAyB,CAAC;AACxD,MAAM,iBAAiB,GAAG,YAAY,IAAI,kBAAkB,CAAC;AA6B7D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,UAAkB,CAAC;IACvB,IAAI,UAAsB,CAAC;IAC3B,IAAI,QAAkB,CAAC;IACvB,IAAI,eAAgC,CAAC;IACrC,IAAI,mBAAwC,CAAC;IAE7C,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,GAAG,aAAa,CAAC;QAE3B,eAAe,GAAG;YAChB,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;gBACtB,IAAI,KAAK,KAAK,QAAQ;oBAAE,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;gBACnE,OAAO,eAAe,CAAC;YACzB,CAAC,CAAC;YACF,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,QAAQ,GAAG;YACT,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC1C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YACtD,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC;YACzC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,eAAe,CAAC;SAC5D,CAAC;QAEF,UAAU,GAAG;YACX,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC;YACzC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;YACvC,IAAI,EAAE,UAAU;SACjB,CAAC;QAEF,mBAAmB,GAAG;YACpB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC;YAC3C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YAC/D,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,UAAU,CAAC,CAAC;SACtD,CAAC;QACF,WAAW,CAAC,eAAe,CAAC,mBAAyC,CAAC,CAAC;QAEvE,UAAU,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxC,gBAAgB,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACjD,qBAAqB,CAAC,eAAe,CAAC;YACpC,cAAc,EAAE,EAAE;YAClB,UAAU,EAAE,WAAW;SACxB,CAAC,CAAC;QACF,GAAG,CAAC,UAAmB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAChD,UAAU,CAAC,OAAgB,CAAC,iBAAiB,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QAC9D,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACvC,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACvC,OAAO,CAAC,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC5C,OAAO,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC/C,YAAY,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACxD,cAAc,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAClD,oBAAoB,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,yBAAyB,CAAC,EAAE,CAAC;YACzC,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC,2BAA2B,CAAC,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,mBAAmB,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACvD,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,yBAAyB,CAAC,EAAE,CAAC;YACzC,MAAM,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC,yCAAyC,CAAC,CACnE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,mBAAmB,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACvD,mBAAmB,CAAC,YAAY,CAAC,iBAAiB,CAChD,IAAI,KAAK,CAAC,eAAe,CAAC,CAC3B,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9D,+DAA+D,CAChE,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,MAAM,QAAQ,GAAY;YACxB,EAAE,EAAE,OAAO;YACX,SAAS,EAAE,MAAM;YACjB,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;YAC5B,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,EAAE,CAAC,IAAI,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAE3B,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,oBAAoB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACtD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CACjD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,CAClD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,OAAO,CAAC,UAAU,CAAC,kBAAkB,CACnC,KAAK,EAAE,IAAI,EAAE,EAAE,CACb,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC,CACrE,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChD,gCAAgC,CACjC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,cAAc,CAAC,eAAe,CAC5B,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,CAAC,iBAAiB,CAAC,EAAE;oBACnB,cAAc,EAAE,EAAE;oBAClB,UAAU,EAAE,WAAW;iBACxB;gBACD,UAAU,EAAE,MAAM;aACnB,CAAC,CACH,CACF,CAAC;YACF,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC1E,QAAQ,CAAC,QAAQ,CAAC,qBAAqB,CAAC;gBACtC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC;aACnC,CAAC,CAAC;YACH,UAAU,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC/B,MAAM,WAAW,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;gBACpC,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;oBAC9B,WAAW,CAAC,QAAQ,GAAG,EAAE;yBACtB,EAAE,EAAE;yBACJ,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;oBAC3D,WAAW,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzD,CAAC;qBAAM,CAAC;oBACN,WAAW,CAAC,QAAQ,GAAG,EAAE;yBACtB,EAAE,EAAE;yBACJ,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;oBAC5D,WAAW,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzD,CAAC;gBACD,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEvC,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC/B,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC1C,6BAA6B,CAC9B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC1C,8BAA8B,CAC/B,CAAC;YACF,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3C,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YAC7B,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAC1C,6BAA6B,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,cAAc,CAAC,eAAe,CAC5B,MAAM,CAAC,IAAI,CACT,IAAI,CAAC,SAAS,CAAC;gBACb,CAAC,iBAAiB,CAAC,EAAE;oBACnB,cAAc,EAAE,EAAE;oBAClB,UAAU,EAAE,WAAW;iBACxB;gBACD,UAAU,EAAE,MAAM;aACnB,CAAC,CACH,CACF,CAAC;YAEF,UAAU,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC/B,MAAM,WAAW,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;gBACpC,IAAI,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBACtC,WAAW,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC1D,CAAC;qBAAM,CAAC;oBACN,WAAW,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;oBACvD,WAAW,CAAC,QAAQ,GAAG,EAAE;yBACtB,EAAE,EAAE;yBACJ,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEvC,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YAC3B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC,6BAA6B,CAAC,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,SAAoB,CAAC;IACzB,IAAI,SAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,iDAAiD;QACjD,SAAS,GAAG;YACV,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAa,CAAC;SAC/D,CAAC;QACF,SAAS,GAAG,IAAI,aAAa,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,QAAQ,GAAY,EAAE,EAAE,EAAE,WAAW,EAAa,CAAC;QACzD,MAAM,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,MAAM,GAAG,UAAU,CAAC;QAC1B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/types.d.ts

@@ -0,0 +1,91 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { MCPServerStatus, ToolConfirmationOutcome } from '@google/gemini-cli-core';
+import type { TaskState } from '@a2a-js/sdk';
+export declare enum CoderAgentEvent {
+    /**
+     * An event requesting one or more tool call confirmations.
+     */
+    ToolCallConfirmationEvent = "tool-call-confirmation",
+    /**
+     * An event updating on the status of one or more tool calls.
+     */
+    ToolCallUpdateEvent = "tool-call-update",
+    /**
+     * An event providing text updates on the task.
+     */
+    TextContentEvent = "text-content",
+    /**
+     * An event that indicates a change in the task's execution state.
+     */
+    StateChangeEvent = "state-change",
+    /**
+     * An user-sent event to initiate the agent.
+     */
+    StateAgentSettingsEvent = "agent-settings",
+    /**
+     * An event that contains a thought from the agent.
+     */
+    ThoughtEvent = "thought"
+}
+export interface AgentSettings {
+    kind: CoderAgentEvent.StateAgentSettingsEvent;
+    workspacePath: string;
+}
+export interface ToolCallConfirmation {
+    kind: CoderAgentEvent.ToolCallConfirmationEvent;
+}
+export interface ToolCallUpdate {
+    kind: CoderAgentEvent.ToolCallUpdateEvent;
+}
+export interface TextContent {
+    kind: CoderAgentEvent.TextContentEvent;
+}
+export interface StateChange {
+    kind: CoderAgentEvent.StateChangeEvent;
+}
+export interface Thought {
+    kind: CoderAgentEvent.ThoughtEvent;
+}
+export type ThoughtSummary = {
+    subject: string;
+    description: string;
+};
+export interface ToolConfirmationResponse {
+    outcome: ToolConfirmationOutcome;
+    callId: string;
+}
+export type CoderAgentMessage = AgentSettings | ToolCallConfirmation | ToolCallUpdate | TextContent | StateChange | Thought;
+export interface TaskMetadata {
+    id: string;
+    contextId: string;
+    taskState: TaskState;
+    model: string;
+    mcpServers: Array<{
+        name: string;
+        status: MCPServerStatus;
+        tools: Array<{
+            name: string;
+            description: string;
+            parameterSchema: unknown;
+        }>;
+    }>;
+    availableTools: Array<{
+        name: string;
+        description: string;
+        parameterSchema: unknown;
+    }>;
+}
+export interface PersistedStateMetadata {
+    _agentSettings: AgentSettings;
+    _taskState: TaskState;
+}
+export type PersistedTaskMetadata = {
+    [k: string]: unknown;
+};
+export declare const METADATA_KEY = "__persistedState";
+export declare function getPersistedState(metadata: PersistedTaskMetadata): PersistedStateMetadata | undefined;
+export declare function setPersistedState(metadata: PersistedTaskMetadata, state: PersistedStateMetadata): PersistedTaskMetadata;

package/dist/src/types.js

@@ -0,0 +1,44 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+// Interfaces and enums for the CoderAgent protocol.
+export var CoderAgentEvent;
+(function (CoderAgentEvent) {
+    /**
+     * An event requesting one or more tool call confirmations.
+     */
+    CoderAgentEvent["ToolCallConfirmationEvent"] = "tool-call-confirmation";
+    /**
+     * An event updating on the status of one or more tool calls.
+     */
+    CoderAgentEvent["ToolCallUpdateEvent"] = "tool-call-update";
+    /**
+     * An event providing text updates on the task.
+     */
+    CoderAgentEvent["TextContentEvent"] = "text-content";
+    /**
+     * An event that indicates a change in the task's execution state.
+     */
+    CoderAgentEvent["StateChangeEvent"] = "state-change";
+    /**
+     * An user-sent event to initiate the agent.
+     */
+    CoderAgentEvent["StateAgentSettingsEvent"] = "agent-settings";
+    /**
+     * An event that contains a thought from the agent.
+     */
+    CoderAgentEvent["ThoughtEvent"] = "thought";
+})(CoderAgentEvent || (CoderAgentEvent = {}));
+export const METADATA_KEY = '__persistedState';
+export function getPersistedState(metadata) {
+    return metadata?.[METADATA_KEY];
+}
+export function setPersistedState(metadata, state) {
+    return {
+        ...metadata,
+        [METADATA_KEY]: state,
+    };
+}
+//# sourceMappingURL=types.js.map

package/dist/src/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAQH,oDAAoD;AAEpD,MAAM,CAAN,IAAY,eAyBX;AAzBD,WAAY,eAAe;IACzB;;OAEG;IACH,uEAAoD,CAAA;IACpD;;OAEG;IACH,2DAAwC,CAAA;IACxC;;OAEG;IACH,oDAAiC,CAAA;IACjC;;OAEG;IACH,oDAAiC,CAAA;IACjC;;OAEG;IACH,6DAA0C,CAAA;IAC1C;;OAEG;IACH,2CAAwB,CAAA;AAC1B,CAAC,EAzBW,eAAe,KAAf,eAAe,QAyB1B;AAyED,MAAM,CAAC,MAAM,YAAY,GAAG,kBAAkB,CAAC;AAE/C,MAAM,UAAU,iBAAiB,CAC/B,QAA+B;IAE/B,OAAO,QAAQ,EAAE,CAAC,YAAY,CAAuC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,QAA+B,EAC/B,KAA6B;IAE7B,OAAO;QACL,GAAG,QAAQ;QACX,CAAC,YAAY,CAAC,EAAE,KAAK;KACtB,CAAC;AACJ,CAAC"}

package/dist/src/utils/executor_utils.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { ExecutionEventBus } from '@a2a-js/sdk/server';
+export declare function pushTaskStateFailed(error: unknown, eventBus: ExecutionEventBus, taskId: string, contextId: string): Promise<void>;

package/dist/src/utils/executor_utils.js

@@ -0,0 +1,41 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { v4 as uuidv4 } from 'uuid';
+import { CoderAgentEvent } from '../types.js';
+export async function pushTaskStateFailed(error, eventBus, taskId, contextId) {
+    const errorMessage = error instanceof Error ? error.message : 'Agent execution error';
+    const stateChange = {
+        kind: CoderAgentEvent.StateChangeEvent,
+    };
+    eventBus.publish({
+        kind: 'status-update',
+        taskId,
+        contextId,
+        status: {
+            state: 'failed',
+            message: {
+                kind: 'message',
+                role: 'agent',
+                parts: [
+                    {
+                        kind: 'text',
+                        text: errorMessage,
+                    },
+                ],
+                messageId: uuidv4(),
+                taskId,
+                contextId,
+            },
+        },
+        final: true,
+        metadata: {
+            coderAgent: stateChange,
+            model: 'unknown',
+            error: errorMessage,
+        },
+    });
+}
+//# sourceMappingURL=executor_utils.js.map

package/dist/src/utils/executor_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor_utils.js","sourceRoot":"","sources":["../../../src/utils/executor_utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AAEpC,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAG9C,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,KAAc,EACd,QAA2B,EAC3B,MAAc,EACd,SAAiB;IAEjB,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAAC;IACnE,MAAM,WAAW,GAAgB;QAC/B,IAAI,EAAE,eAAe,CAAC,gBAAgB;KACvC,CAAC;IACF,QAAQ,CAAC,OAAO,CAAC;QACf,IAAI,EAAE,eAAe;QACrB,MAAM;QACN,SAAS;QACT,MAAM,EAAE;YACN,KAAK,EAAE,QAAQ;YACf,OAAO,EAAE;gBACP,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,YAAY;qBACnB;iBACF;gBACD,SAAS,EAAE,MAAM,EAAE;gBACnB,MAAM;gBACN,SAAS;aACC;SACb;QACD,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE;YACR,UAAU,EAAE,WAAW;YACvB,KAAK,EAAE,SAAS;YAChB,KAAK,EAAE,YAAY;SACpB;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/src/utils/logger.d.ts

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import winston from 'winston';
+declare const logger: winston.Logger;
+export { logger };

package/dist/src/utils/logger.js

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import winston from 'winston';
+const logger = winston.createLogger({
+    level: 'info',
+    format: winston.format.combine(
+    // First, add a timestamp to the log info object
+    winston.format.timestamp({
+        format: 'YYYY-MM-DD HH:mm:ss.SSS A', // Custom timestamp format
+    }), 
+    // Here we define the custom output format
+    winston.format.printf((info) => {
+        const { level, timestamp, message, ...rest } = info;
+        return (`[${level.toUpperCase()}] ${timestamp} -- ${message}` +
+            `${Object.keys(rest).length > 0 ? `\n${JSON.stringify(rest, null, 2)}` : ''}`); // Only print ...rest if present
+    })),
+    transports: [new winston.transports.Console()],
+});
+export { logger };
+//# sourceMappingURL=logger.js.map

package/dist/src/utils/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../../src/utils/logger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,OAAO,MAAM,SAAS,CAAC;AAE9B,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAClC,KAAK,EAAE,MAAM;IACb,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO;IAC5B,gDAAgD;IAChD,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;QACvB,MAAM,EAAE,2BAA2B,EAAE,0BAA0B;KAChE,CAAC;IACF,0CAA0C;IAC1C,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QAC7B,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;QACpD,OAAO,CACL,IAAI,KAAK,CAAC,WAAW,EAAE,KAAK,SAAS,OAAO,OAAO,EAAE;YACrD,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAC9E,CAAC,CAAC,gCAAgC;IACrC,CAAC,CAAC,CACH;IACD,UAAU,EAAE,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;CAC/C,CAAC,CAAC;AAEH,OAAO,EAAE,MAAM,EAAE,CAAC"}

package/dist/src/utils/testing_utils.d.ts

@@ -0,0 +1,33 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { SendStreamingMessageSuccessResponse } from '@a2a-js/sdk';
+import type { Config } from '@google/gemini-cli-core';
+export declare function createMockConfig(overrides?: Partial<Config>): Partial<Config>;
+export declare function createStreamMessageRequest(text: string, messageId: string, taskId?: string): {
+    jsonrpc: string;
+    id: string;
+    method: string;
+    params: {
+        message: {
+            kind: string;
+            role: string;
+            parts: [{
+                kind: string;
+                text: string;
+            }];
+            messageId: string;
+        };
+        metadata: {
+            coderAgent: {
+                kind: string;
+                workspacePath: string;
+            };
+        };
+        taskId?: string;
+    };
+};
+export declare function assertUniqueFinalEventIsLast(events: SendStreamingMessageSuccessResponse[]): void;
+export declare function assertTaskCreationAndWorkingStatus(events: SendStreamingMessageSuccessResponse[]): void;

package/dist/src/utils/testing_utils.js

@@ -0,0 +1,91 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ApprovalMode, DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES, DEFAULT_TRUNCATE_TOOL_OUTPUT_THRESHOLD, GeminiClient, } from '@google/gemini-cli-core';
+import { expect, vi } from 'vitest';
+export function createMockConfig(overrides = {}) {
+    const mockConfig = {
+        getToolRegistry: vi.fn().mockReturnValue({
+            getTool: vi.fn(),
+            getAllToolNames: vi.fn().mockReturnValue([]),
+        }),
+        getApprovalMode: vi.fn().mockReturnValue(ApprovalMode.DEFAULT),
+        getIdeMode: vi.fn().mockReturnValue(false),
+        getAllowedTools: vi.fn().mockReturnValue([]),
+        getWorkspaceContext: vi.fn().mockReturnValue({
+            isPathWithinWorkspace: () => true,
+        }),
+        getTargetDir: () => '/test',
+        storage: {
+            getProjectTempDir: () => '/tmp',
+        },
+        getTruncateToolOutputThreshold: () => DEFAULT_TRUNCATE_TOOL_OUTPUT_THRESHOLD,
+        getTruncateToolOutputLines: () => DEFAULT_TRUNCATE_TOOL_OUTPUT_LINES,
+        getDebugMode: vi.fn().mockReturnValue(false),
+        getContentGeneratorConfig: vi.fn().mockReturnValue({ model: 'gemini-pro' }),
+        getModel: vi.fn().mockReturnValue('gemini-pro'),
+        getUsageStatisticsEnabled: vi.fn().mockReturnValue(false),
+        setFallbackModelHandler: vi.fn(),
+        initialize: vi.fn().mockResolvedValue(undefined),
+        getProxy: vi.fn().mockReturnValue(undefined),
+        getHistory: vi.fn().mockReturnValue([]),
+        getEmbeddingModel: vi.fn().mockReturnValue('text-embedding-004'),
+        getSessionId: vi.fn().mockReturnValue('test-session-id'),
+        getUserTier: vi.fn(),
+        ...overrides,
+    };
+    mockConfig.getGeminiClient = vi
+        .fn()
+        .mockReturnValue(new GeminiClient(mockConfig));
+    return mockConfig;
+}
+export function createStreamMessageRequest(text, messageId, taskId) {
+    const request = {
+        jsonrpc: '2.0',
+        id: '1',
+        method: 'message/stream',
+        params: {
+            message: {
+                kind: 'message',
+                role: 'user',
+                parts: [{ kind: 'text', text }],
+                messageId,
+            },
+            metadata: {
+                coderAgent: {
+                    kind: 'agent-settings',
+                    workspacePath: '/tmp',
+                },
+            },
+        },
+    };
+    if (taskId) {
+        request.params.taskId = taskId;
+    }
+    return request;
+}
+export function assertUniqueFinalEventIsLast(events) {
+    // Final event is input-required & final
+    const finalEvent = events[events.length - 1].result;
+    expect(finalEvent.metadata?.['coderAgent']).toMatchObject({
+        kind: 'state-change',
+    });
+    expect(finalEvent.status?.state).toBe('input-required');
+    expect(finalEvent.final).toBe(true);
+    // There is only one event with final and its the last
+    expect(events.filter((e) => e.result.final).length).toBe(1);
+    expect(events.findIndex((e) => e.result.final)).toBe(events.length - 1);
+}
+export function assertTaskCreationAndWorkingStatus(events) {
+    // Initial task creation event
+    const taskEvent = events[0].result;
+    expect(taskEvent.kind).toBe('task');
+    expect(taskEvent.status.state).toBe('submitted');
+    // Status update: working
+    const workingEvent = events[1].result;
+    expect(workingEvent.kind).toBe('status-update');
+    expect(workingEvent.status.state).toBe('working');
+}
+//# sourceMappingURL=testing_utils.js.map

package/dist/src/utils/testing_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testing_utils.js","sourceRoot":"","sources":["../../../src/utils/testing_utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EACL,YAAY,EACZ,kCAAkC,EAClC,sCAAsC,EACtC,YAAY,GACb,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEpC,MAAM,UAAU,gBAAgB,CAC9B,YAA6B,EAAE;IAE/B,MAAM,UAAU,GAAG;QACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YACvC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;SAC7C,CAAC;QACF,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC;QAC9D,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC1C,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QAC5C,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC3C,qBAAqB,EAAE,GAAG,EAAE,CAAC,IAAI;SAClC,CAAC;QACF,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO;QAC3B,OAAO,EAAE;YACP,iBAAiB,EAAE,GAAG,EAAE,CAAC,MAAM;SACrB;QACZ,8BAA8B,EAAE,GAAG,EAAE,CACnC,sCAAsC;QACxC,0BAA0B,EAAE,GAAG,EAAE,CAAC,kCAAkC;QACpE,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QAC5C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;QAC3E,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC;QAC/C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;QACzD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE;QAChC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAChD,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC;QAC5C,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;QACvC,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,oBAAoB,CAAC;QAChE,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,iBAAiB,CAAC;QACxD,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;QACpB,GAAG,SAAS;KACQ,CAAC;IAEvB,UAAU,CAAC,eAAe,GAAG,EAAE;SAC5B,EAAE,EAAE;SACJ,eAAe,CAAC,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC;IACjD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,IAAY,EACZ,SAAiB,EACjB,MAAe;IAEf,MAAM,OAAO,GAmBT;QACF,OAAO,EAAE,KAAK;QACd,EAAE,EAAE,GAAG;QACP,MAAM,EAAE,gBAAgB;QACxB,MAAM,EAAE;YACN,OAAO,EAAE;gBACP,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;gBAC/B,SAAS;aACV;YACD,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV,IAAI,EAAE,gBAAgB;oBACtB,aAAa,EAAE,MAAM;iBACtB;aACF;SACF;KACF,CAAC;IAEF,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;IACjC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,MAA6C;IAE7C,wCAAwC;IACxC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAA+B,CAAC;IAC7E,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,aAAa,CAAC;QACxD,IAAI,EAAE,cAAc;KACrB,CAAC,CAAC;IACH,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACxD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEpC,sDAAsD;IACtD,MAAM,CACJ,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,MAAgC,CAAC,KAAK,CAAC,CAAC,MAAM,CACvE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACV,MAAM,CACJ,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,MAAgC,CAAC,KAAK,CAAC,CACnE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,kCAAkC,CAChD,MAA6C;IAE7C,8BAA8B;IAC9B,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAAiB,CAAC;IAC9C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAEjD,yBAAyB;IACzB,MAAM,YAAY,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,MAA+B,CAAC;IAC/D,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAChD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACpD,CAAC"}