@theia/ai-core 1.54.0

package/lib/browser/frontend-language-model-registry.js

@@ -0,0 +1,321 @@
+"use strict";
+var FrontendLanguageModelRegistryImpl_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FrontendLanguageModelRegistryImpl = exports.LanguageModelDelegateClientImpl = void 0;
+const tslib_1 = require("tslib");
+// *****************************************************************************
+// Copyright (C) 2024 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
+// *****************************************************************************
+const core_1 = require("@theia/core");
+const inversify_1 = require("@theia/core/shared/inversify");
+const output_channel_1 = require("@theia/output/lib/browser/output-channel");
+const common_1 = require("../common");
+let LanguageModelDelegateClientImpl = class LanguageModelDelegateClientImpl {
+    setReceiver(receiver) {
+        this.receiver = receiver;
+    }
+    send(id, token) {
+        this.receiver.send(id, token);
+    }
+    toolCall(requestId, toolId, args_string) {
+        return this.receiver.toolCall(requestId, toolId, args_string);
+    }
+    languageModelAdded(metadata) {
+        this.receiver.languageModelAdded(metadata);
+    }
+    languageModelRemoved(id) {
+        this.receiver.languageModelRemoved(id);
+    }
+};
+exports.LanguageModelDelegateClientImpl = LanguageModelDelegateClientImpl;
+exports.LanguageModelDelegateClientImpl = LanguageModelDelegateClientImpl = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], LanguageModelDelegateClientImpl);
+let FrontendLanguageModelRegistryImpl = FrontendLanguageModelRegistryImpl_1 = class FrontendLanguageModelRegistryImpl extends common_1.DefaultLanguageModelRegistryImpl {
+    constructor() {
+        super(...arguments);
+        this.streams = new Map();
+        this.requests = new Map();
+    }
+    // called by backend
+    languageModelAdded(metadata) {
+        this.addLanguageModels([metadata]);
+    }
+    // called by backend
+    languageModelRemoved(id) {
+        this.removeLanguageModels([id]);
+    }
+    addLanguageModels(models) {
+        let modelAdded = false;
+        for (const model of models) {
+            if (this.languageModels.find(m => m.id === model.id)) {
+                console.warn(`Tried to add an existing model ${model.id}`);
+                continue;
+            }
+            if (common_1.LanguageModel.is(model)) {
+                this.languageModels.push(new Proxy(model, languageModelOutputHandler(() => this.outputChannelManager.getChannel(model.id))));
+                modelAdded = true;
+            }
+            else {
+                this.languageModels.push(new Proxy(this.createFrontendLanguageModel(model), languageModelOutputHandler(() => this.outputChannelManager.getChannel(model.id))));
+                modelAdded = true;
+            }
+        }
+        if (modelAdded) {
+            this.changeEmitter.fire({ models: this.languageModels });
+        }
+    }
+    init() {
+        this.client.setReceiver(this);
+        const contributions = this.languageModelContributions.getContributions();
+        const promises = contributions.map(provider => provider());
+        const backendDescriptions = this.registryDelegate.getLanguageModelDescriptions();
+        Promise.allSettled([backendDescriptions, ...promises]).then(results => {
+            const backendDescriptionsResult = results[0];
+            if (backendDescriptionsResult.status === 'fulfilled') {
+                this.addLanguageModels(backendDescriptionsResult.value);
+            }
+            else {
+                this.logger.error('Failed to add language models contributed from the backend', backendDescriptionsResult.reason);
+            }
+            for (let i = 1; i < results.length; i++) {
+                // assert that index > 0 contains only language models
+                const languageModelResult = results[i];
+                if (languageModelResult.status === 'fulfilled') {
+                    this.addLanguageModels(languageModelResult.value);
+                }
+                else {
+                    this.logger.error('Failed to add some language models:', languageModelResult.reason);
+                }
+            }
+            this.markInitialized();
+        });
+    }
+    createFrontendLanguageModel(description) {
+        return {
+            ...description,
+            request: async (request, cancellationToken) => {
+                const requestId = `${FrontendLanguageModelRegistryImpl_1.requestCounter++}`;
+                this.requests.set(requestId, request);
+                cancellationToken === null || cancellationToken === void 0 ? void 0 : cancellationToken.onCancellationRequested(() => {
+                    this.providerDelegate.cancel(requestId);
+                });
+                const response = await this.providerDelegate.request(description.id, request, requestId, cancellationToken);
+                if ((0, common_1.isLanguageModelTextResponse)(response) || (0, common_1.isLanguageModelParsedResponse)(response)) {
+                    return response;
+                }
+                if ((0, common_1.isLanguageModelStreamResponseDelegate)(response)) {
+                    if (!this.streams.has(response.streamId)) {
+                        const newStreamState = {
+                            id: response.streamId,
+                            tokens: [],
+                        };
+                        this.streams.set(response.streamId, newStreamState);
+                    }
+                    const streamState = this.streams.get(response.streamId);
+                    return {
+                        stream: this.getIterable(streamState),
+                    };
+                }
+                this.logger.error(`Received unknown response in frontend for request to language model ${description.id}. Trying to continue without touching the response.`, response);
+                return response;
+            },
+        };
+    }
+    async *getIterable(state) {
+        let current = -1;
+        while (true) {
+            if (current < state.tokens.length - 1) {
+                current++;
+                const token = state.tokens[current];
+                if (token === undefined) {
+                    // message is finished
+                    break;
+                }
+                if (token !== undefined) {
+                    yield token;
+                }
+            }
+            else {
+                await new Promise(resolve => {
+                    state.resolve = resolve;
+                });
+            }
+        }
+        this.streams.delete(state.id);
+    }
+    // called by backend via the "delegate client" with new tokens
+    send(id, token) {
+        if (!this.streams.has(id)) {
+            const newStreamState = {
+                id,
+                tokens: [],
+            };
+            this.streams.set(id, newStreamState);
+        }
+        const streamState = this.streams.get(id);
+        streamState.tokens.push(token);
+        if (streamState.resolve) {
+            streamState.resolve(token);
+        }
+    }
+    // called by backend once tool is invoked
+    toolCall(id, toolId, arg_string) {
+        var _a;
+        if (!this.requests.has(id)) {
+            throw new Error('Somehow we got a callback for a non existing request!');
+        }
+        const request = this.requests.get(id);
+        const tool = (_a = request.tools) === null || _a === void 0 ? void 0 : _a.find(t => t.id === toolId);
+        if (tool) {
+            return tool.handler(arg_string);
+        }
+        throw new Error(`Could not find a tool for ${toolId}!`);
+    }
+    async selectLanguageModels(request) {
+        var _a, _b;
+        await this.initialized;
+        const userSettings = (_b = (_a = (await this.settingsService.getAgentSettings(request.agent))) === null || _a === void 0 ? void 0 : _a.languageModelRequirements) === null || _b === void 0 ? void 0 : _b.find(req => req.purpose === request.purpose);
+        if (userSettings === null || userSettings === void 0 ? void 0 : userSettings.identifier) {
+            const model = await this.getLanguageModel(userSettings.identifier);
+            if (model) {
+                return [model];
+            }
+        }
+        return this.languageModels.filter(model => (0, common_1.isModelMatching)(request, model));
+    }
+    async selectLanguageModel(request) {
+        return (await this.selectLanguageModels(request))[0];
+    }
+};
+exports.FrontendLanguageModelRegistryImpl = FrontendLanguageModelRegistryImpl;
+FrontendLanguageModelRegistryImpl.requestCounter = 0;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(common_1.LanguageModelRegistryFrontendDelegate),
+    tslib_1.__metadata("design:type", Object)
+], FrontendLanguageModelRegistryImpl.prototype, "registryDelegate", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(common_1.LanguageModelFrontendDelegate),
+    tslib_1.__metadata("design:type", Object)
+], FrontendLanguageModelRegistryImpl.prototype, "providerDelegate", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(LanguageModelDelegateClientImpl),
+    tslib_1.__metadata("design:type", LanguageModelDelegateClientImpl)
+], FrontendLanguageModelRegistryImpl.prototype, "client", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(core_1.ILogger),
+    tslib_1.__metadata("design:type", Object)
+], FrontendLanguageModelRegistryImpl.prototype, "logger", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(output_channel_1.OutputChannelManager),
+    tslib_1.__metadata("design:type", output_channel_1.OutputChannelManager)
+], FrontendLanguageModelRegistryImpl.prototype, "outputChannelManager", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(common_1.AISettingsService),
+    tslib_1.__metadata("design:type", Object)
+], FrontendLanguageModelRegistryImpl.prototype, "settingsService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.postConstruct)(),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", void 0)
+], FrontendLanguageModelRegistryImpl.prototype, "init", null);
+exports.FrontendLanguageModelRegistryImpl = FrontendLanguageModelRegistryImpl = FrontendLanguageModelRegistryImpl_1 = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], FrontendLanguageModelRegistryImpl);
+const formatJsonWithIndentation = (obj) => {
+    // eslint-disable-next-line no-null/no-null
+    const jsonString = JSON.stringify(obj, null, 2);
+    const lines = jsonString.split('\n');
+    const formattedLines = [];
+    lines.forEach(line => {
+        const subLines = line.split('\\n');
+        const index = indexOfValue(subLines[0]) + 1;
+        formattedLines.push(subLines[0]);
+        const prefix = index > 0 ? ' '.repeat(index) : '';
+        if (index !== -1) {
+            for (let i = 1; i < subLines.length; i++) {
+                formattedLines.push(prefix + subLines[i]);
+            }
+        }
+    });
+    return formattedLines;
+};
+const indexOfValue = (jsonLine) => {
+    const pattern = /"([^"]+)"\s*:\s*/g;
+    const match = pattern.exec(jsonLine);
+    return match ? match.index + match[0].length : -1;
+};
+const languageModelOutputHandler = (outputChannelGetter) => ({
+    get(target, prop) {
+        const original = target[prop];
+        if (prop === 'request' && typeof original === 'function') {
+            return async function (...args) {
+                const outputChannel = outputChannelGetter();
+                outputChannel.appendLine('Sending request:');
+                const formattedRequest = formatJsonWithIndentation(args[0]);
+                formattedRequest.forEach(line => outputChannel.appendLine(line));
+                if (args[1]) {
+                    args[1] = new Proxy(args[1], {
+                        get(cTarget, cProp) {
+                            if (cProp === 'onCancellationRequested') {
+                                return (...cargs) => cTarget.onCancellationRequested(() => {
+                                    outputChannel.appendLine('\nCancel requested', output_channel_1.OutputChannelSeverity.Warning);
+                                    cargs[0]();
+                                }, cargs[1], cargs[2]);
+                            }
+                            return cTarget[cProp];
+                        }
+                    });
+                }
+                try {
+                    const result = await original.apply(target, args);
+                    if ((0, common_1.isLanguageModelStreamResponse)(result)) {
+                        outputChannel.appendLine('Received a response stream');
+                        const stream = result.stream;
+                        const loggedStream = {
+                            async *[Symbol.asyncIterator]() {
+                                for await (const part of stream) {
+                                    outputChannel.append(part.content || '');
+                                    yield part;
+                                }
+                                outputChannel.append('\n');
+                                outputChannel.appendLine('End of stream');
+                            },
+                        };
+                        return {
+                            ...result,
+                            stream: loggedStream,
+                        };
+                    }
+                    else {
+                        outputChannel.appendLine('Received a response');
+                        outputChannel.appendLine(JSON.stringify(result));
+                        return result;
+                    }
+                }
+                catch (err) {
+                    outputChannel.appendLine('An error occurred');
+                    if (err instanceof Error) {
+                        outputChannel.appendLine(err.message, output_channel_1.OutputChannelSeverity.Error);
+                    }
+                    throw err;
+                }
+            };
+        }
+        return original;
+    },
+});
+//# sourceMappingURL=frontend-language-model-registry.js.map

