@aws/lsp-codewhisperer 0.0.29 → 0.0.31

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

@@ -0,0 +1,96 @@
+"use strict";
+/**
+ * Copied from ../qChatServer.ts for the purpose of developing a divergent implementation.
+ * Will be deleted or merged.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QAgenticChatServer = void 0;
+const agenticChatController_1 = require("./agenticChatController");
+const chatSessionManagementService_1 = require("../chat/chatSessionManagementService");
+const quickActions_1 = require("../chat/quickActions");
+const telemetryService_1 = require("../../shared/telemetry/telemetryService");
+const telemetryUtils_1 = require("../../shared/telemetryUtils");
+const AmazonQTokenServiceManager_1 = require("../../shared/amazonQServiceManager/AmazonQTokenServiceManager");
+const utils_1 = require("../../shared/utils");
+const errors_1 = require("../../shared/amazonQServiceManager/errors");
+const QAgenticChatServer = 
+// prettier-ignore
+() => features => {
+    const { chat, credentialsProvider, telemetry, logging, lsp, runtime } = features;
+    let amazonQServiceManager;
+    let chatController;
+    let chatSessionManagementService;
+    let telemetryService;
+    lsp.addInitializer((params) => {
+        return {
+            capabilities: {},
+            awsServerCapabilities: {
+                chatOptions: {
+                    quickActions: {
+                        quickActionsCommandGroups: [
+                            {
+                                commands: [quickActions_1.HELP_QUICK_ACTION, quickActions_1.CLEAR_QUICK_ACTION],
+                            },
+                        ],
+                    },
+                },
+            },
+        };
+    });
+    const updateConfigurationHandler = async () => {
+        const amazonQConfig = await amazonQServiceManager.handleDidChangeConfiguration();
+        chatController.updateConfiguration(amazonQConfig);
+    };
+    lsp.onInitialized(async () => {
+        // Initialize service manager and inject it to chatSessionManagementService to pass it down
+        amazonQServiceManager = AmazonQTokenServiceManager_1.AmazonQTokenServiceManager.getInstance(features);
+        chatSessionManagementService = chatSessionManagementService_1.ChatSessionManagementService
+            .getInstance()
+            .withAmazonQServiceManager(amazonQServiceManager);
+        telemetryService = new telemetryService_1.TelemetryService(amazonQServiceManager, credentialsProvider, telemetry, logging);
+        const clientParams = (0, utils_1.safeGet)(lsp.getClientInitializeParams(), new errors_1.AmazonQServiceInitializationError('TelemetryService initialized before LSP connection was initialized.'));
+        telemetryService.updateUserContext((0, telemetryUtils_1.makeUserContextObject)(clientParams, runtime.platform, 'CHAT'));
+        chatController = new agenticChatController_1.AgenticChatController(chatSessionManagementService, features, telemetryService, amazonQServiceManager);
+        await updateConfigurationHandler();
+    });
+    lsp.didChangeConfiguration(updateConfigurationHandler);
+    chat.onTabAdd(params => {
+        logging.log(`Adding tab: ${params.tabId}`);
+        return chatController.onTabAdd(params);
+    });
+    chat.onTabChange(params => {
+        logging.log(`Changing to tab: ${params.tabId}`);
+        return chatController.onTabChange(params);
+    });
+    chat.onTabRemove(params => {
+        logging.log(`Removing tab: ${params.tabId}`);
+        return chatController.onTabRemove(params);
+    });
+    chat.onEndChat((...params) => {
+        logging.log('Received end chat request');
+        return chatController.onEndChat(...params);
+    });
+    chat.onChatPrompt((...params) => {
+        logging.log('Received chat prompt');
+        return chatController.onChatPrompt(...params);
+    });
+    chat.onInlineChatPrompt((...params) => {
+        logging.log('Received inline chat prompt');
+        return chatController.onInlineChatPrompt(...params);
+    });
+    chat.onQuickAction((...params) => {
+        return chatController.onQuickAction(...params);
+    });
+    chat.onSendFeedback(params => {
+        return chatController.onSendFeedback(params);
+    });
+    chat.onCodeInsertToCursorPosition(params => {
+        return chatController.onCodeInsertToCursorPosition(params);
+    });
+    logging.log('Q Chat server has been initialized');
+    return () => {
+        chatController?.dispose();
+    };
+};
+exports.QAgenticChatServer = QAgenticChatServer;
+//# sourceMappingURL=qAgenticChatServer.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"qAgenticChatServer.js","sourceRoot":"","sources":["../../../src/language-server/agenticChat/qAgenticChatServer.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAGH,mEAA+D;AAC/D,uFAAmF;AACnF,uDAA4E;AAC5E,8EAA0E;AAC1E,gEAAmE;AACnE,8GAA0G;AAC1G,8CAA4C;AAC5C,sEAA6F;AAEtF,MAAM,kBAAkB;AAC3B,kBAAkB;AAClB,GAAW,EAAE,CAAC,QAAQ,CAAC,EAAE;IACrB,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAA;IAEhF,IAAI,qBAAiD,CAAA;IACrD,IAAI,cAAqC,CAAA;IACzC,IAAI,4BAA0D,CAAA;IAC9D,IAAI,gBAAkC,CAAC;IAEvC,GAAG,CAAC,cAAc,CAAC,CAAC,MAAwB,EAAE,EAAE;QAC5C,OAAO;YACH,YAAY,EAAE,EAAE;YAChB,qBAAqB,EAAE;gBACnB,WAAW,EAAE;oBACT,YAAY,EAAE;wBACV,yBAAyB,EAAE;4BACvB;gCACI,QAAQ,EAAE,CAAC,gCAAiB,EAAE,iCAAkB,CAAC;6BACpD;yBACJ;qBACJ;iBACJ;aACJ;SACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,MAAM,0BAA0B,GAAG,KAAK,IAAI,EAAE;QAC1C,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,4BAA4B,EAAE,CAAA;QAChF,cAAc,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAA;IACrD,CAAC,CAAA;IAED,GAAG,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE;QACzB,2FAA2F;QAC3F,qBAAqB,GAAG,uDAA0B,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;QACxE,4BAA4B,GAAG,2DAA4B;aACtD,WAAW,EAAE;aACb,yBAAyB,CAAC,qBAAqB,CAAC,CAAA;QAErD,gBAAgB,GAAG,IAAI,mCAAgB,CACnC,qBAAqB,EACrB,mBAAmB,EACnB,SAAS,EACT,OAAO,CACV,CAAA;QAED,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,CAAC,yBAAyB,EAAE,EAAE,IAAI,0CAAiC,CAC/F,qEAAqE,CAAC,CAAC,CAAA;QAE3E,gBAAgB,CAAC,iBAAiB,CAAC,IAAA,sCAAqB,EAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;QAEjG,cAAc,GAAG,IAAI,6CAAqB,CAAC,4BAA4B,EAAE,QAAQ,EAAE,gBAAgB,EAAE,qBAAqB,CAAC,CAAA;QAE3H,MAAM,0BAA0B,EAAE,CAAA;IACtC,CAAC,CAAC,CAAA;IACF,GAAG,CAAC,sBAAsB,CAAC,0BAA0B,CAAC,CAAA;IAEtD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;QACnB,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAE1C,OAAO,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE;QACtB,OAAO,CAAC,GAAG,CAAC,oBAAoB,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAE/C,OAAO,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE;QACtB,OAAO,CAAC,GAAG,CAAC,iBAAiB,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAE5C,OAAO,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QACzB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAA;QACxC,OAAO,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QAC5B,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAA;QACnC,OAAO,cAAc,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,kBAAkB,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QAClC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAA;QAC1C,OAAO,cAAc,CAAC,kBAAkB,CAAC,GAAG,MAAM,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QAC7B,OAAO,cAAc,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE;QACzB,OAAO,cAAc,CAAC,cAAc,CAAC,MAAM,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,4BAA4B,CAAC,MAAM,CAAC,EAAE;QACvC,OAAO,cAAc,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAA;IAC9D,CAAC,CAAC,CAAA;IAEF,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAA;IAEjD,OAAO,GAAG,EAAE;QACR,cAAc,EAAE,OAAO,EAAE,CAAA;IAC7B,CAAC,CAAA;AACL,CAAC,CAAA;AA3GQ,QAAA,kBAAkB,sBA2G1B"}

package/out/language-server/agenticChat/tools/fsRead.d.ts

@@ -0,0 +1,38 @@
+import { InvokeOutput } from './toolShared';
+import { Features } from '@aws/language-server-runtimes/server-interface/server';
+export interface FsReadParams {
+    path: string;
+    readRange?: number[];
+}
+export declare class FsRead {
+    private readonly logging;
+    private readonly workspace;
+    constructor(features: Pick<Features, 'workspace' | 'logging'> & Partial<Features>);
+    invoke(params: FsReadParams): Promise<InvokeOutput>;
+    private readFile;
+    private handleFileRange;
+    private parseLineRange;
+    private enforceMaxSize;
+    private createOutput;
+    getSpec(): {
+        readonly name: "fsRead";
+        readonly description: "A tool for reading a file. \n* This tool returns the contents of a file, and the optional `readRange` determines what range of lines will be read from the specified file.";
+        readonly inputSchema: {
+            readonly type: "object";
+            readonly parameters: {
+                readonly path: {
+                    readonly description: "Absolute path to a file, e.g. `/repo/file.py`.";
+                    readonly type: "string";
+                };
+                readonly readRange: {
+                    readonly description: "Optional parameter when reading files.\n* If none is given, the full file is shown. If provided, the file will be shown in the indicated line number range, e.g. [11, 12] will show lines 11 and 12. Indexing at 1 to start. Setting `[startLine, -1]` shows all lines from `startLine` to the end of the file.";
+                    readonly items: {
+                        readonly type: "integer";
+                    };
+                    readonly type: "array";
+                };
+            };
+            readonly required: readonly ["path"];
+        };
+    };
+}

package/out/language-server/agenticChat/tools/fsRead.js

@@ -0,0 +1,96 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FsRead = void 0;
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const path_1 = require("@aws/lsp-core/out/util/path");
+const toolShared_1 = require("./toolShared");
+class FsRead {
+    logging;
+    workspace;
+    constructor(features) {
+        this.logging = features.logging;
+        this.workspace = features.workspace;
+    }
+    async invoke(params) {
+        const path = (0, path_1.sanitize)(params.path);
+        const fileContents = await this.readFile(path);
+        this.logging.info(`Read file: ${path}, size: ${fileContents.length}`);
+        return this.handleFileRange(params, fileContents);
+    }
+    async readFile(filePath) {
+        this.logging.info(`Reading file: ${filePath}`);
+        return await this.workspace.fs.readFile(filePath);
+    }
+    handleFileRange(params, fullText) {
+        if (!params.readRange || params.readRange.length === 0) {
+            this.logging.log('No range provided. returning entire file.');
+            return this.createOutput(this.enforceMaxSize(fullText));
+        }
+        const lines = fullText.split('\n');
+        const [start, end] = this.parseLineRange(lines.length, params.readRange);
+        if (start > end) {
+            this.logging.error(`Invalid range: ${params.readRange.join('-')}`);
+            return this.createOutput('');
+        }
+        this.logging.log(`Reading file: ${params.path}, lines ${start + 1}-${end + 1}`);
+        const slice = lines.slice(start, end + 1).join('\n');
+        return this.createOutput(this.enforceMaxSize(slice));
+    }
+    parseLineRange(lineCount, range) {
+        const startIdx = range[0];
+        let endIdx = range.length >= 2 ? range[1] : undefined;
+        if (endIdx === undefined) {
+            endIdx = -1;
+        }
+        const convert = (i) => {
+            return i < 0 ? lineCount + i : i - 1;
+        };
+        const finalStart = Math.max(0, Math.min(lineCount - 1, convert(startIdx)));
+        const finalEnd = Math.max(0, Math.min(lineCount - 1, convert(endIdx)));
+        return [finalStart, finalEnd];
+    }
+    enforceMaxSize(content) {
+        const byteCount = Buffer.byteLength(content, 'utf8');
+        if (byteCount > toolShared_1.maxToolResponseSize) {
+            throw new Error(`This tool only supports reading ${toolShared_1.maxToolResponseSize} bytes at a time.
+                You tried to read ${byteCount} bytes. Try executing with fewer lines specified.`);
+        }
+        return content;
+    }
+    createOutput(content) {
+        return {
+            output: {
+                kind: 'text',
+                content: content,
+            },
+        };
+    }
+    getSpec() {
+        return {
+            name: 'fsRead',
+            description: 'A tool for reading a file. \n* This tool returns the contents of a file, and the optional `readRange` determines what range of lines will be read from the specified file.',
+            inputSchema: {
+                type: 'object',
+                parameters: {
+                    path: {
+                        description: 'Absolute path to a file, e.g. `/repo/file.py`.',
+                        type: 'string',
+                    },
+                    readRange: {
+                        description: 'Optional parameter when reading files.\n* If none is given, the full file is shown. If provided, the file will be shown in the indicated line number range, e.g. [11, 12] will show lines 11 and 12. Indexing at 1 to start. Setting `[startLine, -1]` shows all lines from `startLine` to the end of the file.',
+                        items: {
+                            type: 'integer',
+                        },
+                        type: 'array',
+                    },
+                },
+                required: ['path'],
+            },
+        };
+    }
+}
+exports.FsRead = FsRead;
+//# sourceMappingURL=fsRead.js.map

package/out/language-server/agenticChat/tools/fsRead.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fsRead.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/tools/fsRead.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,sDAAsD;AACtD,6CAAgE;AAUhE,MAAa,MAAM;IACE,OAAO,CAAqB;IAC5B,SAAS,CAAuB;IAEjD,YAAY,QAAqE;QAC7E,IAAI,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAA;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAA;IACvC,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,MAAoB;QACpC,MAAM,IAAI,GAAG,IAAA,eAAQ,EAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QAClC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;QAC9C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,IAAI,WAAW,YAAY,CAAC,MAAM,EAAE,CAAC,CAAA;QACrE,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,YAAY,CAAC,CAAA;IACrD,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,QAAgB;QACnC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAA;QAC9C,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;IACrD,CAAC;IAEO,eAAe,CAAC,MAAoB,EAAE,QAAgB;QAC1D,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAA;YAC7D,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAA;QAC3D,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAClC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAA;QACxE,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,kBAAkB,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YAClE,OAAO,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAA;QAChC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,MAAM,CAAC,IAAI,WAAW,KAAK,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC,CAAA;QAC/E,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpD,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA;IACxD,CAAC;IAEO,cAAc,CAAC,SAAiB,EAAE,KAAe;QACrD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAA;QACzB,IAAI,MAAM,GAAG,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;QAErD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,GAAG,CAAC,CAAC,CAAA;QACf,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,CAAS,EAAU,EAAE;YAClC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAA;QACxC,CAAC,CAAA;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;QAC1E,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;QACtE,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;IACjC,CAAC;IAEO,cAAc,CAAC,OAAe;QAClC,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;QACpD,IAAI,SAAS,GAAG,gCAAmB,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CACX,mCAAmC,gCAAmB;oCAClC,SAAS,mDAAmD,CACnF,CAAA;QACL,CAAC;QACD,OAAO,OAAO,CAAA;IAClB,CAAC;IAEO,YAAY,CAAC,OAAe;QAChC,OAAO;YACH,MAAM,EAAE;gBACJ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;aACnB;SACJ,CAAA;IACL,CAAC;IAEM,OAAO;QACV,OAAO;YACH,IAAI,EAAE,QAAQ;YACd,WAAW,EACP,4KAA4K;YAChL,WAAW,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACR,IAAI,EAAE;wBACF,WAAW,EAAE,gDAAgD;wBAC7D,IAAI,EAAE,QAAQ;qBACjB;oBACD,SAAS,EAAE;wBACP,WAAW,EACP,iTAAiT;wBACrT,KAAK,EAAE;4BACH,IAAI,EAAE,SAAS;yBAClB;wBACD,IAAI,EAAE,OAAO;qBAChB;iBACJ;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACrB;SACK,CAAA;IACd,CAAC;CACJ;AArGD,wBAqGC"}

package/out/language-server/agenticChat/tools/fsWrite.d.ts

@@ -0,0 +1,73 @@
+import { InvokeOutput } from './toolShared';
+import { Features } from '@aws/language-server-runtimes/server-interface/server';
+interface BaseParams {
+    path: string;
+}
+export interface CreateParams extends BaseParams {
+    command: 'create';
+    fileText?: string;
+    newStr?: string;
+}
+export interface StrReplaceParams extends BaseParams {
+    command: 'strReplace';
+    oldStr: string;
+    newStr: string;
+}
+export interface InsertParams extends BaseParams {
+    command: 'insert';
+    insertLine: number;
+    newStr: string;
+}
+export interface AppendParams extends BaseParams {
+    command: 'append';
+    newStr: string;
+}
+export type FsWriteParams = CreateParams | StrReplaceParams | InsertParams | AppendParams;
+export declare class FsWrite {
+    private readonly logging;
+    private readonly workspace;
+    constructor(features: Pick<Features, 'workspace' | 'logging'> & Partial<Features>);
+    invoke(params: FsWriteParams): Promise<InvokeOutput>;
+    private handleCreate;
+    private handleStrReplace;
+    private handleInsert;
+    private handleAppend;
+    private getCreateCommandText;
+    private escapeRegExp;
+    getSpec(): {
+        readonly name: "fsWrite";
+        readonly description: "A tool for creating and editing a file.\n * The `create` command will override the file at `path` if it already exists as a file, and otherwise create a new file\n * The `append` command will add content to the end of an existing file, automatically adding a newline if the file does not end with one. The file must exist.\n Notes for using the `strReplace` command:\n * The `oldStr` parameter should match EXACTLY one or more consecutive lines from the original file. Be mindful of whitespaces!\n * If the `oldStr` parameter is not unique in the file, the replacement will not be performed. Make sure to include enough context in `oldStr` to make it unique\n * The `newStr` parameter should contain the edited lines that should replace the `oldStr`. The `insert` command will insert `newStr` after `insertLine` and place it on its own line.";
+        readonly inputSchema: {
+            readonly type: "object";
+            readonly parameters: {
+                readonly command: {
+                    readonly type: "string";
+                    readonly enum: readonly ["create", "strReplace", "insert", "append"];
+                    readonly description: "The commands to run. Allowed options are: `create`, `strReplace`, `insert`, `append`.";
+                };
+                readonly fileText: {
+                    readonly description: "Required parameter of `create` command, with the content of the file to be created.";
+                    readonly type: "string";
+                };
+                readonly insertLine: {
+                    readonly description: "Required parameter of `insert` command. The `newStr` will be inserted AFTER the line `insertLine` of `path`.";
+                    readonly type: "integer";
+                };
+                readonly newStr: {
+                    readonly description: "Required parameter of `strReplace` command containing the new string. Required parameter of `insert` command containing the string to insert. Required parameter of `append` command containing the content to append to the file.";
+                    readonly type: "string";
+                };
+                readonly oldStr: {
+                    readonly description: "Required parameter of `strReplace` command containing the string in `path` to replace.";
+                    readonly type: "string";
+                };
+                readonly path: {
+                    readonly description: "Absolute path to file or directory, e.g. `/repo/file.py` or `/repo`.";
+                    readonly type: "string";
+                };
+            };
+            readonly required: readonly ["command", "path"];
+        };
+    };
+}
+export {};

package/out/language-server/agenticChat/tools/fsWrite.js

@@ -0,0 +1,128 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FsWrite = void 0;
+const path_1 = require("@aws/lsp-core/out/util/path");
+class FsWrite {
+    logging;
+    workspace;
+    constructor(features) {
+        this.logging = features.logging;
+        this.workspace = features.workspace;
+    }
+    async invoke(params) {
+        const sanitizedPath = (0, path_1.sanitize)(params.path);
+        switch (params.command) {
+            case 'create':
+                await this.handleCreate(params, sanitizedPath);
+                break;
+            case 'strReplace':
+                await this.handleStrReplace(params, sanitizedPath);
+                break;
+            case 'insert':
+                await this.handleInsert(params, sanitizedPath);
+                break;
+            case 'append':
+                await this.handleAppend(params, sanitizedPath);
+                break;
+        }
+        return {
+            output: {
+                kind: 'text',
+                content: '',
+            },
+        };
+    }
+    async handleCreate(params, sanitizedPath) {
+        const content = this.getCreateCommandText(params);
+        await this.workspace.fs.writeFile(sanitizedPath, content);
+    }
+    async handleStrReplace(params, sanitizedPath) {
+        const fileContent = await this.workspace.fs.readFile(sanitizedPath);
+        const matches = [...fileContent.matchAll(new RegExp(this.escapeRegExp(params.oldStr), 'g'))];
+        if (matches.length === 0) {
+            throw new Error(`No occurrences of "${params.oldStr}" were found`);
+        }
+        if (matches.length > 1) {
+            throw new Error(`${matches.length} occurrences of oldStr were found when only 1 is expected`);
+        }
+        const newContent = fileContent.replace(params.oldStr, params.newStr);
+        await this.workspace.fs.writeFile(sanitizedPath, newContent);
+    }
+    async handleInsert(params, sanitizedPath) {
+        const fileContent = await this.workspace.fs.readFile(sanitizedPath);
+        const lines = fileContent.split('\n');
+        const numLines = lines.length;
+        const insertLine = Math.max(0, Math.min(params.insertLine, numLines));
+        let newContent;
+        if (insertLine === 0) {
+            newContent = params.newStr + '\n' + fileContent;
+        }
+        else {
+            newContent = [...lines.slice(0, insertLine), params.newStr, ...lines.slice(insertLine)].join('\n');
+        }
+        await this.workspace.fs.writeFile(sanitizedPath, newContent);
+    }
+    async handleAppend(params, sanitizedPath) {
+        const fileContent = await this.workspace.fs.readFile(sanitizedPath);
+        const needsNewline = fileContent.length !== 0 && !fileContent.endsWith('\n');
+        let contentToAppend = params.newStr;
+        if (needsNewline) {
+            contentToAppend = '\n' + contentToAppend;
+        }
+        const newContent = fileContent + contentToAppend;
+        await this.workspace.fs.writeFile(sanitizedPath, newContent);
+    }
+    getCreateCommandText(params) {
+        if (params.fileText) {
+            return params.fileText;
+        }
+        if (params.newStr) {
+            this.logging.warn('Required field `fileText` is missing, use the provided `newStr` instead');
+            return params.newStr;
+        }
+        this.logging.warn('No content provided for the create command');
+        return '';
+    }
+    escapeRegExp(string) {
+        return string.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+    }
+    getSpec() {
+        return {
+            name: 'fsWrite',
+            description: 'A tool for creating and editing a file.\n * The `create` command will override the file at `path` if it already exists as a file, and otherwise create a new file\n * The `append` command will add content to the end of an existing file, automatically adding a newline if the file does not end with one. The file must exist.\n Notes for using the `strReplace` command:\n * The `oldStr` parameter should match EXACTLY one or more consecutive lines from the original file. Be mindful of whitespaces!\n * If the `oldStr` parameter is not unique in the file, the replacement will not be performed. Make sure to include enough context in `oldStr` to make it unique\n * The `newStr` parameter should contain the edited lines that should replace the `oldStr`. The `insert` command will insert `newStr` after `insertLine` and place it on its own line.',
+            inputSchema: {
+                type: 'object',
+                parameters: {
+                    command: {
+                        type: 'string',
+                        enum: ['create', 'strReplace', 'insert', 'append'],
+                        description: 'The commands to run. Allowed options are: `create`, `strReplace`, `insert`, `append`.',
+                    },
+                    fileText: {
+                        description: 'Required parameter of `create` command, with the content of the file to be created.',
+                        type: 'string',
+                    },
+                    insertLine: {
+                        description: 'Required parameter of `insert` command. The `newStr` will be inserted AFTER the line `insertLine` of `path`.',
+                        type: 'integer',
+                    },
+                    newStr: {
+                        description: 'Required parameter of `strReplace` command containing the new string. Required parameter of `insert` command containing the string to insert. Required parameter of `append` command containing the content to append to the file.',
+                        type: 'string',
+                    },
+                    oldStr: {
+                        description: 'Required parameter of `strReplace` command containing the string in `path` to replace.',
+                        type: 'string',
+                    },
+                    path: {
+                        description: 'Absolute path to file or directory, e.g. `/repo/file.py` or `/repo`.',
+                        type: 'string',
+                    },
+                },
+                required: ['command', 'path'],
+            },
+        };
+    }
+}
+exports.FsWrite = FsWrite;
+//# sourceMappingURL=fsWrite.js.map

package/out/language-server/agenticChat/tools/fsWrite.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fsWrite.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/tools/fsWrite.ts"],"names":[],"mappings":";;;AAEA,sDAAsD;AAiCtD,MAAa,OAAO;IACC,OAAO,CAAqB;IAC5B,SAAS,CAAuB;IAEjD,YAAY,QAAqE;QAC7E,IAAI,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAA;QAC/B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAA;IACvC,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,MAAqB;QACrC,MAAM,aAAa,GAAG,IAAA,eAAQ,EAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QAE3C,QAAQ,MAAM,CAAC,OAAO,EAAE,CAAC;YACrB,KAAK,QAAQ;gBACT,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;gBAC9C,MAAK;YACT,KAAK,YAAY;gBACb,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;gBAClD,MAAK;YACT,KAAK,QAAQ;gBACT,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;gBAC9C,MAAK;YACT,KAAK,QAAQ;gBACT,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;gBAC9C,MAAK;QACb,CAAC;QAED,OAAO;YACH,MAAM,EAAE;gBACJ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,EAAE;aACd;SACJ,CAAA;IACL,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,MAAoB,EAAE,aAAqB;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAA;QAEjD,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,OAAO,CAAC,CAAA;IAC7D,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAwB,EAAE,aAAqB;QAC1E,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QAEnE,MAAM,OAAO,GAAG,CAAC,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAA;QAE5F,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,sBAAsB,MAAM,CAAC,MAAM,cAAc,CAAC,CAAA;QACtE,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,GAAG,OAAO,CAAC,MAAM,2DAA2D,CAAC,CAAA;QACjG,CAAC;QAED,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;QACpE,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;IAChE,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,MAAoB,EAAE,aAAqB;QAClE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QACnE,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAErC,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAA;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAA;QAErE,IAAI,UAAkB,CAAA;QACtB,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;YACnB,UAAU,GAAG,MAAM,CAAC,MAAM,GAAG,IAAI,GAAG,WAAW,CAAA;QACnD,CAAC;aAAM,CAAC;YACJ,UAAU,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACtG,CAAC;QAED,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;IAChE,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,MAAoB,EAAE,aAAqB;QAClE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QACnE,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;QAE5E,IAAI,eAAe,GAAG,MAAM,CAAC,MAAM,CAAA;QACnC,IAAI,YAAY,EAAE,CAAC;YACf,eAAe,GAAG,IAAI,GAAG,eAAe,CAAA;QAC5C,CAAC;QAED,MAAM,UAAU,GAAG,WAAW,GAAG,eAAe,CAAA;QAChD,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;IAChE,CAAC;IAEO,oBAAoB,CAAC,MAAoB;QAC7C,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,MAAM,CAAC,QAAQ,CAAA;QAC1B,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,yEAAyE,CAAC,CAAA;YAC5F,OAAO,MAAM,CAAC,MAAM,CAAA;QACxB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAA;QAC/D,OAAO,EAAE,CAAA;IACb,CAAC;IAEO,YAAY,CAAC,MAAc;QAC/B,OAAO,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAA;IACxD,CAAC;IAEM,OAAO;QACV,OAAO;YACH,IAAI,EAAE,SAAS;YACf,WAAW,EACP,20BAA20B;YAC/0B,WAAW,EAAE;gBACT,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACR,OAAO,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,IAAI,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,QAAQ,CAAC;wBAClD,WAAW,EACP,uFAAuF;qBAC9F;oBACD,QAAQ,EAAE;wBACN,WAAW,EACP,qFAAqF;wBACzF,IAAI,EAAE,QAAQ;qBACjB;oBACD,UAAU,EAAE;wBACR,WAAW,EACP,8GAA8G;wBAClH,IAAI,EAAE,SAAS;qBAClB;oBACD,MAAM,EAAE;wBACJ,WAAW,EACP,oOAAoO;wBACxO,IAAI,EAAE,QAAQ;qBACjB;oBACD,MAAM,EAAE;wBACJ,WAAW,EACP,wFAAwF;wBAC5F,IAAI,EAAE,QAAQ;qBACjB;oBACD,IAAI,EAAE;wBACF,WAAW,EAAE,sEAAsE;wBACnF,IAAI,EAAE,QAAQ;qBACjB;iBACJ;gBACD,QAAQ,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC;aAChC;SACK,CAAA;IACd,CAAC;CACJ;AAlJD,0BAkJC"}

package/out/language-server/agenticChat/tools/toolServer.d.ts

@@ -0,0 +1,2 @@
+import { Server } from '@aws/language-server-runtimes/server-interface';
+export declare const FsToolsServer: Server;

package/out/language-server/agenticChat/tools/toolServer.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FsToolsServer = void 0;
+const fsRead_1 = require("./fsRead");
+const fsWrite_1 = require("./fsWrite");
+const FsToolsServer = ({ workspace, logging, agent }) => {
+    const fsReadTool = new fsRead_1.FsRead({ workspace, logging });
+    const fsWriteTool = new fsWrite_1.FsWrite({ workspace, logging });
+    agent.addTool(fsReadTool.getSpec(), (input) => fsReadTool.invoke(input));
+    agent.addTool(fsWriteTool.getSpec(), (input) => fsWriteTool.invoke(input));
+    return () => { };
+};
+exports.FsToolsServer = FsToolsServer;
+//# sourceMappingURL=toolServer.js.map

package/out/language-server/agenticChat/tools/toolServer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolServer.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/tools/toolServer.ts"],"names":[],"mappings":";;;AACA,qCAA+C;AAC/C,uCAAkD;AAE3C,MAAM,aAAa,GAAW,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;IACnE,MAAM,UAAU,GAAG,IAAI,eAAM,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAA;IACrD,MAAM,WAAW,GAAG,IAAI,iBAAO,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAA;IAEvD,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,KAAmB,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;IAEtF,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC,KAAoB,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;IAEzF,OAAO,GAAG,EAAE,GAAE,CAAC,CAAA;AACnB,CAAC,CAAA;AATY,QAAA,aAAa,iBASzB"}

package/out/language-server/agenticChat/tools/toolShared.d.ts

@@ -0,0 +1,7 @@
+export declare const maxToolResponseSize = 30720;
+export interface InvokeOutput {
+    output: {
+        kind: 'text' | 'json';
+        content: string;
+    };
+}

package/out/language-server/agenticChat/tools/toolShared.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.maxToolResponseSize = void 0;
+exports.maxToolResponseSize = 30720; // 30KB
+//# sourceMappingURL=toolShared.js.map

package/out/language-server/agenticChat/tools/toolShared.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolShared.js","sourceRoot":"","sources":["../../../../src/language-server/agenticChat/tools/toolShared.ts"],"names":[],"mappings":";;;AAAa,QAAA,mBAAmB,GAAG,KAAK,CAAA,CAAC,OAAO"}

package/out/language-server/chat/agents/docGeneration/constants.d.ts

@@ -0,0 +1,59 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { MynahIcons, Status } from '@aws/mynah-ui';
+import { FollowUpTypes, NewFileInfo } from '../shared/types';
+export declare const docChat = "docChat";
+export declare const docScheme = "aws-doc";
+export declare const featureName = "Amazon Q Doc Generation";
+export declare function getFileSummaryPercentage(input: string): number;
+export declare enum DocGenerationStep {
+    UPLOAD_TO_S3 = 0,
+    SUMMARIZING_FILES = 1,
+    GENERATING_ARTIFACTS = 2
+}
+export declare const docGenerationProgressMessage: (currentStep: DocGenerationStep, mode: Mode) => string;
+export declare const docGenerationSuccessMessage: (mode: Mode) => string;
+export declare const docRejectConfirmation = "Your changes have been discarded.";
+export declare const FolderSelectorFollowUps: {
+    icon: MynahIcons;
+    pillText: string;
+    prompt: string;
+    status: Status;
+    type: FollowUpTypes;
+}[];
+export declare const CodeChangeFollowUps: {
+    pillText: string;
+    prompt: string;
+    type: FollowUpTypes;
+    icon: MynahIcons;
+    status: Status;
+}[];
+export declare const NewSessionFollowUps: {
+    pillText: string;
+    type: FollowUpTypes;
+    status: Status;
+}[];
+export declare const SynchronizeDocumentation: {
+    pillText: string;
+    prompt: string;
+    type: FollowUpTypes;
+};
+export declare const EditDocumentation: {
+    pillText: string;
+    prompt: string;
+    type: FollowUpTypes;
+};
+export declare enum Mode {
+    NONE = "None",
+    CREATE = "Create",
+    SYNC = "Sync",
+    EDIT = "Edit"
+}
+/**
+ *
+ * @param paths file paths
+ * @returns the path to a README.md, or undefined if none exist
+ */
+export declare const findReadmePath: (paths?: NewFileInfo[]) => NewFileInfo | undefined;