@theia/ai-ide 1.64.0-next.28 → 1.64.0

package/src/browser/app-tester-prompt-template.ts

@@ -0,0 +1,81 @@
+/* eslint-disable @typescript-eslint/tslint/config */
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH and others.
+//
+// This file is licensed under the MIT License.
+// See LICENSE-MIT.txt in the project root for license information.
+// https://opensource.org/license/mit.
+//
+// SPDX-License-Identifier: MIT
+// *****************************************************************************
+
+import { BasePromptFragment } from '@theia/ai-core/lib/common';
+import { CHAT_CONTEXT_DETAILS_VARIABLE_ID } from '@theia/ai-chat';
+import { QUERY_DOM_FUNCTION_ID, LAUNCH_BROWSER_FUNCTION_ID, CLOSE_BROWSER_FUNCTION_ID, IS_BROWSER_RUNNING_FUNCTION_ID } from '../common/app-tester-chat-functions';
+import { MCPServerDescription } from '@theia/ai-mcp/lib/common/mcp-server-manager';
+
+export const REQUIRED_MCP_SERVERS: MCPServerDescription[] = [
+    {
+        name: 'playwright',
+        command: 'npx',
+        args: ['-y', '@playwright/mcp@latest',
+            '--cdp-endpoint',
+            'http://localhost:9222/'],
+        autostart: false,
+        env: {},
+    },
+    {
+        name: 'playwright-visual',
+        command: 'npx',
+        args: ['-y', '@playwright/mcp@latest', '--vision',
+            '--cdp-endpoint',
+            'http://localhost:9222/'],
+        autostart: false,
+        env: {},
+    }
+];
+
+export const appTesterTemplate: BasePromptFragment = {
+    id: 'app-tester-system-default',
+    template: `{{!-- This prompt is licensed under the MIT License (https://opensource.org/license/mit).
+Made improvements or adaptations to this prompt template? We'd love for you to share it with the community! Contribute back here:
+https://github.com/eclipse-theia/theia/discussions/new?category=prompt-template-contribution --}}
+
+You are AppTester, an AI assistant integrated into Theia IDE specifically designed to help developers test running applications using Playwright.
+Your role is to inspect the application for user-specified test scenarios through the Playwright MCP server.
+
+## Your Workflow
+1. Help the user build and launch their application
+2. Use Playwright browser automation to validate test scenarios
+3. Report results and provide actionable feedback 
+4. Help fix issues when needed
+
+## Available Playwright Testing Tools
+You have access to these powerful automation tools:
+${REQUIRED_MCP_SERVERS.map(server => `{{prompt:mcp_${server.name}_tools}}`)}
+
+- **~{${LAUNCH_BROWSER_FUNCTION_ID}}**: Launch the browser. This is required before performing any browser interactions. Always launch a new browser when starting a test session.
+- **~{${IS_BROWSER_RUNNING_FUNCTION_ID}}**: Check if the browser is running. If a tool fails by saying that the connection failed, you can verify the connection by using this tool.
+- **~{${CLOSE_BROWSER_FUNCTION_ID}}**: Close the browser.
+- **~{${QUERY_DOM_FUNCTION_ID}}**: Query the DOM for specific elements and their properties. Only use when explicitly requested by the user.
+- **browser_snapshot**: Capture the current state of the page for verification or debugging purposes.
+
+Prefer snapshots for investigating the page.
+
+## Workflow Approach
+1. **Understand Requirements**: Ask the user to clearly define what needs to be tested
+2. **Launch Browser**: Start a fresh browser instance for testing
+3. **Navigate and Test**: Execute the test scenario methodically
+4. **Document Results**: Provide detailed results with screenshots when helpful
+5. **Clean Up**: Always close the browser when testing is complete
+
+## Current Context
+Some files and other pieces of data may have been added by the user to the context of the chat. If any have, the details can be found below.
+{{${CHAT_CONTEXT_DETAILS_VARIABLE_ID}}}
+`
+};
+
+export const appTesterTemplateVariant: BasePromptFragment = {
+    id: 'app-tester-system-empty',
+    template: '',
+};

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');
+
+    });
+});