@aws/lsp-codewhisperer 0.0.83 → 0.0.85

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

@@ -0,0 +1,379 @@
+"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
+                            ? {
+                                ...currentSession.requestContext.fileContext,
+                            }
+                            : undefined,
+                        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) {
+            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,+EAS6C;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,cAAc,CAAC,cAAc,CAAC,WAAW;4BAClD,CAAC,CAAC;gCACI,GAAG,cAAc,CAAC,cAAc,CAAC,WAAW;6BAC/C;4BACH,CAAC,CAAC,SAAS;wBACf,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,IAAI,cAAc,CAAC,WAAW,EAAE,CAAC;YAC7C,cAAc,CAAC,WAAW,CAAC,eAAe;gBACtC,YAAY,GAAG,IAAI,GAAG,cAAc,CAAC,WAAW,CAAC,eAAe,CAAA;QACxE,CAAC;QAED,uDAAuD;QACvD,IAAI,qBAAiD,CAAA;QAErD,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,EAAE,gBAAgB,IAAI,EAAE,EAC1D,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;AAzcD,0DAycC"}

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>;
+}

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

@@ -0,0 +1,124 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SessionResultsHandler = void 0;
+const codeWhispererService_1 = require("../../../shared/codeWhispererService");
+const diffUtils_1 = require("../utils/diffUtils");
+const utils_1 = require("../../../shared/utils");
+const telemetry_1 = require("../telemetry/telemetry");
+class SessionResultsHandler {
+    logging;
+    telemetry;
+    telemetryService;
+    completionSessionManager;
+    editSessionManager;
+    codePercentageTracker;
+    codeDiffTracker;
+    rejectedEditTracker;
+    streakTracker;
+    getEditsEnabled;
+    getTimeSinceLastUserModification;
+    constructor(logging, telemetry, telemetryService, completionSessionManager, editSessionManager, codePercentageTracker, codeDiffTracker, rejectedEditTracker, streakTracker, getEditsEnabled, getTimeSinceLastUserModification) {
+        this.logging = logging;
+        this.telemetry = telemetry;
+        this.telemetryService = telemetryService;
+        this.completionSessionManager = completionSessionManager;
+        this.editSessionManager = editSessionManager;
+        this.codePercentageTracker = codePercentageTracker;
+        this.codeDiffTracker = codeDiffTracker;
+        this.rejectedEditTracker = rejectedEditTracker;
+        this.streakTracker = streakTracker;
+        this.getEditsEnabled = getEditsEnabled;
+        this.getTimeSinceLastUserModification = getTimeSinceLastUserModification;
+    }
+    // Schedule tracker for UserModification Telemetry event
+    enqueueCodeDiffEntry(session, acceptedSuggestion, addedCharactersForEdit) {
+        const endPosition = (0, utils_1.getEndPositionForAcceptedSuggestion)(acceptedSuggestion.content, session.startPosition);
+        // use the addedCharactersForEdit if it is EDIT suggestion type
+        const originalString = addedCharactersForEdit ? addedCharactersForEdit : acceptedSuggestion.content;
+        this.codeDiffTracker.enqueue({
+            sessionId: session.codewhispererSessionId || '',
+            requestId: session.responseContext?.requestId || '',
+            fileUrl: session.document.uri,
+            languageId: session.language,
+            time: Date.now(),
+            originalString: originalString ?? '',
+            startPosition: session.startPosition,
+            endPosition: endPosition,
+            customizationArn: session.customizationArn,
+            completionType: (0, utils_1.getCompletionType)(acceptedSuggestion),
+            triggerType: session.triggerType,
+            credentialStartUrl: session.credentialStartUrl,
+        });
+    }
+    async handleSessionResults(params) {
+        const { sessionId, completionSessionResult, firstCompletionDisplayLatency, totalSessionDisplayTime, typeaheadLength, isInlineEdit, addedDiagnostics, removedDiagnostics, } = params;
+        // Comment this out because Edit request might return Completion as well so we can't rely on this flag
+        // const sessionManager = params.isInlineEdit ? this.editSessionManager : this.completionSessionManager
+        // TODO: Not elegant, worth refactoring
+        const editSession = this.editSessionManager.getSessionById(sessionId);
+        const completionSession = this.completionSessionManager.getSessionById(sessionId);
+        const session = editSession ?? completionSession;
+        const sessionManager = editSession ? this.editSessionManager : this.completionSessionManager;
+        if (!session) {
+            this.logging.log(`ERROR: Session ID ${sessionId} was not found`);
+            return;
+        }
+        if (session.state !== 'ACTIVE') {
+            this.logging.log(`ERROR: Trying to record trigger decision for not-active session ${sessionId} with wrong state ${session.state}`);
+            return;
+        }
+        const acceptedItemId = Object.keys(params.completionSessionResult).find(k => params.completionSessionResult[k].accepted);
+        const isAccepted = acceptedItemId ? true : false;
+        const acceptedSuggestion = session.suggestions.find(s => s.itemId === acceptedItemId);
+        let addedLengthForEdits = 0;
+        let deletedLengthForEdits = 0;
+        if (acceptedSuggestion) {
+            this.codePercentageTracker.countSuccess(session.language);
+            if (session.predictionType === codeWhispererService_1.SuggestionType.EDIT && acceptedSuggestion.content) {
+                // [acceptedSuggestion.insertText] will be undefined for NEP suggestion. Use [acceptedSuggestion.content] instead.
+                // Since [acceptedSuggestion.content] is in the form of a diff, transform the content into addedCharacters and deletedCharacters.
+                const { addedLines, deletedLines } = (0, diffUtils_1.getAddedAndDeletedLines)(acceptedSuggestion.content);
+                const charDiffResult = (0, diffUtils_1.getCharacterDifferences)(addedLines, deletedLines);
+                addedLengthForEdits = charDiffResult.charactersAdded;
+                deletedLengthForEdits = charDiffResult.charactersRemoved;
+                this.codePercentageTracker.countAcceptedTokensUsingCount(session.language, charDiffResult.charactersAdded);
+                this.codePercentageTracker.addTotalTokensForEdits(session.language, charDiffResult.charactersAdded);
+                this.enqueueCodeDiffEntry(session, acceptedSuggestion, addedLines.join('\n'));
+            }
+            else if (acceptedSuggestion.insertText) {
+                this.codePercentageTracker.countAcceptedTokens(session.language, acceptedSuggestion.insertText);
+                this.codePercentageTracker.countTotalTokens(session.language, acceptedSuggestion.insertText, true);
+                this.enqueueCodeDiffEntry(session, acceptedSuggestion);
+            }
+        }
+        // Handle rejected edit predictions
+        if (isInlineEdit && !isAccepted) {
+            // Find all rejected suggestions in this session
+            const rejectedSuggestions = session.suggestions.filter(suggestion => {
+                const result = completionSessionResult[suggestion.itemId];
+                return result && result.seen && !result.accepted;
+            });
+            // Record each rejected edit
+            for (const rejectedSuggestion of rejectedSuggestions) {
+                if (rejectedSuggestion.content) {
+                    this.rejectedEditTracker.recordRejectedEdit({
+                        content: rejectedSuggestion.content,
+                        timestamp: Date.now(),
+                        documentUri: session.document.uri,
+                        position: session.startPosition,
+                    });
+                    this.logging.debug(`[EDIT_PREDICTION] Recorded rejected edit: ${rejectedSuggestion.content.substring(0, 20)}...`);
+                }
+            }
+        }
+        session.setClientResultData(completionSessionResult, firstCompletionDisplayLatency, totalSessionDisplayTime, typeaheadLength);
+        if (firstCompletionDisplayLatency)
+            (0, telemetry_1.emitPerceivedLatencyTelemetry)(this.telemetry, session);
+        // Always emit user trigger decision at session close
+        sessionManager.closeSession(session);
+        const streakLength = this.getEditsEnabled() ? this.streakTracker.getAndUpdateStreakLength(isAccepted) : 0;
+        await (0, telemetry_1.emitUserTriggerDecisionTelemetry)(this.telemetry, this.telemetryService, session, this.getTimeSinceLastUserModification(), addedLengthForEdits, deletedLengthForEdits, addedDiagnostics, removedDiagnostics, streakLength, Object.keys(params.completionSessionResult)[0]);
+    }
+}
+exports.SessionResultsHandler = SessionResultsHandler;
+//# sourceMappingURL=sessionResultsHandler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"sessionResultsHandler.js","sourceRoot":"","sources":["../../../../src/language-server/inline-completion/handler/sessionResultsHandler.ts"],"names":[],"mappings":";;;AAYA,+EAAiF;AACjF,kDAAqF;AACrF,iDAA8F;AAC9F,sDAAwG;AAExG,MAAa,qBAAqB;IAET;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAXrB,YACqB,OAAgB,EAChB,SAAoB,EACpB,gBAAkC,EAClC,wBAAwC,EACxC,kBAAkC,EAClC,qBAA4C,EAC5C,eAA+D,EAC/D,mBAAwC,EACxC,aAA4B,EAC5B,eAA8B,EAC9B,gCAA8C;QAV9C,YAAO,GAAP,OAAO,CAAS;QAChB,cAAS,GAAT,SAAS,CAAW;QACpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,6BAAwB,GAAxB,wBAAwB,CAAgB;QACxC,uBAAkB,GAAlB,kBAAkB,CAAgB;QAClC,0BAAqB,GAArB,qBAAqB,CAAuB;QAC5C,oBAAe,GAAf,eAAe,CAAgD;QAC/D,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,kBAAa,GAAb,aAAa,CAAe;QAC5B,oBAAe,GAAf,eAAe,CAAe;QAC9B,qCAAgC,GAAhC,gCAAgC,CAAc;IAChE,CAAC;IAEJ,wDAAwD;IAChD,oBAAoB,CAAC,OAAY,EAAE,kBAA8B,EAAE,sBAA+B;QACtG,MAAM,WAAW,GAAG,IAAA,2CAAmC,EAAC,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,aAAa,CAAC,CAAA;QAC1G,+DAA+D;QAC/D,MAAM,cAAc,GAAG,sBAAsB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAA;QAEnG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;YACzB,SAAS,EAAE,OAAO,CAAC,sBAAsB,IAAI,EAAE;YAC/C,SAAS,EAAE,OAAO,CAAC,eAAe,EAAE,SAAS,IAAI,EAAE;YACnD,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG;YAC7B,UAAU,EAAE,OAAO,CAAC,QAAQ;YAC5B,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE;YAChB,cAAc,EAAE,cAAc,IAAI,EAAE;YACpC,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,WAAW,EAAE,WAAW;YACxB,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,cAAc,EAAE,IAAA,yBAAiB,EAAC,kBAAkB,CAAC;YACrD,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;SACjD,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,MAA+C;QACtE,MAAM,EACF,SAAS,EACT,uBAAuB,EACvB,6BAA6B,EAC7B,uBAAuB,EACvB,eAAe,EACf,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,GACrB,GAAG,MAAM,CAAA;QAEV,sGAAsG;QACtG,uGAAuG;QAEvG,uCAAuC;QACvC,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;QACrE,MAAM,iBAAiB,GAAG,IAAI,CAAC,wBAAwB,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;QAEjF,MAAM,OAAO,GAAG,WAAW,IAAI,iBAAiB,CAAA;QAChD,MAAM,cAAc,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAA;QAC5F,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,SAAS,gBAAgB,CAAC,CAAA;YAChE,OAAM;QACV,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CACZ,mEAAmE,SAAS,qBAAqB,OAAO,CAAC,KAAK,EAAE,CACnH,CAAA;YACD,OAAM;QACV,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,IAAI,CACnE,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,QAAQ,CAClD,CAAA;QACD,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAA;QAChD,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,cAAc,CAAC,CAAA;QACrF,IAAI,mBAAmB,GAAG,CAAC,CAAA;QAC3B,IAAI,qBAAqB,GAAG,CAAC,CAAA;QAE7B,IAAI,kBAAkB,EAAE,CAAC;YACrB,IAAI,CAAC,qBAAqB,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;YACzD,IAAI,OAAO,CAAC,cAAc,KAAK,qCAAc,CAAC,IAAI,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;gBAC/E,kHAAkH;gBAClH,iIAAiI;gBACjI,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,IAAA,mCAAuB,EAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;gBACxF,MAAM,cAAc,GAAG,IAAA,mCAAuB,EAAC,UAAU,EAAE,YAAY,CAAC,CAAA;gBACxE,mBAAmB,GAAG,cAAc,CAAC,eAAe,CAAA;gBACpD,qBAAqB,GAAG,cAAc,CAAC,iBAAiB,CAAA;gBAExD,IAAI,CAAC,qBAAqB,CAAC,6BAA6B,CACpD,OAAO,CAAC,QAAQ,EAChB,cAAc,CAAC,eAAe,CACjC,CAAA;gBACD,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,eAAe,CAAC,CAAA;gBACnG,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,kBAAkB,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;YACjF,CAAC;iBAAM,IAAI,kBAAkB,CAAC,UAAU,EAAE,CAAC;gBACvC,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,CAAC,OAAO,CAAC,QAAQ,EAAE,kBAAkB,CAAC,UAAU,CAAC,CAAA;gBAC/F,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,OAAO,CAAC,QAAQ,EAAE,kBAAkB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAA;gBAClG,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAA;YAC1D,CAAC;QACL,CAAC;QAED,mCAAmC;QACnC,IAAI,YAAY,IAAI,CAAC,UAAU,EAAE,CAAC;YAC9B,gDAAgD;YAChD,MAAM,mBAAmB,GAAG,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE;gBAChE,MAAM,MAAM,GAAG,uBAAuB,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;gBACzD,OAAO,MAAM,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAA;YACpD,CAAC,CAAC,CAAA;YAEF,4BAA4B;YAC5B,KAAK,MAAM,kBAAkB,IAAI,mBAAmB,EAAE,CAAC;gBACnD,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;oBAC7B,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBACxC,OAAO,EAAE,kBAAkB,CAAC,OAAO;wBACnC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;wBACrB,WAAW,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG;wBACjC,QAAQ,EAAE,OAAO,CAAC,aAAa;qBAClC,CAAC,CAAA;oBAEF,IAAI,CAAC,OAAO,CAAC,KAAK,CACd,6CAA6C,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAChG,CAAA;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,CAAC,mBAAmB,CACvB,uBAAuB,EACvB,6BAA6B,EAC7B,uBAAuB,EACvB,eAAe,CAClB,CAAA;QAED,IAAI,6BAA6B;YAAE,IAAA,yCAA6B,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;QAEzF,qDAAqD;QACrD,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,CAAA;QACpC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACzG,MAAM,IAAA,4CAAgC,EAClC,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,gBAAgB,EACrB,OAAO,EACP,IAAI,CAAC,gCAAgC,EAAE,EACvC,mBAAmB,EACnB,qBAAqB,EACrB,gBAAmC,EACnC,kBAAqC,EACrC,YAAY,EACZ,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,CACjD,CAAA;IACL,CAAC;CACJ;AAvJD,sDAuJC"}

