@aws/lsp-codewhisperer 0.0.4 → 0.0.5

package/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [0.0.5] - 2024-05-15
+- Create Chat server export it for consumption
+- Fix duplicate hover message in security scan
+- Clear security scan finding when a project is unloaded
+- Fix the consumption of streaming client
+- Add `json`, `yaml` and `java` to list of supported languages
+
 ## [0.0.4] - 2024-03-28
 - Integrate dependency graph with RunSecurityScan function
 - Add server for transform feature

package/out/client/sigv4/service.json

@@ -192,7 +192,7 @@
             "type": "string",
             "max": 128,
             "min": 1,
-            "pattern": "(python|javascript|java|csharp|typescript|c|cpp|go|kotlin|php|ruby|rust|scala|shell|sql)"
+            "pattern": "(python|javascript|java|csharp|typescript|c|cpp|go|kotlin|php|ruby|rust|scala|shell|sql|json|yaml|vue|tf|tsx|jsx)"
         },
         "Recommendation": {
             "type": "structure",

package/out/client/token/bearer-token-service.json

@@ -1036,7 +1036,7 @@
             "type": "string",
             "max": 128,
             "min": 1,
-            "pattern": "(python|javascript|java|csharp|typescript|c|cpp|go|kotlin|php|ruby|rust|scala|shell|sql|json|yaml|vue|tf)"
+            "pattern": "(python|javascript|java|csharp|typescript|c|cpp|go|kotlin|php|ruby|rust|scala|shell|sql|json|yaml|vue|tf|tsx|jsx)"
         },
         "ProgressUpdates": {
             "type": "list",

package/out/index.d.ts

@@ -1,2 +1,3 @@
 export * from './language-server/codeWhispererSecurityScanServer';
 export * from './language-server/codeWhispererServer';
+export * from './language-server/qChatServer';

package/out/index.js

@@ -16,4 +16,5 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./language-server/codeWhispererSecurityScanServer"), exports);
 __exportStar(require("./language-server/codeWhispererServer"), exports);
+__exportStar(require("./language-server/qChatServer"), exports);
 //# sourceMappingURL=index.js.map

package/out/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,oFAAiE;AACjE,wEAAqD"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,oFAAiE;AACjE,wEAAqD;AACrD,gEAA6C"}

package/out/language-server/chat/chatController.d.ts

@@ -0,0 +1,25 @@
+import { ChatParams } from '@aws/language-server-runtimes-types';
+import { CancellationToken, Chat, ChatResult, EndChatParams, ResponseError, TabAddParams, TabRemoveParams } from '@aws/language-server-runtimes/server-interface';
+import { Features, LspHandlers } from '../types';
+import { ChatSessionManagementService } from './chatSessionManagementService';
+type ChatHandlers = LspHandlers<Chat>;
+export declare class ChatController implements ChatHandlers {
+    #private;
+    constructor(chatSessionManagementService: ChatSessionManagementService, features: Features);
+    onChatPrompt(params: ChatParams, token: CancellationToken): Promise<ChatResult | ResponseError<ChatResult>>;
+    onCodeInsertToCursorPosition(): void;
+    onCopyCodeToClipboard(): void;
+    onEndChat(params: EndChatParams, _token: CancellationToken): boolean;
+    onFollowUpClicked(): void;
+    onInfoLinkClick(): void;
+    onLinkClick(): void;
+    onSendFeedback(): void;
+    onQuickAction(): never;
+    onReady(): void;
+    onSourceLinkClick(): void;
+    onTabAdd(params: TabAddParams): void;
+    onTabChange(): void;
+    onTabRemove(params: TabRemoveParams): void;
+    onVote(): void;
+}
+export {};

package/out/language-server/chat/chatController.js

