@theia/ai-chat 1.59.0-next.62 → 1.59.0

package/src/browser/ai-chat-frontend-module.ts

@@ -42,6 +42,9 @@ import { ChangeSetFileService } from './change-set-file-service';
 import { ContextVariableLabelProvider } from './context-variable-label-provider';
 import { ContextFileVariableLabelProvider } from './context-file-variable-label-provider';
 import { FileChatVariableContribution } from './file-chat-variable-contribution';
+import { ContextSummaryVariableContribution } from '../common/context-summary-variable';
+import { ContextDetailsVariableContribution } from '../common/context-details-variable';
+import { ChangeSetVariableContribution } from './change-set-variable';
 
 export default new ContainerModule(bind => {
     bindContributionProvider(bind, Agent);
@@ -102,4 +105,7 @@ export default new ContainerModule(bind => {
     bind(ResourceResolver).toService(ChangeSetFileResourceResolver);
     bind(ToolCallChatResponseContentFactory).toSelf().inSingletonScope();
     bind(AIVariableContribution).to(FileChatVariableContribution).inSingletonScope();
+    bind(AIVariableContribution).to(ContextSummaryVariableContribution).inSingletonScope();
+    bind(AIVariableContribution).to(ContextDetailsVariableContribution).inSingletonScope();
+    bind(AIVariableContribution).to(ChangeSetVariableContribution).inSingletonScope();
 });

package/src/browser/change-set-file-element.ts

@@ -16,6 +16,7 @@
 
 import { DisposableCollection, Emitter, URI } from '@theia/core';
 import { inject, injectable, postConstruct } from '@theia/core/shared/inversify';
+import { Replacement } from '@theia/core/lib/common/content-replacer';
 import { ChangeSetElement, ChangeSetImpl } from '../common';
 import { ChangeSetFileResourceResolver, createChangeSetFileUri, UpdatableReferenceResource } from './change-set-file-resource';
 import { ChangeSetFileService } from './change-set-file-service';
@@ -39,6 +40,11 @@ export interface ChangeSetElementArgs extends Partial<ChangeSetElement> {
      * If `undefined`, there is no change.
      */
     targetState?: string;
+    /**
+     * An array of replacements used to create the new content for the targetState.
+     * This is only available if the agent was able to provide replacements and we were able to apply them.
+     */
+    replacements?: Replacement[];
 };
 
 @injectable()
@@ -149,6 +155,10 @@ export class ChangeSetFileElement implements ChangeSetElement {
         }
     }
 
+    get replacements(): Replacement[] | undefined {
+        return this.elementProps.replacements;
+    }
+
     get type(): 'add' | 'modify' | 'delete' | undefined {
         return this.elementProps.type;
     }

package/src/browser/change-set-variable.ts

@@ -0,0 +1,54 @@
+// *****************************************************************************
+// 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 { MaybePromise, nls } from '@theia/core';
+import { inject, injectable } from '@theia/core/shared/inversify';
+import { AIVariable, ResolvedAIVariable, AIVariableContribution, AIVariableResolver, AIVariableService, AIVariableResolutionRequest, AIVariableContext } from '@theia/ai-core';
+import { WorkspaceService } from '@theia/workspace/lib/browser';
+import { ChatSessionContext } from '../common';
+
+export const CHANGE_SET_SUMMARY_VARIABLE: AIVariable = {
+    id: 'changeSetSummary',
+    description: nls.localize('theia/ai/core/changeSetSummaryVariable/description', 'Provides a summary of the files in a change set and their contents.'),
+
+    name: 'changeSetSummary',
+};
+
+@injectable()
+export class ChangeSetVariableContribution implements AIVariableContribution, AIVariableResolver {
+    @inject(WorkspaceService)
+    protected readonly workspaceService: WorkspaceService;
+
+    registerVariables(service: AIVariableService): void {
+        service.registerResolver(CHANGE_SET_SUMMARY_VARIABLE, this);
+    }
+
+    canResolve(request: AIVariableResolutionRequest, context: AIVariableContext): MaybePromise<number> {
+        return request.variable.name === CHANGE_SET_SUMMARY_VARIABLE.name ? 50 : 0;
+    }
+
+    async resolve(request: AIVariableResolutionRequest, context: AIVariableContext): Promise<ResolvedAIVariable | undefined> {
+        if (!ChatSessionContext.is(context) || request.variable.name !== CHANGE_SET_SUMMARY_VARIABLE.name || !context.model.changeSet?.getElements().length) { return undefined; }
+        const entries = await Promise.all(
+            context.model.changeSet.getElements().map(async element => `- file: ${await this.workspaceService.getWorkspaceRelativePath(element.uri)}, status: ${element.state}`)
+        );
+        return {
+            variable: CHANGE_SET_SUMMARY_VARIABLE,
+            value: entries.join('\n')
+        };
+    }
+}
+

package/src/common/chat-agents.ts

@@ -21,6 +21,7 @@
 
 import {
     AgentSpecificVariables,
+    AIVariableContext,
     CommunicationRecordingService,
     getTextOfResponse,
     LanguageModel,
@@ -50,7 +51,8 @@ import {
     ChatResponseContent,
     ErrorChatResponseContentImpl,
     MarkdownChatResponseContentImpl,
-    ToolCallChatResponseContentImpl
+    ToolCallChatResponseContentImpl,
+    ChatRequestModel
 } from './chat-model';
 import { findFirstMatch, parseContents } from './parse-contents';
 import { DefaultResponseContentFactory, ResponseContentMatcher, ResponseContentMatcherProvider } from './response-content-matcher';
@@ -86,6 +88,17 @@ export namespace SystemMessageDescription {
     }
 }
 
+export interface ChatSessionContext extends AIVariableContext {
+    request?: ChatRequestModel;
+    model: ChatModel;
+}
+
+export namespace ChatSessionContext {
+    export function is(candidate: unknown): candidate is ChatSessionContext {
+        return typeof candidate === 'object' && !!candidate && 'model' in candidate;
+    }
+}
+
 /**
  * The location from where an chat agent may be invoked.
  * Based on the location, a different context may be available.
@@ -168,8 +181,7 @@ export abstract class AbstractChatAgent implements ChatAgent {
             if (!languageModel) {
                 throw new Error('Couldn\'t find a matching language model. Please check your setup!');
             }
-
-            const systemMessageDescription = await this.getSystemMessageDescription();
+            const systemMessageDescription = await this.getSystemMessageDescription({ model: request.session, request } satisfies ChatSessionContext);
             const messages = await this.getMessages(request.session);
             if (this.defaultLogging) {
                 this.recordingService.recordRequest(
@@ -244,11 +256,11 @@ export abstract class AbstractChatAgent implements ChatAgent {
         return languageModel;
     }
 
-    protected async getSystemMessageDescription(): Promise<SystemMessageDescription | undefined> {
+    protected async getSystemMessageDescription(context: AIVariableContext): Promise<SystemMessageDescription | undefined> {
         if (this.systemPromptId === undefined) {
             return undefined;
         }
-        const resolvedPrompt = await this.promptService.getPrompt(this.systemPromptId);
+        const resolvedPrompt = await this.promptService.getPrompt(this.systemPromptId, undefined, context);
         return resolvedPrompt ? SystemMessageDescription.fromResolvedPromptTemplate(resolvedPrompt) : undefined;
     }
 

package/src/common/chat-model.ts

@@ -24,7 +24,7 @@ import { MarkdownString, MarkdownStringImpl } from '@theia/core/lib/common/markd
 import { Position } from '@theia/core/shared/vscode-languageserver-protocol';
 import { ChatAgentLocation } from './chat-agents';
 import { ParsedChatRequest } from './parsed-chat-request';
-import { ResolvedAIContextVariable } from '@theia/ai-core';
+import { AIVariableResolutionRequest, ResolvedAIContextVariable } from '@theia/ai-core';
 
 /**********************
  * INTERFACES AND TYPE GUARDS
@@ -33,6 +33,8 @@ import { ResolvedAIContextVariable } from '@theia/ai-core';
 export type ChatChangeEvent =
     | ChatAddRequestEvent
     | ChatAddResponseEvent
+    | ChatAddVariableEvent
+    | ChatRemoveVariableEvent
     | ChatRemoveRequestEvent
     | ChatSetChangeSetEvent
     | ChatUpdateChangeSetEvent
@@ -64,6 +66,14 @@ export interface ChatRemoveChangeSetEvent {
     changeSet: ChangeSet;
 }
 
+export interface ChatAddVariableEvent {
+    kind: 'addVariable';
+}
+
+export interface ChatRemoveVariableEvent {
+    kind: 'removeVariable';
+}
+
 export namespace ChatChangeEvent {
     export function isChangeSetEvent(event: ChatChangeEvent): event is ChatSetChangeSetEvent | ChatUpdateChangeSetEvent | ChatRemoveChangeSetEvent {
         return event.kind === 'setChangeSet' || event.kind === 'removeChangeSet' || event.kind === 'updateChangeSet';
@@ -84,17 +94,26 @@ export interface ChatModel {
     readonly id: string;
     readonly location: ChatAgentLocation;
     readonly changeSet?: ChangeSet;
+    readonly context: ChatContextManager;
     getRequests(): ChatRequestModel[];
     isEmpty(): boolean;
 }
 
-export interface ChangeSet {
+export interface ChangeSet extends Disposable {
     onDidChange: Event<ChangeSetChangeEvent>;
     readonly title: string;
     getElements(): ChangeSetElement[];
     dispose(): void;
 }
 
+export interface ChatContextManager {
+    onDidChange: Event<ChatAddVariableEvent | ChatRemoveVariableEvent>;
+    getVariables(): readonly AIVariableResolutionRequest[]
+    addVariables(...variables: AIVariableResolutionRequest[]): void;
+    deleteVariables(...indices: number[]): void;
+    clear(): void;
+}
+
 export interface ChangeSetElement {
     readonly uri: URI;
 
@@ -478,11 +497,13 @@ export class MutableChatModel implements ChatModel, Disposable {
     protected _requests: MutableChatRequestModel[];
     protected _id: string;
     protected _changeSet?: ChangeSetImpl;
+    protected readonly _contextManager = new ChatContextManagerImpl();
 
     constructor(public readonly location = ChatAgentLocation.Panel) {
         // TODO accept serialized data as a parameter to restore a previously saved ChatModel
         this._requests = [];
         this._id = generateUuid();
+        this._contextManager.onDidChange(e => this._onDidChangeEmitter.fire(e));
     }
 
     getRequests(): MutableChatRequestModel[] {
@@ -501,6 +522,10 @@ export class MutableChatModel implements ChatModel, Disposable {
         return this._changeSet;
     }
 
+    get context(): ChatContextManager {
+        return this._contextManager;
+    }
+
     setChangeSet(changeSet: ChangeSetImpl | undefined): void {
         if (!changeSet) {
             return this.removeChangeSet();
@@ -610,8 +635,53 @@ export class ChangeSetImpl implements ChangeSet {
     }
 
     dispose(): void {
-        this._elements.forEach(element => element.dispose?.());
         this._onDidChangeEmitter.dispose();
+        this._elements.forEach(element => element.dispose?.());
+    }
+}
+
+export class ChatContextManagerImpl implements ChatContextManager {
+    protected readonly variables = new Array<AIVariableResolutionRequest>();
+    protected readonly onDidChangeEmitter = new Emitter<ChatAddVariableEvent | ChatRemoveVariableEvent>();
+    get onDidChange(): Event<ChatAddVariableEvent | ChatRemoveVariableEvent> {
+        return this.onDidChangeEmitter.event;
+    }
+
+    getVariables(): readonly AIVariableResolutionRequest[] {
+        const result = this.variables.slice();
+        Object.freeze(result);
+        return result;
+    }
+
+    addVariables(...variables: AIVariableResolutionRequest[]): void {
+        let modified = false;
+        variables.forEach(variable => {
+            if (this.variables.some(existing => existing.variable.id === variable.variable.id && existing.arg === variable.arg)) {
+                return;
+            }
+            this.variables.push(variable);
+            modified = true;
+        });
+        if (modified) {
+            this.onDidChangeEmitter.fire({ kind: 'addVariable' });
+        }
+    }
+
+    deleteVariables(...indices: number[]): void {
+        const toDelete = indices.filter(candidate => candidate <= this.variables.length).sort((left, right) => right - left);
+        if (toDelete.length) {
+            toDelete.forEach(index => {
+                this.variables.splice(index, 1);
+            });
+            this.onDidChangeEmitter.fire({ kind: 'removeVariable' });
+        }
+    }
+
+    clear(): void {
+        if (this.variables.length) {
+            this.variables.length = 0;
+            this.onDidChangeEmitter.fire({ kind: 'removeVariable' });
+        }
     }
 }
 

package/src/common/chat-service.ts

@@ -24,7 +24,7 @@ import { Emitter, ILogger, generateUuid } from '@theia/core';
 import { inject, injectable, optional } from '@theia/core/shared/inversify';
 import { Event } from '@theia/core/shared/vscode-languageserver-protocol';
 import { ChatAgentService } from './chat-agent-service';
-import { ChatAgent, ChatAgentLocation } from './chat-agents';
+import { ChatAgent, ChatAgentLocation, ChatSessionContext } from './chat-agents';
 import {
     ChatModel,
     MutableChatModel,
@@ -60,10 +60,6 @@ export interface ChatSession {
     pinnedAgent?: ChatAgent;
 }
 
-export interface ChatContextRequest {
-    variableRequests: AIVariableResolutionRequest[]
-}
-
 export interface ActiveSessionChangedEvent {
     sessionId: string | undefined;
     focus?: boolean;
@@ -104,8 +100,7 @@ export interface ChatService {
 
     sendRequest(
         sessionId: string,
-        request: ChatRequest,
-        requestedContext?: ChatContextRequest
+        request: ChatRequest
     ): Promise<ChatRequestInvocation | undefined>;
 
     deleteChangeSet(sessionId: string): void;
@@ -169,7 +164,7 @@ export class ChatServiceImpl implements ChatService {
 
     deleteSession(sessionId: string): void {
         const sessionIndex = this._sessions.findIndex(candidate => candidate.id === sessionId);
-        if (~sessionIndex) { return; }
+        if (sessionIndex === -1) { return; }
         const session = this._sessions[sessionIndex];
         // If the removed session is the active one, set the newest one as active
         if (session.isActive) {
@@ -189,7 +184,6 @@ export class ChatServiceImpl implements ChatService {
     async sendRequest(
         sessionId: string,
         request: ChatRequest,
-        requestedContext: ChatContextRequest = { variableRequests: [] }
     ): Promise<ChatRequestInvocation | undefined> {
         const session = this.getSession(sessionId);
         if (!session) {
@@ -211,14 +205,16 @@ export class ChatServiceImpl implements ChatService {
             };
         }
 
-        const resolvedContext = await this.resolveChatContext(requestedContext, request, session);
+        const resolutionContext: ChatSessionContext = { model: session.model };
+        const resolvedContext = await this.resolveChatContext(session.model.context.getVariables(), resolutionContext);
         const requestModel = session.model.addRequest(parsedRequest, agent?.id, resolvedContext);
+        resolutionContext.request = requestModel;
 
         for (const part of parsedRequest.parts) {
             if (part instanceof ParsedChatRequestVariablePart) {
                 const resolvedVariable = await this.variableService.resolveVariable(
                     { variable: part.variableName, arg: part.variableArg },
-                    { request, model: session }
+                    resolutionContext
                 );
                 if (resolvedVariable) {
                     part.resolution = resolvedVariable;
@@ -260,13 +256,12 @@ export class ChatServiceImpl implements ChatService {
     }
 
     protected async resolveChatContext(
-        requestedContext: ChatContextRequest,
-        request: ChatRequest,
-        session: ChatSessionInternal
+        resolutionRequests: readonly AIVariableResolutionRequest[],
+        context: ChatSessionContext,
     ): Promise<ChatContext> {
         const resolvedVariables = await Promise.all(
-            requestedContext.variableRequests.map(async contextVariable => {
-                const resolvedVariable = await this.variableService.resolveVariable(contextVariable, { request, model: session });
+            resolutionRequests.map(async contextVariable => {
+                const resolvedVariable = await this.variableService.resolveVariable(contextVariable, context);
                 if (ResolvedAIContextVariable.is(resolvedVariable)) {
                     return resolvedVariable;
                 }
@@ -317,7 +312,7 @@ export class ChatServiceImpl implements ChatService {
     }
 
     deleteChangeSet(sessionId: string): void {
-        this.getSession(sessionId)?.model.setChangeSet(undefined);
+        this.getSession(sessionId)?.model.removeChangeSet();
     }
 
     deleteChangeSetElement(sessionId: string, index: number): void {

package/src/common/chat-string-utils.ts

@@ -0,0 +1,23 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+export function stringJsonCodeBlock(input: string): string {
+    return `\`\`\`json\n${input}\n\`\`\``;
+}
+
+export function dataToJsonCodeBlock(input: unknown): string {
+    return stringJsonCodeBlock(JSON.stringify(input, undefined, 2));
+}

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

@@ -0,0 +1,53 @@
+// *****************************************************************************
+// 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 { MaybePromise, nls } from '@theia/core';
+import { injectable } from '@theia/core/shared/inversify';
+import { AIVariable, ResolvedAIVariable, AIVariableContribution, AIVariableResolver, AIVariableService, AIVariableResolutionRequest, AIVariableContext } from '@theia/ai-core';
+import { dataToJsonCodeBlock } from './chat-string-utils';
+import { ChatSessionContext } from './chat-agents';
+import { CHAT_CONTEXT_DETAILS_VARIABLE_ID } from './context-variables';
+
+export const CONTEXT_DETAILS_VARIABLE: AIVariable = {
+    id: CHAT_CONTEXT_DETAILS_VARIABLE_ID,
+    description: nls.localize('theia/ai/core/contextDetailsVariable/description', 'Provides full text values and descriptions for all context elements.'),
+    name: CHAT_CONTEXT_DETAILS_VARIABLE_ID,
+};
+
+@injectable()
+export class ContextDetailsVariableContribution implements AIVariableContribution, AIVariableResolver {
+    registerVariables(service: AIVariableService): void {
+        service.registerResolver(CONTEXT_DETAILS_VARIABLE, this);
+    }
+
+    canResolve(request: AIVariableResolutionRequest, context: AIVariableContext): MaybePromise<number> {
+        return request.variable.name === CONTEXT_DETAILS_VARIABLE.name ? 50 : 0;
+    }
+
+    async resolve(request: AIVariableResolutionRequest, context: AIVariableContext): Promise<ResolvedAIVariable | undefined> {
+        /** By expecting context.request, we're assuming that this variable will not be resolved until the context has been resolved. */
+        if (!ChatSessionContext.is(context) || request.variable.name !== CONTEXT_DETAILS_VARIABLE.name || !context.request) { return undefined; }
+        const data = context.request.context.variables.map(variable => ({
+            type: variable.variable.name,
+            ref: variable.value,
+            content: variable.contextValue
+        }));
+        return {
+            variable: CONTEXT_DETAILS_VARIABLE,
+            value: dataToJsonCodeBlock(data)
+        };
+    }
+}

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

