@aws/lsp-codewhisperer 0.0.3 → 0.0.5

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

@@ -0,0 +1,189 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const ts_sinon_1 = require("ts-sinon");
+const chatEventParser_1 = require("./chatEventParser");
+describe('ChatEventParser', () => {
+    const mockMessageId = 'mock-message-id';
+    it('set error if invalidState event is received', () => {
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        ts_sinon_1.default.assert.match(chatEventParser.processPartialEvent({
+            invalidStateEvent: {
+                message: 'Invalid state!',
+                reason: 'Unknown!',
+            },
+        }), {
+            success: false,
+            data: {
+                messageId: mockMessageId,
+                body: undefined,
+                canBeVoted: undefined,
+                codeReference: undefined,
+                followUp: undefined,
+                relatedContent: undefined,
+            },
+            error: ts_sinon_1.default.match.string,
+        });
+    });
+    it('set error if error event is received', () => {
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        ts_sinon_1.default.assert.match(chatEventParser.processPartialEvent({
+            error: {
+                name: 'InternalServerException',
+                message: 'Error!',
+                $fault: 'server',
+                $retryable: false,
+                $metadata: {},
+            },
+        }), {
+            success: false,
+            data: {
+                messageId: mockMessageId,
+                body: undefined,
+                canBeVoted: undefined,
+                codeReference: undefined,
+                followUp: undefined,
+                relatedContent: undefined,
+            },
+            error: ts_sinon_1.default.match.string,
+        });
+        assert.strictEqual(chatEventParser.error, 'Error!');
+    });
+    it('processPartialEvent appends new event on top of the previous result', () => {
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        assert.deepStrictEqual(chatEventParser.processPartialEvent({
+            assistantResponseEvent: {
+                content: 'This is an ',
+            },
+        }), {
+            success: true,
+            data: {
+                messageId: mockMessageId,
+                body: 'This is an ',
+                canBeVoted: undefined,
+                codeReference: undefined,
+                followUp: undefined,
+                relatedContent: undefined,
+            },
+        });
+        assert.deepStrictEqual(chatEventParser.processPartialEvent({
+            assistantResponseEvent: {
+                content: 'assistant response.',
+            },
+        }), {
+            success: true,
+            data: {
+                messageId: mockMessageId,
+                body: 'This is an assistant response.',
+                canBeVoted: undefined,
+                codeReference: undefined,
+                followUp: undefined,
+                relatedContent: undefined,
+            },
+        });
+    });
+    it('getChatResult returns the accumulated result', () => {
+        const chatEventParser = new chatEventParser_1.ChatEventParser(mockMessageId);
+        chatEventParser.processPartialEvent({
+            assistantResponseEvent: {
+                content: 'This is an ',
+            },
+        });
+        chatEventParser.processPartialEvent({
+            followupPromptEvent: {
+                followupPrompt: {
+                    content: 'follow up 1',
+                },
+            },
+        });
+        assert.deepStrictEqual(chatEventParser.getChatResult(), {
+            success: true,
+            data: {
+                messageId: mockMessageId,
+                body: 'This is an ',
+                followUp: {
+                    options: [
+                        {
+                            pillText: 'follow up 1',
+                            prompt: 'follow up 1',
+                        },
+                    ],
+                },
+                canBeVoted: undefined,
+                codeReference: undefined,
+                relatedContent: undefined,
+            },
+        });
+        chatEventParser.processPartialEvent({
+            followupPromptEvent: {
+                followupPrompt: {
+                    content: 'follow up 2',
+                },
+            },
+        });
+        chatEventParser.processPartialEvent({
+            codeReferenceEvent: {
+                references: [
+                    {
+                        licenseName: 'MIT',
+                        repository: 'langauge-servers',
+                    },
+                ],
+            },
+        });
+        chatEventParser.processPartialEvent({
+            assistantResponseEvent: {
+                content: 'assistant response.',
+            },
+        });
+        chatEventParser.processPartialEvent({
+            supplementaryWebLinksEvent: {
+                supplementaryWebLinks: [
+                    {
+                        title: 'weblink1',
+                        snippet: 'snippet1',
+                        url: 'https://link1.com',
+                    },
+                ],
+            },
+        });
+        const expectedData = {
+            messageId: mockMessageId,
+            body: 'This is an assistant response.',
+            canBeVoted: undefined,
+            relatedContent: {
+                content: [
+                    {
+                        title: 'weblink1',
+                        body: 'snippet1',
+                        url: 'https://link1.com',
+                    },
+                ],
+            },
+            codeReference: [
+                {
+                    licenseName: 'MIT',
+                    repository: 'langauge-servers',
+                    information: chatEventParser_1.ChatEventParser.getReferencedInformation({
+                        licenseName: 'MIT',
+                        repository: 'langauge-servers',
+                    }),
+                },
+            ],
+            followUp: {
+                options: [
+                    {
+                        pillText: 'follow up 1',
+                        prompt: 'follow up 1',
+                    },
+                    {
+                        pillText: 'follow up 2',
+                        prompt: 'follow up 2',
+                    },
+                ],
+            },
+        };
+        assert.deepStrictEqual(chatEventParser.getChatResult(), { success: true, data: expectedData });
+    });
+});
+//# sourceMappingURL=chatEventParser.test.js.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,16 @@
+import { CredentialsProvider } from '@aws/language-server-runtimes/server-interface';
+import { Result } from '../types';
+import { ChatSessionService, ChatSessionServiceConfig } from './chatSessionService';
+export declare class ChatSessionManagementService {
+    #private;
+    static getInstance(): ChatSessionManagementService;
+    static reset(): void;
+    private constructor();
+    withConfig(clientConfig?: ChatSessionServiceConfig | (() => ChatSessionServiceConfig)): this;
+    withCredentialsProvider(credentialsProvider: CredentialsProvider): this;
+    hasSession(tabId: string): boolean;
+    createSession(tabId: string): Result<ChatSessionService, string>;
+    getSession(tabId: string): Result<ChatSessionService, string>;
+    deleteSession(tabId: string): Result<void, string>;
+    dispose(): void;
+}

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