@@ -0,0 +1,117 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _ChatController_instances, _ChatController_features, _ChatController_chatSessionManagementService, _ChatController_documentContextExtractor, _ChatController_extractEditorState, _ChatController_processAssistantResponse, _ChatController_log;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatController = void 0;
+const protocol_1 = require("@aws/language-server-runtimes/protocol");
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const chatEventParser_1 = require("./chatEventParser");
+const documentContext_1 = require("./contexts/documentContext");
+const utils_1 = require("./utils");
+class ChatController {
+    constructor(chatSessionManagementService, features) {
+        _ChatController_instances.add(this);
+        _ChatController_features.set(this, void 0);
+        _ChatController_chatSessionManagementService.set(this, void 0);
+        _ChatController_documentContextExtractor.set(this, void 0);
+        __classPrivateFieldSet(this, _ChatController_features, features, "f");
+        __classPrivateFieldSet(this, _ChatController_chatSessionManagementService, chatSessionManagementService, "f");
+        __classPrivateFieldSet(this, _ChatController_documentContextExtractor, new documentContext_1.DocumentContextExtractor(), "f");
+    }
+    async onChatPrompt(params, token) {
+        const sessionResult = __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").getSession(params.tabId);
+        const { data: session } = sessionResult;
+        if (!session) {
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Session not found for tabId', params.tabId);
+            return new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InvalidParams, 'error' in sessionResult ? sessionResult.error : 'Unknown error');
+        }
+        token.onCancellationRequested(() => {
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'cancellation requested');
+            session.abortRequest();
+        });
+        const editorState = await __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_extractEditorState).call(this, params.textDocument, params.cursorState);
+        const requestInput = (0, utils_1.convertChatParamsToRequestInput)(params, editorState);
+        if (!requestInput.success) {
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, requestInput.error);
+            return new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InvalidParams, requestInput.error);
+        }
+        let response;
+        try {
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Request from tab:', params.tabId, 'conversation id:', requestInput.data.conversationState?.conversationId ?? 'undefined');
+            response = await session.generateAssistantResponse(requestInput.data);
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Response to tab:', params.tabId, JSON.stringify(response.$metadata));
+        }
+        catch (err) {
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, `Q api request error ${err instanceof Error ? err.message : 'unknown'}`);
+            return new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InternalError, err instanceof Error ? err.message : 'Internal Server Error');
+        }
+        try {
+            const result = await __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_processAssistantResponse).call(this, response, params.partialResultToken);
+            return result.success
+                ? result.data
+                : new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InternalError, result.error, result.data);
+        }
+        catch (err) {
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Error encountered during response streaming:', err instanceof Error ? err.message : 'unknown');
+            return new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InternalError, err instanceof Error ? err.message : 'Internal Server Error');
+        }
+    }
+    onCodeInsertToCursorPosition() { }
+    onCopyCodeToClipboard() { }
+    onEndChat(params, _token) {
+        const { success } = __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").deleteSession(params.tabId);
+        return success;
+    }
+    onFollowUpClicked() { }
+    onInfoLinkClick() { }
+    onLinkClick() { }
+    onSendFeedback() { }
+    onQuickAction() {
+        throw new Error('Method not implemented.');
+    }
+    onReady() { }
+    onSourceLinkClick() { }
+    onTabAdd(params) {
+        __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").createSession(params.tabId);
+    }
+    onTabChange() { }
+    onTabRemove(params) {
+        __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").deleteSession(params.tabId);
+    }
+    onVote() { }
+}
+exports.ChatController = ChatController;
+_ChatController_features = new WeakMap(), _ChatController_chatSessionManagementService = new WeakMap(), _ChatController_documentContextExtractor = new WeakMap(), _ChatController_instances = new WeakSet(), _ChatController_extractEditorState = async function _ChatController_extractEditorState(identifier, cursorState) {
+    if (!identifier || !Array.isArray(cursorState) || cursorState.length === 0) {
+        return;
+    }
+    const textDocument = await __classPrivateFieldGet(this, _ChatController_features, "f").workspace.getTextDocument(identifier.uri);
+    return textDocument && (await __classPrivateFieldGet(this, _ChatController_documentContextExtractor, "f").extractEditorState(textDocument, cursorState[0]));
+}, _ChatController_processAssistantResponse = async function _ChatController_processAssistantResponse(response, partialResultToken) {
+    const chatEventParser = new chatEventParser_1.ChatEventParser(response.$metadata.requestId);
+    for await (const chatEvent of response.generateAssistantResponseResponse) {
+        __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Streaming partial chat event');
+        const chatResult = chatEventParser.processPartialEvent(chatEvent);
+        // terminate early when there is an error
+        if (!chatResult.success) {
+            return chatResult;
+        }
+        if (partialResultToken) {
+            __classPrivateFieldGet(this, _ChatController_features, "f").lsp.sendProgress(protocol_1.chatRequestType, partialResultToken, chatResult.data);
+        }
+    }
+    return chatEventParser.getChatResult();
+}, _ChatController_log = function _ChatController_log(...messages) {
+    __classPrivateFieldGet(this, _ChatController_features, "f").logging.log(messages.join(' '));
+};
+//# sourceMappingURL=chatController.js.map

