@aws/lsp-codewhisperer 0.0.32 → 0.0.34

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

@@ -0,0 +1,246 @@
+"use strict";
+/**
+ * Copied from chat/contexts/triggerContext.ts for the purpose of developing a divergent implementation.
+ * Will be deleted or merged.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgenticChatTriggerContext = exports.additionalContextMaxLength = exports.workspaceChunkMaxSize = void 0;
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const documentContext_1 = require("../../chat/contexts/documentContext");
+const lsp_core_1 = require("@aws/lsp-core");
+const vscode_uri_1 = require("vscode-uri");
+const localProjectContextController_1 = require("../../../shared/localProjectContextController");
+const path = require("path");
+const languageDetection_1 = require("../../../shared/languageDetection");
+// limit for each chunk of @workspace
+exports.workspaceChunkMaxSize = 40_960;
+// limit for the length of additionalContent
+exports.additionalContextMaxLength = 100;
+class AgenticChatTriggerContext {
+    static DEFAULT_CURSOR_STATE = { position: { line: 0, character: 0 } };
+    #workspace;
+    #lsp;
+    #logging;
+    #documentContextExtractor;
+    constructor({ workspace, lsp, logging }) {
+        this.#workspace = workspace;
+        this.#lsp = lsp;
+        this.#logging = logging;
+        this.#documentContextExtractor = new documentContext_1.DocumentContextExtractor({ logger: logging, workspace });
+    }
+    async getNewTriggerContext(params) {
+        const documentContext = await this.extractDocumentContext(params);
+        return {
+            ...documentContext,
+            userIntent: undefined,
+        };
+    }
+    #mapPlatformToEnvState(platform) {
+        switch (platform) {
+            case 'darwin':
+                return { operatingSystem: 'macos' };
+            case 'linux':
+                return { operatingSystem: 'linux' };
+            case 'win32':
+            case 'cygwin':
+                return { operatingSystem: 'windows' };
+            default:
+                return undefined;
+        }
+    }
+    async getChatParamsFromTrigger(params, triggerContext, chatTriggerType, customizationArn, chatResultStream, profileArn, history = [], tools = [], additionalContent) {
+        const { prompt } = params;
+        const defaultEditorState = { workspaceFolders: lsp_core_1.workspaceUtils.getWorkspaceFolderPaths(this.#lsp) };
+        const hasWorkspace = 'context' in params ? params.context?.some(c => c.command === '@workspace') : false;
+        let promptContent = prompt.escapedPrompt ?? prompt.prompt;
+        // When the user adds @sage context, ** gets prepended and appended to the prompt because of markdown.
+        // This intereferes with routing logic thus we need to remove it
+        if (promptContent && promptContent.includes('@sage')) {
+            promptContent = promptContent.replace(/\*\*@sage\*\*/g, '@sage');
+        }
+        if (hasWorkspace) {
+            promptContent = promptContent?.replace(/^@workspace\/?/, '');
+        }
+        // Get workspace documents if @workspace is used
+        let relevantDocuments = hasWorkspace
+            ? await this.#getRelevantDocuments(promptContent ?? '', chatResultStream)
+            : [];
+        // Process additionalContent items if present
+        if (additionalContent) {
+            for (const item of additionalContent) {
+                // Determine programming language from file extension or type
+                let programmingLanguage = undefined;
+                if (item.relativePath) {
+                    const ext = path.extname(item.relativePath).toLowerCase();
+                    const language = languageDetection_1.languageByExtension[ext];
+                    if (language) {
+                        programmingLanguage = { languageName: language };
+                    }
+                }
+                const filteredType = item.type === 'file'
+                    ? codewhisperer_streaming_1.ContentType.FILE
+                    : item.type === 'rule' || item.type === 'prompt'
+                        ? codewhisperer_streaming_1.ContentType.PROMPT
+                        : item.type === 'code'
+                            ? codewhisperer_streaming_1.ContentType.CODE
+                            : undefined;
+                // Create the relevant text document
+                const relevantTextDocument = {
+                    text: item.innerContext,
+                    relativeFilePath: item.path,
+                    programmingLanguage: programmingLanguage,
+                    type: filteredType,
+                    startLine: item.startLine || -1,
+                    endLine: item.endLine || -1,
+                };
+                relevantDocuments.push(relevantTextDocument);
+            }
+        }
+        const useRelevantDocuments = relevantDocuments.length !== 0;
+        const data = {
+            conversationState: {
+                chatTriggerType: chatTriggerType,
+                currentMessage: {
+                    userInputMessage: {
+                        content: promptContent,
+                        userInputMessageContext: triggerContext.cursorState && triggerContext.relativeFilePath
+                            ? {
+                                editorState: {
+                                    cursorState: triggerContext.cursorState,
+                                    document: {
+                                        text: triggerContext.text,
+                                        programmingLanguage: triggerContext.programmingLanguage,
+                                        relativeFilePath: triggerContext.relativeFilePath,
+                                    },
+                                    relevantDocuments: useRelevantDocuments ? relevantDocuments : undefined,
+                                    useRelevantDocuments: useRelevantDocuments,
+                                    ...defaultEditorState,
+                                },
+                                tools,
+                                envState: this.#mapPlatformToEnvState(process.platform),
+                            }
+                            : {
+                                tools,
+                                editorState: {
+                                    relevantDocuments: useRelevantDocuments ? relevantDocuments : undefined,
+                                    useRelevantDocuments: useRelevantDocuments,
+                                    ...defaultEditorState,
+                                },
+                                envState: this.#mapPlatformToEnvState(process.platform),
+                            },
+                        userIntent: triggerContext.userIntent,
+                        origin: 'IDE',
+                    },
+                },
+                customizationArn,
+                history,
+            },
+            profileArn,
+        };
+        return data;
+    }
+    // public for testing
+    async extractDocumentContext(input) {
+        const { textDocument: textDocumentIdentifier, cursorState } = input;
+        if (textDocumentIdentifier?.uri === undefined) {
+            return;
+        }
+        const textDocument = await this.getTextDocument(textDocumentIdentifier.uri);
+        return textDocument
+            ? this.#documentContextExtractor.extractDocumentContext(textDocument, 
+            // we want to include a default position if a text document is found so users can still ask questions about the opened file
+            // the range will be expanded up to the max characters downstream
+            cursorState?.[0] ?? AgenticChatTriggerContext.DEFAULT_CURSOR_STATE)
+            : 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
+     */
+    async getTextDocument(uri) {
+        // Note: version is unused, and languageId can be determined from file extension.
+        const syncedTextDocument = await this.#workspace.getTextDocument(uri);
+        if (syncedTextDocument) {
+            return syncedTextDocument;
+        }
+        try {
+            const content = await this.#workspace.fs.readFile(vscode_uri_1.URI.parse(uri).fsPath);
+            return server_interface_1.TextDocument.create(uri, '', 0, content);
+        }
+        catch {
+            return;
+        }
+    }
+    async #getRelevantDocuments(prompt, chatResultStream) {
+        const localProjectContextController = await localProjectContextController_1.LocalProjectContextController.getInstance();
+        if (!localProjectContextController.isEnabled && chatResultStream) {
+            await chatResultStream.writeResultBlock({
+                body: `To add your workspace as context, enable local indexing in your IDE settings. After enabling, add @workspace to your question, and I'll generate a response using your workspace as context.`,
+                buttons: [
+                    {
+                        id: server_interface_1.OPEN_WORKSPACE_INDEX_SETTINGS_BUTTON_ID,
+                        text: 'Open settings',
+                        icon: 'external',
+                        keepCardAfterClick: false,
+                        status: 'info',
+                    },
+                ],
+            });
+            return [];
+        }
+        let relevantTextDocuments = await this.#queryRelevantDocuments(prompt, localProjectContextController);
+        relevantTextDocuments = relevantTextDocuments.filter(doc => doc.text && doc.text.length > 0);
+        for (const relevantDocument of relevantTextDocuments) {
+            if (relevantDocument.text && relevantDocument.text.length > exports.workspaceChunkMaxSize) {
+                relevantDocument.text = relevantDocument.text.substring(0, exports.workspaceChunkMaxSize);
+                this.#logging.debug(`Truncating @workspace chunk: ${relevantDocument.relativeFilePath} `);
+            }
+        }
+        return relevantTextDocuments;
+    }
+    async #queryRelevantDocuments(prompt, localProjectContextController) {
+        try {
+            const chunks = await localProjectContextController.queryVectorIndex({ query: prompt });
+            const relevantTextDocuments = [];
+            if (!chunks) {
+                return relevantTextDocuments;
+            }
+            for (const chunk of chunks) {
+                const text = chunk.context ?? chunk.content;
+                const baseDocument = {
+                    text,
+                    relativeFilePath: chunk.relativePath ?? path.basename(chunk.filePath),
+                    startLine: chunk.startLine ?? -1,
+                    endLine: chunk.endLine ?? -1,
+                };
+                if (chunk.programmingLanguage && chunk.programmingLanguage !== 'unknown') {
+                    relevantTextDocuments.push({
+                        ...baseDocument,
+                        programmingLanguage: {
+                            languageName: chunk.programmingLanguage,
+                        },
+                        type: codewhisperer_streaming_1.ContentType.WORKSPACE,
+                    });
+                }
+                else {
+                    relevantTextDocuments.push({
+                        ...baseDocument,
+                        type: codewhisperer_streaming_1.ContentType.WORKSPACE,
+                    });
+                }
+            }
+            return relevantTextDocuments;
+        }
+        catch (e) {
+            this.#logging.error(`Error querying query vector index to get relevant documents: ${e}`);
+            return [];
+        }
+    }
+}
+exports.AgenticChatTriggerContext = AgenticChatTriggerContext;
+//# sourceMappingURL=agenticChatTriggerContext.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"agenticChatTriggerContext.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/context/agenticChatTriggerContext.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAGH,2EASsC;AACtC,qFAQuD;AAEvD,yEAA+F;AAC/F,4CAA8C;AAC9C,2CAAgC;AAChC,iGAA6F;AAC7F,6BAA4B;AAE5B,yEAAuE;AAoBvE,qCAAqC;AACxB,QAAA,qBAAqB,GAAG,MAAM,CAAA;AAE3C,4CAA4C;AAC/B,QAAA,0BAA0B,GAAG,GAAG,CAAA;AAO7C,MAAa,yBAAyB;IAC1B,MAAM,CAAU,oBAAoB,GAAgB,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,CAAA;IAEnG,UAAU,CAAuB;IACjC,IAAI,CAAiB;IACrB,QAAQ,CAAqB;IAC7B,yBAAyB,CAA0B;IAEnD,YAAY,EAAE,SAAS,EAAE,GAAG,EAAE,OAAO,EAAuE;QACxG,IAAI,CAAC,UAAU,GAAG,SAAS,CAAA;QAC3B,IAAI,CAAC,IAAI,GAAG,GAAG,CAAA;QACf,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAA;QACvB,IAAI,CAAC,yBAAyB,GAAG,IAAI,0CAAwB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAA;IACjG,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,MAAqC;QAC5D,MAAM,eAAe,GAAgC,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAA;QAE9F,OAAO;YACH,GAAG,eAAe;YAClB,UAAU,EAAE,SAAS;SACxB,CAAA;IACL,CAAC;IAED,sBAAsB,CAAC,QAAgB;QACnC,QAAQ,QAAQ,EAAE,CAAC;YACf,KAAK,QAAQ;gBACT,OAAO,EAAE,eAAe,EAAE,OAAO,EAAE,CAAA;YACvC,KAAK,OAAO;gBACR,OAAO,EAAE,eAAe,EAAE,OAAO,EAAE,CAAA;YACvC,KAAK,OAAO,CAAC;YACb,KAAK,QAAQ;gBACT,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,CAAA;YACzC;gBACI,OAAO,SAAS,CAAA;QACxB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,wBAAwB,CAC1B,MAAqC,EACrC,cAA8B,EAC9B,eAAgC,EAChC,gBAAyB,EACzB,gBAA0C,EAC1C,UAAmB,EACnB,UAAyB,EAAE,EAC3B,QAAsB,EAAE,EACxB,iBAAoD;QAEpD,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAA;QACzB,MAAM,kBAAkB,GAAG,EAAE,gBAAgB,EAAE,yBAAc,CAAC,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAA;QAElG,MAAM,YAAY,GAAG,SAAS,IAAI,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;QAExG,IAAI,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,MAAM,CAAA;QAEzD,sGAAsG;QACtG,gEAAgE;QAChE,IAAI,aAAa,IAAI,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACnD,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAA;QACpE,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACf,aAAa,GAAG,aAAa,EAAE,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAA;QAChE,CAAC;QAED,gDAAgD;QAChD,IAAI,iBAAiB,GAAG,YAAY;YAChC,CAAC,CAAC,MAAM,IAAI,CAAC,qBAAqB,CAAC,aAAa,IAAI,EAAE,EAAE,gBAAgB,CAAC;YACzE,CAAC,CAAC,EAAE,CAAA;QAER,6CAA6C;QAC7C,IAAI,iBAAiB,EAAE,CAAC;YACpB,KAAK,MAAM,IAAI,IAAI,iBAAiB,EAAE,CAAC;gBACnC,6DAA6D;gBAC7D,IAAI,mBAAmB,GAAoC,SAAS,CAAA;gBAEpE,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACpB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAA;oBACzD,MAAM,QAAQ,GAAG,uCAAmB,CAAC,GAAG,CAAC,CAAA;oBAEzC,IAAI,QAAQ,EAAE,CAAC;wBACX,mBAAmB,GAAG,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAA;oBACpD,CAAC;gBACL,CAAC;gBAED,MAAM,YAAY,GACd,IAAI,CAAC,IAAI,KAAK,MAAM;oBAChB,CAAC,CAAC,qCAAW,CAAC,IAAI;oBAClB,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;wBAC9C,CAAC,CAAC,qCAAW,CAAC,MAAM;wBACpB,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM;4BACpB,CAAC,CAAC,qCAAW,CAAC,IAAI;4BAClB,CAAC,CAAC,SAAS,CAAA;gBACvB,oCAAoC;gBACpC,MAAM,oBAAoB,GAAiC;oBACvD,IAAI,EAAE,IAAI,CAAC,YAAY;oBACvB,gBAAgB,EAAE,IAAI,CAAC,IAAI;oBAC3B,mBAAmB,EAAE,mBAAmB;oBACxC,IAAI,EAAE,YAAY;oBAClB,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC;oBAC/B,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC;iBAC9B,CAAA;gBACD,iBAAiB,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAA;YAChD,CAAC;QACL,CAAC;QACD,MAAM,oBAAoB,GAAG,iBAAiB,CAAC,MAAM,KAAK,CAAC,CAAA;QAE3D,MAAM,IAAI,GAA0C;YAChD,iBAAiB,EAAE;gBACf,eAAe,EAAE,eAAe;gBAChC,cAAc,EAAE;oBACZ,gBAAgB,EAAE;wBACd,OAAO,EAAE,aAAa;wBACtB,uBAAuB,EACnB,cAAc,CAAC,WAAW,IAAI,cAAc,CAAC,gBAAgB;4BACzD,CAAC,CAAC;gCACI,WAAW,EAAE;oCACT,WAAW,EAAE,cAAc,CAAC,WAAW;oCACvC,QAAQ,EAAE;wCACN,IAAI,EAAE,cAAc,CAAC,IAAI;wCACzB,mBAAmB,EAAE,cAAc,CAAC,mBAAmB;wCACvD,gBAAgB,EAAE,cAAc,CAAC,gBAAgB;qCACpD;oCACD,iBAAiB,EAAE,oBAAoB,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;oCACvE,oBAAoB,EAAE,oBAAoB;oCAC1C,GAAG,kBAAkB;iCACxB;gCACD,KAAK;gCACL,QAAQ,EAAE,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,QAAQ,CAAC;6BAC1D;4BACH,CAAC,CAAC;gCACI,KAAK;gCACL,WAAW,EAAE;oCACT,iBAAiB,EAAE,oBAAoB,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;oCACvE,oBAAoB,EAAE,oBAAoB;oCAC1C,GAAG,kBAAkB;iCACxB;gCACD,QAAQ,EAAE,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,QAAQ,CAAC;6BAC1D;wBACX,UAAU,EAAE,cAAc,CAAC,UAAU;wBACrC,MAAM,EAAE,KAAK;qBAChB;iBACJ;gBACD,gBAAgB;gBAChB,OAAO;aACV;YACD,UAAU;SACb,CAAA;QAED,OAAO,IAAI,CAAA;IACf,CAAC;IAED,qBAAqB;IACrB,KAAK,CAAC,sBAAsB,CACxB,KAA0E;QAE1E,MAAM,EAAE,YAAY,EAAE,sBAAsB,EAAE,WAAW,EAAE,GAAG,KAAK,CAAA;QAEnE,IAAI,sBAAsB,EAAE,GAAG,KAAK,SAAS,EAAE,CAAC;YAC5C,OAAM;QACV,CAAC;QACD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAA;QAE3E,OAAO,YAAY;YACf,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,sBAAsB,CACjD,YAAY;YACZ,2HAA2H;YAC3H,iEAAiE;YACjE,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,yBAAyB,CAAC,oBAAoB,CACrE;YACH,CAAC,CAAC,SAAS,CAAA;IACnB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,eAAe,CAAC,GAAW;QAC7B,iFAAiF;QACjF,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;QACrE,IAAI,kBAAkB,EAAE,CAAC;YACrB,OAAO,kBAAkB,CAAA;QAC7B,CAAC;QACD,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;YACxE,OAAO,+BAAY,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,OAAO,CAAC,CAAA;QACnD,CAAC;QAAC,MAAM,CAAC;YACL,OAAM;QACV,CAAC;IACL,CAAC;IAED,KAAK,CAAC,qBAAqB,CACvB,MAAc,EACd,gBAA0C;QAE1C,MAAM,6BAA6B,GAAG,MAAM,6DAA6B,CAAC,WAAW,EAAE,CAAA;QACvF,IAAI,CAAC,6BAA6B,CAAC,SAAS,IAAI,gBAAgB,EAAE,CAAC;YAC/D,MAAM,gBAAgB,CAAC,gBAAgB,CAAC;gBACpC,IAAI,EAAE,8LAA8L;gBACpM,OAAO,EAAE;oBACL;wBACI,EAAE,EAAE,0DAAuC;wBAC3C,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,UAAU;wBAChB,kBAAkB,EAAE,KAAK;wBACzB,MAAM,EAAE,MAAM;qBACjB;iBACJ;aACJ,CAAC,CAAA;YACF,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,qBAAqB,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,6BAA6B,CAAC,CAAA;QACrG,qBAAqB,GAAG,qBAAqB,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QAC5F,KAAK,MAAM,gBAAgB,IAAI,qBAAqB,EAAE,CAAC;YACnD,IAAI,gBAAgB,CAAC,IAAI,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,GAAG,6BAAqB,EAAE,CAAC;gBAChF,gBAAgB,CAAC,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,6BAAqB,CAAC,CAAA;gBACjF,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,gCAAgC,gBAAgB,CAAC,gBAAgB,GAAG,CAAC,CAAA;YAC7F,CAAC;QACL,CAAC;QAED,OAAO,qBAAqB,CAAA;IAChC,CAAC;IAED,KAAK,CAAC,uBAAuB,CACzB,MAAc,EACd,6BAA4D;QAE5D,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,6BAA6B,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;YACtF,MAAM,qBAAqB,GAAmC,EAAE,CAAA;YAChE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,OAAO,qBAAqB,CAAA;YAChC,CAAC;YAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAA;gBAC3C,MAAM,YAAY,GAAG;oBACjB,IAAI;oBACJ,gBAAgB,EAAE,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC;oBACrE,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;oBAChC,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC;iBAC/B,CAAA;gBAED,IAAI,KAAK,CAAC,mBAAmB,IAAI,KAAK,CAAC,mBAAmB,KAAK,SAAS,EAAE,CAAC;oBACvE,qBAAqB,CAAC,IAAI,CAAC;wBACvB,GAAG,YAAY;wBACf,mBAAmB,EAAE;4BACjB,YAAY,EAAE,KAAK,CAAC,mBAAmB;yBAC1C;wBACD,IAAI,EAAE,qCAAW,CAAC,SAAS;qBAC9B,CAAC,CAAA;gBACN,CAAC;qBAAM,CAAC;oBACJ,qBAAqB,CAAC,IAAI,CAAC;wBACvB,GAAG,YAAY;wBACf,IAAI,EAAE,qCAAW,CAAC,SAAS;qBAC9B,CAAC,CAAA;gBACN,CAAC;YACL,CAAC;YAED,OAAO,qBAAqB,CAAA;QAChC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,gEAAgE,CAAC,EAAE,CAAC,CAAA;YACxF,OAAO,EAAE,CAAA;QACb,CAAC;IACL,CAAC;;AA9QL,8DA+QC"}

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