@@ -0,0 +1,92 @@
+"use strict";
+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 __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 _a, _ChatSessionManagementService_instance, _ChatSessionManagementService_sessionByTab, _ChatSessionManagementService_credentialsProvider, _ChatSessionManagementService_clientConfig;
+Object.defineProperty(exports, "__esModule", { value: true });
+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);
+        }
+        return __classPrivateFieldGet(_a, _a, "f", _ChatSessionManagementService_instance);
+    }
+    static reset() {
+        __classPrivateFieldSet(_a, _a, undefined, "f", _ChatSessionManagementService_instance);
+    }
+    constructor() {
+        _ChatSessionManagementService_sessionByTab.set(this, new Map());
+        _ChatSessionManagementService_credentialsProvider.set(this, void 0);
+        _ChatSessionManagementService_clientConfig.set(this, void 0);
+    }
+    withConfig(clientConfig) {
+        __classPrivateFieldSet(this, _ChatSessionManagementService_clientConfig, clientConfig, "f");
+        return this;
+    }
+    withCredentialsProvider(credentialsProvider) {
+        __classPrivateFieldSet(this, _ChatSessionManagementService_credentialsProvider, credentialsProvider, "f");
+        return this;
+    }
+    hasSession(tabId) {
+        return __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").has(tabId);
+    }
+    createSession(tabId) {
+        if (!__classPrivateFieldGet(this, _ChatSessionManagementService_credentialsProvider, "f")) {
+            return {
+                success: false,
+                error: 'Credentials provider is not set',
+            };
+        }
+        else if (__classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").has(tabId)) {
+            return {
+                success: false,
+                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");
+        const newSession = new chatSessionService_1.ChatSessionService(__classPrivateFieldGet(this, _ChatSessionManagementService_credentialsProvider, "f"), clientConfig);
+        __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").set(tabId, newSession);
+        return {
+            success: true,
+            data: newSession,
+        };
+    }
+    getSession(tabId) {
+        const session = __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").get(tabId);
+        return session
+            ? {
+                success: true,
+                data: session,
+            }
+            : {
+                success: false,
+                error: 'Session does not exist',
+            };
+    }
+    deleteSession(tabId) {
+        __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").get(tabId)?.dispose();
+        __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").delete(tabId);
+        return {
+            success: true,
+            data: undefined,
+        };
+    }
+    dispose() {
+        __classPrivateFieldGet(this, _ChatSessionManagementService_sessionByTab, "f").forEach(session => session.dispose());
+    }
+}
+exports.ChatSessionManagementService = ChatSessionManagementService;
+_a = ChatSessionManagementService, _ChatSessionManagementService_sessionByTab = new WeakMap(), _ChatSessionManagementService_credentialsProvider = new WeakMap(), _ChatSessionManagementService_clientConfig = new WeakMap();
+_ChatSessionManagementService_instance = { value: void 0 };
+//# sourceMappingURL=chatSessionManagementService.js.map

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

@@ -0,0 +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"}

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

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

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

@@ -0,0 +1,73 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const assert = require("assert");
+const ts_sinon_1 = require("ts-sinon");
+const chatSessionManagementService_1 = require("./chatSessionManagementService");
+const chatSessionService_1 = require("./chatSessionService");
+describe('ChatSessionManagementService', () => {
+    const mockSessionId = 'mockSessionId';
+    it('getInstance should return the same instance if initialized', () => {
+        const instance = chatSessionManagementService_1.ChatSessionManagementService.getInstance();
+        const instance2 = chatSessionManagementService_1.ChatSessionManagementService.getInstance();
+        assert.strictEqual(instance, instance2);
+    });
+    it('creating a session without credentials provider should throw an error', () => {
+        const createSessionResult = chatSessionManagementService_1.ChatSessionManagementService.getInstance().createSession(mockSessionId);
+        ts_sinon_1.default.assert.match(createSessionResult, { success: false, error: ts_sinon_1.default.match.string });
+    });
+    describe('Session interface', () => {
+        const mockCredentialsProvider = {
+            hasCredentials: ts_sinon_1.default.stub().returns(true),
+            getCredentials: ts_sinon_1.default.stub().returns(Promise.resolve({ token: 'mockToken' })),
+            getConnectionMetadata: ts_sinon_1.default.stub(),
+        };
+        const mockSessionId2 = 'mockSessionId2';
+        let disposeStub;
+        let chatSessionManagementService;
+        beforeEach(() => {
+            disposeStub = ts_sinon_1.default.stub(chatSessionService_1.ChatSessionService.prototype, 'dispose');
+            chatSessionManagementService =
+                chatSessionManagementService_1.ChatSessionManagementService.getInstance().withCredentialsProvider(mockCredentialsProvider);
+        });
+        afterEach(() => {
+            chatSessionManagementService_1.ChatSessionManagementService.reset();
+            disposeStub.restore();
+        });
+        it('getSession should return an 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, {
+                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);
+        });
+        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('deleting session should dispose the chat session service and delete from map', () => {
+            chatSessionManagementService.createSession(mockSessionId);
+            assert.ok(chatSessionManagementService.hasSession(mockSessionId));
+            chatSessionManagementService.deleteSession(mockSessionId);
+            assert.ok(!chatSessionManagementService.hasSession(mockSessionId));
+            ts_sinon_1.default.assert.calledOnce(disposeStub);
+        });
+        it('disposing the chat session management should dispose all the chat session services', () => {
+            chatSessionManagementService.createSession(mockSessionId);
+            chatSessionManagementService.createSession(mockSessionId2);
+            chatSessionManagementService.dispose();
+            ts_sinon_1.default.assert.calledTwice(disposeStub);
+        });
+    });
+});
+//# sourceMappingURL=chatSessionManagementService.test.js.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,13 @@
+import { CodeWhispererStreamingClientConfig, GenerateAssistantResponseCommandInput, GenerateAssistantResponseCommandOutput } from '@amzn/codewhisperer-streaming';
+import { CredentialsProvider } from '@aws/language-server-runtimes/server-interface';
+export type ChatSessionServiceConfig = CodeWhispererStreamingClientConfig;
+export declare class ChatSessionService {
+    #private;
+    shareCodeWhispererContentWithAWS: boolean;
+    get sessionId(): string | undefined;
+    set sessionId(value: string | undefined);
+    constructor(credentialsProvider: CredentialsProvider, config?: CodeWhispererStreamingClientConfig);
+    generateAssistantResponse(request: GenerateAssistantResponseCommandInput): Promise<GenerateAssistantResponseCommandOutput>;
+    dispose(): void;
+    abortRequest(): void;
+}

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

