@theia/ai-code-completion 1.55.1 → 1.57.0-next.112

package/lib/common/code-completion-agent.js

@@ -1,150 +0,0 @@
-"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.CodeCompletionAgentImpl = exports.CodeCompletionAgent = void 0;
-const tslib_1 = require("tslib");
-const common_1 = require("@theia/ai-core/lib/common");
-const core_1 = require("@theia/core");
-const inversify_1 = require("@theia/core/shared/inversify");
-exports.CodeCompletionAgent = Symbol('CodeCompletionAgent');
-let CodeCompletionAgentImpl = class CodeCompletionAgentImpl {
-    constructor() {
-        this.id = 'Code Completion';
-        this.name = 'Code Completion';
-        this.description = 'This agent provides inline code completion in the code editor in the Theia IDE.';
-        this.promptTemplates = [
-            {
-                id: 'code-completion-prompt',
-                template: `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.
-Finish the following code snippet.
-
-{{textUntilCurrentPosition}}[[MARKER]]{{textAfterCurrentPosition}}
-
-Only return the exact replacement for [[MARKER]] to complete the snippet.`,
-            },
-        ];
-        this.languageModelRequirements = [
-            {
-                purpose: 'code-completion',
-                identifier: 'openai/gpt-4o',
-            },
-        ];
-        this.variables = [];
-        this.functions = [];
-        this.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: 'textUntilCurrentPosition', usedInPrompt: true, description: 'The code before the current position of the cursor.' },
-            { name: 'textAfterCurrentPosition', usedInPrompt: true, description: 'The code after the current position of the cursor.' }
-        ];
-    }
-    async provideInlineCompletions(model, position, context, token) {
-        const languageModel = await this.languageModelRegistry.selectLanguageModel({
-            agent: this.id,
-            ...this.languageModelRequirements[0],
-        });
-        if (!languageModel) {
-            this.logger.error('No language model found for code-completion-agent');
-            return undefined;
-        }
-        // Get text until the given position
-        const textUntilCurrentPosition = model.getValueInRange({
-            startLineNumber: 1,
-            startColumn: 1,
-            endLineNumber: position.lineNumber,
-            endColumn: position.column,
-        });
-        // Get text after the given position
-        const textAfterCurrentPosition = model.getValueInRange({
-            startLineNumber: position.lineNumber,
-            startColumn: position.column,
-            endLineNumber: model.getLineCount(),
-            endColumn: model.getLineMaxColumn(model.getLineCount()),
-        });
-        const file = model.uri.toString(false);
-        const language = model.getLanguageId();
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        const prompt = await this.promptService
-            .getPrompt('code-completion-prompt', { textUntilCurrentPosition, textAfterCurrentPosition, file, language })
-            .then(p => p === null || p === void 0 ? void 0 : p.text);
-        if (!prompt) {
-            this.logger.error('No prompt found for code-completion-agent');
-            return undefined;
-        }
-        // since we do not actually hold complete conversions, the request/response pair is considered a session
-        const sessionId = (0, core_1.generateUuid)();
-        const requestId = (0, core_1.generateUuid)();
-        const request = {
-            messages: [{ type: 'text', actor: 'user', query: prompt }],
-        };
-        const requestEntry = {
-            agentId: this.id,
-            sessionId,
-            timestamp: Date.now(),
-            requestId,
-            request: prompt,
-        };
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        this.recordingService.recordRequest(requestEntry);
-        const response = await languageModel.request(request, token);
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        const completionText = await (0, common_1.getTextOfResponse)(response);
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        this.recordingService.recordResponse({
-            agentId: this.id,
-            sessionId,
-            timestamp: Date.now(),
-            requestId,
-            response: completionText,
-        });
-        return {
-            items: [{ insertText: completionText }],
-            enableForwardStability: true,
-        };
-    }
-};
-exports.CodeCompletionAgentImpl = CodeCompletionAgentImpl;
-tslib_1.__decorate([
-    (0, inversify_1.inject)(core_1.ILogger),
-    (0, inversify_1.named)('code-completion-agent'),
-    tslib_1.__metadata("design:type", Object)
-], CodeCompletionAgentImpl.prototype, "logger", void 0);
-tslib_1.__decorate([
-    (0, inversify_1.inject)(common_1.LanguageModelRegistry),
-    tslib_1.__metadata("design:type", Object)
-], CodeCompletionAgentImpl.prototype, "languageModelRegistry", void 0);
-tslib_1.__decorate([
-    (0, inversify_1.inject)(common_1.PromptService),
-    tslib_1.__metadata("design:type", Object)
-], CodeCompletionAgentImpl.prototype, "promptService", void 0);
-tslib_1.__decorate([
-    (0, inversify_1.inject)(common_1.CommunicationRecordingService),
-    tslib_1.__metadata("design:type", Object)
-], CodeCompletionAgentImpl.prototype, "recordingService", void 0);
-exports.CodeCompletionAgentImpl = CodeCompletionAgentImpl = tslib_1.__decorate([
-    (0, inversify_1.injectable)()
-], CodeCompletionAgentImpl);
-//# sourceMappingURL=code-completion-agent.js.map