package/out/language-server/chat/chatController.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatController.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAEA,qEAAwE;AACxE,qFAUuD;AAEvD,uDAAmD;AAEnD,gEAAqE;AACrE,mCAAyD;AAIzD,MAAa,cAAc;IAKvB,YAAY,4BAA0D,EAAE,QAAkB;;QAJ1F,2CAAmB;QACnB,+DAA2D;QAC3D,2DAAmD;QAG/C,uBAAA,IAAI,4BAAa,QAAQ,MAAA,CAAA;QACzB,uBAAA,IAAI,gDAAiC,4BAA4B,MAAA,CAAA;QACjE,uBAAA,IAAI,4CAA6B,IAAI,0CAAwB,EAAE,MAAA,CAAA;IACnE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAkB,EAAE,KAAwB;QAC3D,MAAM,aAAa,GAAG,uBAAA,IAAI,oDAA8B,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEjF,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,aAAa,CAAA;QAEvC,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,6BAA6B,EAAE,MAAM,CAAC,KAAK,CAAC,CAAA;YACtD,OAAO,IAAI,gCAAa,CACpB,6BAAU,CAAC,aAAa,EACxB,OAAO,IAAI,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,eAAe,CACnE,CAAA;QACL,CAAC;QAED,KAAK,CAAC,uBAAuB,CAAC,GAAG,EAAE;YAC/B,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,wBAAwB,CAAC,CAAA;YACnC,OAAO,CAAC,YAAY,EAAE,CAAA;QAC1B,CAAC,CAAC,CAAA;QAEF,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,qEAAoB,MAAxB,IAAI,EAAqB,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,WAAW,CAAC,CAAA;QAC3F,MAAM,YAAY,GAAG,IAAA,uCAA+B,EAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAEzE,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YACxB,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,YAAY,CAAC,KAAK,CAAC,CAAA;YAC7B,OAAO,IAAI,gCAAa,CAAa,6BAAU,CAAC,aAAa,EAAE,YAAY,CAAC,KAAK,CAAC,CAAA;QACtF,CAAC;QAED,IAAI,QAAgD,CAAA;QAEpD,IAAI,CAAC;YACD,uBAAA,IAAI,sDAAK,MAAT,IAAI,EACA,mBAAmB,EACnB,MAAM,CAAC,KAAK,EACZ,kBAAkB,EAClB,YAAY,CAAC,IAAI,CAAC,iBAAiB,EAAE,cAAc,IAAI,WAAW,CACrE,CAAA;YAED,QAAQ,GAAG,MAAM,OAAO,CAAC,yBAAyB,CAAC,YAAY,CAAC,IAAI,CAAC,CAAA;YACrE,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,kBAAkB,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAA;QACnF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,uBAAuB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAA;YAElF,OAAO,IAAI,gCAAa,CACpB,6BAAU,CAAC,aAAa,EACxB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAC/D,CAAA;QACL,CAAC;QAED,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,2EAA0B,MAA9B,IAAI,EAA2B,QAAQ,EAAE,MAAM,CAAC,kBAAkB,CAAC,CAAA;YAExF,OAAO,MAAM,CAAC,OAAO;gBACjB,CAAC,CAAC,MAAM,CAAC,IAAI;gBACb,CAAC,CAAC,IAAI,gCAAa,CAAa,6BAAU,CAAC,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,CAAA;QAC5F,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,8CAA8C,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;YAEzG,OAAO,IAAI,gCAAa,CACpB,6BAAU,CAAC,aAAa,EACxB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAC/D,CAAA;QACL,CAAC;IACL,CAAC;IAED,4BAA4B,KAAI,CAAC;IACjC,qBAAqB,KAAI,CAAC;IAE1B,SAAS,CAAC,MAAqB,EAAE,MAAyB;QACtD,MAAM,EAAE,OAAO,EAAE,GAAG,uBAAA,IAAI,oDAA8B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAElF,OAAO,OAAO,CAAA;IAClB,CAAC;IAED,iBAAiB,KAAI,CAAC;IAEtB,eAAe,KAAI,CAAC;IAEpB,WAAW,KAAI,CAAC;IAEhB,cAAc,KAAI,CAAC;IAEnB,aAAa;QACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;IAC9C,CAAC;IAED,OAAO,KAAI,CAAC;IACZ,iBAAiB,KAAI,CAAC;IAEtB,QAAQ,CAAC,MAAoB;QACzB,uBAAA,IAAI,oDAA8B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAClE,CAAC;IAED,WAAW,KAAI,CAAC;IAEhB,WAAW,CAAC,MAAuB;QAC/B,uBAAA,IAAI,oDAA8B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAClE,CAAC;IAED,MAAM,KAAI,CAAC;CAyCd;AArJD,wCAqJC;kPAvCG,KAAK,6CACD,UAAmC,EACnC,WAA2B;IAE3B,IAAI,CAAC,UAAU,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzE,OAAM;IACV,CAAC;IACD,MAAM,YAAY,GAAG,MAAM,uBAAA,IAAI,gCAAU,CAAC,SAAS,CAAC,eAAe,CAAC,UAAU,CAAC,GAAG,CAAC,CAAA;IAEnF,OAAO,YAAY,IAAI,CAAC,MAAM,uBAAA,IAAI,gDAA0B,CAAC,kBAAkB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;AAClH,CAAC,6CAED,KAAK,mDACD,QAAgD,EAChD,kBAAoC;IAEpC,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAU,CAAC,CAAA;IAE1E,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,QAAQ,CAAC,iCAAkC,EAAE,CAAC;QACxE,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,8BAA8B,CAAC,CAAA;QAEzC,MAAM,UAAU,GAAG,eAAe,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAA;QAEjE,yCAAyC;QACzC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACtB,OAAO,UAAU,CAAA;QACrB,CAAC;QAED,IAAI,kBAAkB,EAAE,CAAC;YACrB,uBAAA,IAAI,gCAAU,CAAC,GAAG,CAAC,YAAY,CAAC,0BAAe,EAAE,kBAAkB,EAAE,UAAU,CAAC,IAAI,CAAC,CAAA;QACzF,CAAC;IACL,CAAC;IAED,OAAO,eAAe,CAAC,aAAa,EAAE,CAAA;AAC1C,CAAC,qDAEI,GAAG,QAAkB;IACtB,uBAAA,IAAI,gCAAU,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;AAClD,CAAC"}

package/out/language-server/chat/chatController.test.d.ts

@@ -0,0 +1 @@
+export {};

package/out/language-server/chat/chatController.test.js

