@aws/lsp-codewhisperer 0.0.31 → 0.0.33

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

@@ -0,0 +1,176 @@
+"use strict";
+var _a;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgenticChatResultStream = void 0;
+const crypto_1 = require("crypto");
+class AgenticChatResultStream {
+    static resultDelimiter = '\n\n';
+    #state = {
+        chatResultBlocks: [],
+        isLocked: false,
+        uuid: (0, crypto_1.randomUUID)(),
+        messageId: undefined,
+        messageIdToUpdateForTool: new Map(),
+        messageOperations: new Map(),
+    };
+    #sendProgress;
+    constructor(sendProgress) {
+        this.#sendProgress = sendProgress;
+    }
+    getResult(only) {
+        return this.#joinResults(this.#state.chatResultBlocks, only);
+    }
+    setMessageIdToUpdateForTool(toolName, messageId) {
+        this.#state.messageIdToUpdateForTool.set(toolName, messageId);
+    }
+    getMessageIdToUpdateForTool(toolName) {
+        return this.#state.messageIdToUpdateForTool.get(toolName);
+    }
+    /**
+     * Adds a file operation for a specific message
+     * @param messageId The ID of the message
+     * @param type The type of operation ('fsRead' or 'listDirectory' or 'fsWrite')
+     * @param filePaths Array of FileDetailsWithPath involved in the operation
+     */
+    addMessageOperation(messageId, type, filePaths) {
+        this.#state.messageOperations.set(messageId, { type: type, filePaths });
+    }
+    /**
+     * Gets the file operation details for a specific message
+     * @param messageId The ID of the message
+     * @returns The file operation details or undefined if not found
+     */
+    getMessageOperation(messageId) {
+        return this.#state.messageOperations.get(messageId);
+    }
+    #joinResults(chatResults, only) {
+        const result = {
+            body: '',
+            additionalMessages: [],
+            messageId: this.#state.messageId || this.#state.uuid,
+        };
+        return chatResults
+            .filter(cr => cr.messageId === this.#state.messageId || only === undefined || only === cr.messageId)
+            .reduce((acc, c) => {
+            if (c.messageId === this.#state.messageId) {
+                return {
+                    ...acc,
+                    buttons: [...(acc.buttons ?? []), ...(c.buttons ?? [])],
+                    body: acc.body + _a.resultDelimiter + c.body,
+                    ...(c.contextList && { contextList: c.contextList }),
+                    header: Object.prototype.hasOwnProperty.call(c, 'header') ? c.header : acc.header,
+                    codeReference: [...(acc.codeReference ?? []), ...(c.codeReference ?? [])],
+                };
+            }
+            else if (acc.additionalMessages.some(am => am.messageId === c.messageId)) {
+                return {
+                    ...acc,
+                    additionalMessages: acc.additionalMessages.map(am => ({
+                        ...am,
+                        buttons: am.messageId === c.messageId
+                            ? [...(am.buttons ?? []), ...(c.buttons ?? [])]
+                            : am.buttons,
+                        body: am.messageId === c.messageId
+                            ? am.body + _a.resultDelimiter + c.body
+                            : am.body,
+                        ...(am.messageId === c.messageId &&
+                            (c.contextList || acc.contextList) && {
+                            contextList: {
+                                filePaths: [
+                                    ...(acc.contextList?.filePaths ?? []),
+                                    ...(c.contextList?.filePaths ?? []),
+                                ],
+                                rootFolderTitle: c.contextList?.rootFolderTitle
+                                    ? c.contextList.rootFolderTitle
+                                    : (acc.contextList?.rootFolderTitle ?? ''),
+                                details: {
+                                    ...(acc.contextList?.details ?? {}),
+                                    ...(c.contextList?.details ?? {}),
+                                },
+                            },
+                        }),
+                        ...(am.messageId === c.messageId &&
+                            (c.fileList || acc.fileList) && {
+                            fileList: {
+                                filePaths: [
+                                    ...(acc.fileList?.filePaths ?? []),
+                                    ...(c.fileList?.filePaths ?? []),
+                                ],
+                                rootFolderTitle: c.fileList?.rootFolderTitle
+                                    ? c.fileList.rootFolderTitle
+                                    : (acc.fileList?.rootFolderTitle ?? ''),
+                                details: {
+                                    ...(acc.fileList?.details ?? {}),
+                                    ...(c.fileList?.details ?? {}),
+                                },
+                            },
+                        }),
+                        header: Object.prototype.hasOwnProperty.call(c, 'header') ? c.header : am.header,
+                    })),
+                };
+            }
+            else {
+                return {
+                    ...acc,
+                    additionalMessages: [...acc.additionalMessages, c],
+                };
+            }
+        }, result);
+    }
+    /**
+     * Add a block to the message block store and send it to the client.
+     * @param result the blockId associated with the block such that it can be overwritten later
+     * @returns
+     */
+    async writeResultBlock(result) {
+        this.#state.chatResultBlocks.push(result);
+        await this.#sendProgress(this.getResult(result.messageId));
+        // TODO: We should use chat messageId as blockId instead of nummber for more predictable updates.
+        return this.#state.chatResultBlocks.length - 1;
+    }
+    /**
+     * Overwrites a specific blockId and re-sends the resulting blocks to the client.
+     * @param result
+     * @param blockId
+     */
+    async overwriteResultBlock(result, blockId) {
+        this.#state.chatResultBlocks[blockId] = result;
+        await this.#sendProgress(this.getResult(result.messageId));
+    }
+    /**
+     * Removes a specific messageId and re-sends the result to the client.
+     * @param messageId
+     */
+    async removeResultBlock(messageId) {
+        this.#state.chatResultBlocks = this.#state.chatResultBlocks.filter(block => block.messageId !== messageId);
+    }
+    getResultStreamWriter() {
+        // Note: if write calls are not awaited, stream can be out-of-order.
+        if (this.#state.isLocked) {
+            throw new Error('AgenticChatResultStream: already locked');
+        }
+        this.#state.isLocked = true;
+        let lastResult;
+        return {
+            write: async (intermediateChatResult) => {
+                const isLoading = intermediateChatResult.messageId?.startsWith('loading-');
+                if (isLoading) {
+                    return await this.#sendProgress(intermediateChatResult);
+                }
+                this.#state.messageId = intermediateChatResult.messageId;
+                const combinedResult = this.#joinResults([...this.#state.chatResultBlocks, intermediateChatResult], intermediateChatResult.messageId);
+                lastResult = intermediateChatResult;
+                return await this.#sendProgress(combinedResult);
+            },
+            close: async () => {
+                if (lastResult) {
+                    this.#state.chatResultBlocks.push(lastResult);
+                }
+                this.#state.isLocked = false;
+            },
+        };
+    }
+}
+exports.AgenticChatResultStream = AgenticChatResultStream;
+_a = AgenticChatResultStream;
+//# sourceMappingURL=agenticChatResultStream.js.map

