@theia/ai-ide 1.64.0-next.35 → 1.65.0-next.1

package/src/browser/ai-ide-activation-service.ts

@@ -0,0 +1,65 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+import { inject, injectable } from '@theia/core/shared/inversify';
+import { FrontendApplicationContribution, PreferenceService } from '@theia/core/lib/browser';
+import { Emitter, MaybePromise, Event, } from '@theia/core';
+import { ContextKeyService, ContextKey } from '@theia/core/lib/browser/context-key-service';
+import { AIActivationService, ENABLE_AI_CONTEXT_KEY } from '@theia/ai-core/lib/browser/ai-activation-service';
+import { PREFERENCE_NAME_ENABLE_AI } from './ai-ide-preferences';
+
+/**
+ * Implements AI Activation Service based on preferences.
+ */
+@injectable()
+export class AIIdeActivationServiceImpl implements AIActivationService, FrontendApplicationContribution {
+    @inject(ContextKeyService)
+    protected readonly contextKeyService: ContextKeyService;
+
+    @inject(PreferenceService)
+    protected preferenceService: PreferenceService;
+
+    protected isAiEnabledKey: ContextKey<boolean>;
+
+    protected onDidChangeAIEnabled = new Emitter<boolean>();
+    get onDidChangeActiveStatus(): Event<boolean> {
+        return this.onDidChangeAIEnabled.event;
+    }
+
+    get isActive(): boolean {
+        return this.isAiEnabledKey.get() ?? false;
+    }
+
+    protected updateEnableValue(value: boolean): void {
+        if (value !== this.isAiEnabledKey.get()) {
+            this.isAiEnabledKey.set(value);
+            this.onDidChangeAIEnabled.fire(value);
+        }
+    }
+
+    initialize(): MaybePromise<void> {
+        this.isAiEnabledKey = this.contextKeyService.createKey(ENABLE_AI_CONTEXT_KEY, false);
+        // make sure we don't miss once preferences are ready
+        this.preferenceService.ready.then(() => {
+            const enableValue = this.preferenceService.get<boolean>(PREFERENCE_NAME_ENABLE_AI, false);
+            this.updateEnableValue(enableValue);
+        });
+        this.preferenceService.onPreferenceChanged(e => {
+            if (e.preferenceName === PREFERENCE_NAME_ENABLE_AI) {
+                this.updateEnableValue(e.newValue);
+            }
+        });
+    }
+}

package/src/browser/ai-ide-preferences.ts

@@ -0,0 +1,44 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+import { AI_CORE_PREFERENCES_TITLE, ENABLE_AI_CONTEXT_KEY } from '@theia/ai-core/lib/browser';
+import { nls } from '@theia/core';
+import { PreferenceSchema } from '@theia/core/lib/browser';
+
+// We reuse the context key for the preference name
+export const PREFERENCE_NAME_ENABLE_AI = ENABLE_AI_CONTEXT_KEY;
+
+export const aiIdePreferenceSchema: PreferenceSchema = {
+    type: 'object',
+    properties: {
+        [PREFERENCE_NAME_ENABLE_AI]: {
+            title: AI_CORE_PREFERENCES_TITLE,
+            markdownDescription: nls.localize('theia/ai/ide/enableAI/mdDescription',
+                '❗ This setting allows you to access the latest AI capabilities (Beta version).\
+            \n\
+            Please note that these features are in a beta phase, which means they may \
+            undergo changes and will be further improved. It is important to be aware that these features may generate\
+            continuous requests to the language models (LLMs) you provide access to. This might incur costs that you\
+            need to monitor closely. By enabling this option, you acknowledge these risks.\
+            \n\
+            **Please note! The settings below in this section will only take effect\n\
+            once the main feature setting is enabled. After enabling the feature, you need to configure at least one\
+            LLM provider below. Also see [the documentation](https://theia-ide.org/docs/user_ai/)**.'),
+            type: 'boolean',
+            default: false,
+        }
+    }
+};

package/src/browser/app-tester-chat-agent.ts