@@ -0,0 +1,240 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const testing_1 = require("@aws/language-server-runtimes/testing");
+const assert = require("assert");
+const ts_sinon_1 = require("ts-sinon");
+const testUtils_1 = require("../testUtils");
+const chatController_1 = require("./chatController");
+const chatSessionManagementService_1 = require("./chatSessionManagementService");
+const chatSessionService_1 = require("./chatSessionService");
+const documentContext_1 = require("./contexts/documentContext");
+describe('ChatController', () => {
+    const mockTabId = 'tab-1';
+    const mockConversationId = 'mock-conversation-id';
+    const mockMessageId = 'mock-message-id';
+    const mockAssistantResponseList = [
+        {
+            assistantResponseEvent: {
+                content: 'Hello ',
+            },
+        },
+        {
+            assistantResponseEvent: {
+                content: 'World',
+            },
+        },
+        {
+            assistantResponseEvent: {
+                content: '!',
+            },
+        },
+    ];
+    const expectedCompleteChatResult = {
+        messageId: mockMessageId,
+        body: 'Hello World!',
+        canBeVoted: undefined,
+        codeReference: undefined,
+        followUp: undefined,
+        relatedContent: undefined,
+    };
+    const mockCancellationToken = {
+        isCancellationRequested: false,
+        onCancellationRequested: () => ({ dispose: () => null }),
+    };
+    let generateAssistantResponseStub;
+    let disposeStub;
+    let testFeatures;
+    let chatSessionManagementService;
+    let chatController;
+    beforeEach(() => {
+        generateAssistantResponseStub = ts_sinon_1.default
+            .stub(codewhisperer_streaming_1.CodeWhispererStreaming.prototype, 'generateAssistantResponse')
+            .callsFake(() => {
+            return new Promise(resolve => setTimeout(() => {
+                resolve({
+                    conversationId: mockConversationId,
+                    $metadata: {
+                        requestId: mockMessageId,
+                    },
+                    generateAssistantResponseResponse: (0, testUtils_1.createIterableResponse)(mockAssistantResponseList),
+                });
+            }));
+        });
+        testFeatures = new testing_1.TestFeatures();
+        disposeStub = ts_sinon_1.default.stub(chatSessionService_1.ChatSessionService.prototype, 'dispose');
+        chatSessionManagementService = chatSessionManagementService_1.ChatSessionManagementService.getInstance().withCredentialsProvider(testFeatures.credentialsProvider);
+        chatController = new chatController_1.ChatController(chatSessionManagementService, testFeatures);
+    });
+    afterEach(() => {
+        generateAssistantResponseStub.restore();
+        disposeStub.restore();
+        chatSessionManagementService_1.ChatSessionManagementService.reset();
+    });
+    it('creates a session when a tab add notifcation is received', () => {
+        chatController.onTabAdd({ tabId: mockTabId });
+        const sessionResult = chatSessionManagementService.getSession(mockTabId);
+        ts_sinon_1.default.assert.match(sessionResult, {
+            success: true,
+            data: ts_sinon_1.default.match.instanceOf(chatSessionService_1.ChatSessionService),
+        });
+    });
+    it('deletes a session by tab id when a tab remove notifcation is received', () => {
+        chatController.onTabAdd({ tabId: mockTabId });
+        assert.ok(chatSessionManagementService.getSession(mockTabId).data instanceof chatSessionService_1.ChatSessionService);
+        chatController.onTabRemove({ tabId: mockTabId });
+        ts_sinon_1.default.assert.calledOnce(disposeStub);
+        const sessionResult = chatSessionManagementService.getSession(mockTabId);
+        ts_sinon_1.default.assert.match(sessionResult, {
+            success: false,
+            error: ts_sinon_1.default.match.string,
+        });
+    });
+    it('deletes a session by tab id a end chat request is received', () => {
+        chatController.onTabAdd({ tabId: mockTabId });
+        chatController.onEndChat({ tabId: mockTabId }, mockCancellationToken);
+        ts_sinon_1.default.assert.calledOnce(disposeStub);
+        const sessionResult = chatSessionManagementService.getSession(mockTabId);
+        ts_sinon_1.default.assert.match(sessionResult, {
+            success: false,
+            error: ts_sinon_1.default.match.string,
+        });
+    });
+    describe('onChatPrompt', () => {
+        beforeEach(() => {
+            chatController.onTabAdd({ tabId: mockTabId });
+        });
+        it('throw error if session is not found', async () => {
+            const result = await chatController.onChatPrompt({ tabId: 'XXXX', prompt: { prompt: 'Hello' } }, mockCancellationToken);
+            assert.ok(result instanceof server_interface_1.ResponseError);
+        });
+        it('read all the response streams and return compiled results', async () => {
+            const chatResultPromise = chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' } }, mockCancellationToken);
+            const chatResult = await chatResultPromise;
+            ts_sinon_1.default.assert.callCount(testFeatures.lsp.sendProgress, 0);
+            assert.deepStrictEqual(chatResult, expectedCompleteChatResult);
+        });
+        it('read all the response streams and send progress as partial result is received', async () => {
+            const chatResultPromise = chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' }, partialResultToken: 1 }, mockCancellationToken);
+            const chatResult = await chatResultPromise;
+            ts_sinon_1.default.assert.callCount(testFeatures.lsp.sendProgress, mockAssistantResponseList.length);
+            assert.deepStrictEqual(chatResult, expectedCompleteChatResult);
+        });
+        it('returns a ResponseError if request input is not valid', async () => {
+            const chatResultPromise = chatController.onChatPrompt({ tabId: mockTabId, prompt: {} }, mockCancellationToken);
+            const chatResult = await chatResultPromise;
+            assert.ok(chatResult instanceof server_interface_1.ResponseError);
+        });
+        it('returns a ResponseError if generateAssistantResponse returns an error', async () => {
+            generateAssistantResponseStub.callsFake(() => {
+                throw new Error('Error');
+            });
+            const chatResult = await chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' } }, mockCancellationToken);
+            assert.ok(chatResult instanceof server_interface_1.ResponseError);
+        });
+        it('returns a ResponseError if response streams return an error event', async () => {
+            generateAssistantResponseStub.callsFake(() => {
+                return Promise.resolve({
+                    conversationId: mockConversationId,
+                    $metadata: {
+                        requestId: mockMessageId,
+                    },
+                    generateAssistantResponseResponse: (0, testUtils_1.createIterableResponse)([
+                        // ["Hello ", "World"]
+                        ...mockAssistantResponseList.slice(0, 2),
+                        { error: { message: 'some error' } },
+                        // ["!"]
+                        ...mockAssistantResponseList.slice(2),
+                    ]),
+                });
+            });
+            const chatResult = await chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' } }, mockCancellationToken);
+            assert.deepStrictEqual(chatResult, new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InternalError, 'some error', {
+                ...expectedCompleteChatResult,
+                body: 'Hello World',
+            }));
+        });
+        it('returns a ResponseError if response streams return an invalid state event', async () => {
+            generateAssistantResponseStub.callsFake(() => {
+                return Promise.resolve({
+                    conversationId: mockConversationId,
+                    $metadata: {
+                        requestId: mockMessageId,
+                    },
+                    generateAssistantResponseResponse: (0, testUtils_1.createIterableResponse)([
+                        // ["Hello ", "World"]
+                        ...mockAssistantResponseList.slice(0, 2),
+                        { invalidStateEvent: { message: 'invalid state' } },
+                        // ["!"]
+                        ...mockAssistantResponseList.slice(2),
+                    ]),
+                });
+            });
+            const chatResult = await chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' } }, mockCancellationToken);
+            assert.deepStrictEqual(chatResult, new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InternalError, 'invalid state', {
+                ...expectedCompleteChatResult,
+                body: 'Hello World',
+            }));
+        });
+        describe('#extractEditorState', () => {
+            const typescriptDocument = server_interface_1.TextDocument.create('file:///test.ts', 'typescript', 1, 'test');
+            let extractEditorStateStub;
+            const editorStateObject = {};
+            const mockCursorState = {
+                range: {
+                    start: {
+                        line: 1,
+                        character: 1,
+                    },
+                    end: {
+                        line: 1,
+                        character: 1,
+                    },
+                },
+            };
+            beforeEach(() => {
+                extractEditorStateStub = ts_sinon_1.default.stub(documentContext_1.DocumentContextExtractor.prototype, 'extractEditorState');
+                testFeatures.openDocument(typescriptDocument);
+                extractEditorStateStub.resolves(editorStateObject);
+            });
+            afterEach(() => {
+                extractEditorStateStub.restore();
+            });
+            it('leaves editor state as undefined if cursorState is not passed', async () => {
+                await chatController.onChatPrompt({
+                    tabId: mockTabId,
+                    prompt: { prompt: 'Hello' },
+                    textDocument: { uri: 'file:///test.ts' },
+                    cursorState: undefined,
+                }, mockCancellationToken);
+                const calledRequestInput = generateAssistantResponseStub.firstCall.firstArg;
+                assert.strictEqual(calledRequestInput.conversationState?.currentMessage?.userInputMessage?.userInputMessageContext
+                    ?.editorState, undefined);
+            });
+            it('leaves editor state as undefined if document identified is not passed', async () => {
+                await chatController.onChatPrompt({
+                    tabId: mockTabId,
+                    prompt: { prompt: 'Hello' },
+                    cursorState: [mockCursorState],
+                }, mockCancellationToken);
+                const calledRequestInput = generateAssistantResponseStub.firstCall.firstArg;
+                assert.strictEqual(calledRequestInput.conversationState?.currentMessage?.userInputMessage?.userInputMessageContext
+                    ?.editorState, undefined);
+            });
+            it('parses editor state context and includes as requestInput if both cursor state and text document are found', async () => {
+                await chatController.onChatPrompt({
+                    tabId: mockTabId,
+                    prompt: { prompt: 'Hello' },
+                    textDocument: { uri: 'file:///test.ts' },
+                    cursorState: [mockCursorState],
+                }, mockCancellationToken);
+                const calledRequestInput = generateAssistantResponseStub.firstCall.firstArg;
+                // asserting object reference equality
+                assert.strictEqual(calledRequestInput.conversationState?.currentMessage?.userInputMessage?.userInputMessageContext
+                    ?.editorState, editorStateObject);
+            });
+        });
+    });
+});
+//# sourceMappingURL=chatController.test.js.map