package/out/language-server/agenticChat/agenticChatResultStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agenticChatResultStream.js","sourceRoot":"","sources":["../../../src/language-server/agenticChat/agenticChatResultStream.ts"],"names":[],"mappings":";;;;AACA,mCAAmC;AAiDnC,MAAa,uBAAuB;IAChC,MAAM,CAAU,eAAe,GAAG,MAAM,CAAA;IACxC,MAAM,GAAG;QACL,gBAAgB,EAAE,EAAmB;QACrC,QAAQ,EAAE,KAAK;QACf,IAAI,EAAE,IAAA,mBAAU,GAAE;QAClB,SAAS,EAAE,SAA+B;QAC1C,wBAAwB,EAAE,IAAI,GAAG,EAAyB;QAC1D,iBAAiB,EAAE,IAAI,GAAG,EAAyB;KACtD,CAAA;IACQ,aAAa,CAAuD;IAE7E,YAAY,YAAmE;QAC3E,IAAI,CAAC,aAAa,GAAG,YAAY,CAAA;IACrC,CAAC;IAED,SAAS,CAAC,IAAa;QACnB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAA;IAChE,CAAC;IAED,2BAA2B,CAAC,QAAgB,EAAE,SAAiB;QAC3D,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAyB,EAAE,SAAS,CAAC,CAAA;IAClF,CAAC;IAED,2BAA2B,CAAC,QAAgB;QACxC,OAAO,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,GAAG,CAAC,QAAyB,CAAC,CAAA;IAC9E,CAAC;IAED;;;;;OAKG;IACH,mBAAmB,CAAC,SAAiB,EAAE,IAAY,EAAE,SAAgC;QACjF,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,IAAqB,EAAE,SAAS,EAAE,CAAC,CAAA;IAC5F,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CAAC,SAAiB;QACjC,OAAO,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;IACvD,CAAC;IAED,YAAY,CAAC,WAA0B,EAAE,IAAa;QAClD,MAAM,MAAM,GAAe;YACvB,IAAI,EAAE,EAAE;YACR,kBAAkB,EAAE,EAAE;YACtB,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;SACvD,CAAA;QAED,OAAO,WAAW;aACb,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,EAAE,CAAC,SAAS,CAAC;aACnG,MAAM,CAAa,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;YAC3B,IAAI,CAAC,CAAC,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBACxC,OAAO;oBACH,GAAG,GAAG;oBACN,OAAO,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;oBACvD,IAAI,EAAE,GAAG,CAAC,IAAI,GAAG,EAAuB,CAAC,eAAe,GAAG,CAAC,CAAC,IAAI;oBACjE,GAAG,CAAC,CAAC,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;oBACpD,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM;oBACjF,aAAa,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;iBAC5E,CAAA;YACL,CAAC;iBAAM,IAAI,GAAG,CAAC,kBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC1E,OAAO;oBACH,GAAG,GAAG;oBACN,kBAAkB,EAAE,GAAG,CAAC,kBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;wBACnD,GAAG,EAAE;wBACL,OAAO,EACH,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS;4BACxB,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;4BAC/C,CAAC,CAAC,EAAE,CAAC,OAAO;wBACpB,IAAI,EACA,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS;4BACxB,CAAC,CAAC,EAAE,CAAC,IAAI,GAAG,EAAuB,CAAC,eAAe,GAAG,CAAC,CAAC,IAAI;4BAC5D,CAAC,CAAC,EAAE,CAAC,IAAI;wBACjB,GAAG,CAAC,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS;4BAC5B,CAAC,CAAC,CAAC,WAAW,IAAI,GAAG,CAAC,WAAW,CAAC,IAAI;4BAClC,WAAW,EAAE;gCACT,SAAS,EAAE;oCACP,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,EAAE,CAAC;oCACrC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,SAAS,IAAI,EAAE,CAAC;iCACtC;gCACD,eAAe,EAAE,CAAC,CAAC,WAAW,EAAE,eAAe;oCAC3C,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,eAAe;oCAC/B,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,eAAe,IAAI,EAAE,CAAC;gCAC9C,OAAO,EAAE;oCACL,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,IAAI,EAAE,CAAC;oCACnC,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,OAAO,IAAI,EAAE,CAAC;iCACpC;6BACJ;yBACJ,CAAC;wBACN,GAAG,CAAC,EAAE,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS;4BAC5B,CAAC,CAAC,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI;4BAC5B,QAAQ,EAAE;gCACN,SAAS,EAAE;oCACP,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC;oCAClC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC;iCACnC;gCACD,eAAe,EAAE,CAAC,CAAC,QAAQ,EAAE,eAAe;oCACxC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe;oCAC5B,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,eAAe,IAAI,EAAE,CAAC;gCAC3C,OAAO,EAAE;oCACL,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;oCAChC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;iCACjC;6BACJ;yBACJ,CAAC;wBACN,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM;qBACnF,CAAC,CAAC;iBACN,CAAA;YACL,CAAC;iBAAM,CAAC;gBACJ,OAAO;oBACH,GAAG,GAAG;oBACN,kBAAkB,EAAE,CAAC,GAAG,GAAG,CAAC,kBAAmB,EAAE,CAAC,CAAC;iBACtD,CAAA;YACL,CAAC;QACL,CAAC,EAAE,MAAM,CAAC,CAAA;IAClB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,gBAAgB,CAAC,MAAmB;QACtC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACzC,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAA;QAC1D,iGAAiG;QACjG,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAA;IAClD,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,oBAAoB,CAAC,MAAmB,EAAE,OAAe;QAC3D,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,GAAG,MAAM,CAAA;QAC9C,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAA;IAC9D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QACrC,IAAI,CAAC,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAA;IAC9G,CAAC;IAED,qBAAqB;QACjB,oEAAoE;QACpE,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAA;QAC9D,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAA;QAC3B,IAAI,UAAkC,CAAA;QAEtC,OAAO;YACH,KAAK,EAAE,KAAK,EAAE,sBAAmC,EAAE,EAAE;gBACjD,MAAM,SAAS,GAAG,sBAAsB,CAAC,SAAS,EAAE,UAAU,CAAC,UAAU,CAAC,CAAA;gBAC1E,IAAI,SAAS,EAAE,CAAC;oBACZ,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAA;gBAC3D,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,sBAAsB,CAAC,SAAS,CAAA;gBACxD,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CACpC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,sBAAsB,CAAC,EACzD,sBAAsB,CAAC,SAAS,CACnC,CAAA;gBACD,UAAU,GAAG,sBAAsB,CAAA;gBACnC,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,CAAA;YACnD,CAAC;YACD,KAAK,EAAE,KAAK,IAAI,EAAE;gBACd,IAAI,UAAU,EAAE,CAAC;oBACb,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;gBACjD,CAAC;gBACD,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,KAAK,CAAA;YAChC,CAAC;SACJ,CAAA;IACL,CAAC;;AAtLL,0DAuLC"}