@@ -0,0 +1,20 @@
+import { ContextCommand, ContextCommandGroup } from '@aws/language-server-runtimes/protocol';
+import { Disposable } from 'vscode-languageclient/node';
+import { Chat, Logging, Lsp, Workspace } from '@aws/language-server-runtimes/server-interface';
+import { ContextCommandItem } from 'local-indexing';
+export declare class ContextCommandsProvider implements Disposable {
+    private readonly logging;
+    private readonly chat;
+    private readonly workspace;
+    private readonly lsp;
+    private promptFileWatcher?;
+    private cachedContextCommands?;
+    constructor(logging: Logging, chat: Chat, workspace: Workspace, lsp: Lsp);
+    updateContextCommandsOnFileChange(filePaths: string[], isAdd: boolean): Promise<void>;
+    registerPromptFileWatcher(): void;
+    getUserPromptsCommand(): Promise<ContextCommand[]>;
+    processContextCommandUpdate(items: ContextCommandItem[]): Promise<void>;
+    mapContextCommandItems(items: ContextCommandItem[]): Promise<ContextCommandGroup[]>;
+    maybeUpdateCodeSymbols(): Promise<void>;
+    dispose(): void;
+}

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

@@ -0,0 +1,206 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextCommandsProvider = void 0;
+const path = require("path");
+const chokidar_1 = require("chokidar");
+const contextUtils_1 = require("./contextUtils");
+const localProjectContextController_1 = require("../../../shared/localProjectContextController");
+const vscode_uri_1 = require("vscode-uri");
+class ContextCommandsProvider {
+    logging;
+    chat;
+    workspace;
+    lsp;
+    promptFileWatcher;
+    cachedContextCommands;
+    constructor(logging, chat, workspace, lsp) {
+        this.logging = logging;
+        this.chat = chat;
+        this.workspace = workspace;
+        this.lsp = lsp;
+        this.registerPromptFileWatcher();
+        this.lsp.workspace.onDidCreateFiles(async (event) => {
+            const filePaths = event.files.map(file => vscode_uri_1.URI.parse(file.uri).fsPath);
+            await this.updateContextCommandsOnFileChange(filePaths, true);
+        });
+        this.lsp.workspace.onDidDeleteFiles(async (event) => {
+            const filePaths = event.files.map(file => vscode_uri_1.URI.parse(file.uri).fsPath);
+            await this.updateContextCommandsOnFileChange(filePaths, false);
+        });
+        this.lsp.workspace.onDidRenameFiles(async (event) => {
+            const oldFiles = event.files.map(file => vscode_uri_1.URI.parse(file.oldUri).fsPath);
+            await this.updateContextCommandsOnFileChange(oldFiles, false);
+            const newFiles = event.files.map(file => vscode_uri_1.URI.parse(file.newUri).fsPath);
+            await this.updateContextCommandsOnFileChange(newFiles, true);
+        });
+    }
+    async updateContextCommandsOnFileChange(filePaths, isAdd) {
+        try {
+            const localProjectContextController = await localProjectContextController_1.LocalProjectContextController.getInstance();
+            const shouldUpdate = await localProjectContextController.shouldUpdateContextCommand(filePaths, isAdd);
+            if (shouldUpdate) {
+                const contextItems = await localProjectContextController.getContextCommandItems();
+                await this.processContextCommandUpdate(contextItems);
+            }
+        }
+        catch (error) {
+            this.logging.error(`Error updating context commands on file change: ${error}`);
+        }
+    }
+    registerPromptFileWatcher() {
+        this.promptFileWatcher = (0, chokidar_1.watch)((0, contextUtils_1.getUserPromptsDirectory)(), {
+            persistent: true,
+            interval: 2000,
+            ignoreInitial: true,
+        });
+        this.promptFileWatcher.on('add', async () => {
+            await this.processContextCommandUpdate(this.cachedContextCommands ?? []);
+        });
+        this.promptFileWatcher.on('unlink', async () => {
+            await this.processContextCommandUpdate(this.cachedContextCommands ?? []);
+        });
+    }
+    async getUserPromptsCommand() {
+        const createPromptCommand = {
+            command: 'Create a new prompt',
+            id: 'create-saved-prompt',
+            icon: 'list-add',
+        };
+        try {
+            const userPromptsDirectory = (0, contextUtils_1.getUserPromptsDirectory)();
+            const directoryExists = await this.workspace.fs.exists(userPromptsDirectory);
+            if (directoryExists) {
+                const files = await this.workspace.fs.readdir(userPromptsDirectory);
+                const systemPromptFiles = files.filter(file => file.name.endsWith(contextUtils_1.promptFileExtension));
+                const promptCommands = systemPromptFiles.map(file => ({
+                    command: path.basename(file.name, contextUtils_1.promptFileExtension),
+                    icon: 'magic',
+                    label: 'file',
+                    id: 'prompt',
+                    route: [userPromptsDirectory, file.name],
+                }));
+                promptCommands.push(createPromptCommand);
+                return promptCommands;
+            }
+            else {
+                return [createPromptCommand];
+            }
+        }
+        catch (e) {
+            this.logging.warn(`Error reading user prompts directory: ${e}`);
+            return [createPromptCommand];
+        }
+    }
+    async processContextCommandUpdate(items) {
+        const allItems = await this.mapContextCommandItems(items);
+        this.chat.sendContextCommands({ contextCommandGroups: allItems });
+        this.cachedContextCommands = items;
+    }
+    async mapContextCommandItems(items) {
+        const folderCmds = [];
+        const folderCmdGroup = {
+            command: 'Folders',
+            children: [
+                {
+                    groupName: 'Folders',
+                    commands: folderCmds,
+                },
+            ],
+            description: 'Add all files in a folder to context',
+            icon: 'folder',
+        };
+        const fileCmds = [];
+        const fileCmdGroup = {
+            command: 'Files',
+            children: [
+                {
+                    groupName: 'Files',
+                    commands: fileCmds,
+                },
+            ],
+            description: 'Add a file to context',
+            icon: 'file',
+        };
+        const codeCmds = [];
+        const codeCmdGroup = {
+            command: 'Code',
+            children: [
+                {
+                    groupName: 'Code',
+                    commands: codeCmds,
+                },
+            ],
+            description: 'Add code to context',
+            icon: 'code-block',
+        };
+        const promptCmds = [];
+        const promptCmdGroup = {
+            command: 'Prompts',
+            children: [
+                {
+                    groupName: 'Prompts',
+                    commands: promptCmds,
+                },
+            ],
+            description: 'Add a saved prompt to context',
+            icon: 'magic',
+        };
+        const workspaceCmd = {
+            command: '@workspace',
+            description: 'Reference all code in workspace.',
+        };
+        const commands = [workspaceCmd, folderCmdGroup, fileCmdGroup, codeCmdGroup, promptCmdGroup];
+        const allCommands = [
+            {
+                commands: commands,
+            },
+        ];
+        for (const item of items) {
+            const wsFolderName = path.basename(item.workspaceFolder);
+            let baseItem = {
+                command: path.basename(item.relativePath),
+                description: path.join(wsFolderName, item.relativePath),
+                route: [item.workspaceFolder, item.relativePath],
+                id: item.id,
+            };
+            if (item.type === 'file') {
+                fileCmds.push({
+                    ...baseItem,
+                    label: 'file',
+                    icon: 'file',
+                });
+            }
+            else if (item.type === 'folder') {
+                folderCmds.push({
+                    ...baseItem,
+                    label: 'folder',
+                    icon: 'folder',
+                });
+            }
+            else if (item.symbol) {
+                codeCmds.push({
+                    ...baseItem,
+                    command: item.symbol.name,
+                    description: `${item.symbol.kind}, ${path.join(wsFolderName, item.relativePath)}, L${item.symbol.range.start.line}-${item.symbol.range.end.line}`,
+                    label: 'code',
+                    icon: 'code-block',
+                });
+            }
+        }
+        const userPromptsItem = await this.getUserPromptsCommand();
+        promptCmds.push(...userPromptsItem);
+        return allCommands;
+    }
+    async maybeUpdateCodeSymbols() {
+        const needUpdate = await (await localProjectContextController_1.LocalProjectContextController.getInstance()).shouldUpdateContextCommandSymbolsOnce();
+        if (needUpdate) {
+            const items = await (await localProjectContextController_1.LocalProjectContextController.getInstance()).getContextCommandItems();
+            await this.processContextCommandUpdate(items);
+        }
+    }
+    dispose() {
+        void this.promptFileWatcher?.close();
+    }
+}
+exports.ContextCommandsProvider = ContextCommandsProvider;
+//# sourceMappingURL=contextCommandsProvider.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"contextCommandsProvider.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/context/contextCommandsProvider.ts"],"names":[],"mappings":";;;AAAA,6BAA4B;AAC5B,uCAA2C;AAU3C,iDAA6E;AAE7E,iGAA6F;AAC7F,2CAAgC;AAEhC,MAAa,uBAAuB;IAIX;IACA;IACA;IACA;IANb,iBAAiB,CAAY;IAC7B,qBAAqB,CAAuB;IACpD,YACqB,OAAgB,EAChB,IAAU,EACV,SAAoB,EACpB,GAAQ;QAHR,YAAO,GAAP,OAAO,CAAS;QAChB,SAAI,GAAJ,IAAI,CAAM;QACV,cAAS,GAAT,SAAS,CAAW;QACpB,QAAG,GAAH,GAAG,CAAK;QAEzB,IAAI,CAAC,yBAAyB,EAAE,CAAA;QAChC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,KAAK,EAAC,KAAK,EAAC,EAAE;YAC9C,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAG,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;YACrE,MAAM,IAAI,CAAC,iCAAiC,CAAC,SAAS,EAAE,IAAI,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,KAAK,EAAC,KAAK,EAAC,EAAE;YAC9C,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAG,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;YACrE,MAAM,IAAI,CAAC,iCAAiC,CAAC,SAAS,EAAE,KAAK,CAAC,CAAA;QAClE,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,KAAK,EAAC,KAAK,EAAC,EAAE;YAC9C,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAA;YACvE,MAAM,IAAI,CAAC,iCAAiC,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;YAC7D,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAA;YACvE,MAAM,IAAI,CAAC,iCAAiC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QAChE,CAAC,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,iCAAiC,CAAC,SAAmB,EAAE,KAAc;QACvE,IAAI,CAAC;YACD,MAAM,6BAA6B,GAAG,MAAM,6DAA6B,CAAC,WAAW,EAAE,CAAA;YACvF,MAAM,YAAY,GAAG,MAAM,6BAA6B,CAAC,0BAA0B,CAAC,SAAS,EAAE,KAAK,CAAC,CAAA;YACrG,IAAI,YAAY,EAAE,CAAC;gBACf,MAAM,YAAY,GAAG,MAAM,6BAA6B,CAAC,sBAAsB,EAAE,CAAA;gBACjF,MAAM,IAAI,CAAC,2BAA2B,CAAC,YAAY,CAAC,CAAA;YACxD,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,mDAAmD,KAAK,EAAE,CAAC,CAAA;QAClF,CAAC;IACL,CAAC;IAED,yBAAyB;QACrB,IAAI,CAAC,iBAAiB,GAAG,IAAA,gBAAK,EAAC,IAAA,sCAAuB,GAAE,EAAE;YACtD,UAAU,EAAE,IAAI;YAChB,QAAQ,EAAE,IAAI;YACd,aAAa,EAAE,IAAI;SACtB,CAAC,CAAA;QAEF,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,qBAAqB,IAAI,EAAE,CAAC,CAAA;QAC5E,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,qBAAqB,IAAI,EAAE,CAAC,CAAA;QAC5E,CAAC,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,qBAAqB;QACvB,MAAM,mBAAmB,GAAG;YACxB,OAAO,EAAE,qBAAqB;YAC9B,EAAE,EAAE,qBAAqB;YACzB,IAAI,EAAE,UAAU;SACnB,CAAA;QACD,IAAI,CAAC;YACD,MAAM,oBAAoB,GAAG,IAAA,sCAAuB,GAAE,CAAA;YACtD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAA;YAC5E,IAAI,eAAe,EAAE,CAAC;gBAClB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAA;gBACnE,MAAM,iBAAiB,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,kCAAmB,CAAC,CAAC,CAAA;gBACvF,MAAM,cAAc,GAAG,iBAAiB,CAAC,GAAG,CACxC,IAAI,CAAC,EAAE,CACH,CAAC;oBACG,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,kCAAmB,CAAC;oBACtD,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,MAAM;oBACb,EAAE,EAAE,QAAQ;oBACZ,KAAK,EAAE,CAAC,oBAAoB,EAAE,IAAI,CAAC,IAAI,CAAC;iBAC3C,CAAmB,CAC3B,CAAA;gBACD,cAAc,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;gBACxC,OAAO,cAAc,CAAA;YACzB,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,mBAAmB,CAAC,CAAA;YAChC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,yCAAyC,CAAC,EAAE,CAAC,CAAA;YAC/D,OAAO,CAAC,mBAAmB,CAAC,CAAA;QAChC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,KAA2B;QACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAA;QACzD,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,EAAE,oBAAoB,EAAE,QAAQ,EAAE,CAAC,CAAA;QACjE,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,KAA2B;QACpD,MAAM,UAAU,GAAqB,EAAE,CAAA;QACvC,MAAM,cAAc,GAAmB;YACnC,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE;gBACN;oBACI,SAAS,EAAE,SAAS;oBACpB,QAAQ,EAAE,UAAU;iBACvB;aACJ;YACD,WAAW,EAAE,sCAAsC;YACnD,IAAI,EAAE,QAAQ;SACjB,CAAA;QAED,MAAM,QAAQ,GAAqB,EAAE,CAAA;QACrC,MAAM,YAAY,GAAmB;YACjC,OAAO,EAAE,OAAO;YAChB,QAAQ,EAAE;gBACN;oBACI,SAAS,EAAE,OAAO;oBAClB,QAAQ,EAAE,QAAQ;iBACrB;aACJ;YACD,WAAW,EAAE,uBAAuB;YACpC,IAAI,EAAE,MAAM;SACf,CAAA;QAED,MAAM,QAAQ,GAAqB,EAAE,CAAA;QACrC,MAAM,YAAY,GAAmB;YACjC,OAAO,EAAE,MAAM;YACf,QAAQ,EAAE;gBACN;oBACI,SAAS,EAAE,MAAM;oBACjB,QAAQ,EAAE,QAAQ;iBACrB;aACJ;YACD,WAAW,EAAE,qBAAqB;YAClC,IAAI,EAAE,YAAY;SACrB,CAAA;QAED,MAAM,UAAU,GAAqB,EAAE,CAAA;QACvC,MAAM,cAAc,GAAmB;YACnC,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE;gBACN;oBACI,SAAS,EAAE,SAAS;oBACpB,QAAQ,EAAE,UAAU;iBACvB;aACJ;YACD,WAAW,EAAE,+BAA+B;YAC5C,IAAI,EAAE,OAAO;SAChB,CAAA;QACD,MAAM,YAAY,GAAG;YACjB,OAAO,EAAE,YAAY;YACrB,WAAW,EAAE,kCAAkC;SAClD,CAAA;QACD,MAAM,QAAQ,GAAG,CAAC,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,CAAC,CAAA;QAC3F,MAAM,WAAW,GAA0B;YACvC;gBACI,QAAQ,EAAE,QAAQ;aACrB;SACJ,CAAA;QAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;YACxD,IAAI,QAAQ,GAAG;gBACX,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC;gBACzC,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC;gBACvD,KAAK,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,YAAY,CAAC;gBAChD,EAAE,EAAE,IAAI,CAAC,EAAE;aACd,CAAA;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACvB,QAAQ,CAAC,IAAI,CAAC;oBACV,GAAG,QAAQ;oBACX,KAAK,EAAE,MAAM;oBACb,IAAI,EAAE,MAAM;iBACf,CAAC,CAAA;YACN,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAChC,UAAU,CAAC,IAAI,CAAC;oBACZ,GAAG,QAAQ;oBACX,KAAK,EAAE,QAAQ;oBACf,IAAI,EAAE,QAAQ;iBACjB,CAAC,CAAA;YACN,CAAC;iBAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC;oBACV,GAAG,QAAQ;oBACX,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;oBACzB,WAAW,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE;oBACjJ,KAAK,EAAE,MAAM;oBACb,IAAI,EAAE,YAAY;iBACrB,CAAC,CAAA;YACN,CAAC;QACL,CAAC;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAA;QAC1D,UAAU,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAA;QACnC,OAAO,WAAW,CAAA;IACtB,CAAC;IAED,KAAK,CAAC,sBAAsB;QACxB,MAAM,UAAU,GAAG,MAAM,CACrB,MAAM,6DAA6B,CAAC,WAAW,EAAE,CACpD,CAAC,qCAAqC,EAAE,CAAA;QACzC,IAAI,UAAU,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,6DAA6B,CAAC,WAAW,EAAE,CAAC,CAAC,sBAAsB,EAAE,CAAA;YAChG,MAAM,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAA;QACjD,CAAC;IACL,CAAC;IAED,OAAO;QACH,KAAK,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAA;IACxC,CAAC;CACJ;AA7MD,0DA6MC"}

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

