@aws/lsp-codewhisperer 0.0.83 → 0.0.85

package/out/language-server/inline-completion/handler/editCompletionHandler.d.ts

@@ -0,0 +1,54 @@
+import { CancellationToken, InitializeParams, InlineCompletionListWithReferences, InlineCompletionWithReferencesParams, Range, TextDocument } from '@aws/language-server-runtimes/protocol';
+import { RecentEditTracker } from '../tracker/codeEditTracker';
+import { CredentialsProvider, Logging, Telemetry, Workspace } from '@aws/language-server-runtimes/server-interface';
+import { GenerateSuggestionsResponse } from '../../../shared/codeWhispererService';
+import { CodeWhispererSession, SessionManager } from '../session/sessionManager';
+import { CursorTracker } from '../tracker/cursorTracker';
+import { CodewhispererLanguage } from '../../../shared/languageDetection';
+import { TelemetryService } from '../../../shared/telemetry/telemetryService';
+import { AmazonQBaseServiceManager } from '../../../shared/amazonQServiceManager/BaseAmazonQServiceManager';
+import { RejectedEditTracker } from '../tracker/rejectedEditTracker';
+import { DocumentChangedListener } from '../documentChangedListener';
+export declare class EditCompletionHandler {
+    readonly logging: Logging;
+    readonly clientMetadata: InitializeParams;
+    readonly workspace: Workspace;
+    readonly amazonQServiceManager: AmazonQBaseServiceManager;
+    readonly sessionManager: SessionManager;
+    readonly cursorTracker: CursorTracker;
+    readonly recentEditsTracker: RecentEditTracker;
+    readonly rejectedEditTracker: RejectedEditTracker;
+    readonly documentChangedListener: DocumentChangedListener;
+    readonly telemetry: Telemetry;
+    readonly telemetryService: TelemetryService;
+    readonly credentialsProvider: CredentialsProvider;
+    private readonly editsEnabled;
+    private debounceTimeout;
+    private isWaiting;
+    private hasDocumentChangedSinceInvocation;
+    private readonly streakTracker;
+    private isInProgress;
+    constructor(logging: Logging, clientMetadata: InitializeParams, workspace: Workspace, amazonQServiceManager: AmazonQBaseServiceManager, sessionManager: SessionManager, cursorTracker: CursorTracker, recentEditsTracker: RecentEditTracker, rejectedEditTracker: RejectedEditTracker, documentChangedListener: DocumentChangedListener, telemetry: Telemetry, telemetryService: TelemetryService, credentialsProvider: CredentialsProvider);
+    get codeWhispererService(): import("../../../shared/codeWhispererService").CodeWhispererServiceBase;
+    /**
+     * This is a workaround to refresh the debounce timer when user is typing quickly.
+     * Adding debounce at function call doesnt work because server won't process second request until first request is processed.
+     * Also as a followup, ideally it should be a message/event publish/subscribe pattern instead of manual invocation like this
+     */
+    documentChanged(): void;
+    onEditCompletion(params: InlineCompletionWithReferencesParams, token: CancellationToken): Promise<InlineCompletionListWithReferences>;
+    _invoke(params: InlineCompletionWithReferencesParams, startPreprocessTimestamp: number, token: CancellationToken, textDocument: TextDocument, inferredLanguageId: CodewhispererLanguage, currentSession: CodeWhispererSession | undefined): Promise<InlineCompletionListWithReferences>;
+    processSuggestionResponse(suggestionResponse: GenerateSuggestionsResponse, session: CodeWhispererSession, isNewSession: boolean, selectionRange?: Range, textDocument?: TextDocument): Promise<{
+        sessionId: string;
+        items: never[];
+    } | {
+        items: {
+            insertText: string;
+            isInlineEdit: boolean;
+            itemId: string;
+        }[];
+        sessionId: string;
+        partialResultToken: string | undefined;
+    }>;
+    handleSuggestionsErrors(error: Error, session: CodeWhispererSession): InlineCompletionListWithReferences;
+}

package/out/language-server/inline-completion/handler/editCompletionHandler.js