package/out/language-server/agenticChat/constants.d.ts

@@ -0,0 +1,4 @@
+export declare const genericErrorMsg = "An unexpected error occurred, check the logs for more information.";
+export declare const maxAgentLoopIterations = 100;
+export declare const loadingThresholdMs = 2000;
+export declare const generateAssistantResponseInputLimit = 600000;

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

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateAssistantResponseInputLimit = exports.loadingThresholdMs = exports.maxAgentLoopIterations = exports.genericErrorMsg = void 0;
+exports.genericErrorMsg = 'An unexpected error occurred, check the logs for more information.';
+exports.maxAgentLoopIterations = 100;
+exports.loadingThresholdMs = 2000;
+exports.generateAssistantResponseInputLimit = 600_000;
+//# sourceMappingURL=constants.js.map

package/out/language-server/agenticChat/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/language-server/agenticChat/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,eAAe,GAAG,oEAAoE,CAAA;AACtF,QAAA,sBAAsB,GAAG,GAAG,CAAA;AAC5B,QAAA,kBAAkB,GAAG,IAAI,CAAA;AACzB,QAAA,mCAAmC,GAAG,OAAO,CAAA"}

package/out/language-server/agenticChat/context/addtionalContextProvider.d.ts

@@ -0,0 +1,14 @@
+import { FileList, ContextCommand } from '@aws/language-server-runtimes/protocol';
+import { AdditionalContextPrompt, ContextCommandItem } from 'local-indexing';
+import { AdditionalContentEntryAddition, TriggerContext } from './agenticChatTriggerContext';
+import { Lsp, Workspace } from '@aws/language-server-runtimes/server-interface';
+export declare class AdditionalContextProvider {
+    private readonly workspace;
+    private readonly lsp;
+    constructor(workspace: Workspace, lsp: Lsp);
+    collectWorkspaceRules(): Promise<ContextCommandItem[]>;
+    getContextType(prompt: AdditionalContextPrompt): string;
+    getAdditionalContext(triggerContext: TriggerContext, context?: ContextCommand[]): Promise<AdditionalContentEntryAddition[]>;
+    getFileListFromContext(context: AdditionalContentEntryAddition[]): FileList;
+    mapToContextCommandItems(context: ContextCommand[], workspaceFolderPath: string): ContextCommandItem[];
+}

