@aws/lsp-codewhisperer 0.0.6 → 0.0.10

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

@@ -1,117 +1,245 @@
 "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 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_2 = require("../utils");
+const metric_1 = require("../telemetry/metric");
+const triggerContext_1 = require("./contexts/triggerContext");
+const constants_1 = require("./constants");
 class ChatController {
+    #features;
+    #chatSessionManagementService;
+    #telemetryController;
+    #triggerContext;
     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");
+        this.#features = features;
+        this.#chatSessionManagementService = chatSessionManagementService;
+        this.#triggerContext = new triggerContext_1.QChatTriggerContext(features.workspace, features.logging);
+        this.#telemetryController = new chatTelemetryController_1.ChatTelemetryController(features);
+    }
+    dispose() {
+        this.#chatSessionManagementService.dispose();
+        this.#triggerContext.dispose();
+        this.#telemetryController.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 maybeDefaultResponse = (0, utils_1.getDefaultChatResponse)(params.prompt.prompt);
+        if (maybeDefaultResponse) {
+            return maybeDefaultResponse;
+        }
+        const sessionResult = this.#chatSessionManagementService.getSession(params.tabId);
+        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 this.#getTriggerContext(params, metric);
+        const isNewConversation = !session.sessionId;
         token.onCancellationRequested(() => {
-            __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_log).call(this, 'cancellation requested');
+            this.#log('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));
+            this.#log('Request for conversation id:', conversationIdentifier);
+            const requestInput = this.#triggerContext.getChatParamsFromTrigger(params, triggerContext);
+            metric.recordStart();
+            response = await session.generateAssistantResponse(requestInput);
+            this.#log('Response for conversationId:', conversationIdentifier, 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');
+            if ((0, utils_2.isAwsError)(err) || ((0, utils_2.isObject)(err) && 'statusCode' in err && typeof err.statusCode === 'number')) {
+                metric.setDimension('cwsprChatRepsonseCode', err.statusCode ?? 400);
+                this.#telemetryController.emitMessageResponseError(params.tabId, metric.metric);
+            }
+            const authFollowType = (0, utils_1.getAuthFollowUpType)(err);
+            if (authFollowType) {
+                this.#log(`Q auth error: ${(0, utils_2.getErrorMessage)(err)}`);
+                return (0, utils_1.createAuthFollowUpResult)(authFollowType);
+            }
+            this.#log(`Q api request error ${err instanceof Error ? err.message : 'unknown'}`);
+            return new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, err instanceof Error ? err.message : 'Unknown request error');
+        }
+        if (response.conversationId) {
+            this.#telemetryController.setConversationId(params.tabId, response.conversationId);
+            if (isNewConversation) {
+                this.#telemetryController.updateTriggerInfo(params.tabId, {
+                    startTrigger: {
+                        hasUserSnippet: metric.metric.cwsprChatHasCodeSnippet ?? false,
+                        triggerType: triggerContext.triggerType,
+                    },
+                });
+                this.#telemetryController.emitStartConversationMetric(params.tabId, metric.metric);
+            }
         }
         try {
-            const result = await __classPrivateFieldGet(this, _ChatController_instances, "m", _ChatController_processAssistantResponse).call(this, response, params.partialResultToken);
+            const result = await this.#processAssistantResponse(response, metric.mergeWith({
+                cwsprChatResponseCode: response.$metadata.httpStatusCode,
+                cwsprChatMessageId: response.$metadata.requestId,
+            }), params.partialResultToken);
+            this.#telemetryController.emitAddMessageMetric(params.tabId, metric.metric);
+            this.#telemetryController.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');
+            this.#log('Error encountered during response streaming:', err instanceof Error ? err.message : 'unknown');
+            return new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, err instanceof Error ? err.message : 'Unknown error occured during response stream');
         }
     }
     onCodeInsertToCursorPosition() { }
     onCopyCodeToClipboard() { }
     onEndChat(params, _token) {
-        const { success } = __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").deleteSession(params.tabId);
+        const { success } = this.#chatSessionManagementService.deleteSession(params.tabId);
         return success;
     }
     onFollowUpClicked() { }
     onInfoLinkClick() { }
     onLinkClick() { }