@@ -0,0 +1,63 @@
+"use strict";
+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 __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 _ChatSessionService_codeWhispererRegion, _ChatSessionService_codeWhispererEndpoint, _ChatSessionService_client, _ChatSessionService_abortController, _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() {
+        return __classPrivateFieldGet(this, _ChatSessionService_sessionId, "f");
+    }
+    set sessionId(value) {
+        __classPrivateFieldSet(this, _ChatSessionService_sessionId, value, "f");
+    }
+    constructor(credentialsProvider, config) {
+        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_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");
+    }
+    async generateAssistantResponse(request) {
+        __classPrivateFieldSet(this, _ChatSessionService_abortController, new abort_controller_1.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, {
+            abortSignal: __classPrivateFieldGet(this, _ChatSessionService_abortController, "f")?.signal,
+        });
+        __classPrivateFieldSet(this, _ChatSessionService_sessionId, response.conversationId, "f");
+        return response;
+    }
+    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();
+//# sourceMappingURL=chatSessionService.js.map

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

@@ -0,0 +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"}

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

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

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

@@ -0,0 +1,80 @@
+"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),
+        getCredentials: ts_sinon_1.default.stub().returns(Promise.resolve({ token: 'mockToken ' })),
+        getConnectionMetadata: ts_sinon_1.default.stub(),
+    };
+    const mockConversationId = 'mockConversationId';
+    const mockRequestParams = {
+        conversationState: {
+            chatTriggerType: 'MANUAL',
+            currentMessage: {
+                userInputMessage: {
+                    content: 'hello',
+                },
+            },
+        },
+    };
+    const mockRequestResponse = {
+        conversationId: mockConversationId,
+        $metadata: {},
+        generateAssistantResponseResponse: undefined,
+    };
+    beforeEach(() => {
+        abortStub = ts_sinon_1.default.stub(abort_controller_1.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', () => {
+        it('should call generate assistant response from the streaming client and set the session id ', async () => {
+            await chatSessionService.generateAssistantResponse(mockRequestParams);
+            ts_sinon_1.default.assert.calledOnceWithExactly(generateAssistantResponseStub, mockRequestParams, ts_sinon_1.default.match.object);
+            assert.strictEqual(chatSessionService.sessionId, mockConversationId);
+        });
+        it('should fill in conversationId with session id in the request if exists', async () => {
+            await chatSessionService.generateAssistantResponse(mockRequestParams);
+            ts_sinon_1.default.assert.calledOnceWithExactly(generateAssistantResponseStub, mockRequestParams, ts_sinon_1.default.match.object);
+            await chatSessionService.generateAssistantResponse(mockRequestParams);
+            const requestParamsWithConversationId = {
+                conversationState: {
+                    ...mockRequestParams.conversationState,
+                    conversationId: mockConversationId,
+                },
+            };
+            assert.ok(generateAssistantResponseStub
+                .getCall(1)
+                .calledWithExactly(requestParamsWithConversationId, ts_sinon_1.default.match.object));
+        });
+    });
+    it('calling .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 () => {
+        await chatSessionService.generateAssistantResponse(mockRequestParams);
+        chatSessionService.dispose();
+        ts_sinon_1.default.assert.calledOnce(abortStub);
+        ts_sinon_1.default.assert.calledOnce(destroyClientStub);
+    });
+});
+//# sourceMappingURL=chatSessionService.test.js.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,17 @@
+import { EditorState } from '@amzn/codewhisperer-streaming';
+import { CursorState } from '@aws/language-server-runtimes-types';
+import { Range, TextDocument } from 'vscode-languageserver-textdocument';
+export declare class DocumentContextExtractor {
+    #private;
+    private static readonly DEFAULT_CHARACTER_LIMIT;
+    constructor(characterLimits?: number);
+    /**
+     * 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']>;
+}