@theia/ai-ide 1.67.0-next.56 → 1.67.0-next.86

package/lib/browser/context-functions.spec.js

@@ -0,0 +1,213 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+const jsdom_1 = require("@theia/core/lib/browser/test/jsdom");
+let disableJSDOM = (0, jsdom_1.enableJSDOM)();
+const frontend_application_config_provider_1 = require("@theia/core/lib/browser/frontend-application-config-provider");
+frontend_application_config_provider_1.FrontendApplicationConfigProvider.set({});
+const chai_1 = require("chai");
+const context_functions_1 = require("./context-functions");
+const core_1 = require("@theia/core");
+const assert_1 = require("assert");
+const context_file_validation_service_1 = require("@theia/ai-chat/lib/browser/context-file-validation-service");
+disableJSDOM();
+describe('Context Functions Cancellation Tests', () => {
+    let cancellationTokenSource;
+    let mockCtx;
+    before(() => {
+        disableJSDOM = (0, jsdom_1.enableJSDOM)();
+    });
+    after(() => {
+        // Disable JSDOM after all tests
+        disableJSDOM();
+    });
+    beforeEach(() => {
+        cancellationTokenSource = new core_1.CancellationTokenSource();
+        const context = {
+            addVariables: () => { },
+            getVariables: () => { var _a; return (_a = mockCtx.context) === null || _a === void 0 ? void 0 : _a.variables; }
+        };
+        mockCtx = {
+            response: {
+                cancellationToken: cancellationTokenSource.token
+            },
+            context: {
+                variables: [{
+                        variable: { id: 'file1', name: 'File' },
+                        arg: '/path/to/file',
+                        contextValue: 'file content'
+                    }]
+            },
+            session: {
+                context
+            }
+        };
+    });
+    afterEach(() => {
+        cancellationTokenSource.dispose();
+    });
+    it('ListChatContext should respect cancellation token', async () => {
+        const listChatContext = new context_functions_1.ListChatContext();
+        cancellationTokenSource.cancel();
+        const result = await listChatContext.getTool().handler('', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+    it('ResolveChatContext should respect cancellation token', async () => {
+        const resolveChatContext = new context_functions_1.ResolveChatContext();
+        cancellationTokenSource.cancel();
+        const result = await resolveChatContext.getTool().handler('{"contextElementId":"file1/path/to/file"}', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+    it('AddFileToChatContext should respect cancellation token', async () => {
+        const addFileToChatContext = new context_functions_1.AddFileToChatContext();
+        cancellationTokenSource.cancel();
+        const result = await addFileToChatContext.getTool().handler('{"filesToAdd":["/new/path/to/file"]}', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+});
+describe('AddFileToChatContext Validation Tests', () => {
+    let mockCtx;
+    let addedFiles;
+    before(() => {
+        disableJSDOM = (0, jsdom_1.enableJSDOM)();
+    });
+    after(() => {
+        disableJSDOM();
+    });
+    beforeEach(() => {
+        addedFiles = [];
+        const context = {
+            addVariables: (...vars) => {
+                addedFiles.push(...vars);
+            },
+            getVariables: () => []
+        };
+        mockCtx = {
+            response: {
+                cancellationToken: new core_1.CancellationTokenSource().token
+            },
+            context: {
+                variables: []
+            },
+            session: {
+                context
+            }
+        };
+    });
+    it('should add valid files to context', async () => {
+        const mockValidationService = {
+            validateFile: async () => ({ state: context_file_validation_service_1.FileValidationState.VALID })
+        };
+        const addFileToChatContext = new context_functions_1.AddFileToChatContext();
+        addFileToChatContext.validationService = mockValidationService;
+        const result = await addFileToChatContext.getTool().handler('{"filesToAdd":["/valid/file1.ts","/valid/file2.ts"]}', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.added).to.have.lengthOf(2);
+        (0, chai_1.expect)(jsonResponse.added).to.include('/valid/file1.ts');
+        (0, chai_1.expect)(jsonResponse.added).to.include('/valid/file2.ts');
+        (0, chai_1.expect)(jsonResponse.rejected).to.have.lengthOf(0);
+        (0, chai_1.expect)(jsonResponse.summary.totalRequested).to.equal(2);
+        (0, chai_1.expect)(jsonResponse.summary.added).to.equal(2);
+        (0, chai_1.expect)(jsonResponse.summary.rejected).to.equal(0);
+        (0, chai_1.expect)(addedFiles).to.have.lengthOf(2);
+    });
+    it('should reject non-existent files', async () => {
+        const mockValidationService = {
+            validateFile: async (file) => {
+                if (file === '/nonexistent/file.ts') {
+                    return {
+                        state: context_file_validation_service_1.FileValidationState.INVALID_NOT_FOUND,
+                        message: 'File does not exist'
+                    };
+                }
+                return { state: context_file_validation_service_1.FileValidationState.VALID };
+            }
+        };
+        const addFileToChatContext = new context_functions_1.AddFileToChatContext();
+        addFileToChatContext.validationService = mockValidationService;
+        const result = await addFileToChatContext.getTool().handler('{"filesToAdd":["/valid/file.ts","/nonexistent/file.ts"]}', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.added).to.have.lengthOf(1);
+        (0, chai_1.expect)(jsonResponse.added).to.include('/valid/file.ts');
+        (0, chai_1.expect)(jsonResponse.rejected).to.have.lengthOf(1);
+        (0, chai_1.expect)(jsonResponse.rejected[0].file).to.equal('/nonexistent/file.ts');
+        (0, chai_1.expect)(jsonResponse.rejected[0].reason).to.equal('File does not exist');
+        (0, chai_1.expect)(jsonResponse.rejected[0].state).to.equal(context_file_validation_service_1.FileValidationState.INVALID_NOT_FOUND);
+        (0, chai_1.expect)(jsonResponse.summary.totalRequested).to.equal(2);
+        (0, chai_1.expect)(jsonResponse.summary.added).to.equal(1);
+        (0, chai_1.expect)(jsonResponse.summary.rejected).to.equal(1);
+        (0, chai_1.expect)(addedFiles).to.have.lengthOf(1);
+    });
+    it('should reject files in secondary workspace roots', async () => {
+        const mockValidationService = {
+            validateFile: async (file) => {
+                if (file === '/secondary/root/file.ts') {
+                    return {
+                        state: context_file_validation_service_1.FileValidationState.INVALID_SECONDARY,
+                        message: 'File is in a secondary workspace root. AI agents can only access files in the first workspace root.'
+                    };
+                }
+                return { state: context_file_validation_service_1.FileValidationState.VALID };
+            }
+        };
+        const addFileToChatContext = new context_functions_1.AddFileToChatContext();
+        addFileToChatContext.validationService = mockValidationService;
+        const result = await addFileToChatContext.getTool().handler('{"filesToAdd":["/secondary/root/file.ts"]}', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.added).to.have.lengthOf(0);
+        (0, chai_1.expect)(jsonResponse.rejected).to.have.lengthOf(1);
+        (0, chai_1.expect)(jsonResponse.rejected[0].file).to.equal('/secondary/root/file.ts');
+        (0, chai_1.expect)(jsonResponse.rejected[0].state).to.equal(context_file_validation_service_1.FileValidationState.INVALID_SECONDARY);
+        (0, chai_1.expect)(addedFiles).to.have.lengthOf(0);
+    });
+    it('should add all files when validation service is not available', async () => {
+        const addFileToChatContext = new context_functions_1.AddFileToChatContext();
+        const result = await addFileToChatContext.getTool().handler('{"filesToAdd":["/file1.ts","/file2.ts"]}', mockCtx);
+        if (typeof result !== 'string') {
+            (0, assert_1.fail)(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        (0, chai_1.expect)(jsonResponse.added).to.have.lengthOf(2);
+        (0, chai_1.expect)(jsonResponse.rejected).to.have.lengthOf(0);
+        (0, chai_1.expect)(jsonResponse.summary.totalRequested).to.equal(2);
+        (0, chai_1.expect)(jsonResponse.summary.added).to.equal(2);
+        (0, chai_1.expect)(jsonResponse.summary.rejected).to.equal(0);
+        (0, chai_1.expect)(addedFiles).to.have.lengthOf(2);
+    });
+});
+//# sourceMappingURL=context-functions.spec.js.map

package/lib/browser/context-functions.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-functions.spec.js","sourceRoot":"","sources":["../../src/browser/context-functions.spec.ts"],"names":[],"mappings":";;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,8DAAiE;AACjE,IAAI,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;AACjC,uHAAiH;AACjH,wEAAiC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAC1C,+BAA8B;AAC9B,2DAAgG;AAChG,sCAAsD;AAEtD,mCAA8B;AAE9B,gHAA+H;AAC/H,YAAY,EAAE,CAAC;AAEf,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IAClD,IAAI,uBAAgD,CAAC;IACrD,IAAI,OAAyC,CAAC;IAE9C,MAAM,CAAC,GAAG,EAAE;QACR,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,GAAG,EAAE;QACP,gCAAgC;QAChC,YAAY,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACZ,uBAAuB,GAAG,IAAI,8BAAuB,EAAE,CAAC;QACxD,MAAM,OAAO,GAAgC;YACzC,YAAY,EAAE,GAAG,EAAE,GAAG,CAAC;YACvB,YAAY,EAAE,GAAG,EAAE,WAAC,OAAA,MAAA,OAAO,CAAC,OAAO,0CAAE,SAAwC,CAAA,EAAA;SAChF,CAAC;QACF,OAAO,GAAG;YACN,QAAQ,EAAE;gBACN,iBAAiB,EAAE,uBAAuB,CAAC,KAAK;aACvB;YAC7B,OAAO,EAAE;gBACL,SAAS,EAAE,CAAC;wBACR,QAAQ,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE;wBACvC,GAAG,EAAE,eAAe;wBACpB,YAAY,EAAE,cAAc;qBACF,CAAC;aAClC;YACD,OAAO,EAAE;gBACL,OAAO;aACU;SACxB,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACX,uBAAuB,CAAC,OAAO,EAAE,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,eAAe,GAAG,IAAI,mCAAe,EAAE,CAAC;QAC9C,uBAAuB,CAAC,MAAM,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpE,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,kBAAkB,GAAG,IAAI,sCAAkB,EAAE,CAAC;QACpD,uBAAuB,CAAC,MAAM,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,2CAA2C,EAAE,OAAO,CAAC,CAAC;QAChH,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,oBAAoB,GAAG,IAAI,wCAAoB,EAAE,CAAC;QACxD,uBAAuB,CAAC,MAAM,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,sCAAsC,EAAE,OAAO,CAAC,CAAC;QAC7G,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACnD,IAAI,OAAyC,CAAC;IAC9C,IAAI,UAAyC,CAAC;IAE9C,MAAM,CAAC,GAAG,EAAE;QACR,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,GAAG,EAAE;QACP,YAAY,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACZ,UAAU,GAAG,EAAE,CAAC;QAChB,MAAM,OAAO,GAAgC;YACzC,YAAY,EAAE,CAAC,GAAG,IAAmC,EAAE,EAAE;gBACrD,UAAU,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;YAC7B,CAAC;YACD,YAAY,EAAE,GAAG,EAAE,CAAC,EAAE;SACzB,CAAC;QACF,OAAO,GAAG;YACN,QAAQ,EAAE;gBACN,iBAAiB,EAAE,IAAI,8BAAuB,EAAE,CAAC,KAAK;aAC7B;YAC7B,OAAO,EAAE;gBACL,SAAS,EAAE,EAAE;aAChB;YACD,OAAO,EAAE;gBACL,OAAO;aACU;SACxB,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,qBAAqB,GAAiC;YACxD,YAAY,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,qDAAmB,CAAC,KAAK,EAAE,CAAC;SACnE,CAAC;QAEF,MAAM,oBAAoB,GAAG,IAAI,wCAAoB,EAAE,CAAC;QACvD,oBAAuF,CAAC,iBAAiB,GAAG,qBAAqB,CAAC;QAEnI,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,CAAC,OAAO,CACvD,sDAAsD,EACtD,OAAO,CACV,CAAC;QAEF,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACzD,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACzD,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxD,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,UAAU,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,qBAAqB,GAAiC;YACxD,YAAY,EAAE,KAAK,EAAC,IAAI,EAAC,EAAE;gBACvB,IAAI,IAAI,KAAK,sBAAsB,EAAE,CAAC;oBAClC,OAAO;wBACH,KAAK,EAAE,qDAAmB,CAAC,iBAAiB;wBAC5C,OAAO,EAAE,qBAAqB;qBACjC,CAAC;gBACN,CAAC;gBACD,OAAO,EAAE,KAAK,EAAE,qDAAmB,CAAC,KAAK,EAAE,CAAC;YAChD,CAAC;SACJ,CAAC;QAEF,MAAM,oBAAoB,GAAG,IAAI,wCAAoB,EAAE,CAAC;QACvD,oBAAuF,CAAC,iBAAiB,GAAG,qBAAqB,CAAC;QAEnI,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,CAAC,OAAO,CACvD,0DAA0D,EAC1D,OAAO,CACV,CAAC;QAEF,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACxD,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACvE,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACxE,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,qDAAmB,CAAC,iBAAiB,CAAC,CAAC;QACvF,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxD,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,UAAU,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,qBAAqB,GAAiC;YACxD,YAAY,EAAE,KAAK,EAAC,IAAI,EAAC,EAAE;gBACvB,IAAI,IAAI,KAAK,yBAAyB,EAAE,CAAC;oBACrC,OAAO;wBACH,KAAK,EAAE,qDAAmB,CAAC,iBAAiB;wBAC5C,OAAO,EAAE,qGAAqG;qBACjH,CAAC;gBACN,CAAC;gBACD,OAAO,EAAE,KAAK,EAAE,qDAAmB,CAAC,KAAK,EAAE,CAAC;YAChD,CAAC;SACJ,CAAC;QAEF,MAAM,oBAAoB,GAAG,IAAI,wCAAoB,EAAE,CAAC;QACvD,oBAAuF,CAAC,iBAAiB,GAAG,qBAAqB,CAAC;QAEnI,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,CAAC,OAAO,CACvD,4CAA4C,EAC5C,OAAO,CACV,CAAC;QAEF,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC1E,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,qDAAmB,CAAC,iBAAiB,CAAC,CAAC;QACvF,IAAA,aAAM,EAAC,UAAU,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,oBAAoB,GAAG,IAAI,wCAAoB,EAAE,CAAC;QAExD,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,CAAC,OAAO,CACvD,0CAA0C,EAC1C,OAAO,CACV,CAAC;QAEF,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAA,aAAI,EAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACxC,IAAA,aAAM,EAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxD,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/C,IAAA,aAAM,EAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClD,IAAA,aAAM,EAAC,UAAU,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/lib/browser/file-changeset-function.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=file-changeset-function.spec.d.ts.map

package/lib/browser/file-changeset-function.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-changeset-function.spec.d.ts","sourceRoot":"","sources":["../../src/browser/file-changeset-function.spec.ts"],"names":[],"mappings":""}

package/lib/browser/file-changeset-function.spec.js

@@ -0,0 +1,45 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2025 Lonti.com Pty Ltd.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+const frontend_application_config_provider_1 = require("@theia/core/lib/browser/frontend-application-config-provider");
+const jsdom_1 = require("@theia/core/lib/browser/test/jsdom");
+let disableJSDOM = (0, jsdom_1.enableJSDOM)();
+frontend_application_config_provider_1.FrontendApplicationConfigProvider.set({});
+const inversify_1 = require("@theia/core/shared/inversify");
+const chai_1 = require("chai");
+const file_changeset_functions_1 = require("./file-changeset-functions");
+disableJSDOM();
+describe('DefaultFileChangeSetTitleProvider', () => {
+    let provider;
+    before(() => {
+        const container = new inversify_1.Container();
+        container.bind(file_changeset_functions_1.DefaultFileChangeSetTitleProvider).toSelf();
+        provider = container.get(file_changeset_functions_1.DefaultFileChangeSetTitleProvider);
+        disableJSDOM = (0, jsdom_1.enableJSDOM)();
+    });
+    after(() => {
+        disableJSDOM();
+    });
+    it('should provide the title', () => {
+        const ctx = {
+            agentId: 'test-agent',
+        };
+        const title = provider.getChangeSetTitle(ctx);
+        (0, chai_1.expect)(title).to.equal('Changes proposed');
+    });
+});
+//# sourceMappingURL=file-changeset-function.spec.js.map

package/lib/browser/file-changeset-function.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-changeset-function.spec.js","sourceRoot":"","sources":["../../src/browser/file-changeset-function.spec.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,wCAAwC;AACxC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;AAEhF,uHAAiH;AACjH,8DAAiE;AACjE,IAAI,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;AACjC,wEAAiC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAG1C,4DAAyD;AACzD,+BAA8B;AAC9B,yEAA+E;AAE/E,YAAY,EAAE,CAAC;AAEf,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAC/C,IAAI,QAA2C,CAAC;IAEhD,MAAM,CAAC,GAAG,EAAE;QACR,MAAM,SAAS,GAAG,IAAI,qBAAS,EAAE,CAAC;QAClC,SAAS,CAAC,IAAI,CAAC,4DAAiC,CAAC,CAAC,MAAM,EAAE,CAAC;QAE3D,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,4DAAiC,CAAC,CAAC;QAC5D,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE;QACP,YAAY,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAChC,MAAM,GAAG,GAAG;YACR,OAAO,EAAE,YAAY;SACG,CAAC;QAE7B,MAAM,KAAK,GAAG,QAAQ,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAC9C,IAAA,aAAM,EAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/lib/browser/file-changeset-functions.d.ts

@@ -0,0 +1,87 @@
+import { MutableChatRequestModel } from '@theia/ai-chat';
+import { ChangeSetFileElementFactory } from '@theia/ai-chat/lib/browser/change-set-file-element';
+import { ToolProvider, ToolRequest, ToolRequestParameters } from '@theia/ai-core';
+import { ContentReplacer } from '@theia/core/lib/common/content-replacer';
+import { FileService } from '@theia/filesystem/lib/browser/file-service';
+import { WorkspaceFunctionScope } from './workspace-functions';
+export declare const FileChangeSetTitleProvider: unique symbol;
+export interface FileChangeSetTitleProvider {
+    getChangeSetTitle(ctx: MutableChatRequestModel): string;
+}
+export declare class SuggestFileContent implements ToolProvider {
+    static ID: string;
+    protected readonly workspaceFunctionScope: WorkspaceFunctionScope;
+    fileService: FileService;
+    protected readonly fileChangeFactory: ChangeSetFileElementFactory;
+    protected readonly fileChangeSetTitleProvider: FileChangeSetTitleProvider;
+    getTool(): ToolRequest;
+}
+export declare class WriteFileContent implements ToolProvider {
+    static ID: string;
+    protected readonly workspaceFunctionScope: WorkspaceFunctionScope;
+    fileService: FileService;
+    protected readonly fileChangeFactory: ChangeSetFileElementFactory;
+    protected readonly fileChangeSetTitleProvider: FileChangeSetTitleProvider;
+    getTool(): ToolRequest;
+}
+export declare class ReplaceContentInFileFunctionHelper {
+    protected readonly workspaceFunctionScope: WorkspaceFunctionScope;
+    fileService: FileService;
+    protected readonly fileChangeFactory: ChangeSetFileElementFactory;
+    protected readonly fileChangeSetTitleProvider: FileChangeSetTitleProvider;
+    private replacer;
+    constructor();
+    protected setReplacer(replacer: ContentReplacer): void;
+    getToolMetadata(supportMultipleReplace?: boolean, immediateApplication?: boolean): {
+        description: string;
+        parameters: ToolRequestParameters;
+    };
+    createChangesetFromToolCall(toolCallString: string, ctx: MutableChatRequestModel): Promise<string>;
+    writeChangesetFromToolCall(toolCallString: string, ctx: MutableChatRequestModel): Promise<string>;
+    private processReplacementsCommon;
+    private findExistingChangeElement;
+    clearFileChanges(path: string, ctx: MutableChatRequestModel): Promise<string>;
+    getProposedFileState(path: string, ctx: MutableChatRequestModel): Promise<string>;
+}
+export declare class SimpleSuggestFileReplacements implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelper;
+    getTool(): ToolRequest;
+}
+export declare class SimpleWriteFileReplacements implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelper;
+    getTool(): ToolRequest;
+}
+export declare class SuggestFileReplacements_Simple implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelper;
+    getTool(): ToolRequest;
+}
+export declare class WriteFileReplacements implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelper;
+    getTool(): ToolRequest;
+}
+export declare class ClearFileChanges implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelper;
+    getTool(): ToolRequest;
+}
+export declare class GetProposedFileState implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelper;
+    getTool(): ToolRequest;
+}
+export declare class ReplaceContentInFileFunctionHelperV2 extends ReplaceContentInFileFunctionHelper {
+    constructor();
+}
+export declare class SuggestFileReplacements implements ToolProvider {
+    static ID: string;
+    protected readonly replaceContentInFileFunctionHelper: ReplaceContentInFileFunctionHelperV2;
+    getTool(): ToolRequest;
+}
+export declare class DefaultFileChangeSetTitleProvider implements FileChangeSetTitleProvider {
+    getChangeSetTitle(ctx: MutableChatRequestModel): string;
+}
+//# sourceMappingURL=file-changeset-functions.d.ts.map

package/lib/browser/file-changeset-functions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-changeset-functions.d.ts","sourceRoot":"","sources":["../../src/browser/file-changeset-functions.ts"],"names":[],"mappings":"AAeA,OAAO,EAAa,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AACpE,OAAO,EAA8C,2BAA2B,EAAE,MAAM,oDAAoD,CAAC;AAC7I,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,qBAAqB,EAAmC,MAAM,gBAAgB,CAAC;AACnH,OAAO,EAAsC,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAI9G,OAAO,EAAE,WAAW,EAAE,MAAM,4CAA4C,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAa/D,eAAO,MAAM,0BAA0B,eAAuC,CAAC;AAE/E,MAAM,WAAW,0BAA0B;IACvC,iBAAiB,CAAC,GAAG,EAAE,uBAAuB,GAAG,MAAM,CAAC;CAC3D;AAED,qBACa,kBAAmB,YAAW,YAAY;IACnD,MAAM,CAAC,EAAE,SAA2B;IAGpC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE,sBAAsB,CAAC;IAGlE,WAAW,EAAE,WAAW,CAAC;IAGzB,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,2BAA2B,CAAC;IAGlE,SAAS,CAAC,QAAQ,CAAC,0BAA0B,EAAE,0BAA0B,CAAC;IAE1E,OAAO,IAAI,WAAW;CAqDzB;AAED,qBACa,gBAAiB,YAAW,YAAY;IACjD,MAAM,CAAC,EAAE,SAAyB;IAGlC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE,sBAAsB,CAAC;IAGlE,WAAW,EAAE,WAAW,CAAC;IAGzB,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,2BAA2B,CAAC;IAGlE,SAAS,CAAC,QAAQ,CAAC,0BAA0B,EAAE,0BAA0B,CAAC;IAE1E,OAAO,IAAI,WAAW;CA8DzB;AAED,qBACa,kCAAkC;IAE3C,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE,sBAAsB,CAAC;IAGlE,WAAW,EAAE,WAAW,CAAC;IAGzB,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,2BAA2B,CAAC;IAGlE,SAAS,CAAC,QAAQ,CAAC,0BAA0B,EAAE,0BAA0B,CAAC;IAE1E,OAAO,CAAC,QAAQ,CAAkB;;IAMlC,SAAS,CAAC,WAAW,CAAC,QAAQ,EAAE,eAAe,GAAG,IAAI;IAItD,eAAe,CAAC,sBAAsB,GAAE,OAAe,EAAE,oBAAoB,GAAE,OAAe,GAAG;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,qBAAqB,CAAA;KAAE;IAkErJ,2BAA2B,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,uBAAuB,GAAG,OAAO,CAAC,MAAM,CAAC;IAwBlG,0BAA0B,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,uBAAuB,GAAG,OAAO,CAAC,MAAM,CAAC;YA2BzF,yBAAyB;IA2DvC,OAAO,CAAC,yBAAyB;IAO3B,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,uBAAuB,GAAG,OAAO,CAAC,MAAM,CAAC;IAkB7E,oBAAoB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,uBAAuB,GAAG,OAAO,CAAC,MAAM,CAAC;CA2B1F;AAED,qBACa,6BAA8B,YAAW,YAAY;IAC9D,MAAM,CAAC,EAAE,SAAmC;IAE5C,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,kCAAkC,CAAC;IAE1F,OAAO,IAAI,WAAW;CAezB;AAED,qBACa,2BAA4B,YAAW,YAAY;IAC5D,MAAM,CAAC,EAAE,SAAiC;IAE1C,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,kCAAkC,CAAC;IAE1F,OAAO,IAAI,WAAW;CAezB;AAED,qBACa,8BAA+B,YAAW,YAAY;IAC/D,MAAM,CAAC,EAAE,SAAuC;IAEhD,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,kCAAkC,CAAC;IAE1F,OAAO,IAAI,WAAW;CAezB;AAED,qBACa,qBAAsB,YAAW,YAAY;IACtD,MAAM,CAAC,EAAE,SAA8B;IAEvC,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,kCAAkC,CAAC;IAE1F,OAAO,IAAI,WAAW;CAezB;AAED,qBACa,gBAAiB,YAAW,YAAY;IACjD,MAAM,CAAC,EAAE,SAAyB;IAElC,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,kCAAkC,CAAC;IAE1F,OAAO,IAAI,WAAW;CAwBzB;AAED,qBACa,oBAAqB,YAAW,YAAY;IACrD,MAAM,CAAC,EAAE,SAA2B;IAEpC,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,kCAAkC,CAAC;IAE1F,OAAO,IAAI,WAAW;CAyBzB;AAED,qBACa,oCAAqC,SAAQ,kCAAkC;;CAK3F;AAED,qBACa,uBAAwB,YAAW,YAAY;IACxD,MAAM,CAAC,EAAE,SAAgC;IAGzC,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,oCAAoC,CAAC;IAE5F,OAAO,IAAI,WAAW;CAqBzB;AAED,qBACa,iCAAkC,YAAW,0BAA0B;IAChF,iBAAiB,CAAC,GAAG,EAAE,uBAAuB,GAAG,MAAM;CAG1D"}