@@ -0,0 +1,53 @@
+// *****************************************************************************
+// 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 { MaybePromise, nls } from '@theia/core';
+import { injectable } from '@theia/core/shared/inversify';
+import { AIVariable, ResolvedAIVariable, AIVariableContribution, AIVariableResolver, AIVariableService, AIVariableResolutionRequest, AIVariableContext } from '@theia/ai-core';
+import { dataToJsonCodeBlock } from './chat-string-utils';
+import { ChatSessionContext } from './chat-agents';
+
+export const CONTEXT_SUMMARY_VARIABLE: AIVariable = {
+    id: 'contextSummary',
+    description: nls.localize('theia/ai/core/contextSummaryVariable/description', 'Describes files in the context for a given session.'),
+    name: 'contextSummary',
+};
+
+@injectable()
+export class ContextSummaryVariableContribution implements AIVariableContribution, AIVariableResolver {
+    registerVariables(service: AIVariableService): void {
+        service.registerResolver(CONTEXT_SUMMARY_VARIABLE, this);
+    }
+
+    canResolve(request: AIVariableResolutionRequest, context: AIVariableContext): MaybePromise<number> {
+        return request.variable.name === CONTEXT_SUMMARY_VARIABLE.name ? 50 : 0;
+    }
+
+    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)
+            .map(variable => ({
+                type: variable.variable.name,
+                // eslint-disable-next-line no-null/no-null
+                instanceData: variable.arg || null,
+                contextElementId: variable.variable.id + variable.arg
+            }));
+        return {
+            variable: CONTEXT_SUMMARY_VARIABLE,
+            value: dataToJsonCodeBlock(data)
+        };
+    }
+}

package/src/common/context-variables.ts

@@ -0,0 +1,19 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+export const CHAT_CONTEXT_VARIABLE_ID = 'contextSummary';
+export const CHAT_CONTEXT_DETAILS_VARIABLE_ID = 'contextDetails';
+export const CHANGE_SET_SUMMARY_VARIABLE_ID = 'changeSetSummary';

package/src/common/custom-chat-agent.ts

@@ -24,9 +24,10 @@ export class CustomChatAgent extends AbstractStreamParsingChatAgent {
     name: string = 'CustomChatAgent';
     languageModelRequirements: LanguageModelRequirement[] = [{ purpose: 'chat' }];
     protected defaultLanguageModelPurpose: string = 'chat';
-    protected override systemPromptId: string = `${this.name}_prompt`;
 
     set prompt(prompt: string) {
+        // the name is dynamic, so we set the propmptId here
+        this.systemPromptId = `${this.name}_prompt`;
         this.promptTemplates.push({ id: `${this.name}_prompt`, template: prompt });
     }
 }

package/src/common/index.ts

@@ -21,3 +21,4 @@ export * from './chat-request-parser';
 export * from './chat-service';
 export * from './custom-chat-agent';
 export * from './parsed-chat-request';
+export * from './context-variables';