-    onSendFeedback() { }
-    onQuickAction() {
-        throw new Error('Method not implemented.');
-    }
     onReady() { }
+    onSendFeedback({ tabId, feedbackPayload }) {
+        this.#features.telemetry.emitMetric({
+            name: 'amazonq_sendFeedback',
+            data: {
+                comment: JSON.stringify({
+                    type: 'codewhisperer-chat-answer-feedback',
+                    conversationId: this.#telemetryController.getConversationId(tabId) ?? '',
+                    messageId: feedbackPayload.messageId,
+                    reason: feedbackPayload.selectedOption,
+                    userComment: feedbackPayload.comment,
+                }),
+                // this is always Negative because only thumbs down has a form
+                sentiment: 'Negative',
+            },
+        });
+    }
     onSourceLinkClick() { }
     onTabAdd(params) {
-        __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").createSession(params.tabId);
+        this.#telemetryController.activeTabId = params.tabId;
+        this.#chatSessionManagementService.createSession(params.tabId);
+    }
+    onTabChange(params) {
+        this.#telemetryController.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.ExitFocusConversation,
+            data: {},
+        });
+        this.#telemetryController.activeTabId = params.tabId;
+        this.#telemetryController.emitConversationMetric({
+            name: types_1.ChatTelemetryEventName.EnterFocusConversation,
+            data: {},
+        });
     }
-    onTabChange() { }
     onTabRemove(params) {
-        __classPrivateFieldGet(this, _ChatController_chatSessionManagementService, "f").deleteSession(params.tabId);
+        if (this.#telemetryController.activeTabId === params.tabId) {
+            this.#telemetryController.emitConversationMetric({
+                name: types_1.ChatTelemetryEventName.ExitFocusConversation,
+                data: {},
+            });
+            this.#telemetryController.activeTabId = undefined;
+        }
+        this.#chatSessionManagementService.deleteSession(params.tabId);
+        this.#telemetryController.removeConversation(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;
+    onQuickAction(params, _cancellationToken) {
+        switch (params.quickAction) {
+            case quickActions_1.QuickAction.Clear: {
+                const sessionResult = this.#chatSessionManagementService.getSession(params.tabId);
+                this.#telemetryController.emitChatMetric({
+                    name: types_1.ChatTelemetryEventName.RunCommand,
+                    data: {
+                        cwsprChatCommandType: params.quickAction,
+                    },
+                });
+                this.#telemetryController.removeConversation(params.tabId);
+                sessionResult.data?.clear();
+                return {};
+            }
+            case quickActions_1.QuickAction.Help:
+                this.#telemetryController.emitChatMetric({
+                    name: types_1.ChatTelemetryEventName.RunCommand,
+                    data: {
+                        cwsprChatCommandType: params.quickAction,
+                    },
+                });
+                return {
+                    messageId: (0, uuid_1.v4)(),
+                    body: constants_1.HELP_MESSAGE,
+                };
+            default:
+                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;
+    async #getTriggerContext(params, metric) {
+        const lastMessageTrigger = this.#telemetryController.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 ?? '')) {
+            this.#telemetryController.emitInteractWithMessageMetric(params.tabId, {
+                cwsprChatMessageId: lastMessageTrigger.messageId,
+                cwsprChatInteractionType: types_1.ChatInteractionType.ClickFollowUp,
+            });
+            triggerContext = lastMessageTrigger;
         }
-        if (partialResultToken) {
-            __classPrivateFieldGet(this, _ChatController_features, "f").lsp.sendProgress(protocol_1.chatRequestType, partialResultToken, chatResult.data);
+        else {
+            triggerContext = await this.#triggerContext.getNewTriggerContext(params);
+            triggerContext.triggerType = this.#telemetryController.getCurrentTrigger(params.tabId) ?? 'click';
         }
+        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;
     }