package/lib/browser/frontend-language-model-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-language-model-registry.js","sourceRoot":"","sources":["../../src/browser/frontend-language-model-registry.ts"],"names":[],"mappings":";;;;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,sCAAyD;AACzD,4DAIsC;AACtC,6EAIkD;AAClD,sCAkBmB;AAGZ,IAAM,+BAA+B,GAArC,MAAM,+BAA+B;IAIxC,WAAW,CAAC,QAA2C;QACnD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,EAAU,EAAE,KAAkD;QAC/D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,QAAQ,CAAC,SAAiB,EAAE,MAAc,EAAE,WAAmB;QAC3D,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;IAED,kBAAkB,CAAC,QAA+B;QAC9C,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,oBAAoB,CAAC,EAAU;QAC3B,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC;CACJ,CAAA;AAvBY,0EAA+B;0CAA/B,+BAA+B;IAD3C,IAAA,sBAAU,GAAE;GACA,+BAA+B,CAuB3C;AASM,IAAM,iCAAiC,yCAAvC,MAAM,iCACT,SAAQ,yCAAgC;IADrC;;QAyJK,YAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;QACzC,aAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IAsE/D,CAAC;IA7NG,oBAAoB;IACpB,kBAAkB,CAAC,QAA+B;QAC9C,IAAI,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,oBAAoB;IACpB,oBAAoB,CAAC,EAAU;QAC3B,IAAI,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACpC,CAAC;IAqBQ,iBAAiB,CAAC,MAAiD;QACxE,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnD,OAAO,CAAC,IAAI,CAAC,kCAAkC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC3D,SAAS;YACb,CAAC;YACD,IAAI,sBAAa,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC,cAAc,CAAC,IAAI,CACpB,IAAI,KAAK,CACL,KAAK,EACL,0BAA0B,CACtB,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,UAAU,CACtC,KAAK,CAAC,EAAE,CACX,CACJ,CACJ,CACJ,CAAC;gBACF,UAAU,GAAG,IAAI,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,cAAc,CAAC,IAAI,CACpB,IAAI,KAAK,CACL,IAAI,CAAC,2BAA2B,CAC5B,KAAK,CACR,EACD,0BAA0B,CACtB,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,UAAU,CACtC,KAAK,CAAC,EAAE,CACX,CACJ,CACJ,CACJ,CAAC;gBACF,UAAU,GAAG,IAAI,CAAC;YACtB,CAAC;QACL,CAAC;QACD,IAAI,UAAU,EAAE,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;IAGkB,IAAI;QACnB,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAE9B,MAAM,aAAa,GACf,IAAI,CAAC,0BAA0B,CAAC,gBAAgB,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC3D,MAAM,mBAAmB,GACrB,IAAI,CAAC,gBAAgB,CAAC,4BAA4B,EAAE,CAAC;QAEzD,OAAO,CAAC,UAAU,CAAC,CAAC,mBAAmB,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CACvD,OAAO,CAAC,EAAE;YACN,MAAM,yBAAyB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC7C,IAAI,yBAAyB,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACnD,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,MAAM,CAAC,KAAK,CACb,4DAA4D,EAC5D,yBAAyB,CAAC,MAAM,CACnC,CAAC;YACN,CAAC;YACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,sDAAsD;gBACtD,MAAM,mBAAmB,GAAG,OAAO,CAAC,CAAC,CAEQ,CAAC;gBAC9C,IAAI,mBAAmB,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBAC7C,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;gBACtD,CAAC;qBAAM,CAAC;oBACJ,IAAI,CAAC,MAAM,CAAC,KAAK,CACb,qCAAqC,EACrC,mBAAmB,CAAC,MAAM,CAC7B,CAAC;gBACN,CAAC;YACL,CAAC;YACD,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3B,CAAC,CACJ,CAAC;IACN,CAAC;IAED,2BAA2B,CACvB,WAAkC;QAElC,OAAO;YACH,GAAG,WAAW;YACd,OAAO,EAAE,KAAK,EAAE,OAA6B,EAAE,iBAAqC,EAAE,EAAE;gBACpF,MAAM,SAAS,GAAG,GAAG,mCAAiC,CAAC,cAAc,EAAE,EAAE,CAAC;gBAC1E,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBACtC,iBAAiB,aAAjB,iBAAiB,uBAAjB,iBAAiB,CAAE,uBAAuB,CAAC,GAAG,EAAE;oBAC5C,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBAC5C,CAAC,CAAC,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAChD,WAAW,CAAC,EAAE,EACd,OAAO,EACP,SAAS,EACT,iBAAiB,CACpB,CAAC;gBACF,IAAI,IAAA,oCAA2B,EAAC,QAAQ,CAAC,IAAI,IAAA,sCAA6B,EAAC,QAAQ,CAAC,EAAE,CAAC;oBACnF,OAAO,QAAQ,CAAC;gBACpB,CAAC;gBACD,IAAI,IAAA,8CAAqC,EAAC,QAAQ,CAAC,EAAE,CAAC;oBAClD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;wBACvC,MAAM,cAAc,GAAG;4BACnB,EAAE,EAAE,QAAQ,CAAC,QAAQ;4BACrB,MAAM,EAAE,EAAE;yBACb,CAAC;wBACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;oBACxD,CAAC;oBACD,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAE,CAAC;oBACzD,OAAO;wBACH,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC;qBACxC,CAAC;gBACN,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,KAAK,CACb,uEAAuE,WAAW,CAAC,EAAE,qDAAqD,EAC1I,QAAQ,CACX,CAAC;gBACF,OAAO,QAAQ,CAAC;YACpB,CAAC;SACJ,CAAC;IACN,CAAC;IAKD,KAAK,CAAC,CAAC,WAAW,CACd,KAAkB;QAElB,IAAI,OAAO,GAAG,CAAC,CAAC,CAAC;QACjB,OAAO,IAAI,EAAE,CAAC;YACV,IAAI,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,OAAO,EAAE,CAAC;gBACV,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACpC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACtB,sBAAsB;oBACtB,MAAM;gBACV,CAAC;gBACD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACtB,MAAM,KAAK,CAAC;gBAChB,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;oBACxB,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC5B,CAAC,CAAC,CAAC;YACP,CAAC;QACL,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,8DAA8D;IAC9D,IAAI,CAAC,EAAU,EAAE,KAAkD;QAC/D,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACxB,MAAM,cAAc,GAAG;gBACnB,EAAE;gBACF,MAAM,EAAE,EAAE;aACb,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QAC1C,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;YACtB,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACL,CAAC;IAED,yCAAyC;IACzC,QAAQ,CAAC,EAAU,EAAE,MAAc,EAAE,UAAkB;;QACnD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC7E,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QACvC,MAAM,IAAI,GAAG,MAAA,OAAO,CAAC,KAAK,0CAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;QACvD,IAAI,IAAI,EAAE,CAAC;YACP,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,6BAA6B,MAAM,GAAG,CAAC,CAAC;IAC5D,CAAC;IAEQ,KAAK,CAAC,oBAAoB,CAAC,OAA8B;;QAC9D,MAAM,IAAI,CAAC,WAAW,CAAC;QACvB,MAAM,YAAY,GAAG,MAAA,MAAA,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,0CAAE,yBAAyB,0CAAE,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3J,IAAI,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,UAAU,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACnE,IAAI,KAAK,EAAE,CAAC;gBACR,OAAO,CAAC,KAAK,CAAC,CAAC;YACnB,CAAC;QACL,CAAC;QACD,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,IAAA,wBAAe,EAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IAChF,CAAC;IAEQ,KAAK,CAAC,mBAAmB,CAAC,OAA8B;QAC7D,OAAO,CAAC,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;;AA/NQ,8EAAiC;AA6B3B,gDAAc,GAAW,CAAC,AAAZ,CAAa;AAjBhC;IADT,IAAA,kBAAM,EAAC,8CAAqC,CAAC;;2EACoB;AAGxD;IADT,IAAA,kBAAM,EAAC,sCAA6B,CAAC;;2EACoB;AAGhD;IADT,IAAA,kBAAM,EAAC,+BAA+B,CAAC;sCACtB,+BAA+B;iEAAC;AAG/B;IADlB,IAAA,kBAAM,EAAC,cAAO,CAAC;;iEACmB;AAGzB;IADT,IAAA,kBAAM,EAAC,qCAAoB,CAAC;sCACG,qCAAoB;+EAAC;AAG3C;IADT,IAAA,kBAAM,EAAC,0BAAiB,CAAC;;0EACmB;AA6C1B;IADlB,IAAA,yBAAa,GAAE;;;;6DAsCf;4CA7GQ,iCAAiC;IAD7C,IAAA,sBAAU,GAAE;GACA,iCAAiC,CAgO7C;AAED,MAAM,yBAAyB,GAAG,CAAC,GAAY,EAAY,EAAE;IACzD,2CAA2C;IAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAChD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,cAAc,GAAa,EAAE,CAAC;IAEpC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;QACjB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC5C,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,MAAM,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAClD,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,cAAc,CAAC,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9C,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,YAAY,GAAG,CAAC,QAAgB,EAAU,EAAE;IAC9C,MAAM,OAAO,GAAG,mBAAmB,CAAC;IACpC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACtD,CAAC,CAAC;AAEF,MAAM,0BAA0B,GAAG,CAC/B,mBAAwC,EACb,EAAE,CAAC,CAAC;IAC/B,GAAG,CACC,MAAqB,EACrB,IAAO;QAEP,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,IAAI,KAAK,SAAS,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;YACvD,OAAO,KAAK,WACR,GAAG,IAA0C;gBAE7C,MAAM,aAAa,GAAG,mBAAmB,EAAE,CAAC;gBAC5C,aAAa,CAAC,UAAU,CACpB,kBAAkB,CACrB,CAAC;gBACF,MAAM,gBAAgB,GAAG,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5D,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;gBACjE,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;oBACV,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE;wBACzB,GAAG,CACC,OAA0B,EAC1B,KAAS;4BAET,IAAI,KAAK,KAAK,yBAAyB,EAAE,CAAC;gCACtC,OAAO,CAAC,GAAG,KAA+D,EAAE,EAAE,CAAC,OAAO,CAAC,uBAAuB,CAAC,GAAG,EAAE;oCAChH,aAAa,CAAC,UAAU,CAAC,oBAAoB,EAAE,sCAAqB,CAAC,OAAO,CAAC,CAAC;oCAC9E,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gCACf,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;4BAC3B,CAAC;4BACD,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;wBAC1B,CAAC;qBACJ,CAAC,CAAC;gBACP,CAAC;gBACD,IAAI,CAAC;oBACD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;oBAClD,IAAI,IAAA,sCAA6B,EAAC,MAAM,CAAC,EAAE,CAAC;wBACxC,aAAa,CAAC,UAAU,CAAC,4BAA4B,CAAC,CAAC;wBACvD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;wBAC7B,MAAM,YAAY,GAAG;4BACjB,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;gCACzB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;oCAC9B,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;oCACzC,MAAM,IAAI,CAAC;gCACf,CAAC;gCACD,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gCAC3B,aAAa,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;4BAC9C,CAAC;yBACJ,CAAC;wBACF,OAAO;4BACH,GAAG,MAAM;4BACT,MAAM,EAAE,YAAY;yBACvB,CAAC;oBACN,CAAC;yBAAM,CAAC;wBACJ,aAAa,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC;wBAChD,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;wBACjD,OAAO,MAAM,CAAC;oBAClB,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACX,aAAa,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;oBAC9C,IAAI,GAAG,YAAY,KAAK,EAAE,CAAC;wBACvB,aAAa,CAAC,UAAU,CACpB,GAAG,CAAC,OAAO,EACX,sCAAqB,CAAC,KAAK,CAC9B,CAAC;oBACN,CAAC;oBACD,MAAM,GAAG,CAAC;gBACd,CAAC;YACL,CAAC,CAAC;QACN,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;CACJ,CAAC,CAAC"}