package/out/language-server/chat/chatController.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatController.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatController.test.ts"],"names":[],"mappings":";;AAAA,2EAIsC;AACtC,qFAAoH;AACpH,mEAAoE;AACpE,iCAAgC;AAChC,uCAA4B;AAC5B,4CAAqD;AACrD,qDAAiD;AACjD,iFAA6E;AAC7E,6DAAyD;AACzD,gEAAqE;AAErE,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC5B,MAAM,SAAS,GAAG,OAAO,CAAA;IACzB,MAAM,kBAAkB,GAAG,sBAAsB,CAAA;IACjD,MAAM,aAAa,GAAG,iBAAiB,CAAA;IAEvC,MAAM,yBAAyB,GAAyB;QACpD;YACI,sBAAsB,EAAE;gBACpB,OAAO,EAAE,QAAQ;aACpB;SACJ;QACD;YACI,sBAAsB,EAAE;gBACpB,OAAO,EAAE,OAAO;aACnB;SACJ;QACD;YACI,sBAAsB,EAAE;gBACpB,OAAO,EAAE,GAAG;aACf;SACJ;KACJ,CAAA;IAED,MAAM,0BAA0B,GAAe;QAC3C,SAAS,EAAE,aAAa;QACxB,IAAI,EAAE,cAAc;QACpB,UAAU,EAAE,SAAS;QACrB,aAAa,EAAE,SAAS;QACxB,QAAQ,EAAE,SAAS;QACnB,cAAc,EAAE,SAAS;KAC5B,CAAA;IAED,MAAM,qBAAqB,GAAG;QAC1B,uBAAuB,EAAE,KAAK;QAC9B,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC;KAC3D,CAAA;IAED,IAAI,6BAA8C,CAAA;IAClD,IAAI,WAA4B,CAAA;IAEhC,IAAI,YAA0B,CAAA;IAC9B,IAAI,4BAA0D,CAAA;IAC9D,IAAI,cAA8B,CAAA;IAElC,UAAU,CAAC,GAAG,EAAE;QACZ,6BAA6B,GAAG,kBAAK;aAChC,IAAI,CAAC,gDAAsB,CAAC,SAAS,EAAE,2BAA2B,CAAC;aACnE,SAAS,CAAC,GAAG,EAAE;YACZ,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CACzB,UAAU,CAAC,GAAG,EAAE;gBACZ,OAAO,CAAC;oBACJ,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE;wBACP,SAAS,EAAE,aAAa;qBAC3B;oBACD,iCAAiC,EAAE,IAAA,kCAAsB,EAAC,yBAAyB,CAAC;iBACvF,CAAC,CAAA;YACN,CAAC,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEN,YAAY,GAAG,IAAI,sBAAY,EAAE,CAAA;QAEjC,WAAW,GAAG,kBAAK,CAAC,IAAI,CAAC,uCAAkB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;QAEjE,4BAA4B,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAC,uBAAuB,CAC7F,YAAY,CAAC,mBAAmB,CACnC,CAAA;QAED,cAAc,GAAG,IAAI,+BAAc,CAAC,4BAA4B,EAAE,YAAY,CAAC,CAAA;IACnF,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,6BAA6B,CAAC,OAAO,EAAE,CAAA;QACvC,WAAW,CAAC,OAAO,EAAE,CAAA;QACrB,2DAA4B,CAAC,KAAK,EAAE,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAChE,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAE7C,MAAM,aAAa,GAAG,4BAA4B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QACxE,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE;YAC9B,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,kBAAK,CAAC,KAAK,CAAC,UAAU,CAAC,uCAAkB,CAAC;SACnD,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC7E,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAE7C,MAAM,CAAC,EAAE,CAAC,4BAA4B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,YAAY,uCAAkB,CAAC,CAAA;QAEhG,cAAc,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAEhD,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;QAEpC,MAAM,aAAa,GAAG,4BAA4B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAExE,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE;YAC9B,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;SAC5B,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QAClE,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAE7C,cAAc,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,qBAAqB,CAAC,CAAA;QAErE,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;QAEpC,MAAM,aAAa,GAAG,4BAA4B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAExE,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE;YAC9B,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;SAC5B,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC1B,UAAU,CAAC,GAAG,EAAE;YACZ,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QACjD,CAAC,CAAC,CAAA;QACF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAC5C,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAC9C,qBAAqB,CACxB,CAAA;YAED,MAAM,CAAC,EAAE,CAAC,MAAM,YAAY,gCAAa,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,iBAAiB,GAAG,cAAc,CAAC,YAAY,CACjD,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EACjD,qBAAqB,CACxB,CAAA;YAED,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAA;YAE1C,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC,CAAA;YACxD,MAAM,CAAC,eAAe,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAA;QAClE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC3F,MAAM,iBAAiB,GAAG,cAAc,CAAC,YAAY,CACjD,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,kBAAkB,EAAE,CAAC,EAAE,EACxE,qBAAqB,CACxB,CAAA;YAED,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAA;YAE1C,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,yBAAyB,CAAC,MAAM,CAAC,CAAA;YACvF,MAAM,CAAC,eAAe,CAAC,UAAU,EAAE,0BAA0B,CAAC,CAAA;QAClE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,iBAAiB,GAAG,cAAc,CAAC,YAAY,CACjD,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,EAChC,qBAAqB,CACxB,CAAA;YAED,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAA;YAC1C,MAAM,CAAC,EAAE,CAAC,UAAU,YAAY,gCAAa,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACnF,6BAA6B,CAAC,SAAS,CAAC,GAAG,EAAE;gBACzC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAA;YAC5B,CAAC,CAAC,CAAA;YAEF,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,YAAY,CAChD,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EACjD,qBAAqB,CACxB,CAAA;YAED,MAAM,CAAC,EAAE,CAAC,UAAU,YAAY,gCAAa,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YAC/E,6BAA6B,CAAC,SAAS,CAAC,GAAG,EAAE;gBACzC,OAAO,OAAO,CAAC,OAAO,CAAC;oBACnB,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE;wBACP,SAAS,EAAE,aAAa;qBAC3B;oBACD,iCAAiC,EAAE,IAAA,kCAAsB,EAAC;wBACtD,sBAAsB;wBACtB,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;wBACxC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE;wBACpC,QAAQ;wBACR,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC;qBACxC,CAAC;iBACL,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YAEF,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,YAAY,CAChD,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EACjD,qBAAqB,CACxB,CAAA;YAED,MAAM,CAAC,eAAe,CAClB,UAAU,EACV,IAAI,gCAAa,CAAC,6BAAU,CAAC,aAAa,EAAE,YAAY,EAAE;gBACtD,GAAG,0BAA0B;gBAC7B,IAAI,EAAE,aAAa;aACtB,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACvF,6BAA6B,CAAC,SAAS,CAAC,GAAG,EAAE;gBACzC,OAAO,OAAO,CAAC,OAAO,CAAC;oBACnB,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE;wBACP,SAAS,EAAE,aAAa;qBAC3B;oBACD,iCAAiC,EAAE,IAAA,kCAAsB,EAAC;wBACtD,sBAAsB;wBACtB,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;wBACxC,EAAE,iBAAiB,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE;wBACnD,QAAQ;wBACR,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC;qBACxC,CAAC;iBACL,CAAC,CAAA;YACN,CAAC,CAAC,CAAA;YAEF,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,YAAY,CAChD,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EACjD,qBAAqB,CACxB,CAAA;YAED,MAAM,CAAC,eAAe,CAClB,UAAU,EACV,IAAI,gCAAa,CAAC,6BAAU,CAAC,aAAa,EAAE,eAAe,EAAE;gBACzD,GAAG,0BAA0B;gBAC7B,IAAI,EAAE,aAAa;aACtB,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;YACjC,MAAM,kBAAkB,GAAG,+BAAY,CAAC,MAAM,CAAC,iBAAiB,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,CAAC,CAAA;YAC1F,IAAI,sBAAuC,CAAA;YAC3C,MAAM,iBAAiB,GAAG,EAAE,CAAA;YAC5B,MAAM,eAAe,GAAG;gBACpB,KAAK,EAAE;oBACH,KAAK,EAAE;wBACH,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,CAAC;qBACf;oBACD,GAAG,EAAE;wBACD,IAAI,EAAE,CAAC;wBACP,SAAS,EAAE,CAAC;qBACf;iBACJ;aACJ,CAAA;YAED,UAAU,CAAC,GAAG,EAAE;gBACZ,sBAAsB,GAAG,kBAAK,CAAC,IAAI,CAAC,0CAAwB,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAA;gBAC7F,YAAY,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAA;gBAC7C,sBAAsB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAA;YACtD,CAAC,CAAC,CAAA;YAEF,SAAS,CAAC,GAAG,EAAE;gBACX,sBAAsB,CAAC,OAAO,EAAE,CAAA;YACpC,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;gBAC3E,MAAM,cAAc,CAAC,YAAY,CAC7B;oBACI,KAAK,EAAE,SAAS;oBAChB,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;oBAC3B,YAAY,EAAE,EAAE,GAAG,EAAE,iBAAiB,EAAE;oBACxC,WAAW,EAAE,SAAS;iBACzB,EACD,qBAAqB,CACxB,CAAA;gBAED,MAAM,kBAAkB,GACpB,6BAA6B,CAAC,SAAS,CAAC,QAAQ,CAAA;gBAEpD,MAAM,CAAC,WAAW,CACd,kBAAkB,CAAC,iBAAiB,EAAE,cAAc,EAAE,gBAAgB,EAAE,uBAAuB;oBAC3F,EAAE,WAAW,EACjB,SAAS,CACZ,CAAA;YACL,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;gBACnF,MAAM,cAAc,CAAC,YAAY,CAC7B;oBACI,KAAK,EAAE,SAAS;oBAChB,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;oBAC3B,WAAW,EAAE,CAAC,eAAe,CAAC;iBACjC,EACD,qBAAqB,CACxB,CAAA;gBAED,MAAM,kBAAkB,GACpB,6BAA6B,CAAC,SAAS,CAAC,QAAQ,CAAA;gBAEpD,MAAM,CAAC,WAAW,CACd,kBAAkB,CAAC,iBAAiB,EAAE,cAAc,EAAE,gBAAgB,EAAE,uBAAuB;oBAC3F,EAAE,WAAW,EACjB,SAAS,CACZ,CAAA;YACL,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,2GAA2G,EAAE,KAAK,IAAI,EAAE;gBACvH,MAAM,cAAc,CAAC,YAAY,CAC7B;oBACI,KAAK,EAAE,SAAS;oBAChB,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;oBAC3B,YAAY,EAAE,EAAE,GAAG,EAAE,iBAAiB,EAAE;oBACxC,WAAW,EAAE,CAAC,eAAe,CAAC;iBACjC,EACD,qBAAqB,CACxB,CAAA;gBAED,MAAM,kBAAkB,GACpB,6BAA6B,CAAC,SAAS,CAAC,QAAQ,CAAA;gBAEpD,sCAAsC;gBACtC,MAAM,CAAC,WAAW,CACd,kBAAkB,CAAC,iBAAiB,EAAE,cAAc,EAAE,gBAAgB,EAAE,uBAAuB;oBAC3F,EAAE,WAAW,EACjB,iBAAiB,CACpB,CAAA;YACL,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/chat/chatEventParser.d.ts

