@aws/lsp-codewhisperer 0.0.81 → 0.0.83

package/CHANGELOG.md

@@ -1,5 +1,48 @@
 # Changelog
 
+## [0.0.83](https://github.com/aws/language-servers/compare/lsp-codewhisperer/v0.0.82...lsp-codewhisperer/v0.0.83) (2025-10-01)
+
+
+### Bug Fixes
+
+* **amazonq:** escaping user input to mitigate xss issue ([#2360](https://github.com/aws/language-servers/issues/2360)) ([45b86be](https://github.com/aws/language-servers/commit/45b86bef1a93cf9ced6fbf0c222cf5410de04c81))
+* **amazonq:** fix to add opt-out header to streaming client ([#2365](https://github.com/aws/language-servers/issues/2365)) ([692e77b](https://github.com/aws/language-servers/commit/692e77bc99770ac7d676928e95e3dc43bb91e7f0))
+* **amazonq:** handle IAM credentials expiration field to be aws sdk versions compatible and add refresh logic to codewhisperer IAM client ([#2349](https://github.com/aws/language-servers/issues/2349)) ([5eb3768](https://github.com/aws/language-servers/commit/5eb3768bf020d61d0ade767d62e13839048146e4))
+* **amazonq:** send full finding details to plugin, partial to agent ([#2356](https://github.com/aws/language-servers/issues/2356)) ([961e6ca](https://github.com/aws/language-servers/commit/961e6ca11b122481685f9f65b3da14c6a2497cc4))
+* improve history management ([#2312](https://github.com/aws/language-servers/issues/2312)) ([#2357](https://github.com/aws/language-servers/issues/2357)) ([e7aa2a6](https://github.com/aws/language-servers/commit/e7aa2a6545bcb1a8238abfde69a05432be0b6615))
+* optimize memory bank token usage and add new tab support ([#2366](https://github.com/aws/language-servers/issues/2366)) ([3057d56](https://github.com/aws/language-servers/commit/3057d56e4a3047d1715d6e3560e9f934d1de469c))
+* private package mapping during artifact generation ([#2348](https://github.com/aws/language-servers/issues/2348)) ([d56bfa1](https://github.com/aws/language-servers/commit/d56bfa191954fac8068e2bf390c2d0b88ef8b168))
+* trim new line when emitting error message ([#2359](https://github.com/aws/language-servers/issues/2359)) ([d8733a7](https://github.com/aws/language-servers/commit/d8733a75487f74815302b838802eccbf3ffec55e))
+
+
+### Reverts
+
+* fix to add opt-out header to streaming client ([#2365](https://github.com/aws/language-servers/issues/2365)) ([#2370](https://github.com/aws/language-servers/issues/2370)) ([b29478f](https://github.com/aws/language-servers/commit/b29478fa1ecc58e331ff330ff79f46b0d8c38d9e))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aws/lsp-core bumped from ^0.0.15 to ^0.0.16
+
+## [0.0.82](https://github.com/aws/language-servers/compare/lsp-codewhisperer/v0.0.81...lsp-codewhisperer/v0.0.82) (2025-09-24)
+
+
+### Features
+
+* memory bank support ([#2314](https://github.com/aws/language-servers/issues/2314)) ([0e215fc](https://github.com/aws/language-servers/commit/0e215fc0e475b4c40a8237492371716982d4d532))
+
+
+### Bug Fixes
+
+* **amazonq:** fix to emit event for same region profile switch ([#2320](https://github.com/aws/language-servers/issues/2320)) ([aa1a482](https://github.com/aws/language-servers/commit/aa1a4827871a1cfa9fcd76f7ba420107a5d44b01))
+* **amazonq:** reduce number of findings to 30 as a quick fix ([#2318](https://github.com/aws/language-servers/issues/2318)) ([b31cf67](https://github.com/aws/language-servers/commit/b31cf67ddc68a2ca2e0a4ebd9ee94d0545afc656))
+* **amazonq:** removing a bracket from full review message ([#2317](https://github.com/aws/language-servers/issues/2317)) ([6d321ac](https://github.com/aws/language-servers/commit/6d321ac6f318c27b01f9f97eee45a62798a60cf5))
+* emit error code on failed user messages ([#2322](https://github.com/aws/language-servers/issues/2322)) ([a949ac0](https://github.com/aws/language-servers/commit/a949ac0a9d7a4dbce5fb7c8480952cee0a674b55))
+* inline latency telemetry should account for preprocess time ([#2323](https://github.com/aws/language-servers/issues/2323)) ([68c6d14](https://github.com/aws/language-servers/commit/68c6d1465a3325612052740496cc1e6e50f56b9a))
+* userTriggerDecision STE suggestionType validation error ([#2313](https://github.com/aws/language-servers/issues/2313)) ([8f30ac0](https://github.com/aws/language-servers/commit/8f30ac0ec5f4f7b7c343f5e889aec64a282897ea))
+
 ## [0.0.81](https://github.com/aws/language-servers/compare/lsp-codewhisperer/v0.0.80...lsp-codewhisperer/v0.0.81) (2025-09-19)
 
 

package/out/language-server/agenticChat/agenticChatController.js

@@ -66,8 +66,8 @@ const activeUserTracker_1 = require("../../shared/activeUserTracker");
 const displayFindings_1 = require("./tools/qCodeAnalysis/displayFindings");
 const constants_3 = require("../../shared/constants");
 const IdleWorkspaceManager_1 = require("../workspaceContext/IdleWorkspaceManager");
-const escapeHTML = require("escape-html");
 const semanticSearch_1 = require("./tools/workspaceContext/semanticSearch");
+const memoryBankController_1 = require("./context/memorybank/memoryBankController");
 class AgenticChatController {
     #features;
     #chatSessionManagementService;
@@ -80,6 +80,7 @@ class AgenticChatController {
     #chatHistoryDb;
     #additionalContextProvider;
     #contextCommandsProvider;
+    #memoryBankController;
     #stoppedToolUses = new Set();
     #userWrittenCodeTracker;
     #toolUseStartTimes = {};
@@ -154,6 +155,7 @@ class AgenticChatController {
         this.#mcpEventHandler = new mcpEventHandler_1.McpEventHandler(features, telemetryService);
         this.#origin = (0, utils_2.getOriginFromClientInfo)((0, utils_2.getClientName)(this.#features.lsp.getClientInitializeParams()));
         this.#activeUserTracker = activeUserTracker_1.ActiveUserTracker.getInstance(this.#features);
+        this.#memoryBankController = memoryBankController_1.MemoryBankController.getInstance(features);
     }
     async onExecuteCommand(params, _token) {
         this.#log(`onExecuteCommand: ${params.command}`);
@@ -562,17 +564,94 @@ class AgenticChatController {
     }
     async onChatPrompt(params, token) {
         // Phase 1: Initial Setup - This happens only once
-        params.prompt.prompt = (0, utils_2.sanitizeInput)(params.prompt.prompt || '');
+        params.prompt.prompt = (0, utils_2.sanitizeInput)(params.prompt.prompt || '', true);
         IdleWorkspaceManager_1.IdleWorkspaceManager.recordActivityTimestamp();
-        const maybeDefaultResponse = !params.prompt.command && (0, utils_1.getDefaultChatResponse)(params.prompt.prompt);
-        if (maybeDefaultResponse) {
-            return maybeDefaultResponse;
-        }
         const sessionResult = this.#chatSessionManagementService.getSession(params.tabId);
         const { data: session, success } = sessionResult;
         if (!success) {
             return new server_interface_1.ResponseError(protocol_1.ErrorCodes.InternalError, sessionResult.error);
         }
+        // Memory Bank Creation Flow - Delegate to MemoryBankController
+        if (this.#memoryBankController.isMemoryBankCreationRequest(params.prompt.prompt)) {
+            this.#features.logging.info(`Memory Bank creation request detected for tabId: ${params.tabId}`);
+            session.isMemoryBankGeneration = true;
+            // Store original prompt to prevent data loss on failure
+            const originalPrompt = params.prompt.prompt;
+            try {
+                const workspaceFolders = lsp_core_1.workspaceUtils.getWorkspaceFolderPaths(this.#features.workspace);
+                const workspaceUri = workspaceFolders.length > 0 ? workspaceFolders[0] : '';
+                if (!workspaceUri) {
+                    throw new Error('No workspace folder found for Memory Bank creation');
+                }
+                // Check if memory bank already exists to provide appropriate user feedback
+                const memoryBankExists = await this.#memoryBankController.memoryBankExists(workspaceUri);
+                const actionType = memoryBankExists ? 'Regenerating' : 'Generating';
+                this.#features.logging.info(`${actionType} Memory Bank for workspace: ${workspaceUri}`);
+                const resultStream = this.#getChatResultStream(params.partialResultToken);
+                await resultStream.writeResultBlock({
+                    body: `Preparing to analyze your project...`,
+                    type: 'answer',
+                    messageId: crypto.randomUUID(),
+                });
+                const comprehensivePrompt = await this.#memoryBankController.prepareComprehensiveMemoryBankPrompt(workspaceUri, async (prompt) => {
+                    // Direct LLM call for ranking - no agentic loop
+                    try {
+                        if (!this.#serviceManager) {
+                            throw new Error('amazonQServiceManager is not initialized');
+                        }
+                        const client = this.#serviceManager.getStreamingClient();
+                        const requestInput = {
+                            conversationState: {
+                                chatTriggerType: codewhisperer_streaming_1.ChatTriggerType.MANUAL,
+                                currentMessage: {
+                                    userInputMessage: {
+                                        content: prompt,
+                                    },
+                                },
+                            },
+                        };
+                        const response = await client.sendMessage(requestInput);
+                        let responseContent = '';
+                        const maxResponseSize = 50000; // 50KB limit
+                        if (response.sendMessageResponse) {
+                            for await (const chatEvent of response.sendMessageResponse) {
+                                if (chatEvent.assistantResponseEvent?.content) {
+                                    responseContent += chatEvent.assistantResponseEvent.content;
+                                    if (responseContent.length > maxResponseSize) {
+                                        this.#features.logging.warn('LLM response exceeded size limit, truncating');
+                                        break;
+                                    }
+                                }
+                            }
+                        }
+                        return responseContent.trim();
+                    }
+                    catch (error) {
+                        this.#features.logging.error(`Memory Bank LLM ranking failed: ${error}`);
+                        return ''; // Empty string triggers TF-IDF fallback
+                    }
+                });
+                // Only update prompt if we got a valid comprehensive prompt
+                if (comprehensivePrompt && comprehensivePrompt.trim().length > 0) {
+                    params.prompt.prompt = comprehensivePrompt;
+                }
+                else {
+                    this.#features.logging.warn('Empty comprehensive prompt received, using original prompt');
+                    params.prompt.prompt = originalPrompt;
+                }
+            }
+            catch (error) {
+                this.#features.logging.error(`Memory Bank preparation failed: ${error}`);
+                // Restore original prompt to ensure no data loss
+                params.prompt.prompt = originalPrompt;
+                // Reset memory bank flag since preparation failed
+                session.isMemoryBankGeneration = false;
+            }
+        }
+        const maybeDefaultResponse = !params.prompt.command && (0, utils_1.getDefaultChatResponse)(params.prompt.prompt);
+        if (maybeDefaultResponse) {
+            return maybeDefaultResponse;
+        }
         const compactIds = session.getAllDeferredCompactMessageIds();
         await this.#invalidateCompactCommand(params.tabId, compactIds);
         session.rejectAllDeferredToolExecutions(new toolShared_1.ToolApprovalException('Command ignored: new prompt', false));
@@ -619,7 +698,7 @@ class AgenticChatController {
                 await this.#telemetryController.emitAddMessageMetric(params.tabId, metric.metric, 'Cancelled');
             });
             session.setConversationType('AgenticChat');
-            const additionalContext = await this.#additionalContextProvider.getAdditionalContext(triggerContext, params.tabId, params.context);
+            const additionalContext = await this.#additionalContextProvider.getAdditionalContext(triggerContext, params.tabId, params.context, params.prompt.prompt);
             // Add active file to context list if it's not already there
             const activeFile = triggerContext.text &&
                 triggerContext.relativeFilePath &&
@@ -714,8 +793,7 @@ class AgenticChatController {
      * Runs the compaction, making requests and processing tool uses until completion
      */
     #shouldCompact(currentRequestCount) {
-        // 80% of 570K limit
-        if (currentRequestCount > 456_000) {
+        if (currentRequestCount > constants_2.COMPACTION_CHARACTER_THRESHOLD) {
             this.#debug(`Current request total character count is: ${currentRequestCount}, prompting user to compact`);
             return true;
         }
@@ -744,7 +822,7 @@ class AgenticChatController {
         if (currentMessage) {
             //  Get and process the messages from history DB to maintain invariants for service requests
             try {
-                const { history: historyMessages, historyCount: historyCharCount } = this.#chatHistoryDb.fixAndGetHistory(tabId, conversationIdentifier ?? '', currentMessage, []);
+                const { history: historyMessages, historyCount: historyCharCount } = this.#chatHistoryDb.fixAndGetHistory(tabId, currentMessage, []);
                 messages = historyMessages;
                 characterCount = historyCharCount;
             }
@@ -862,7 +940,7 @@ class AgenticChatController {
             if (currentMessage) {
                 //  Get and process the messages from history DB to maintain invariants for service requests
                 try {
-                    const { history: historyMessages, historyCount: historyCharacterCount, currentCount: currentInputCount, } = this.#chatHistoryDb.fixAndGetHistory(tabId, conversationId, currentMessage, pinnedContextMessages);
+                    const { history: historyMessages, historyCount: historyCharacterCount, currentCount: currentInputCount, } = this.#chatHistoryDb.fixAndGetHistory(tabId, currentMessage, pinnedContextMessages);
                     messages = historyMessages;
                     currentRequestCount = currentInputCount + historyCharacterCount;
                     this.#debug(`Request total character count: ${currentRequestCount}`);
@@ -903,7 +981,7 @@ class AgenticChatController {
                 }
                 else {
                     this.#chatHistoryDb.addMessage(tabId, 'cwc', conversationIdentifier, {
-                        body: escapeHTML(currentMessage.userInputMessage?.content ?? ''),
+                        body: currentMessage.userInputMessage?.content ?? '',
                         type: 'prompt',
                         userIntent: currentMessage.userInputMessage?.userIntent,
                         origin: currentMessage.userInputMessage?.origin,
@@ -1245,7 +1323,9 @@ class AgenticChatController {
                             [toolConstants_1.FILE_SEARCH]: { Tool: fileSearch_1.FileSearch },
                         };
                         const { Tool } = toolMap[toolUse.name];
-                        const tool = new Tool(this.#features);
+                        const tool = toolUse.name === toolConstants_1.FS_READ && session.isMemoryBankGeneration
+                            ? new Tool(this.#features, constants_2.FSREAD_MEMORY_BANK_MAX_PER_FILE, constants_2.FSREAD_MEMORY_BANK_MAX_TOTAL)
+                            : new Tool(this.#features);
                         // For MCP tools, get the permission from McpManager
                         // const permission = McpManager.instance.getToolPerm('Built-in', toolUse.name)
                         // If permission is 'alwaysAllow', we don't need to ask for acceptance
@@ -1351,6 +1431,25 @@ class AgenticChatController {
                 const ws = this.#getWritableStream(chatResultStream, toolUse);
                 const result = await this.#features.agent.runTool(toolUse.name, toolUse.input, token, ws);
                 let toolResultContent;
+                if (toolUse.name === codeReview_1.CodeReview.toolName) {
+                    // no need to write tool result for code review, this is handled by model via chat
+                    // Push result in message so that it is picked by IDE plugin to show in issues panel
+                    const codeReviewResult = result;
+                    if (codeReviewResult?.output?.kind === 'json' &&
+                        codeReviewResult.output.success &&
+                        codeReviewResult.output.content?.findingsByFile) {
+                        await chatResultStream.writeResultBlock({
+                            type: 'tool',
+                            messageId: toolUse.toolUseId + codeReviewConstants_1.CODE_REVIEW_FINDINGS_MESSAGE_SUFFIX,
+                            body: codeReviewResult.output.content.findingsByFile,
+                        });
+                        codeReviewResult.output.content = {
+                            codeReviewId: codeReviewResult.output.content.codeReviewId,
+                            message: codeReviewResult.output.content.message,
+                            findingsByFileSimplified: codeReviewResult.output.content.findingsByFileSimplified,
+                        };
+                    }
+                }
                 if (typeof result === 'string') {
                     toolResultContent = { text: result };
                 }
@@ -1422,18 +1521,6 @@ class AgenticChatController {
                         await chatResultStream.writeResultBlock(chatResult);
                         break;
                     case codeReview_1.CodeReview.toolName:
-                        // no need to write tool result for code review, this is handled by model via chat
-                        // Push result in message so that it is picked by IDE plugin to show in issues panel
-                        const codeReviewResult = result;
-                        if (codeReviewResult?.output?.kind === 'json' &&
-                            codeReviewResult.output.success &&
-                            codeReviewResult.output.content?.findingsByFile) {
-                            await chatResultStream.writeResultBlock({
-                                type: 'tool',
-                                messageId: toolUse.toolUseId + codeReviewConstants_1.CODE_REVIEW_FINDINGS_MESSAGE_SUFFIX,
-                                body: codeReviewResult.output.content.findingsByFile,
-                            });
-                        }
                         break;
                     case displayFindings_1.DisplayFindings.toolName:
                         // no need to write tool result for code review, this is handled by model via chat
@@ -1940,7 +2027,7 @@ class AgenticChatController {
             body: constants_2.COMPACTION_HEADER_BODY,
             buttons,
         };
-        const body = (0, constants_2.COMPACTION_BODY)(Math.round((characterCount / chatDb_1.MaxOverallCharacters) * 100));
+        const body = (0, constants_2.COMPACTION_BODY)(Math.round((characterCount / constants_2.MAX_OVERALL_CHARACTERS) * 100));
         return {
             type: 'tool',
             messageId,
@@ -2491,13 +2578,15 @@ class AgenticChatController {
                     .filter(prompt => prompt.length > 0)),
             },
         });
+        // Reset memory bank flag after completion
+        session.isMemoryBankGeneration = false;
         return chatResultStream.getResult();
     }
     /**
      * Handles errors that occur during the request
      */
     async #handleRequestError(conversationId, err, errorMessageId, tabId, metric, agenticCodingMode) {
-        const errorMessage = (0, utils_2.getErrorMsg)(err) ?? constants_2.GENERIC_ERROR_MS;
+        const errorMessage = ((0, utils_2.getErrorMsg)(err) ?? constants_2.GENERIC_ERROR_MS).replace(/[\r\n]+/g, ' '); // replace new lines with empty space
         const requestID = (0, utils_2.getRequestID)(err) ?? '';
         metric.setDimension('cwsprChatResponseCode', (0, utils_2.getHttpStatusCode)(err) ?? 0);
         metric.setDimension('languageServerVersion', this.#features.runtime.serverInfo.version);
@@ -2506,7 +2595,13 @@ class AgenticChatController {
         metric.metric.requestIds = [requestID];
         metric.metric.cwsprChatMessageId = errorMessageId;
         metric.metric.cwsprChatConversationId = conversationId;
-        await this.#telemetryController.emitAddMessageMetric(tabId, metric.metric, 'Failed', errorMessage);
+        const errorCode = err.code ?? '';
+        await this.#telemetryController.emitAddMessageMetric(tabId, metric.metric, 'Failed', errorMessage, errorCode);
+        // Reset memory bank flag on request error
+        const sessionResult = this.#chatSessionManagementService.getSession(tabId);
+        if (sessionResult.success) {
+            sessionResult.data.isMemoryBankGeneration = false;
+        }
         if ((0, utils_2.isUsageLimitError)(err)) {
             if (this.#paidTierMode !== 'paidtier') {
                 this.setPaidTierMode(tabId, 'freetier-limit');