package/lib/browser/frontend-prompt-customization-service.d.ts

@@ -0,0 +1,31 @@
+import { DisposableCollection, URI, Event } from '@theia/core';
+import { OpenerService } from '@theia/core/lib/browser';
+import { PromptCustomizationService, PromptTemplate } from '../common';
+import { FileService } from '@theia/filesystem/lib/browser/file-service';
+import { AICorePreferences } from './ai-core-preferences';
+import { AgentService } from '../common/agent-service';
+import { EnvVariablesServer } from '@theia/core/lib/common/env-variables';
+export declare class FrontendPromptCustomizationServiceImpl implements PromptCustomizationService {
+    protected readonly envVariablesServer: EnvVariablesServer;
+    protected readonly preferences: AICorePreferences;
+    protected readonly fileService: FileService;
+    protected readonly openerService: OpenerService;
+    protected readonly agentService: AgentService;
+    protected readonly trackedTemplateURIs: Set<string>;
+    protected templates: Map<string, string>;
+    protected toDispose: DisposableCollection;
+    private readonly onDidChangePromptEmitter;
+    readonly onDidChangePrompt: Event<string>;
+    protected init(): void;
+    protected update(): Promise<void>;
+    protected getTemplatesDirectoryURI(): Promise<URI>;
+    protected getTemplateURI(templateId: string): Promise<URI>;
+    protected removePromptTemplateSuffix(filename: string): string;
+    isPromptTemplateCustomized(id: string): boolean;
+    getCustomizedPromptTemplate(id: string): string | undefined;
+    editTemplate(id: string, content?: string): Promise<void>;
+    resetTemplate(id: string): Promise<void>;
+    getOriginalTemplate(id: string): PromptTemplate | undefined;
+    getTemplateIDFromURI(uri: URI): string | undefined;
+}
+//# sourceMappingURL=frontend-prompt-customization-service.d.ts.map

