@theia/ai-code-completion 1.63.0-next.0 → 1.63.0-next.52

package/lib/browser/code-completion-variable-context.js

@@ -0,0 +1,26 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2025 Lonti.com Pty Ltd.
+//
+// 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.CodeCompletionVariableContext = void 0;
+var CodeCompletionVariableContext;
+(function (CodeCompletionVariableContext) {
+    function is(context) {
+        return !!context && 'model' in context && 'position' in context && 'context' in context;
+    }
+    CodeCompletionVariableContext.is = is;
+})(CodeCompletionVariableContext || (exports.CodeCompletionVariableContext = CodeCompletionVariableContext = {}));
+//# sourceMappingURL=code-completion-variable-context.js.map

package/lib/browser/code-completion-variable-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variable-context.js","sourceRoot":"","sources":["../../src/browser/code-completion-variable-context.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,wCAAwC;AACxC,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;;;AAWhF,IAAiB,6BAA6B,CAI7C;AAJD,WAAiB,6BAA6B;IAC1C,SAAgB,EAAE,CAAC,OAA0B;QACzC,OAAO,CAAC,CAAC,OAAO,IAAI,OAAO,IAAI,OAAO,IAAI,UAAU,IAAI,OAAO,IAAI,SAAS,IAAI,OAAO,CAAC;IAC5F,CAAC;IAFe,gCAAE,KAEjB,CAAA;AACL,CAAC,EAJgB,6BAA6B,6CAA7B,6BAA6B,QAI7C"}

package/lib/browser/code-completion-variable-contribution.d.ts