@@ -0,0 +1,321 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EditCompletionHandler = void 0;
+const protocol_1 = require("@aws/language-server-runtimes/protocol");
+const codeWhispererService_1 = require("../../../shared/codeWhispererService");
+const sessionManager_1 = require("../session/sessionManager");
+const languageDetection_1 = require("../../../shared/languageDetection");
+const workspaceFolderManager_1 = require("../../workspaceContext/workspaceFolderManager");
+const triggerUtils_1 = require("../utils/triggerUtils");
+const telemetry_1 = require("../telemetry/telemetry");
+const lsp_core_1 = require("@aws/lsp-core");
+const utils_1 = require("../../../shared/utils");
+const errors_1 = require("../../../shared/amazonQServiceManager/errors");
+const constants_1 = require("../contants/constants");
+const streakTracker_1 = require("../tracker/streakTracker");
+const diffUtils_1 = require("../utils/diffUtils");
+class EditCompletionHandler {
+    logging;
+    clientMetadata;
+    workspace;
+    amazonQServiceManager;
+    sessionManager;
+    cursorTracker;
+    recentEditsTracker;
+    rejectedEditTracker;
+    documentChangedListener;
+    telemetry;
+    telemetryService;
+    credentialsProvider;
+    editsEnabled;
+    debounceTimeout;
+    isWaiting = false;
+    hasDocumentChangedSinceInvocation = false;
+    streakTracker;
+    isInProgress = false;
+    constructor(logging, clientMetadata, workspace, amazonQServiceManager, sessionManager, cursorTracker, recentEditsTracker, rejectedEditTracker, documentChangedListener, telemetry, telemetryService, credentialsProvider) {
+        this.logging = logging;
+        this.clientMetadata = clientMetadata;
+        this.workspace = workspace;
+        this.amazonQServiceManager = amazonQServiceManager;
+        this.sessionManager = sessionManager;
+        this.cursorTracker = cursorTracker;
+        this.recentEditsTracker = recentEditsTracker;
+        this.rejectedEditTracker = rejectedEditTracker;
+        this.documentChangedListener = documentChangedListener;
+        this.telemetry = telemetry;
+        this.telemetryService = telemetryService;
+        this.credentialsProvider = credentialsProvider;
+        this.editsEnabled =
+            this.clientMetadata.initializationOptions?.aws?.awsClientCapabilities?.textDocument
+                ?.inlineCompletionWithReferences?.inlineEditSupport ?? false;
+        this.streakTracker = streakTracker_1.StreakTracker.getInstance();
+    }
+    get codeWhispererService() {
+        return this.amazonQServiceManager.getCodewhispererService();
+    }
+    /**
+     * This is a workaround to refresh the debounce timer when user is typing quickly.
+     * Adding debounce at function call doesnt work because server won't process second request until first request is processed.
+     * Also as a followup, ideally it should be a message/event publish/subscribe pattern instead of manual invocation like this
+     */
+    documentChanged() {
+        if (this.debounceTimeout) {
+            if (this.isWaiting) {
+                this.hasDocumentChangedSinceInvocation = true;
+            }
+            else {
+                this.logging.info(`refresh and debounce edits suggestion for another ${constants_1.EDIT_DEBOUNCE_INTERVAL_MS}`);
+                this.debounceTimeout.refresh();
+            }
+        }
+    }
+    async onEditCompletion(params, token) {
+        if (this.isInProgress) {
+            this.logging.info(`editCompletionHandler is WIP, skip the request`);
+            return constants_1.EMPTY_RESULT;
+        }
+        // On every new completion request close current inflight session.
+        const currentSession = this.sessionManager.getCurrentSession();
+        if (currentSession && currentSession.state == 'REQUESTING' && !params.partialResultToken) {
+            // this REQUESTING state only happens when the session is initialized, which is rare
+            currentSession.discardInflightSessionOnNewInvocation = true;
+        }
+        if (this.cursorTracker) {
+            this.cursorTracker.trackPosition(params.textDocument.uri, params.position);
+        }
+        const textDocument = await this.workspace.getTextDocument(params.textDocument.uri);
+        if (!textDocument) {
+            this.logging.warn(`textDocument [${params.textDocument.uri}] not found`);
+            return constants_1.EMPTY_RESULT;
+        }
+        if (!(this.codeWhispererService instanceof codeWhispererService_1.CodeWhispererServiceToken)) {
+            return constants_1.EMPTY_RESULT;
+        }
+        // request for new session
+        const inferredLanguageId = (0, languageDetection_1.getSupportedLanguageId)(textDocument);
+        if (!inferredLanguageId) {
+            this.logging.log(`textDocument [${params.textDocument.uri}] with languageId [${textDocument.languageId}] not supported`);
+            return constants_1.EMPTY_RESULT;
+        }
+        // Not ideally to rely on a state, should improve it and simply make it a debounced API
+        this.isInProgress = true;
+        const startPreprocessTimestamp = Date.now();
+        if (params.partialResultToken && currentSession) {
+            // Close ACTIVE session. We shouldn't record Discard trigger decision for trigger with nextToken.
+            if (currentSession && currentSession.state === 'ACTIVE') {
+                this.sessionManager.discardSession(currentSession);
+            }
+            const newSession = this.sessionManager.createSession({
+                document: textDocument,
+                startPosition: params.position,
+                startPreprocessTimestamp: startPreprocessTimestamp,
+                triggerType: 'AutoTrigger',
+                language: currentSession.language,
+                requestContext: currentSession.requestContext,
+                autoTriggerType: undefined,
+                triggerCharacter: '',
+                classifierResult: undefined,
+                classifierThreshold: undefined,
+                credentialStartUrl: currentSession.credentialStartUrl,
+                supplementalMetadata: currentSession.supplementalMetadata,
+                customizationArn: currentSession.customizationArn,
+            });
+            // subsequent paginated requests for current session
+            try {
+                const suggestionResponse = await this.codeWhispererService.generateSuggestions({
+                    ...newSession.requestContext,
+                    nextToken: `${params.partialResultToken}`,
+                });
+                return await this.processSuggestionResponse(suggestionResponse, newSession, true, params.context.selectedCompletionInfo?.range);
+            }
+            catch (error) {
+                return this.handleSuggestionsErrors(error, currentSession);
+            }
+            finally {
+                this.isInProgress = false;
+            }
+        }
+        return new Promise(async (resolve) => {
+            this.debounceTimeout = setTimeout(async () => {
+                try {
+                    this.isWaiting = true;
+                    const result = await this._invoke(params, startPreprocessTimestamp, token, textDocument, inferredLanguageId, currentSession).finally(() => {
+                        this.isWaiting = false;
+                    });
+                    if (this.hasDocumentChangedSinceInvocation) {
+                        this.logging.info('EditCompletionHandler - Document changed during execution, resolving empty result');
+                        resolve({
+                            sessionId: sessionManager_1.SessionManager.getInstance('EDITS').getActiveSession()?.id ?? '',
+                            items: [],
+                        });
+                    }
+                    else {
+                        this.logging.info('EditCompletionHandler - No document changes, resolving result');
+                        resolve(result);
+                    }
+                }
+                finally {
+                    this.debounceTimeout = undefined;
+                    this.hasDocumentChangedSinceInvocation = false;
+                }
+            }, constants_1.EDIT_DEBOUNCE_INTERVAL_MS);
+        }).finally(() => {
+            this.isInProgress = false;
+        });
+    }
+    async _invoke(params, startPreprocessTimestamp, token, textDocument, inferredLanguageId, currentSession) {
+        // Build request context
+        const isAutomaticLspTriggerKind = params.context.triggerKind == protocol_1.InlineCompletionTriggerKind.Automatic;
+        const maxResults = isAutomaticLspTriggerKind ? 1 : 5;
+        const fileContext = (0, codeWhispererService_1.getFileContext)({
+            textDocument,
+            inferredLanguageId,
+            position: params.position,
+            workspaceFolder: this.workspace.getWorkspaceFolder(textDocument.uri),
+        });
+        const workspaceState = workspaceFolderManager_1.WorkspaceFolderManager.getInstance()?.getWorkspaceState();
+        const workspaceId = workspaceState?.webSocketClient?.isConnected() ? workspaceState.workspaceId : undefined;
+        const qEditsTrigger = (0, triggerUtils_1.shouldTriggerEdits)(this.codeWhispererService, fileContext, params, this.cursorTracker, this.recentEditsTracker, this.sessionManager, true);
+        if (!qEditsTrigger) {
+            return constants_1.EMPTY_RESULT;
+        }
+        const generateCompletionReq = {
+            fileContext: fileContext,
+            maxResults: maxResults,
+            predictionTypes: ['EDITS'],
+            workspaceId: workspaceId,
+        };
+        if (qEditsTrigger) {
+            generateCompletionReq.editorState = {
+                document: {
+                    relativeFilePath: textDocument.uri,
+                    programmingLanguage: {
+                        languageName: generateCompletionReq.fileContext?.programmingLanguage?.languageName,
+                    },
+                    text: textDocument.getText(),
+                },
+                cursorState: {
+                    position: {
+                        line: params.position.line,
+                        character: params.position.character,
+                    },
+                },
+            };
+        }
+        const supplementalContext = await this.codeWhispererService.constructSupplementalContext(textDocument, params.position, this.workspace, this.recentEditsTracker, this.logging, token, params.openTabFilepaths, {
+            includeRecentEdits: true,
+        });
+        if (supplementalContext) {
+            generateCompletionReq.supplementalContexts = supplementalContext.items;
+        }
+        // Close ACTIVE session and record Discard trigger decision immediately
+        if (currentSession && currentSession.state === 'ACTIVE') {
+            // Emit user trigger decision at session close time for active session
+            this.sessionManager.discardSession(currentSession);
+            const streakLength = this.editsEnabled ? this.streakTracker.getAndUpdateStreakLength(false) : 0;
+            await (0, telemetry_1.emitUserTriggerDecisionTelemetry)(this.telemetry, this.telemetryService, currentSession, this.documentChangedListener.timeSinceLastUserModification, 0, 0, [], [], streakLength);
+        }
+        const newSession = this.sessionManager.createSession({
+            document: textDocument,
+            startPreprocessTimestamp: startPreprocessTimestamp,
+            startPosition: params.position,
+            triggerType: isAutomaticLspTriggerKind ? 'AutoTrigger' : 'OnDemand',
+            language: fileContext.programmingLanguage.languageName,
+            requestContext: generateCompletionReq,
+            autoTriggerType: undefined,
+            triggerCharacter: '',
+            classifierResult: undefined,
+            classifierThreshold: undefined,
+            credentialStartUrl: this.credentialsProvider.getConnectionMetadata?.()?.sso?.startUrl ?? undefined,
+            supplementalMetadata: supplementalContext?.supContextData,
+            customizationArn: lsp_core_1.textUtils.undefinedIfEmpty(this.codeWhispererService.customizationArn),
+        });
+        try {
+            const suggestionResponse = await this.codeWhispererService.generateSuggestions(generateCompletionReq);
+            return await this.processSuggestionResponse(suggestionResponse, newSession, true, params.context.selectedCompletionInfo?.range);
+        }
+        catch (error) {
+            return this.handleSuggestionsErrors(error, newSession);
+        }
+    }
+    async processSuggestionResponse(suggestionResponse, session, isNewSession, selectionRange, textDocument) {
+        // TODO: we haven't decided how to do these telemetry for Edits suggestions
+        // codePercentageTracker.countInvocation(session.language)
+        // userWrittenCodeTracker?.recordUsageCount(session.language)
+        if (isNewSession) {
+            // Populate the session with information from codewhisperer response
+            session.suggestions = suggestionResponse.suggestions;
+            session.responseContext = suggestionResponse.responseContext;
+            session.codewhispererSessionId = suggestionResponse.responseContext.codewhispererSessionId;
+            session.setTimeToFirstRecommendation();
+            session.predictionType = codeWhispererService_1.SuggestionType.EDIT;
+        }
+        else {
+            session.suggestions = [...session.suggestions, ...suggestionResponse.suggestions];
+        }
+        // Emit service invocation telemetry for every request sent to backend
+        (0, telemetry_1.emitServiceInvocationTelemetry)(this.telemetry, session, suggestionResponse.responseContext.requestId);
+        // Discard previous inflight API response due to new trigger
+        if (session.discardInflightSessionOnNewInvocation) {
+            session.discardInflightSessionOnNewInvocation = false;
+            this.sessionManager.discardSession(session);
+            const streakLength = this.editsEnabled ? this.streakTracker.getAndUpdateStreakLength(false) : 0;
+            await (0, telemetry_1.emitUserTriggerDecisionTelemetry)(this.telemetry, this.telemetryService, session, this.documentChangedListener.timeSinceLastUserModification, 0, 0, [], [], streakLength);
+        }
+        // API response was recieved, we can activate session now
+        this.sessionManager.activateSession(session);
+        // Process suggestions to apply Empty or Filter filters
+        if (suggestionResponse.suggestions.length === 0) {
+            this.sessionManager.closeSession(session);
+            await (0, telemetry_1.emitEmptyUserTriggerDecisionTelemetry)(this.telemetryService, session, this.documentChangedListener.timeSinceLastUserModification, this.editsEnabled ? this.streakTracker.getAndUpdateStreakLength(false) : 0);
+            return constants_1.EMPTY_RESULT;
+        }
+        return {
+            items: suggestionResponse.suggestions
+                .map(suggestion => {
+                // Check if this suggestion is similar to a previously rejected edit
+                const isSimilarToRejected = this.rejectedEditTracker.isSimilarToRejected(suggestion.content ?? '', textDocument?.uri || '');
+                const processedSuggestion = (0, diffUtils_1.processEditSuggestion)(suggestion.content ?? '', session.startPosition, session.document, session.requestContext.fileContext?.rightFileContent ?? '');
+                const isInlineEdit = processedSuggestion.type === codeWhispererService_1.SuggestionType.EDIT;
+                if (isSimilarToRejected) {
+                    // Mark as rejected in the session
+                    session.setSuggestionState(suggestion.itemId, 'Reject');
+                    this.logging.debug(`[EDIT_PREDICTION] Filtered out suggestion similar to previously rejected edit`);
+                    // Return empty item that will be filtered out
+                    return {
+                        insertText: '',
+                        isInlineEdit: isInlineEdit,
+                        itemId: suggestion.itemId,
+                    };
+                }
+                return {
+                    insertText: processedSuggestion.suggestionContent,
+                    isInlineEdit: isInlineEdit,
+                    itemId: suggestion.itemId,
+                };
+            })
+                .filter(item => item.insertText !== ''),
+            sessionId: session.id,
+            partialResultToken: suggestionResponse.responseContext.nextToken,
+        };
+    }
+    handleSuggestionsErrors(error, session) {
+        this.logging.log('Recommendation failure: ' + error);
+        (0, telemetry_1.emitServiceInvocationFailure)(this.telemetry, session, error);
+        // UTDE telemetry is not needed here because in error cases we don't care about UTDE for errored out sessions
+        this.sessionManager.closeSession(session);
+        let translatedError = error;
+        if ((0, utils_1.hasConnectionExpired)(error)) {
+            translatedError = new errors_1.AmazonQServiceConnectionExpiredError((0, utils_1.getErrorMessage)(error));
+        }
+        if (translatedError instanceof errors_1.AmazonQError) {
+            throw new protocol_1.ResponseError(protocol_1.LSPErrorCodes.RequestFailed, translatedError.message || 'Error processing suggestion requests', {
+                awsErrorCode: translatedError.code,
+            });
+        }
+        return constants_1.EMPTY_RESULT;
+    }
+}
+exports.EditCompletionHandler = EditCompletionHandler;
+//# sourceMappingURL=editCompletionHandler.js.map