package/lib/browser/frontend-prompt-customization-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-prompt-customization-service.d.ts","sourceRoot":"","sources":["../../src/browser/frontend-prompt-customization-service.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,oBAAoB,EAAE,GAAG,EAAE,KAAK,EAAW,MAAM,aAAa,CAAC;AACxE,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAAE,0BAA0B,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAEvE,OAAO,EAAE,WAAW,EAAE,MAAM,4CAA4C,CAAC;AAEzE,OAAO,EAAE,iBAAiB,EAAoC,MAAM,uBAAuB,CAAC;AAC5F,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sCAAsC,CAAC;AAE1E,qBACa,sCAAuC,YAAW,0BAA0B;IAGrF,SAAS,CAAC,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,CAAC;IAG1D,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,iBAAiB,CAAC;IAGlD,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAG5C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IAGhD,SAAS,CAAC,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAC;IAE9C,SAAS,CAAC,QAAQ,CAAC,mBAAmB,cAAqB;IAC3D,SAAS,CAAC,SAAS,sBAA6B;IAEhD,SAAS,CAAC,SAAS,uBAA8B;IAEjD,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAAyB;IAClE,QAAQ,CAAC,iBAAiB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAuC;IAGhF,SAAS,CAAC,IAAI,IAAI,IAAI;cASN,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;cA8DvB,wBAAwB,IAAI,OAAO,CAAC,GAAG,CAAC;cASxC,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IAIhE,SAAS,CAAC,0BAA0B,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM;IAQ9D,0BAA0B,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAI/C,2BAA2B,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAIrD,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBzD,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAO9C,mBAAmB,CAAC,EAAE,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAW3D,oBAAoB,CAAC,GAAG,EAAE,GAAG,GAAG,MAAM,GAAG,SAAS;CAQrD"}