@@ -0,0 +1,16 @@
+import { AIVariableContext, AIVariableResolutionRequest, AIVariableResolver, ResolvedAIVariable } from '@theia/ai-core';
+import { FrontendVariableContribution, FrontendVariableService } from '@theia/ai-core/lib/browser';
+import { MaybePromise } from '@theia/core';
+import { PreferenceService } from '@theia/core/lib/browser/preferences/preference-service';
+import { CodeCompletionVariableContext } from './code-completion-variable-context';
+export declare class CodeCompletionVariableContribution implements FrontendVariableContribution, AIVariableResolver {
+    protected preferences: PreferenceService;
+    registerVariables(service: FrontendVariableService): void;
+    canResolve(_request: AIVariableResolutionRequest, context: AIVariableContext): MaybePromise<number>;
+    resolve(request: AIVariableResolutionRequest, context: AIVariableContext): Promise<ResolvedAIVariable | undefined>;
+    protected resolvePrefix(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined;
+    protected resolveSuffix(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined;
+    protected resolveLanguage(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined;
+    protected resolveFile(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined;
+}
+//# sourceMappingURL=code-completion-variable-contribution.d.ts.map

package/lib/browser/code-completion-variable-contribution.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variable-contribution.d.ts","sourceRoot":"","sources":["../../src/browser/code-completion-variable-contribution.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,iBAAiB,EAAE,2BAA2B,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACxH,OAAO,EAAE,4BAA4B,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACnG,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,iBAAiB,EAAE,MAAM,wDAAwD,CAAC;AAG3F,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAGnF,qBACa,kCAAmC,YAAW,4BAA4B,EAAE,kBAAkB;IAEvG,SAAS,CAAC,WAAW,EAAE,iBAAiB,CAAC;IAEzC,iBAAiB,CAAC,OAAO,EAAE,uBAAuB,GAAG,IAAI;IAWzD,UAAU,CAAC,QAAQ,EAAE,2BAA2B,EAAE,OAAO,EAAE,iBAAiB,GAAG,YAAY,CAAC,MAAM,CAAC;IAI7F,OAAO,CAAC,OAAO,EAAE,2BAA2B,EAAE,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC;IAmBxH,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,6BAA6B,GAAG,kBAAkB,GAAG,SAAS;IAkC/F,SAAS,CAAC,aAAa,CAAC,OAAO,EAAE,6BAA6B,GAAG,kBAAkB,GAAG,SAAS;IAgC/F,SAAS,CAAC,eAAe,CAAC,OAAO,EAAE,6BAA6B,GAAG,kBAAkB,GAAG,SAAS;IAOjG,SAAS,CAAC,WAAW,CAAC,OAAO,EAAE,6BAA6B,GAAG,kBAAkB,GAAG,SAAS;CAOhG"}

package/lib/browser/code-completion-variable-contribution.js

@@ -0,0 +1,127 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2025 Lonti.com Pty Ltd.
+//
+// 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.CodeCompletionVariableContribution = void 0;
+const tslib_1 = require("tslib");
+const preference_service_1 = require("@theia/core/lib/browser/preferences/preference-service");
+const inversify_1 = require("@theia/core/shared/inversify");
+const ai_code_completion_preference_1 = require("./ai-code-completion-preference");
+const code_completion_variable_context_1 = require("./code-completion-variable-context");
+const code_completion_variables_1 = require("./code-completion-variables");
+let CodeCompletionVariableContribution = class CodeCompletionVariableContribution {
+    registerVariables(service) {
+        [
+            code_completion_variables_1.FILE,
+            code_completion_variables_1.PREFIX,
+            code_completion_variables_1.SUFFIX,
+            code_completion_variables_1.LANGUAGE
+        ].forEach(variable => {
+            service.registerResolver(variable, this);
+        });
+    }
+    canResolve(_request, context) {
+        return code_completion_variable_context_1.CodeCompletionVariableContext.is(context) ? 1 : 0;
+    }
+    async resolve(request, context) {
+        if (!code_completion_variable_context_1.CodeCompletionVariableContext.is(context)) {
+            return Promise.resolve(undefined);
+        }
+        switch (request.variable.id) {
+            case code_completion_variables_1.FILE.id:
+                return this.resolveFile(context);
+            case code_completion_variables_1.LANGUAGE.id:
+                return this.resolveLanguage(context);
+            case code_completion_variables_1.PREFIX.id:
+                return this.resolvePrefix(context);
+            case code_completion_variables_1.SUFFIX.id:
+                return this.resolveSuffix(context);
+            default:
+                return undefined;
+        }
+    }
+    resolvePrefix(context) {
+        const position = context.position;
+        const model = context.model;
+        const maxContextLines = this.preferences.get(ai_code_completion_preference_1.PREF_AI_INLINE_COMPLETION_MAX_CONTEXT_LINES, -1);
+        let prefixStartLine = 1;
+        if (maxContextLines === 0) {
+            // Only the cursor line
+            prefixStartLine = position.lineNumber;
+        }
+        else if (maxContextLines > 0) {
+            const linesBeforeCursor = position.lineNumber - 1;
+            // Allocate one more line to the prefix in case of an odd maxContextLines
+            const prefixLines = Math.min(Math.ceil(maxContextLines / 2), linesBeforeCursor);
+            prefixStartLine = Math.max(1, position.lineNumber - prefixLines);
+        }
+        const prefix = model.getValueInRange({
+            startLineNumber: prefixStartLine,
+            startColumn: 1,
+            endLineNumber: position.lineNumber,
+            endColumn: position.column,
+        });
+        return {
+            variable: code_completion_variables_1.PREFIX,
+            value: prefix
+        };
+    }
+    resolveSuffix(context) {
+        const position = context.position;
+        const model = context.model;
+        const maxContextLines = this.preferences.get(ai_code_completion_preference_1.PREF_AI_INLINE_COMPLETION_MAX_CONTEXT_LINES, -1);
+        let suffixEndLine = model.getLineCount();
+        if (maxContextLines === 0) {
+            suffixEndLine = position.lineNumber;
+        }
+        else if (maxContextLines > 0) {
+            const linesAfterCursor = model.getLineCount() - position.lineNumber;
+            const suffixLines = Math.min(Math.floor(maxContextLines / 2), linesAfterCursor);
+            suffixEndLine = Math.min(model.getLineCount(), position.lineNumber + suffixLines);
+        }
+        const suffix = model.getValueInRange({
+            startLineNumber: position.lineNumber,
+            startColumn: position.column,
+            endLineNumber: suffixEndLine,
+            endColumn: model.getLineMaxColumn(suffixEndLine),
+        });
+        return {
+            variable: code_completion_variables_1.SUFFIX,
+            value: suffix
+        };
+    }
+    resolveLanguage(context) {
+        return {
+            variable: code_completion_variables_1.LANGUAGE,
+            value: context.model.getLanguageId()
+        };
+    }
+    resolveFile(context) {
+        return {
+            variable: code_completion_variables_1.FILE,
+            value: context.model.uri.toString(false)
+        };
+    }
+};
+exports.CodeCompletionVariableContribution = CodeCompletionVariableContribution;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(preference_service_1.PreferenceService),
+    tslib_1.__metadata("design:type", Object)
+], CodeCompletionVariableContribution.prototype, "preferences", void 0);
+exports.CodeCompletionVariableContribution = CodeCompletionVariableContribution = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], CodeCompletionVariableContribution);
+//# sourceMappingURL=code-completion-variable-contribution.js.map

