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

package/src/browser/code-completion-agent.ts

@@ -22,12 +22,11 @@ import {
     UserRequest
 } from '@theia/ai-core/lib/common';
 import { generateUuid, ILogger, nls, ProgressService } from '@theia/core';
-import { PreferenceService } from '@theia/core/lib/browser';
 import { inject, injectable, named } from '@theia/core/shared/inversify';
 import * as monaco from '@theia/monaco-editor-core';
-import { PREF_AI_INLINE_COMPLETION_MAX_CONTEXT_LINES } from './ai-code-completion-preference';
 import { codeCompletionPrompts } from './code-completion-prompt-template';
 import { CodeCompletionPostProcessor } from './code-completion-postprocessor';
+import { CodeCompletionVariableContext } from './code-completion-variable-context';
 
 export const CodeCompletionAgent = Symbol('CodeCompletionAgent');
 export interface CodeCompletionAgent extends Agent {
@@ -62,56 +61,17 @@ export class CodeCompletionAgentImpl implements CodeCompletionAgent {
                 return undefined;
             }
 
-            const maxContextLines = this.preferences.get<number>(PREF_AI_INLINE_COMPLETION_MAX_CONTEXT_LINES, -1);
-
-            let prefixStartLine = 1;
-            let suffixEndLine = model.getLineCount();
-            // if maxContextLines is -1, use the full file as context without any line limit
-
-            if (maxContextLines === 0) {
-                // Only the cursor line
-                prefixStartLine = position.lineNumber;
-                suffixEndLine = position.lineNumber;
-            } else if (maxContextLines > 0) {
-                const linesBeforeCursor = position.lineNumber - 1;
-                const linesAfterCursor = model.getLineCount() - position.lineNumber;
-
-                // Allocate one more line to the prefix in case of an odd maxContextLines
-                const prefixLines = Math.min(
-                    Math.ceil(maxContextLines / 2),
-                    linesBeforeCursor
-                );
-                const suffixLines = Math.min(
-                    Math.floor(maxContextLines / 2),
-                    linesAfterCursor
-                );
-
-                prefixStartLine = Math.max(1, position.lineNumber - prefixLines);
-                suffixEndLine = Math.min(model.getLineCount(), position.lineNumber + suffixLines);
-            }
-
-            const prefix = model.getValueInRange({
-                startLineNumber: prefixStartLine,
-                startColumn: 1,
-                endLineNumber: position.lineNumber,
-                endColumn: position.column,
-            });
-
-            const suffix = model.getValueInRange({
-                startLineNumber: position.lineNumber,
-                startColumn: position.column,
-                endLineNumber: suffixEndLine,
-                endColumn: model.getLineMaxColumn(suffixEndLine),
-            });
-
-            const file = model.uri.toString(false);
-            const language = model.getLanguageId();
+            const variableContext: CodeCompletionVariableContext = {
+                model,
+                position,
+                context
+            };
 
             if (token.isCancellationRequested) {
                 return undefined;
             }
             const prompt = await this.promptService
-                .getResolvedPromptFragment('code-completion-prompt', { prefix, suffix, file, language })
+                .getResolvedPromptFragment('code-completion-prompt', undefined, variableContext)
                 .then(p => p?.text);
             if (!prompt) {
                 this.logger.error('No prompt found for code-completion-agent');
@@ -174,9 +134,6 @@ export class CodeCompletionAgentImpl implements CodeCompletionAgent {
     @inject(ProgressService)
     protected progressService: ProgressService;
 
-    @inject(PreferenceService)
-    protected preferences: PreferenceService;
-
     @inject(CodeCompletionPostProcessor)
     protected postProcessor: CodeCompletionPostProcessor;
 
@@ -193,11 +150,5 @@ export class CodeCompletionAgentImpl implements CodeCompletionAgent {
     ];
     readonly variables: string[] = [];
     readonly functions: string[] = [];
-    readonly agentSpecificVariables: AgentSpecificVariables[] = [
-        { name: 'file', usedInPrompt: true, description: 'The uri of the file being edited.' },
-        { name: 'language', usedInPrompt: true, description: 'The languageId of the file being edited.' },
-        { name: 'prefix', usedInPrompt: true, description: 'The code before the current position of the cursor.' },
-        { name: 'suffix', usedInPrompt: true, description: 'The code after the current position of the cursor.' }
-    ];
-    readonly tags?: string[] | undefined;
+    readonly agentSpecificVariables: AgentSpecificVariables[] = [];
 }

package/src/browser/code-completion-cache.ts

@@ -0,0 +1,131 @@
+// *****************************************************************************
+// 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 { injectable } from '@theia/core/shared/inversify';
+import * as monaco from '@theia/monaco-editor-core';
+
+@injectable()
+export class CodeCompletionCache {
+    private cache: Map<string, CacheEntry>;
+    private maxSize = 100;
+
+    constructor() {
+        this.cache = new Map<string, CacheEntry>();
+    }
+
+    /**
+     * Generate a unique cache key based on input parameters
+     * @param filePath Path of the current file
+     * @param lineNumber Current line number
+     * @param lineText Context prefix for completion
+     * @returns Unique cache key
+     */
+    generateKey(filePath: string, model: monaco.editor.ITextModel, position: monaco.Position): string {
+        const lineNumber = position.lineNumber;
+        const beforeCursorLineRange = new monaco.Range(
+            position.lineNumber, 1,
+            position.lineNumber, position.column
+        );
+        const prefix = model.getValueInRange(beforeCursorLineRange);
+        const afterCursorLineRange = new monaco.Range(
+            position.lineNumber, position.column,
+            position.lineNumber, model.getLineMaxColumn(position.lineNumber)
+        );
+        const suffix = model.getValueInRange(afterCursorLineRange);
+        const key = JSON.stringify({
+            filePath,
+            lineNumber,
+            prefix,
+            suffix
+        });
+        return key;
+    }
+
+    /**
+     * Get a cached completion if available
+     * @param key Cache key
+     * @returns Cached completion or undefined
+     */
+    get(key: string): monaco.languages.InlineCompletions | undefined {
+        const entry = this.cache.get(key);
+        if (entry) {
+            // Update the entry's last accessed time
+            entry.lastAccessed = Date.now();
+            return entry.value;
+        }
+        return undefined;
+    }
+
+    /**
+     * Store a completion in the cache
+     * @param key Cache key
+     * @param value Completion value to cache
+     */
+    put(key: string, value: monaco.languages.InlineCompletions | undefined): void {
+        // If cache is full, remove the least recently used entry
+        if (this.cache.size >= this.maxSize) {
+            this.removeLeastRecentlyUsed();
+        }
+
+        this.cache.set(key, {
+            value,
+            lastAccessed: Date.now()
+        });
+    }
+
+    /**
+     * Clear the entire cache
+     */
+    clear(): void {
+        this.cache.clear();
+    }
+
+    /**
+     * Remove the least recently used entry from the cache
+     */
+    private removeLeastRecentlyUsed(): void {
+        let oldestKey: string | undefined;
+        let oldestTime = Infinity;
+
+        for (const [key, entry] of this.cache.entries()) {
+            if (entry.lastAccessed < oldestTime) {
+                oldestKey = key;
+                oldestTime = entry.lastAccessed;
+            }
+        }
+
+        if (oldestKey) {
+            this.cache.delete(oldestKey);
+        }
+    }
+
+    /**
+     * Set the maximum cache size
+     * @param size New maximum cache size
+     */
+    setMaxSize(size: number): void {
+        this.maxSize = size;
+        // Trim cache if it exceeds new size
+        while (this.cache.size > this.maxSize) {
+            this.removeLeastRecentlyUsed();
+        }
+    }
+}
+
+interface CacheEntry {
+    value: monaco.languages.InlineCompletions | undefined;
+    lastAccessed: number;
+}

package/src/browser/code-completion-prompt-template.ts

@@ -10,6 +10,7 @@
 // *****************************************************************************
 
 import { PromptVariantSet } from '@theia/ai-core/lib/common';
+import { FILE, LANGUAGE, PREFIX, SUFFIX } from './code-completion-variables';
 
 export const codeCompletionPrompts: PromptVariantSet[] = [{
     id: 'code-completion-prompt',
@@ -18,13 +19,47 @@ export const codeCompletionPrompts: PromptVariantSet[] = [{
         template: `{{!-- This prompt is licensed under the MIT License (https://opensource.org/license/mit).
 Made improvements or adaptations to this prompt template? We’d love for you to share it with the community! Contribute back here:
 https://github.com/eclipse-theia/theia/discussions/new?category=prompt-template-contribution --}}
-You are a code completion agent. The current file you have to complete is named {{file}}.
-The language of the file is {{language}}. Return your result as plain text without markdown formatting.
+You are a code completion agent. The current file you have to complete is named {{${FILE.id}}}.
+The language of the file is {{${LANGUAGE.id}}}. Return your result as plain text without markdown formatting.
 Finish the following code snippet.
 
-{{prefix}}[[MARKER]]{{suffix}}
+{{${PREFIX.id}}}[[MARKER]]{{${SUFFIX.id}}}
 
 Only return the exact replacement for [[MARKER]] to complete the snippet.`
+    },
+    {
+        id: 'code-completion-prompt-next',
+        template: `{{!-- This prompt is licensed under the MIT License (https://opensource.org/license/mit).
+Made improvements or adaptations to this prompt template? We'd love for you to share it with the community! Contribute back here:
+https://github.com/eclipse-theia/theia/discussions/new?category=prompt-template-contribution --}}
+# System Role
+You are an expert AI code completion assistant focused on generating precise, contextually appropriate code snippets.
+
+## Code Context
+\`\`\`
+{{${PREFIX.id}}}[[MARKER]]{{${SUFFIX.id}}}
+\`\`\`
+
+## Metadata
+- File: {{${FILE.id}}}
+- Programming Language: {{${LANGUAGE.id}}}
+- Project Context: {{prompt:project-info}}
+
+# Completion Guidelines
+1. Analyze the surrounding code context carefully.
+2. Generate ONLY the code that should replace [[MARKER]].
+3. Ensure the completion:
+   - Maintains the exact syntax of the surrounding code
+   - Follows best practices for the specific programming language
+   - Completes the code snippet logically and efficiently
+4. Do NOT include any explanatory text, comments, or additional instructions.
+5. Return ONLY the raw code replacement.
+
+# Constraints
+- Return strictly the code for [[MARKER]]
+- Match indentation and style of surrounding code
+- Prioritize readability and maintainability
+- Consider language-specific idioms and patterns`
     }],
     defaultVariant: {
         id: 'code-completion-prompt-default',
@@ -33,12 +68,12 @@ Made improvements or adaptations to this prompt template? We’d love for you to
 https://github.com/eclipse-theia/theia/discussions/new?category=prompt-template-contribution --}}
 ## Code snippet
 \`\`\`
-{{ prefix }}[[MARKER]]{{ suffix }}
+{{${PREFIX.id}}}[[MARKER]]{{${SUFFIX.id}}}
 \`\`\`
 
 ## Meta Data
-- File: {{file}}
-- Language: {{language}}
+- File: {{${FILE.id}}}
+- Language: {{${LANGUAGE.id}}}
 
 Replace [[MARKER]] with the exact code to complete the code snippet. Return only the replacement of [[MARKER]] as plain text.`,
     },

package/src/browser/code-completion-variable-context.ts

@@ -0,0 +1,30 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+import { AIVariableContext } from '@theia/ai-core';
+import * as monaco from '@theia/monaco-editor-core';
+
+export interface CodeCompletionVariableContext {
+    model: monaco.editor.ITextModel,
+    position: monaco.Position,
+    context: monaco.languages.InlineCompletionContext
+}
+
+export namespace CodeCompletionVariableContext {
+    export function is(context: AIVariableContext): context is CodeCompletionVariableContext {
+        return !!context && 'model' in context && 'position' in context && 'context' in context;
+    }
+}

package/src/browser/code-completion-variable-contribution.spec.ts

@@ -0,0 +1,160 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+import { FrontendApplicationConfigProvider } from '@theia/core/lib/browser/frontend-application-config-provider';
+import { enableJSDOM } from '@theia/core/lib/browser/test/jsdom';
+let disableJSDOM = enableJSDOM();
+FrontendApplicationConfigProvider.set({});
+
+import { PreferenceService } from '@theia/core/lib/browser';
+import { Container } from '@theia/core/shared/inversify';
+import { editor, languages, Uri } from '@theia/monaco-editor-core/esm/vs/editor/editor.api';
+import { expect } from 'chai';
+import * as sinon from 'sinon';
+import { CodeCompletionVariableContext } from './code-completion-variable-context';
+import { CodeCompletionVariableContribution } from './code-completion-variable-contribution';
+import { FILE, LANGUAGE, PREFIX, SUFFIX } from './code-completion-variables';
+
+disableJSDOM();
+
+describe('CodeCompletionVariableContribution', () => {
+    let contribution: CodeCompletionVariableContribution;
+    let model: editor.ITextModel;
+
+    before(() => {
+        disableJSDOM = enableJSDOM();
+        const container = new Container();
+        container.bind(PreferenceService).toConstantValue({
+            get: () => 1000,
+        });
+        container.bind(CodeCompletionVariableContribution).toSelf().inSingletonScope();
+        contribution = container.get(CodeCompletionVariableContribution);
+    });
+
+    beforeEach(() => {
+        model = editor.createModel('//line 1\nconsole.\n//line 2', 'javascript', 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: CodeCompletionVariableContext = {
+                model,
+                position: model.getPositionAt(8),
+                context: {
+                    triggerKind: languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+
+            expect(contribution.canResolve({ variable: FILE }, context)).to.equal(1);
+        });
+
+        it('should not be able to resolve the file from unknown context', () => {
+            expect(contribution.canResolve({ variable: FILE }, {})).to.equal(0);
+        });
+    });
+
+    describe('resolve', () => {
+        it('should resolve the file variable', async () => {
+            const context: CodeCompletionVariableContext = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+
+            const resolved = await contribution.resolve({ variable: FILE }, context);
+            expect(resolved).to.deep.equal({
+                variable: FILE,
+                value: 'file:///home/user/workspace/test.js'
+            });
+        });
+
+        it('should resolve the language variable', async () => {
+            const context: CodeCompletionVariableContext = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+
+            const resolved = await contribution.resolve({ variable: LANGUAGE }, context);
+            expect(resolved).to.deep.equal({
+                variable: LANGUAGE,
+                value: 'javascript'
+            });
+        });
+
+        it('should resolve the prefix variable', async () => {
+            const context: CodeCompletionVariableContext = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+
+            const resolved = await contribution.resolve({ variable: PREFIX }, context);
+            expect(resolved).to.deep.equal({
+                variable: PREFIX,
+                value: '//line 1\nconsole.'
+            });
+        });
+
+        it('should resolve the suffix variable', async () => {
+            const context: CodeCompletionVariableContext = {
+                model,
+                position: model.getPositionAt(17),
+                context: {
+                    triggerKind: languages.InlineCompletionTriggerKind.Automatic,
+                    selectedSuggestionInfo: undefined,
+                    includeInlineEdits: false,
+                    includeInlineCompletions: false
+                }
+            };
+
+            const resolved = await contribution.resolve({ variable: SUFFIX }, context);
+            expect(resolved).to.deep.equal({
+                variable: SUFFIX,
+                value: '\n//line 2'
+            });
+        });
+    });
+});

package/src/browser/code-completion-variable-contribution.ts

@@ -0,0 +1,145 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+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 { inject, injectable } from '@theia/core/shared/inversify';
+import { PREF_AI_INLINE_COMPLETION_MAX_CONTEXT_LINES } from './ai-code-completion-preference';
+import { CodeCompletionVariableContext } from './code-completion-variable-context';
+import { FILE, LANGUAGE, PREFIX, SUFFIX } from './code-completion-variables';
+
+@injectable()
+export class CodeCompletionVariableContribution implements FrontendVariableContribution, AIVariableResolver {
+    @inject(PreferenceService)
+    protected preferences: PreferenceService;
+
+    registerVariables(service: FrontendVariableService): void {
+        [
+            FILE,
+            PREFIX,
+            SUFFIX,
+            LANGUAGE
+        ].forEach(variable => {
+            service.registerResolver(variable, this);
+        });
+    }
+
+    canResolve(_request: AIVariableResolutionRequest, context: AIVariableContext): MaybePromise<number> {
+        return CodeCompletionVariableContext.is(context) ? 1 : 0;
+    }
+
+    async resolve(request: AIVariableResolutionRequest, context: AIVariableContext): Promise<ResolvedAIVariable | undefined> {
+        if (!CodeCompletionVariableContext.is(context)) {
+            return Promise.resolve(undefined);
+        }
+
+        switch (request.variable.id) {
+            case FILE.id:
+                return this.resolveFile(context);
+            case LANGUAGE.id:
+                return this.resolveLanguage(context);
+            case PREFIX.id:
+                return this.resolvePrefix(context);
+            case SUFFIX.id:
+                return this.resolveSuffix(context);
+            default:
+                return undefined;
+        }
+    }
+
+    protected resolvePrefix(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined {
+        const position = context.position;
+        const model = context.model;
+        const maxContextLines = this.preferences.get<number>(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: PREFIX,
+            value: prefix
+        };
+    }
+
+    protected resolveSuffix(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined {
+        const position = context.position;
+        const model = context.model;
+        const maxContextLines = this.preferences.get<number>(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: SUFFIX,
+            value: suffix
+        };
+    }
+
+    protected resolveLanguage(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined {
+        return {
+            variable: LANGUAGE,
+            value: context.model.getLanguageId()
+        };
+    }
+
+    protected resolveFile(context: CodeCompletionVariableContext): ResolvedAIVariable | undefined {
+        return {
+            variable: FILE,
+            value: context.model.uri.toString(false)
+        };
+    }
+
+}

package/src/browser/code-completion-variables.ts

@@ -0,0 +1,41 @@
+// *****************************************************************************
+// 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
+// *****************************************************************************
+
+import { AIVariable } from '@theia/ai-core/lib/common/variable-service';
+
+export const FILE: AIVariable = {
+    id: 'codeCompletionFile',
+    name: 'codeCompletionFile',
+    description: 'The uri of the file being edited.',
+};
+
+export const PREFIX: AIVariable = {
+    id: 'codeCompletionPrefix',
+    name: 'codeCompletionPrefix',
+    description: 'The code before the current position of the cursor.',
+};
+
+export const SUFFIX: AIVariable = {
+    id: 'codeCompletionSuffix',
+    name: 'codeCompletionSuffix',
+    description: 'The code after the current position of the cursor.',
+};
+
+export const LANGUAGE: AIVariable = {
+    id: 'codeCompletionLanguage',
+    name: 'codeCompletionLanguage',
+    description: 'The languageId of the file being edited.',
+};