package/lib/browser/frontend-prompt-customization-service.js

@@ -0,0 +1,194 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2024 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
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FrontendPromptCustomizationServiceImpl = void 0;
+const tslib_1 = require("tslib");
+const core_1 = require("@theia/core");
+const browser_1 = require("@theia/core/lib/browser");
+const inversify_1 = require("@theia/core/shared/inversify");
+const buffer_1 = require("@theia/core/lib/common/buffer");
+const file_service_1 = require("@theia/filesystem/lib/browser/file-service");
+const ai_core_preferences_1 = require("./ai-core-preferences");
+const agent_service_1 = require("../common/agent-service");
+const env_variables_1 = require("@theia/core/lib/common/env-variables");
+let FrontendPromptCustomizationServiceImpl = class FrontendPromptCustomizationServiceImpl {
+    constructor() {
+        this.trackedTemplateURIs = new Set();
+        this.templates = new Map();
+        this.toDispose = new core_1.DisposableCollection();
+        this.onDidChangePromptEmitter = new core_1.Emitter();
+        this.onDidChangePrompt = this.onDidChangePromptEmitter.event;
+    }
+    init() {
+        this.preferences.onPreferenceChanged(event => {
+            if (event.preferenceName === ai_core_preferences_1.PREFERENCE_NAME_PROMPT_TEMPLATES) {
+                this.update();
+            }
+        });
+        this.update();
+    }
+    async update() {
+        this.toDispose.dispose();
+        // we need to assign a local variable, so that updates running in parallel don't interfere with each other
+        const _templates = new Map();
+        this.templates = _templates;
+        this.trackedTemplateURIs.clear();
+        const templateURI = await this.getTemplatesDirectoryURI();
+        this.toDispose.push(this.fileService.watch(templateURI, { recursive: true, excludes: [] }));
+        this.toDispose.push(this.fileService.onDidFilesChange(async (event) => {
+            // check deleted templates
+            for (const deletedFile of event.getDeleted()) {
+                if (this.trackedTemplateURIs.has(deletedFile.resource.toString())) {
+                    this.trackedTemplateURIs.delete(deletedFile.resource.toString());
+                    const templateId = this.removePromptTemplateSuffix(deletedFile.resource.path.name);
+                    _templates.delete(templateId);
+                    this.onDidChangePromptEmitter.fire(templateId);
+                }
+            }
+            // check updated templates
+            for (const updatedFile of event.getUpdated()) {
+                if (this.trackedTemplateURIs.has(updatedFile.resource.toString())) {
+                    const filecontent = await this.fileService.read(updatedFile.resource);
+                    const templateId = this.removePromptTemplateSuffix(updatedFile.resource.path.name);
+                    _templates.set(templateId, filecontent.value);
+                    this.onDidChangePromptEmitter.fire(templateId);
+                }
+            }
+            // check new templates
+            for (const addedFile of event.getAdded()) {
+                if (addedFile.resource.parent.toString() === templateURI.toString() && addedFile.resource.path.ext === '.prompttemplate') {
+                    this.trackedTemplateURIs.add(addedFile.resource.toString());
+                    const filecontent = await this.fileService.read(addedFile.resource);
+                    const templateId = this.removePromptTemplateSuffix(addedFile.resource.path.name);
+                    _templates.set(templateId, filecontent.value);
+                    this.onDidChangePromptEmitter.fire(templateId);
+                }
+            }
+        }));
+        const stat = await this.fileService.resolve(templateURI);
+        if (stat.children === undefined) {
+            return;
+        }
+        for (const file of stat.children) {
+            if (!file.isFile) {
+                continue;
+            }
+            const fileURI = file.resource;
+            if (fileURI.path.ext === '.prompttemplate') {
+                this.trackedTemplateURIs.add(fileURI.toString());
+                const filecontent = await this.fileService.read(fileURI);
+                const templateId = this.removePromptTemplateSuffix(file.name);
+                _templates.set(templateId, filecontent.value);
+                this.onDidChangePromptEmitter.fire(templateId);
+            }
+        }
+    }
+    async getTemplatesDirectoryURI() {
+        const templatesFolder = this.preferences[ai_core_preferences_1.PREFERENCE_NAME_PROMPT_TEMPLATES];
+        if (templatesFolder && templatesFolder.trim().length > 0) {
+            return core_1.URI.fromFilePath(templatesFolder);
+        }
+        const theiaConfigDir = await this.envVariablesServer.getConfigDirUri();
+        return new core_1.URI(theiaConfigDir).resolve('prompt-templates');
+    }
+    async getTemplateURI(templateId) {
+        return (await this.getTemplatesDirectoryURI()).resolve(`${templateId}.prompttemplate`);
+    }
+    removePromptTemplateSuffix(filename) {
+        const suffix = '.prompttemplate';
+        if (filename.endsWith(suffix)) {
+            return filename.slice(0, -suffix.length);
+        }
+        return filename;
+    }
+    isPromptTemplateCustomized(id) {
+        return this.templates.has(id);
+    }
+    getCustomizedPromptTemplate(id) {
+        return this.templates.get(id);
+    }
+    async editTemplate(id, content) {
+        const template = this.getOriginalTemplate(id);
+        if (template === undefined) {
+            throw new Error(`Unable to edit template ${id}: template not found.`);
+        }
+        const editorUri = await this.getTemplateURI(id);
+        if (!await this.fileService.exists(editorUri)) {
+            await this.fileService.createFile(editorUri, buffer_1.BinaryBuffer.fromString(content !== null && content !== void 0 ? content : template.template));
+        }
+        else if (content) {
+            // Write content to the file before opening it
+            await this.fileService.writeFile(editorUri, buffer_1.BinaryBuffer.fromString(content));
+        }
+        const openHandler = await this.openerService.getOpener(editorUri);
+        openHandler.open(editorUri);
+    }
+    async resetTemplate(id) {
+        const editorUri = await this.getTemplateURI(id);
+        if (await this.fileService.exists(editorUri)) {
+            await this.fileService.delete(editorUri);
+        }
+    }
+    getOriginalTemplate(id) {
+        for (const agent of this.agentService.getAllAgents()) {
+            for (const template of agent.promptTemplates) {
+                if (template.id === id) {
+                    return template;
+                }
+            }
+        }
+        return undefined;
+    }
+    getTemplateIDFromURI(uri) {
+        const id = this.removePromptTemplateSuffix(uri.path.name);
+        if (this.templates.has(id)) {
+            return id;
+        }
+        return undefined;
+    }
+};
+exports.FrontendPromptCustomizationServiceImpl = FrontendPromptCustomizationServiceImpl;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(env_variables_1.EnvVariablesServer),
+    tslib_1.__metadata("design:type", Object)
+], FrontendPromptCustomizationServiceImpl.prototype, "envVariablesServer", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(ai_core_preferences_1.AICorePreferences),
+    tslib_1.__metadata("design:type", Object)
+], FrontendPromptCustomizationServiceImpl.prototype, "preferences", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(file_service_1.FileService),
+    tslib_1.__metadata("design:type", file_service_1.FileService)
+], FrontendPromptCustomizationServiceImpl.prototype, "fileService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(browser_1.OpenerService),
+    tslib_1.__metadata("design:type", Object)
+], FrontendPromptCustomizationServiceImpl.prototype, "openerService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.inject)(agent_service_1.AgentService),
+    tslib_1.__metadata("design:type", Object)
+], FrontendPromptCustomizationServiceImpl.prototype, "agentService", void 0);
+tslib_1.__decorate([
+    (0, inversify_1.postConstruct)(),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", void 0)
+], FrontendPromptCustomizationServiceImpl.prototype, "init", null);
+exports.FrontendPromptCustomizationServiceImpl = FrontendPromptCustomizationServiceImpl = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], FrontendPromptCustomizationServiceImpl);
+//# sourceMappingURL=frontend-prompt-customization-service.js.map