-    return chatEventParser.getChatResult();
-}, _ChatController_log = function _ChatController_log(...messages) {
-    __classPrivateFieldGet(this, _ChatController_features, "f").logging.log(messages.join(' '));
-};
+    async #processAssistantResponse(response, metric, partialResultToken) {
+        const requestId = response.$metadata.requestId;
+        const chatEventParser = new chatEventParser_1.ChatEventParser(requestId, metric);
+        for await (const chatEvent of response.generateAssistantResponseResponse) {
+            const chatResult = chatEventParser.processPartialEvent(chatEvent);
+            // terminate early when there is an error
+            if (!chatResult.success) {
+                return chatResult;
+            }
+            if (!(0, utils_2.isNullish)(partialResultToken)) {
+                this.#features.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();
+    }
+    #log(...messages) {
+        this.#features.logging.log(messages.join(' '));
+    }
+}
+exports.ChatController = ChatController;
 //# sourceMappingURL=chatController.js.map

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;AACnD,mCAA+F;AAE/F,iFAA6E;AAC7E,iDAA4C;AAC5C,oCAA2E;AAC3E,gDAA4C;AAC5C,8DAA+E;AAC/E,2CAA0C;AAI1C,MAAa,cAAc;IACvB,SAAS,CAAU;IACnB,6BAA6B,CAA8B;IAC3D,oBAAoB,CAAyB;IAC7C,eAAe,CAAqB;IAEpC,YAAY,4BAA0D,EAAE,QAAkB;QACtF,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAA;QACzB,IAAI,CAAC,6BAA6B,GAAG,4BAA4B,CAAA;QACjE,IAAI,CAAC,eAAe,GAAG,IAAI,oCAAmB,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAA;QACpF,IAAI,CAAC,oBAAoB,GAAG,IAAI,iDAAuB,CAAC,QAAQ,CAAC,CAAA;IACrE,CAAC;IAED,OAAO;QACH,IAAI,CAAC,6BAA6B,CAAC,OAAO,EAAE,CAAA;QAC5C,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,CAAA;QAC9B,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,CAAA;IACvC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAkB,EAAE,KAAwB;QAC3D,MAAM,oBAAoB,GAAG,IAAA,8BAAsB,EAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QAEzE,IAAI,oBAAoB,EAAE;YACtB,OAAO,oBAAoB,CAAA;SAC9B;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,6BAA6B,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,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QACpE,MAAM,iBAAiB,GAAG,CAAC,OAAO,CAAC,SAAS,CAAA;QAE5C,KAAK,CAAC,uBAAuB,CAAC,GAAG,EAAE;YAC/B,IAAI,CAAC,IAAI,CAAC,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,IAAI,CAAC,IAAI,CAAC,8BAA8B,EAAE,sBAAsB,CAAC,CAAA;YACjE,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,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,IAAI,CAAC,IAAI,CAAC,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,IAAI,CAAC,oBAAoB,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,IAAI,CAAC,IAAI,CAAC,iBAAiB,IAAA,uBAAe,EAAC,GAAG,CAAC,EAAE,CAAC,CAAA;gBAElD,OAAO,IAAA,gCAAwB,EAAC,cAAc,CAAC,CAAA;aAClD;YAED,IAAI,CAAC,IAAI,CAAC,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,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,cAAc,CAAC,CAAA;YAElF,IAAI,iBAAiB,EAAE;gBACnB,IAAI,CAAC,oBAAoB,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,IAAI,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;aACrF;SACJ;QAED,IAAI;YACA,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAC/C,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,IAAI,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;YAE3E,IAAI,CAAC,oBAAoB,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,IAAI,CAAC,IAAI,CAAC,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,IAAI,CAAC,6BAA6B,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,IAAI,CAAC,SAAS,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,IAAI,CAAC,oBAAoB,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,8DAA8D;gBAC9D,SAAS,EAAE,UAAU;aACxB;SACJ,CAAC,CAAA;IACN,CAAC;IAED,iBAAiB,KAAI,CAAC;IAEtB,QAAQ,CAAC,MAAoB;QACzB,IAAI,CAAC,oBAAoB,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAA;QAEpD,IAAI,CAAC,6BAA6B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAClE,CAAC;IAED,WAAW,CAAC,MAAuB;QAC/B,IAAI,CAAC,oBAAoB,CAAC,sBAAsB,CAAC;YAC7C,IAAI,EAAE,8BAAsB,CAAC,qBAAqB;YAClD,IAAI,EAAE,EAAE;SACX,CAAC,CAAA;QAEF,IAAI,CAAC,oBAAoB,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAA;QAEpD,IAAI,CAAC,oBAAoB,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,IAAI,CAAC,oBAAoB,CAAC,WAAW,KAAK,MAAM,CAAC,KAAK,EAAE;YACxD,IAAI,CAAC,oBAAoB,CAAC,sBAAsB,CAAC;gBAC7C,IAAI,EAAE,8BAAsB,CAAC,qBAAqB;gBAClD,IAAI,EAAE,EAAE;aACX,CAAC,CAAA;YACF,IAAI,CAAC,oBAAoB,CAAC,WAAW,GAAG,SAAS,CAAA;SACpD;QAED,IAAI,CAAC,6BAA6B,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC9D,IAAI,CAAC,oBAAoB,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,IAAI,CAAC,6BAA6B,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBAEjF,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC;oBACrC,IAAI,EAAE,8BAAsB,CAAC,UAAU;oBACvC,IAAI,EAAE;wBACF,oBAAoB,EAAE,MAAM,CAAC,WAAW;qBAC3C;iBACJ,CAAC,CAAA;gBAEF,IAAI,CAAC,oBAAoB,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,IAAI,CAAC,oBAAoB,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,wBAAY;iBACrB,CAAA;YACL;gBACI,OAAO,EAAE,CAAA;SAChB;IACL,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAAkB,EAAE,MAAyC;QAClF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAExF,IAAI,cAA8B,CAAA;QAElC,wDAAwD;QACxD,4CAA4C;QAC5C,IAAI,kBAAkB,EAAE,eAAe,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC,EAAE;YACvE,IAAI,CAAC,oBAAoB,CAAC,6BAA6B,CAAC,MAAM,CAAC,KAAK,EAAE;gBAClE,kBAAkB,EAAE,kBAAkB,CAAC,SAAU;gBACjD,wBAAwB,EAAE,2BAAmB,CAAC,aAAa;aAC9D,CAAC,CAAA;YAEF,cAAc,GAAG,kBAAkB,CAAA;SACtC;aAAM;YACH,cAAc,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAA;YACxE,cAAc,CAAC,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,CAAA;SACpG;QAED,MAAM,CAAC,SAAS,CAAC;YACb,mBAAmB,EAAE,cAAc,EAAE,UAAU;YAC/C,4BAA4B,EAAE,cAAc,EAAE,mBAAmB,EAAE,YAAY;YAC/E,sBAAsB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;YAC1D,2BAA2B,EAAE,cAAc,EAAE,WAAW;YACxD,uBAAuB,EAAE,cAAc,CAAC,cAAc,IAAI,KAAK;YAC/D,gCAAgC,EAAE,cAAc,CAAC,eAAe,EAAE,MAAM,IAAI,CAAC;YAC7E,oCAAoC,EAAE,cAAc,CAAC,qBAAqB,IAAI,CAAC;SAClF,CAAC,CAAA;QAEF,OAAO,cAAc,CAAA;IACzB,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC3B,QAAgD,EAChD,MAA+B,EAC/B,kBAAoC;QAEpC,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,SAAU,CAAA;QAC/C,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;QAE9D,IAAI,KAAK,EAAE,MAAM,SAAS,IAAI,QAAQ,CAAC,iCAAkC,EAAE;YACvE,MAAM,UAAU,GAAG,eAAe,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAA;YAEjE,yCAAyC;YACzC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE;gBACrB,OAAO,UAAU,CAAA;aACpB;YAED,IAAI,CAAC,IAAA,iBAAS,EAAC,kBAAkB,CAAC,EAAE;gBAChC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,0BAAe,EAAE,kBAAkB,EAAE,UAAU,CAAC,IAAI,CAAC,CAAA;aACxF;SACJ;QAED,MAAM,CAAC,SAAS,CAAC;YACb,4BAA4B,EAAE,MAAM,CAAC,cAAc,EAAE;YACrD,sBAAsB,EAAE,eAAe,CAAC,WAAW,CAAC,mBAAmB;YACvE,wBAAwB,EAAE,eAAe,CAAC,WAAW,CAAC,kBAAkB;YACxE,wBAAwB,EAAE,eAAe,CAAC,WAAW,CAAC,0BAA0B;YAChF,uBAAuB,EAAE,eAAe,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC;SAC7D,CAAC,CAAA;QAEF,OAAO,eAAe,CAAC,aAAa,EAAE,CAAA;IAC1C,CAAC;IAED,IAAI,CAAC,GAAG,QAAkB;QACtB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;IAClD,CAAC;CACJ;AA5SD,wCA4SC"}

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

