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

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

@@ -13,24 +13,31 @@
 //
 // SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
 // *****************************************************************************
-import { injectable, inject } from '@theia/core/shared/inversify';
-import { ToolProvider, ToolRequest, ToolRequestParameters, ToolRequestParametersProperties } from '@theia/ai-core';
-import { WorkspaceFunctionScope } from './workspace-functions';
-import { ChangeSetElementArgs, ChangeSetFileElement, ChangeSetFileElementFactory } from '@theia/ai-chat/lib/browser/change-set-file-element';
 import { ChangeSet, MutableChatRequestModel } from '@theia/ai-chat';
-import { FileService } from '@theia/filesystem/lib/browser/file-service';
+import { ChangeSetElementArgs, ChangeSetFileElement, ChangeSetFileElementFactory } from '@theia/ai-chat/lib/browser/change-set-file-element';
+import { ToolProvider, ToolRequest, ToolRequestParameters, ToolRequestParametersProperties } from '@theia/ai-core';
 import { ContentReplacer, Replacement } from '@theia/core/lib/common/content-replacer';
 import { URI } from '@theia/core/lib/common/uri';
+import { inject, injectable } from '@theia/core/shared/inversify';
+import { FileService } from '@theia/filesystem/lib/browser/file-service';
+import { WorkspaceFunctionScope } from './workspace-functions';
 
+import { nls } from '@theia/core';
 import {
+    CLEAR_FILE_CHANGES_ID,
+    GET_PROPOSED_CHANGES_ID,
     SUGGEST_FILE_CONTENT_ID,
-    WRITE_FILE_CONTENT_ID,
     SUGGEST_FILE_REPLACEMENTS_ID,
-    WRITE_FILE_REPLACEMENTS_ID,
-    CLEAR_FILE_CHANGES_ID,
-    GET_PROPOSED_CHANGES_ID
+    WRITE_FILE_CONTENT_ID,
+    WRITE_FILE_REPLACEMENTS_ID
 } from '../common/file-changeset-function-ids';
 
