@aws/lsp-codewhisperer 0.0.5 → 0.0.9

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

@@ -10,60 +10,107 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     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;
+var _ChatController_instances, _ChatController_features, _ChatController_chatSessionManagementService, _ChatController_telemetryController, _ChatController_triggerContext, _ChatController_getTriggerContext, _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 uuid_1 = require("uuid");
+const types_1 = require("../telemetry/types");
 const chatEventParser_1 = require("./chatEventParser");
-const documentContext_1 = require("./contexts/documentContext");
-const utils_1 = require("./utils");
+const chatTelemetryController_1 = require("./telemetry/chatTelemetryController");
+const quickActions_1 = require("./quickActions");
+const utils_1 = require("../utils");
+const metric_1 = require("../telemetry/metric");
+const triggerContext_1 = require("./contexts/triggerContext");
 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);
+        _ChatController_telemetryController.set(this, void 0);
+        _ChatController_triggerContext.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");
+        __classPrivateFieldSet(this, _ChatController_triggerContext, new triggerContext_1.QChatTriggerContext(features.workspace, features.logging), "f");
+        __classPrivateFieldSet(this, _ChatController_telemetryController, new chatTelemetryController_1.ChatTelemetryController(features), "f");
+    }
+    dispose() {
+        __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").dispose();
+        __classPrivateFieldGet(this, _ChatController_triggerContext, "f").dispose();
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").dispose();
     }
     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');
+        const { data: session, success } = sessionResult;
+        if (!success) {
+            return new server_interface_1.ResponseError(protocol_1.ErrorCodes.InternalError, sessionResult.error);
         }
+        const metric = new metric_1.Metric({
+            cwsprChatConversationType: 'Chat',
+        });
+        const triggerContext = await __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_getTriggerContext).call(this, params, metric);
+        const isNewConversation = !session.sessionId;
         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;
+        const conversationIdentifier = session?.sessionId ?? 'New session';
         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));
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Request for conversation id:', conversationIdentifier);
+            const requestInput = __classPrivateFieldGet(this, _ChatController_triggerContext, "f").getChatParamsFromTrigger(params, triggerContext);
+            metric.recordStart();
+            response = await session.generateAssistantResponse(requestInput);
+            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'Response for conversationId:', conversationIdentifier, JSON.stringify(response.$metadata));
         }
         catch (err) {
+            if ((0, utils_1.isAwsError)(err) || ((0, utils_1.isObject)(err) && 'statusCode' in err && typeof err.statusCode === 'number')) {
+                metric.setDimension('cwsprChatRepsonseCode', err.statusCode ?? 400);
+                __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitMessageResponseError(params.tabId, metric.metric);
+            }
+            const authFollowType = (0, utils_1.getAuthFollowUpType)(err);
+            if (authFollowType) {
+                __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, `Q auth error: ${(0, utils_1.getErrorMessage)(err)}`);
+                return (0, utils_1.createAuthFollowUpResult)(authFollowType);
+            }
             __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');
+            return new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, err instanceof Error ? err.message : 'Unknown request error');
+        }
+        if (response.conversationId) {
+            __classPrivateFieldGet(this, _ChatController_telemetryController, "f").setConversationId(params.tabId, response.conversationId);
+            if (isNewConversation) {
+                __classPrivateFieldGet(this, _ChatController_telemetryController, "f").updateTriggerInfo(params.tabId, {
+                    startTrigger: {
+                        hasUserSnippet: metric.metric.cwsprChatHasCodeSnippet ?? false,
+                        triggerType: triggerContext.triggerType,
+                    },
+                });
+                __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitStartConversationMetric(params.tabId, metric.metric);
+            }
         }
         try {
-            const result = await __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_processAssistantResponse).call(this, response, params.partialResultToken);
+            const result = await __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_processAssistantResponse).call(this, response, metric.mergeWith({
+                cwsprChatResponseCode: response.$metadata.httpStatusCode,
+                cwsprChatMessageId: response.$metadata.requestId,
+            }), params.partialResultToken);
+            __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitAddMessageMetric(params.tabId, metric.metric);
+            __classPrivateFieldGet(this, _ChatController_telemetryController, "f").updateTriggerInfo(params.tabId, {
+                lastMessageTrigger: {
+                    ...triggerContext,
+                    messageId: response.$metadata.requestId,
+                    followUpActions: new Set(result.data?.followUp?.options
+                        ?.map(option => option.prompt ?? '')
+                        .filter(prompt => prompt.length > 0)),
+                },
+            });
             return result.success
                 ? result.data
-                : new server_interface_1.ResponseError(server_interface_1.ErrorCodes.InternalError, result.error, result.data);
+                : new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, result.error);
         }
         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');
+            return new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, err instanceof Error ? err.message : 'Unknown error occured during response stream');
         }
     }
     onCodeInsertToCursorPosition() { }
