@aws/lsp-codewhisperer 0.0.83 → 0.0.84

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

@@ -1,14 +1,14 @@
 import { CancellationToken, InitializeParams, InlineCompletionListWithReferences, InlineCompletionWithReferencesParams, Range, TextDocument } from '@aws/language-server-runtimes/protocol';
-import { RecentEditTracker } from './tracker/codeEditTracker';
+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';
+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;
@@ -29,7 +29,7 @@ export declare class EditCompletionHandler {
     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;
+    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.

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

@@ -2,17 +2,18 @@
 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 trigger_1 = require("./trigger");
-const telemetry_1 = require("./telemetry/telemetry");
+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 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;
@@ -175,7 +176,7 @@ class EditCompletionHandler {
         });
         const workspaceState = workspaceFolderManager_1.WorkspaceFolderManager.getInstance()?.getWorkspaceState();
         const workspaceId = workspaceState?.webSocketClient?.isConnected() ? workspaceState.workspaceId : undefined;
-        const qEditsTrigger = (0, trigger_1.shouldTriggerEdits)(this.codeWhispererService, fileContext, params, this.cursorTracker, this.recentEditsTracker, this.sessionManager, true);
+        const qEditsTrigger = (0, triggerUtils_1.shouldTriggerEdits)(this.codeWhispererService, fileContext, params, this.cursorTracker, this.recentEditsTracker, this.sessionManager, true);
         if (!qEditsTrigger) {
             return constants_1.EMPTY_RESULT;
         }
@@ -275,6 +276,8 @@ class EditCompletionHandler {
                 .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');
@@ -282,13 +285,13 @@ class EditCompletionHandler {
                     // Return empty item that will be filtered out
                     return {
                         insertText: '',
-                        isInlineEdit: true,
+                        isInlineEdit: isInlineEdit,
                         itemId: suggestion.itemId,
                     };
                 }
                 return {
-                    insertText: suggestion.content,
-                    isInlineEdit: true,
+                    insertText: processedSuggestion.suggestionContent,
+                    isInlineEdit: isInlineEdit,
                     itemId: suggestion.itemId,
                 };
             })

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,CAAC,mBAAmB,CAAC,YAAY;qBACnF;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,EAClB,YAAY,EAAE,GAAG,IAAI,EAAE,CAC1B,CAAA;gBAED,MAAM,mBAAmB,GAAG,IAAA,iCAAqB,EAC7C,UAAU,CAAC,OAAO,EAClB,OAAO,CAAC,aAAa,EACrB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,gBAAgB,CACtD,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;
+}

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