+export const FileChangeSetTitleProvider = Symbol('FileChangeSetTitleProvider');
+
+export interface FileChangeSetTitleProvider {
+    getChangeSetTitle(ctx: MutableChatRequestModel): string;
+}
+
 @injectable()
 export class SuggestFileContent implements ToolProvider {
     static ID = SUGGEST_FILE_CONTENT_ID;
@@ -44,6 +51,9 @@ export class SuggestFileContent implements ToolProvider {
     @inject(ChangeSetFileElementFactory)
     protected readonly fileChangeFactory: ChangeSetFileElementFactory;
 
+    @inject(FileChangeSetTitleProvider)
+    protected readonly fileChangeSetTitleProvider: FileChangeSetTitleProvider;
+
     getTool(): ToolRequest {
         return {
             id: SuggestFileContent.ID,
@@ -68,6 +78,9 @@ export class SuggestFileContent implements ToolProvider {
                 required: ['path', 'content']
             },
             handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
                 const { path, content } = JSON.parse(args);
                 const chatSessionId = ctx.session.id;
                 const uri = await this.workspaceFunctionScope.resolveRelativePath(path);
@@ -75,7 +88,7 @@ export class SuggestFileContent implements ToolProvider {
                 if (content === '') {
                     type = 'delete';
                 }
-                if (!await this.fileService.exists(uri)) {
+                if (!(await this.fileService.exists(uri))) {
                     type = 'add';
                 }
                 ctx.session.changeSet.addElements(
@@ -88,7 +101,8 @@ export class SuggestFileContent implements ToolProvider {
                         chatSessionId
                     })
                 );
-                ctx.session.changeSet.setTitle('Changes proposed by Coder');
+
+                ctx.session.changeSet.setTitle(this.fileChangeSetTitleProvider.getChangeSetTitle(ctx));
                 return `Proposed writing to file ${path}. The user will review and potentially apply the changes`;
             }
         };
@@ -108,6 +122,9 @@ export class WriteFileContent implements ToolProvider {
     @inject(ChangeSetFileElementFactory)
     protected readonly fileChangeFactory: ChangeSetFileElementFactory;
 
+    @inject(FileChangeSetTitleProvider)
+    protected readonly fileChangeSetTitleProvider: FileChangeSetTitleProvider;
+
     getTool(): ToolRequest {
         return {
             id: WriteFileContent.ID,
@@ -132,6 +149,9 @@ export class WriteFileContent implements ToolProvider {
                 required: ['path', 'content']
             },
             handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
                 const { path, content } = JSON.parse(args);
                 const chatSessionId = ctx.session.id;
                 const uri = await this.workspaceFunctionScope.resolveRelativePath(path);
@@ -139,7 +159,7 @@ export class WriteFileContent implements ToolProvider {
                 if (content === '') {
                     type = 'delete';
                 }
-                if (!await this.fileService.exists(uri)) {
+                if (!(await this.fileService.exists(uri))) {
                     type = 'add';
                 }
 
@@ -153,7 +173,7 @@ export class WriteFileContent implements ToolProvider {
                     chatSessionId
                 });
 
-                ctx.session.changeSet.setTitle('Changes applied by Coder');
+                ctx.session.changeSet.setTitle(this.fileChangeSetTitleProvider.getChangeSetTitle(ctx));
                 // Add the element to the change set
                 ctx.session.changeSet.addElements(fileElement);
 
@@ -180,6 +200,9 @@ export class ReplaceContentInFileFunctionHelper {
     @inject(ChangeSetFileElementFactory)
     protected readonly fileChangeFactory: ChangeSetFileElementFactory;
 
+    @inject(FileChangeSetTitleProvider)
+    protected readonly fileChangeSetTitleProvider: FileChangeSetTitleProvider;
+
     private replacer: ContentReplacer;
 
     constructor() {
@@ -249,12 +272,15 @@ export class ReplaceContentInFileFunctionHelper {
             description: replacementDescription,
             parameters: replacementParameters
         };
-
     }
 
     async createChangesetFromToolCall(toolCallString: string, ctx: MutableChatRequestModel): Promise<string> {
         try {
-            const result = await this.processReplacementsCommon(toolCallString, ctx, 'Changes proposed by Coder');
+            if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                return JSON.stringify({ error: 'Operation cancelled by user' });
+            }
+
+            const result = await this.processReplacementsCommon(toolCallString, ctx, this.fileChangeSetTitleProvider.getChangeSetTitle(ctx));
 
             if (result.errors.length > 0) {
                 return `Errors encountered: ${result.errors.join('; ')}`;
@@ -274,7 +300,11 @@ export class ReplaceContentInFileFunctionHelper {
 
     async writeChangesetFromToolCall(toolCallString: string, ctx: MutableChatRequestModel): Promise<string> {
         try {
-            const result = await this.processReplacementsCommon(toolCallString, ctx, 'Changes applied by Coder');
+            if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                return JSON.stringify({ error: 'Operation cancelled by user' });
+            }
+
+            const result = await this.processReplacementsCommon(toolCallString, ctx, this.fileChangeSetTitleProvider.getChangeSetTitle(ctx));
 
             if (result.errors.length > 0) {
                 return `Errors encountered: ${result.errors.join('; ')}`;
@@ -300,6 +330,10 @@ export class ReplaceContentInFileFunctionHelper {
         ctx: MutableChatRequestModel,
         changeSetTitle: string
     ): Promise<{ fileElement: ChangeSetFileElement | undefined, path: string, reset: boolean, errors: string[] }> {
+        if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+            throw new Error('Operation cancelled by user');
+        }
+
         const { path, replacements, reset } = JSON.parse(toolCallString) as { path: string, replacements: Replacement[], reset?: boolean };
         const fileUri = await this.workspaceFunctionScope.resolveRelativePath(path);
 
@@ -318,6 +352,10 @@ export class ReplaceContentInFileFunctionHelper {
             }
         }
 
+        if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+            throw new Error('Operation cancelled by user');
+        }
+
         const { updatedContent, errors } = this.replacer.applyReplacements(startingContent, replacements);
 
         if (errors.length > 0) {
@@ -348,11 +386,17 @@ export class ReplaceContentInFileFunctionHelper {
 
     private findExistingChangeElement(changeSet: ChangeSet, fileUri: URI): ChangeSetFileElement | undefined {
         const element = changeSet.getElementByURI(fileUri);
-        if (element instanceof ChangeSetFileElement) { return element; }
+        if (element instanceof ChangeSetFileElement) {
+            return element;
+        }
     }
 
     async clearFileChanges(path: string, ctx: MutableChatRequestModel): Promise<string> {
         try {
+            if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                return JSON.stringify({ error: 'Operation cancelled by user' });
+            }
+
             const fileUri = await this.workspaceFunctionScope.resolveRelativePath(path);
             if (ctx.session.changeSet.removeElements(fileUri)) {
                 return `Cleared pending change(s) for file ${path}.`;
@@ -367,6 +411,10 @@ export class ReplaceContentInFileFunctionHelper {
 
     async getProposedFileState(path: string, ctx: MutableChatRequestModel): Promise<string> {
         try {
+            if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                return JSON.stringify({ error: 'Operation cancelled by user' });
+            }
+
             const fileUri = await this.workspaceFunctionScope.resolveRelativePath(path);
 
             if (!ctx.session.changeSet) {
@@ -403,8 +451,12 @@ export class SimpleSuggestFileReplacements implements ToolProvider {
             name: SimpleSuggestFileReplacements.ID,
             description: metadata.description,
             parameters: metadata.parameters,
-            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> =>
-                this.replaceContentInFileFunctionHelper.createChangesetFromToolCall(args, ctx)
+            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
+                return this.replaceContentInFileFunctionHelper.createChangesetFromToolCall(args, ctx);
+            }
         };
     }
 }
@@ -422,8 +474,12 @@ export class SimpleWriteFileReplacements implements ToolProvider {
             name: SimpleWriteFileReplacements.ID,
             description: metadata.description,
             parameters: metadata.parameters,
-            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> =>
-                this.replaceContentInFileFunctionHelper.writeChangesetFromToolCall(args, ctx)
+            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
+                return this.replaceContentInFileFunctionHelper.writeChangesetFromToolCall(args, ctx);
+            }
         };
     }
 }
@@ -441,8 +497,12 @@ export class SuggestFileReplacements implements ToolProvider {
             name: SuggestFileReplacements.ID,
             description: metadata.description,
             parameters: metadata.parameters,
-            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> =>
-                this.replaceContentInFileFunctionHelper.createChangesetFromToolCall(args, ctx)
+            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
+                return this.replaceContentInFileFunctionHelper.createChangesetFromToolCall(args, ctx);
+            }
         };
     }
 }
@@ -460,8 +520,12 @@ export class WriteFileReplacements implements ToolProvider {
             name: WriteFileReplacements.ID,
             description: metadata.description,
             parameters: metadata.parameters,
-            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> =>
-                this.replaceContentInFileFunctionHelper.writeChangesetFromToolCall(args, ctx)
+            handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
+                return this.replaceContentInFileFunctionHelper.writeChangesetFromToolCall(args, ctx);
+            }
         };
     }
 }
@@ -488,6 +552,9 @@ export class ClearFileChanges implements ToolProvider {
                 required: ['path']
             },
             handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
                 const { path } = JSON.parse(args);
                 return this.replaceContentInFileFunctionHelper.clearFileChanges(path, ctx);
             }
@@ -518,9 +585,19 @@ export class GetProposedFileState implements ToolProvider {
                 required: ['path']
             },
             handler: async (args: string, ctx: MutableChatRequestModel): Promise<string> => {
+                if (ctx?.response?.cancellationToken?.isCancellationRequested) {
+                    return JSON.stringify({ error: 'Operation cancelled by user' });
+                }
                 const { path } = JSON.parse(args);
                 return this.replaceContentInFileFunctionHelper.getProposedFileState(path, ctx);
             }
         };
     }
 }
+
+@injectable()
+export class DefaultFileChangeSetTitleProvider implements FileChangeSetTitleProvider {
+    getChangeSetTitle(ctx: MutableChatRequestModel): string {
+        return nls.localize('theia/ai-chat/fileChangeSetTitle', 'Changes proposed');
+    }
+}

package/src/browser/frontend-module.ts

@@ -33,6 +33,11 @@ import {
     ServiceConnectionProvider
 } from '@theia/core/lib/browser';
 import { TaskListProvider, TaskRunnerProvider } from './workspace-task-provider';
+import {
+    LaunchListProvider,
+    LaunchRunnerProvider,
+    LaunchStopProvider,
+} from './workspace-launch-provider';
 import { WorkspacePreferencesSchema } from './workspace-preferences';
 import {
     ClearFileChanges,
@@ -43,7 +48,9 @@ import {
     SuggestFileContent,
     WriteFileContent,
     WriteFileReplacements,
-    SimpleWriteFileReplacements
+    SimpleWriteFileReplacements,
+    FileChangeSetTitleProvider,
+    DefaultFileChangeSetTitleProvider
 } from './file-changeset-functions';
 import { OrchestratorChatAgent, OrchestratorChatAgentId } from '../common/orchestrator-chat-agent';
 import { UniversalChatAgent, UniversalChatAgentId } from '../common/universal-chat-agent';
@@ -71,10 +78,19 @@ import { CommandContribution } from '@theia/core';
 import { AIPromptFragmentsConfigurationWidget } from './ai-configuration/prompt-fragments-configuration-widget';
 import { BrowserAutomation, browserAutomationPath } from '../common/browser-automation-protocol';
 import { CloseBrowserProvider, IsBrowserRunningProvider, LaunchBrowserProvider, QueryDomProvider } from './app-tester-chat-functions';
+import { ModelAliasesConfigurationWidget } from './ai-configuration/model-aliases-configuration-widget';
+import { aiIdePreferenceSchema } from './ai-ide-preferences';
+import { AIActivationService } from '@theia/ai-core/lib/browser';
+import { AIIdeActivationServiceImpl } from './ai-ide-activation-service';
 
 export default new ContainerModule((bind, _unbind, _isBound, rebind) => {
+    bind(PreferenceContribution).toConstantValue({ schema: aiIdePreferenceSchema });
     bind(PreferenceContribution).toConstantValue({ schema: WorkspacePreferencesSchema });
 
+    bind(AIIdeActivationServiceImpl).toSelf().inSingletonScope();
+    // rebinds the default implementation of '@theia/ai-core'
+    rebind(AIActivationService).toService(AIIdeActivationServiceImpl);
+
     bind(ArchitectAgent).toSelf().inSingletonScope();
     bind(Agent).toService(ArchitectAgent);
     bind(ChatAgent).toService(ArchitectAgent);
@@ -119,7 +135,11 @@ export default new ContainerModule((bind, _unbind, _isBound, rebind) => {
     bindToolProvider(WriteFileContent, bind);
     bindToolProvider(TaskListProvider, bind);
     bindToolProvider(TaskRunnerProvider, bind);
+    bindToolProvider(LaunchListProvider, bind);
+    bindToolProvider(LaunchRunnerProvider, bind);
+    bindToolProvider(LaunchStopProvider, bind);
     bind(ReplaceContentInFileFunctionHelper).toSelf().inSingletonScope();
+    bind(FileChangeSetTitleProvider).to(DefaultFileChangeSetTitleProvider).inSingletonScope();
     bindToolProvider(SuggestFileReplacements, bind);
     bindToolProvider(WriteFileReplacements, bind);
     bindToolProvider(ListChatContext, bind);
@@ -157,6 +177,14 @@ export default new ContainerModule((bind, _unbind, _isBound, rebind) => {
         }))
         .inSingletonScope();
 
+    bind(ModelAliasesConfigurationWidget).toSelf();
+    bind(WidgetFactory)
+        .toDynamicValue(ctx => ({
+            id: ModelAliasesConfigurationWidget.ID,
+            createWidget: () => ctx.container.get(ModelAliasesConfigurationWidget)
+        }))
+        .inSingletonScope();
+
     bindToolProvider(SimpleSuggestFileReplacements, bind);
     bindToolProvider(SimpleWriteFileReplacements, bind);
     bindToolProvider(ClearFileChanges, bind);

package/src/browser/ide-chat-welcome-message-provider.tsx

@@ -85,7 +85,7 @@ export class IdeChatWelcomeMessageProvider implements ChatWelcomeMessageProvider
             <div className='theia-ResponseNode-Content' key={'disabled-message'}>
                 <div className="disable-message">
                     <span className="section-header">{
-                        nls.localize('theia/ai/chat-ui/chat-view-tree-widget/aiFeatureHeader', '🚀 AI Features Available (Alpha Version)!')}
+                        nls.localize('theia/ai/chat-ui/chat-view-tree-widget/aiFeatureHeader', '🚀 AI Features Available (Beta Version)!')}
                     </span>
                     <div className="section-title">
                         <p><code>{nls.localize('theia/ai/chat-ui/chat-view-tree-widget/featuresDisabled', 'Currently, all AI Features are disabled!')}</code></p>
@@ -96,14 +96,14 @@ export class IdeChatWelcomeMessageProvider implements ChatWelcomeMessageProvider
                     <div className="section-content">
                         <p>To enable the AI features, please go to the AI features section of the&nbsp;
                             {this.renderLinkButton(nls.localize('theia/ai/chat-ui/chat-view-tree-widget/settingsMenu', 'the settings menu'),
-                               CommonCommands.OPEN_PREFERENCES.id, 'ai-features')}&nbsp;and
-                            </p>
+                                CommonCommands.OPEN_PREFERENCES.id, 'ai-features')}&nbsp;and
+                        </p>
                         <ol>
                             <li>Toggle the switch for <strong>{nls.localize('theia/ai/chat-ui/chat-view-tree-widget/aiFeaturesEnable', 'Ai-features: Enable')}</strong>.</li>
                             <li>Provide at least one LLM provider (e.g. OpenAI). See <a href="https://theia-ide.org/docs/user_ai/" target="_blank">the documentation</a>
                                 &nbsp;for more information.</li>
                         </ol>
-                        <p>This will activate the AI capabilities in the app. Please remember, these features are <strong>in an alpha state</strong>,
+                        <p>This will activate the AI capabilities in the app. Please remember, these features are <strong>in a beta state</strong>,
                             so they may change and we are working on improving them 🚧.<br></br>
                             Please support us by <a href="https://github.com/eclipse-theia/theia">providing feedback
                             </a>!</p>

package/src/browser/style/index.css

@@ -64,7 +64,8 @@
 }
 
 #ai-variable-configuration-container-widget,
-#ai-agent-configuration-container-widget {
+#ai-agent-configuration-container-widget,
+#ai-model-aliases-configuration-widget {
   margin-top: 5px;
 }
 
@@ -86,21 +87,24 @@
   grid-template-columns: 1fr 2fr;
 }
 