package/out/language-server/agenticChat/context/addtionalContextProvider.js

@@ -0,0 +1,191 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AdditionalContextProvider = void 0;
+const path = require("path");
+const agenticChatTriggerContext_1 = require("./agenticChatTriggerContext");
+const vscode_uri_1 = require("vscode-uri");
+const lsp_core_1 = require("@aws/lsp-core");
+const contextUtils_1 = require("./contextUtils");
+const localProjectContextController_1 = require("../../../shared/localProjectContextController");
+class AdditionalContextProvider {
+    workspace;
+    lsp;
+    constructor(workspace, lsp) {
+        this.workspace = workspace;
+        this.lsp = lsp;
+    }
+    async collectWorkspaceRules() {
+        const rulesFiles = [];
+        let workspaceFolders = lsp_core_1.workspaceUtils.getWorkspaceFolderPaths(this.lsp);
+        if (!workspaceFolders.length) {
+            return rulesFiles;
+        }
+        for (const workspaceFolder of workspaceFolders) {
+            const rulesPath = path.join(workspaceFolder, '.amazonq', 'rules');
+            const folderExists = await this.workspace.fs.exists(rulesPath);
+            if (folderExists) {
+                const entries = await this.workspace.fs.readdir(rulesPath);
+                for (const entry of entries) {
+                    if (entry.isFile() && entry.name.endsWith(contextUtils_1.promptFileExtension)) {
+                        rulesFiles.push({
+                            workspaceFolder: workspaceFolder,
+                            type: 'file',
+                            relativePath: path.relative(workspaceFolder, path.join(rulesPath, entry.name)),
+                            id: '',
+                        });
+                    }
+                }
+            }
+        }
+        return rulesFiles;
+    }
+    getContextType(prompt) {
+        if (prompt.name === 'symbol') {
+            return 'code';
+        }
+        if (prompt.filePath.endsWith(contextUtils_1.promptFileExtension)) {
+            if (lsp_core_1.pathUtils.isInDirectory(path.join('.amazonq', 'rules'), prompt.relativePath)) {
+                return 'rule';
+            }
+            else if (lsp_core_1.pathUtils.isInDirectory((0, contextUtils_1.getUserPromptsDirectory)(), prompt.filePath)) {
+                return 'prompt';
+            }
+        }
+        return 'file';
+    }
+    async getAdditionalContext(triggerContext, context) {
+        if (!triggerContext.contextInfo) {
+            triggerContext.contextInfo = contextUtils_1.initialContextInfo;
+        }
+        const additionalContextCommands = [];
+        const workspaceRules = await this.collectWorkspaceRules();
+        let workspaceFolderPath = triggerContext.workspaceFolder?.uri
+            ? vscode_uri_1.URI.parse(triggerContext.workspaceFolder.uri).fsPath
+            : lsp_core_1.workspaceUtils.getWorkspaceFolderPaths(this.lsp)[0];
+        if (workspaceRules.length > 0) {
+            additionalContextCommands.push(...workspaceRules);
+        }
+        triggerContext.contextInfo.contextCount.ruleContextCount = workspaceRules.length;
+        if (context) {
+            let fileContextCount = 0;
+            let folderContextCount = 0;
+            let promptContextCount = 0;
+            let codeContextCount = 0;
+            additionalContextCommands.push(...this.mapToContextCommandItems(context, workspaceFolderPath));
+            for (const c of context) {
+                if (typeof context !== 'string') {
+                    if (c.id === 'prompt') {
+                        promptContextCount++;
+                    }
+                    else if (c.label === 'file') {
+                        fileContextCount++;
+                    }
+                    else if (c.label === 'folder') {
+                        folderContextCount++;
+                    }
+                    else if (c.label === 'code') {
+                        codeContextCount++;
+                    }
+                }
+            }
+            triggerContext.contextInfo.contextCount = {
+                ...triggerContext.contextInfo.contextCount,
+                fileContextCount,
+                folderContextCount,
+                promptContextCount,
+                codeContextCount,
+            };
+        }
+        if (additionalContextCommands.length === 0) {
+            return [];
+        }
+        let prompts = [];
+        try {
+            const localProjectContextController = await localProjectContextController_1.LocalProjectContextController.getInstance();
+            prompts = await localProjectContextController.getContextCommandPrompt(additionalContextCommands);
+        }
+        catch (error) {
+            // do nothing
+        }
+        const contextEntry = [];
+        let ruleContextLength = 0;
+        let fileContextLength = 0;
+        let promptContextLength = 0;
+        let codeContextLength = 0;
+        for (const prompt of prompts.slice(0, agenticChatTriggerContext_1.additionalContextMaxLength)) {
+            const contextType = this.getContextType(prompt);
+            const description = contextType === 'rule' || contextType === 'prompt'
+                ? `You must follow the instructions in ${prompt.relativePath}. Below are lines ${prompt.startLine}-${prompt.endLine} of this file:\n`
+                : prompt.description;
+            const relativePath = prompt.filePath.startsWith((0, contextUtils_1.getUserPromptsDirectory)())
+                ? path.basename(prompt.filePath)
+                : path.relative(workspaceFolderPath, prompt.filePath);
+            const entry = {
+                name: prompt.name.substring(0, contextUtils_1.additionalContentNameLimit),
+                description: description.substring(0, contextUtils_1.additionalContentNameLimit),
+                innerContext: prompt.content.substring(0, agenticChatTriggerContext_1.workspaceChunkMaxSize),
+                type: contextType,
+                path: prompt.filePath,
+                relativePath: relativePath,
+                startLine: prompt.startLine,
+                endLine: prompt.endLine,
+            };
+            contextEntry.push(entry);
+            if (contextType === 'rule') {
+                ruleContextLength += prompt.content.length;
+            }
+            else if (contextType === 'prompt') {
+                promptContextLength += prompt.content.length;
+            }
+            else if (contextType === 'code') {
+                codeContextLength += prompt.content.length;
+            }
+            else {
+                fileContextLength += prompt.content.length;
+            }
+        }
+        triggerContext.contextInfo.contextLength = {
+            ruleContextLength,
+            fileContextLength,
+            promptContextLength,
+            codeContextLength,
+        };
+        return contextEntry;
+    }
+    getFileListFromContext(context) {
+        const fileDetails = {};
+        for (const item of context) {
+            fileDetails[item.relativePath] = {
+                lineRanges: [
+                    {
+                        first: item.name === 'symbol' ? item.startLine : -1,
+                        second: item.name === 'symbol' ? item.endLine : -1,
+                    },
+                ],
+                description: item.path,
+                fullPath: item.path,
+            };
+        }
+        const fileList = {
+            filePaths: [...new Set(context.map(item => item.relativePath))],
+            details: fileDetails,
+        };
+        return fileList;
+    }
+    mapToContextCommandItems(context, workspaceFolderPath) {
+        const contextCommands = [];
+        for (const item of context) {
+            if (item.route && item.route.length === 2) {
+                contextCommands.push({
+                    workspaceFolder: item.route?.[0] ?? workspaceFolderPath,
+                    type: item.label ?? '',
+                    relativePath: item.route?.[1] ?? '',
+                    id: item.id ?? '',
+                });
+            }
+        }
+        return contextCommands;
+    }
+}
+exports.AdditionalContextProvider = AdditionalContextProvider;
+//# sourceMappingURL=addtionalContextProvider.js.map