package/lib/browser/code-completion-variable-contribution.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variable-contribution.js","sourceRoot":"","sources":["../../src/browser/code-completion-variable-contribution.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,wCAAwC;AACxC,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;;;;AAKhF,+FAA2F;AAC3F,4DAAkE;AAClE,mFAA8F;AAC9F,yFAAmF;AACnF,2EAA6E;AAGtE,IAAM,kCAAkC,GAAxC,MAAM,kCAAkC;IAI3C,iBAAiB,CAAC,OAAgC;QAC9C;YACI,gCAAI;YACJ,kCAAM;YACN,kCAAM;YACN,oCAAQ;SACX,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YACjB,OAAO,CAAC,gBAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACP,CAAC;IAED,UAAU,CAAC,QAAqC,EAAE,OAA0B;QACxE,OAAO,gEAA6B,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAoC,EAAE,OAA0B;QAC1E,IAAI,CAAC,gEAA6B,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;YAC7C,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC;QAED,QAAQ,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAC1B,KAAK,gCAAI,CAAC,EAAE;gBACR,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACrC,KAAK,oCAAQ,CAAC,EAAE;gBACZ,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACzC,KAAK,kCAAM,CAAC,EAAE;gBACV,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YACvC,KAAK,kCAAM,CAAC,EAAE;gBACV,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YACvC;gBACI,OAAO,SAAS,CAAC;QACzB,CAAC;IACL,CAAC;IAES,aAAa,CAAC,OAAsC;QAC1D,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAS,2EAA2C,EAAE,CAAC,CAAC,CAAC,CAAC;QACtG,IAAI,eAAe,GAAG,CAAC,CAAC;QAExB,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;YACxB,uBAAuB;YACvB,eAAe,GAAG,QAAQ,CAAC,UAAU,CAAC;QAC1C,CAAC;aAAM,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,iBAAiB,GAAG,QAAQ,CAAC,UAAU,GAAG,CAAC,CAAC;YAElD,yEAAyE;YACzE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CACxB,IAAI,CAAC,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC,EAC9B,iBAAiB,CACpB,CAAC;YAEF,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,UAAU,GAAG,WAAW,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC;YACjC,eAAe,EAAE,eAAe;YAChC,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,QAAQ,CAAC,UAAU;YAClC,SAAS,EAAE,QAAQ,CAAC,MAAM;SAC7B,CAAC,CAAC;QAEH,OAAO;YACH,QAAQ,EAAE,kCAAM;YAChB,KAAK,EAAE,MAAM;SAChB,CAAC;IACN,CAAC;IAES,aAAa,CAAC,OAAsC;QAC1D,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5B,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAS,2EAA2C,EAAE,CAAC,CAAC,CAAC,CAAC;QACtG,IAAI,aAAa,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC;QAEzC,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;YACxB,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC;QACxC,CAAC;aAAM,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,gBAAgB,GAAG,KAAK,CAAC,YAAY,EAAE,GAAG,QAAQ,CAAC,UAAU,CAAC;YAEpE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CACxB,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,CAAC,CAAC,EAC/B,gBAAgB,CACnB,CAAC;YAEF,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,QAAQ,CAAC,UAAU,GAAG,WAAW,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC;YACjC,eAAe,EAAE,QAAQ,CAAC,UAAU;YACpC,WAAW,EAAE,QAAQ,CAAC,MAAM;YAC5B,aAAa,EAAE,aAAa;YAC5B,SAAS,EAAE,KAAK,CAAC,gBAAgB,CAAC,aAAa,CAAC;SACnD,CAAC,CAAC;QAEH,OAAO;YACH,QAAQ,EAAE,kCAAM;YAChB,KAAK,EAAE,MAAM;SAChB,CAAC;IACN,CAAC;IAES,eAAe,CAAC,OAAsC;QAC5D,OAAO;YACH,QAAQ,EAAE,oCAAQ;YAClB,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,aAAa,EAAE;SACvC,CAAC;IACN,CAAC;IAES,WAAW,CAAC,OAAsC;QACxD,OAAO;YACH,QAAQ,EAAE,gCAAI;YACd,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC;SAC3C,CAAC;IACN,CAAC;CAEJ,CAAA;AAtHY,gFAAkC;AAEjC;IADT,IAAA,kBAAM,EAAC,sCAAiB,CAAC;;uEACe;6CAFhC,kCAAkC;IAD9C,IAAA,sBAAU,GAAE;GACA,kCAAkC,CAsH9C"}