package/lib/common/code-completion-agent.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"code-completion-agent.js","sourceRoot":"","sources":["../../src/common/code-completion-agent.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,sDAGmC;AACnC,sCAAoD;AACpD,4DAAyE;AAG5D,QAAA,mBAAmB,GAAG,MAAM,CAAC,qBAAqB,CAAC,CAAC;AAO1D,IAAM,uBAAuB,GAA7B,MAAM,uBAAuB;IAA7B;QAqGH,OAAE,GAAG,iBAAiB,CAAC;QACvB,SAAI,GAAG,iBAAiB,CAAC;QACzB,gBAAW,GACP,iFAAiF,CAAC;QACtF,oBAAe,GAAqB;YAChC;gBACI,EAAE,EAAE,wBAAwB;gBAC5B,QAAQ,EAAE;;;;;;0EAMoD;aACjE;SACJ,CAAC;QACF,8BAAyB,GAA+B;YACpD;gBACI,OAAO,EAAE,iBAAiB;gBAC1B,UAAU,EAAE,eAAe;aAC9B;SACJ,CAAC;QACO,cAAS,GAAa,EAAE,CAAC;QACzB,cAAS,GAAa,EAAE,CAAC;QACzB,2BAAsB,GAA6B;YACxD,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,mCAAmC,EAAE;YACtF,EAAE,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,0CAA0C,EAAE;YACjG,EAAE,IAAI,EAAE,0BAA0B,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,qDAAqD,EAAE;YAC5H,EAAE,IAAI,EAAE,0BAA0B,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,oDAAoD,EAAE;SAC9H,CAAC;IAEN,CAAC;IAnIG,KAAK,CAAC,wBAAwB,CAC1B,KAA+B,EAC/B,QAAyB,EACzB,OAAiD,EACjD,KAA+B;QAE/B,MAAM,aAAa,GACf,MAAM,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,EAAE;YACd,GAAG,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC;SACvC,CAAC,CAAC;QACP,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,KAAK,CACb,mDAAmD,CACtD,CAAC;YACF,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,oCAAoC;QACpC,MAAM,wBAAwB,GAAG,KAAK,CAAC,eAAe,CAAC;YACnD,eAAe,EAAE,CAAC;YAClB,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,QAAQ,CAAC,UAAU;YAClC,SAAS,EAAE,QAAQ,CAAC,MAAM;SAC7B,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,wBAAwB,GAAG,KAAK,CAAC,eAAe,CAAC;YACnD,eAAe,EAAE,QAAQ,CAAC,UAAU;YACpC,WAAW,EAAE,QAAQ,CAAC,MAAM;YAC5B,aAAa,EAAE,KAAK,CAAC,YAAY,EAAE;YACnC,SAAS,EAAE,KAAK,CAAC,gBAAgB,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;SAC1D,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvC,MAAM,QAAQ,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;QAEvC,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa;aAClC,SAAS,CAAC,wBAAwB,EAAE,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;aAC3G,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,aAAD,CAAC,uBAAD,CAAC,CAAE,IAAI,CAAC,CAAC;QACxB,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;YAC/D,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,wGAAwG;QACxG,MAAM,SAAS,GAAG,IAAA,mBAAY,GAAE,CAAC;QACjC,MAAM,SAAS,GAAG,IAAA,mBAAY,GAAE,CAAC;QACjC,MAAM,OAAO,GAAyB;YAClC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;SAC7D,CAAC;QACF,MAAM,YAAY,GAA8B;YAC5C,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS;YACT,OAAO,EAAE,MAAM;SAClB,CAAC;QACF,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC7D,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,MAAM,cAAc,GAAG,MAAM,IAAA,0BAAiB,EAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC;YACjC,OAAO,EAAE,IAAI,CAAC,EAAE;YAChB,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS;YACT,QAAQ,EAAE,cAAc;SAC3B,CAAC,CAAC;QAEH,OAAO;YACH,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;YACvC,sBAAsB,EAAE,IAAI;SAC/B,CAAC;IACN,CAAC;CA8CJ,CAAA;AApIY,0DAAuB;AA0FtB;IAFT,IAAA,kBAAM,EAAC,cAAO,CAAC;IACf,IAAA,iBAAK,EAAC,uBAAuB,CAAC;;uDACL;AAGhB;IADT,IAAA,kBAAM,EAAC,8BAAqB,CAAC;;sEACyB;AAG7C;IADT,IAAA,kBAAM,EAAC,sBAAa,CAAC;;8DACiB;AAG7B;IADT,IAAA,kBAAM,EAAC,sCAA6B,CAAC;;iEACoB;kCAnGjD,uBAAuB;IADnC,IAAA,sBAAU,GAAE;GACA,uBAAuB,CAoInC"}

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