@@ -0,0 +1,24 @@
+import { ChatResponseStream, Reference, SupplementaryWebLink } from '@amzn/codewhisperer-streaming';
+import { ChatItemAction, ChatResult, ReferenceTrackerInformation, SourceLink } from '@aws/language-server-runtimes/protocol';
+import { Result } from '../types';
+export declare class ChatEventParser implements ChatResult {
+    error?: string;
+    messageId?: string;
+    body?: string;
+    canBeVoted?: boolean;
+    relatedContent?: {
+        title?: string;
+        content: SourceLink[];
+    };
+    followUp?: {
+        text?: string;
+        options?: ChatItemAction[];
+    };
+    codeReference?: ReferenceTrackerInformation[];
+    static getReferencedInformation(reference: Reference): string;
+    static mapRelatedData({ title, url, snippet }: SupplementaryWebLink): SourceLink;
+    static mapReferenceData(reference: Reference): ReferenceTrackerInformation;
+    constructor(messageId: string);
+    processPartialEvent(chatEvent: ChatResponseStream): Result<ChatResult, string>;
+    getChatResult(): Result<ChatResult, string>;
+}

package/out/language-server/chat/chatEventParser.js

@@ -0,0 +1,84 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatEventParser = void 0;
+class ChatEventParser {
+    static getReferencedInformation(reference) {
+        return `Reference code under **${reference.licenseName}** license from repository \`${reference.repository}\``;
+    }
+    static mapRelatedData({ title = '', url = '', snippet }) {
+        return {
+            title,
+            url,
+            body: snippet,
+        };
+    }
+    static mapReferenceData(reference) {
+        return {
+            ...reference,
+            information: ChatEventParser.getReferencedInformation(reference),
+        };
+    }
+    constructor(messageId) {
+        this.messageId = messageId;
+    }
+    processPartialEvent(chatEvent) {
+        const { followupPromptEvent, supplementaryWebLinksEvent, codeReferenceEvent, assistantResponseEvent, error, invalidStateEvent, } = chatEvent;
+        if (error) {
+            this.error = error.message;
+        }
+        else if (invalidStateEvent) {
+            this.error = invalidStateEvent.message ?? invalidStateEvent.reason ?? 'Invalid state';
+        }
+        else if (assistantResponseEvent?.content) {
+            this.body = (this.body ?? '') + assistantResponseEvent.content;
+        }
+        else if (followupPromptEvent?.followupPrompt) {
+            const { content } = followupPromptEvent.followupPrompt;
+            this.followUp = {
+                ...this.followUp,
+                options: [
+                    ...(this.followUp?.options ?? []),
+                    {
+                        pillText: content ?? '',
+                        prompt: content ?? '',
+                    },
+                ],
+            };
+        }
+        else if (supplementaryWebLinksEvent?.supplementaryWebLinks &&
+            supplementaryWebLinksEvent.supplementaryWebLinks.length > 0) {
+            const sourceLinks = supplementaryWebLinksEvent.supplementaryWebLinks.map(ChatEventParser.mapRelatedData);
+            this.relatedContent = {
+                ...this.relatedContent,
+                content: [...(this.relatedContent?.content ?? []), ...sourceLinks],
+            };
+        }
+        else if (codeReferenceEvent?.references && codeReferenceEvent.references.length > 0) {
+            const references = codeReferenceEvent.references.map(ChatEventParser.mapReferenceData);
+            this.codeReference = [...(this.codeReference ?? []), ...references];
+        }
+        return this.getChatResult();
+    }
+    getChatResult() {
+        const chatResult = {
+            messageId: this.messageId,
+            body: this.body,
+            canBeVoted: this.canBeVoted,
+            relatedContent: this.relatedContent,
+            followUp: this.followUp,
+            codeReference: this.codeReference,
+        };
+        return this.error
+            ? {
+                success: false,
+                data: chatResult,
+                error: this.error,
+            }
+            : {
+                success: true,
+                data: chatResult,
+            };
+    }
+}
+exports.ChatEventParser = ChatEventParser;
+//# sourceMappingURL=chatEventParser.js.map