package/out/language-server/inline-completion/telemetry/telemetry.d.ts

@@ -1,9 +1,10 @@
-import { Telemetry, IdeDiagnostic } from '@aws/language-server-runtimes/server-interface';
-import { AWSError } from 'aws-sdk';
+import { Telemetry } from '@aws/language-server-runtimes/server-interface';
+import { IdeDiagnostic } from '@amzn/codewhisperer-runtime';
+import { ServiceException } from '@smithy/smithy-client';
 import { CodeWhispererSession, UserTriggerDecision } from '../session/sessionManager';
 import { TelemetryService } from '../../../shared/telemetry/telemetryService';
-export declare const emitServiceInvocationTelemetry: (telemetry: Telemetry, session: CodeWhispererSession, requestId: string) => void;
-export declare const emitServiceInvocationFailure: (telemetry: Telemetry, session: CodeWhispererSession, error: Error | AWSError) => void;
+export declare const emitServiceInvocationTelemetry: (telemetry: Telemetry, session: CodeWhispererSession, requestId: string | undefined) => void;
+export declare const emitServiceInvocationFailure: (telemetry: Telemetry, session: CodeWhispererSession, error: Error | ServiceException) => void;
 export declare const emitPerceivedLatencyTelemetry: (telemetry: Telemetry, session: CodeWhispererSession) => void;
 export declare function emitEmptyUserTriggerDecisionTelemetry(telemetryService: TelemetryService, session: CodeWhispererSession, timeSinceLastUserModification?: number, streakLength?: number): Promise<void>;
 export declare const emitUserTriggerDecisionTelemetry: (telemetry: Telemetry, telemetryService: TelemetryService, session: CodeWhispererSession, timeSinceLastUserModification?: number, addedCharsCountForEditSuggestion?: number, deletedCharsCountForEditSuggestion?: number, addedIdeDiagnostics?: IdeDiagnostic[], removedIdeDiagnostics?: IdeDiagnostic[], streakLength?: number, itemId?: string) => Promise<void>;