@@ -1,164 +0,0 @@
-// *****************************************************************************
-// 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
-// *****************************************************************************
-
-import {
-    Agent, AgentSpecificVariables, CommunicationHistoryEntry, CommunicationRecordingService, getTextOfResponse,
-    LanguageModelRegistry, LanguageModelRequest, LanguageModelRequirement, PromptService, PromptTemplate
-} from '@theia/ai-core/lib/common';
-import { generateUuid, ILogger } from '@theia/core';
-import { inject, injectable, named } from '@theia/core/shared/inversify';
-import * as monaco from '@theia/monaco-editor-core';
-
-export const CodeCompletionAgent = Symbol('CodeCompletionAgent');
-export interface CodeCompletionAgent extends Agent {
-    provideInlineCompletions(model: monaco.editor.ITextModel, position: monaco.Position,
-        context: monaco.languages.InlineCompletionContext, token: monaco.CancellationToken): Promise<monaco.languages.InlineCompletions | undefined>
-}
-
-@injectable()
-export class CodeCompletionAgentImpl implements CodeCompletionAgent {
-    async provideInlineCompletions(
-        model: monaco.editor.ITextModel,
-        position: monaco.Position,
-        context: monaco.languages.InlineCompletionContext,
-        token: monaco.CancellationToken
-    ): Promise<monaco.languages.InlineCompletions | undefined> {
-        const languageModel =
-            await this.languageModelRegistry.selectLanguageModel({
-                agent: this.id,
-                ...this.languageModelRequirements[0],
-            });
-        if (!languageModel) {
-            this.logger.error(
-                'No language model found for code-completion-agent'
-            );
-            return undefined;
-        }
-
-        // Get text until the given position
-        const textUntilCurrentPosition = model.getValueInRange({
-            startLineNumber: 1,
-            startColumn: 1,
-            endLineNumber: position.lineNumber,
-            endColumn: position.column,
-        });
-
-        // Get text after the given position
-        const textAfterCurrentPosition = model.getValueInRange({
-            startLineNumber: position.lineNumber,
-            startColumn: position.column,
-            endLineNumber: model.getLineCount(),
-            endColumn: model.getLineMaxColumn(model.getLineCount()),
-        });
-
-        const file = model.uri.toString(false);
-        const language = model.getLanguageId();
-
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        const prompt = await this.promptService
-            .getPrompt('code-completion-prompt', { textUntilCurrentPosition, textAfterCurrentPosition, file, language })
-            .then(p => p?.text);
-        if (!prompt) {
-            this.logger.error('No prompt found for code-completion-agent');
-            return undefined;
-        }
-
-        // since we do not actually hold complete conversions, the request/response pair is considered a session
-        const sessionId = generateUuid();
-        const requestId = generateUuid();
-        const request: LanguageModelRequest = {
-            messages: [{ type: 'text', actor: 'user', query: prompt }],
-        };
-        const requestEntry: CommunicationHistoryEntry = {
-            agentId: this.id,
-            sessionId,
-            timestamp: Date.now(),
-            requestId,
-            request: prompt,
-        };
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        this.recordingService.recordRequest(requestEntry);
-        const response = await languageModel.request(request, token);
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        const completionText = await getTextOfResponse(response);
-        if (token.isCancellationRequested) {
-            return undefined;
-        }
-        this.recordingService.recordResponse({
-            agentId: this.id,
-            sessionId,
-            timestamp: Date.now(),
-            requestId,
-            response: completionText,
-        });
-
-        return {
-            items: [{ insertText: completionText }],
-            enableForwardStability: true,
-        };
-    }
-
-    @inject(ILogger)
-    @named('code-completion-agent')
-    protected logger: ILogger;
-
-    @inject(LanguageModelRegistry)
-    protected languageModelRegistry: LanguageModelRegistry;
-
-    @inject(PromptService)
-    protected promptService: PromptService;
-
-    @inject(CommunicationRecordingService)
-    protected recordingService: CommunicationRecordingService;
-
-    id = 'Code Completion';
-    name = 'Code Completion';
-    description =
-        'This agent provides inline code completion in the code editor in the Theia IDE.';
-    promptTemplates: PromptTemplate[] = [
-        {
-            id: 'code-completion-prompt',
-            template: `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.
-Finish the following code snippet.
-
-{{textUntilCurrentPosition}}[[MARKER]]{{textAfterCurrentPosition}}
-
-Only return the exact replacement for [[MARKER]] to complete the snippet.`,
-        },
-    ];
-    languageModelRequirements: LanguageModelRequirement[] = [
-        {
-            purpose: 'code-completion',
-            identifier: 'openai/gpt-4o',
-        },
-    ];
-    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: 'textUntilCurrentPosition', usedInPrompt: true, description: 'The code before the current position of the cursor.' },
-        { name: 'textAfterCurrentPosition', usedInPrompt: true, description: 'The code after the current position of the cursor.' }
-    ];
-    readonly tags?: String[] | undefined;
-}