package/lib/browser/frontend-prompt-customization-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-prompt-customization-service.js","sourceRoot":"","sources":["../../src/browser/frontend-prompt-customization-service.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,sCAAwE;AACxE,qDAAwD;AACxD,4DAAiF;AAEjF,0DAA6D;AAC7D,6EAAyE;AAEzE,+DAA4F;AAC5F,2DAAuD;AACvD,wEAA0E;AAGnE,IAAM,sCAAsC,GAA5C,MAAM,sCAAsC;IAA5C;QAiBgB,wBAAmB,GAAG,IAAI,GAAG,EAAU,CAAC;QACjD,cAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAEtC,cAAS,GAAG,IAAI,2BAAoB,EAAE,CAAC;QAEhC,6BAAwB,GAAG,IAAI,cAAO,EAAU,CAAC;QACzD,sBAAiB,GAAkB,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC;IAiJpF,CAAC;IA9Ia,IAAI;QACV,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,KAAK,CAAC,EAAE;YACzC,IAAI,KAAK,CAAC,cAAc,KAAK,sDAAgC,EAAE,CAAC;gBAC5D,IAAI,CAAC,MAAM,EAAE,CAAC;YAClB,CAAC;QACL,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,CAAC;IAClB,CAAC;IAES,KAAK,CAAC,MAAM;QAClB,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QACzB,0GAA0G;QAC1G,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC;QAC5B,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAEjC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAE1D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC5F,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,KAAK,EAAE,KAAuB,EAAE,EAAE;YACpF,0BAA0B;YAC1B,KAAK,MAAM,WAAW,IAAI,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC;gBAC3C,IAAI,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC;oBAChE,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;oBACjE,MAAM,UAAU,GAAG,IAAI,CAAC,0BAA0B,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACnF,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;oBAC9B,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnD,CAAC;YACL,CAAC;YACD,0BAA0B;YAC1B,KAAK,MAAM,WAAW,IAAI,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC;gBAC3C,IAAI,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC;oBAChE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;oBACtE,MAAM,UAAU,GAAG,IAAI,CAAC,0BAA0B,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACnF,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;oBAC9C,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnD,CAAC;YACL,CAAC;YACD,sBAAsB;YACtB,KAAK,MAAM,SAAS,IAAI,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC;gBACvC,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,KAAK,WAAW,CAAC,QAAQ,EAAE,IAAI,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,KAAK,iBAAiB,EAAE,CAAC;oBACvH,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAC5D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;oBACpE,MAAM,UAAU,GAAG,IAAI,CAAC,0BAA0B,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACjF,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;oBAC9C,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnD,CAAC;YACL,CAAC;QAEL,CAAC,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACzD,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC9B,OAAO;QACX,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACf,SAAS;YACb,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC9B,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,iBAAiB,EAAE,CAAC;gBACzC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACjD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzD,MAAM,UAAU,GAAG,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC9D,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;gBAC9C,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;QACL,CAAC;IACL,CAAC;IAES,KAAK,CAAC,wBAAwB;QACpC,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,CAAC,sDAAgC,CAAC,CAAC;QAC3E,IAAI,eAAe,IAAI,eAAe,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,OAAO,UAAG,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QAC7C,CAAC;QACD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;QACvE,OAAO,IAAI,UAAG,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC/D,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,UAAkB;QAC7C,OAAO,CAAC,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,UAAU,iBAAiB,CAAC,CAAC;IAC3F,CAAC;IAES,0BAA0B,CAAC,QAAgB;QACjD,MAAM,MAAM,GAAG,iBAAiB,CAAC;QACjC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,0BAA0B,CAAC,EAAU;QACjC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,2BAA2B,CAAC,EAAU;QAClC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,EAAU,EAAE,OAAgB;QAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,2BAA2B,EAAE,uBAAuB,CAAC,CAAC;QAC1E,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;QAChD,IAAI,CAAE,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,SAAS,EAAE,qBAAY,CAAC,UAAU,CAAC,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxG,CAAC;aAAM,IAAI,OAAO,EAAE,CAAC;YACjB,8CAA8C;YAC9C,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,SAAS,EAAE,qBAAY,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;QAClF,CAAC;QACD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAClE,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,EAAU;QAC1B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;QAChD,IAAI,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3C,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;IACL,CAAC;IAED,mBAAmB,CAAC,EAAU;QAC1B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,EAAE,CAAC;YACnD,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;gBAC3C,IAAI,QAAQ,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBACrB,OAAO,QAAQ,CAAC;gBACpB,CAAC;YACL,CAAC;QACL,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,oBAAoB,CAAC,GAAQ;QACzB,MAAM,EAAE,GAAG,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACd,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;CAEJ,CAAA;AAxKY,wFAAsC;AAG5B;IADlB,IAAA,kBAAM,EAAC,kCAAkB,CAAC;;kFAC+B;AAGvC;IADlB,IAAA,kBAAM,EAAC,uCAAiB,CAAC;;2EACwB;AAG/B;IADlB,IAAA,kBAAM,EAAC,0BAAW,CAAC;sCACY,0BAAW;2EAAC;AAGzB;IADlB,IAAA,kBAAM,EAAC,uBAAa,CAAC;;6EAC0B;AAG7B;IADlB,IAAA,kBAAM,EAAC,4BAAY,CAAC;;4EACyB;AAWpC;IADT,IAAA,yBAAa,GAAE;;;;kEAQf;iDAjCQ,sCAAsC;IADlD,IAAA,sBAAU,GAAE;GACA,sCAAsC,CAwKlD"}