-/* Agent Settings */
-#ai-agent-configuration-container-widget ul {
+/* Agent and Model Alias Settings */
+#ai-agent-configuration-container-widget ul,
+#ai-model-aliases-configuration-widget .model-alias-configuration-list ul {
   list-style: none;
   padding: 0;
   margin: 0;
 }
 
-.ai-agent-configuration-main {
+.ai-agent-configuration-main,
+.model-alias-configuration-main {
   display: flex;
   flex-direction: row;
 }
 
 #ai-agent-configuration-container-widget
   .ai-agent-configuration-main
-  .configuration-agents-list {
+  .configuration-agents-list,
+#ai-model-aliases-configuration-widget .model-alias-configuration-list {
   min-width: 160px;
   overflow: hidden;
   white-space: nowrap;
@@ -109,7 +113,8 @@
   padding: var(--theia-ui-padding);
 }
 
-.configuration-agent-panel {
+.configuration-agent-panel,
+.model-alias-configuration-panel {
   flex: 1;
 }
 
@@ -337,6 +342,32 @@
   font-style: italic;
 }
 
+.prompt-variant-warning {
+  color: var(--theia-warningForeground);
+  background: var(--theia-warningBackground);
+  border-left: 4px solid var(--theia-warningBorder);
+  padding: 6px 12px;
+  margin-bottom: 8px;
+  display: flex;
+  align-items: center;
+  gap: 8px;
+  border-radius: 3px;
+  font-size: 0.95em;
+}
+
+.prompt-variant-error {
+  color: var(--theia-errorForeground);
+  background: var(--theia-errorBackground);
+  border-left: 4px solid var(--theia-errorBorder);
+  padding: 6px 12px;
+  margin-bottom: 8px;
+  display: flex;
+  align-items: center;
+  gap: 8px;
+  border-radius: 3px;
+  font-size: 0.95em;
+}
+
 /* Template content collapsable styles */
 .template-content-container {
   padding: 10px;
@@ -746,6 +777,80 @@ h4 {
 
 /* End AI Tools Configuration Widget Styles */
 
+/* AI Model Aliases and Language Model Renderer extracted styles */
+.ai-model-alias-list {
+  width: 25%;
+}
+.ai-alias-detail-title {
+  padding-left: 0;
+  padding-bottom: 10px;
+}
+.ai-alias-detail-description {
+  padding-bottom: 10px;
+}
+.ai-alias-detail-selected-model {
+  margin-bottom: 20px;
+}
+.ai-language-model-item-ready {
+  font-style: normal;
+}
+.ai-language-model-item-not-ready {
+  font-style: italic;
+}
+.ai-alias-priority-item-resolved {
+  font-weight: bold;
+}
+.ai-alias-priority-item-ready {
+  font-style: inherit;
+}
+.ai-alias-priority-item-not-ready {
+  font-style: italic;
+}
+.ai-alias-detail-defaults {
+  margin-bottom: 10px;
+}
+.ai-model-default-not-ready {
+  font-style: italic;
+  color: var(--theia-descriptionForeground);
+}
+.ai-alias-defaults-hint {
+  color: var(--theia-descriptionForeground);
+  margin-top: 8px;
+}
+.ai-alias-evaluates-to-container {
+  margin-top: 8px;
+  margin-bottom: 8px;
+}
+.ai-alias-evaluates-to-label {
+  font-weight: 600;
+}
+.ai-alias-evaluates-to-value {
+  margin-left: 8px;
+}
+.ai-model-status-ready {
+  color: green;
+  margin-left: 6px;
+}
+.ai-model-status-not-ready {
+  color: red;
+  margin-left: 6px;
+}
+.ai-alias-evaluates-to-unresolved {
+  margin-left: 8px;
+  color: var(--theia-descriptionForeground);
+}
+.ai-alias-detail-agents {
+  margin-bottom: 10px;
+}
+.ai-alias-agent-id {
+  color: var(--theia-descriptionForeground);
+  margin-left: 8px;
+}
+.ai-alias-no-agents {
+  color: var(--theia-descriptionForeground);
+}
+/* End AI Model Aliases and Language Model Renderer extracted styles */
+
 /* Token Usage Configuration Styles */
 .token-usage-configuration-title {
   margin: 0 0 16px 0;

package/src/browser/test/tool-provider-cancellation-test-util.spec.ts

@@ -0,0 +1,60 @@
+// *****************************************************************************
+// 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 { CancellationTokenSource } from '@theia/core';
+import { expect } from 'chai';
+
+// Simple test for cancellation handling
+describe('Tool Provider Cancellation Tests', () => {
+    it('should verify basic cancellation token functionality', () => {
+        // Create a cancellation token source
+        const cts = new CancellationTokenSource();
+
+        // Initially the token should not be cancelled
+        expect(cts.token.isCancellationRequested).to.be.false;
+
+        // After cancellation, the token should report as cancelled
+        cts.cancel();
+        expect(cts.token.isCancellationRequested).to.be.true;
+
+        // Cleanup
+        cts.dispose();
+    });
+
+    it('should trigger cancellation callback when cancelled', async () => {
+        // Create a cancellation token source
+        const cts = new CancellationTokenSource();
+
+        // Create a flag to track if the callback was called
+        let callbackCalled = false;
+
+        // Register a cancellation callback
+        const disposable = cts.token.onCancellationRequested(() => {
+            callbackCalled = true;
+        });
+
+        // Initially the callback should not have been called
+        expect(callbackCalled).to.be.false;
+
+        // After cancellation, the callback should be called
+        cts.cancel();
+        expect(callbackCalled).to.be.true;
+
+        // Cleanup
+        disposable.dispose();
+        cts.dispose();
+    });
+});