@theia/ai-chat 1.66.0-next.41 → 1.66.0-next.67

package/src/browser/task-context-service.ts

@@ -15,7 +15,7 @@
 // *****************************************************************************
 
 import { inject, injectable } from '@theia/core/shared/inversify';
-import { MaybePromise, ProgressService, URI, generateUuid, Event, EOL } from '@theia/core';
+import { MaybePromise, ProgressService, URI, generateUuid, Event, EOL, nls } from '@theia/core';
 import { ChatAgent, ChatAgentLocation, ChatService, ChatSession, MutableChatModel, MutableChatRequestModel, ParsedChatRequestTextPart } from '../common';
 import { PreferenceService } from '@theia/core/lib/common';
 import { ChatSessionSummaryAgent } from '../common/chat-session-summary-agent';
@@ -90,7 +90,10 @@ export class TaskContextService {
         if (existing && !override) { return existing.id; }
         const summaryId = generateUuid();
         const summaryDeferred = new Deferred<Summary>();
-        const progress = await this.progressService.showProgress({ text: `Summarize: ${session.title || session.id}`, options: { location: 'ai-chat' } });
+        const progress = await this.progressService.showProgress({
+            text: nls.localize('theia/ai/chat/taskContextService/summarizeProgressMessage', 'Summarize: {0}', session.title || session.id),
+            options: { location: 'ai-chat' }
+        });
         this.pendingSummaries.set(session.id, summaryDeferred.promise);
         try {
             const prompt = await this.getSystemPrompt(session, promptId);
@@ -127,7 +130,10 @@ export class TaskContextService {
             return this.summarize(session, promptId, agent, override);
         }
 
-        const progress = await this.progressService.showProgress({ text: `Updating: ${session.title || session.id}`, options: { location: 'ai-chat' } });
+        const progress = await this.progressService.showProgress({
+            text: nls.localize('theia/ai/chat/taskContextService/updatingProgressMessage', 'Updating: {0}', session.title || session.id),
+            options: { location: 'ai-chat' }
+        });
         try {
             const prompt = await this.getSystemPrompt(session, promptId);
             if (!prompt) {

package/src/browser/task-context-variable.ts

@@ -15,14 +15,19 @@
 // *****************************************************************************
 
 import { AIVariable } from '@theia/ai-core';
+import { nls } from '@theia/core';
 import { codiconArray } from '@theia/core/lib/browser';
 
 export const TASK_CONTEXT_VARIABLE: AIVariable = {
     id: 'taskContext',
-    description: 'Provides context information for a task, e.g. the plan for completing a task or a summary of a previous sessions',
+    description: nls.localize('theia/chat/taskContextVariable/description',
+        'Provides context information for a task, e.g. the plan for completing a task or a summary of a previous sessions'),
     name: 'taskContext',
-    label: 'Task Context',
+    label: nls.localize('theia/chat/taskContextVariable/label', 'Task Context'),
     iconClasses: codiconArray('clippy'),
     isContextVariable: true,
-    args: [{ name: 'context-id', description: 'The ID of the task context to retrieve, or a chat session to summarize.' }]
+    args: [{
+        name: 'context-id',
+        description: nls.localize('theia/chat/taskContextVariable/args/contextId/description', 'The ID of the task context to retrieve, or a chat session to summarize.')
+    }]
 };

package/src/common/change-set.ts

@@ -14,7 +14,7 @@
 // SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
 // *****************************************************************************
 
-import { ArrayUtils, Disposable, Emitter, Event, URI } from '@theia/core';
+import { ArrayUtils, Disposable, Emitter, Event, nls, URI } from '@theia/core';
 
 export interface ChangeSetElement {
     readonly uri: URI;
@@ -95,7 +95,7 @@ export class ChangeSetImpl implements ChangeSet {
 
     protected hasBeenSet = false;
     protected _elements = new Map<string, ChangeSetElement | undefined>();
-    protected _title = 'Suggested Changes';
+    protected _title = nls.localize('theia/ai/chat/changeSetDefaultTitle', 'Suggested Changes');
     get title(): string {
         return this._title;
     }

package/src/common/chat-agents-variable-contribution.ts

@@ -13,7 +13,7 @@
 //
 // SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
 // *****************************************************************************
-import { MaybePromise } from '@theia/core';
+import { MaybePromise, nls } from '@theia/core';
 import { inject, injectable } from '@theia/core/shared/inversify';
 import {
     AIVariable,
@@ -29,7 +29,7 @@ import { ChatAgentService } from './chat-agent-service';
 export const CHAT_AGENTS_VARIABLE: AIVariable = {
     id: 'chatAgents',
     name: 'chatAgents',
-    description: 'Returns the list of chat agents available in the system'
+    description: nls.localize('theia/ai/chat/chatAgentsVariable/description', 'Returns the list of chat agents available in the system')
 };
 
 export interface ChatAgentDescriptor {

package/src/common/chat-agents.ts

@@ -49,7 +49,7 @@ import {
     LanguageModelRegistry,
     LanguageModelStreamResponsePart
 } from '@theia/ai-core/lib/common';
-import { ContributionProvider, ILogger, isArray } from '@theia/core';
+import { ContributionProvider, ILogger, isArray, nls } from '@theia/core';
 import { inject, injectable, named, postConstruct } from '@theia/core/shared/inversify';
 import { ChatAgentService } from './chat-agent-service';
 import {
@@ -155,7 +155,7 @@ export abstract class AbstractChatAgent implements ChatAgent {
     readonly abstract languageModelRequirements: LanguageModelRequirement[];
     iconClass: string = 'codicon codicon-copilot';
     locations: ChatAgentLocation[] = ChatAgentLocation.ALL;
-    tags: string[] = ['Chat'];
+    tags: string[] = [nls.localizeByDefault('Chat')];
     description: string = '';
     variables: string[] = [];
     prompts: PromptVariantSet[] = [];
@@ -180,7 +180,7 @@ export abstract class AbstractChatAgent implements ChatAgent {
         try {
             const languageModel = await this.getLanguageModel(this.defaultLanguageModelPurpose);
             if (!languageModel) {
-                throw new Error('Couldn\'t find a matching language model. Please check your setup!');
+                throw new Error(nls.localize('theia/ai/chat/couldNotFindMatchingLM', 'Couldn\'t find a matching language model. Please check your setup!'));
             }
             const systemMessageDescription = await this.getSystemMessageDescription({ model: request.session, request } satisfies ChatSessionContext);
             const messages = await this.getMessages(request.session);
@@ -237,7 +237,7 @@ export abstract class AbstractChatAgent implements ChatAgent {
     protected async selectLanguageModel(selector: LanguageModelRequirement): Promise<LanguageModel> {
         const languageModel = await this.languageModelRegistry.selectLanguageModel({ agent: this.id, ...selector });
         if (!languageModel) {
-            throw new Error(`Couldn\'t find a ready language model for agent ${this.id}. Please check your setup!`);
+            throw new Error(nls.localize('theia/ai/chat/couldNotFindReadyLMforAgent', 'Couldn\'t find a ready language model for agent {0}. Please check your setup!', this.id));
         }
         return languageModel;
     }

package/src/common/chat-request-parser.spec.ts

@@ -204,4 +204,65 @@ describe('ChatRequestParserImpl', () => {
         expect(result.toolRequests.get(testTool1.id)).to.deep.equal(testTool1);
         expect(result.toolRequests.get(testTool2.id)).to.deep.equal(testTool2);
     });
+
+    it('parses simple command without arguments', async () => {
+        const req: ChatRequest = {
+            text: '/hello'
+        };
+        const context: ChatContext = { variables: [] };
+        const result = await parser.parseChatRequest(req, ChatAgentLocation.Panel, context);
+
+        expect(result.parts.length).to.equal(1);
+        expect(result.parts[0] instanceof ParsedChatRequestVariablePart).to.be.true;
+        const varPart = result.parts[0] as ParsedChatRequestVariablePart;
+        expect(varPart.variableName).to.equal('prompt');
+        expect(varPart.variableArg).to.equal('hello');
+    });
+
+    it('parses command with single argument', async () => {
+        const req: ChatRequest = {
+            text: '/explain topic'
+        };
+        const context: ChatContext = { variables: [] };
+        const result = await parser.parseChatRequest(req, ChatAgentLocation.Panel, context);
+
+        expect(result.parts.length).to.equal(1);
+        const varPart = result.parts[0] as ParsedChatRequestVariablePart;
+        expect(varPart.variableName).to.equal('prompt');
+        expect(varPart.variableArg).to.equal('explain|topic');
+    });
+
+    it('parses command with multiple arguments', async () => {
+        const req: ChatRequest = {
+            text: '/compare item1 item2'
+        };
+        const context: ChatContext = { variables: [] };
+        const result = await parser.parseChatRequest(req, ChatAgentLocation.Panel, context);
+
+        const varPart = result.parts[0] as ParsedChatRequestVariablePart;
+        expect(varPart.variableName).to.equal('prompt');
+        expect(varPart.variableArg).to.equal('compare|item1 item2');
+    });
+
+    it('parses command with quoted arguments', async () => {
+        const req: ChatRequest = {
+            text: '/cmd "arg with spaces" other'
+        };
+        const context: ChatContext = { variables: [] };
+        const result = await parser.parseChatRequest(req, ChatAgentLocation.Panel, context);
+
+        const varPart = result.parts[0] as ParsedChatRequestVariablePart;
+        expect(varPart.variableArg).to.equal('cmd|"arg with spaces" other');
+    });
+
+    it('handles command with escaped quotes', async () => {
+        const req: ChatRequest = {
+            text: '/cmd "arg with \\"quote\\"" other'
+        };
+        const context: ChatContext = { variables: [] };
+        const result = await parser.parseChatRequest(req, ChatAgentLocation.Panel, context);
+
+        const varPart = result.parts[0] as ParsedChatRequestVariablePart;
+        expect(varPart.variableArg).to.equal('cmd|"arg with \\"quote\\"" other');
+    });
 });

package/src/common/chat-request-parser.ts

@@ -31,6 +31,7 @@ import {
     ParsedChatRequestTextPart,
     ParsedChatRequestVariablePart,
     chatVariableLeader,
+    chatSubcommandLeader,
     OffsetRange,
     ParsedChatRequest,
     ParsedChatRequestPart,
@@ -42,6 +43,7 @@ const agentReg = /^@([\w_\-\.]+)(?=(\s|$|\b))/i; // An @-agent
 const functionReg = /^~([\w_\-\.]+)(?=(\s|$|\b))/i; // A ~ tool function
 const functionPromptFormatReg = /^\~\{\s*(.*?)\s*\}/i; // A ~{} prompt-format tool function
 const variableReg = /^#([\w_\-]+)(?::([\w_\-_\/\\.:]+))?(?=(\s|$|\b))/i; // A #-variable with an optional : arg (#file:workspace/path/name.ext)
+const commandReg = /^\/([\w_\-]+)(?:\s+(.+?))?(?=\s*$)/; // A /-command with optional arguments (/commandname arg1 arg2)
 
 export const ChatRequestParser = Symbol('ChatRequestParser');
 export interface ChatRequestParser {
@@ -114,7 +116,21 @@ export class ChatRequestParserImpl implements ChatRequestParser {
             let newPart: ParsedChatRequestPart | undefined;
 
             if (previousChar.match(/\s/) || i === 0) {
-                if (char === chatFunctionLeader) {
+                if (char === chatSubcommandLeader) {
+                    // Try to parse as command - commands are syntactic sugar for #prompt:commandName|args
+                    const commandPart = this.tryToParseCommand(
+                        message.slice(i),
+                        i,
+                        parts
+                    );
+                    if (commandPart) {
+                        newPart = commandPart;
+                        const variable = this.variableService.getVariable(commandPart.variableName);
+                        if (variable) {
+                            variables.set(variable.name, variable);
+                        }
+                    }
+                } else if (char === chatFunctionLeader) {
                     const functionPart = this.tryToParseFunction(
                         message.slice(i),
                         i
@@ -248,6 +264,29 @@ export class ChatRequestParserImpl implements ChatRequestParser {
         return new ParsedChatRequestVariablePart(varRange, name, variableArg);
     }
 
+    /**
+     * Try to parse a command at the start of the given message.
+     *
+     * Commands are syntactic sugar for `#prompt:commandName|args`.
+     * The prompt variable resolver will handle the actual resolution.
+     */
+    protected tryToParseCommand(
+        message: string,
+        offset: number,
+        _parts: ReadonlyArray<ParsedChatRequestPart>
+    ): ParsedChatRequestVariablePart | undefined {
+        const nextCommandMatch = message.match(commandReg);
+        if (!nextCommandMatch) {
+            return;
+        }
+
+        const [full, commandName, commandArgs] = nextCommandMatch;
+        const commandRange = offsetRange(offset, offset + full.length);
+
+        const variableArg = commandArgs ? `${commandName}|${commandArgs}` : commandName;
+        return new ParsedChatRequestVariablePart(commandRange, 'prompt', variableArg);
+    }
+
     private tryToParseFunction(message: string, offset: number): ParsedChatRequestFunctionPart | undefined {
         // Support both the and chat and prompt formats for functions
         const nextFunctionMatch = message.match(functionPromptFormatReg) || message.match(functionReg);

package/src/common/chat-session-naming-service.ts

@@ -26,7 +26,7 @@ import {
 } from '@theia/ai-core';
 import { inject, injectable } from '@theia/core/shared/inversify';
 import { ChatSession } from './chat-service';
-import { generateUuid } from '@theia/core';
+import { generateUuid, nls } from '@theia/core';
 
 import { CHAT_SESSION_NAMING_PROMPT } from './chat-session-naming-prompt-template';
 
@@ -47,7 +47,7 @@ export class ChatSessionNamingAgent implements Agent {
     static ID = 'Chat Session Naming';
     id = ChatSessionNamingAgent.ID;
     name = ChatSessionNamingAgent.ID;
-    description = 'Agent for generating chat session names';
+    description = nls.localize('theia/ai/chat/chatSessionNamingAgent/description', 'Agent for generating chat session names');
     variables = [];
     prompts = [CHAT_SESSION_NAMING_PROMPT];
     languageModelRequirements: LanguageModelRequirement[] = [{
@@ -55,8 +55,16 @@ export class ChatSessionNamingAgent implements Agent {
         identifier: 'default/summarize',
     }];
     agentSpecificVariables = [
-        { name: 'conversation', usedInPrompt: true, description: 'The content of the chat conversation.' },
-        { name: 'listOfSessionNames', usedInPrompt: true, description: 'The list of existing session names.' }
+        {
+            name: 'conversation',
+            usedInPrompt: true,
+            description: nls.localize('theia/ai/chat/chatSessionNamingAgent/vars/conversation/description', 'The content of the chat conversation.')
+        },
+        {
+            name: 'listOfSessionNames',
+            usedInPrompt: true,
+            description: nls.localize('theia/ai/chat/chatSessionNamingAgent/vars/listOfSessionNames/description', 'The list of existing session names.')
+        }
     ];
     functions = [];
 

package/src/common/chat-session-summary-agent-prompt.ts

@@ -24,7 +24,7 @@ export const CHAT_SESSION_SUMMARY_PROMPT = {
             'The summary will primarily be used by other AI agents, so tailor your response for use by AI agents. ' +
             'Also consider the system message. ' +
             'Make sure you include all necessary context information and use unique references (such as URIs, file paths, etc.). ' +
-            'If the conversation was about a task, describe the state of the task, i.e.what has been completed and what is open. ' +
+            'If the conversation was about a task, describe the state of the task, i.e. what has been completed and what is open. ' +
             'If a changeset is open in the session, describe the state of the suggested changes. ' +
             `\n\n{{${CHANGE_SET_SUMMARY_VARIABLE_ID}}}`,
     }

package/src/common/chat-session-summary-agent.ts

@@ -21,13 +21,14 @@ import {
 import { injectable } from '@theia/core/shared/inversify';
 import { AbstractStreamParsingChatAgent, ChatAgent } from './chat-agents';
 import { CHAT_SESSION_SUMMARY_PROMPT } from './chat-session-summary-agent-prompt';
+import { nls } from '@theia/core';
 
 @injectable()
 export class ChatSessionSummaryAgent extends AbstractStreamParsingChatAgent implements ChatAgent {
     static ID = 'chat-session-summary-agent';
     id = ChatSessionSummaryAgent.ID;
     name = 'Chat Session Summary';
-    override description = 'Agent for generating chat session summaries.';
+    override description = nls.localize('theia/ai/chat/chatSessionSummaryAgent/description', 'Agent for generating chat session summaries.');
     override prompts: PromptVariantSet[] = [CHAT_SESSION_SUMMARY_PROMPT];
     protected readonly defaultLanguageModelPurpose = 'chat-session-summary';
     languageModelRequirements: LanguageModelRequirement[] = [{

package/src/common/chat-tool-preferences.ts

@@ -70,7 +70,7 @@ export const chatToolPreferences: PreferenceSchema = {
             },
             default: {},
             description: nls.localize('theia/ai/chat/toolConfirmation/description',
-                'Configure confirmation behavior for different tools. Key is the tool ID, value is the confirmation mode.' +
+                'Configure confirmation behavior for different tools. Key is the tool ID, value is the confirmation mode. ' +
                 'Use "*" as the key to set a global default for all tools.'),
             title: AI_CORE_PREFERENCES_TITLE,
         }
@@ -80,4 +80,3 @@ export const chatToolPreferences: PreferenceSchema = {
 export interface ChatToolConfiguration {
     [TOOL_CONFIRMATION_PREFERENCE]: { [toolId: string]: ToolConfirmationMode };
 }
-

package/src/common/image-context-variable.ts

@@ -19,19 +19,34 @@ import {
     AIVariableResolutionRequest,
     ResolvedAIContextVariable
 } from '@theia/ai-core';
+import { nls } from '@theia/core';
 
 export const IMAGE_CONTEXT_VARIABLE: AIVariable = {
     id: 'imageContext',
-    description: 'Provides context information for an image',
+    description: nls.localize('theia/ai/chat/imageContextVariable/description', 'Provides context information for an image'),
     name: 'imageContext',
-    label: 'Image File',
+    label: nls.localize('theia/ai/chat/imageContextVariable/label', 'Image File'),
     iconClasses: ['codicon', 'codicon-file-media'],
     isContextVariable: true,
     args: [
-        { name: 'name', description: 'The name of the image file if available.', isOptional: true },
-        { name: 'wsRelativePath', description: 'The workspace-relative path of the image file if available.', isOptional: true },
-        { name: 'data', description: 'The image data in base64.' },
-        { name: 'mimeType', description: 'The mimetype of the image.' }
+        {
+            name: 'name',
+            description: nls.localize('theia/ai/chat/imageContextVariable/args/name/description', 'The name of the image file if available.'),
+            isOptional: true
+        },
+        {
+            name: 'wsRelativePath',
+            description: nls.localize('theia/ai/chat/imageContextVariable/args/wsRelativePath/description', 'The workspace-relative path of the image file if available.'),
+            isOptional: true
+        },
+        {
+            name: 'data',
+            description: nls.localize('theia/ai/chat/imageContextVariable/args/data/description', 'The image data in base64.')
+        },
+        {
+            name: 'mimeType',
+            description: nls.localize('theia/ai/chat/imageContextVariable/args/mimeType/description', 'The mimetype of the image.')
+        }
     ]
 };