package/lib/browser/frontend-variable-service.d.ts

@@ -0,0 +1,6 @@
+import { DefaultAIVariableService } from '../common';
+import { FrontendApplicationContribution } from '@theia/core/lib/browser';
+export declare class FrontendVariableService extends DefaultAIVariableService implements FrontendApplicationContribution {
+    onStart(): void;
+}
+//# sourceMappingURL=frontend-variable-service.d.ts.map

package/lib/browser/frontend-variable-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-variable-service.d.ts","sourceRoot":"","sources":["../../src/browser/frontend-variable-service.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,wBAAwB,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,+BAA+B,EAAE,MAAM,yBAAyB,CAAC;AAE1E,qBACa,uBAAwB,SAAQ,wBAAyB,YAAW,+BAA+B;IAC5G,OAAO,IAAI,IAAI;CAGlB"}

package/lib/browser/frontend-variable-service.js

@@ -0,0 +1,31 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2024 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
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FrontendVariableService = void 0;
+const tslib_1 = require("tslib");
+const inversify_1 = require("@theia/core/shared/inversify");
+const common_1 = require("../common");
+let FrontendVariableService = class FrontendVariableService extends common_1.DefaultAIVariableService {
+    onStart() {
+        this.initContributions();
+    }
+};
+exports.FrontendVariableService = FrontendVariableService;
+exports.FrontendVariableService = FrontendVariableService = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], FrontendVariableService);
+//# sourceMappingURL=frontend-variable-service.js.map