package/out/language-server/agenticChat/context/addtionalContextProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"addtionalContextProvider.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/context/addtionalContextProvider.ts"],"names":[],"mappings":";;;AAEA,6BAA4B;AAC5B,2EAKoC;AACpC,2CAAgC;AAEhC,4CAAyD;AACzD,iDAKuB;AACvB,iGAA6F;AAE7F,MAAa,yBAAyB;IAEb;IACA;IAFrB,YACqB,SAAoB,EACpB,GAAQ;QADR,cAAS,GAAT,SAAS,CAAW;QACpB,QAAG,GAAH,GAAG,CAAK;IAC1B,CAAC;IAEJ,KAAK,CAAC,qBAAqB;QACvB,MAAM,UAAU,GAAyB,EAAE,CAAA;QAC3C,IAAI,gBAAgB,GAAG,yBAAc,CAAC,uBAAuB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAEvE,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC;YAC3B,OAAO,UAAU,CAAA;QACrB,CAAC;QACD,KAAK,MAAM,eAAe,IAAI,gBAAgB,EAAE,CAAC;YAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,UAAU,EAAE,OAAO,CAAC,CAAA;YACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAE9D,IAAI,YAAY,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;gBAE1D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC1B,IAAI,KAAK,CAAC,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,kCAAmB,CAAC,EAAE,CAAC;wBAC7D,UAAU,CAAC,IAAI,CAAC;4BACZ,eAAe,EAAE,eAAe;4BAChC,IAAI,EAAE,MAAM;4BACZ,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;4BAC9E,EAAE,EAAE,EAAE;yBACT,CAAC,CAAA;oBACN,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QACD,OAAO,UAAU,CAAA;IACrB,CAAC;IAED,cAAc,CAAC,MAA+B;QAC1C,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC3B,OAAO,MAAM,CAAA;QACjB,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,kCAAmB,CAAC,EAAE,CAAC;YAChD,IAAI,oBAAS,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC/E,OAAO,MAAM,CAAA;YACjB,CAAC;iBAAM,IAAI,oBAAS,CAAC,aAAa,CAAC,IAAA,sCAAuB,GAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC7E,OAAO,QAAQ,CAAA;YACnB,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,oBAAoB,CACtB,cAA8B,EAC9B,OAA0B;QAE1B,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC;YAC9B,cAAc,CAAC,WAAW,GAAG,iCAAkB,CAAA;QACnD,CAAC;QACD,MAAM,yBAAyB,GAAyB,EAAE,CAAA;QAC1D,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAA;QACzD,IAAI,mBAAmB,GAAG,cAAc,CAAC,eAAe,EAAE,GAAG;YACzD,CAAC,CAAC,gBAAG,CAAC,KAAK,CAAC,cAAc,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM;YACtD,CAAC,CAAC,yBAAc,CAAC,uBAAuB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;QAEzD,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,yBAAyB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAA;QACrD,CAAC;QACD,cAAc,CAAC,WAAW,CAAC,YAAY,CAAC,gBAAgB,GAAG,cAAc,CAAC,MAAM,CAAA;QAChF,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,gBAAgB,GAAG,CAAC,CAAA;YACxB,IAAI,kBAAkB,GAAG,CAAC,CAAA;YAC1B,IAAI,kBAAkB,GAAG,CAAC,CAAA;YAC1B,IAAI,gBAAgB,GAAG,CAAC,CAAA;YACxB,yBAAyB,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,wBAAwB,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC,CAAA;YAC9F,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACtB,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAC9B,IAAI,CAAC,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;wBACpB,kBAAkB,EAAE,CAAA;oBACxB,CAAC;yBAAM,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;wBAC5B,gBAAgB,EAAE,CAAA;oBACtB,CAAC;yBAAM,IAAI,CAAC,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;wBAC9B,kBAAkB,EAAE,CAAA;oBACxB,CAAC;yBAAM,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;wBAC5B,gBAAgB,EAAE,CAAA;oBACtB,CAAC;gBACL,CAAC;YACL,CAAC;YACD,cAAc,CAAC,WAAY,CAAC,YAAY,GAAG;gBACvC,GAAG,cAAc,CAAC,WAAY,CAAC,YAAY;gBAC3C,gBAAgB;gBAChB,kBAAkB;gBAClB,kBAAkB;gBAClB,gBAAgB;aACnB,CAAA;QACL,CAAC;QAED,IAAI,yBAAyB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzC,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,OAAO,GAA8B,EAAE,CAAA;QAC3C,IAAI,CAAC;YACD,MAAM,6BAA6B,GAAG,MAAM,6DAA6B,CAAC,WAAW,EAAE,CAAA;YACvF,OAAO,GAAG,MAAM,6BAA6B,CAAC,uBAAuB,CAAC,yBAAyB,CAAC,CAAA;QACpG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,aAAa;QACjB,CAAC;QAED,MAAM,YAAY,GAAqC,EAAE,CAAA;QACzD,IAAI,iBAAiB,GAAG,CAAC,CAAA;QACzB,IAAI,iBAAiB,GAAG,CAAC,CAAA;QACzB,IAAI,mBAAmB,GAAG,CAAC,CAAA;QAC3B,IAAI,iBAAiB,GAAG,CAAC,CAAA;QACzB,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,sDAA0B,CAAC,EAAE,CAAC;YAChE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAA;YAC/C,MAAM,WAAW,GACb,WAAW,KAAK,MAAM,IAAI,WAAW,KAAK,QAAQ;gBAC9C,CAAC,CAAC,uCAAuC,MAAM,CAAC,YAAY,qBAAqB,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,OAAO,kBAAkB;gBACrI,CAAC,CAAC,MAAM,CAAC,WAAW,CAAA;YAE5B,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAA,sCAAuB,GAAE,CAAC;gBACtE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC;gBAChC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;YACzD,MAAM,KAAK,GAAG;gBACV,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,yCAA0B,CAAC;gBAC1D,WAAW,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,yCAA0B,CAAC;gBACjE,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,iDAAqB,CAAC;gBAChE,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,MAAM,CAAC,QAAQ;gBACrB,YAAY,EAAE,YAAY;gBAC1B,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;aAC1B,CAAA;YACD,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAExB,IAAI,WAAW,KAAK,MAAM,EAAE,CAAC;gBACzB,iBAAiB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAA;YAC9C,CAAC;iBAAM,IAAI,WAAW,KAAK,QAAQ,EAAE,CAAC;gBAClC,mBAAmB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAA;YAChD,CAAC;iBAAM,IAAI,WAAW,KAAK,MAAM,EAAE,CAAC;gBAChC,iBAAiB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAA;YAC9C,CAAC;iBAAM,CAAC;gBACJ,iBAAiB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAA;YAC9C,CAAC;QACL,CAAC;QACD,cAAc,CAAC,WAAW,CAAC,aAAa,GAAG;YACvC,iBAAiB;YACjB,iBAAiB;YACjB,mBAAmB;YACnB,iBAAiB;SACpB,CAAA;QACD,OAAO,YAAY,CAAA;IACvB,CAAC;IAED,sBAAsB,CAAC,OAAyC;QAC5D,MAAM,WAAW,GAAgC,EAAE,CAAA;QACnD,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YACzB,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG;gBAC7B,UAAU,EAAE;oBACR;wBACI,KAAK,EAAE,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;wBACnD,MAAM,EAAE,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;qBACrD;iBACJ;gBACD,WAAW,EAAE,IAAI,CAAC,IAAI;gBACtB,QAAQ,EAAE,IAAI,CAAC,IAAI;aACtB,CAAA;QACL,CAAC;QACD,MAAM,QAAQ,GAAa;YACvB,SAAS,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;YAC/D,OAAO,EAAE,WAAW;SACvB,CAAA;QACD,OAAO,QAAQ,CAAA;IACnB,CAAC;IAED,wBAAwB,CAAC,OAAyB,EAAE,mBAA2B;QAC3E,MAAM,eAAe,GAAyB,EAAE,CAAA;QAChD,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxC,eAAe,CAAC,IAAI,CAAC;oBACjB,eAAe,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,mBAAmB;oBACvD,IAAI,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;oBACtB,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;oBACnC,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE;iBACE,CAAC,CAAA;YAC5B,CAAC;QACL,CAAC;QACD,OAAO,eAAe,CAAA;IAC1B,CAAC;CACJ;AA3LD,8DA2LC"}