package/out/language-server/inline-completion/telemetry/telemetry.js

@@ -38,7 +38,7 @@ const emitServiceInvocationTelemetry = (telemetry, session, requestId) => {
 exports.emitServiceInvocationTelemetry = emitServiceInvocationTelemetry;
 const emitServiceInvocationFailure = (telemetry, session, error) => {
     const duration = new Date().getTime() - session.startTime;
-    const codewhispererRequestId = (0, utils_1.isAwsError)(error) ? error.requestId : undefined;
+    const codewhispererRequestId = (0, utils_1.isServiceException)(error) ? error.$metadata.requestId : undefined;
     const data = {
         codewhispererRequestId: codewhispererRequestId,
         codewhispererSessionId: undefined,
@@ -66,8 +66,8 @@ const emitServiceInvocationFailure = (telemetry, session, error) => {
         data,
         errorData: {
             reason: error.name || 'UnknownError',
-            errorCode: (0, utils_1.isAwsError)(error) ? error.code : undefined,
-            httpStatusCode: (0, utils_1.isAwsError)(error) ? error.statusCode : undefined,
+            errorCode: (0, utils_1.getErrorId)(error),
+            httpStatusCode: (0, utils_1.isServiceException)(error) ? error.$metadata.httpStatusCode : undefined,
         },
     });
 };

package/out/language-server/inline-completion/telemetry/telemetry.js.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../../../src/language-server/inline-completion/telemetry/telemetry.ts"],"names":[],"mappings":";;;AA+GA,sFA4BC;AApID,iDAAqE;AAErE,+EAAqE;AAE9D,MAAM,8BAA8B,GAAG,CAC1C,SAAoB,EACpB,OAA6B,EAC7B,SAAiB,EACnB,EAAE;IACA,MAAM,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAA;IACzD,MAAM,IAAI,GAAwC;QAC9C,sBAAsB,EAAE,SAAS;QACjC,sBAAsB,EAAE,OAAO,CAAC,eAAe,EAAE,sBAAsB;QACvE,gCAAgC,EAAE,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;QAChE,2BAA2B,EACvB,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAA,yBAAiB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;QAC1F,wBAAwB,EAAE,OAAO,CAAC,WAAW;QAC7C,iCAAiC,EAAE,OAAO,CAAC,eAAe;QAC1D,QAAQ;QACR,uBAAuB,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI;QACnD,yBAAyB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;QAC1D,qBAAqB,EAAE,OAAO,CAAC,QAAQ;QACvC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,gBAAgB;QACvF,qCAAqC,EAAE,OAAO,CAAC,oBAAoB,EAAE,KAAK;QAC1E,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,OAAO;QAC9E,sCAAsC,EAAE,OAAO,CAAC,oBAAoB,EAAE,cAAc;QACpF,6BAA6B,EAAE,OAAO,CAAC,gBAAgB;QACvD,MAAM,EAAE,WAAW;QACnB,wCAAwC,EAAE,OAAO,CAAC,6BAA6B;KAClF,CAAA;IACD,SAAS,CAAC,UAAU,CAAC;QACjB,IAAI,EAAE,iCAAiC;QACvC,MAAM,EAAE,WAAW;QACnB,IAAI,EAAE;YACF,GAAG,IAAI;YACP,wCAAwC,EAAE,OAAO,CAAC,6BAA6B;SAClF;KACJ,CAAC,CAAA;AACN,CAAC,CAAA;AAnCY,QAAA,8BAA8B,kCAmC1C;AAEM,MAAM,4BAA4B,GAAG,CACxC,SAAoB,EACpB,OAA6B,EAC7B,KAAuB,EACzB,EAAE;IACA,MAAM,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAA;IACzD,MAAM,sBAAsB,GAAG,IAAA,kBAAU,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAA;IAE9E,MAAM,IAAI,GAAwC;QAC9C,sBAAsB,EAAE,sBAAsB;QAC9C,sBAAsB,EAAE,SAAS;QACjC,gCAAgC,EAAE,CAAC,CAAC;QACpC,wBAAwB,EAAE,OAAO,CAAC,WAAW;QAC7C,iCAAiC,EAAE,OAAO,CAAC,eAAe;QAC1D,MAAM,EAAE,uCAAuC,KAAK,CAAC,IAAI,IAAI,cAAc,EAAE;QAC7E,QAAQ;QACR,uBAAuB,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI;QACnD,yBAAyB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;QAC1D,qBAAqB,EAAE,OAAO,CAAC,QAAQ;QACvC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,gBAAgB;QACvF,qCAAqC,EAAE,OAAO,CAAC,oBAAoB,EAAE,KAAK;QAC1E,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,OAAO;QAC9E,sCAAsC,EAAE,OAAO,CAAC,oBAAoB,EAAE,cAAc;QACpF,6BAA6B,EAAE,OAAO,CAAC,gBAAgB;QACvD,wCAAwC,EAAE,OAAO,CAAC,6BAA6B;QAC/E,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,QAAQ;KACpB,CAAA;IAED,SAAS,CAAC,UAAU,CAAC;QACjB,IAAI,EAAE,iCAAiC;QACvC,MAAM,EAAE,QAAQ;QAChB,IAAI;QACJ,SAAS,EAAE;YACP,MAAM,EAAE,KAAK,CAAC,IAAI,IAAI,cAAc;YACpC,SAAS,EAAE,IAAA,kBAAU,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YACrD,cAAc,EAAE,IAAA,kBAAU,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;SACnE;KACJ,CAAC,CAAA;AACN,CAAC,CAAA;AAxCY,QAAA,4BAA4B,gCAwCxC;AAEM,MAAM,6BAA6B,GAAG,CAAC,SAAoB,EAAE,OAA6B,EAAE,EAAE;IACjG,MAAM,IAAI,GAAuC;QAC7C,sBAAsB,EAAE,OAAO,CAAC,eAAe,EAAE,SAAS;QAC1D,sBAAsB,EAAE,OAAO,CAAC,eAAe,EAAE,sBAAsB;QACvE,2BAA2B,EACvB,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAA,yBAAiB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;QAC1F,wBAAwB,EAAE,OAAO,CAAC,WAAW;QAC7C,QAAQ,EAAE,OAAO,CAAC,6BAA6B;QAC/C,qBAAqB,EAAE,OAAO,CAAC,QAAQ;QACvC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,6BAA6B,EAAE,OAAO,CAAC,gBAAgB;QACvD,MAAM,EAAE,WAAW;QACnB,OAAO,EAAE,IAAI;KAChB,CAAA;IAED,SAAS,CAAC,UAAU,CAAC;QACjB,IAAI,EAAE,gCAAgC;QACtC,IAAI;KACP,CAAC,CAAA;AACN,CAAC,CAAA;AAnBY,QAAA,6BAA6B,iCAmBzC;AAEM,KAAK,UAAU,qCAAqC,CACvD,gBAAkC,EAClC,OAA6B,EAC7B,6BAAsC,EACtC,YAAqB;IAErB,yDAAyD;IACzD,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC/B,OAAM;IACV,CAAC;IAED,eAAe;IACf,IAAA,kDAA0C,EACtC,gBAAgB,EAChB,OAAO,EACP,OAAO,EACP,6BAA6B,EAC7B,CAAC,EACD,CAAC,EACD,EAAE,EACF,EAAE,EACF,YAAY,CACf;SACI,IAAI,EAAE;SACN,KAAK,CAAC,CAAC,CAAC,EAAE,GAAE,CAAC,CAAC;SACd,OAAO,CAAC,GAAG,EAAE;QACV,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAA;IACvC,CAAC,CAAC,CAAA;AACV,CAAC;AAEM,MAAM,gCAAgC,GAAG,KAAK,EACjD,SAAoB,EACpB,gBAAkC,EAClC,OAA6B,EAC7B,6BAAsC,EACtC,gCAAyC,EACzC,kCAA2C,EAC3C,mBAAqC,EACrC,qBAAuC,EACvC,YAAqB,EACrB,MAAe,EACjB,EAAE;IACA,yDAAyD;IACzD,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC/B,OAAM;IACV,CAAC;IAED,wFAAwF;IACxF,0EAA0E;IAC1E,MAAM,mBAAmB,GACrB,OAAO,CAAC,cAAc,KAAK,qCAAc,CAAC,IAAI;QAC1C,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,MAAM,CAAC;QACxC,CAAC,CAAC,OAAO,CAAC,gCAAgC,EAAE,CAAA;IAEpD,8EAA8E;IAC9E,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACvB,OAAM;IACV,CAAC;IAED,MAAM,IAAA,kDAA0C,EAC5C,gBAAgB,EAChB,OAAO,EACP,mBAAmB,EACnB,6BAA6B,EAC7B,gCAAgC,EAChC,kCAAkC,EAClC,mBAAmB,EACnB,qBAAqB,EACrB,YAAY,CACf,CAAA;IAED,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAA;AACvC,CAAC,CAAA;AA1CY,QAAA,gCAAgC,oCA0C5C;AAEM,MAAM,0CAA0C,GAAG,CACtD,gBAAkC,EAClC,OAA6B,EAC7B,mBAAwC,EACxC,6BAAsC,EACtC,gCAAyC,EACzC,kCAA2C,EAC3C,mBAAqC,EACrC,qBAAuC,EACvC,YAAqB,EACvB,EAAE;IACA,OAAO,gBAAgB,CAAC,uBAAuB,CAC3C,OAAO,EACP,mBAAmB,EACnB,6BAA6B,EAC7B,gCAAgC,EAChC,kCAAkC,EAClC,mBAAmB,EACnB,qBAAqB,EACrB,YAAY,CACf,CAAA;AACL,CAAC,CAAA;AArBY,QAAA,0CAA0C,8CAqBtD"}
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../../../src/language-server/inline-completion/telemetry/telemetry.ts"],"names":[],"mappings":";;;AAgHA,sFA4BC;AApID,iDAAyF;AAEzF,+EAAqE;AAE9D,MAAM,8BAA8B,GAAG,CAC1C,SAAoB,EACpB,OAA6B,EAC7B,SAA6B,EAC/B,EAAE;IACA,MAAM,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAA;IACzD,MAAM,IAAI,GAAwC;QAC9C,sBAAsB,EAAE,SAAS;QACjC,sBAAsB,EAAE,OAAO,CAAC,eAAe,EAAE,sBAAsB;QACvE,gCAAgC,EAAE,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;QAChE,2BAA2B,EACvB,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAA,yBAAiB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;QAC1F,wBAAwB,EAAE,OAAO,CAAC,WAAW;QAC7C,iCAAiC,EAAE,OAAO,CAAC,eAAe;QAC1D,QAAQ;QACR,uBAAuB,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI;QACnD,yBAAyB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;QAC1D,qBAAqB,EAAE,OAAO,CAAC,QAAQ;QACvC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,gBAAgB;QACvF,qCAAqC,EAAE,OAAO,CAAC,oBAAoB,EAAE,KAAK;QAC1E,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,OAAO;QAC9E,sCAAsC,EAAE,OAAO,CAAC,oBAAoB,EAAE,cAAc;QACpF,6BAA6B,EAAE,OAAO,CAAC,gBAAgB;QACvD,MAAM,EAAE,WAAW;QACnB,wCAAwC,EAAE,OAAO,CAAC,6BAA6B;KAClF,CAAA;IACD,SAAS,CAAC,UAAU,CAAC;QACjB,IAAI,EAAE,iCAAiC;QACvC,MAAM,EAAE,WAAW;QACnB,IAAI,EAAE;YACF,GAAG,IAAI;YACP,wCAAwC,EAAE,OAAO,CAAC,6BAA6B;SAClF;KACJ,CAAC,CAAA;AACN,CAAC,CAAA;AAnCY,QAAA,8BAA8B,kCAmC1C;AAEM,MAAM,4BAA4B,GAAG,CACxC,SAAoB,EACpB,OAA6B,EAC7B,KAA+B,EACjC,EAAE;IACA,MAAM,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAA;IACzD,MAAM,sBAAsB,GAAG,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAA;IAEhG,MAAM,IAAI,GAAwC;QAC9C,sBAAsB,EAAE,sBAAsB;QAC9C,sBAAsB,EAAE,SAAS;QACjC,gCAAgC,EAAE,CAAC,CAAC;QACpC,wBAAwB,EAAE,OAAO,CAAC,WAAW;QAC7C,iCAAiC,EAAE,OAAO,CAAC,eAAe;QAC1D,MAAM,EAAE,uCAAuC,KAAK,CAAC,IAAI,IAAI,cAAc,EAAE;QAC7E,QAAQ;QACR,uBAAuB,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI;QACnD,yBAAyB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;QAC1D,qBAAqB,EAAE,OAAO,CAAC,QAAQ;QACvC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,gBAAgB;QACvF,qCAAqC,EAAE,OAAO,CAAC,oBAAoB,EAAE,KAAK;QAC1E,uCAAuC,EAAE,OAAO,CAAC,oBAAoB,EAAE,OAAO;QAC9E,sCAAsC,EAAE,OAAO,CAAC,oBAAoB,EAAE,cAAc;QACpF,6BAA6B,EAAE,OAAO,CAAC,gBAAgB;QACvD,wCAAwC,EAAE,OAAO,CAAC,6BAA6B;QAC/E,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,QAAQ;KACpB,CAAA;IAED,SAAS,CAAC,UAAU,CAAC;QACjB,IAAI,EAAE,iCAAiC;QACvC,MAAM,EAAE,QAAQ;QAChB,IAAI;QACJ,SAAS,EAAE;YACP,MAAM,EAAE,KAAK,CAAC,IAAI,IAAI,cAAc;YACpC,SAAS,EAAE,IAAA,kBAAU,EAAC,KAAK,CAAC;YAC5B,cAAc,EAAE,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;SACzF;KACJ,CAAC,CAAA;AACN,CAAC,CAAA;AAxCY,QAAA,4BAA4B,gCAwCxC;AAEM,MAAM,6BAA6B,GAAG,CAAC,SAAoB,EAAE,OAA6B,EAAE,EAAE;IACjG,MAAM,IAAI,GAAuC;QAC7C,sBAAsB,EAAE,OAAO,CAAC,eAAe,EAAE,SAAS;QAC1D,sBAAsB,EAAE,OAAO,CAAC,eAAe,EAAE,sBAAsB;QACvE,2BAA2B,EACvB,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAA,yBAAiB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;QAC1F,wBAAwB,EAAE,OAAO,CAAC,WAAW;QAC7C,QAAQ,EAAE,OAAO,CAAC,6BAA6B;QAC/C,qBAAqB,EAAE,OAAO,CAAC,QAAQ;QACvC,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,6BAA6B,EAAE,OAAO,CAAC,gBAAgB;QACvD,MAAM,EAAE,WAAW;QACnB,OAAO,EAAE,IAAI;KAChB,CAAA;IAED,SAAS,CAAC,UAAU,CAAC;QACjB,IAAI,EAAE,gCAAgC;QACtC,IAAI;KACP,CAAC,CAAA;AACN,CAAC,CAAA;AAnBY,QAAA,6BAA6B,iCAmBzC;AAEM,KAAK,UAAU,qCAAqC,CACvD,gBAAkC,EAClC,OAA6B,EAC7B,6BAAsC,EACtC,YAAqB;IAErB,yDAAyD;IACzD,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC/B,OAAM;IACV,CAAC;IAED,eAAe;IACf,IAAA,kDAA0C,EACtC,gBAAgB,EAChB,OAAO,EACP,OAAO,EACP,6BAA6B,EAC7B,CAAC,EACD,CAAC,EACD,EAAE,EACF,EAAE,EACF,YAAY,CACf;SACI,IAAI,EAAE;SACN,KAAK,CAAC,CAAC,CAAC,EAAE,GAAE,CAAC,CAAC;SACd,OAAO,CAAC,GAAG,EAAE;QACV,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAA;IACvC,CAAC,CAAC,CAAA;AACV,CAAC;AAEM,MAAM,gCAAgC,GAAG,KAAK,EACjD,SAAoB,EACpB,gBAAkC,EAClC,OAA6B,EAC7B,6BAAsC,EACtC,gCAAyC,EACzC,kCAA2C,EAC3C,mBAAqC,EACrC,qBAAuC,EACvC,YAAqB,EACrB,MAAe,EACjB,EAAE;IACA,yDAAyD;IACzD,IAAI,OAAO,CAAC,oBAAoB,EAAE,CAAC;QAC/B,OAAM;IACV,CAAC;IAED,wFAAwF;IACxF,0EAA0E;IAC1E,MAAM,mBAAmB,GACrB,OAAO,CAAC,cAAc,KAAK,qCAAc,CAAC,IAAI;QAC1C,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,MAAM,CAAC;QACxC,CAAC,CAAC,OAAO,CAAC,gCAAgC,EAAE,CAAA;IAEpD,8EAA8E;IAC9E,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACvB,OAAM;IACV,CAAC;IAED,MAAM,IAAA,kDAA0C,EAC5C,gBAAgB,EAChB,OAAO,EACP,mBAAmB,EACnB,6BAA6B,EAC7B,gCAAgC,EAChC,kCAAkC,EAClC,mBAAmB,EACnB,qBAAqB,EACrB,YAAY,CACf,CAAA;IAED,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAA;AACvC,CAAC,CAAA;AA1CY,QAAA,gCAAgC,oCA0C5C;AAEM,MAAM,0CAA0C,GAAG,CACtD,gBAAkC,EAClC,OAA6B,EAC7B,mBAAwC,EACxC,6BAAsC,EACtC,gCAAyC,EACzC,kCAA2C,EAC3C,mBAAqC,EACrC,qBAAuC,EACvC,YAAqB,EACvB,EAAE;IACA,OAAO,gBAAgB,CAAC,uBAAuB,CAC3C,OAAO,EACP,mBAAmB,EACnB,6BAA6B,EAC7B,gCAAgC,EAChC,kCAAkC,EAClC,mBAAmB,EACnB,qBAAqB,EACrB,YAAY,CACf,CAAA;AACL,CAAC,CAAA;AArBY,QAAA,0CAA0C,8CAqBtD"}