@@ -0,0 +1,377 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InlineCompletionHandler = void 0;
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const autoTrigger_1 = require("../auto-trigger/autoTrigger");
+const codeWhispererService_1 = require("../../../shared/codeWhispererService");
+const languageDetection_1 = require("../../../shared/languageDetection");
+const utils_1 = require("../../../shared/utils");
+const telemetryUtils_1 = require("../../../shared/telemetryUtils");
+const lsp_core_1 = require("@aws/lsp-core");
+const errors_1 = require("../../../shared/amazonQServiceManager/errors");
+const utils_2 = require("../../../shared/utils");
+const workspaceFolderManager_1 = require("../../workspaceContext/workspaceFolderManager");
+const telemetry_1 = require("../telemetry/telemetry");
+const constants_1 = require("../contants/constants");
+const IdleWorkspaceManager_1 = require("../../workspaceContext/IdleWorkspaceManager");
+const mergeRightUtils_1 = require("../utils/mergeRightUtils");
+const textDocumentUtils_1 = require("../utils/textDocumentUtils");
+class InlineCompletionHandler {
+    logging;
+    workspace;
+    amazonQServiceManager;
+    completionSessionManager;
+    codePercentageTracker;
+    userWrittenCodeTracker;
+    recentEditTracker;
+    cursorTracker;
+    streakTracker;
+    telemetry;
+    telemetryService;
+    credentialsProvider;
+    getEditsEnabled;
+    getTimeSinceLastUserModification;
+    lsp;
+    isOnInlineCompletionHandlerInProgress = false;
+    constructor(logging, workspace, amazonQServiceManager, completionSessionManager, codePercentageTracker, userWrittenCodeTracker, recentEditTracker, cursorTracker, streakTracker, telemetry, telemetryService, credentialsProvider, getEditsEnabled, getTimeSinceLastUserModification, lsp) {
+        this.logging = logging;
+        this.workspace = workspace;
+        this.amazonQServiceManager = amazonQServiceManager;
+        this.completionSessionManager = completionSessionManager;
+        this.codePercentageTracker = codePercentageTracker;
+        this.userWrittenCodeTracker = userWrittenCodeTracker;
+        this.recentEditTracker = recentEditTracker;
+        this.cursorTracker = cursorTracker;
+        this.streakTracker = streakTracker;
+        this.telemetry = telemetry;
+        this.telemetryService = telemetryService;
+        this.credentialsProvider = credentialsProvider;
+        this.getEditsEnabled = getEditsEnabled;
+        this.getTimeSinceLastUserModification = getTimeSinceLastUserModification;
+        this.lsp = lsp;
+    }
+    async onInlineCompletion(params, token) {
+        // this handle should not run concurrently because
+        // 1. it would create a high volume of traffic, causing throttling
+        // 2. it is not designed to handle concurrent changes to these state variables.
+        // when one handler is at the API call stage, it has not yet update the session state
+        // but another request can start, causing the state to be incorrect.
+        IdleWorkspaceManager_1.IdleWorkspaceManager.recordActivityTimestamp();
+        if (this.isOnInlineCompletionHandlerInProgress) {
+            this.logging.log(`Skip concurrent inline completion`);
+            return constants_1.EMPTY_RESULT;
+        }
+        // Add this check to ensure service manager is initialized
+        if (!this.amazonQServiceManager) {
+            this.logging.log('Amazon Q Service Manager not initialized yet');
+            return constants_1.EMPTY_RESULT;
+        }
+        this.isOnInlineCompletionHandlerInProgress = true;
+        try {
+            // On every new completion request close current inflight session.
+            const currentSession = this.completionSessionManager.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 (0, textDocumentUtils_1.getTextDocument)(params.textDocument.uri, this.workspace, this.logging);
+            const codeWhispererService = this.amazonQServiceManager.getCodewhispererService();
+            const authType = codeWhispererService instanceof codeWhispererService_1.CodeWhispererServiceToken ? 'token' : 'iam';
+            this.logging.debug(`[INLINE_COMPLETION] Service ready - auth: ${authType}, partial token: ${!!params.partialResultToken}`);
+            if (params.partialResultToken && currentSession) {
+                // subsequent paginated requests for current session
+                try {
+                    const suggestionResponse = await codeWhispererService.generateSuggestions({
+                        ...currentSession.requestContext,
+                        fileContext: {
+                            ...currentSession.requestContext.fileContext,
+                        },
+                        nextToken: `${params.partialResultToken}`,
+                    });
+                    return await this.processSuggestionResponse(suggestionResponse, currentSession, false, params.context.selectedCompletionInfo?.range);
+                }
+                catch (error) {
+                    return this.handleSuggestionsErrors(error, currentSession);
+                }
+            }
+            else {
+                return await this.processNewCompletionRequest(params, textDocument, token);
+            }
+        }
+        finally {
+            this.isOnInlineCompletionHandlerInProgress = false;
+        }
+    }
+    async processNewCompletionRequest(params, textDocument, token) {
+        // request for new session
+        if (!textDocument) {
+            this.logging.log(`textDocument [${params.textDocument.uri}] not found`);
+            return constants_1.EMPTY_RESULT;
+        }
+        let inferredLanguageId = (0, languageDetection_1.getSupportedLanguageId)(textDocument);
+        if (params.fileContextOverride?.programmingLanguage) {
+            inferredLanguageId = params.fileContextOverride?.programmingLanguage;
+        }
+        if (!inferredLanguageId) {
+            this.logging.log(`textDocument [${params.textDocument.uri}] with languageId [${textDocument.languageId}] not supported`);
+            return constants_1.EMPTY_RESULT;
+        }
+        // Build request context
+        const isAutomaticLspTriggerKind = params.context.triggerKind == server_interface_1.InlineCompletionTriggerKind.Automatic;
+        const maxResults = isAutomaticLspTriggerKind ? 1 : 5;
+        const selectionRange = params.context.selectedCompletionInfo?.range;
+        const startPreprocessTimestamp = Date.now();
+        // For Jupyter Notebook in VSC, the language server does not have access to
+        // its internal states including current active cell index, etc
+        // we rely on VSC to calculate file context
+        let fileContext = undefined;
+        if (params.fileContextOverride) {
+            fileContext = {
+                leftFileContent: params.fileContextOverride.leftFileContent,
+                rightFileContent: params.fileContextOverride.rightFileContent,
+                filename: params.fileContextOverride.filename,
+                fileUri: params.fileContextOverride.fileUri,
+                programmingLanguage: {
+                    languageName: inferredLanguageId,
+                },
+            };
+        }
+        else {
+            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 previousSession = this.completionSessionManager.getPreviousSession();
+        // Only refer to decisions in the past 2 mins
+        const previousDecisionForClassifier = previousSession && Date.now() - previousSession.decisionMadeTimestamp <= 2 * 60 * 1000
+            ? previousSession.getAggregatedUserTriggerDecision()
+            : undefined;
+        let ideCategory = '';
+        const initializeParams = this.lsp.getClientInitializeParams();
+        if (initializeParams !== undefined) {
+            ideCategory = (0, telemetryUtils_1.getIdeCategory)(initializeParams);
+        }
+        // auto trigger code path
+        let codewhispererAutoTriggerType = undefined;
+        let triggerCharacters = '';
+        let autoTriggerResult = undefined;
+        if (isAutomaticLspTriggerKind) {
+            // Reference: https://github.com/aws/aws-toolkit-vscode/blob/amazonq/v1.74.0/packages/core/src/codewhisperer/service/classifierTrigger.ts#L477
+            if (params.documentChangeParams?.contentChanges &&
+                params.documentChangeParams.contentChanges.length > 0 &&
+                params.documentChangeParams.contentChanges[0].text !== undefined) {
+                triggerCharacters = params.documentChangeParams.contentChanges[0].text;
+                codewhispererAutoTriggerType = (0, autoTrigger_1.getAutoTriggerType)(params.documentChangeParams.contentChanges);
+            }
+            else {
+                // if the client does not emit document change for the trigger, use left most character.
+                triggerCharacters = fileContext.leftFileContent.trim().at(-1) ?? '';
+                codewhispererAutoTriggerType = (0, autoTrigger_1.triggerType)(fileContext);
+            }
+            // See: https://github.com/aws/aws-toolkit-vscode/blob/amazonq/v1.74.0/packages/core/src/codewhisperer/service/keyStrokeHandler.ts#L132
+            // In such cases, do not auto trigger.
+            if (codewhispererAutoTriggerType === undefined) {
+                return constants_1.EMPTY_RESULT;
+            }
+            autoTriggerResult = (0, autoTrigger_1.autoTrigger)({
+                fileContext, // The left/right file context and programming language
+                lineNum: params.position.line, // the line number of the invocation, this is the line of the cursor
+                char: triggerCharacters, // Add the character just inserted, if any, before the invication position
+                ide: ideCategory ?? '',
+                os: (0, autoTrigger_1.getNormalizeOsName)(),
+                previousDecision: previousDecisionForClassifier, // The last decision by the user on the previous invocation
+                triggerType: codewhispererAutoTriggerType, // The 2 trigger types currently influencing the Auto-Trigger are SpecialCharacter and Enter
+            }, this.logging);
+            if (codewhispererAutoTriggerType === 'Classifier' && !autoTriggerResult.shouldTrigger) {
+                return constants_1.EMPTY_RESULT;
+            }
+        }
+        let requestContext = {
+            fileContext,
+            maxResults,
+        };
+        const codeWhispererService = this.amazonQServiceManager.getCodewhispererService();
+        const supplementalContext = await codeWhispererService.constructSupplementalContext(textDocument, params.position, this.workspace, this.recentEditTracker, this.logging, token, params.openTabFilepaths, { includeRecentEdits: false });
+        if (supplementalContext?.items) {
+            requestContext.supplementalContexts = supplementalContext.items;
+        }
+        // Close ACTIVE session and record Discard trigger decision immediately
+        const currentSession = this.completionSessionManager.getCurrentSession();
+        if (currentSession && currentSession.state === 'ACTIVE') {
+            // Emit user trigger decision at session close time for active session
+            // TODO: yuxqiang workaround to exclude JB from this logic because JB and VSC handle a
+            // bit differently in the case when there's a new trigger while a reject/discard event is sent
+            // for the previous trigger
+            if (ideCategory !== 'JETBRAINS') {
+                this.completionSessionManager.discardSession(currentSession);
+                const streakLength = this.getEditsEnabled() ? this.streakTracker.getAndUpdateStreakLength(false) : 0;
+                await (0, telemetry_1.emitUserTriggerDecisionTelemetry)(this.telemetry, this.telemetryService, currentSession, this.getTimeSinceLastUserModification(), 0, 0, [], [], streakLength);
+            }
+        }
+        const supplementalMetadata = supplementalContext?.supContextData;
+        const newSession = this.completionSessionManager.createSession({
+            document: textDocument,
+            startPreprocessTimestamp: startPreprocessTimestamp,
+            startPosition: params.position,
+            triggerType: isAutomaticLspTriggerKind ? 'AutoTrigger' : 'OnDemand',
+            language: fileContext.programmingLanguage.languageName,
+            requestContext: requestContext,
+            autoTriggerType: isAutomaticLspTriggerKind ? codewhispererAutoTriggerType : undefined,
+            triggerCharacter: triggerCharacters,
+            classifierResult: autoTriggerResult?.classifierResult,
+            classifierThreshold: autoTriggerResult?.classifierThreshold,
+            credentialStartUrl: this.credentialsProvider.getConnectionMetadata?.()?.sso?.startUrl ?? undefined,
+            supplementalMetadata: supplementalMetadata,
+            customizationArn: lsp_core_1.textUtils.undefinedIfEmpty(codeWhispererService.customizationArn),
+        });
+        // Add extra context to request context
+        const { extraContext } = this.amazonQServiceManager.getConfiguration().inlineSuggestions;
+        if (extraContext) {
+            requestContext.fileContext.leftFileContent =
+                extraContext + '\n' + requestContext.fileContext.leftFileContent;
+        }
+        // Create the appropriate request based on service type
+        let generateCompletionReq;
+        if (codeWhispererService instanceof codeWhispererService_1.CodeWhispererServiceToken) {
+            const tokenRequest = requestContext;
+            generateCompletionReq = {
+                ...tokenRequest,
+                ...(workspaceId ? { workspaceId } : {}),
+            };
+        }
+        else {
+            const iamRequest = requestContext;
+            generateCompletionReq = {
+                ...iamRequest,
+            };
+        }
+        try {
+            const authType = codeWhispererService instanceof codeWhispererService_1.CodeWhispererServiceToken ? 'token' : 'iam';
+            this.logging.debug(`[INLINE_COMPLETION] API call - generateSuggestions (new session, ${authType})`);
+            const suggestionResponse = await codeWhispererService.generateSuggestions(generateCompletionReq);
+            return await this.processSuggestionResponse(suggestionResponse, newSession, true, selectionRange);
+        }
+        catch (error) {
+            return this.handleSuggestionsErrors(error, newSession);
+        }
+    }
+    async processSuggestionResponse(suggestionResponse, session, isNewSession, selectionRange) {
+        this.codePercentageTracker.countInvocation(session.language);
+        this.userWrittenCodeTracker?.recordUsageCount(session.language);
+        session.includeImportsWithSuggestions =
+            this.amazonQServiceManager.getConfiguration().includeImportsWithSuggestions;
+        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.COMPLETION;
+        }
+        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.completionSessionManager.discardSession(session);
+            const streakLength = this.getEditsEnabled() ? this.streakTracker.getAndUpdateStreakLength(false) : 0;
+            await (0, telemetry_1.emitUserTriggerDecisionTelemetry)(this.telemetry, this.telemetryService, session, this.getTimeSinceLastUserModification(), 0, 0, [], [], streakLength);
+        }
+        // session was closed by user already made decisions consequent completion request before new paginated API response was received
+        if (session.predictionType !== codeWhispererService_1.SuggestionType.EDIT && // TODO: this is a shorterm fix to allow Edits tabtabtab experience, however the real solution is to manage such sessions correctly
+            (session.state === 'CLOSED' || session.state === 'DISCARD')) {
+            return constants_1.EMPTY_RESULT;
+        }
+        // API response was recieved, we can activate session now
+        this.completionSessionManager.activateSession(session);
+        // Process suggestions to apply Empty or Filter filters
+        const filteredSuggestions = suggestionResponse.suggestions
+            // Empty suggestion filter
+            .filter(suggestion => {
+            if (suggestion.content === '') {
+                session.setSuggestionState(suggestion.itemId, 'Empty');
+                return false;
+            }
+            return true;
+        })
+            // References setting filter
+            .filter(suggestion => {
+            // State to track whether code with references should be included in
+            // the response. No locking or concurrency controls, filtering is done
+            // right before returning and is only guaranteed to be consistent within
+            // the context of a single response.
+            const { includeSuggestionsWithCodeReferences } = this.amazonQServiceManager.getConfiguration();
+            if (includeSuggestionsWithCodeReferences) {
+                return true;
+            }
+            if (suggestion.references == null || suggestion.references.length === 0) {
+                return true;
+            }
+            // Filter out suggestions that have references when includeSuggestionsWithCodeReferences setting is true
+            session.setSuggestionState(suggestion.itemId, 'Filter');
+            return false;
+        });
+        const { includeImportsWithSuggestions } = this.amazonQServiceManager.getConfiguration();
+        const suggestionsWithRightContext = (0, mergeRightUtils_1.mergeSuggestionsWithRightContext)(session.requestContext.fileContext.rightFileContent, filteredSuggestions, includeImportsWithSuggestions, selectionRange).filter(suggestion => {
+            // Discard suggestions that have empty string insertText after right context merge and can't be displayed anymore
+            if (suggestion.insertText === '') {
+                session.setSuggestionState(suggestion.itemId, 'Discard');
+                return false;
+            }
+            return true;
+        });
+        suggestionsWithRightContext.forEach(suggestion => {
+            const cachedSuggestion = session.suggestions.find(s => s.itemId === suggestion.itemId);
+            if (cachedSuggestion)
+                cachedSuggestion.insertText = suggestion.insertText.toString();
+        });
+        // TODO: need dedupe after right context merging but I don't see one
+        session.suggestionsAfterRightContextMerge.push(...suggestionsWithRightContext);
+        session.codewhispererSuggestionImportCount =
+            session.codewhispererSuggestionImportCount +
+                suggestionsWithRightContext.reduce((total, suggestion) => {
+                    return total + (suggestion.mostRelevantMissingImports?.length || 0);
+                }, 0);
+        // If after all server-side filtering no suggestions can be displayed, and there is no nextToken
+        // close session and return empty results
+        if (session.suggestionsAfterRightContextMerge.length === 0 && !suggestionResponse.responseContext.nextToken) {
+            this.completionSessionManager.closeSession(session);
+            await (0, telemetry_1.emitUserTriggerDecisionTelemetry)(this.telemetry, this.telemetryService, session, this.getTimeSinceLastUserModification());
+            return constants_1.EMPTY_RESULT;
+        }
+        return {
+            items: suggestionsWithRightContext,
+            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.completionSessionManager.closeSession(session);
+        let translatedError = error;
+        if ((0, utils_2.hasConnectionExpired)(error)) {
+            translatedError = new errors_1.AmazonQServiceConnectionExpiredError((0, utils_1.getErrorMessage)(error));
+        }
+        if (translatedError instanceof errors_1.AmazonQError) {
+            throw new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, translatedError.message || 'Error processing suggestion requests', {
+                awsErrorCode: translatedError.code,
+            });
+        }
+        return constants_1.EMPTY_RESULT;
+    }
+}
+exports.InlineCompletionHandler = InlineCompletionHandler;
+//# sourceMappingURL=inlineCompletionHandler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"inlineCompletionHandler.js","sourceRoot":"","sources":["../../../../src/language-server/inline-completion/handler/inlineCompletionHandler.ts"],"names":[],"mappings":";;;AAAA,qFAcuD;AACvD,6DAA8G;AAC9G,+EAU6C;AAC7C,yEAAiG;AAGjG,iDAAuD;AACvD,mEAA+D;AAC/D,4CAAyC;AAMzC,yEAAiH;AAEjH,iDAA4D;AAC5D,0FAAsF;AACtF,sDAI+B;AAC/B,qDAAoD;AACpD,sFAAkF;AAClF,8DAA2E;AAC3E,kEAA4D;AAE5D,MAAa,uBAAuB;IAIX;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAjBb,qCAAqC,GAAG,KAAK,CAAA;IAErD,YACqB,OAAgB,EAChB,SAAoB,EACpB,qBAAgD,EAChD,wBAAwC,EACxC,qBAA4C,EAC5C,sBAA0D,EAC1D,iBAAoC,EACpC,aAA4B,EAC5B,aAA4B,EAC5B,SAAoB,EACpB,gBAAkC,EAClC,mBAAwC,EACxC,eAA8B,EAC9B,gCAA8C,EAC9C,GAAQ;QAdR,YAAO,GAAP,OAAO,CAAS;QAChB,cAAS,GAAT,SAAS,CAAW;QACpB,0BAAqB,GAArB,qBAAqB,CAA2B;QAChD,6BAAwB,GAAxB,wBAAwB,CAAgB;QACxC,0BAAqB,GAArB,qBAAqB,CAAuB;QAC5C,2BAAsB,GAAtB,sBAAsB,CAAoC;QAC1D,sBAAiB,GAAjB,iBAAiB,CAAmB;QACpC,kBAAa,GAAb,aAAa,CAAe;QAC5B,kBAAa,GAAb,aAAa,CAAe;QAC5B,cAAS,GAAT,SAAS,CAAW;QACpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,oBAAe,GAAf,eAAe,CAAe;QAC9B,qCAAgC,GAAhC,gCAAgC,CAAc;QAC9C,QAAG,GAAH,GAAG,CAAK;IAC1B,CAAC;IAEJ,KAAK,CAAC,kBAAkB,CACpB,MAA4C,EAC5C,KAAwB;QAExB,kDAAkD;QAClD,kEAAkE;QAClE,+EAA+E;QAC/E,qFAAqF;QACrF,oEAAoE;QACpE,2CAAoB,CAAC,uBAAuB,EAAE,CAAA;QAE9C,IAAI,IAAI,CAAC,qCAAqC,EAAE,CAAC;YAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAA;YACrD,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,0DAA0D;QAC1D,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAA;YAChE,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,IAAI,CAAC,qCAAqC,GAAG,IAAI,CAAA;QAEjD,IAAI,CAAC;YACD,kEAAkE;YAClE,MAAM,cAAc,GAAG,IAAI,CAAC,wBAAwB,CAAC,iBAAiB,EAAE,CAAA;YACxE,IAAI,cAAc,IAAI,cAAc,CAAC,KAAK,IAAI,YAAY,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBACvF,oFAAoF;gBACpF,cAAc,CAAC,qCAAqC,GAAG,IAAI,CAAA;YAC/D,CAAC;YAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACrB,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;YAC9E,CAAC;YACD,MAAM,YAAY,GAAG,MAAM,IAAA,mCAAe,EAAC,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;YAEjG,MAAM,oBAAoB,GAAG,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,EAAE,CAAA;YACjF,MAAM,QAAQ,GAAG,oBAAoB,YAAY,gDAAyB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAA;YAC5F,IAAI,CAAC,OAAO,CAAC,KAAK,CACd,6CAA6C,QAAQ,oBAAoB,CAAC,CAAC,MAAM,CAAC,kBAAkB,EAAE,CACzG,CAAA;YAED,IAAI,MAAM,CAAC,kBAAkB,IAAI,cAAc,EAAE,CAAC;gBAC9C,oDAAoD;gBACpD,IAAI,CAAC;oBACD,MAAM,kBAAkB,GAAG,MAAM,oBAAoB,CAAC,mBAAmB,CAAC;wBACtE,GAAG,cAAc,CAAC,cAAc;wBAChC,WAAW,EAAE;4BACT,GAAG,cAAc,CAAC,cAAc,CAAC,WAAW;yBAC/C;wBACD,SAAS,EAAE,GAAG,MAAM,CAAC,kBAAkB,EAAE;qBAC5C,CAAC,CAAA;oBACF,OAAO,MAAM,IAAI,CAAC,yBAAyB,CACvC,kBAAkB,EAClB,cAAc,EACd,KAAK,EACL,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAC/C,CAAA;gBACL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,KAAc,EAAE,cAAc,CAAC,CAAA;gBACvE,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,OAAO,MAAM,IAAI,CAAC,2BAA2B,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,CAAC,CAAA;YAC9E,CAAC;QACL,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,qCAAqC,GAAG,KAAK,CAAA;QACtD,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,2BAA2B,CACrC,MAA4C,EAC5C,YAAsC,EACtC,KAAwB;QAExB,0BAA0B;QAC1B,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,MAAM,CAAC,YAAY,CAAC,GAAG,aAAa,CAAC,CAAA;YACvE,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,IAAI,kBAAkB,GAAG,IAAA,0CAAsB,EAAC,YAAY,CAAC,CAAA;QAC7D,IAAI,MAAM,CAAC,mBAAmB,EAAE,mBAAmB,EAAE,CAAC;YAClD,kBAAkB,GAAG,MAAM,CAAC,mBAAmB,EAAE,mBAA4C,CAAA;QACjG,CAAC;QACD,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,wBAAwB;QACxB,MAAM,yBAAyB,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,IAAI,8CAA2B,CAAC,SAAS,CAAA;QACrG,MAAM,UAAU,GAAG,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACpD,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAAA;QAEnE,MAAM,wBAAwB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAE3C,2EAA2E;QAC3E,+DAA+D;QAC/D,2CAA2C;QAC3C,IAAI,WAAW,GAA4B,SAAS,CAAA;QACpD,IAAI,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC7B,WAAW,GAAG;gBACV,eAAe,EAAE,MAAM,CAAC,mBAAmB,CAAC,eAAe;gBAC3D,gBAAgB,EAAE,MAAM,CAAC,mBAAmB,CAAC,gBAAgB;gBAC7D,QAAQ,EAAE,MAAM,CAAC,mBAAmB,CAAC,QAAQ;gBAC7C,OAAO,EAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO;gBAC3C,mBAAmB,EAAE;oBACjB,YAAY,EAAE,kBAAkB;iBACnC;aACJ,CAAA;QACL,CAAC;aAAM,CAAC;YACJ,WAAW,GAAG,IAAA,qCAAc,EAAC;gBACzB,YAAY;gBACZ,kBAAkB;gBAClB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,GAAG,CAAC;aACvE,CAAC,CAAA;QACN,CAAC;QAED,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,eAAe,GAAG,IAAI,CAAC,wBAAwB,CAAC,kBAAkB,EAAE,CAAA;QAC1E,6CAA6C;QAC7C,MAAM,6BAA6B,GAC/B,eAAe,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,eAAe,CAAC,qBAAqB,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI;YAClF,CAAC,CAAC,eAAe,CAAC,gCAAgC,EAAE;YACpD,CAAC,CAAC,SAAS,CAAA;QACnB,IAAI,WAAW,GAAuB,EAAE,CAAA;QACxC,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,yBAAyB,EAAE,CAAA;QAC7D,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACjC,WAAW,GAAG,IAAA,+BAAc,EAAC,gBAAgB,CAAC,CAAA;QAClD,CAAC;QAED,yBAAyB;QACzB,IAAI,4BAA4B,GAAG,SAAS,CAAA;QAC5C,IAAI,iBAAiB,GAAG,EAAE,CAAA;QAC1B,IAAI,iBAAiB,GAAG,SAAS,CAAA;QAEjC,IAAI,yBAAyB,EAAE,CAAC;YAC5B,8IAA8I;YAC9I,IACI,MAAM,CAAC,oBAAoB,EAAE,cAAc;gBAC3C,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC;gBACrD,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,EAClE,CAAC;gBACC,iBAAiB,GAAG,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;gBACtE,4BAA4B,GAAG,IAAA,gCAAkB,EAAC,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAA;YACjG,CAAC;iBAAM,CAAC;gBACJ,wFAAwF;gBACxF,iBAAiB,GAAG,WAAW,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;gBACnE,4BAA4B,GAAG,IAAA,yBAAW,EAAC,WAAW,CAAC,CAAA;YAC3D,CAAC;YACD,uIAAuI;YACvI,sCAAsC;YACtC,IAAI,4BAA4B,KAAK,SAAS,EAAE,CAAC;gBAC7C,OAAO,wBAAY,CAAA;YACvB,CAAC;YAED,iBAAiB,GAAG,IAAA,yBAAW,EAC3B;gBACI,WAAW,EAAE,uDAAuD;gBACpE,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,oEAAoE;gBACnG,IAAI,EAAE,iBAAiB,EAAE,0EAA0E;gBACnG,GAAG,EAAE,WAAW,IAAI,EAAE;gBACtB,EAAE,EAAE,IAAA,gCAAkB,GAAE;gBACxB,gBAAgB,EAAE,6BAA6B,EAAE,2DAA2D;gBAC5G,WAAW,EAAE,4BAA4B,EAAE,4FAA4F;aAC1I,EACD,IAAI,CAAC,OAAO,CACf,CAAA;YAED,IAAI,4BAA4B,KAAK,YAAY,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC;gBACpF,OAAO,wBAAY,CAAA;YACvB,CAAC;QACL,CAAC;QAED,IAAI,cAAc,GAA+B;YAC7C,WAAW;YACX,UAAU;SACb,CAAA;QAED,MAAM,oBAAoB,GAAG,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,EAAE,CAAA;QACjF,MAAM,mBAAmB,GAAG,MAAM,oBAAoB,CAAC,4BAA4B,CAC/E,YAAY,EACZ,MAAM,CAAC,QAAQ,EACf,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,OAAO,EACZ,KAAK,EACL,MAAM,CAAC,gBAAgB,EACvB,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAChC,CAAA;QAED,IAAI,mBAAmB,EAAE,KAAK,EAAE,CAAC;YAC7B,cAAc,CAAC,oBAAoB,GAAG,mBAAmB,CAAC,KAAK,CAAA;QACnE,CAAC;QAED,uEAAuE;QACvE,MAAM,cAAc,GAAG,IAAI,CAAC,wBAAwB,CAAC,iBAAiB,EAAE,CAAA;QACxE,IAAI,cAAc,IAAI,cAAc,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YACtD,sEAAsE;YACtE,sFAAsF;YACtF,8FAA8F;YAC9F,2BAA2B;YAC3B,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;gBAC9B,IAAI,CAAC,wBAAwB,CAAC,cAAc,CAAC,cAAc,CAAC,CAAA;gBAC5D,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;gBACpG,MAAM,IAAA,4CAAgC,EAClC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,gBAAgB,EACrB,cAAc,EACd,IAAI,CAAC,gCAAgC,EAAE,EACvC,CAAC,EACD,CAAC,EACD,EAAE,EACF,EAAE,EACF,YAAY,CACf,CAAA;YACL,CAAC;QACL,CAAC;QAED,MAAM,oBAAoB,GAAG,mBAAmB,EAAE,cAAc,CAAA;QAEhE,MAAM,UAAU,GAAG,IAAI,CAAC,wBAAwB,CAAC,aAAa,CAAC;YAC3D,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,YAAqC;YAC/E,cAAc,EAAE,cAAc;YAC9B,eAAe,EAAE,yBAAyB,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,SAAS;YACrF,gBAAgB,EAAE,iBAAiB;YACnC,gBAAgB,EAAE,iBAAiB,EAAE,gBAAgB;YACrD,mBAAmB,EAAE,iBAAiB,EAAE,mBAAmB;YAC3D,kBAAkB,EAAE,IAAI,CAAC,mBAAmB,CAAC,qBAAqB,EAAE,EAAE,EAAE,GAAG,EAAE,QAAQ,IAAI,SAAS;YAClG,oBAAoB,EAAE,oBAAoB;YAC1C,gBAAgB,EAAE,oBAAS,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,gBAAgB,CAAC;SACtF,CAAC,CAAA;QAEF,uCAAuC;QACvC,MAAM,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,CAAC,iBAAiB,CAAA;QACxF,IAAI,YAAY,EAAE,CAAC;YACf,cAAc,CAAC,WAAW,CAAC,eAAe;gBACtC,YAAY,GAAG,IAAI,GAAG,cAAc,CAAC,WAAW,CAAC,eAAe,CAAA;QACxE,CAAC;QAED,uDAAuD;QACvD,IAAI,qBAAqD,CAAA;QAEzD,IAAI,oBAAoB,YAAY,gDAAyB,EAAE,CAAC;YAC5D,MAAM,YAAY,GAAG,cAAiD,CAAA;YACtE,qBAAqB,GAAG;gBACpB,GAAG,YAAY;gBACf,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1C,CAAA;QACL,CAAC;aAAM,CAAC;YACJ,MAAM,UAAU,GAAG,cAA+C,CAAA;YAClE,qBAAqB,GAAG;gBACpB,GAAG,UAAU;aAChB,CAAA;QACL,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,oBAAoB,YAAY,gDAAyB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAA;YAC5F,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,oEAAoE,QAAQ,GAAG,CAAC,CAAA;YACnG,MAAM,kBAAkB,GAAG,MAAM,oBAAoB,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,CAAA;YAChG,OAAO,MAAM,IAAI,CAAC,yBAAyB,CAAC,kBAAkB,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,CAAC,CAAA;QACrG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,KAAc,EAAE,UAAU,CAAC,CAAA;QACnE,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,yBAAyB,CACnC,kBAA+C,EAC/C,OAA6B,EAC7B,YAAqB,EACrB,cAAsB;QAEtB,IAAI,CAAC,qBAAqB,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QAE5D,IAAI,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QAC/D,OAAO,CAAC,6BAA6B;YACjC,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,CAAC,6BAA6B,CAAA;QAE/E,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,UAAU,CAAA;QACtD,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,wBAAwB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;YACrD,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;YACpG,MAAM,IAAA,4CAAgC,EAClC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,gBAAgB,EACrB,OAAO,EACP,IAAI,CAAC,gCAAgC,EAAE,EACvC,CAAC,EACD,CAAC,EACD,EAAE,EACF,EAAE,EACF,YAAY,CACf,CAAA;QACL,CAAC;QAED,iIAAiI;QACjI,IACI,OAAO,CAAC,cAAc,KAAK,qCAAc,CAAC,IAAI,IAAI,mIAAmI;YACrL,CAAC,OAAO,CAAC,KAAK,KAAK,QAAQ,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,EAC7D,CAAC;YACC,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,yDAAyD;QACzD,IAAI,CAAC,wBAAwB,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;QAEtD,uDAAuD;QACvD,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,WAAW;YACtD,0BAA0B;aACzB,MAAM,CAAC,UAAU,CAAC,EAAE;YACjB,IAAI,UAAU,CAAC,OAAO,KAAK,EAAE,EAAE,CAAC;gBAC5B,OAAO,CAAC,kBAAkB,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;gBACtD,OAAO,KAAK,CAAA;YAChB,CAAC;YACD,OAAO,IAAI,CAAA;QACf,CAAC,CAAC;YACF,4BAA4B;aAC3B,MAAM,CAAC,UAAU,CAAC,EAAE;YACjB,oEAAoE;YACpE,sEAAsE;YACtE,wEAAwE;YACxE,oCAAoC;YACpC,MAAM,EAAE,oCAAoC,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,CAAA;YAC9F,IAAI,oCAAoC,EAAE,CAAC;gBACvC,OAAO,IAAI,CAAA;YACf,CAAC;YACD,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtE,OAAO,IAAI,CAAA;YACf,CAAC;YACD,wGAAwG;YACxG,OAAO,CAAC,kBAAkB,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;YACvD,OAAO,KAAK,CAAA;QAChB,CAAC,CAAC,CAAA;QAEN,MAAM,EAAE,6BAA6B,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,CAAA;QACvF,MAAM,2BAA2B,GAAG,IAAA,kDAAgC,EAChE,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC,gBAAgB,EACnD,mBAAmB,EACnB,6BAA6B,EAC7B,cAAc,CACjB,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE;YAClB,iHAAiH;YACjH,IAAI,UAAU,CAAC,UAAU,KAAK,EAAE,EAAE,CAAC;gBAC/B,OAAO,CAAC,kBAAkB,CAAC,UAAU,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;gBACxD,OAAO,KAAK,CAAA;YAChB,CAAC;YACD,OAAO,IAAI,CAAA;QACf,CAAC,CAAC,CAAA;QAEF,2BAA2B,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;YAC7C,MAAM,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,MAAM,CAAC,CAAA;YACtF,IAAI,gBAAgB;gBAAE,gBAAgB,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAA;QACxF,CAAC,CAAC,CAAA;QAEF,oEAAoE;QACpE,OAAO,CAAC,iCAAiC,CAAC,IAAI,CAAC,GAAG,2BAA2B,CAAC,CAAA;QAE9E,OAAO,CAAC,kCAAkC;YACtC,OAAO,CAAC,kCAAkC;gBAC1C,2BAA2B,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;oBACrD,OAAO,KAAK,GAAG,CAAC,UAAU,CAAC,0BAA0B,EAAE,MAAM,IAAI,CAAC,CAAC,CAAA;gBACvE,CAAC,EAAE,CAAC,CAAC,CAAA;QAET,gGAAgG;QAChG,yCAAyC;QACzC,IAAI,OAAO,CAAC,iCAAiC,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC;YAC1G,IAAI,CAAC,wBAAwB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAA;YACnD,MAAM,IAAA,4CAAgC,EAClC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,gBAAgB,EACrB,OAAO,EACP,IAAI,CAAC,gCAAgC,EAAE,CAC1C,CAAA;YACD,OAAO,wBAAY,CAAA;QACvB,CAAC;QAED,OAAO;YACH,KAAK,EAAE,2BAA2B;YAClC,SAAS,EAAE,OAAO,CAAC,EAAE;YACrB,kBAAkB,EAAE,kBAAkB,CAAC,eAAe,CAAC,SAAS;SACnE,CAAA;IACL,CAAC;IAEO,uBAAuB,CAAC,KAAY,EAAE,OAA6B;QACvE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,GAAG,KAAK,CAAC,CAAA;QAEpD,IAAA,wCAA4B,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,CAAA;QAE5D,6GAA6G;QAC7G,IAAI,CAAC,wBAAwB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAA;QAEnD,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,gCAAa,CACnB,gCAAa,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;AAvcD,0DAucC"}

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