package/lib/browser/frontend-variable-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-variable-service.js","sourceRoot":"","sources":["../../src/browser/frontend-variable-service.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,4DAA0D;AAC1D,sCAAqD;AAI9C,IAAM,uBAAuB,GAA7B,MAAM,uBAAwB,SAAQ,iCAAwB;IACjE,OAAO;QACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC7B,CAAC;CACJ,CAAA;AAJY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,sBAAU,GAAE;GACA,uBAAuB,CAInC"}

package/lib/browser/index.d.ts

@@ -0,0 +1,11 @@
+export * from './ai-activation-service';
+export * from './ai-core-frontend-application-contribution';
+export * from './ai-core-frontend-module';
+export * from './ai-core-preferences';
+export * from './ai-settings-service';
+export * from './ai-view-contribution';
+export * from './frontend-language-model-registry';
+export * from './frontend-variable-service';
+export * from './prompttemplate-contribution';
+export * from './theia-variable-contribution';
+//# sourceMappingURL=index.d.ts.map

package/lib/browser/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/browser/index.ts"],"names":[],"mappings":"AAgBA,cAAc,yBAAyB,CAAC;AACxC,cAAc,6CAA6C,CAAC;AAC5D,cAAc,2BAA2B,CAAC;AAC1C,cAAc,uBAAuB,CAAC;AACtC,cAAc,uBAAuB,CAAC;AACtC,cAAc,wBAAwB,CAAC;AACvC,cAAc,oCAAoC,CAAC;AACnD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,+BAA+B,CAAC"}