@@ -75,32 +122,110 @@ class ChatController {
     onFollowUpClicked() { }
     onInfoLinkClick() { }
     onLinkClick() { }
-    onSendFeedback() { }
-    onQuickAction() {
-        throw new Error('Method not implemented.');
-    }
     onReady() { }
+    onSendFeedback({ tabId, feedbackPayload }) {
+        __classPrivateFieldGet(this, _ChatController_features, "f").telemetry.emitMetric({
+            name: 'amazonq_sendFeedback',
+            data: {
+                comment: JSON.stringify({
+                    type: 'codewhisperer-chat-answer-feedback',
+                    conversationId: __classPrivateFieldGet(this, _ChatController_telemetryController, "f").getConversationId(tabId) ?? '',
+                    messageId: feedbackPayload.messageId,
+                    reason: feedbackPayload.selectedOption,
+                    userComment: feedbackPayload.comment,
+                }),
+                sentiment: 'Negative',
+            },
+        });
+    }
     onSourceLinkClick() { }
     onTabAdd(params) {
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").activeTabId = params.tabId;
         __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").createSession(params.tabId);
     }
-    onTabChange() { }
+    onTabChange(params) {
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.ExitFocusConversation,
+            data: {},
+        });
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").activeTabId = params.tabId;
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.EnterFocusConversation,
+            data: {},
+        });
+    }
     onTabRemove(params) {
+        if (__classPrivateFieldGet(this, _ChatController_telemetryController, "f").activeTabId === params.tabId) {
+            __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitConversationMetric({
+                name: types_1.ChatTelemetryEventName.ExitFocusConversation,
+                data: {},
+            });
+            __classPrivateFieldGet(this, _ChatController_telemetryController, "f").activeTabId = undefined;
+        }
         __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").deleteSession(params.tabId);
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").removeConversation(params.tabId);
+    }
+    onQuickAction(params, _cancellationToken) {
+        switch (params.quickAction) {
+            case quickActions_1.QuickAction.Clear: {
+                const sessionResult = __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").getSession(params.tabId);
+                __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitChatMetric({
+                    name: types_1.ChatTelemetryEventName.RunCommand,
+                    data: {
+                        cwsprChatCommandType: params.quickAction,
+                    },
+                });
+                __classPrivateFieldGet(this, _ChatController_telemetryController, "f").removeConversation(params.tabId);
+                sessionResult.data?.clear();
+                return {};
+            }
+            case quickActions_1.QuickAction.Help:
+                __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitChatMetric({
+                    name: types_1.ChatTelemetryEventName.RunCommand,
+                    data: {
+                        cwsprChatCommandType: params.quickAction,
+                    },
+                });
+                return {
+                    messageId: (0, uuid_1.v4)(),
+                    body: quickActions_1.HELP_MESSAGE,
+                };
+            default:
+                return {};
+        }
     }
-    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;
+_ChatController_features = new WeakMap(), _ChatController_chatSessionManagementService = new WeakMap(), _ChatController_telemetryController = new WeakMap(), _ChatController_triggerContext = new WeakMap(), _ChatController_instances = new WeakSet(), _ChatController_getTriggerContext = async function _ChatController_getTriggerContext(params, metric) {
+    const lastMessageTrigger = __classPrivateFieldGet(this, _ChatController_telemetryController, "f").getLastMessageTrigger(params.tabId);
+    let triggerContext;
+    // this is the only way we can detect a follow up action
+    // we can reuse previous trigger information
+    if (lastMessageTrigger?.followUpActions?.has(params.prompt?.prompt ?? '')) {
+        __classPrivateFieldGet(this, _ChatController_telemetryController, "f").emitInteractWithMessageMetric(params.tabId, {
+            cwsprChatMessageId: lastMessageTrigger.messageId,
+            cwsprChatInteractionType: types_1.ChatInteractionType.ClickFollowUp,
+        });
+        triggerContext = lastMessageTrigger;
+    }
+    else {
+        triggerContext = await __classPrivateFieldGet(this, _ChatController_triggerContext, "f").getNewTriggerContext(params);
+        triggerContext.triggerType = __classPrivateFieldGet(this, _ChatController_telemetryController, "f").getCurrentTrigger(params.tabId) ?? 'click';
     }
-    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);
+    metric.mergeWith({
+        cwsprChatUserIntent: triggerContext?.userIntent,
+        cwsprChatProgrammingLanguage: triggerContext?.programmingLanguage?.languageName,
+        cwsprChatRequestLength: params.prompt?.prompt?.length ?? 0,
+        cwsprChatTriggerInteraction: triggerContext?.triggerType,
+        cwsprChatHasCodeSnippet: triggerContext.hasCodeSnippet ?? false,
+        cwsprChatActiveEditorImportCount: triggerContext.documentSymbols?.length ?? 0,
+        cwsprChatActiveEditorTotalCharacters: triggerContext.totalEditorCharacters ?? 0,
+    });
+    return triggerContext;
+}, _ChatController_processAssistantResponse = async function _ChatController_processAssistantResponse(response, metric, partialResultToken) {
+    const requestId = response.$metadata.requestId;
+    const chatEventParser = new chatEventParser_1.ChatEventParser(requestId, metric);
     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) {
@@ -110,6 +235,13 @@ _ChatController_features = new WeakMap(), _ChatController_chatSessionManagementS
             __classPrivateFieldGet(this, _ChatController_features, "f").lsp.sendProgress(protocol_1.chatRequestType, partialResultToken, chatResult.data);
         }
     }