@@ -0,0 +1,27 @@
+export interface ContextInfo {
+    contextCount: {
+        fileContextCount: number;
+        folderContextCount: number;
+        promptContextCount: number;
+        ruleContextCount: number;
+        codeContextCount: number;
+    };
+    contextLength: {
+        fileContextLength: number;
+        ruleContextLength: number;
+        promptContextLength: number;
+        codeContextLength: number;
+    };
+}
+export declare const initialContextInfo: ContextInfo;
+export declare const promptFileExtension = ".md";
+export declare const additionalContentInnerContextLimit = 8192;
+export declare const additionalContentNameLimit = 1024;
+export declare const getUserPromptsDirectory: () => string;
+/**
+ * Creates a secure file path for a new prompt file.
+ *
+ * @param promptName - The user-provided name for the prompt
+ * @returns A sanitized file path within the user prompts directory
+ */
+export declare const getNewPromptFilePath: (promptName: string) => string;

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

@@ -0,0 +1,44 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getNewPromptFilePath = exports.getUserPromptsDirectory = exports.additionalContentNameLimit = exports.additionalContentInnerContextLimit = exports.promptFileExtension = exports.initialContextInfo = void 0;
+const path_1 = require("@aws/lsp-core/out/util/path");
+const path = require("path");
+const text_1 = require("@aws/lsp-core/out/util/text");
+exports.initialContextInfo = {
+    contextCount: {
+        fileContextCount: 0,
+        folderContextCount: 0,
+        promptContextCount: 0,
+        ruleContextCount: 0,
+        codeContextCount: 0,
+    },
+    contextLength: {
+        fileContextLength: 0,
+        ruleContextLength: 0,
+        promptContextLength: 0,
+        codeContextLength: 0,
+    },
+};
+exports.promptFileExtension = '.md';
+exports.additionalContentInnerContextLimit = 8192;
+exports.additionalContentNameLimit = 1024;
+const getUserPromptsDirectory = () => {
+    return path.join((0, path_1.getUserHomeDir)(), '.aws', 'amazonq', 'prompts');
+};
+exports.getUserPromptsDirectory = getUserPromptsDirectory;
+/**
+ * Creates a secure file path for a new prompt file.
+ *
+ * @param promptName - The user-provided name for the prompt
+ * @returns A sanitized file path within the user prompts directory
+ */
+const getNewPromptFilePath = (promptName) => {
+    const userPromptsDirectory = (0, exports.getUserPromptsDirectory)();
+    const trimmedName = promptName?.trim() || '';
+    const truncatedName = trimmedName.slice(0, 100);
+    const safePromptName = truncatedName ? (0, text_1.sanitizeFilename)(path.basename(truncatedName)) : 'default';
+    const finalPath = path.join(userPromptsDirectory, `${safePromptName}${exports.promptFileExtension}`);
+    return finalPath;
+};
+exports.getNewPromptFilePath = getNewPromptFilePath;
+//# sourceMappingURL=contextUtils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"contextUtils.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/context/contextUtils.ts"],"names":[],"mappings":";;;AAAA,sDAA4D;AAC5D,6BAA4B;AAC5B,sDAA8D;AAkBjD,QAAA,kBAAkB,GAAgB;IAC3C,YAAY,EAAE;QACV,gBAAgB,EAAE,CAAC;QACnB,kBAAkB,EAAE,CAAC;QACrB,kBAAkB,EAAE,CAAC;QACrB,gBAAgB,EAAE,CAAC;QACnB,gBAAgB,EAAE,CAAC;KACtB;IACD,aAAa,EAAE;QACX,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,mBAAmB,EAAE,CAAC;QACtB,iBAAiB,EAAE,CAAC;KACvB;CACJ,CAAA;AAEY,QAAA,mBAAmB,GAAG,KAAK,CAAA;AAC3B,QAAA,kCAAkC,GAAG,IAAI,CAAA;AACzC,QAAA,0BAA0B,GAAG,IAAI,CAAA;AAEvC,MAAM,uBAAuB,GAAG,GAAW,EAAE;IAChD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAA,qBAAc,GAAE,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAA;AACpE,CAAC,CAAA;AAFY,QAAA,uBAAuB,2BAEnC;AAED;;;;;GAKG;AACI,MAAM,oBAAoB,GAAG,CAAC,UAAkB,EAAU,EAAE;IAC/D,MAAM,oBAAoB,GAAG,IAAA,+BAAuB,GAAE,CAAA;IAEtD,MAAM,WAAW,GAAG,UAAU,EAAE,IAAI,EAAE,IAAI,EAAE,CAAA;IAE5C,MAAM,aAAa,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA;IAE/C,MAAM,cAAc,GAAG,aAAa,CAAC,CAAC,CAAC,IAAA,uBAAgB,EAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;IAEjG,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,cAAc,GAAG,2BAAmB,EAAE,CAAC,CAAA;IAE5F,OAAO,SAAS,CAAA;AACpB,CAAC,CAAA;AAZY,QAAA,oBAAoB,wBAYhC"}

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