@@ -0,0 +1,23 @@
+import { Logging, LogInlineCompletionSessionResultsParams, Telemetry } from '@aws/language-server-runtimes/server-interface';
+import { SessionManager } from '../session/sessionManager';
+import { CodePercentageTracker } from '../tracker/codePercentageTracker';
+import { RejectedEditTracker } from '../tracker/rejectedEditTracker';
+import { StreakTracker } from '../tracker/streakTracker';
+import { TelemetryService } from '../../../shared/telemetry/telemetryService';
+import { AcceptedInlineSuggestionEntry, CodeDiffTracker } from '../tracker/codeDiffTracker';
+export declare class SessionResultsHandler {
+    private readonly logging;
+    private readonly telemetry;
+    private readonly telemetryService;
+    private readonly completionSessionManager;
+    private readonly editSessionManager;
+    private readonly codePercentageTracker;
+    private readonly codeDiffTracker;
+    private readonly rejectedEditTracker;
+    private readonly streakTracker;
+    private readonly getEditsEnabled;
+    private readonly getTimeSinceLastUserModification;
+    constructor(logging: Logging, telemetry: Telemetry, telemetryService: TelemetryService, completionSessionManager: SessionManager, editSessionManager: SessionManager, codePercentageTracker: CodePercentageTracker, codeDiffTracker: CodeDiffTracker<AcceptedInlineSuggestionEntry>, rejectedEditTracker: RejectedEditTracker, streakTracker: StreakTracker, getEditsEnabled: () => boolean, getTimeSinceLastUserModification: () => number);
+    private enqueueCodeDiffEntry;
+    handleSessionResults(params: LogInlineCompletionSessionResultsParams): Promise<void>;
+}