package/out/language-server/inline-completion/handler/editCompletionHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"editCompletionHandler.js","sourceRoot":"","sources":["../../../../src/language-server/inline-completion/handler/editCompletionHandler.ts"],"names":[],"mappings":";;;AAAA,qEAU+C;AAG/C,+EAM6C;AAC7C,8DAAgF;AAEhF,yEAAiG;AACjG,0FAAsF;AACtF,wDAA0D;AAC1D,sDAK+B;AAE/B,4CAAyC;AAGzC,iDAA6E;AAC7E,yEAAiH;AAEjH,qDAA+E;AAC/E,4DAAwD;AACxD,kDAA0D;AAE1D,MAAa,qBAAqB;IAUjB;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IApBI,YAAY,CAAS;IAC9B,eAAe,CAA4B;IAC3C,SAAS,GAAY,KAAK,CAAA;IAC1B,iCAAiC,GAAY,KAAK,CAAA;IACzC,aAAa,CAAe;IAErC,YAAY,GAAG,KAAK,CAAA;IAE5B,YACa,OAAgB,EAChB,cAAgC,EAChC,SAAoB,EACpB,qBAAgD,EAChD,cAA8B,EAC9B,aAA4B,EAC5B,kBAAqC,EACrC,mBAAwC,EACxC,uBAAgD,EAChD,SAAoB,EACpB,gBAAkC,EAClC,mBAAwC;QAXxC,YAAO,GAAP,OAAO,CAAS;QAChB,mBAAc,GAAd,cAAc,CAAkB;QAChC,cAAS,GAAT,SAAS,CAAW;QACpB,0BAAqB,GAArB,qBAAqB,CAA2B;QAChD,mBAAc,GAAd,cAAc,CAAgB;QAC9B,kBAAa,GAAb,aAAa,CAAe;QAC5B,uBAAkB,GAAlB,kBAAkB,CAAmB;QACrC,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,cAAS,GAAT,SAAS,CAAW;QACpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,wBAAmB,GAAnB,mBAAmB,CAAqB;QAEjD,IAAI,CAAC,YAAY;YACb,IAAI,CAAC,cAAc,CAAC,qBAAqB,EAAE,GAAG,EAAE,qBAAqB,EAAE,YAAY;gBAC/E,EAAE,8BAA8B,EAAE,iBAAiB,IAAI,KAAK,CAAA;QACpE,IAAI,CAAC,aAAa,GAAG,6BAAa,CAAC,WAAW,EAAE,CAAA;IACpD,CAAC;IAED,IAAI,oBAAoB;QACpB,OAAO,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,EAAE,CAAA;IAC/D,CAAC;IAED;;;;OAIG;IACH,eAAe;QACX,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACjB,IAAI,CAAC,iCAAiC,GAAG,IAAI,CAAA;YACjD,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qDAAqD,qCAAyB,EAAE,CAAC,CAAA;gBACnG,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,CAAA;YAClC,CAAC;QACL,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CAClB,MAA4C,EAC5C,KAAwB;QAExB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAA;YACnE,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,kEAAkE;QAClE,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAA;QAC9D,IAAI,cAAc,IAAI,cAAc,CAAC,KAAK,IAAI,YAAY,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACvF,oFAAoF;YACpF,cAAc,CAAC,qCAAqC,GAAG,IAAI,CAAA;QAC/D,CAAC;QAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;QAC9E,CAAC;QACD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;QAClF,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,MAAM,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC,CAAA;YACxE,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,IAAI,CAAC,CAAC,IAAI,CAAC,oBAAoB,YAAY,gDAAyB,CAAC,EAAE,CAAC;YACpE,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,0BAA0B;QAC1B,MAAM,kBAAkB,GAAG,IAAA,0CAAsB,EAAC,YAAY,CAAC,CAAA;QAC/D,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,GAAG,CACZ,iBAAiB,MAAM,CAAC,YAAY,CAAC,GAAG,sBAAsB,YAAY,CAAC,UAAU,iBAAiB,CACzG,CAAA;YACD,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,uFAAuF;QACvF,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QACxB,MAAM,wBAAwB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAE3C,IAAI,MAAM,CAAC,kBAAkB,IAAI,cAAc,EAAE,CAAC;YAC9C,iGAAiG;YACjG,IAAI,cAAc,IAAI,cAAc,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACtD,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,cAAc,CAAC,CAAA;YACtD,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC;gBACjD,QAAQ,EAAE,YAAY;gBACtB,aAAa,EAAE,MAAM,CAAC,QAAQ;gBAC9B,wBAAwB,EAAE,wBAAwB;gBAClD,WAAW,EAAE,aAAa;gBAC1B,QAAQ,EAAE,cAAc,CAAC,QAAQ;gBACjC,cAAc,EAAE,cAAc,CAAC,cAAc;gBAC7C,eAAe,EAAE,SAAS;gBAC1B,gBAAgB,EAAE,EAAE;gBACpB,gBAAgB,EAAE,SAAS;gBAC3B,mBAAmB,EAAE,SAAS;gBAC9B,kBAAkB,EAAE,cAAc,CAAC,kBAAkB;gBACrD,oBAAoB,EAAE,cAAc,CAAC,oBAAoB;gBACzD,gBAAgB,EAAE,cAAc,CAAC,gBAAgB;aACpD,CAAC,CAAA;YACF,oDAAoD;YACpD,IAAI,CAAC;gBACD,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,mBAAmB,CAAC;oBAC3E,GAAG,UAAU,CAAC,cAAc;oBAC5B,SAAS,EAAE,GAAG,MAAM,CAAC,kBAAkB,EAAE;iBAC5C,CAAC,CAAA;gBACF,OAAO,MAAM,IAAI,CAAC,yBAAyB,CACvC,kBAAkB,EAClB,UAAU,EACV,IAAI,EACJ,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAC/C,CAAA;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,KAAc,EAAE,cAAc,CAAC,CAAA;YACvE,CAAC;oBAAS,CAAC;gBACP,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;YAC7B,CAAC;QACL,CAAC;QAED,OAAO,IAAI,OAAO,CAAqC,KAAK,EAAC,OAAO,EAAC,EAAE;YACnE,IAAI,CAAC,eAAe,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;gBACzC,IAAI,CAAC;oBACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA;oBACrB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,MAAM,EACN,wBAAwB,EACxB,KAAK,EACL,YAAY,EACZ,kBAAkB,EAClB,cAAc,CACjB,CAAC,OAAO,CAAC,GAAG,EAAE;wBACX,IAAI,CAAC,SAAS,GAAG,KAAK,CAAA;oBAC1B,CAAC,CAAC,CAAA;oBACF,IAAI,IAAI,CAAC,iCAAiC,EAAE,CAAC;wBACzC,IAAI,CAAC,OAAO,CAAC,IAAI,CACb,mFAAmF,CACtF,CAAA;wBACD,OAAO,CAAC;4BACJ,SAAS,EAAE,+BAAc,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,EAAE,EAAE,IAAI,EAAE;4BAC3E,KAAK,EAAE,EAAE;yBACZ,CAAC,CAAA;oBACN,CAAC;yBAAM,CAAC;wBACJ,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,+DAA+D,CAAC,CAAA;wBAClF,OAAO,CAAC,MAAM,CAAC,CAAA;oBACnB,CAAC;gBACL,CAAC;wBAAS,CAAC;oBACP,IAAI,CAAC,eAAe,GAAG,SAAS,CAAA;oBAChC,IAAI,CAAC,iCAAiC,GAAG,KAAK,CAAA;gBAClD,CAAC;YACL,CAAC,EAAE,qCAAyB,CAAC,CAAA;QACjC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;QAC7B,CAAC,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,OAAO,CACT,MAA4C,EAC5C,wBAAgC,EAChC,KAAwB,EACxB,YAA0B,EAC1B,kBAAyC,EACzC,cAAgD;QAEhD,wBAAwB;QACxB,MAAM,yBAAyB,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,IAAI,sCAA2B,CAAC,SAAS,CAAA;QACrG,MAAM,UAAU,GAAG,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACpD,MAAM,WAAW,GAAG,IAAA,qCAAc,EAAC;YAC/B,YAAY;YACZ,kBAAkB;YAClB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,GAAG,CAAC;SACvE,CAAC,CAAA;QAEF,MAAM,cAAc,GAAG,+CAAsB,CAAC,WAAW,EAAE,EAAE,iBAAiB,EAAE,CAAA;QAChF,MAAM,WAAW,GAAG,cAAc,EAAE,eAAe,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAA;QAE3G,MAAM,aAAa,GAAG,IAAA,iCAAkB,EACpC,IAAI,CAAC,oBAAoB,EACzB,WAAW,EACX,MAAM,EACN,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,cAAc,EACnB,IAAI,CACP,CAAA;QAED,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,MAAM,qBAAqB,GAA+B;YACtD,WAAW,EAAE,WAAW;YACxB,UAAU,EAAE,UAAU;YACtB,eAAe,EAAE,CAAC,OAAO,CAAC;YAC1B,WAAW,EAAE,WAAW;SAC3B,CAAA;QAED,IAAI,aAAa,EAAE,CAAC;YAChB,qBAAqB,CAAC,WAAW,GAAG;gBAChC,QAAQ,EAAE;oBACN,gBAAgB,EAAE,YAAY,CAAC,GAAG;oBAClC,mBAAmB,EAAE;wBACjB,YAAY,EAAE,qBAAqB,CAAC,WAAW,EAAE,mBAAmB,EAAE,YAAY;qBACrF;oBACD,IAAI,EAAE,YAAY,CAAC,OAAO,EAAE;iBAC/B;gBACD,WAAW,EAAE;oBACT,QAAQ,EAAE;wBACN,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;wBAC1B,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;qBACvC;iBACJ;aACJ,CAAA;QACL,CAAC;QAED,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,4BAA4B,CACpF,YAAY,EACZ,MAAM,CAAC,QAAQ,EACf,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,OAAO,EACZ,KAAK,EACL,MAAM,CAAC,gBAAgB,EACvB;YACI,kBAAkB,EAAE,IAAI;SAC3B,CACJ,CAAA;QACD,IAAI,mBAAmB,EAAE,CAAC;YACtB,qBAAqB,CAAC,oBAAoB,GAAG,mBAAmB,CAAC,KAAK,CAAA;QAC1E,CAAC;QAED,uEAAuE;QACvE,IAAI,cAAc,IAAI,cAAc,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YACtD,sEAAsE;YACtE,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,cAAc,CAAC,CAAA;YAClD,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;YAC/F,MAAM,IAAA,4CAAgC,EAClC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,gBAAgB,EACrB,cAAc,EACd,IAAI,CAAC,uBAAuB,CAAC,6BAA6B,EAC1D,CAAC,EACD,CAAC,EACD,EAAE,EACF,EAAE,EACF,YAAY,CACf,CAAA;QACL,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC;YACjD,QAAQ,EAAE,YAAY;YACtB,wBAAwB,EAAE,wBAAwB;YAClD,aAAa,EAAE,MAAM,CAAC,QAAQ;YAC9B,WAAW,EAAE,yBAAyB,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,UAAU;YACnE,QAAQ,EAAE,WAAW,CAAC,mBAAmB,CAAC,YAAY;YACtD,cAAc,EAAE,qBAAqB;YACrC,eAAe,EAAE,SAAS;YAC1B,gBAAgB,EAAE,EAAE;YACpB,gBAAgB,EAAE,SAAS;YAC3B,mBAAmB,EAAE,SAAS;YAC9B,kBAAkB,EAAE,IAAI,CAAC,mBAAmB,CAAC,qBAAqB,EAAE,EAAE,EAAE,GAAG,EAAE,QAAQ,IAAI,SAAS;YAClG,oBAAoB,EAAE,mBAAmB,EAAE,cAAc;YACzD,gBAAgB,EAAE,oBAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC,oBAAoB,CAAC,gBAAgB,CAAC;SAC3F,CAAC,CAAA;QAEF,IAAI,CAAC;YACD,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,CAAA;YACrG,OAAO,MAAM,IAAI,CAAC,yBAAyB,CACvC,kBAAkB,EAClB,UAAU,EACV,IAAI,EACJ,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAC/C,CAAA;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,KAAc,EAAE,UAAU,CAAC,CAAA;QACnE,CAAC;IACL,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC3B,kBAA+C,EAC/C,OAA6B,EAC7B,YAAqB,EACrB,cAAsB,EACtB,YAA2B;QAE3B,2EAA2E;QAC3E,0DAA0D;QAC1D,6DAA6D;QAE7D,IAAI,YAAY,EAAE,CAAC;YACf,oEAAoE;YACpE,OAAO,CAAC,WAAW,GAAG,kBAAkB,CAAC,WAAW,CAAA;YACpD,OAAO,CAAC,eAAe,GAAG,kBAAkB,CAAC,eAAe,CAAA;YAC5D,OAAO,CAAC,sBAAsB,GAAG,kBAAkB,CAAC,eAAe,CAAC,sBAAsB,CAAA;YAC1F,OAAO,CAAC,4BAA4B,EAAE,CAAA;YACtC,OAAO,CAAC,cAAc,GAAG,qCAAc,CAAC,IAAI,CAAA;QAChD,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,WAAW,GAAG,CAAC,GAAG,OAAO,CAAC,WAAW,EAAE,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAA;QACrF,CAAC;QAED,sEAAsE;QACtE,IAAA,0CAA8B,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,kBAAkB,CAAC,eAAe,CAAC,SAAS,CAAC,CAAA;QAErG,4DAA4D;QAC5D,IAAI,OAAO,CAAC,qCAAqC,EAAE,CAAC;YAChD,OAAO,CAAC,qCAAqC,GAAG,KAAK,CAAA;YACrD,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;YAC3C,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;YAC/F,MAAM,IAAA,4CAAgC,EAClC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,gBAAgB,EACrB,OAAO,EACP,IAAI,CAAC,uBAAuB,CAAC,6BAA6B,EAC1D,CAAC,EACD,CAAC,EACD,EAAE,EACF,EAAE,EACF,YAAY,CACf,CAAA;QACL,CAAC;QAED,yDAAyD;QACzD,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;QAE5C,uDAAuD;QACvD,IAAI,kBAAkB,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9C,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,CAAA;YACzC,MAAM,IAAA,iDAAqC,EACvC,IAAI,CAAC,gBAAgB,EACrB,OAAO,EACP,IAAI,CAAC,uBAAuB,CAAC,6BAA6B,EAC1D,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAC7E,CAAA;YACD,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,OAAO;YACH,KAAK,EAAE,kBAAkB,CAAC,WAAW;iBAChC,GAAG,CAAC,UAAU,CAAC,EAAE;gBACd,oEAAoE;gBACpE,MAAM,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CACpE,UAAU,CAAC,OAAO,IAAI,EAAE,EACxB,YAAY,EAAE,GAAG,IAAI,EAAE,CAC1B,CAAA;gBAED,MAAM,mBAAmB,GAAG,IAAA,iCAAqB,EAC7C,UAAU,CAAC,OAAO,IAAI,EAAE,EACxB,OAAO,CAAC,aAAa,EACrB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,cAAc,CAAC,WAAW,EAAE,gBAAgB,IAAI,EAAE,CAC7D,CAAA;gBACD,MAAM,YAAY,GAAG,mBAAmB,CAAC,IAAI,KAAK,qCAAc,CAAC,IAAI,CAAA;gBAErE,IAAI,mBAAmB,EAAE,CAAC;oBACtB,kCAAkC;oBAClC,OAAO,CAAC,kBAAkB,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;oBACvD,IAAI,CAAC,OAAO,CAAC,KAAK,CACd,+EAA+E,CAClF,CAAA;oBACD,8CAA8C;oBAC9C,OAAO;wBACH,UAAU,EAAE,EAAE;wBACd,YAAY,EAAE,YAAY;wBAC1B,MAAM,EAAE,UAAU,CAAC,MAAM;qBAC5B,CAAA;gBACL,CAAC;gBAED,OAAO;oBACH,UAAU,EAAE,mBAAmB,CAAC,iBAAiB;oBACjD,YAAY,EAAE,YAAY;oBAC1B,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC5B,CAAA;YACL,CAAC,CAAC;iBACD,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,KAAK,EAAE,CAAC;YAC3C,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,kBAAkB,EAAE,kBAAkB,CAAC,eAAe,CAAC,SAAS;SACnE,CAAA;IACL,CAAC;IAED,uBAAuB,CAAC,KAAY,EAAE,OAA6B;QAC/D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,GAAG,KAAK,CAAC,CAAA;QACpD,IAAA,wCAA4B,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,CAAA;QAE5D,6GAA6G;QAC7G,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,CAAA;QAEzC,IAAI,eAAe,GAAG,KAAK,CAAA;QAE3B,IAAI,IAAA,4BAAoB,EAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,eAAe,GAAG,IAAI,6CAAoC,CAAC,IAAA,uBAAe,EAAC,KAAK,CAAC,CAAC,CAAA;QACtF,CAAC;QAED,IAAI,eAAe,YAAY,qBAAY,EAAE,CAAC;YAC1C,MAAM,IAAI,wBAAa,CACnB,wBAAa,CAAC,aAAa,EAC3B,eAAe,CAAC,OAAO,IAAI,sCAAsC,EACjE;gBACI,YAAY,EAAE,eAAe,CAAC,IAAI;aACrC,CACJ,CAAA;QACL,CAAC;QAED,OAAO,wBAAY,CAAA;IACvB,CAAC;CACJ;AAhaD,sDAgaC"}