package/out/language-server/chat/chatEventParser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatEventParser.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatEventParser.ts"],"names":[],"mappings":";;;AASA,MAAa,eAAe;IASxB,MAAM,CAAC,wBAAwB,CAAC,SAAoB;QAChD,OAAO,0BAA0B,SAAS,CAAC,WAAW,gCAAgC,SAAS,CAAC,UAAU,IAAI,CAAA;IAClH,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,EAAE,OAAO,EAAwB;QACzE,OAAO;YACH,KAAK;YACL,GAAG;YACH,IAAI,EAAE,OAAO;SAChB,CAAA;IACL,CAAC;IACD,MAAM,CAAC,gBAAgB,CAAC,SAAoB;QACxC,OAAO;YACH,GAAG,SAAS;YACZ,WAAW,EAAE,eAAe,CAAC,wBAAwB,CAAC,SAAS,CAAC;SACnE,CAAA;IACL,CAAC;IAED,YAAY,SAAiB;QACzB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC9B,CAAC;IAEM,mBAAmB,CAAC,SAA6B;QACpD,MAAM,EACF,mBAAmB,EACnB,0BAA0B,EAC1B,kBAAkB,EAClB,sBAAsB,EACtB,KAAK,EACL,iBAAiB,GACpB,GAAG,SAAS,CAAA;QAEb,IAAI,KAAK,EAAE,CAAC;YACR,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAA;QAC9B,CAAC;aAAM,IAAI,iBAAiB,EAAE,CAAC;YAC3B,IAAI,CAAC,KAAK,GAAG,iBAAiB,CAAC,OAAO,IAAI,iBAAiB,CAAC,MAAM,IAAI,eAAe,CAAA;QACzF,CAAC;aAAM,IAAI,sBAAsB,EAAE,OAAO,EAAE,CAAC;YACzC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,sBAAsB,CAAC,OAAO,CAAA;QAClE,CAAC;aAAM,IAAI,mBAAmB,EAAE,cAAc,EAAE,CAAC;YAC7C,MAAM,EAAE,OAAO,EAAE,GAAG,mBAAmB,CAAC,cAAc,CAAA;YAEtD,IAAI,CAAC,QAAQ,GAAG;gBACZ,GAAG,IAAI,CAAC,QAAQ;gBAChB,OAAO,EAAE;oBACL,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;oBACjC;wBACI,QAAQ,EAAE,OAAO,IAAI,EAAE;wBACvB,MAAM,EAAE,OAAO,IAAI,EAAE;qBACxB;iBACJ;aACJ,CAAA;QACL,CAAC;aAAM,IACH,0BAA0B,EAAE,qBAAqB;YACjD,0BAA0B,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,EAC7D,CAAC;YACC,MAAM,WAAW,GAAG,0BAA0B,CAAC,qBAAqB,CAAC,GAAG,CAAC,eAAe,CAAC,cAAc,CAAC,CAAA;YAExG,IAAI,CAAC,cAAc,GAAG;gBAClB,GAAG,IAAI,CAAC,cAAc;gBACtB,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC;aACrE,CAAA;QACL,CAAC;aAAM,IAAI,kBAAkB,EAAE,UAAU,IAAI,kBAAkB,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpF,MAAM,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAA;YACtF,IAAI,CAAC,aAAa,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,GAAG,UAAU,CAAC,CAAA;QACvE,CAAC;QAED,OAAO,IAAI,CAAC,aAAa,EAAE,CAAA;IAC/B,CAAC;IAEM,aAAa;QAChB,MAAM,UAAU,GAAe;YAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,aAAa,EAAE,IAAI,CAAC,aAAa;SACpC,CAAA;QAED,OAAO,IAAI,CAAC,KAAK;YACb,CAAC,CAAC;gBACI,OAAO,EAAE,KAAK;gBACd,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,IAAI,CAAC,KAAK;aACpB;YACH,CAAC,CAAC;gBACI,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,UAAU;aACnB,CAAA;IACX,CAAC;CACJ;AAnGD,0CAmGC"}

package/out/language-server/chat/chatEventParser.test.d.ts

@@ -0,0 +1 @@
+export {};