@@ -0,0 +1,10 @@
+type AgenticChatErrorCode = 'QModelResponse' | 'AmazonQServiceManager' | 'FailedResult' | 'MaxAgentLoopIterations' | 'InputTooLong' | 'PromptCharacterLimit';
+export declare const customerFacingErrorCodes: AgenticChatErrorCode[];
+export declare class AgenticChatError extends Error {
+    readonly code: AgenticChatErrorCode;
+    readonly requestId?: string | undefined;
+    constructor(message: string, code: AgenticChatErrorCode, cause?: Error, requestId?: string | undefined);
+}
+export declare function wrapErrorWithCode(error: unknown, code: AgenticChatErrorCode): AgenticChatError;
+export declare function isInputTooLongError(error: unknown): boolean;
+export {};

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

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgenticChatError = exports.customerFacingErrorCodes = void 0;
+exports.wrapErrorWithCode = wrapErrorWithCode;
+exports.isInputTooLongError = isInputTooLongError;
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+exports.customerFacingErrorCodes = [
+    'QModelResponse',
+    'MaxAgentLoopIterations',
+    'InputTooLong',
+    'PromptCharacterLimit',
+];
+class AgenticChatError extends Error {
+    code;
+    requestId;
+    constructor(message, code, cause, requestId) {
+        super(message, { cause: cause });
+        this.code = code;
+        this.requestId = requestId;
+    }
+}
+exports.AgenticChatError = AgenticChatError;
+function wrapErrorWithCode(error, code) {
+    if (error instanceof codewhisperer_streaming_1.CodeWhispererStreamingServiceException) {
+        return new AgenticChatError(error.message, code, error, error.$metadata?.requestId);
+    }
+    if (error instanceof Error) {
+        return new AgenticChatError(error.message, code, error);
+    }
+    return new AgenticChatError(String(error), code);
+}
+function isInputTooLongError(error) {
+    if (error instanceof AgenticChatError && error.code === 'InputTooLong') {
+        return true;
+    }
+    if (error instanceof Error) {
+        //  This is fragile (breaks if the backend changes their error message wording)
+        return error.message.includes('Input is too long');
+    }
+    return false;
+}
+//# sourceMappingURL=errors.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../src/language-server/agenticChat/errors.ts"],"names":[],"mappings":";;;AA2BA,8CASC;AAED,kDAWC;AAjDD,2EAAsF;AAUzE,QAAA,wBAAwB,GAA2B;IAC5D,gBAAgB;IAChB,wBAAwB;IACxB,cAAc;IACd,sBAAsB;CACzB,CAAA;AACD,MAAa,gBAAiB,SAAQ,KAAK;IAGnB;IAEA;IAJpB,YACI,OAAe,EACC,IAA0B,EAC1C,KAAa,EACG,SAAkB;QAElC,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;QAJhB,SAAI,GAAJ,IAAI,CAAsB;QAE1B,cAAS,GAAT,SAAS,CAAS;IAGtC,CAAC;CACJ;AATD,4CASC;AAED,SAAgB,iBAAiB,CAAC,KAAc,EAAE,IAA0B;IACxE,IAAI,KAAK,YAAY,gEAAsC,EAAE,CAAC;QAC1D,OAAO,IAAI,gBAAgB,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;IACvF,CAAC;IAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QACzB,OAAO,IAAI,gBAAgB,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,CAAA;IAC3D,CAAC;IACD,OAAO,IAAI,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAA;AACpD,CAAC;AAED,SAAgB,mBAAmB,CAAC,KAAc;IAC9C,IAAI,KAAK,YAAY,gBAAgB,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;QACrE,OAAO,IAAI,CAAA;IACf,CAAC;IAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QACzB,+EAA+E;QAC/E,OAAO,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAA;IACtD,CAAC;IAED,OAAO,KAAK,CAAA;AAChB,CAAC"}