package/out/language-server/agenticChat/context/agenticChatTriggerContext.d.ts

@@ -0,0 +1,54 @@
+/**
+ * Copied from chat/contexts/triggerContext.ts for the purpose of developing a divergent implementation.
+ * Will be deleted or merged.
+ */
+import { TriggerType } from '@aws/chat-client-ui-types';
+import { ChatTriggerType, UserIntent, AdditionalContentEntry, GenerateAssistantResponseCommandInput, ChatMessage } from '@amzn/codewhisperer-streaming';
+import { BedrockTools, ChatParams, InlineChatParams, FileList, TextDocument } from '@aws/language-server-runtimes/server-interface';
+import { Features } from '../../types';
+import { DocumentContext } from '../../chat/contexts/documentContext';
+import { RelevantTextDocument } from '@amzn/codewhisperer-streaming';
+import { AgenticChatResultStream } from '../agenticChatResultStream';
+import { ContextInfo } from './contextUtils';
+export interface TriggerContext extends Partial<DocumentContext> {
+    userIntent?: UserIntent;
+    triggerType?: TriggerType;
+    contextInfo?: ContextInfo;
+    documentReference?: FileList;
+}
+export type LineInfo = {
+    startLine: number;
+    endLine: number;
+};
+export type AdditionalContentEntryAddition = AdditionalContentEntry & {
+    type: string;
+    relativePath: string;
+    path: string;
+} & LineInfo;
+export type RelevantTextDocumentAddition = RelevantTextDocument & LineInfo;
+export declare const workspaceChunkMaxSize = 40960;
+export declare const additionalContextMaxLength = 100;
+export interface DocumentReference {
+    readonly relativeFilePath: string;
+    readonly lineRanges: Array<{
+        first: number;
+        second: number;
+    }>;
+}
+export declare class AgenticChatTriggerContext {
+    #private;
+    private static readonly DEFAULT_CURSOR_STATE;
+    constructor({ workspace, lsp, logging }: Pick<Features, 'workspace' | 'lsp' | 'logging'> & Partial<Features>);
+    getNewTriggerContext(params: ChatParams | InlineChatParams): Promise<TriggerContext>;
+    getChatParamsFromTrigger(params: ChatParams | InlineChatParams, triggerContext: TriggerContext, chatTriggerType: ChatTriggerType, customizationArn?: string, chatResultStream?: AgenticChatResultStream, profileArn?: string, history?: ChatMessage[], tools?: BedrockTools, additionalContent?: AdditionalContentEntryAddition[]): Promise<GenerateAssistantResponseCommandInput>;
+    extractDocumentContext(input: Pick<ChatParams | InlineChatParams, 'cursorState' | 'textDocument'>): Promise<DocumentContext | undefined>;
+    /**
+     * Fetch the current textDocument such that:
+     * 1. If the document is synced with LSP, return the synced textDocument
+     * 2. If the document is not synced with LSP, read the file from the file system
+     * 3. If the file cannot be read, return undefined
+     * @param uri
+     * @returns
+     */
+    getTextDocument(uri: string): Promise<TextDocument | undefined>;
+}