+    metric.mergeWith({
+        cwsprChatFullResponseLatency: metric.getTimeElapsed(),
+        cwsprChatFollowUpCount: chatEventParser.totalEvents.followupPromptEvent,
+        cwsprChatReferencesCount: chatEventParser.totalEvents.codeReferenceEvent,
+        cwsprChatSourceLinkCount: chatEventParser.totalEvents.supplementaryWebLinksEvent,
+        cwsprChatResponseLength: chatEventParser.body?.length ?? 0,
+    });
     return chatEventParser.getChatResult();
 }, _ChatController_log = function _ChatController_log(...messages) {
     __classPrivateFieldGet(this, _ChatController_features, "f").logging.log(messages.join(' '));

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

@@ -1 +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"}
+{"version":3,"file":"chatController.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatController.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,qEAAoG;AACpG,qFAYuD;AACvD,+BAAiC;AACjC,8CAK2B;AAE3B,uDAAmD;AAEnD,iFAA6E;AAC7E,iDAA0D;AAC1D,oCAA+G;AAC/G,gDAA4C;AAC5C,8DAA+E;AAI/E,MAAa,cAAc;IAMvB,YAAY,4BAA0D,EAAE,QAAkB;;QAL1F,2CAAmB;QACnB,+DAA2D;QAC3D,sDAA6C;QAC7C,iDAAoC;QAGhC,uBAAA,IAAI,4BAAa,QAAQ,MAAA,CAAA;QACzB,uBAAA,IAAI,gDAAiC,4BAA4B,MAAA,CAAA;QACjE,uBAAA,IAAI,kCAAmB,IAAI,oCAAmB,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAA,CAAA;QACpF,uBAAA,IAAI,uCAAwB,IAAI,iDAAuB,CAAC,QAAQ,CAAC,MAAA,CAAA;IACrE,CAAC;IAED,OAAO;QACH,uBAAA,IAAI,oDAA8B,CAAC,OAAO,EAAE,CAAA;QAC5C,uBAAA,IAAI,sCAAgB,CAAC,OAAO,EAAE,CAAA;QAC9B,uBAAA,IAAI,2CAAqB,CAAC,OAAO,EAAE,CAAA;IACvC,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,OAAO,EAAE,GAAG,aAAa,CAAA;QAEhD,IAAI,CAAC,OAAO,EAAE;YACV,OAAO,IAAI,gCAAa,CAAa,qBAAU,CAAC,aAAa,EAAE,aAAa,CAAC,KAAK,CAAC,CAAA;SACtF;QAED,MAAM,MAAM,GAAG,IAAI,eAAM,CAA4B;YACjD,yBAAyB,EAAE,MAAM;SACpC,CAAC,CAAA;QAEF,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,EAAoB,MAAM,EAAE,MAAM,CAAC,CAAA;QACpE,MAAM,iBAAiB,GAAG,CAAC,OAAO,CAAC,SAAS,CAAA;QAE5C,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,IAAI,QAAgD,CAAA;QAEpD,MAAM,sBAAsB,GAAG,OAAO,EAAE,SAAS,IAAI,aAAa,CAAA;QAClE,IAAI;YACA,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,8BAA8B,EAAE,sBAAsB,CAAC,CAAA;YACjE,MAAM,YAAY,GAAG,uBAAA,IAAI,sCAAgB,CAAC,wBAAwB,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;YAE1F,MAAM,CAAC,WAAW,EAAE,CAAA;YACpB,QAAQ,GAAG,MAAM,OAAO,CAAC,yBAAyB,CAAC,YAAY,CAAC,CAAA;YAChE,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,8BAA8B,EAAE,sBAAsB,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAA;SACxG;QAAC,OAAO,GAAG,EAAE;YACV,IAAI,IAAA,kBAAU,EAAC,GAAG,CAAC,IAAI,CAAC,IAAA,gBAAQ,EAAC,GAAG,CAAC,IAAI,YAAY,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ,CAAC,EAAE;gBACjG,MAAM,CAAC,YAAY,CAAC,uBAAuB,EAAE,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,CAAA;gBACnE,uBAAA,IAAI,2CAAqB,CAAC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;aAClF;YAED,MAAM,cAAc,GAAG,IAAA,2BAAmB,EAAC,GAAG,CAAC,CAAA;YAE/C,IAAI,cAAc,EAAE;gBAChB,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,iBAAiB,IAAA,uBAAe,EAAC,GAAG,CAAC,EAAE,CAAC,CAAA;gBAElD,OAAO,IAAA,gCAAwB,EAAC,cAAc,CAAC,CAAA;aAClD;YAED,uBAAA,IAAI,sDAAK,MAAT,IAAI,EAAM,uBAAuB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAA;YAClF,OAAO,IAAI,gCAAa,CACpB,gCAAa,CAAC,aAAa,EAC3B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAC/D,CAAA;SACJ;QAED,IAAI,QAAQ,CAAC,cAAc,EAAE;YACzB,uBAAA,IAAI,2CAAqB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,cAAc,CAAC,CAAA;YAElF,IAAI,iBAAiB,EAAE;gBACnB,uBAAA,IAAI,2CAAqB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE;oBACtD,YAAY,EAAE;wBACV,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,uBAAuB,IAAI,KAAK;wBAC9D,WAAW,EAAE,cAAc,CAAC,WAAW;qBAC1C;iBACJ,CAAC,CAAA;gBAEF,uBAAA,IAAI,2CAAqB,CAAC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;aACrF;SACJ;QAED,IAAI;YACA,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,2EAA0B,MAA9B,IAAI,EACrB,QAAQ,EACR,MAAM,CAAC,SAAS,CAAC;gBACb,qBAAqB,EAAE,QAAQ,CAAC,SAAS,CAAC,cAAc;gBACxD,kBAAkB,EAAE,QAAQ,CAAC,SAAS,CAAC,SAAS;aACnD,CAAC,EACF,MAAM,CAAC,kBAAkB,CAC5B,CAAA;YAED,uBAAA,IAAI,2CAAqB,CAAC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;YAE3E,uBAAA,IAAI,2CAAqB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE;gBACtD,kBAAkB,EAAE;oBAChB,GAAG,cAAc;oBACjB,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,SAAS;oBACvC,eAAe,EAAE,IAAI,GAAG,CACpB,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO;wBAC1B,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;yBACnC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAC3C;iBACJ;aACJ,CAAC,CAAA;YAEF,OAAO,MAAM,CAAC,OAAO;gBACjB,CAAC,CAAC,MAAM,CAAC,IAAI;gBACb,CAAC,CAAC,IAAI,gCAAa,CAAa,gCAAa,CAAC,aAAa,EAAE,MAAM,CAAC,KAAK,CAAC,CAAA;SACjF;QAAC,OAAO,GAAG,EAAE;YACV,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,gCAAa,CAAC,aAAa,EAC3B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,8CAA8C,CACtF,CAAA;SACJ;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,OAAO,KAAI,CAAC;IAEZ,cAAc,CAAC,EAAE,KAAK,EAAE,eAAe,EAAkB;QACrD,uBAAA,IAAI,gCAAU,CAAC,SAAS,CAAC,UAAU,CAAC;YAChC,IAAI,EAAE,sBAAsB;YAC5B,IAAI,EAAE;gBACF,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;oBACpB,IAAI,EAAE,oCAAoC;oBAC1C,cAAc,EAAE,uBAAA,IAAI,2CAAqB,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,EAAE;oBACxE,SAAS,EAAE,eAAe,CAAC,SAAS;oBACpC,MAAM,EAAE,eAAe,CAAC,cAAc;oBACtC,WAAW,EAAE,eAAe,CAAC,OAAO;iBACvC,CAAC;gBACF,SAAS,EAAE,UAAU;aACxB;SACJ,CAAC,CAAA;IACN,CAAC;IAED,iBAAiB,KAAI,CAAC;IAEtB,QAAQ,CAAC,MAAoB;QACzB,uBAAA,IAAI,2CAAqB,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAA;QAEpD,uBAAA,IAAI,oDAA8B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAClE,CAAC;IAED,WAAW,CAAC,MAAuB;QAC/B,uBAAA,IAAI,2CAAqB,CAAC,sBAAsB,CAAC;YAC7C,IAAI,EAAE,8BAAsB,CAAC,qBAAqB;YAClD,IAAI,EAAE,EAAE;SACX,CAAC,CAAA;QAEF,uBAAA,IAAI,2CAAqB,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAA;QAEpD,uBAAA,IAAI,2CAAqB,CAAC,sBAAsB,CAAC;YAC7C,IAAI,EAAE,8BAAsB,CAAC,sBAAsB;YACnD,IAAI,EAAE,EAAE;SACX,CAAC,CAAA;IACN,CAAC;IAED,WAAW,CAAC,MAAuB;QAC/B,IAAI,uBAAA,IAAI,2CAAqB,CAAC,WAAW,KAAK,MAAM,CAAC,KAAK,EAAE;YACxD,uBAAA,IAAI,2CAAqB,CAAC,sBAAsB,CAAC;gBAC7C,IAAI,EAAE,8BAAsB,CAAC,qBAAqB;gBAClD,IAAI,EAAE,EAAE;aACX,CAAC,CAAA;YACF,uBAAA,IAAI,2CAAqB,CAAC,WAAW,GAAG,SAAS,CAAA;SACpD;QAED,uBAAA,IAAI,oDAA8B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC9D,uBAAA,IAAI,2CAAqB,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAC9D,CAAC;IAED,aAAa,CAAC,MAAyB,EAAE,kBAAqC;QAC1E,QAAQ,MAAM,CAAC,WAAW,EAAE;YACxB,KAAK,0BAAW,CAAC,KAAK,CAAC,CAAC;gBACpB,MAAM,aAAa,GAAG,uBAAA,IAAI,oDAA8B,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBAEjF,uBAAA,IAAI,2CAAqB,CAAC,cAAc,CAAC;oBACrC,IAAI,EAAE,8BAAsB,CAAC,UAAU;oBACvC,IAAI,EAAE;wBACF,oBAAoB,EAAE,MAAM,CAAC,WAAW;qBAC3C;iBACJ,CAAC,CAAA;gBAEF,uBAAA,IAAI,2CAAqB,CAAC,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBAE1D,aAAa,CAAC,IAAI,EAAE,KAAK,EAAE,CAAA;gBAE3B,OAAO,EAAE,CAAA;aACZ;YAED,KAAK,0BAAW,CAAC,IAAI;gBACjB,uBAAA,IAAI,2CAAqB,CAAC,cAAc,CAAC;oBACrC,IAAI,EAAE,8BAAsB,CAAC,UAAU;oBACvC,IAAI,EAAE;wBACF,oBAAoB,EAAE,MAAM,CAAC,WAAW;qBAC3C;iBACJ,CAAC,CAAA;gBACF,OAAO;oBACH,SAAS,EAAE,IAAA,SAAI,GAAE;oBACjB,IAAI,EAAE,2BAAY;iBACrB,CAAA;YACL;gBACI,OAAO,EAAE,CAAA;SAChB;IACL,CAAC;CAqEJ;AArSD,wCAqSC;4RAnEG,KAAK,4CAAoB,MAAkB,EAAE,MAAyC;IAClF,MAAM,kBAAkB,GAAG,uBAAA,IAAI,2CAAqB,CAAC,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAExF,IAAI,cAA8B,CAAA;IAElC,wDAAwD;IACxD,4CAA4C;IAC5C,IAAI,kBAAkB,EAAE,eAAe,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC,EAAE;QACvE,uBAAA,IAAI,2CAAqB,CAAC,6BAA6B,CAAC,MAAM,CAAC,KAAK,EAAE;YAClE,kBAAkB,EAAE,kBAAkB,CAAC,SAAU;YACjD,wBAAwB,EAAE,2BAAmB,CAAC,aAAa;SAC9D,CAAC,CAAA;QAEF,cAAc,GAAG,kBAAkB,CAAA;KACtC;SAAM;QACH,cAAc,GAAG,MAAM,uBAAA,IAAI,sCAAgB,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAA;QACxE,cAAc,CAAC,WAAW,GAAG,uBAAA,IAAI,2CAAqB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,CAAA;KACpG;IAED,MAAM,CAAC,SAAS,CAAC;QACb,mBAAmB,EAAE,cAAc,EAAE,UAAU;QAC/C,4BAA4B,EAAE,cAAc,EAAE,mBAAmB,EAAE,YAAY;QAC/E,sBAAsB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;QAC1D,2BAA2B,EAAE,cAAc,EAAE,WAAW;QACxD,uBAAuB,EAAE,cAAc,CAAC,cAAc,IAAI,KAAK;QAC/D,gCAAgC,EAAE,cAAc,CAAC,eAAe,EAAE,MAAM,IAAI,CAAC;QAC7E,oCAAoC,EAAE,cAAc,CAAC,qBAAqB,IAAI,CAAC;KAClF,CAAC,CAAA;IAEF,OAAO,cAAc,CAAA;AACzB,CAAC,6CAED,KAAK,mDACD,QAAgD,EAChD,MAA+B,EAC/B,kBAAoC;IAEpC,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,SAAU,CAAA;IAC/C,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;IAE9D,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,QAAQ,CAAC,iCAAkC,EAAE;QACvE,MAAM,UAAU,GAAG,eAAe,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAA;QAEjE,yCAAyC;QACzC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE;YACrB,OAAO,UAAU,CAAA;SACpB;QAED,IAAI,kBAAkB,EAAE;YACpB,uBAAA,IAAI,gCAAU,CAAC,GAAG,CAAC,YAAY,CAAC,0BAAe,EAAE,kBAAkB,EAAE,UAAU,CAAC,IAAI,CAAC,CAAA;SACxF;KACJ;IAED,MAAM,CAAC,SAAS,CAAC;QACb,4BAA4B,EAAE,MAAM,CAAC,cAAc,EAAE;QACrD,sBAAsB,EAAE,eAAe,CAAC,WAAW,CAAC,mBAAmB;QACvE,wBAAwB,EAAE,eAAe,CAAC,WAAW,CAAC,kBAAkB;QACxE,wBAAwB,EAAE,eAAe,CAAC,WAAW,CAAC,0BAA0B;QAChF,uBAAuB,EAAE,eAAe,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC;KAC7D,CAAC,CAAA;IAEF,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.js

@@ -9,7 +9,9 @@ const testUtils_1 = require("../testUtils");
 const chatController_1 = require("./chatController");
 const chatSessionManagementService_1 = require("./chatSessionManagementService");
 const chatSessionService_1 = require("./chatSessionService");
+const chatTelemetryController_1 = require("./telemetry/chatTelemetryController");
 const documentContext_1 = require("./contexts/documentContext");
+const utils = require("../utils");
 describe('ChatController', () => {
     const mockTabId = 'tab-1';
     const mockConversationId = 'mock-conversation-id';
@@ -34,7 +36,7 @@ describe('ChatController', () => {
     const expectedCompleteChatResult = {
         messageId: mockMessageId,
         body: 'Hello World!',
-        canBeVoted: undefined,
+        canBeVoted: true,
         codeReference: undefined,
         followUp: undefined,
         relatedContent: undefined,
@@ -45,6 +47,9 @@ describe('ChatController', () => {
     };
     let generateAssistantResponseStub;
     let disposeStub;
+    let activeTabSpy;
+    let removeConversationSpy;
+    let emitConversationMetricStub;
     let testFeatures;
     let chatSessionManagementService;
     let chatController;
@@ -63,13 +68,15 @@ describe('ChatController', () => {
             }));
         });
         testFeatures = new testing_1.TestFeatures();
+        activeTabSpy = ts_sinon_1.default.spy(chatTelemetryController_1.ChatTelemetryController.prototype, 'activeTabId', ['get', 'set']);
+        removeConversationSpy = ts_sinon_1.default.spy(chatTelemetryController_1.ChatTelemetryController.prototype, 'removeConversation');
+        emitConversationMetricStub = ts_sinon_1.default.stub(chatTelemetryController_1.ChatTelemetryController.prototype, 'emitConversationMetric');
         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();
+        ts_sinon_1.default.restore();
         chatSessionManagementService_1.ChatSessionManagementService.reset();
     });
     it('creates a session when a tab add notifcation is received', () => {
@@ -85,27 +92,49 @@ describe('ChatController', () => {
         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,
-        });
+        const hasSession = chatSessionManagementService.hasSession(mockTabId);
+        assert.ok(!hasSession);
     });
-    it('deletes a session by tab id a end chat request is received', () => {
+    it('deletes a session by tab id an 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,
-        });
+        const hasSession = chatSessionManagementService.hasSession(mockTabId);
+        assert.ok(!hasSession);
+    });
+    it('onTabAdd sets active tab id in telemetryController', () => {
+        chatController.onTabAdd({ tabId: mockTabId });
+        ts_sinon_1.default.assert.calledWithExactly(activeTabSpy.set, mockTabId);
+    });
+    it('onTabChange sets active tab id in telemetryController and emits metrics', () => {
+        chatController.onTabChange({ tabId: mockTabId });
+        ts_sinon_1.default.assert.calledWithExactly(activeTabSpy.set, mockTabId);
+        ts_sinon_1.default.assert.calledTwice(emitConversationMetricStub);
+    });
+    it('onTabRemove unsets tab id if current tab is removed and emits metrics', () => {
+        chatController.onTabAdd({ tabId: mockTabId });
+        emitConversationMetricStub.resetHistory();
+        activeTabSpy.set.resetHistory();
+        chatController.onTabRemove({ tabId: mockTabId });
+        ts_sinon_1.default.assert.calledWithExactly(removeConversationSpy, mockTabId);
+        ts_sinon_1.default.assert.calledOnce(emitConversationMetricStub);
+    });
+    it('onTabRemove does not unset tabId if current tab is not being removed', () => {
+        chatController.onTabAdd({ tabId: mockTabId });
+        chatController.onTabAdd({ tabId: 'mockTabId-2' });
+        testFeatures.telemetry.emitMetric.resetHistory();
+        activeTabSpy.set.resetHistory();
+        chatController.onTabRemove({ tabId: mockTabId });
+        ts_sinon_1.default.assert.notCalled(activeTabSpy.set);
+        ts_sinon_1.default.assert.calledWithExactly(removeConversationSpy, mockTabId);
+        ts_sinon_1.default.assert.notCalled(emitConversationMetricStub);
     });
     describe('onChatPrompt', () => {
         beforeEach(() => {
             chatController.onTabAdd({ tabId: mockTabId });
         });
-        it('throw error if session is not found', async () => {
+        it("throws error if credentials provider doesn't exist", async () => {
+            chatSessionManagementService_1.ChatSessionManagementService.getInstance().withCredentialsProvider(undefined);
             const result = await chatController.onChatPrompt({ tabId: 'XXXX', prompt: { prompt: 'Hello' } }, mockCancellationToken);
             assert.ok(result instanceof server_interface_1.ResponseError);
         });
@@ -121,11 +150,6 @@ describe('ChatController', () => {
             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');
@@ -133,6 +157,16 @@ describe('ChatController', () => {
             const chatResult = await chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' } }, mockCancellationToken);
             assert.ok(chatResult instanceof server_interface_1.ResponseError);
         });
+        it('returns a auth follow up action if generateAssistantResponse returns an auth error', async () => {
+            generateAssistantResponseStub.callsFake(() => {
+                throw new Error('Error');
+            });
+            ts_sinon_1.default.stub(utils, 'getAuthFollowUpType').returns('full-auth');
+            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, 0);
+            assert.deepStrictEqual(chatResult, utils.createAuthFollowUpResult('full-auth'));
+        });
         it('returns a ResponseError if response streams return an error event', async () => {
             generateAssistantResponseStub.callsFake(() => {
                 return Promise.resolve({
@@ -150,10 +184,7 @@ describe('ChatController', () => {
                 });
             });
             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',
-            }));
+            assert.deepStrictEqual(chatResult, new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, 'some error'));
         });
         it('returns a ResponseError if response streams return an invalid state event', async () => {
             generateAssistantResponseStub.callsFake(() => {
@@ -172,15 +203,11 @@ describe('ChatController', () => {
                 });
             });
             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',
-            }));
+            assert.deepStrictEqual(chatResult, new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, 'invalid state'));
         });
-        describe('#extractEditorState', () => {
+        describe('#extractDocumentContext', () => {
             const typescriptDocument = server_interface_1.TextDocument.create('file:///test.ts', 'typescript', 1, 'test');
-            let extractEditorStateStub;
-            const editorStateObject = {};
+            let extractDocumentContextStub;
             const mockCursorState = {
                 range: {
                     start: {
@@ -194,14 +221,19 @@ describe('ChatController', () => {
                 },
             };
             beforeEach(() => {
-                extractEditorStateStub = ts_sinon_1.default.stub(documentContext_1.DocumentContextExtractor.prototype, 'extractEditorState');
+                extractDocumentContextStub = ts_sinon_1.default.stub(documentContext_1.DocumentContextExtractor.prototype, 'extractDocumentContext');
                 testFeatures.openDocument(typescriptDocument);
-                extractEditorStateStub.resolves(editorStateObject);
             });
             afterEach(() => {
-                extractEditorStateStub.restore();
+                extractDocumentContextStub.restore();
             });
             it('leaves editor state as undefined if cursorState is not passed', async () => {
+                const documentContextObject = {
+                    programmingLanguage: 'typescript',
+                    cursorState: undefined,
+                    relativeFilePath: 'file:///test.ts',
+                };
+                extractDocumentContextStub.resolves(documentContextObject);
                 await chatController.onChatPrompt({
                     tabId: mockTabId,
                     prompt: { prompt: 'Hello' },
@@ -212,7 +244,13 @@ describe('ChatController', () => {
                 assert.strictEqual(calledRequestInput.conversationState?.currentMessage?.userInputMessage?.userInputMessageContext
                     ?.editorState, undefined);
             });
-            it('leaves editor state as undefined if document identified is not passed', async () => {
+            it('leaves editor state as undefined if relative file path is undefined', async () => {
+                const documentContextObject = {
+                    programmingLanguage: 'typescript',
+                    cursorState: [],
+                    relativeFilePath: undefined,
+                };
+                extractDocumentContextStub.resolves(documentContextObject);
                 await chatController.onChatPrompt({
                     tabId: mockTabId,
                     prompt: { prompt: 'Hello' },
@@ -223,6 +261,12 @@ describe('ChatController', () => {
                     ?.editorState, undefined);
             });
             it('parses editor state context and includes as requestInput if both cursor state and text document are found', async () => {
+                const documentContextObject = {
+                    programmingLanguage: 'typescript',
+                    cursorState: [],
+                    relativeFilePath: typescriptDocument.uri,
+                };
+                extractDocumentContextStub.resolves(documentContextObject);
                 await chatController.onChatPrompt({
                     tabId: mockTabId,
                     prompt: { prompt: 'Hello' },
@@ -230,9 +274,16 @@ describe('ChatController', () => {
                     cursorState: [mockCursorState],
                 }, mockCancellationToken);
                 const calledRequestInput = generateAssistantResponseStub.firstCall.firstArg;
-                // asserting object reference equality
-                assert.strictEqual(calledRequestInput.conversationState?.currentMessage?.userInputMessage?.userInputMessageContext
-                    ?.editorState, editorStateObject);
+                assert.deepStrictEqual(calledRequestInput.conversationState?.currentMessage?.userInputMessage?.userInputMessageContext
+                    ?.editorState, {
+                    cursorState: [],
+                    document: {
+                        documentSymbols: undefined,
+                        programmingLanguage: 'typescript',
+                        relativeFilePath: 'file:///test.ts',
+                        text: undefined,
+                    },
+                });
             });
         });
     });

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

@@ -1 +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"}
+{"version":3,"file":"chatController.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatController.test.ts"],"names":[],"mappings":";;AAAA,2EAIsC;AACtC,qFAAuH;AACvH,mEAAoE;AACpE,iCAAgC;AAChC,uCAA4B;AAC5B,4CAAqD;AACrD,qDAAiD;AACjD,iFAA6E;AAC7E,6DAAyD;AACzD,iFAA6E;AAC7E,gEAAqE;AACrE,kCAAiC;AAEjC,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,IAAI;QAChB,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;IAChC,IAAI,YAGH,CAAA;IACD,IAAI,qBAAqC,CAAA;IACzC,IAAI,0BAA2C,CAAA;IAE/C,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,YAAY,GAAG,kBAAK,CAAC,GAAG,CAAC,iDAAuB,CAAC,SAAS,EAAE,aAAa,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAA;QAC1F,qBAAqB,GAAG,kBAAK,CAAC,GAAG,CAAC,iDAAuB,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAA;QAC1F,0BAA0B,GAAG,kBAAK,CAAC,IAAI,CAAC,iDAAuB,CAAC,SAAS,EAAE,wBAAwB,CAAC,CAAA;QAEpG,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,kBAAK,CAAC,OAAO,EAAE,CAAA;QACf,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,UAAU,GAAG,4BAA4B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAErE,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACnE,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,UAAU,GAAG,4BAA4B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAErE,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC1D,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAE7C,kBAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAA;IAC/D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QAC/E,cAAc,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAEhD,kBAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,YAAY,CAAC,GAAG,EAAE,SAAS,CAAC,CAAA;QAC3D,kBAAK,CAAC,MAAM,CAAC,WAAW,CAAC,0BAA0B,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC7E,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAE7C,0BAA0B,CAAC,YAAY,EAAE,CAAA;QACzC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,CAAA;QAE/B,cAAc,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAEhD,kBAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAA;QAChE,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC5E,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAC7C,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAA;QAEjD,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,YAAY,EAAE,CAAA;QAChD,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,CAAA;QAE/B,cAAc,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAEhD,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAA;QACxC,kBAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAA;QAChE,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAA;IACtD,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,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAChE,2DAA4B,CAAC,WAAW,EAAE,CAAC,uBAAuB,CAAC,SAAgB,CAAC,CAAA;YACpF,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,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,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAChG,6BAA6B,CAAC,SAAS,CAAC,GAAG,EAAE;gBACzC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAA;YAC5B,CAAC,CAAC,CAAA;YAEF,kBAAK,CAAC,IAAI,CAAC,KAAK,EAAE,qBAAqB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAA;YAC7D,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,CAAC,CAAC,CAAA;YACxD,MAAM,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,wBAAwB,CAAC,WAAW,CAAC,CAAC,CAAA;QACnF,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,CAAC,UAAU,EAAE,IAAI,gCAAa,CAAC,gCAAa,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC,CAAA;QACpG,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,CAAC,UAAU,EAAE,IAAI,gCAAa,CAAC,gCAAa,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC,CAAA;QACvG,CAAC,CAAC,CAAA;QAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACrC,MAAM,kBAAkB,GAAG,+BAAY,CAAC,MAAM,CAAC,iBAAiB,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,CAAC,CAAA;YAC1F,IAAI,0BAA2C,CAAA;YAE/C,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,0BAA0B,GAAG,kBAAK,CAAC,IAAI,CAAC,0CAAwB,CAAC,SAAS,EAAE,wBAAwB,CAAC,CAAA;gBACrG,YAAY,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAA;YACjD,CAAC,CAAC,CAAA;YAEF,SAAS,CAAC,GAAG,EAAE;gBACX,0BAA0B,CAAC,OAAO,EAAE,CAAA;YACxC,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;gBAC3E,MAAM,qBAAqB,GAAG;oBAC1B,mBAAmB,EAAE,YAAY;oBACjC,WAAW,EAAE,SAAS;oBACtB,gBAAgB,EAAE,iBAAiB;iBACtC,CAAA;gBACD,0BAA0B,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAA;gBAE1D,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,qEAAqE,EAAE,KAAK,IAAI,EAAE;gBACjF,MAAM,qBAAqB,GAAG;oBAC1B,mBAAmB,EAAE,YAAY;oBACjC,WAAW,EAAE,EAAE;oBACf,gBAAgB,EAAE,SAAS;iBAC9B,CAAA;gBACD,0BAA0B,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAA;gBAE1D,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,qBAAqB,GAAG;oBAC1B,mBAAmB,EAAE,YAAY;oBACjC,WAAW,EAAE,EAAE;oBACf,gBAAgB,EAAE,kBAAkB,CAAC,GAAG;iBAC3C,CAAA;gBACD,0BAA0B,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAA;gBAE1D,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,MAAM,CAAC,eAAe,CAClB,kBAAkB,CAAC,iBAAiB,EAAE,cAAc,EAAE,gBAAgB,EAAE,uBAAuB;oBAC3F,EAAE,WAAW,EACjB;oBACI,WAAW,EAAE,EAAE;oBACf,QAAQ,EAAE;wBACN,eAAe,EAAE,SAAS;wBAC1B,mBAAmB,EAAE,YAAY;wBACjC,gBAAgB,EAAE,iBAAiB;wBACnC,IAAI,EAAE,SAAS;qBAClB;iBACJ,CACJ,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

@@ -1,7 +1,11 @@
 import { ChatResponseStream, Reference, SupplementaryWebLink } from '@amzn/codewhisperer-streaming';
 import { ChatItemAction, ChatResult, ReferenceTrackerInformation, SourceLink } from '@aws/language-server-runtimes/protocol';
 import { Result } from '../types';
+import { AddMessageEvent } from '../telemetry/types';
+import { Metric } from '../telemetry/metric';
 export declare class ChatEventParser implements ChatResult {
+    #private;
+    static readonly FOLLOW_UP_TEXT = "Suggested follow up questions:";
     error?: string;
     messageId?: string;
     body?: string;
@@ -18,7 +22,13 @@ export declare class ChatEventParser implements ChatResult {
     static getReferencedInformation(reference: Reference): string;
     static mapRelatedData({ title, url, snippet }: SupplementaryWebLink): SourceLink;
     static mapReferenceData(reference: Reference): ReferenceTrackerInformation;
-    constructor(messageId: string);
+    constructor(messageId: string, metric: Metric<AddMessageEvent>);
+    get totalEvents(): {
+        followupPromptEvent: number;
+        supplementaryWebLinksEvent: number;
+        codeReferenceEvent: number;
+        assistantResponseEvent: number;
+    };
     processPartialEvent(chatEvent: ChatResponseStream): Result<ChatResult, string>;
     getChatResult(): Result<ChatResult, string>;
 }