package/out/language-server/inline-completion/handler/inlineCompletionHandler.d.ts

@@ -0,0 +1,32 @@
+import { CancellationToken, InlineCompletionListWithReferences, InlineCompletionWithReferencesParams, Logging, Telemetry, Workspace, CredentialsProvider, Lsp } from '@aws/language-server-runtimes/server-interface';
+import { SessionManager } from '../session/sessionManager';
+import { CodePercentageTracker } from '../tracker/codePercentageTracker';
+import { TelemetryService } from '../../../shared/telemetry/telemetryService';
+import { UserWrittenCodeTracker } from '../../../shared/userWrittenCodeTracker';
+import { RecentEditTracker } from '../tracker/codeEditTracker';
+import { CursorTracker } from '../tracker/cursorTracker';
+import { StreakTracker } from '../tracker/streakTracker';
+import { AmazonQBaseServiceManager } from '../../../shared/amazonQServiceManager/BaseAmazonQServiceManager';
+export declare class InlineCompletionHandler {
+    private readonly logging;
+    private readonly workspace;
+    private readonly amazonQServiceManager;
+    private readonly completionSessionManager;
+    private readonly codePercentageTracker;
+    private readonly userWrittenCodeTracker;
+    private readonly recentEditTracker;
+    private readonly cursorTracker;
+    private readonly streakTracker;
+    private readonly telemetry;
+    private readonly telemetryService;
+    private readonly credentialsProvider;
+    private readonly getEditsEnabled;
+    private readonly getTimeSinceLastUserModification;
+    private readonly lsp;
+    private isOnInlineCompletionHandlerInProgress;
+    constructor(logging: Logging, workspace: Workspace, amazonQServiceManager: AmazonQBaseServiceManager, completionSessionManager: SessionManager, codePercentageTracker: CodePercentageTracker, userWrittenCodeTracker: UserWrittenCodeTracker | undefined, recentEditTracker: RecentEditTracker, cursorTracker: CursorTracker, streakTracker: StreakTracker, telemetry: Telemetry, telemetryService: TelemetryService, credentialsProvider: CredentialsProvider, getEditsEnabled: () => boolean, getTimeSinceLastUserModification: () => number, lsp: Lsp);
+    onInlineCompletion(params: InlineCompletionWithReferencesParams, token: CancellationToken): Promise<InlineCompletionListWithReferences>;
+    private processNewCompletionRequest;
+    private processSuggestionResponse;
+    private handleSuggestionsErrors;
+}