@@ -40,7 +40,7 @@ export class AppTesterChatAgent extends AbstractStreamParsingChatAgent {
     name = AppTesterChatAgentId;
     languageModelRequirements: LanguageModelRequirement[] = [{
         purpose: 'chat',
-        identifier: 'openai/gpt-4o',
+        identifier: 'default/code',
     }];
     protected defaultLanguageModelPurpose: string = 'chat';
     override description = nls.localize('theia/ai/chat/app-tester/description', 'This agent tests your application user interface to verify user-specified test scenarios through the Playwright MCP server. '

package/src/browser/architect-agent.ts

@@ -30,7 +30,7 @@ export class ArchitectAgent extends AbstractStreamParsingChatAgent {
     id = 'Architect';
     languageModelRequirements: LanguageModelRequirement[] = [{
         purpose: 'chat',
-        identifier: 'openai/gpt-4o',
+        identifier: 'default/code',
     }];
     protected defaultLanguageModelPurpose: string = 'chat';
 

package/src/browser/coder-agent.ts

@@ -30,7 +30,7 @@ export class CoderAgent extends AbstractStreamParsingChatAgent {
     name = 'Coder';
     languageModelRequirements: LanguageModelRequirement[] = [{
         purpose: 'chat',
-        identifier: 'openai/gpt-4o',
+        identifier: 'default/code',
     }];
     protected defaultLanguageModelPurpose: string = 'chat';
 

package/src/browser/context-functions.spec.ts

@@ -0,0 +1,102 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+import { enableJSDOM } from '@theia/core/lib/browser/test/jsdom';
+let disableJSDOM = enableJSDOM();
+import { FrontendApplicationConfigProvider } from '@theia/core/lib/browser/frontend-application-config-provider';
+FrontendApplicationConfigProvider.set({});
+import { expect } from 'chai';
+import { ListChatContext, ResolveChatContext, AddFileToChatContext } from './context-functions';
+import { CancellationTokenSource } from '@theia/core';
+import { ChatContextManager, MutableChatModel, MutableChatRequestModel, MutableChatResponseModel } from '@theia/ai-chat';
+import { fail } from 'assert';
+import { ResolvedAIContextVariable } from '@theia/ai-core';
+disableJSDOM();
+
+describe('Context Functions Cancellation Tests', () => {
+    let cancellationTokenSource: CancellationTokenSource;
+    let mockCtx: Partial<MutableChatRequestModel>;
+
+    before(() => {
+        disableJSDOM = enableJSDOM();
+    });
+    after(() => {
+        // Disable JSDOM after all tests
+        disableJSDOM();
+    });
+
+    beforeEach(() => {
+        cancellationTokenSource = new CancellationTokenSource();
+        const context: Partial<ChatContextManager> = {
+            addVariables: () => { },
+            getVariables: () => mockCtx.context?.variables as ResolvedAIContextVariable[]
+        };
+        mockCtx = {
+            response: {
+                cancellationToken: cancellationTokenSource.token
+            } as MutableChatResponseModel,
+            context: {
+                variables: [{
+                    variable: { id: 'file1', name: 'File' },
+                    arg: '/path/to/file',
+                    contextValue: 'file content'
+                } as ResolvedAIContextVariable]
+            },
+            session: {
+                context
+            } as MutableChatModel
+        };
+    });
+
+    afterEach(() => {
+        cancellationTokenSource.dispose();
+    });
+
+    it('ListChatContext should respect cancellation token', async () => {
+        const listChatContext = new ListChatContext();
+        cancellationTokenSource.cancel();
+
+        const result = await listChatContext.getTool().handler('', mockCtx);
+        if (typeof result !== 'string') {
+            fail(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('ResolveChatContext should respect cancellation token', async () => {
+        const resolveChatContext = new ResolveChatContext();
+        cancellationTokenSource.cancel();
+
+        const result = await resolveChatContext.getTool().handler('{"contextElementId":"file1/path/to/file"}', mockCtx);
+        if (typeof result !== 'string') {
+            fail(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('AddFileToChatContext should respect cancellation token', async () => {
+        const addFileToChatContext = new AddFileToChatContext();
+        cancellationTokenSource.cancel();
+
+        const result = await addFileToChatContext.getTool().handler('{"filesToAdd":["/new/path/to/file"]}', mockCtx);
+        if (typeof result !== 'string') {
+            fail(`Wrong tool call result type: ${result}`);
+        }
+        const jsonResponse = JSON.parse(result);
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+});

package/src/browser/context-functions.ts

@@ -30,6 +30,9 @@ export class ListChatContext implements ToolProvider {
             name: ListChatContext.ID,
             description: 'Returns the list of context elements (such as files) specified by the user manually as part of the chat request.',
             handler: async (_: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
                 const result = ctx.context.variables.map(contextElement => ({
                     id: contextElement.variable.id + contextElement.arg,
                     type: contextElement.variable.name
@@ -64,6 +67,10 @@ export class ResolveChatContext implements ToolProvider {
                 required: ['contextElementId']
             },
             handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
+
                 const { contextElementId } = JSON.parse(args) as { contextElementId: string };
                 const variable = ctx.context.variables.find(contextElement => contextElement.variable.id + contextElement.arg === contextElementId);
                 if (variable) {
@@ -101,6 +108,10 @@ export class AddFileToChatContext implements ToolProvider {
             },
             description: 'Adds one or more files to the context of the current chat session, and returns the current list of files in the context.',
             handler: async (arg: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
+
                 const { filesToAdd } = JSON.parse(arg) as { filesToAdd: string[] };
 
                 ctx.session.context.addVariables(...filesToAdd.map(file => ({ arg: file, variable: FILE_VARIABLE })));

package/src/browser/file-changeset-function.spec.ts

@@ -0,0 +1,52 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+import { FrontendApplicationConfigProvider } from '@theia/core/lib/browser/frontend-application-config-provider';
+import { enableJSDOM } from '@theia/core/lib/browser/test/jsdom';
+let disableJSDOM = enableJSDOM();
+FrontendApplicationConfigProvider.set({});
+
+import { MutableChatRequestModel } from '@theia/ai-chat';
+import { Container } from '@theia/core/shared/inversify';
+import { expect } from 'chai';
+import { DefaultFileChangeSetTitleProvider } from './file-changeset-functions';
+
+disableJSDOM();
+
+describe('DefaultFileChangeSetTitleProvider', () => {
+    let provider: DefaultFileChangeSetTitleProvider;
+
+    before(() => {
+        const container = new Container();
+        container.bind(DefaultFileChangeSetTitleProvider).toSelf();
+
+        provider = container.get(DefaultFileChangeSetTitleProvider);
+        disableJSDOM = enableJSDOM();
+    });
+
+    after(() => {
+        disableJSDOM();
+    });
+
+    it('should provide the title', () => {
+        const ctx = {
+            agentId: 'test-agent',
+        } as MutableChatRequestModel;
+
+        const title = provider.getChangeSetTitle(ctx);
+        expect(title).to.equal('Changes proposed');
+    });
+});

package/src/browser/file-changeset-functions.spec.ts

@@ -0,0 +1,212 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+import { enableJSDOM } from '@theia/core/lib/browser/test/jsdom';
+let disableJSDOM = enableJSDOM();
+import { FrontendApplicationConfigProvider } from '@theia/core/lib/browser/frontend-application-config-provider';
+FrontendApplicationConfigProvider.set({});
+
+import { expect } from 'chai';
+import { CancellationTokenSource } from '@theia/core';
+import {
+    SuggestFileContent,
+    WriteFileContent,
+    SuggestFileReplacements,
+    WriteFileReplacements,
+    ClearFileChanges,
+    GetProposedFileState,
+    ReplaceContentInFileFunctionHelper,
+    FileChangeSetTitleProvider,
+    DefaultFileChangeSetTitleProvider
+} from './file-changeset-functions';
+import { MutableChatRequestModel, MutableChatResponseModel, ChangeSet, ChangeSetElement, MutableChatModel } from '@theia/ai-chat';
+import { Container } from '@theia/core/shared/inversify';
+import { WorkspaceFunctionScope } from './workspace-functions';
+import { FileService } from '@theia/filesystem/lib/browser/file-service';
+import { ChangeSetFileElementFactory, ChangeSetFileElement } from '@theia/ai-chat/lib/browser/change-set-file-element';
+import { URI } from '@theia/core/lib/common/uri';
+
+disableJSDOM();
+
+describe('File Changeset Functions Cancellation Tests', () => {
+    let cancellationTokenSource: CancellationTokenSource;
+    let mockCtx: Partial<MutableChatRequestModel>;
+    let container: Container;
+    before(() => {
+        disableJSDOM = enableJSDOM();
+    });
+    after(() => {
+        // Disable JSDOM after all tests
+        disableJSDOM();
+    });
+    beforeEach(() => {
+        cancellationTokenSource = new CancellationTokenSource();
+
+        // Create a mock change set that doesn't do anything
+        const mockChangeSet: Partial<ChangeSet> = {
+            addElements: (...elements: ChangeSetElement[]) => true,
+            setTitle: () => { },
+            removeElements: () => true,
+            getElementByURI: () => undefined
+        };
+
+        // Setup mock context
+        mockCtx = {
+            id: 'test-request-id',
+            response: {
+                cancellationToken: cancellationTokenSource.token
+            } as MutableChatResponseModel,
+            session: {
+                id: 'test-session-id',
+                changeSet: mockChangeSet as ChangeSet
+            } as MutableChatModel
+        };
+
+        // Create a new container for each test
+        container = new Container();
+
+        // Mock dependencies
+        const mockWorkspaceScope = {
+            resolveRelativePath: async () => new URI('file:///workspace/test.txt')
+        } as unknown as WorkspaceFunctionScope;
+
+        const mockFileService = {
+            exists: async () => true,
+            read: async () => ({ value: { toString: () => 'test content' } })
+        } as unknown as FileService;
+
+        const mockFileChangeFactory: ChangeSetFileElementFactory = () => ({
+            uri: new URI('file:///workspace/test.txt'),
+            type: 'modify',
+            state: 'pending',
+            targetState: 'new content',
+            apply: async () => { },
+        } as ChangeSetFileElement);
+
+        // Register mocks in the container
+        container.bind(WorkspaceFunctionScope).toConstantValue(mockWorkspaceScope);
+        container.bind(FileService).toConstantValue(mockFileService);
+        container.bind(ChangeSetFileElementFactory).toConstantValue(mockFileChangeFactory);
+        container.bind(FileChangeSetTitleProvider).to(DefaultFileChangeSetTitleProvider).inSingletonScope();
+        container.bind(ReplaceContentInFileFunctionHelper).toSelf();
+        container.bind(SuggestFileContent).toSelf();
+        container.bind(WriteFileContent).toSelf();
+        container.bind(SuggestFileReplacements).toSelf();
+        container.bind(WriteFileReplacements).toSelf();
+        container.bind(ClearFileChanges).toSelf();
+        container.bind(GetProposedFileState).toSelf();
+    });
+
+    afterEach(() => {
+        cancellationTokenSource.dispose();
+    });
+
+    it('SuggestFileContent should respect cancellation token', async () => {
+        const suggestFileContent = container.get(SuggestFileContent);
+        cancellationTokenSource.cancel();
+
+        const handler = suggestFileContent.getTool().handler;
+        const result = await handler(JSON.stringify({ path: 'test.txt', content: 'test content' }), mockCtx as MutableChatRequestModel);
+
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('WriteFileContent should respect cancellation token', async () => {
+        const writeFileContent = container.get(WriteFileContent);
+        cancellationTokenSource.cancel();
+
+        const handler = writeFileContent.getTool().handler;
+        const result = await handler(JSON.stringify({ path: 'test.txt', content: 'test content' }), mockCtx as MutableChatRequestModel);
+
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('SuggestFileReplacements should respect cancellation token', async () => {
+        const suggestFileReplacements = container.get(SuggestFileReplacements);
+        cancellationTokenSource.cancel();
+
+        const handler = suggestFileReplacements.getTool().handler;
+        const result = await handler(
+            JSON.stringify({
+                path: 'test.txt',
+                replacements: [{ oldContent: 'old', newContent: 'new' }]
+            }),
+            mockCtx as MutableChatRequestModel
+        );
+
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('WriteFileReplacements should respect cancellation token', async () => {
+        const writeFileReplacements = container.get(WriteFileReplacements);
+        cancellationTokenSource.cancel();
+
+        const handler = writeFileReplacements.getTool().handler;
+        const result = await handler(
+            JSON.stringify({
+                path: 'test.txt',
+                replacements: [{ oldContent: 'old', newContent: 'new' }]
+            }),
+            mockCtx as MutableChatRequestModel
+        );
+
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('ClearFileChanges should respect cancellation token', async () => {
+        const clearFileChanges = container.get(ClearFileChanges);
+        cancellationTokenSource.cancel();
+
+        const handler = clearFileChanges.getTool().handler;
+        const result = await handler(JSON.stringify({ path: 'test.txt' }), mockCtx as MutableChatRequestModel);
+
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('GetProposedFileState should respect cancellation token', async () => {
+        const getProposedFileState = container.get(GetProposedFileState);
+        cancellationTokenSource.cancel();
+
+        const handler = getProposedFileState.getTool().handler;
+        const result = await handler(JSON.stringify({ path: 'test.txt' }), mockCtx as MutableChatRequestModel);
+
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+    });
+
+    it('ReplaceContentInFileFunctionHelper should handle cancellation in common processing', async () => {
+        const helper = container.get(ReplaceContentInFileFunctionHelper);
+        cancellationTokenSource.cancel();
+
+        // Test the underlying helper method through the public methods
+
+        const result = await helper.createChangesetFromToolCall(
+            JSON.stringify({
+                path: 'test.txt',
+                replacements: [{ oldContent: 'old', newContent: 'new' }]
+            }),
+            mockCtx as MutableChatRequestModel
+        );
+        const jsonResponse = typeof result === 'string' ? JSON.parse(result) : result;
+        expect(jsonResponse.error).to.equal('Operation cancelled by user');
+
+    });
+});