@aws/lsp-codewhisperer 0.0.5 → 0.0.9

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

@@ -1,4 +1,16 @@
 "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 _ChatEventParser_metric, _ChatEventParser_lastChunkTime, _ChatEventParser_totalEvents;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatEventParser = void 0;
 class ChatEventParser {
@@ -15,14 +27,44 @@ class ChatEventParser {
     static mapReferenceData(reference) {
         return {
             ...reference,
+            recommendationContentSpan: reference.recommendationContentSpan
+                ? {
+                    start: reference.recommendationContentSpan.start ?? 0,
+                    end: reference.recommendationContentSpan.end ?? 0,
+                }
+                : undefined,
             information: ChatEventParser.getReferencedInformation(reference),
         };
     }
-    constructor(messageId) {
+    constructor(messageId, metric) {
+        _ChatEventParser_metric.set(this, void 0);
+        _ChatEventParser_lastChunkTime.set(this, 0);
+        _ChatEventParser_totalEvents.set(this, {
+            followupPromptEvent: 0,
+            supplementaryWebLinksEvent: 0,
+            codeReferenceEvent: 0,
+            assistantResponseEvent: 0,
+        });
         this.messageId = messageId;
+        __classPrivateFieldSet(this, _ChatEventParser_metric, metric, "f");
+    }
+    get totalEvents() {
+        return __classPrivateFieldGet(this, _ChatEventParser_totalEvents, "f");
     }
     processPartialEvent(chatEvent) {
         const { followupPromptEvent, supplementaryWebLinksEvent, codeReferenceEvent, assistantResponseEvent, error, invalidStateEvent, } = chatEvent;
+        if (!__classPrivateFieldGet(this, _ChatEventParser_metric, "f").metric.cwsprTimeToFirstChunk) {
+            __classPrivateFieldGet(this, _ChatEventParser_metric, "f").mergeWith({
+                cwsprTimeToFirstChunk: __classPrivateFieldGet(this, _ChatEventParser_metric, "f").getTimeElapsed(),
+                cwsprChatTimeBetweenChunks: [],
+            });
+        }
+        else {
+            __classPrivateFieldGet(this, _ChatEventParser_metric, "f").mergeWith({
+                cwsprChatTimeBetweenChunks: [Date.now() - __classPrivateFieldGet(this, _ChatEventParser_lastChunkTime, "f")],
+            });
+        }
+        __classPrivateFieldSet(this, _ChatEventParser_lastChunkTime, Date.now(), "f");
         if (error) {
             this.error = error.message;
         }
@@ -30,12 +72,14 @@ class ChatEventParser {
             this.error = invalidStateEvent.message ?? invalidStateEvent.reason ?? 'Invalid state';
         }
         else if (assistantResponseEvent?.content) {
+            __classPrivateFieldGet(this, _ChatEventParser_totalEvents, "f").assistantResponseEvent += 1;
             this.body = (this.body ?? '') + assistantResponseEvent.content;
         }
         else if (followupPromptEvent?.followupPrompt) {
+            __classPrivateFieldGet(this, _ChatEventParser_totalEvents, "f").followupPromptEvent += 1;
             const { content } = followupPromptEvent.followupPrompt;
             this.followUp = {
-                ...this.followUp,
+                text: ChatEventParser.FOLLOW_UP_TEXT,
                 options: [
                     ...(this.followUp?.options ?? []),
                     {
@@ -47,6 +91,7 @@ class ChatEventParser {
         }
         else if (supplementaryWebLinksEvent?.supplementaryWebLinks &&
             supplementaryWebLinksEvent.supplementaryWebLinks.length > 0) {
+            __classPrivateFieldGet(this, _ChatEventParser_totalEvents, "f").supplementaryWebLinksEvent += 1;
             const sourceLinks = supplementaryWebLinksEvent.supplementaryWebLinks.map(ChatEventParser.mapRelatedData);
             this.relatedContent = {
                 ...this.relatedContent,
@@ -54,6 +99,7 @@ class ChatEventParser {
             };
         }
         else if (codeReferenceEvent?.references && codeReferenceEvent.references.length > 0) {
+            __classPrivateFieldGet(this, _ChatEventParser_totalEvents, "f").codeReferenceEvent += 1;
             const references = codeReferenceEvent.references.map(ChatEventParser.mapReferenceData);
             this.codeReference = [...(this.codeReference ?? []), ...references];
         }
@@ -63,7 +109,7 @@ class ChatEventParser {
         const chatResult = {
             messageId: this.messageId,
             body: this.body,
-            canBeVoted: this.canBeVoted,
+            canBeVoted: this.canBeVoted ?? true,
             relatedContent: this.relatedContent,
             followUp: this.followUp,
             codeReference: this.codeReference,
@@ -81,4 +127,6 @@ class ChatEventParser {
     }
 }
 exports.ChatEventParser = ChatEventParser;
+_ChatEventParser_metric = new WeakMap(), _ChatEventParser_lastChunkTime = new WeakMap(), _ChatEventParser_totalEvents = new WeakMap();
+ChatEventParser.FOLLOW_UP_TEXT = 'Suggested follow up questions:';
 //# sourceMappingURL=chatEventParser.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"chatEventParser.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatEventParser.ts"],"names":[],"mappings":";;;AASA,MAAa,eAAe;IASxB,MAAM,CAAC,wBAAwB,CAAC,SAAoB;QAChD,OAAO,0BAA0B,SAAS,CAAC,WAAW,gCAAgC,SAAS,CAAC,UAAU,IAAI,CAAA;IAClH,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,EAAE,OAAO,EAAwB;QACzE,OAAO;YACH,KAAK;YACL,GAAG;YACH,IAAI,EAAE,OAAO;SAChB,CAAA;IACL,CAAC;IACD,MAAM,CAAC,gBAAgB,CAAC,SAAoB;QACxC,OAAO;YACH,GAAG,SAAS;YACZ,WAAW,EAAE,eAAe,CAAC,wBAAwB,CAAC,SAAS,CAAC;SACnE,CAAA;IACL,CAAC;IAED,YAAY,SAAiB;QACzB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC9B,CAAC;IAEM,mBAAmB,CAAC,SAA6B;QACpD,MAAM,EACF,mBAAmB,EACnB,0BAA0B,EAC1B,kBAAkB,EAClB,sBAAsB,EACtB,KAAK,EACL,iBAAiB,GACpB,GAAG,SAAS,CAAA;QAEb,IAAI,KAAK,EAAE,CAAC;YACR,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAA;QAC9B,CAAC;aAAM,IAAI,iBAAiB,EAAE,CAAC;YAC3B,IAAI,CAAC,KAAK,GAAG,iBAAiB,CAAC,OAAO,IAAI,iBAAiB,CAAC,MAAM,IAAI,eAAe,CAAA;QACzF,CAAC;aAAM,IAAI,sBAAsB,EAAE,OAAO,EAAE,CAAC;YACzC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,sBAAsB,CAAC,OAAO,CAAA;QAClE,CAAC;aAAM,IAAI,mBAAmB,EAAE,cAAc,EAAE,CAAC;YAC7C,MAAM,EAAE,OAAO,EAAE,GAAG,mBAAmB,CAAC,cAAc,CAAA;YAEtD,IAAI,CAAC,QAAQ,GAAG;gBACZ,GAAG,IAAI,CAAC,QAAQ;gBAChB,OAAO,EAAE;oBACL,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;oBACjC;wBACI,QAAQ,EAAE,OAAO,IAAI,EAAE;wBACvB,MAAM,EAAE,OAAO,IAAI,EAAE;qBACxB;iBACJ;aACJ,CAAA;QACL,CAAC;aAAM,IACH,0BAA0B,EAAE,qBAAqB;YACjD,0BAA0B,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,EAC7D,CAAC;YACC,MAAM,WAAW,GAAG,0BAA0B,CAAC,qBAAqB,CAAC,GAAG,CAAC,eAAe,CAAC,cAAc,CAAC,CAAA;YAExG,IAAI,CAAC,cAAc,GAAG;gBAClB,GAAG,IAAI,CAAC,cAAc;gBACtB,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC;aACrE,CAAA;QACL,CAAC;aAAM,IAAI,kBAAkB,EAAE,UAAU,IAAI,kBAAkB,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpF,MAAM,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAA;YACtF,IAAI,CAAC,aAAa,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,GAAG,UAAU,CAAC,CAAA;QACvE,CAAC;QAED,OAAO,IAAI,CAAC,aAAa,EAAE,CAAA;IAC/B,CAAC;IAEM,aAAa;QAChB,MAAM,UAAU,GAAe;YAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,aAAa,EAAE,IAAI,CAAC,aAAa;SACpC,CAAA;QAED,OAAO,IAAI,CAAC,KAAK;YACb,CAAC,CAAC;gBACI,OAAO,EAAE,KAAK;gBACd,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,IAAI,CAAC,KAAK;aACpB;YACH,CAAC,CAAC;gBACI,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,UAAU;aACnB,CAAA;IACX,CAAC;CACJ;AAnGD,0CAmGC"}
+{"version":3,"file":"chatEventParser.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatEventParser.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAWA,MAAa,eAAe;IAoBxB,MAAM,CAAC,wBAAwB,CAAC,SAAoB;QAChD,OAAO,0BAA0B,SAAS,CAAC,WAAW,gCAAgC,SAAS,CAAC,UAAU,IAAI,CAAA;IAClH,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,EAAE,KAAK,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,EAAE,OAAO,EAAwB;QACzE,OAAO;YACH,KAAK;YACL,GAAG;YACH,IAAI,EAAE,OAAO;SAChB,CAAA;IACL,CAAC;IAED,MAAM,CAAC,gBAAgB,CAAC,SAAoB;QACxC,OAAO;YACH,GAAG,SAAS;YACZ,yBAAyB,EAAE,SAAS,CAAC,yBAAyB;gBAC1D,CAAC,CAAC;oBACI,KAAK,EAAE,SAAS,CAAC,yBAAyB,CAAC,KAAK,IAAI,CAAC;oBACrD,GAAG,EAAE,SAAS,CAAC,yBAAyB,CAAC,GAAG,IAAI,CAAC;iBACpD;gBACH,CAAC,CAAC,SAAS;YACf,WAAW,EAAE,eAAe,CAAC,wBAAwB,CAAC,SAAS,CAAC;SACnE,CAAA;IACL,CAAC;IAED,YAAY,SAAiB,EAAE,MAA+B;QAlC9D,0CAAgC;QAChC,yCAAyB,CAAC,EAAA;QAC1B,uCAAe;YACX,mBAAmB,EAAE,CAAC;YACtB,0BAA0B,EAAE,CAAC;YAC7B,kBAAkB,EAAE,CAAC;YACrB,sBAAsB,EAAE,CAAC;SAC5B,EAAA;QA4BG,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;QAC1B,uBAAA,IAAI,2BAAW,MAAM,MAAA,CAAA;IACzB,CAAC;IAED,IAAW,WAAW;QAClB,OAAO,uBAAA,IAAI,oCAAa,CAAA;IAC5B,CAAC;IAEM,mBAAmB,CAAC,SAA6B;QACpD,MAAM,EACF,mBAAmB,EACnB,0BAA0B,EAC1B,kBAAkB,EAClB,sBAAsB,EACtB,KAAK,EACL,iBAAiB,GACpB,GAAG,SAAS,CAAA;QAEb,IAAI,CAAC,uBAAA,IAAI,+BAAQ,CAAC,MAAM,CAAC,qBAAqB,EAAE;YAC5C,uBAAA,IAAI,+BAAQ,CAAC,SAAS,CAAC;gBACnB,qBAAqB,EAAE,uBAAA,IAAI,+BAAQ,CAAC,cAAc,EAAE;gBACpD,0BAA0B,EAAE,EAAE;aACjC,CAAC,CAAA;SACL;aAAM;YACH,uBAAA,IAAI,+BAAQ,CAAC,SAAS,CAAC;gBACnB,0BAA0B,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,uBAAA,IAAI,sCAAe,CAAC;aACjE,CAAC,CAAA;SACL;QAED,uBAAA,IAAI,kCAAkB,IAAI,CAAC,GAAG,EAAE,MAAA,CAAA;QAEhC,IAAI,KAAK,EAAE;YACP,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAA;SAC7B;aAAM,IAAI,iBAAiB,EAAE;YAC1B,IAAI,CAAC,KAAK,GAAG,iBAAiB,CAAC,OAAO,IAAI,iBAAiB,CAAC,MAAM,IAAI,eAAe,CAAA;SACxF;aAAM,IAAI,sBAAsB,EAAE,OAAO,EAAE;YACxC,uBAAA,IAAI,oCAAa,CAAC,sBAAsB,IAAI,CAAC,CAAA;YAC7C,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,sBAAsB,CAAC,OAAO,CAAA;SACjE;aAAM,IAAI,mBAAmB,EAAE,cAAc,EAAE;YAC5C,uBAAA,IAAI,oCAAa,CAAC,mBAAmB,IAAI,CAAC,CAAA;YAC1C,MAAM,EAAE,OAAO,EAAE,GAAG,mBAAmB,CAAC,cAAc,CAAA;YAEtD,IAAI,CAAC,QAAQ,GAAG;gBACZ,IAAI,EAAE,eAAe,CAAC,cAAc;gBACpC,OAAO,EAAE;oBACL,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,IAAI,EAAE,CAAC;oBACjC;wBACI,QAAQ,EAAE,OAAO,IAAI,EAAE;wBACvB,MAAM,EAAE,OAAO,IAAI,EAAE;qBACxB;iBACJ;aACJ,CAAA;SACJ;aAAM,IACH,0BAA0B,EAAE,qBAAqB;YACjD,0BAA0B,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,EAC7D;YACE,uBAAA,IAAI,oCAAa,CAAC,0BAA0B,IAAI,CAAC,CAAA;YACjD,MAAM,WAAW,GAAG,0BAA0B,CAAC,qBAAqB,CAAC,GAAG,CAAC,eAAe,CAAC,cAAc,CAAC,CAAA;YAExG,IAAI,CAAC,cAAc,GAAG;gBAClB,GAAG,IAAI,CAAC,cAAc;gBACtB,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,GAAG,WAAW,CAAC;aACrE,CAAA;SACJ;aAAM,IAAI,kBAAkB,EAAE,UAAU,IAAI,kBAAkB,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;YACnF,uBAAA,IAAI,oCAAa,CAAC,kBAAkB,IAAI,CAAC,CAAA;YACzC,MAAM,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAA;YACtF,IAAI,CAAC,aAAa,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,EAAE,GAAG,UAAU,CAAC,CAAA;SACtE;QAED,OAAO,IAAI,CAAC,aAAa,EAAE,CAAA;IAC/B,CAAC;IAEM,aAAa;QAChB,MAAM,UAAU,GAAe;YAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;YACnC,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,aAAa,EAAE,IAAI,CAAC,aAAa;SACpC,CAAA;QAED,OAAO,IAAI,CAAC,KAAK;YACb,CAAC,CAAC;gBACI,OAAO,EAAE,KAAK;gBACd,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,IAAI,CAAC,KAAK;aACpB;YACH,CAAC,CAAC;gBACI,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,UAAU;aACnB,CAAA;IACX,CAAC;;AA1IL,0CA2IC;;AA1ImB,8BAAc,GAAG,gCAAgC,CAAA"}

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

@@ -3,10 +3,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const assert = require("assert");
 const ts_sinon_1 = require("ts-sinon");
 const chatEventParser_1 = require("./chatEventParser");
+const metric_1 = require("../telemetry/metric");
 describe('ChatEventParser', () => {
     const mockMessageId = 'mock-message-id';
     it('set error if invalidState event is received', () => {
-        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId, new metric_1.Metric());
         ts_sinon_1.default.assert.match(chatEventParser.processPartialEvent({
             invalidStateEvent: {
                 message: 'Invalid state!',
@@ -17,7 +18,7 @@ describe('ChatEventParser', () => {
             data: {
                 messageId: mockMessageId,
                 body: undefined,
-                canBeVoted: undefined,
+                canBeVoted: true,
                 codeReference: undefined,
                 followUp: undefined,
                 relatedContent: undefined,
@@ -26,7 +27,7 @@ describe('ChatEventParser', () => {
         });
     });
     it('set error if error event is received', () => {
-        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId, new metric_1.Metric());
         ts_sinon_1.default.assert.match(chatEventParser.processPartialEvent({
             error: {
                 name: 'InternalServerException',
@@ -40,7 +41,7 @@ describe('ChatEventParser', () => {
             data: {
                 messageId: mockMessageId,
                 body: undefined,
-                canBeVoted: undefined,
+                canBeVoted: true,
                 codeReference: undefined,
                 followUp: undefined,
                 relatedContent: undefined,
@@ -50,7 +51,7 @@ describe('ChatEventParser', () => {
         assert.strictEqual(chatEventParser.error, 'Error!');
     });
     it('processPartialEvent appends new event on top of the previous result', () => {
-        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId, new metric_1.Metric());
         assert.deepStrictEqual(chatEventParser.processPartialEvent({
             assistantResponseEvent: {
                 content: 'This is an ',
@@ -60,7 +61,7 @@ describe('ChatEventParser', () => {
             data: {
                 messageId: mockMessageId,
                 body: 'This is an ',
-                canBeVoted: undefined,
+                canBeVoted: true,
                 codeReference: undefined,
                 followUp: undefined,
                 relatedContent: undefined,
@@ -75,7 +76,7 @@ describe('ChatEventParser', () => {
             data: {
                 messageId: mockMessageId,
                 body: 'This is an assistant response.',
-                canBeVoted: undefined,
+                canBeVoted: true,
                 codeReference: undefined,
                 followUp: undefined,
                 relatedContent: undefined,
@@ -83,7 +84,7 @@ describe('ChatEventParser', () => {
         });
     });
     it('getChatResult returns the accumulated result', () => {
-        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId, new metric_1.Metric());
         chatEventParser.processPartialEvent({
             assistantResponseEvent: {
                 content: 'This is an ',
@@ -102,6 +103,7 @@ describe('ChatEventParser', () => {
                 messageId: mockMessageId,
                 body: 'This is an ',
                 followUp: {
+                    text: chatEventParser_1.ChatEventParser.FOLLOW_UP_TEXT,
                     options: [
                         {
                             pillText: 'follow up 1',
@@ -109,7 +111,7 @@ describe('ChatEventParser', () => {
                         },
                     ],
                 },
-                canBeVoted: undefined,
+                canBeVoted: true,
                 codeReference: undefined,
                 relatedContent: undefined,
             },
@@ -150,7 +152,7 @@ describe('ChatEventParser', () => {
         const expectedData = {
             messageId: mockMessageId,
             body: 'This is an assistant response.',
-            canBeVoted: undefined,
+            canBeVoted: true,
             relatedContent: {
                 content: [
                     {
@@ -164,6 +166,7 @@ describe('ChatEventParser', () => {
                 {
                     licenseName: 'MIT',
                     repository: 'langauge-servers',
+                    recommendationContentSpan: undefined,
                     information: chatEventParser_1.ChatEventParser.getReferencedInformation({
                         licenseName: 'MIT',
                         repository: 'langauge-servers',
@@ -171,6 +174,7 @@ describe('ChatEventParser', () => {
                 },
             ],
             followUp: {
+                text: chatEventParser_1.ChatEventParser.FOLLOW_UP_TEXT,
                 options: [
                     {
                         pillText: 'follow up 1',

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

@@ -1 +1 @@
-{"version":3,"file":"chatEventParser.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatEventParser.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,uCAA4B;AAE5B,uDAAmD;AAEnD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,MAAM,aAAa,GAAG,iBAAiB,CAAA;IAEvC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACnD,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,CAAC,CAAA;QAE1D,kBAAK,CAAC,MAAM,CAAC,KAAK,CACd,eAAe,CAAC,mBAAmB,CAAC;YAChC,iBAAiB,EAAE;gBACf,OAAO,EAAE,gBAAgB;gBACzB,MAAM,EAAE,UAAU;aACrB;SACJ,CAAC,EACF;YACI,OAAO,EAAE,KAAK;YACd,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;YACD,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;SAC5B,CACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC5C,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,CAAC,CAAA;QAE1D,kBAAK,CAAC,MAAM,CAAC,KAAK,CACd,eAAe,CAAC,mBAAmB,CAAC;YAChC,KAAK,EAAE;gBACH,IAAI,EAAE,yBAAyB;gBAC/B,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,QAAQ;gBAChB,UAAU,EAAE,KAAK;gBACjB,SAAS,EAAE,EAAE;aAChB;SACJ,CAAC,EACF;YACI,OAAO,EAAE,KAAK;YACd,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;YACD,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;SAC5B,CACJ,CAAA;QAED,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC3E,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,CAAC,CAAA;QAE1D,MAAM,CAAC,eAAe,CAClB,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,aAAa;aACzB;SACJ,CAAC,EACF;YACI,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;SACJ,CACJ,CAAA;QAED,MAAM,CAAC,eAAe,CAClB,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,qBAAqB;aACjC;SACJ,CAAC,EACF;YACI,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,gCAAgC;gBACtC,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;SACJ,CACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACpD,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,CAAC,CAAA;QAE1D,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,aAAa;aACzB;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,mBAAmB,EAAE;gBACjB,cAAc,EAAE;oBACZ,OAAO,EAAE,aAAa;iBACzB;aACJ;SACJ,CAAC,CAAA;QAEF,MAAM,CAAC,eAAe,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE;YACpD,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,aAAa;gBACnB,QAAQ,EAAE;oBACN,OAAO,EAAE;wBACL;4BACI,QAAQ,EAAE,aAAa;4BACvB,MAAM,EAAE,aAAa;yBACxB;qBACJ;iBACJ;gBACD,UAAU,EAAE,SAAS;gBACrB,aAAa,EAAE,SAAS;gBACxB,cAAc,EAAE,SAAS;aAC5B;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,mBAAmB,EAAE;gBACjB,cAAc,EAAE;oBACZ,OAAO,EAAE,aAAa;iBACzB;aACJ;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,kBAAkB,EAAE;gBAChB,UAAU,EAAE;oBACR;wBACI,WAAW,EAAE,KAAK;wBAClB,UAAU,EAAE,kBAAkB;qBACjC;iBACJ;aACJ;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,qBAAqB;aACjC;SACJ,CAAC,CAAA;QACF,eAAe,CAAC,mBAAmB,CAAC;YAChC,0BAA0B,EAAE;gBACxB,qBAAqB,EAAE;oBACnB;wBACI,KAAK,EAAE,UAAU;wBACjB,OAAO,EAAE,UAAU;wBACnB,GAAG,EAAE,mBAAmB;qBAC3B;iBACJ;aACJ;SACJ,CAAC,CAAA;QAEF,MAAM,YAAY,GAAe;YAC7B,SAAS,EAAE,aAAa;YACxB,IAAI,EAAE,gCAAgC;YACtC,UAAU,EAAE,SAAS;YACrB,cAAc,EAAE;gBACZ,OAAO,EAAE;oBACL;wBACI,KAAK,EAAE,UAAU;wBACjB,IAAI,EAAE,UAAU;wBAChB,GAAG,EAAE,mBAAmB;qBAC3B;iBACJ;aACJ;YACD,aAAa,EAAE;gBACX;oBACI,WAAW,EAAE,KAAK;oBAClB,UAAU,EAAE,kBAAkB;oBAC9B,WAAW,EAAE,iCAAe,CAAC,wBAAwB,CAAC;wBAClD,WAAW,EAAE,KAAK;wBAClB,UAAU,EAAE,kBAAkB;qBACjC,CAAC;iBACL;aACJ;YACD,QAAQ,EAAE;gBACN,OAAO,EAAE;oBACL;wBACI,QAAQ,EAAE,aAAa;wBACvB,MAAM,EAAE,aAAa;qBACxB;oBACD;wBACI,QAAQ,EAAE,aAAa;wBACvB,MAAM,EAAE,aAAa;qBACxB;iBACJ;aACJ;SACJ,CAAA;QAED,MAAM,CAAC,eAAe,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAA;IAClG,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}
+{"version":3,"file":"chatEventParser.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatEventParser.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,uCAA4B;AAE5B,uDAAmD;AACnD,gDAA4C;AAG5C,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7B,MAAM,aAAa,GAAG,iBAAiB,CAAA;IAEvC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACnD,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,EAAE,IAAI,eAAM,EAAmB,CAAC,CAAA;QAEzF,kBAAK,CAAC,MAAM,CAAC,KAAK,CACd,eAAe,CAAC,mBAAmB,CAAC;YAChC,iBAAiB,EAAE;gBACf,OAAO,EAAE,gBAAgB;gBACzB,MAAM,EAAE,UAAU;aACrB;SACJ,CAAC,EACF;YACI,OAAO,EAAE,KAAK;YACd,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,IAAI;gBAChB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;YACD,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;SAC5B,CACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC5C,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,EAAE,IAAI,eAAM,EAAmB,CAAC,CAAA;QAEzF,kBAAK,CAAC,MAAM,CAAC,KAAK,CACd,eAAe,CAAC,mBAAmB,CAAC;YAChC,KAAK,EAAE;gBACH,IAAI,EAAE,yBAAyB;gBAC/B,OAAO,EAAE,QAAQ;gBACjB,MAAM,EAAE,QAAQ;gBAChB,UAAU,EAAE,KAAK;gBACjB,SAAS,EAAE,EAAE;aAChB;SACJ,CAAC,EACF;YACI,OAAO,EAAE,KAAK;YACd,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,IAAI;gBAChB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;YACD,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;SAC5B,CACJ,CAAA;QAED,MAAM,CAAC,WAAW,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC3E,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,EAAE,IAAI,eAAM,EAAmB,CAAC,CAAA;QAEzF,MAAM,CAAC,eAAe,CAClB,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,aAAa;aACzB;SACJ,CAAC,EACF;YACI,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,IAAI;gBAChB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;SACJ,CACJ,CAAA;QAED,MAAM,CAAC,eAAe,CAClB,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,qBAAqB;aACjC;SACJ,CAAC,EACF;YACI,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,gCAAgC;gBACtC,UAAU,EAAE,IAAI;gBAChB,aAAa,EAAE,SAAS;gBACxB,QAAQ,EAAE,SAAS;gBACnB,cAAc,EAAE,SAAS;aAC5B;SACJ,CACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACpD,MAAM,eAAe,GAAG,IAAI,iCAAe,CAAC,aAAa,EAAE,IAAI,eAAM,EAAmB,CAAC,CAAA;QAEzF,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,aAAa;aACzB;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,mBAAmB,EAAE;gBACjB,cAAc,EAAE;oBACZ,OAAO,EAAE,aAAa;iBACzB;aACJ;SACJ,CAAC,CAAA;QAEF,MAAM,CAAC,eAAe,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE;YACpD,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACF,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,aAAa;gBACnB,QAAQ,EAAE;oBACN,IAAI,EAAE,iCAAe,CAAC,cAAc;oBACpC,OAAO,EAAE;wBACL;4BACI,QAAQ,EAAE,aAAa;4BACvB,MAAM,EAAE,aAAa;yBACxB;qBACJ;iBACJ;gBACD,UAAU,EAAE,IAAI;gBAChB,aAAa,EAAE,SAAS;gBACxB,cAAc,EAAE,SAAS;aAC5B;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,mBAAmB,EAAE;gBACjB,cAAc,EAAE;oBACZ,OAAO,EAAE,aAAa;iBACzB;aACJ;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,kBAAkB,EAAE;gBAChB,UAAU,EAAE;oBACR;wBACI,WAAW,EAAE,KAAK;wBAClB,UAAU,EAAE,kBAAkB;qBACjC;iBACJ;aACJ;SACJ,CAAC,CAAA;QAEF,eAAe,CAAC,mBAAmB,CAAC;YAChC,sBAAsB,EAAE;gBACpB,OAAO,EAAE,qBAAqB;aACjC;SACJ,CAAC,CAAA;QACF,eAAe,CAAC,mBAAmB,CAAC;YAChC,0BAA0B,EAAE;gBACxB,qBAAqB,EAAE;oBACnB;wBACI,KAAK,EAAE,UAAU;wBACjB,OAAO,EAAE,UAAU;wBACnB,GAAG,EAAE,mBAAmB;qBAC3B;iBACJ;aACJ;SACJ,CAAC,CAAA;QAEF,MAAM,YAAY,GAAe;YAC7B,SAAS,EAAE,aAAa;YACxB,IAAI,EAAE,gCAAgC;YACtC,UAAU,EAAE,IAAI;YAChB,cAAc,EAAE;gBACZ,OAAO,EAAE;oBACL;wBACI,KAAK,EAAE,UAAU;wBACjB,IAAI,EAAE,UAAU;wBAChB,GAAG,EAAE,mBAAmB;qBAC3B;iBACJ;aACJ;YACD,aAAa,EAAE;gBACX;oBACI,WAAW,EAAE,KAAK;oBAClB,UAAU,EAAE,kBAAkB;oBAC9B,yBAAyB,EAAE,SAAS;oBACpC,WAAW,EAAE,iCAAe,CAAC,wBAAwB,CAAC;wBAClD,WAAW,EAAE,KAAK;wBAClB,UAAU,EAAE,kBAAkB;qBACjC,CAAC;iBACL;aACJ;YACD,QAAQ,EAAE;gBACN,IAAI,EAAE,iCAAe,CAAC,cAAc;gBACpC,OAAO,EAAE;oBACL;wBACI,QAAQ,EAAE,aAAa;wBACvB,MAAM,EAAE,aAAa;qBACxB;oBACD;wBACI,QAAQ,EAAE,aAAa;wBACvB,MAAM,EAAE,aAAa;qBACxB;iBACJ;aACJ;SACJ,CAAA;QAED,MAAM,CAAC,eAAe,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAA;IAClG,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

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

@@ -16,13 +16,13 @@ exports.ChatSessionManagementService = void 0;
 const chatSessionService_1 = require("./chatSessionService");
 class ChatSessionManagementService {
     static getInstance() {
-        if (!__classPrivateFieldGet(_a, _a, "f", _ChatSessionManagementService_instance)) {
-            __classPrivateFieldSet(_a, _a, new _a(), "f", _ChatSessionManagementService_instance);
+        if (!__classPrivateFieldGet(ChatSessionManagementService, _a, "f", _ChatSessionManagementService_instance)) {
+            __classPrivateFieldSet(ChatSessionManagementService, _a, new ChatSessionManagementService(), "f", _ChatSessionManagementService_instance);
         }
-        return __classPrivateFieldGet(_a, _a, "f", _ChatSessionManagementService_instance);
+        return __classPrivateFieldGet(ChatSessionManagementService, _a, "f", _ChatSessionManagementService_instance);
     }
     static reset() {
-        __classPrivateFieldSet(_a, _a, undefined, "f", _ChatSessionManagementService_instance);
+        __classPrivateFieldSet(ChatSessionManagementService, _a, undefined, "f", _ChatSessionManagementService_instance);
     }
     constructor() {
         _ChatSessionManagementService_sessionByTab.set(this, new Map());
@@ -49,9 +49,8 @@ class ChatSessionManagementService {
         }
         else if (__classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").has(tabId)) {
             return {
-                success: false,
+                success: true,
                 data: __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").get(tabId),
-                error: 'Session already exists',
             };
         }
         const clientConfig = typeof __classPrivateFieldGet(this, _ChatSessionManagementService_clientConfig, "f") === 'function' ? __classPrivateFieldGet(this, _ChatSessionManagementService_clientConfig, "f").call(this) : __classPrivateFieldGet(this, _ChatSessionManagementService_clientConfig, "f");
@@ -64,15 +63,7 @@ class ChatSessionManagementService {
     }
     getSession(tabId) {
         const session = __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").get(tabId);
-        return session
-            ? {
-                success: true,
-                data: session,
-            }
-            : {
-                success: false,
-                error: 'Session does not exist',
-            };
+        return session ? { success: true, data: session } : this.createSession(tabId);
     }
     deleteSession(tabId) {
         __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").get(tabId)?.dispose();

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

@@ -1 +1 @@
-{"version":3,"file":"chatSessionManagementService.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionManagementService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAEA,6DAAmF;AAEnF,MAAa,4BAA4B;IAM9B,MAAM,CAAC,WAAW;QACrB,IAAI,CAAC,uBAAA,EAA4B,kDAAU,EAAE,CAAC;YAC1C,uBAAA,EAA4B,MAAa,IAAI,EAA4B,EAAE,8CAAA,CAAA;QAC/E,CAAC;QAED,OAAO,uBAAA,EAA4B,kDAAU,CAAA;IACjD,CAAC;IAEM,MAAM,CAAC,KAAK;QACf,uBAAA,EAA4B,MAAa,SAAS,8CAAA,CAAA;IACtD,CAAC;IAED;QAhBA,qDAAiD,IAAI,GAAG,EAAe,EAAA;QACvE,oEAA0C;QAC1C,6DAA2E;IAcpD,CAAC;IAEjB,UAAU,CAAC,YAA0E;QACxF,uBAAA,IAAI,8CAAiB,YAAY,MAAA,CAAA;QAEjC,OAAO,IAAI,CAAA;IACf,CAAC;IAEM,uBAAuB,CAAC,mBAAwC;QACnE,uBAAA,IAAI,qDAAwB,mBAAmB,MAAA,CAAA;QAE/C,OAAO,IAAI,CAAA;IACf,CAAC;IAEM,UAAU,CAAC,KAAa;QAC3B,OAAO,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;IACxC,CAAC;IAEM,aAAa,CAAC,KAAa;QAC9B,IAAI,CAAC,uBAAA,IAAI,yDAAqB,EAAE,CAAC;YAC7B,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,iCAAiC;aAC3C,CAAA;QACL,CAAC;aAAM,IAAI,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,IAAI,EAAE,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC;gBACnC,KAAK,EAAE,wBAAwB;aAClC,CAAA;QACL,CAAC;QAED,MAAM,YAAY,GAAG,OAAO,uBAAA,IAAI,kDAAc,KAAK,UAAU,CAAC,CAAC,CAAC,uBAAA,IAAI,kDAAc,MAAlB,IAAI,CAAgB,CAAC,CAAC,CAAC,uBAAA,IAAI,kDAAc,CAAA;QACzG,MAAM,UAAU,GAAG,IAAI,uCAAkB,CAAC,uBAAA,IAAI,yDAAqB,EAAE,YAAY,CAAC,CAAA;QAElF,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAA;QAEzC,OAAO;YACH,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,UAAU;SACnB,CAAA;IACL,CAAC;IAEM,UAAU,CAAC,KAAa;QAC3B,MAAM,OAAO,GAAG,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QAC7C,OAAO,OAAO;YACV,CAAC,CAAC;gBACI,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,OAAO;aAChB;YACH,CAAC,CAAC;gBACI,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,wBAAwB;aAClC,CAAA;IACX,CAAC;IAEM,aAAa,CAAC,KAAa;QAC9B,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,CAAA;QACxC,uBAAA,IAAI,kDAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEhC,OAAO;YACH,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,SAAS;SAClB,CAAA;IACL,CAAC;IAEM,OAAO;QACV,uBAAA,IAAI,kDAAc,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAA;IAC5D,CAAC;CACJ;AAvFD,oEAuFC;;AAtFU,0DAAS,CAA+B"}
+{"version":3,"file":"chatSessionManagementService.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionManagementService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAEA,6DAAmF;AAEnF,MAAa,4BAA4B;IAM9B,MAAM,CAAC,WAAW;QACrB,IAAI,CAAC,uBAAA,4BAA4B,kDAAU,EAAE;YACzC,uBAAA,4BAA4B,MAAa,IAAI,4BAA4B,EAAE,8CAAA,CAAA;SAC9E;QAED,OAAO,uBAAA,4BAA4B,kDAAU,CAAA;IACjD,CAAC;IAEM,MAAM,CAAC,KAAK;QACf,uBAAA,4BAA4B,MAAa,SAAS,8CAAA,CAAA;IACtD,CAAC;IAED;QAhBA,qDAAiD,IAAI,GAAG,EAAe,EAAA;QACvE,oEAA0C;QAC1C,6DAA2E;IAcpD,CAAC;IAEjB,UAAU,CAAC,YAA0E;QACxF,uBAAA,IAAI,8CAAiB,YAAY,MAAA,CAAA;QAEjC,OAAO,IAAI,CAAA;IACf,CAAC;IAEM,uBAAuB,CAAC,mBAAwC;QACnE,uBAAA,IAAI,qDAAwB,mBAAmB,MAAA,CAAA;QAE/C,OAAO,IAAI,CAAA;IACf,CAAC;IAEM,UAAU,CAAC,KAAa;QAC3B,OAAO,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;IACxC,CAAC;IAEM,aAAa,CAAC,KAAa;QAC9B,IAAI,CAAC,uBAAA,IAAI,yDAAqB,EAAE;YAC5B,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,iCAAiC;aAC3C,CAAA;SACJ;aAAM,IAAI,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YACtC,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAE;aACvC,CAAA;SACJ;QAED,MAAM,YAAY,GAAG,OAAO,uBAAA,IAAI,kDAAc,KAAK,UAAU,CAAC,CAAC,CAAC,uBAAA,IAAI,kDAAc,MAAlB,IAAI,CAAgB,CAAC,CAAC,CAAC,uBAAA,IAAI,kDAAc,CAAA;QACzG,MAAM,UAAU,GAAG,IAAI,uCAAkB,CAAC,uBAAA,IAAI,yDAAqB,EAAE,YAAY,CAAC,CAAA;QAElF,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAA;QAEzC,OAAO;YACH,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,UAAU;SACnB,CAAA;IACL,CAAC;IAEM,UAAU,CAAC,KAAa;QAC3B,MAAM,OAAO,GAAG,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QAE7C,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAA;IACjF,CAAC;IAEM,aAAa,CAAC,KAAa;QAC9B,uBAAA,IAAI,kDAAc,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,CAAA;QACxC,uBAAA,IAAI,kDAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEhC,OAAO;YACH,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,SAAS;SAClB,CAAA;IACL,CAAC;IAEM,OAAO;QACV,uBAAA,IAAI,kDAAc,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAA;IAC5D,CAAC;CACJ;AA/ED,oEA+EC;;AA9EU,2DAAwC"}

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

@@ -33,27 +33,19 @@ describe('ChatSessionManagementService', () => {
             chatSessionManagementService_1.ChatSessionManagementService.reset();
             disposeStub.restore();
         });
-        it('getSession should return an existing client if found', () => {
+        it('getSession should create a client if not found and returns existing client if found', () => {
             assert.ok(!chatSessionManagementService.hasSession(mockSessionId));
-            ts_sinon_1.default.assert.match(chatSessionManagementService.getSession(mockSessionId), {
-                success: false,
-                error: ts_sinon_1.default.match.string,
-            });
-            const chatClientData = chatSessionManagementService.createSession(mockSessionId);
-            ts_sinon_1.default.assert.match(chatClientData, {
+            const result = chatSessionManagementService.getSession(mockSessionId);
+            ts_sinon_1.default.assert.match(result, {
                 success: true,
                 data: ts_sinon_1.default.match.instanceOf(chatSessionService_1.ChatSessionService),
             });
             assert.ok(chatSessionManagementService.hasSession(mockSessionId));
-            // asserting object reference
-            assert.strictEqual(chatSessionManagementService.getSession(mockSessionId).data, chatClientData.data);
+            assert.strictEqual(chatSessionManagementService.getSession(mockSessionId).data, result.data);
         });
-        it('creating a session with an existing id should return an error', () => {
-            chatSessionManagementService.createSession(mockSessionId);
-            ts_sinon_1.default.assert.match(chatSessionManagementService.createSession(mockSessionId), {
-                success: false,
-                error: ts_sinon_1.default.match.string,
-            });
+        it('creating a session with an existing id should return existing session', () => {
+            const createdSession = chatSessionManagementService.createSession(mockSessionId);
+            assert.deepStrictEqual(chatSessionManagementService.createSession(mockSessionId), createdSession);
         });
         it('deleting session should dispose the chat session service and delete from map', () => {
             chatSessionManagementService.createSession(mockSessionId);

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

@@ -1 +1 @@
-{"version":3,"file":"chatSessionManagementService.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionManagementService.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,uCAA4B;AAC5B,iFAA6E;AAC7E,6DAAyD;AAEzD,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC1C,MAAM,aAAa,GAAG,eAAe,CAAA;IAErC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QAClE,MAAM,QAAQ,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAA;QAC3D,MAAM,SAAS,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAA;QAE5D,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC7E,MAAM,mBAAmB,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;QAEnG,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAA;IAC1F,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC/B,MAAM,uBAAuB,GAAwB;YACjD,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;YAC1C,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;YAC7E,qBAAqB,EAAE,kBAAK,CAAC,IAAI,EAAE;SACtC,CAAA;QAED,MAAM,cAAc,GAAG,gBAAgB,CAAA;QACvC,IAAI,WAA4B,CAAA;QAChC,IAAI,4BAA0D,CAAA;QAE9D,UAAU,CAAC,GAAG,EAAE;YACZ,WAAW,GAAG,kBAAK,CAAC,IAAI,CAAC,uCAAkB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;YACjE,4BAA4B;gBACxB,2DAA4B,CAAC,WAAW,EAAE,CAAC,uBAAuB,CAAC,uBAAuB,CAAC,CAAA;QACnG,CAAC,CAAC,CAAA;QAEF,SAAS,CAAC,GAAG,EAAE;YACX,2DAA4B,CAAC,KAAK,EAAE,CAAA;YACpC,WAAW,CAAC,OAAO,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,EAAE,CAAC,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAElE,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE;gBACvE,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;aAC5B,CAAC,CAAA;YAEF,MAAM,cAAc,GAAG,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEhF,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;gBAC/B,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,kBAAK,CAAC,KAAK,CAAC,UAAU,CAAC,uCAAkB,CAAC;aACnD,CAAC,CAAA;YAEF,MAAM,CAAC,EAAE,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAEjE,6BAA6B;YAC7B,MAAM,CAAC,WAAW,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,CAAC,CAAA;QACxG,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACrE,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEzD,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE;gBAC1E,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM;aAC5B,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACpF,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEzD,MAAM,CAAC,EAAE,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAEjE,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEzD,MAAM,CAAC,EAAE,CAAC,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAElE,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oFAAoF,EAAE,GAAG,EAAE;YAC1F,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YACzD,4BAA4B,CAAC,aAAa,CAAC,cAAc,CAAC,CAAA;YAE1D,4BAA4B,CAAC,OAAO,EAAE,CAAA;YAEtC,kBAAK,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}
+{"version":3,"file":"chatSessionManagementService.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionManagementService.test.ts"],"names":[],"mappings":";;AACA,iCAAgC;AAChC,uCAA4B;AAC5B,iFAA6E;AAC7E,6DAAyD;AAEzD,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC1C,MAAM,aAAa,GAAG,eAAe,CAAA;IAErC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QAClE,MAAM,QAAQ,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAA;QAC3D,MAAM,SAAS,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAA;QAE5D,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC7E,MAAM,mBAAmB,GAAG,2DAA4B,CAAC,WAAW,EAAE,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;QAEnG,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAA;IAC1F,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC/B,MAAM,uBAAuB,GAAwB;YACjD,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;YAC1C,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;YAC7E,qBAAqB,EAAE,kBAAK,CAAC,IAAI,EAAE;SACtC,CAAA;QAED,MAAM,cAAc,GAAG,gBAAgB,CAAA;QACvC,IAAI,WAA4B,CAAA;QAChC,IAAI,4BAA0D,CAAA;QAE9D,UAAU,CAAC,GAAG,EAAE;YACZ,WAAW,GAAG,kBAAK,CAAC,IAAI,CAAC,uCAAkB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;YACjE,4BAA4B;gBACxB,2DAA4B,CAAC,WAAW,EAAE,CAAC,uBAAuB,CAAC,uBAAuB,CAAC,CAAA;QACnG,CAAC,CAAC,CAAA;QAEF,SAAS,CAAC,GAAG,EAAE;YACX,2DAA4B,CAAC,KAAK,EAAE,CAAA;YACpC,WAAW,CAAC,OAAO,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qFAAqF,EAAE,GAAG,EAAE;YAC3F,MAAM,CAAC,EAAE,CAAC,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAElE,MAAM,MAAM,GAAG,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAA;YAErE,kBAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE;gBACvB,OAAO,EAAE,IAAI;gBACb,IAAI,EAAE,kBAAK,CAAC,KAAK,CAAC,UAAU,CAAC,uCAAkB,CAAC;aACnD,CAAC,CAAA;YAEF,MAAM,CAAC,EAAE,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,WAAW,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAA;QAChG,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC7E,MAAM,cAAc,GAAG,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEhF,MAAM,CAAC,eAAe,CAAC,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,cAAc,CAAC,CAAA;QACrG,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACpF,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEzD,MAAM,CAAC,EAAE,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAEjE,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YAEzD,MAAM,CAAC,EAAE,CAAC,CAAC,4BAA4B,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAA;YAElE,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oFAAoF,EAAE,GAAG,EAAE;YAC1F,4BAA4B,CAAC,aAAa,CAAC,aAAa,CAAC,CAAA;YACzD,4BAA4B,CAAC,aAAa,CAAC,cAAc,CAAC,CAAA;YAE1D,4BAA4B,CAAC,OAAO,EAAE,CAAA;YAEtC,kBAAK,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

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

@@ -8,6 +8,7 @@ export declare class ChatSessionService {
     set sessionId(value: string | undefined);
     constructor(credentialsProvider: CredentialsProvider, config?: CodeWhispererStreamingClientConfig);
     generateAssistantResponse(request: GenerateAssistantResponseCommandInput): Promise<GenerateAssistantResponseCommandOutput>;
+    clear(): void;
     dispose(): void;
     abortRequest(): void;
 }

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

@@ -10,12 +10,11 @@ var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (
     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 _ChatSessionService_codeWhispererRegion, _ChatSessionService_codeWhispererEndpoint, _ChatSessionService_client, _ChatSessionService_abortController, _ChatSessionService_sessionId;
+var _ChatSessionService_codeWhispererRegion, _ChatSessionService_codeWhispererEndpoint, _ChatSessionService_abortController, _ChatSessionService_credentialsProvider, _ChatSessionService_config, _ChatSessionService_sessionId;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatSessionService = void 0;
 const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
 const util_retry_1 = require("@aws-sdk/util-retry");
-const abort_controller_1 = require("@smithy/abort-controller");
 const utils_1 = require("../utils");
 class ChatSessionService {
     get sessionId() {
@@ -28,36 +27,43 @@ class ChatSessionService {
         this.shareCodeWhispererContentWithAWS = false;
         _ChatSessionService_codeWhispererRegion.set(this, 'us-east-1');
         _ChatSessionService_codeWhispererEndpoint.set(this, 'https://codewhisperer.us-east-1.amazonaws.com/');
-        _ChatSessionService_client.set(this, void 0);
         _ChatSessionService_abortController.set(this, void 0);
+        _ChatSessionService_credentialsProvider.set(this, void 0);
+        _ChatSessionService_config.set(this, void 0);
         _ChatSessionService_sessionId.set(this, void 0);
-        __classPrivateFieldSet(this, _ChatSessionService_client, new codewhisperer_streaming_1.CodeWhispererStreaming({
-            region: __classPrivateFieldGet(this, _ChatSessionService_codeWhispererRegion, "f"),
-            endpoint: __classPrivateFieldGet(this, _ChatSessionService_codeWhispererEndpoint, "f"),
-            token: () => Promise.resolve({ token: (0, utils_1.getBearerTokenFromProvider)(credentialsProvider) }),
-            retryStrategy: new util_retry_1.ConfiguredRetryStrategy(0, (attempt) => 500 + attempt ** 10),
-            ...config,
-        }), "f");
+        __classPrivateFieldSet(this, _ChatSessionService_credentialsProvider, credentialsProvider, "f");
+        __classPrivateFieldSet(this, _ChatSessionService_config, config, "f");
     }
     async generateAssistantResponse(request) {
-        __classPrivateFieldSet(this, _ChatSessionService_abortController, new abort_controller_1.AbortController(), "f");
+        __classPrivateFieldSet(this, _ChatSessionService_abortController, new AbortController(), "f");
         if (__classPrivateFieldGet(this, _ChatSessionService_sessionId, "f") && request.conversationState) {
             request.conversationState.conversationId = __classPrivateFieldGet(this, _ChatSessionService_sessionId, "f");
         }
-        const response = await __classPrivateFieldGet(this, _ChatSessionService_client, "f").generateAssistantResponse(request, {
+        const client = new codewhisperer_streaming_1.CodeWhispererStreaming({
+            region: __classPrivateFieldGet(this, _ChatSessionService_codeWhispererRegion, "f"),
+            endpoint: __classPrivateFieldGet(this, _ChatSessionService_codeWhispererEndpoint, "f"),
+            token: () => Promise.resolve({ token: (0, utils_1.getBearerTokenFromProvider)(__classPrivateFieldGet(this, _ChatSessionService_credentialsProvider, "f")) }),
+            retryStrategy: new util_retry_1.ConfiguredRetryStrategy(0, (attempt) => 500 + attempt ** 10),
+            customUserAgent: '%Amazon-Q-For-LanguageServers%',
+            ...__classPrivateFieldGet(this, _ChatSessionService_config, "f"),
+        });
+        const response = await client.generateAssistantResponse(request, {
             abortSignal: __classPrivateFieldGet(this, _ChatSessionService_abortController, "f")?.signal,
         });
         __classPrivateFieldSet(this, _ChatSessionService_sessionId, response.conversationId, "f");
         return response;
     }
+    clear() {
+        __classPrivateFieldGet(this, _ChatSessionService_abortController, "f")?.abort();
+        __classPrivateFieldSet(this, _ChatSessionService_sessionId, undefined, "f");
+    }
     dispose() {
         __classPrivateFieldGet(this, _ChatSessionService_abortController, "f")?.abort();
-        __classPrivateFieldGet(this, _ChatSessionService_client, "f").destroy();
     }
     abortRequest() {
         __classPrivateFieldGet(this, _ChatSessionService_abortController, "f")?.abort();
     }
 }
 exports.ChatSessionService = ChatSessionService;
-_ChatSessionService_codeWhispererRegion = new WeakMap(), _ChatSessionService_codeWhispererEndpoint = new WeakMap(), _ChatSessionService_client = new WeakMap(), _ChatSessionService_abortController = new WeakMap(), _ChatSessionService_sessionId = new WeakMap();
+_ChatSessionService_codeWhispererRegion = new WeakMap(), _ChatSessionService_codeWhispererEndpoint = new WeakMap(), _ChatSessionService_abortController = new WeakMap(), _ChatSessionService_credentialsProvider = new WeakMap(), _ChatSessionService_config = new WeakMap(), _ChatSessionService_sessionId = new WeakMap();
 //# sourceMappingURL=chatSessionService.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"chatSessionService.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2EAKsC;AACtC,oDAA6D;AAE7D,+DAA0D;AAC1D,oCAAqD;AAGrD,MAAa,kBAAkB;IAQ3B,IAAW,SAAS;QAChB,OAAO,uBAAA,IAAI,qCAAW,CAAA;IAC1B,CAAC;IAED,IAAW,SAAS,CAAC,KAAyB;QAC1C,uBAAA,IAAI,iCAAc,KAAK,MAAA,CAAA;IAC3B,CAAC;IAED,YAAY,mBAAwC,EAAE,MAA2C;QAf1F,qCAAgC,GAAG,KAAK,CAAA;QACtC,kDAAuB,WAAW,EAAA;QAClC,oDAAyB,gDAAgD,EAAA;QAClF,6CAA+B;QAC/B,sDAAkC;QAClC,gDAAmB;QAWf,uBAAA,IAAI,8BAAW,IAAI,gDAAsB,CAAC;YACtC,MAAM,EAAE,uBAAA,IAAI,+CAAqB;YACjC,QAAQ,EAAE,uBAAA,IAAI,iDAAuB;YACrC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAA,kCAA0B,EAAC,mBAAmB,CAAC,EAAE,CAAC;YACxF,aAAa,EAAE,IAAI,oCAAuB,CAAC,CAAC,EAAE,CAAC,OAAe,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,IAAI,EAAE,CAAC;YACvF,GAAG,MAAM;SACZ,CAAC,MAAA,CAAA;IACN,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAClC,OAA8C;QAE9C,uBAAA,IAAI,uCAAoB,IAAI,kCAAe,EAAE,MAAA,CAAA;QAE7C,IAAI,uBAAA,IAAI,qCAAW,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC;YAC/C,OAAO,CAAC,iBAAiB,CAAC,cAAc,GAAG,uBAAA,IAAI,qCAAW,CAAA;QAC9D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,uBAAA,IAAI,kCAAQ,CAAC,yBAAyB,CAAC,OAAO,EAAE;YACnE,WAAW,EAAE,uBAAA,IAAI,2CAAiB,EAAE,MAAM;SAC7C,CAAC,CAAA;QAEF,uBAAA,IAAI,iCAAc,QAAQ,CAAC,cAAc,MAAA,CAAA;QAEzC,OAAO,QAAQ,CAAA;IACnB,CAAC;IAEM,OAAO;QACV,uBAAA,IAAI,2CAAiB,EAAE,KAAK,EAAE,CAAA;QAC9B,uBAAA,IAAI,kCAAQ,CAAC,OAAO,EAAE,CAAA;IAC1B,CAAC;IAEM,YAAY;QACf,uBAAA,IAAI,2CAAiB,EAAE,KAAK,EAAE,CAAA;IAClC,CAAC;CACJ;AApDD,gDAoDC"}
+{"version":3,"file":"chatSessionService.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2EAKsC;AACtC,oDAA6D;AAE7D,oCAAqD;AAGrD,MAAa,kBAAkB;IAS3B,IAAW,SAAS;QAChB,OAAO,uBAAA,IAAI,qCAAW,CAAA;IAC1B,CAAC;IAED,IAAW,SAAS,CAAC,KAAyB;QAC1C,uBAAA,IAAI,iCAAc,KAAK,MAAA,CAAA;IAC3B,CAAC;IAED,YAAY,mBAAwC,EAAE,MAA2C;QAhB1F,qCAAgC,GAAG,KAAK,CAAA;QAC/C,kDAAgC,WAAW,EAAA;QAC3C,oDAAkC,gDAAgD,EAAA;QAClF,sDAAkC;QAClC,0DAAyC;QACzC,6CAA4C;QAC5C,gDAAmB;QAWf,uBAAA,IAAI,2CAAwB,mBAAmB,MAAA,CAAA;QAC/C,uBAAA,IAAI,8BAAW,MAAM,MAAA,CAAA;IACzB,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAClC,OAA8C;QAE9C,uBAAA,IAAI,uCAAoB,IAAI,eAAe,EAAE,MAAA,CAAA;QAE7C,IAAI,uBAAA,IAAI,qCAAW,IAAI,OAAO,CAAC,iBAAiB,EAAE;YAC9C,OAAO,CAAC,iBAAiB,CAAC,cAAc,GAAG,uBAAA,IAAI,qCAAW,CAAA;SAC7D;QAED,MAAM,MAAM,GAAG,IAAI,gDAAsB,CAAC;YACtC,MAAM,EAAE,uBAAA,IAAI,+CAAqB;YACjC,QAAQ,EAAE,uBAAA,IAAI,iDAAuB;YACrC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,IAAA,kCAA0B,EAAC,uBAAA,IAAI,+CAAqB,CAAC,EAAE,CAAC;YAC9F,aAAa,EAAE,IAAI,oCAAuB,CAAC,CAAC,EAAE,CAAC,OAAe,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,IAAI,EAAE,CAAC;YACvF,eAAe,EAAE,gCAAgC;YACjD,GAAG,uBAAA,IAAI,kCAAQ;SAClB,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,OAAO,EAAE;YAC7D,WAAW,EAAE,uBAAA,IAAI,2CAAiB,EAAE,MAAM;SAC7C,CAAC,CAAA;QAEF,uBAAA,IAAI,iCAAc,QAAQ,CAAC,cAAc,MAAA,CAAA;QAEzC,OAAO,QAAQ,CAAA;IACnB,CAAC;IAEM,KAAK;QACR,uBAAA,IAAI,2CAAiB,EAAE,KAAK,EAAE,CAAA;QAC9B,uBAAA,IAAI,iCAAc,SAAS,MAAA,CAAA;IAC/B,CAAC;IAEM,OAAO;QACV,uBAAA,IAAI,2CAAiB,EAAE,KAAK,EAAE,CAAA;IAClC,CAAC;IAEM,YAAY;QACf,uBAAA,IAAI,2CAAiB,EAAE,KAAK,EAAE,CAAA;IAClC,CAAC;CACJ;AA7DD,gDA6DC"}

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

@@ -1,14 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
-const abort_controller_1 = require("@smithy/abort-controller");
 const assert = require("assert");
 const ts_sinon_1 = require("ts-sinon");
 const chatSessionService_1 = require("./chatSessionService");
 describe('Chat Session Service', () => {
     let generateAssistantResponseStub;
     let abortStub;
-    let destroyClientStub;
     let chatSessionService;
     const mockCredentialsProvider = {
         hasCredentials: ts_sinon_1.default.stub().returns(true),
@@ -32,16 +30,14 @@ describe('Chat Session Service', () => {
         generateAssistantResponseResponse: undefined,
     };
     beforeEach(() => {
-        abortStub = ts_sinon_1.default.stub(abort_controller_1.AbortController.prototype, 'abort');
+        abortStub = ts_sinon_1.default.stub(AbortController.prototype, 'abort');
         generateAssistantResponseStub = ts_sinon_1.default
             .stub(codewhisperer_streaming_1.CodeWhispererStreaming.prototype, 'generateAssistantResponse')
             .callsFake(() => Promise.resolve(mockRequestResponse));
-        destroyClientStub = ts_sinon_1.default.stub(codewhisperer_streaming_1.CodeWhispererStreaming.prototype, 'destroy');
         chatSessionService = new chatSessionService_1.ChatSessionService(mockCredentialsProvider);
     });
     afterEach(() => {
         generateAssistantResponseStub.restore();
-        destroyClientStub.restore();
         abortStub.restore();
     });
     describe('calling GenerateAssistantResponse', () => {
@@ -65,16 +61,22 @@ describe('Chat Session Service', () => {
                 .calledWithExactly(requestParamsWithConversationId, ts_sinon_1.default.match.object));
         });
     });
-    it('calling .abortRequest() aborts request with AbortController', async () => {
+    it('abortRequest() aborts request with AbortController', async () => {
         await chatSessionService.generateAssistantResponse(mockRequestParams);
         chatSessionService.abortRequest();
         ts_sinon_1.default.assert.calledOnce(abortStub);
     });
-    it('calling .dispose() calls client.destroy and aborts outgoing requests', async () => {
+    it('dispose() calls aborts outgoing requests', async () => {
         await chatSessionService.generateAssistantResponse(mockRequestParams);
         chatSessionService.dispose();
         ts_sinon_1.default.assert.calledOnce(abortStub);
-        ts_sinon_1.default.assert.calledOnce(destroyClientStub);
+    });
+    it('clear() reset session id and aborts outgoing request', async () => {
+        await chatSessionService.generateAssistantResponse(mockRequestParams);
+        assert.strictEqual(chatSessionService.sessionId, mockConversationId);
+        chatSessionService.clear();
+        ts_sinon_1.default.assert.calledOnce(abortStub);
+        assert.strictEqual(chatSessionService.sessionId, undefined);
     });
 });
 //# sourceMappingURL=chatSessionService.test.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"chatSessionService.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionService.test.ts"],"names":[],"mappings":";;AAAA,2EAIsC;AAEtC,+DAA0D;AAC1D,iCAAgC;AAChC,uCAA4B;AAC5B,6DAAyD;AAEzD,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IAClC,IAAI,6BAAwD,CAAA;IAC5D,IAAI,SAAoC,CAAA;IACxC,IAAI,iBAA4C,CAAA;IAChD,IAAI,kBAAsC,CAAA;IAC1C,MAAM,uBAAuB,GAAwB;QACjD,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;QAC1C,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAC9E,qBAAqB,EAAE,kBAAK,CAAC,IAAI,EAAE;KACtC,CAAA;IAED,MAAM,kBAAkB,GAAG,oBAAoB,CAAA;IAE/C,MAAM,iBAAiB,GAA0C;QAC7D,iBAAiB,EAAE;YACf,eAAe,EAAE,QAAQ;YACzB,cAAc,EAAE;gBACZ,gBAAgB,EAAE;oBACd,OAAO,EAAE,OAAO;iBACnB;aACJ;SACJ;KACJ,CAAA;IAED,MAAM,mBAAmB,GAA2C;QAChE,cAAc,EAAE,kBAAkB;QAClC,SAAS,EAAE,EAAE;QACb,iCAAiC,EAAE,SAAS;KAC/C,CAAA;IAED,UAAU,CAAC,GAAG,EAAE;QACZ,SAAS,GAAG,kBAAK,CAAC,IAAI,CAAC,kCAAe,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;QAE1D,6BAA6B,GAAG,kBAAK;aAChC,IAAI,CAAC,gDAAsB,CAAC,SAAS,EAAE,2BAA2B,CAAC;aACnE,SAAS,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAA;QAE1D,iBAAiB,GAAG,kBAAK,CAAC,IAAI,CAAC,gDAAsB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;QAE3E,kBAAkB,GAAG,IAAI,uCAAkB,CAAC,uBAAuB,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,6BAA6B,CAAC,OAAO,EAAE,CAAA;QACvC,iBAAiB,CAAC,OAAO,EAAE,CAAA;QAC3B,SAAS,CAAC,OAAO,EAAE,CAAA;IACvB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;YACvG,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;YAErE,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,6BAA6B,EAAE,iBAAiB,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;YACxG,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAA;QACxE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;YAErE,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,6BAA6B,EAAE,iBAAiB,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;YAExG,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;YAErE,MAAM,+BAA+B,GAAG;gBACpC,iBAAiB,EAAE;oBACf,GAAG,iBAAiB,CAAC,iBAAiB;oBACtC,cAAc,EAAE,kBAAkB;iBACrC;aACJ,CAAA;YAED,MAAM,CAAC,EAAE,CACL,6BAA6B;iBACxB,OAAO,CAAC,CAAC,CAAC;iBACV,iBAAiB,CAAC,+BAA+B,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAC9E,CAAA;QACL,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;QAErE,kBAAkB,CAAC,YAAY,EAAE,CAAA;QAEjC,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;QAErE,kBAAkB,CAAC,OAAO,EAAE,CAAA;QAE5B,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAClC,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}
+{"version":3,"file":"chatSessionService.test.js","sourceRoot":"","sources":["../../../src/language-server/chat/chatSessionService.test.ts"],"names":[],"mappings":";;AAAA,2EAIsC;AAEtC,iCAAgC;AAChC,uCAA4B;AAC5B,6DAAyD;AAEzD,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IAClC,IAAI,6BAAwD,CAAA;IAC5D,IAAI,SAAoC,CAAA;IACxC,IAAI,kBAAsC,CAAA;IAC1C,MAAM,uBAAuB,GAAwB;QACjD,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;QAC1C,cAAc,EAAE,kBAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAC9E,qBAAqB,EAAE,kBAAK,CAAC,IAAI,EAAE;KACtC,CAAA;IAED,MAAM,kBAAkB,GAAG,oBAAoB,CAAA;IAE/C,MAAM,iBAAiB,GAA0C;QAC7D,iBAAiB,EAAE;YACf,eAAe,EAAE,QAAQ;YACzB,cAAc,EAAE;gBACZ,gBAAgB,EAAE;oBACd,OAAO,EAAE,OAAO;iBACnB;aACJ;SACJ;KACJ,CAAA;IAED,MAAM,mBAAmB,GAA2C;QAChE,cAAc,EAAE,kBAAkB;QAClC,SAAS,EAAE,EAAE;QACb,iCAAiC,EAAE,SAAS;KAC/C,CAAA;IAED,UAAU,CAAC,GAAG,EAAE;QACZ,SAAS,GAAG,kBAAK,CAAC,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;QAE1D,6BAA6B,GAAG,kBAAK;aAChC,IAAI,CAAC,gDAAsB,CAAC,SAAS,EAAE,2BAA2B,CAAC;aACnE,SAAS,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAA;QAE1D,kBAAkB,GAAG,IAAI,uCAAkB,CAAC,uBAAuB,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACX,6BAA6B,CAAC,OAAO,EAAE,CAAA;QACvC,SAAS,CAAC,OAAO,EAAE,CAAA;IACvB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,2FAA2F,EAAE,KAAK,IAAI,EAAE;YACvG,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;YAErE,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,6BAA6B,EAAE,iBAAiB,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;YACxG,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAA;QACxE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;YAErE,kBAAK,CAAC,MAAM,CAAC,qBAAqB,CAAC,6BAA6B,EAAE,iBAAiB,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;YAExG,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;YAErE,MAAM,+BAA+B,GAAG;gBACpC,iBAAiB,EAAE;oBACf,GAAG,iBAAiB,CAAC,iBAAiB;oBACtC,cAAc,EAAE,kBAAkB;iBACrC;aACJ,CAAA;YAED,MAAM,CAAC,EAAE,CACL,6BAA6B;iBACxB,OAAO,CAAC,CAAC,CAAC;iBACV,iBAAiB,CAAC,+BAA+B,EAAE,kBAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAC9E,CAAA;QACL,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;QAErE,kBAAkB,CAAC,YAAY,EAAE,CAAA;QAEjC,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;QAErE,kBAAkB,CAAC,OAAO,EAAE,CAAA;QAE5B,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,kBAAkB,CAAC,yBAAyB,CAAC,iBAAiB,CAAC,CAAA;QAErE,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAA;QAEpE,kBAAkB,CAAC,KAAK,EAAE,CAAA;QAE1B,kBAAK,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAClC,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;IAC/D,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/chat/contexts/documentContext.d.ts

@@ -1,17 +1,26 @@
-import { EditorState } from '@amzn/codewhisperer-streaming';
-import { CursorState } from '@aws/language-server-runtimes-types';
-import { Range, TextDocument } from 'vscode-languageserver-textdocument';
+import { EditorState, TextDocument as CwsprTextDocument } from '@amzn/codewhisperer-streaming';
+import { CursorState } from '@aws/language-server-runtimes/server-interface';
+import { TextDocument } from 'vscode-languageserver-textdocument';
+import { Features } from '../../types';
+import { DocumentFqnExtractorConfig } from './documentFqnExtractor';
+export type DocumentContext = CwsprTextDocument & {
+    cursorState?: EditorState['cursorState'];
+    hasCodeSnippet: boolean;
+    totalEditorCharacters: number;
+};
+export interface DocumentContextExtractorConfig extends DocumentFqnExtractorConfig {
+    config?: DocumentFqnExtractorConfig;
+    logger?: Features['logging'];
+    characterLimits?: number;
+}
 export declare class DocumentContextExtractor {
     #private;
     private static readonly DEFAULT_CHARACTER_LIMIT;
-    constructor(characterLimits?: number);
+    constructor(config?: DocumentContextExtractorConfig);
+    dispose(): void;
     /**
      * From the given the cursor state, we want to give Q context up to the characters limit
      * on both sides of the cursor.
      */
-    extractEditorState(document: TextDocument, cursorState: CursorState): Promise<EditorState>;
-    /**
-     * Extract document context from the given range inside a document
-     */
-    extractDocumentContext(document: TextDocument, codeBlockRange: Range): Promise<EditorState['document']>;
+    extractDocumentContext(document: TextDocument, cursorState: CursorState): Promise<DocumentContext>;
 }