@theia/ai-chat 1.61.0-next.8 → 1.61.0

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

@@ -55,7 +55,7 @@ function offsetRange(start: number, endExclusive: number): OffsetRange {
     return { start, endExclusive };
 }
 @injectable()
-export class ChatRequestParserImpl {
+export class ChatRequestParserImpl implements ChatRequestParser {
     constructor(
         @inject(ChatAgentService) private readonly agentService: ChatAgentService,
         @inject(AIVariableService) private readonly variableService: AIVariableService,
@@ -90,7 +90,7 @@ export class ChatRequestParserImpl {
         }
 
         // Get resolved variables from variable cache after all variables have been resolved.
-        // We want to return all recursilvely resolved variables, thus use the whole cache.
+        // We want to return all recursively resolved variables, thus use the whole cache.
         const resolvedVariables = await getAllResolvedAIVariables(variableCache);
 
         return { request, parts, toolRequests, variables: resolvedVariables };

package/src/common/chat-service.ts

@@ -38,6 +38,7 @@ import {
 import { ChatRequestParser } from './chat-request-parser';
 import { ParsedChatRequest, ParsedChatRequestAgentPart } from './parsed-chat-request';
 import { ChatSessionNamingService } from './chat-session-naming-service';
+import { Deferred } from '@theia/core/lib/common/promise-util';
 
 export interface ChatRequestInvocation {
     /**
@@ -125,6 +126,7 @@ export interface ChatService {
     getSessions(): ChatSession[];
     createSession(location?: ChatAgentLocation, options?: SessionOptions, pinnedAgent?: ChatAgent): ChatSession;
     deleteSession(sessionId: string): void;
+    getActiveSession(): ChatSession | undefined;
     setActiveSession(sessionId: string, options?: SessionOptions): void;
 
     sendRequest(
@@ -208,6 +210,12 @@ export class ChatServiceImpl implements ChatService {
         this.onSessionEventEmitter.fire({ type: 'deleted', sessionId: sessionId });
     }
 
+    getActiveSession(): ChatSession | undefined {
+        const activeSessions = this._sessions.filter(candidate => candidate.isActive);
+        if (activeSessions.length > 1) { throw new Error('More than one session marked as active. This indicates an error in ChatService.'); }
+        return activeSessions.at(0);
+    }
+
     setActiveSession(sessionId: string | undefined, options?: SessionOptions): void {
         this._sessions.forEach(session => {
             session.isActive = session.id === sessionId;
@@ -225,7 +233,7 @@ export class ChatServiceImpl implements ChatService {
         }
 
         const resolutionContext: ChatSessionContext = { model: session.model };
-        const resolvedContext = await this.resolveChatContext(session.model.context.getVariables(), resolutionContext);
+        const resolvedContext = await this.resolveChatContext(request.variables ?? session.model.context.getVariables(), resolutionContext);
         const parsedRequest = await this.chatRequestParser.parseChatRequest(request, session.model.location, resolvedContext);
         const agent = this.getAgent(parsedRequest, session);
 
@@ -244,33 +252,23 @@ export class ChatServiceImpl implements ChatService {
         this.updateSessionMetadata(session, requestModel);
         resolutionContext.request = requestModel;
 
-        let resolveResponseCreated: (responseModel: ChatResponseModel) => void;
-        let resolveResponseCompleted: (responseModel: ChatResponseModel) => void;
+        const responseCompletionDeferred = new Deferred<ChatResponseModel>();
         const invocation: ChatRequestInvocation = {
             requestCompleted: Promise.resolve(requestModel),
-            responseCreated: new Promise(resolve => {
-                resolveResponseCreated = resolve;
-            }),
-            responseCompleted: new Promise(resolve => {
-                resolveResponseCompleted = resolve;
-            }),
+            responseCreated: Promise.resolve(requestModel.response),
+            responseCompleted: responseCompletionDeferred.promise,
         };
 
-        resolveResponseCreated!(requestModel.response);
         requestModel.response.onDidChange(() => {
             if (requestModel.response.isComplete) {
-                resolveResponseCompleted!(requestModel.response);
+                responseCompletionDeferred.resolve(requestModel.response);
             }
             if (requestModel.response.isError) {
-                resolveResponseCompleted!(requestModel.response);
+                responseCompletionDeferred.resolve(requestModel.response);
             }
         });
 
-        if (agent) {
-            agent.invoke(requestModel).catch(error => requestModel.response.error(error));
-        } else {
-            this.logger.error('No ChatAgents available to handle request!', requestModel);
-        }
+        agent.invoke(requestModel).catch(error => requestModel.response.error(error));
 
         return invocation;
     }
@@ -304,15 +302,8 @@ export class ChatServiceImpl implements ChatService {
         context: ChatSessionContext,
     ): Promise<ChatContext> {
         // TODO use a common cache to resolve variables and return recursively resolved variables?
-        const resolvedVariables = await Promise.all(
-            resolutionRequests.map(async contextVariable => {
-                const resolvedVariable = await this.variableService.resolveVariable(contextVariable, context);
-                if (ResolvedAIContextVariable.is(resolvedVariable)) {
-                    return resolvedVariable;
-                }
-                return undefined;
-            })
-        ).then(results => results.filter((result): result is ResolvedAIContextVariable => result !== undefined));
+        const resolvedVariables = await Promise.all(resolutionRequests.map(async contextVariable => this.variableService.resolveVariable(contextVariable, context)))
+            .then(results => results.filter(ResolvedAIContextVariable.is));
         return { variables: resolvedVariables };
     }
 

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

@@ -18,6 +18,7 @@ import {
     Agent,
     AgentService,
     CommunicationRecordingService,
+    CommunicationRequestEntryParam,
     getTextOfResponse,
     LanguageModelRegistry,
     LanguageModelRequirement,
@@ -38,7 +39,7 @@ const CHAT_SESSION_NAMING_PROMPT = {
         'Use the same language for the chat conversation name as used in the provided conversation, if in doubt default to English. ' +
         'Start the chat conversation name with an upper-case letter. ' +
         'Below we also provide the already existing other conversation names, make sure your suggestion for a name is unique with respect to the existing ones.\n\n' +
-        'IMPORTANT: Your answer MUST ONLY CONTAIN THE PROPOSED NAME and must not be preceded or succeeded with any other text.' +
+        'IMPORTANT: Your answer MUST ONLY CONTAIN THE PROPOSED NAME and must not be preceded or followed by any other text.' +
         '\n\nOther session names:\n{{listOfSessionNames}}' +
         '\n\nConversation:\n{{conversation}}',
 };
@@ -92,7 +93,7 @@ export class ChatSessionNamingAgent implements Agent {
         }
 
         const conversation = chatSession.model.getRequests()
-            .map(req => `<user>${req.request.text}</user>` +
+            .map(req => `<user>${req.message.parts.map(chunk => chunk.promptText).join('')}</user>` +
                 (req.response.response ? `<assistant>${req.response.response.asString()}</assistant>` : ''))
             .join('\n\n');
         const listOfSessionNames = otherNames.map(name => name).join(', ');
@@ -115,7 +116,7 @@ export class ChatSessionNamingAgent implements Agent {
             sessionId,
             agentId: this.id
         };
-        this.recordingService.recordRequest(request);
+        this.recordingService.recordRequest({ ...request, request: request.messages } satisfies CommunicationRequestEntryParam);
 
         const result = await lm.request(request);
         const response = await getTextOfResponse(result);

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

@@ -0,0 +1,28 @@
+/* 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 { CHANGE_SET_SUMMARY_VARIABLE_ID } from './context-variables';
+
+export const CHAT_SESSION_SUMMARY_PROMPT = {
+    id: 'chat-session-summary-prompt',
+    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 --}}\n\n' +
+        'You are a chat agent for summarizing AI agent chat sessions for later use. \
+Review the conversation above and generate a concise summary that captures every crucial detail, \
+including all requirements, decisions, and pending tasks. \
+Ensure that the summary is sufficiently comprehensive to allow seamless continuation of the workflow. 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 a changeset is open in the session, describe the state of the suggested changes.
+{{${CHANGE_SET_SUMMARY_VARIABLE_ID}}}`,
+};

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

@@ -0,0 +1,42 @@
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH and others.
+//
+// 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 {
+    LanguageModelRequirement,
+    PromptTemplate
+} from '@theia/ai-core';
+import { injectable } from '@theia/core/shared/inversify';
+import { AbstractStreamParsingChatAgent, ChatAgent } from './chat-agents';
+import { CHAT_SESSION_SUMMARY_PROMPT } from './chat-session-summary-agent-prompt';
+
+@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 variables = [];
+    override promptTemplates: PromptTemplate[] = [CHAT_SESSION_SUMMARY_PROMPT];
+    protected readonly defaultLanguageModelPurpose = 'chat-session-summary';
+    languageModelRequirements: LanguageModelRequirement[] = [{
+        purpose: 'chat-session-summary',
+        identifier: 'openai/gpt-4o-mini',
+    }];
+    override agentSpecificVariables = [];
+    override functions = [];
+    override locations = [];
+    override tags = [];
+}

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

@@ -38,7 +38,7 @@ export class ContextSummaryVariableContribution implements AIVariableContributio
 
     async resolve(request: AIVariableResolutionRequest, context: AIVariableContext): Promise<ResolvedAIVariable | undefined> {
         if (!ChatSessionContext.is(context) || request.variable.name !== CONTEXT_SUMMARY_VARIABLE.name) { return undefined; }
-        const data = context.model.context.getVariables().filter(variable => variable.variable.isContextVariable)
+        const data = ChatSessionContext.getVariables(context).filter(variable => variable.variable.isContextVariable)
             .map(variable => ({
                 type: variable.variable.name,
                 // eslint-disable-next-line no-null/no-null