package/lib/browser/code-completion-variable-contribution.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=code-completion-variable-contribution.spec.d.ts.map

package/lib/browser/code-completion-variable-contribution.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variable-contribution.spec.d.ts","sourceRoot":"","sources":["../../src/browser/code-completion-variable-contribution.spec.ts"],"names":[],"mappings":""}

package/lib/browser/code-completion-variable-contribution.spec.js

@@ -0,0 +1,143 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2025 Lonti.com Pty Ltd.
+//
+// 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 });
+const frontend_application_config_provider_1 = require("@theia/core/lib/browser/frontend-application-config-provider");
+const jsdom_1 = require("@theia/core/lib/browser/test/jsdom");
+let disableJSDOM = (0, jsdom_1.enableJSDOM)();
+frontend_application_config_provider_1.FrontendApplicationConfigProvider.set({});
+const browser_1 = require("@theia/core/lib/browser");
+const inversify_1 = require("@theia/core/shared/inversify");
+const editor_api_1 = require("@theia/monaco-editor-core/esm/vs/editor/editor.api");
+const chai_1 = require("chai");
+const sinon = require("sinon");
+const code_completion_variable_contribution_1 = require("./code-completion-variable-contribution");
+const code_completion_variables_1 = require("./code-completion-variables");
+disableJSDOM();
+describe('CodeCompletionVariableContribution', () => {
+    let contribution;
+    let model;
+    before(() => {
+        disableJSDOM = (0, jsdom_1.enableJSDOM)();
+        const container = new inversify_1.Container();
+        container.bind(browser_1.PreferenceService).toConstantValue({
+            get: () => 1000,
+        });
+        container.bind(code_completion_variable_contribution_1.CodeCompletionVariableContribution).toSelf().inSingletonScope();
+        contribution = container.get(code_completion_variable_contribution_1.CodeCompletionVariableContribution);
+    });
+    beforeEach(() => {
+        model = editor_api_1.editor.createModel('//line 1\nconsole.\n//line 2', 'javascript', editor_api_1.Uri.file('/home/user/workspace/test.js'));
+        sinon.stub(model, 'getLanguageId').returns('javascript');
+    });
+    afterEach(() => {
+        model.dispose();
+    });
+    after(() => {
+        // Disable JSDOM after all tests
+        disableJSDOM();
+        model.dispose();
+    });
+    describe('canResolve', () => {
+        it('should be able to resolve the file from the CodeCompletionVariableContext', () => {
+            const context = {
+                model,
+                position: model.getPositionAt(8),
+                context: {
+                    triggerKind: editor_api_1.languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+            (0, chai_1.expect)(contribution.canResolve({ variable: code_completion_variables_1.FILE }, context)).to.equal(1);
+        });
+        it('should not be able to resolve the file from unknown context', () => {
+            (0, chai_1.expect)(contribution.canResolve({ variable: code_completion_variables_1.FILE }, {})).to.equal(0);
+        });
+    });
+    describe('resolve', () => {
+        it('should resolve the file variable', async () => {
+            const context = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: editor_api_1.languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+            const resolved = await contribution.resolve({ variable: code_completion_variables_1.FILE }, context);
+            (0, chai_1.expect)(resolved).to.deep.equal({
+                variable: code_completion_variables_1.FILE,
+                value: 'file:///home/user/workspace/test.js'
+            });
+        });
+        it('should resolve the language variable', async () => {
+            const context = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: editor_api_1.languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+            const resolved = await contribution.resolve({ variable: code_completion_variables_1.LANGUAGE }, context);
+            (0, chai_1.expect)(resolved).to.deep.equal({
+                variable: code_completion_variables_1.LANGUAGE,
+                value: 'javascript'
+            });
+        });
+        it('should resolve the prefix variable', async () => {
+            const context = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: editor_api_1.languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+            const resolved = await contribution.resolve({ variable: code_completion_variables_1.PREFIX }, context);
+            (0, chai_1.expect)(resolved).to.deep.equal({
+                variable: code_completion_variables_1.PREFIX,
+                value: '//line 1\nconsole.'
+            });
+        });
+        it('should resolve the suffix variable', async () => {
+            const context = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: editor_api_1.languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+            const resolved = await contribution.resolve({ variable: code_completion_variables_1.SUFFIX }, context);
+            (0, chai_1.expect)(resolved).to.deep.equal({
+                variable: code_completion_variables_1.SUFFIX,
+                value: '\n//line 2'
+            });
+        });
+    });
+});
+//# sourceMappingURL=code-completion-variable-contribution.spec.js.map

package/lib/browser/code-completion-variable-contribution.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variable-contribution.spec.js","sourceRoot":"","sources":["../../src/browser/code-completion-variable-contribution.spec.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,wCAAwC;AACxC,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,uHAAiH;AACjH,8DAAiE;AACjE,IAAI,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;AACjC,wEAAiC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;AAE1C,qDAA4D;AAC5D,4DAAyD;AACzD,mFAA4F;AAC5F,+BAA8B;AAC9B,+BAA+B;AAE/B,mGAA6F;AAC7F,2EAA6E;AAE7E,YAAY,EAAE,CAAC;AAEf,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAChD,IAAI,YAAgD,CAAC;IACrD,IAAI,KAAwB,CAAC;IAE7B,MAAM,CAAC,GAAG,EAAE;QACR,YAAY,GAAG,IAAA,mBAAW,GAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,qBAAS,EAAE,CAAC;QAClC,SAAS,CAAC,IAAI,CAAC,2BAAiB,CAAC,CAAC,eAAe,CAAC;YAC9C,GAAG,EAAE,GAAG,EAAE,CAAC,IAAI;SAClB,CAAC,CAAC;QACH,SAAS,CAAC,IAAI,CAAC,0EAAkC,CAAC,CAAC,MAAM,EAAE,CAAC,gBAAgB,EAAE,CAAC;QAC/E,YAAY,GAAG,SAAS,CAAC,GAAG,CAAC,0EAAkC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACZ,KAAK,GAAG,mBAAM,CAAC,WAAW,CAAC,8BAA8B,EAAE,YAAY,EAAE,gBAAG,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC,CAAC;QACnH,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACX,KAAK,CAAC,OAAO,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE;QACP,gCAAgC;QAChC,YAAY,EAAE,CAAC;QACf,KAAK,CAAC,OAAO,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACjF,MAAM,OAAO,GAAkC;gBAC3C,KAAK;gBACL,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;gBAChC,OAAO,EAAE;oBACL,WAAW,EAAE,sBAAS,CAAC,2BAA2B,CAAC,SAAS;oBAC5D,sBAAsB,EAAE,SAAS;oBACjC,kBAAkB,EAAE,KAAK;oBACzB,wBAAwB,EAAE,KAAK;iBAClC;aACJ,CAAC;YAEF,IAAA,aAAM,EAAC,YAAY,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,gCAAI,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACnE,IAAA,aAAM,EAAC,YAAY,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,gCAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,OAAO,GAAkC;gBAC3C,KAAK;gBACL,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE;oBACL,WAAW,EAAE,sBAAS,CAAC,2BAA2B,CAAC,SAAS;oBAC5D,sBAAsB,EAAE,SAAS;oBACjC,kBAAkB,EAAE,KAAK;oBACzB,wBAAwB,EAAE,KAAK;iBAClC;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,gCAAI,EAAE,EAAE,OAAO,CAAC,CAAC;YACzE,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC3B,QAAQ,EAAE,gCAAI;gBACd,KAAK,EAAE,qCAAqC;aAC/C,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,OAAO,GAAkC;gBAC3C,KAAK;gBACL,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE;oBACL,WAAW,EAAE,sBAAS,CAAC,2BAA2B,CAAC,SAAS;oBAC5D,sBAAsB,EAAE,SAAS;oBACjC,kBAAkB,EAAE,KAAK;oBACzB,wBAAwB,EAAE,KAAK;iBAClC;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,oCAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;YAC7E,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC3B,QAAQ,EAAE,oCAAQ;gBAClB,KAAK,EAAE,YAAY;aACtB,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,OAAO,GAAkC;gBAC3C,KAAK;gBACL,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE;oBACL,WAAW,EAAE,sBAAS,CAAC,2BAA2B,CAAC,SAAS;oBAC5D,sBAAsB,EAAE,SAAS;oBACjC,kBAAkB,EAAE,KAAK;oBACzB,wBAAwB,EAAE,KAAK;iBAClC;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,kCAAM,EAAE,EAAE,OAAO,CAAC,CAAC;YAC3E,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC3B,QAAQ,EAAE,kCAAM;gBAChB,KAAK,EAAE,oBAAoB;aAC9B,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,OAAO,GAAkC;gBAC3C,KAAK;gBACL,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE;oBACL,WAAW,EAAE,sBAAS,CAAC,2BAA2B,CAAC,SAAS;oBAC5D,sBAAsB,EAAE,SAAS;oBACjC,kBAAkB,EAAE,KAAK;oBACzB,wBAAwB,EAAE,KAAK;iBAClC;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,kCAAM,EAAE,EAAE,OAAO,CAAC,CAAC;YAC3E,IAAA,aAAM,EAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC;gBAC3B,QAAQ,EAAE,kCAAM;gBAChB,KAAK,EAAE,YAAY;aACtB,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/lib/browser/code-completion-variables.d.ts

@@ -0,0 +1,6 @@
+import { AIVariable } from '@theia/ai-core/lib/common/variable-service';
+export declare const FILE: AIVariable;
+export declare const PREFIX: AIVariable;
+export declare const SUFFIX: AIVariable;
+export declare const LANGUAGE: AIVariable;
+//# sourceMappingURL=code-completion-variables.d.ts.map

package/lib/browser/code-completion-variables.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variables.d.ts","sourceRoot":"","sources":["../../src/browser/code-completion-variables.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,UAAU,EAAE,MAAM,4CAA4C,CAAC;AAExE,eAAO,MAAM,IAAI,EAAE,UAIlB,CAAC;AAEF,eAAO,MAAM,MAAM,EAAE,UAIpB,CAAC;AAEF,eAAO,MAAM,MAAM,EAAE,UAIpB,CAAC;AAEF,eAAO,MAAM,QAAQ,EAAE,UAItB,CAAC"}

package/lib/browser/code-completion-variables.js

@@ -0,0 +1,39 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2025 Lonti.com Pty Ltd.
+//
+// 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.LANGUAGE = exports.SUFFIX = exports.PREFIX = exports.FILE = void 0;
+exports.FILE = {
+    id: 'codeCompletionFile',
+    name: 'codeCompletionFile',
+    description: 'The uri of the file being edited.',
+};
+exports.PREFIX = {
+    id: 'codeCompletionPrefix',
+    name: 'codeCompletionPrefix',
+    description: 'The code before the current position of the cursor.',
+};
+exports.SUFFIX = {
+    id: 'codeCompletionSuffix',
+    name: 'codeCompletionSuffix',
+    description: 'The code after the current position of the cursor.',
+};
+exports.LANGUAGE = {
+    id: 'codeCompletionLanguage',
+    name: 'codeCompletionLanguage',
+    description: 'The languageId of the file being edited.',
+};
+//# sourceMappingURL=code-completion-variables.js.map

package/lib/browser/code-completion-variables.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"code-completion-variables.js","sourceRoot":"","sources":["../../src/browser/code-completion-variables.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,wCAAwC;AACxC,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;;;AAInE,QAAA,IAAI,GAAe;IAC5B,EAAE,EAAE,oBAAoB;IACxB,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EAAE,mCAAmC;CACnD,CAAC;AAEW,QAAA,MAAM,GAAe;IAC9B,EAAE,EAAE,sBAAsB;IAC1B,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EAAE,qDAAqD;CACrE,CAAC;AAEW,QAAA,MAAM,GAAe;IAC9B,EAAE,EAAE,sBAAsB;IAC1B,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EAAE,oDAAoD;CACpE,CAAC;AAEW,QAAA,QAAQ,GAAe;IAChC,EAAE,EAAE,wBAAwB;IAC5B,IAAI,EAAE,wBAAwB;IAC9B,WAAW,EAAE,0CAA0C;CAC1D,CAAC"}

package/package.json

@@ -1,14 +1,14 @@
 {
   "name": "@theia/ai-code-completion",
-  "version": "1.63.0-next.0+c02e6a4f5",
+  "version": "1.63.0-next.52+176018e53",
   "description": "Theia - AI Core",
   "dependencies": {
-    "@theia/ai-core": "1.63.0-next.0+c02e6a4f5",
-    "@theia/core": "1.63.0-next.0+c02e6a4f5",
-    "@theia/filesystem": "1.63.0-next.0+c02e6a4f5",
+    "@theia/ai-core": "1.63.0-next.52+176018e53",
+    "@theia/core": "1.63.0-next.52+176018e53",
+    "@theia/filesystem": "1.63.0-next.52+176018e53",
     "@theia/monaco-editor-core": "1.96.302",
-    "@theia/output": "1.63.0-next.0+c02e6a4f5",
-    "@theia/workspace": "1.63.0-next.0+c02e6a4f5",
+    "@theia/output": "1.63.0-next.52+176018e53",
+    "@theia/workspace": "1.63.0-next.52+176018e53",
     "minimatch": "^5.1.0",
     "tslib": "^2.6.2"
   },
@@ -51,5 +51,5 @@
   "nyc": {
     "extends": "../../configs/nyc.json"
   },
-  "gitHead": "c02e6a4f58728b4067cbd267d598550192abb249"
+  "gitHead": "176018e53e62cafec9b25b23ac9f0b4826b6d0f7"
 }

package/src/browser/ai-code-completion-frontend-module.ts

@@ -14,14 +14,15 @@
 // SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
 // *****************************************************************************
 
-import { ContainerModule } from '@theia/core/shared/inversify';
-import { CodeCompletionAgent, CodeCompletionAgentImpl } from './code-completion-agent';
-import { AIFrontendApplicationContribution } from './ai-code-frontend-application-contribution';
+import { Agent, AIVariableContribution } from '@theia/ai-core';
 import { FrontendApplicationContribution, KeybindingContribution, PreferenceContribution } from '@theia/core/lib/browser';
-import { Agent } from '@theia/ai-core';
+import { ContainerModule } from '@theia/core/shared/inversify';
 import { AICodeCompletionPreferencesSchema } from './ai-code-completion-preference';
+import { AIFrontendApplicationContribution } from './ai-code-frontend-application-contribution';
 import { AICodeInlineCompletionsProvider } from './ai-code-inline-completion-provider';
+import { CodeCompletionAgent, CodeCompletionAgentImpl } from './code-completion-agent';
 import { CodeCompletionPostProcessor, DefaultCodeCompletionPostProcessor } from './code-completion-postprocessor';
+import { CodeCompletionVariableContribution } from './code-completion-variable-contribution';
 
 export default new ContainerModule(bind => {
     bind(CodeCompletionAgentImpl).toSelf().inSingletonScope();
@@ -33,4 +34,5 @@ export default new ContainerModule(bind => {
     bind(KeybindingContribution).toService(AIFrontendApplicationContribution);
     bind(PreferenceContribution).toConstantValue({ schema: AICodeCompletionPreferencesSchema });
     bind(CodeCompletionPostProcessor).to(DefaultCodeCompletionPostProcessor).inSingletonScope();
+    bind(AIVariableContribution).to(CodeCompletionVariableContribution).inSingletonScope();
 });

package/src/browser/ai-code-completion-preference.ts

@@ -23,6 +23,7 @@ export const PREF_AI_INLINE_COMPLETION_DEBOUNCE_DELAY = 'ai-features.codeComplet
 export const PREF_AI_INLINE_COMPLETION_EXCLUDED_EXTENSIONS = 'ai-features.codeCompletion.excludedFileExtensions';
 export const PREF_AI_INLINE_COMPLETION_MAX_CONTEXT_LINES = 'ai-features.codeCompletion.maxContextLines';
 export const PREF_AI_INLINE_COMPLETION_STRIP_BACKTICKS = 'ai-features.codeCompletion.stripBackticks';
+export const PREF_AI_INLINE_COMPLETION_CACHE_CAPACITY = 'ai-features.codeCompletion.cacheCapacity';
 
 export const AICodeCompletionPreferencesSchema: PreferenceSchema = {
     type: 'object',
@@ -69,6 +70,15 @@ export const AICodeCompletionPreferencesSchema: PreferenceSchema = {
                 'Remove surrounding backticks from the code returned by some LLMs. If a backtick is detected, all content after the closing\
              backtick is stripped as well. This setting helps ensure plain code is returned when language models use markdown-like formatting.'),
             default: true
+        },
+        [PREF_AI_INLINE_COMPLETION_CACHE_CAPACITY]: {
+            title: nls.localize('theia/ai/completion/cacheCapacity/title', 'Code Completion Cache Capacity'),
+            type: 'number',
+            description: nls.localize('theia/ai/completion/cacheCapacity/description',
+                'Maximum number of code completions to store in the cache. A higher number can improve performance but will consume more memory.\
+                Minimum value is 10, recommended range is between 50-200.'),
+            default: 100,
+            minimum: 10
         }
     }
 };

package/src/browser/ai-code-frontend-application-contribution.ts

@@ -24,10 +24,12 @@ import { InlineCompletionTriggerKind } from '@theia/monaco-editor-core/esm/vs/ed
 import {
     PREF_AI_INLINE_COMPLETION_AUTOMATIC_ENABLE,
     PREF_AI_INLINE_COMPLETION_DEBOUNCE_DELAY,
-    PREF_AI_INLINE_COMPLETION_EXCLUDED_EXTENSIONS
+    PREF_AI_INLINE_COMPLETION_EXCLUDED_EXTENSIONS,
+    PREF_AI_INLINE_COMPLETION_CACHE_CAPACITY
 } from './ai-code-completion-preference';
 import { AICodeInlineCompletionsProvider } from './ai-code-inline-completion-provider';
 import { InlineCompletionDebouncer } from './code-completion-debouncer';
+import { CodeCompletionCache } from './code-completion-cache';
 
 @injectable()
 export class AIFrontendApplicationContribution implements FrontendApplicationContribution, KeybindingContribution {
@@ -40,6 +42,7 @@ export class AIFrontendApplicationContribution implements FrontendApplicationCon
     @inject(AIActivationService)
     protected readonly activationService: AIActivationService;
 
+    private completionCache = new CodeCompletionCache();
     private debouncer = new InlineCompletionDebouncer();
     private debounceDelay: number;
 
@@ -58,6 +61,9 @@ export class AIFrontendApplicationContribution implements FrontendApplicationCon
 
         this.debounceDelay = this.preferenceService.get<number>(PREF_AI_INLINE_COMPLETION_DEBOUNCE_DELAY, 300);
 
+        const cacheCapacity = this.preferenceService.get<number>(PREF_AI_INLINE_COMPLETION_CACHE_CAPACITY, 100);
+        this.completionCache.setMaxSize(cacheCapacity);
+
         this.preferenceService.onPreferenceChanged(event => {
             if (event.preferenceName === PREF_AI_INLINE_COMPLETION_AUTOMATIC_ENABLE
                 || event.preferenceName === PREF_AI_INLINE_COMPLETION_EXCLUDED_EXTENSIONS) {
@@ -67,6 +73,9 @@ export class AIFrontendApplicationContribution implements FrontendApplicationCon
             if (event.preferenceName === PREF_AI_INLINE_COMPLETION_DEBOUNCE_DELAY) {
                 this.debounceDelay = event.newValue;
             }
+            if (event.preferenceName === PREF_AI_INLINE_COMPLETION_CACHE_CAPACITY) {
+                this.completionCache.setMaxSize(event.newValue);
+            }
         });
 
         this.activationService.onDidChangeActiveStatus(change => {
@@ -102,14 +111,27 @@ export class AIFrontendApplicationContribution implements FrontendApplicationCon
                         return { items: [] };
                     }
 
-                    const completionHandler = () => {
+                    const completionHandler = async () => {
                         try {
-                            return this.inlineCodeCompletionProvider.provideInlineCompletions(
+                            const cacheKey = this.completionCache.generateKey(fileName, model, position);
+                            const cachedCompletion = this.completionCache.get(cacheKey);
+
+                            if (cachedCompletion) {
+                                return cachedCompletion;
+                            }
+
+                            const completion = await this.inlineCodeCompletionProvider.provideInlineCompletions(
                                 model,
                                 position,
                                 context,
                                 token
                             );
+
+                            if (completion && completion.items.length > 0) {
+                                this.completionCache.put(cacheKey, completion);
+                            }
+
+                            return completion;
                         } catch (error) {
                             console.error('Error providing inline completions:', error);
                             return { items: [] };