@@ -9,7 +9,10 @@ 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");
+const constants_1 = require("./constants");
 describe('ChatController', () => {
     const mockTabId = 'tab-1';
     const mockConversationId = 'mock-conversation-id';
@@ -34,7 +37,7 @@ describe('ChatController', () => {
     const expectedCompleteChatResult = {
         messageId: mockMessageId,
         body: 'Hello World!',
-        canBeVoted: undefined,
+        canBeVoted: true,
         codeReference: undefined,
         followUp: undefined,
         relatedContent: undefined,
@@ -45,6 +48,9 @@ describe('ChatController', () => {
     };
     let generateAssistantResponseStub;
     let disposeStub;
+    let activeTabSpy;
+    let removeConversationSpy;
+    let emitConversationMetricStub;
     let testFeatures;
     let chatSessionManagementService;
     let chatController;
@@ -63,13 +69,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 +93,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);
         });
@@ -115,16 +145,25 @@ describe('ChatController', () => {
             ts_sinon_1.default.assert.callCount(testFeatures.lsp.sendProgress, 0);
             assert.deepStrictEqual(chatResult, expectedCompleteChatResult);
         });
+        it('returns help message if it is a help follow up action', async () => {
+            const chatResultPromise = chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: constants_1.DEFAULT_HELP_FOLLOW_UP_PROMPT } }, mockCancellationToken);
+            const chatResult = await chatResultPromise;
+            ts_sinon_1.default.assert.match(chatResult, {
+                messageId: ts_sinon_1.default.match.string,
+                body: constants_1.HELP_MESSAGE,
+            });
+        });
         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);
+        it('can use 0 as progress token', async () => {
+            const chatResultPromise = chatController.onChatPrompt({ tabId: mockTabId, prompt: { prompt: 'Hello' }, partialResultToken: 0 }, mockCancellationToken);
             const chatResult = await chatResultPromise;
-            assert.ok(chatResult instanceof server_interface_1.ResponseError);
+            ts_sinon_1.default.assert.callCount(testFeatures.lsp.sendProgress, mockAssistantResponseList.length);
+            assert.deepStrictEqual(chatResult, expectedCompleteChatResult);
         });
         it('returns a ResponseError if generateAssistantResponse returns an error', async () => {
             generateAssistantResponseStub.callsFake(() => {
@@ -133,6 +172,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 +199,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 +218,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 +236,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 +259,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 +276,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 +289,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,
+                    },
+                });
             });
         });
     });