@aws/lsp-codewhisperer 0.0.28 → 0.0.30

package/out/language-server/amazonQServiceManager/errors.js

@@ -0,0 +1,70 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AmazonQServiceNoProfileSupportError = exports.AmazonQServiceInvalidProfileError = exports.AmazonQServiceProfileUpdateCancelled = exports.AmazonQServicePendingProfileUpdateError = exports.AmazonQServicePendingProfileError = exports.AmazonQServicePendingSigninError = exports.AmazonQServiceNotInitializedError = exports.AmazonQServiceInitializationError = exports.AmazonQError = void 0;
+// Base error class for Amazon Q
+class AmazonQError extends Error {
+    code;
+    constructor(message, code) {
+        super(message);
+        this.name = 'AmazonQError';
+        this.code = code;
+    }
+}
+exports.AmazonQError = AmazonQError;
+class AmazonQServiceInitializationError extends AmazonQError {
+    constructor(message = 'Amazon Q service manager initilization error') {
+        super(message, 'E_AMAZON_Q_INITIALIZATION_ERROR');
+        this.name = 'AmazonQServiceInitializationError';
+    }
+}
+exports.AmazonQServiceInitializationError = AmazonQServiceInitializationError;
+class AmazonQServiceNotInitializedError extends AmazonQError {
+    constructor(message = 'Amazon Q service SDK is not initialized') {
+        super(message, 'E_AMAZON_Q_NOT_INITIALIZED');
+        this.name = 'AmazonQServiceNotInitializedError';
+    }
+}
+exports.AmazonQServiceNotInitializedError = AmazonQServiceNotInitializedError;
+class AmazonQServicePendingSigninError extends AmazonQError {
+    constructor(message = 'Amazon Q service is not signed in') {
+        super(message, 'E_AMAZON_Q_PENDING_CONNECTION');
+        this.name = 'AmazonQServicePendingSigninError';
+    }
+}
+exports.AmazonQServicePendingSigninError = AmazonQServicePendingSigninError;
+class AmazonQServicePendingProfileError extends AmazonQError {
+    constructor(message = 'Amazon Q Profile is not selected for IDC connection type') {
+        super(message, 'E_AMAZON_Q_PENDING_PROFILE');
+        this.name = 'AmazonQServicePendingProfileError';
+    }
+}
+exports.AmazonQServicePendingProfileError = AmazonQServicePendingProfileError;
+class AmazonQServicePendingProfileUpdateError extends AmazonQError {
+    constructor(message = 'Amazon Q Profile is pending update') {
+        super(message, 'E_AMAZON_Q_PENDING_PROFILE_UPDATE');
+        this.name = 'AmazonQServicePendingProfileUpdateError';
+    }
+}
+exports.AmazonQServicePendingProfileUpdateError = AmazonQServicePendingProfileUpdateError;
+class AmazonQServiceProfileUpdateCancelled extends AmazonQError {
+    constructor(message = 'Amazon Q Profile cancelled') {
+        super(message, 'E_AMAZON_Q_PROFILE_UPDATE_CANCELLED');
+        this.name = 'AmazonQServiceProfileUpdateCancelled';
+    }
+}
+exports.AmazonQServiceProfileUpdateCancelled = AmazonQServiceProfileUpdateCancelled;
+class AmazonQServiceInvalidProfileError extends AmazonQError {
+    constructor(message = 'Selected Amazon Q Profile is invalid') {
+        super(message, 'E_AMAZON_Q_INVALID_PROFILE');
+        this.name = 'AmazonQServiceInvalidProfileError';
+    }
+}
+exports.AmazonQServiceInvalidProfileError = AmazonQServiceInvalidProfileError;
+class AmazonQServiceNoProfileSupportError extends AmazonQError {
+    constructor(message = 'Current Connection type does not support Amazon Q Profiles') {
+        super(message, 'E_AMAZON_Q_CONNECTION_NO_PROFILE_SUPPORT');
+        this.name = 'AmazonQServiceNoProfileSupportError';
+    }
+}
+exports.AmazonQServiceNoProfileSupportError = AmazonQServiceNoProfileSupportError;
+//# sourceMappingURL=errors.js.map

package/out/language-server/amazonQServiceManager/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../src/language-server/amazonQServiceManager/errors.ts"],"names":[],"mappings":";;;AAAA,gCAAgC;AAChC,MAAa,YAAa,SAAQ,KAAK;IAC5B,IAAI,CAAQ;IACnB,YAAY,OAAe,EAAE,IAAY;QACrC,KAAK,CAAC,OAAO,CAAC,CAAA;QACd,IAAI,CAAC,IAAI,GAAG,cAAc,CAAA;QAC1B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;IACpB,CAAC;CACJ;AAPD,oCAOC;AAED,MAAa,iCAAkC,SAAQ,YAAY;IAC/D,YAAY,UAAkB,8CAA8C;QACxE,KAAK,CAAC,OAAO,EAAE,iCAAiC,CAAC,CAAA;QACjD,IAAI,CAAC,IAAI,GAAG,mCAAmC,CAAA;IACnD,CAAC;CACJ;AALD,8EAKC;AAED,MAAa,iCAAkC,SAAQ,YAAY;IAC/D,YAAY,UAAkB,yCAAyC;QACnE,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAA;QAC5C,IAAI,CAAC,IAAI,GAAG,mCAAmC,CAAA;IACnD,CAAC;CACJ;AALD,8EAKC;AAED,MAAa,gCAAiC,SAAQ,YAAY;IAC9D,YAAY,UAAkB,mCAAmC;QAC7D,KAAK,CAAC,OAAO,EAAE,+BAA+B,CAAC,CAAA;QAC/C,IAAI,CAAC,IAAI,GAAG,kCAAkC,CAAA;IAClD,CAAC;CACJ;AALD,4EAKC;AAED,MAAa,iCAAkC,SAAQ,YAAY;IAC/D,YAAY,UAAkB,0DAA0D;QACpF,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAA;QAC5C,IAAI,CAAC,IAAI,GAAG,mCAAmC,CAAA;IACnD,CAAC;CACJ;AALD,8EAKC;AAED,MAAa,uCAAwC,SAAQ,YAAY;IACrE,YAAY,UAAkB,oCAAoC;QAC9D,KAAK,CAAC,OAAO,EAAE,mCAAmC,CAAC,CAAA;QACnD,IAAI,CAAC,IAAI,GAAG,yCAAyC,CAAA;IACzD,CAAC;CACJ;AALD,0FAKC;AAED,MAAa,oCAAqC,SAAQ,YAAY;IAClE,YAAY,UAAkB,4BAA4B;QACtD,KAAK,CAAC,OAAO,EAAE,qCAAqC,CAAC,CAAA;QACrD,IAAI,CAAC,IAAI,GAAG,sCAAsC,CAAA;IACtD,CAAC;CACJ;AALD,oFAKC;AAED,MAAa,iCAAkC,SAAQ,YAAY;IAC/D,YAAY,UAAkB,sCAAsC;QAChE,KAAK,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAAA;QAC5C,IAAI,CAAC,IAAI,GAAG,mCAAmC,CAAA;IACnD,CAAC;CACJ;AALD,8EAKC;AAED,MAAa,mCAAoC,SAAQ,YAAY;IACjE,YAAY,UAAkB,4DAA4D;QACtF,KAAK,CAAC,OAAO,EAAE,0CAA0C,CAAC,CAAA;QAC1D,IAAI,CAAC,IAAI,GAAG,qCAAqC,CAAA;IACrD,CAAC;CACJ;AALD,kFAKC"}

package/out/language-server/amazonQServiceManager/qDeveloperProfiles.js

@@ -44,9 +44,9 @@ async function fetchProfilesFromRegion(service, region, logging) {
                 },
             }));
             allRegionalProfiles.push(...profiles);
+            logging.debug(`Fetched profiles from ${region}: ${JSON.stringify(response)} (iteration: ${numberOfPages})`);
             nextToken = response.nextToken;
             numberOfPages++;
-            logging.debug(`Fetched profiles from ${region}: ${JSON.stringify(response)} (iteration: ${numberOfPages})`);
         } while (nextToken !== undefined && numberOfPages < exports.MAX_Q_DEVELOPER_PROFILE_PAGES);
         return allRegionalProfiles;
     }
@@ -73,12 +73,12 @@ const developerProfilesEnabledKey = 'developerProfiles';
  * ```
  */
 function signalsAWSQDeveloperProfilesEnabled(initializationOptions) {
-    const qCapibilities = initializationOptions.awsClientCapabilities?.[AWSQCapabilitiesKey];
-    if ((0, utils_1.isObject)(qCapibilities) &&
-        !(qCapibilities instanceof Array) &&
-        developerProfilesEnabledKey in qCapibilities &&
-        (0, utils_1.isBool)(qCapibilities[developerProfilesEnabledKey])) {
-        return qCapibilities[developerProfilesEnabledKey];
+    const qCapabilities = initializationOptions.awsClientCapabilities?.[AWSQCapabilitiesKey];
+    if ((0, utils_1.isObject)(qCapabilities) &&
+        !(qCapabilities instanceof Array) &&
+        developerProfilesEnabledKey in qCapabilities &&
+        (0, utils_1.isBool)(qCapabilities[developerProfilesEnabledKey])) {
+        return qCapabilities[developerProfilesEnabledKey];
     }
     return false;
 }

package/out/language-server/amazonQServiceManager/qDeveloperProfiles.js.map

@@ -1 +1 @@
-{"version":3,"file":"qDeveloperProfiles.js","sourceRoot":"","sources":["../../../src/language-server/amazonQServiceManager/qDeveloperProfiles.ts"],"names":[],"mappings":";;;AAyHA,kFAaC;AAtID,qFAKuD;AACvD,oCAA8D;AAC9D,+CAAiD;AAuBpC,QAAA,6BAA6B,GAAG,EAAE,CAAA;AAC/C,MAAM,iCAAiC,GAAG,EAAE,CAAA;AAErC,MAAM,kCAAkC,GAC3C,CAAC,OAAwE,EAAmC,EAAE,CAC9G,KAAK,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE;IAC7C,IAAI,CAAC,cAAc,IAAI,cAAc,KAAK,gBAAgB,EAAE,CAAC;QACzD,OAAO,CAAC,KAAK,CAAC,8EAA8E,CAAC,CAAA;QAC7F,OAAO,EAAE,CAAA;IACb,CAAC;IAED,IAAI,WAAW,GAA8B,EAAE,CAAA;IAC/C,MAAM,UAAU,GAAG,SAAS,IAAI,2BAAe,CAAA;IAE/C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CACnC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,EAAE;QAClD,MAAM,oBAAoB,GAAG,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QACtD,OAAO,uBAAuB,CAAC,oBAAoB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACzE,CAAC,CAAC,CACL,CAAA;IAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC,aAAa,CAAC,MAAM,KAAK,WAAW,CAAC,CAAA;IAE7F,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,gCAAa,CAAC,gCAAa,CAAC,aAAa,EAAE,sDAAsD,CAAC,CAAA;IAChH,CAAC;IAED,gBAAgB,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAA;IAEvF,OAAO,WAAW,CAAA;AACtB,CAAC,CAAA;AA3BQ,QAAA,kCAAkC,sCA2B1C;AAEL,KAAK,UAAU,uBAAuB,CAClC,OAAkC,EAClC,MAAc,EACd,OAAgB;IAEhB,IAAI,mBAAmB,GAA8B,EAAE,CAAA;IACvD,IAAI,SAAS,GAAuB,SAAS,CAAA;IAC7C,IAAI,aAAa,GAAG,CAAC,CAAA;IAErB,IAAI,CAAC;QACD,GAAG,CAAC;YACA,OAAO,CAAC,KAAK,CAAC,kCAAkC,MAAM,gBAAgB,aAAa,GAAG,CAAC,CAAA;YAEvF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,qBAAqB,CAAC;gBACjD,UAAU,EAAE,iCAAiC;aAChD,CAAC,CAAA;YAEF,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC/C,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,IAAI,EAAE,OAAO,CAAC,WAAW;gBACzB,eAAe,EAAE;oBACb,MAAM;iBACT;aACJ,CAAC,CAAC,CAAA;YAEH,mBAAmB,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAA;YAErC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAA;YAC9B,aAAa,EAAE,CAAA;YAEf,OAAO,CAAC,KAAK,CAAC,yBAAyB,MAAM,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,gBAAgB,aAAa,GAAG,CAAC,CAAA;QAC/G,CAAC,QAAQ,SAAS,KAAK,SAAS,IAAI,aAAa,GAAG,qCAA6B,EAAC;QAElF,OAAO,mBAAmB,CAAA;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,MAAM,KAAK,KAAK,EAAE,CAAC,CAAA;QAEjE,MAAM,KAAK,CAAA;IACf,CAAC;AACL,CAAC;AAED,MAAM,mBAAmB,GAAG,GAAG,CAAA;AAC/B,MAAM,2BAA2B,GAAG,mBAAmB,CAAA;AAEvD;;;;;;;;;;;;;;GAcG;AACH,SAAgB,mCAAmC,CAAC,qBAA+C;IAC/F,MAAM,aAAa,GAAG,qBAAqB,CAAC,qBAAqB,EAAE,CAAC,mBAAmB,CAAC,CAAA;IAExF,IACI,IAAA,gBAAQ,EAAC,aAAa,CAAC;QACvB,CAAC,CAAC,aAAa,YAAY,KAAK,CAAC;QACjC,2BAA2B,IAAI,aAAa;QAC5C,IAAA,cAAM,EAAC,aAAa,CAAC,2BAA2B,CAAC,CAAC,EACpD,CAAC;QACC,OAAO,aAAa,CAAC,2BAA2B,CAAC,CAAA;IACrD,CAAC;IAED,OAAO,KAAK,CAAA;AAChB,CAAC"}
+{"version":3,"file":"qDeveloperProfiles.js","sourceRoot":"","sources":["../../../src/language-server/amazonQServiceManager/qDeveloperProfiles.ts"],"names":[],"mappings":";;;AAwHA,kFAaC;AArID,qFAKuD;AACvD,oCAA8D;AAC9D,+CAAiD;AAuBpC,QAAA,6BAA6B,GAAG,EAAE,CAAA;AAC/C,MAAM,iCAAiC,GAAG,EAAE,CAAA;AAErC,MAAM,kCAAkC,GAC3C,CAAC,OAAwE,EAAmC,EAAE,CAC9G,KAAK,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE;IAC7C,IAAI,CAAC,cAAc,IAAI,cAAc,KAAK,gBAAgB,EAAE,CAAC;QACzD,OAAO,CAAC,KAAK,CAAC,8EAA8E,CAAC,CAAA;QAC7F,OAAO,EAAE,CAAA;IACb,CAAC;IAED,IAAI,WAAW,GAA8B,EAAE,CAAA;IAC/C,MAAM,UAAU,GAAG,SAAS,IAAI,2BAAe,CAAA;IAE/C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CACnC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,EAAE;QAClD,MAAM,oBAAoB,GAAG,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QACtD,OAAO,uBAAuB,CAAC,oBAAoB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACzE,CAAC,CAAC,CACL,CAAA;IAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC,aAAa,CAAC,MAAM,KAAK,WAAW,CAAC,CAAA;IAE7F,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,gCAAa,CAAC,gCAAa,CAAC,aAAa,EAAE,sDAAsD,CAAC,CAAA;IAChH,CAAC;IAED,gBAAgB,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAA;IAEvF,OAAO,WAAW,CAAA;AACtB,CAAC,CAAA;AA3BQ,QAAA,kCAAkC,sCA2B1C;AAEL,KAAK,UAAU,uBAAuB,CAClC,OAAkC,EAClC,MAAc,EACd,OAAgB;IAEhB,IAAI,mBAAmB,GAA8B,EAAE,CAAA;IACvD,IAAI,SAAS,GAAuB,SAAS,CAAA;IAC7C,IAAI,aAAa,GAAG,CAAC,CAAA;IAErB,IAAI,CAAC;QACD,GAAG,CAAC;YACA,OAAO,CAAC,KAAK,CAAC,kCAAkC,MAAM,gBAAgB,aAAa,GAAG,CAAC,CAAA;YAEvF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,qBAAqB,CAAC;gBACjD,UAAU,EAAE,iCAAiC;aAChD,CAAC,CAAA;YAEF,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC/C,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,IAAI,EAAE,OAAO,CAAC,WAAW;gBACzB,eAAe,EAAE;oBACb,MAAM;iBACT;aACJ,CAAC,CAAC,CAAA;YAEH,mBAAmB,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAA;YAErC,OAAO,CAAC,KAAK,CAAC,yBAAyB,MAAM,KAAK,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,gBAAgB,aAAa,GAAG,CAAC,CAAA;YAC3G,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAA;YAC9B,aAAa,EAAE,CAAA;QACnB,CAAC,QAAQ,SAAS,KAAK,SAAS,IAAI,aAAa,GAAG,qCAA6B,EAAC;QAElF,OAAO,mBAAmB,CAAA;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,MAAM,KAAK,KAAK,EAAE,CAAC,CAAA;QAEjE,MAAM,KAAK,CAAA;IACf,CAAC;AACL,CAAC;AAED,MAAM,mBAAmB,GAAG,GAAG,CAAA;AAC/B,MAAM,2BAA2B,GAAG,mBAAmB,CAAA;AAEvD;;;;;;;;;;;;;;GAcG;AACH,SAAgB,mCAAmC,CAAC,qBAA+C;IAC/F,MAAM,aAAa,GAAG,qBAAqB,CAAC,qBAAqB,EAAE,CAAC,mBAAmB,CAAC,CAAA;IAExF,IACI,IAAA,gBAAQ,EAAC,aAAa,CAAC;QACvB,CAAC,CAAC,aAAa,YAAY,KAAK,CAAC;QACjC,2BAA2B,IAAI,aAAa;QAC5C,IAAA,cAAM,EAAC,aAAa,CAAC,2BAA2B,CAAC,CAAC,EACpD,CAAC;QACC,OAAO,aAAa,CAAC,2BAA2B,CAAC,CAAA;IACrD,CAAC;IAED,OAAO,KAAK,CAAA;AAChB,CAAC"}

package/out/language-server/chat/agents/docGeneration/constants.d.ts

@@ -0,0 +1,59 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { MynahIcons, Status } from '@aws/mynah-ui';
+import { FollowUpTypes, NewFileInfo } from '../shared/types';
+export declare const docChat = "docChat";
+export declare const docScheme = "aws-doc";
+export declare const featureName = "Amazon Q Doc Generation";
+export declare function getFileSummaryPercentage(input: string): number;
+export declare enum DocGenerationStep {
+    UPLOAD_TO_S3 = 0,
+    SUMMARIZING_FILES = 1,
+    GENERATING_ARTIFACTS = 2
+}
+export declare const docGenerationProgressMessage: (currentStep: DocGenerationStep, mode: Mode) => string;
+export declare const docGenerationSuccessMessage: (mode: Mode) => string;
+export declare const docRejectConfirmation = "Your changes have been discarded.";
+export declare const FolderSelectorFollowUps: {
+    icon: MynahIcons;
+    pillText: string;
+    prompt: string;
+    status: Status;
+    type: FollowUpTypes;
+}[];
+export declare const CodeChangeFollowUps: {
+    pillText: string;
+    prompt: string;
+    type: FollowUpTypes;
+    icon: MynahIcons;
+    status: Status;
+}[];
+export declare const NewSessionFollowUps: {
+    pillText: string;
+    type: FollowUpTypes;
+    status: Status;
+}[];
+export declare const SynchronizeDocumentation: {
+    pillText: string;
+    prompt: string;
+    type: FollowUpTypes;
+};
+export declare const EditDocumentation: {
+    pillText: string;
+    prompt: string;
+    type: FollowUpTypes;
+};
+export declare enum Mode {
+    NONE = "None",
+    CREATE = "Create",
+    SYNC = "Sync",
+    EDIT = "Edit"
+}
+/**
+ *
+ * @param paths file paths
+ * @returns the path to a README.md, or undefined if none exist
+ */
+export declare const findReadmePath: (paths?: NewFileInfo[]) => NewFileInfo | undefined;

package/out/language-server/chat/agents/docGeneration/constants.js

@@ -0,0 +1,147 @@
+"use strict";
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findReadmePath = exports.Mode = exports.EditDocumentation = exports.SynchronizeDocumentation = exports.NewSessionFollowUps = exports.CodeChangeFollowUps = exports.FolderSelectorFollowUps = exports.docRejectConfirmation = exports.docGenerationSuccessMessage = exports.docGenerationProgressMessage = exports.DocGenerationStep = exports.featureName = exports.docScheme = exports.docChat = void 0;
+exports.getFileSummaryPercentage = getFileSummaryPercentage;
+const types_1 = require("../shared/types");
+const i18n_1 = require("../shared/i18n");
+// For uniquely identifiying which chat messages should be routed to Doc
+exports.docChat = 'docChat';
+exports.docScheme = 'aws-doc';
+exports.featureName = 'Amazon Q Doc Generation';
+function getFileSummaryPercentage(input) {
+    // Split the input string by newline characters
+    const lines = input.split('\n');
+    // Find the line containing "summarized:"
+    const summaryLine = lines.find(line => line.includes('summarized:'));
+    // If the line is not found, return null
+    if (!summaryLine) {
+        return -1;
+    }
+    // Extract the numbers from the summary line
+    const [summarized, total] = summaryLine.split(':')[1].trim().split(' of ').map(Number);
+    // Calculate the percentage
+    const percentage = (summarized / total) * 100;
+    return percentage;
+}
+const checkIcons = {
+    wait: '☐',
+    current: '☐',
+    done: '☑',
+};
+const getIconForStep = (targetStep, currentStep) => {
+    return currentStep === targetStep
+        ? checkIcons.current
+        : currentStep > targetStep
+            ? checkIcons.done
+            : checkIcons.wait;
+};
+var DocGenerationStep;
+(function (DocGenerationStep) {
+    DocGenerationStep[DocGenerationStep["UPLOAD_TO_S3"] = 0] = "UPLOAD_TO_S3";
+    DocGenerationStep[DocGenerationStep["SUMMARIZING_FILES"] = 1] = "SUMMARIZING_FILES";
+    DocGenerationStep[DocGenerationStep["GENERATING_ARTIFACTS"] = 2] = "GENERATING_ARTIFACTS";
+})(DocGenerationStep || (exports.DocGenerationStep = DocGenerationStep = {}));
+const docGenerationProgressMessage = (currentStep, mode) => `
+${mode === Mode.CREATE ? (0, i18n_1.i18n)('AWS.amazonq.doc.answer.creating') : (0, i18n_1.i18n)('AWS.amazonq.doc.answer.updating')}
+
+${getIconForStep(DocGenerationStep.UPLOAD_TO_S3, currentStep)} ${(0, i18n_1.i18n)('AWS.amazonq.doc.answer.scanning')}
+
+${getIconForStep(DocGenerationStep.SUMMARIZING_FILES, currentStep)} ${(0, i18n_1.i18n)('AWS.amazonq.doc.answer.summarizing')}
+
+${getIconForStep(DocGenerationStep.GENERATING_ARTIFACTS, currentStep)} ${(0, i18n_1.i18n)('AWS.amazonq.doc.answer.generating')}
+
+
+`;
+exports.docGenerationProgressMessage = docGenerationProgressMessage;
+const docGenerationSuccessMessage = (mode) => mode === Mode.CREATE ? (0, i18n_1.i18n)('AWS.amazonq.doc.answer.readmeCreated') : (0, i18n_1.i18n)('AWS.amazonq.doc.answer.readmeUpdated');
+exports.docGenerationSuccessMessage = docGenerationSuccessMessage;
+exports.docRejectConfirmation = 'Your changes have been discarded.';
+exports.FolderSelectorFollowUps = [
+    {
+        icon: 'ok',
+        pillText: 'Yes',
+        prompt: 'Yes',
+        status: 'success',
+        type: types_1.FollowUpTypes.ProceedFolderSelection,
+    },
+    {
+        icon: 'refresh',
+        pillText: 'Change folder',
+        prompt: 'Change folder',
+        status: 'info',
+        type: types_1.FollowUpTypes.ChooseFolder,
+    },
+    {
+        icon: 'cancel',
+        pillText: 'Cancel',
+        prompt: 'Cancel',
+        status: 'error',
+        type: types_1.FollowUpTypes.CancelFolderSelection,
+    },
+];
+exports.CodeChangeFollowUps = [
+    {
+        pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.accept'),
+        prompt: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.accept'),
+        type: types_1.FollowUpTypes.AcceptChanges,
+        icon: 'ok',
+        status: 'success',
+    },
+    {
+        pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.makeChanges'),
+        prompt: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.makeChanges'),
+        type: types_1.FollowUpTypes.MakeChanges,
+        icon: 'refresh',
+        status: 'info',
+    },
+    {
+        pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.reject'),
+        prompt: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.reject'),
+        type: types_1.FollowUpTypes.RejectChanges,
+        icon: 'cancel',
+        status: 'error',
+    },
+];
+exports.NewSessionFollowUps = [
+    {
+        pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.newTask'),
+        type: types_1.FollowUpTypes.NewTask,
+        status: 'info',
+    },
+    {
+        pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.closeSession'),
+        type: types_1.FollowUpTypes.CloseSession,
+        status: 'info',
+    },
+];
+exports.SynchronizeDocumentation = {
+    pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.update'),
+    prompt: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.update'),
+    type: types_1.FollowUpTypes.SynchronizeDocumentation,
+};
+exports.EditDocumentation = {
+    pillText: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.makeChange'),
+    prompt: (0, i18n_1.i18n)('AWS.amazonq.doc.pillText.makeChange'),
+    type: types_1.FollowUpTypes.EditDocumentation,
+};
+var Mode;
+(function (Mode) {
+    Mode["NONE"] = "None";
+    Mode["CREATE"] = "Create";
+    Mode["SYNC"] = "Sync";
+    Mode["EDIT"] = "Edit";
+})(Mode || (exports.Mode = Mode = {}));
+/**
+ *
+ * @param paths file paths
+ * @returns the path to a README.md, or undefined if none exist
+ */
+const findReadmePath = (paths) => {
+    return paths?.find(path => /readme\.md$/i.test(path.relativePath));
+};
+exports.findReadmePath = findReadmePath;
+//# sourceMappingURL=constants.js.map

package/out/language-server/chat/agents/docGeneration/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../../src/language-server/chat/agents/docGeneration/constants.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAaH,4DAmBC;AA7BD,2CAA4D;AAC5D,yCAAqC;AAErC,wEAAwE;AAC3D,QAAA,OAAO,GAAG,SAAS,CAAA;AAEnB,QAAA,SAAS,GAAG,SAAS,CAAA;AAErB,QAAA,WAAW,GAAG,yBAAyB,CAAA;AAEpD,SAAgB,wBAAwB,CAAC,KAAa;IAClD,+CAA+C;IAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IAE/B,yCAAyC;IACzC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAA;IAEpE,wCAAwC;IACxC,IAAI,CAAC,WAAW,EAAE,CAAC;QACf,OAAO,CAAC,CAAC,CAAA;IACb,CAAC;IAED,4CAA4C;IAC5C,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;IAEtF,2BAA2B;IAC3B,MAAM,UAAU,GAAG,CAAC,UAAU,GAAG,KAAK,CAAC,GAAG,GAAG,CAAA;IAE7C,OAAO,UAAU,CAAA;AACrB,CAAC;AAED,MAAM,UAAU,GAAG;IACf,IAAI,EAAE,SAAS;IACf,OAAO,EAAE,SAAS;IAClB,IAAI,EAAE,SAAS;CAClB,CAAA;AAED,MAAM,cAAc,GAAG,CAAC,UAAkB,EAAE,WAAmB,EAAE,EAAE;IAC/D,OAAO,WAAW,KAAK,UAAU;QAC7B,CAAC,CAAC,UAAU,CAAC,OAAO;QACpB,CAAC,CAAC,WAAW,GAAG,UAAU;YACxB,CAAC,CAAC,UAAU,CAAC,IAAI;YACjB,CAAC,CAAC,UAAU,CAAC,IAAI,CAAA;AAC3B,CAAC,CAAA;AAED,IAAY,iBAIX;AAJD,WAAY,iBAAiB;IACzB,yEAAY,CAAA;IACZ,mFAAiB,CAAA;IACjB,yFAAoB,CAAA;AACxB,CAAC,EAJW,iBAAiB,iCAAjB,iBAAiB,QAI5B;AAEM,MAAM,4BAA4B,GAAG,CAAC,WAA8B,EAAE,IAAU,EAAE,EAAE,CAAC;EAC1F,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAA,WAAI,EAAC,iCAAiC,CAAC,CAAC,CAAC,CAAC,IAAA,WAAI,EAAC,iCAAiC,CAAC;;EAExG,cAAc,CAAC,iBAAiB,CAAC,YAAY,EAAE,WAAW,CAAC,IAAI,IAAA,WAAI,EAAC,iCAAiC,CAAC;;EAEtG,cAAc,CAAC,iBAAiB,CAAC,iBAAiB,EAAE,WAAW,CAAC,IAAI,IAAA,WAAI,EAAC,oCAAoC,CAAC;;EAE9G,cAAc,CAAC,iBAAiB,CAAC,oBAAoB,EAAE,WAAW,CAAC,IAAI,IAAA,WAAI,EAAC,mCAAmC,CAAC;;;CAGjH,CAAA;AAVY,QAAA,4BAA4B,gCAUxC;AAEM,MAAM,2BAA2B,GAAG,CAAC,IAAU,EAAE,EAAE,CACtD,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAA,WAAI,EAAC,sCAAsC,CAAC,CAAC,CAAC,CAAC,IAAA,WAAI,EAAC,sCAAsC,CAAC,CAAA;AADzG,QAAA,2BAA2B,+BAC8E;AAEzG,QAAA,qBAAqB,GAAG,mCAAmC,CAAA;AAE3D,QAAA,uBAAuB,GAAG;IACnC;QACI,IAAI,EAAE,IAAkB;QACxB,QAAQ,EAAE,KAAK;QACf,MAAM,EAAE,KAAK;QACb,MAAM,EAAE,SAAmB;QAC3B,IAAI,EAAE,qBAAa,CAAC,sBAAsB;KAC7C;IACD;QACI,IAAI,EAAE,SAAuB;QAC7B,QAAQ,EAAE,eAAe;QACzB,MAAM,EAAE,eAAe;QACvB,MAAM,EAAE,MAAgB;QACxB,IAAI,EAAE,qBAAa,CAAC,YAAY;KACnC;IACD;QACI,IAAI,EAAE,QAAsB;QAC5B,QAAQ,EAAE,QAAQ;QAClB,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE,OAAiB;QACzB,IAAI,EAAE,qBAAa,CAAC,qBAAqB;KAC5C;CACJ,CAAA;AAEY,QAAA,mBAAmB,GAAG;IAC/B;QACI,QAAQ,EAAE,IAAA,WAAI,EAAC,iCAAiC,CAAC;QACjD,MAAM,EAAE,IAAA,WAAI,EAAC,iCAAiC,CAAC;QAC/C,IAAI,EAAE,qBAAa,CAAC,aAAa;QACjC,IAAI,EAAE,IAAkB;QACxB,MAAM,EAAE,SAAmB;KAC9B;IACD;QACI,QAAQ,EAAE,IAAA,WAAI,EAAC,sCAAsC,CAAC;QACtD,MAAM,EAAE,IAAA,WAAI,EAAC,sCAAsC,CAAC;QACpD,IAAI,EAAE,qBAAa,CAAC,WAAW;QAC/B,IAAI,EAAE,SAAuB;QAC7B,MAAM,EAAE,MAAgB;KAC3B;IACD;QACI,QAAQ,EAAE,IAAA,WAAI,EAAC,iCAAiC,CAAC;QACjD,MAAM,EAAE,IAAA,WAAI,EAAC,iCAAiC,CAAC;QAC/C,IAAI,EAAE,qBAAa,CAAC,aAAa;QACjC,IAAI,EAAE,QAAsB;QAC5B,MAAM,EAAE,OAAiB;KAC5B;CACJ,CAAA;AAEY,QAAA,mBAAmB,GAAG;IAC/B;QACI,QAAQ,EAAE,IAAA,WAAI,EAAC,kCAAkC,CAAC;QAClD,IAAI,EAAE,qBAAa,CAAC,OAAO;QAC3B,MAAM,EAAE,MAAgB;KAC3B;IACD;QACI,QAAQ,EAAE,IAAA,WAAI,EAAC,uCAAuC,CAAC;QACvD,IAAI,EAAE,qBAAa,CAAC,YAAY;QAChC,MAAM,EAAE,MAAgB;KAC3B;CACJ,CAAA;AAEY,QAAA,wBAAwB,GAAG;IACpC,QAAQ,EAAE,IAAA,WAAI,EAAC,iCAAiC,CAAC;IACjD,MAAM,EAAE,IAAA,WAAI,EAAC,iCAAiC,CAAC;IAC/C,IAAI,EAAE,qBAAa,CAAC,wBAAwB;CAC/C,CAAA;AAEY,QAAA,iBAAiB,GAAG;IAC7B,QAAQ,EAAE,IAAA,WAAI,EAAC,qCAAqC,CAAC;IACrD,MAAM,EAAE,IAAA,WAAI,EAAC,qCAAqC,CAAC;IACnD,IAAI,EAAE,qBAAa,CAAC,iBAAiB;CACxC,CAAA;AAED,IAAY,IAKX;AALD,WAAY,IAAI;IACZ,qBAAa,CAAA;IACb,yBAAiB,CAAA;IACjB,qBAAa,CAAA;IACb,qBAAa,CAAA;AACjB,CAAC,EALW,IAAI,oBAAJ,IAAI,QAKf;AAED;;;;GAIG;AACI,MAAM,cAAc,GAAG,CAAC,KAAqB,EAAE,EAAE;IACpD,OAAO,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAA;AACtE,CAAC,CAAA;AAFY,QAAA,cAAc,kBAE1B"}

package/out/language-server/chat/agents/docGeneration/errors.d.ts

@@ -0,0 +1,29 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare class DocClientError {
+    remainingIterations?: number;
+    constructor(message: string, code: string, remainingIterations?: number);
+}
+export declare class ReadmeTooLargeError extends DocClientError {
+    constructor();
+}
+export declare class ReadmeUpdateTooLargeError extends DocClientError {
+    constructor(remainingIterations: number);
+}
+export declare class WorkspaceEmptyError extends DocClientError {
+    constructor();
+}
+export declare class NoChangeRequiredException extends DocClientError {
+    constructor();
+}
+export declare class PromptRefusalException extends DocClientError {
+    constructor(remainingIterations: number);
+}
+export declare class PromptTooVagueError extends DocClientError {
+    constructor(remainingIterations: number);
+}
+export declare class PromptUnrelatedError extends DocClientError {
+    constructor(remainingIterations: number);
+}

package/out/language-server/chat/agents/docGeneration/errors.js

@@ -0,0 +1,67 @@
+"use strict";
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PromptUnrelatedError = exports.PromptTooVagueError = exports.PromptRefusalException = exports.NoChangeRequiredException = exports.WorkspaceEmptyError = exports.ReadmeUpdateTooLargeError = exports.ReadmeTooLargeError = exports.DocClientError = void 0;
+//TODO: extend the toolkit error once ready
+// import { ClientError, ContentLengthError as CommonContentLengthError } from '../shared/errors'
+const i18n_1 = require("../shared/i18n");
+class DocClientError {
+    remainingIterations;
+    constructor(message, code, remainingIterations) {
+        // super(message, { code })
+        this.remainingIterations = remainingIterations;
+    }
+}
+exports.DocClientError = DocClientError;
+class ReadmeTooLargeError extends DocClientError {
+    constructor() {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.readmeTooLarge'), ReadmeTooLargeError.name);
+    }
+}
+exports.ReadmeTooLargeError = ReadmeTooLargeError;
+class ReadmeUpdateTooLargeError extends DocClientError {
+    constructor(remainingIterations) {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.readmeUpdateTooLarge'), ReadmeUpdateTooLargeError.name, remainingIterations);
+    }
+}
+exports.ReadmeUpdateTooLargeError = ReadmeUpdateTooLargeError;
+class WorkspaceEmptyError extends DocClientError {
+    constructor() {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.workspaceEmpty'), WorkspaceEmptyError.name);
+    }
+}
+exports.WorkspaceEmptyError = WorkspaceEmptyError;
+class NoChangeRequiredException extends DocClientError {
+    constructor() {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.noChangeRequiredException'), NoChangeRequiredException.name);
+    }
+}
+exports.NoChangeRequiredException = NoChangeRequiredException;
+class PromptRefusalException extends DocClientError {
+    constructor(remainingIterations) {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.promptRefusal'), PromptRefusalException.name, remainingIterations);
+    }
+}
+exports.PromptRefusalException = PromptRefusalException;
+// TODO: move the common content length error to shared folder
+// export class ContentLengthError extends CommonContentLengthError {
+//     constructor() {
+//         super(i18n('AWS.amazonq.doc.error.contentLengthError'), { code: ContentLengthError.name })
+//     }
+// }
+class PromptTooVagueError extends DocClientError {
+    constructor(remainingIterations) {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.promptTooVague'), PromptTooVagueError.name, remainingIterations);
+    }
+}
+exports.PromptTooVagueError = PromptTooVagueError;
+class PromptUnrelatedError extends DocClientError {
+    constructor(remainingIterations) {
+        super((0, i18n_1.i18n)('AWS.amazonq.doc.error.promptUnrelated'), PromptUnrelatedError.name, remainingIterations);
+    }
+}
+exports.PromptUnrelatedError = PromptUnrelatedError;
+//# sourceMappingURL=errors.js.map

package/out/language-server/chat/agents/docGeneration/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../../../src/language-server/chat/agents/docGeneration/errors.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,2CAA2C;AAC3C,iGAAiG;AACjG,yCAAqC;AAErC,MAAa,cAAc;IACvB,mBAAmB,CAAS;IAC5B,YAAY,OAAe,EAAE,IAAY,EAAE,mBAA4B;QACnE,2BAA2B;QAC3B,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAA;IAClD,CAAC;CACJ;AAND,wCAMC;AAED,MAAa,mBAAoB,SAAQ,cAAc;IACnD;QACI,KAAK,CAAC,IAAA,WAAI,EAAC,sCAAsC,CAAC,EAAE,mBAAmB,CAAC,IAAI,CAAC,CAAA;IACjF,CAAC;CACJ;AAJD,kDAIC;AAED,MAAa,yBAA0B,SAAQ,cAAc;IACzD,YAAY,mBAA2B;QACnC,KAAK,CAAC,IAAA,WAAI,EAAC,4CAA4C,CAAC,EAAE,yBAAyB,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAA;IAClH,CAAC;CACJ;AAJD,8DAIC;AAED,MAAa,mBAAoB,SAAQ,cAAc;IACnD;QACI,KAAK,CAAC,IAAA,WAAI,EAAC,sCAAsC,CAAC,EAAE,mBAAmB,CAAC,IAAI,CAAC,CAAA;IACjF,CAAC;CACJ;AAJD,kDAIC;AAED,MAAa,yBAA0B,SAAQ,cAAc;IACzD;QACI,KAAK,CAAC,IAAA,WAAI,EAAC,iDAAiD,CAAC,EAAE,yBAAyB,CAAC,IAAI,CAAC,CAAA;IAClG,CAAC;CACJ;AAJD,8DAIC;AAED,MAAa,sBAAuB,SAAQ,cAAc;IACtD,YAAY,mBAA2B;QACnC,KAAK,CAAC,IAAA,WAAI,EAAC,qCAAqC,CAAC,EAAE,sBAAsB,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAA;IACxG,CAAC;CACJ;AAJD,wDAIC;AAED,8DAA8D;AAC9D,qEAAqE;AACrE,sBAAsB;AACtB,qGAAqG;AACrG,QAAQ;AACR,IAAI;AACJ,MAAa,mBAAoB,SAAQ,cAAc;IACnD,YAAY,mBAA2B;QACnC,KAAK,CAAC,IAAA,WAAI,EAAC,sCAAsC,CAAC,EAAE,mBAAmB,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAA;IACtG,CAAC;CACJ;AAJD,kDAIC;AAED,MAAa,oBAAqB,SAAQ,cAAc;IACpD,YAAY,mBAA2B;QACnC,KAAK,CAAC,IAAA,WAAI,EAAC,uCAAuC,CAAC,EAAE,oBAAoB,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAA;IACxG,CAAC;CACJ;AAJD,oDAIC"}

package/out/language-server/chat/agents/shared/i18n.d.ts

@@ -0,0 +1 @@
+export declare function i18n(text: string): string;

package/out/language-server/chat/agents/shared/i18n.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.i18n = i18n;
+function i18n(text) {
+    return text;
+}
+//# sourceMappingURL=i18n.js.map

package/out/language-server/chat/agents/shared/i18n.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"i18n.js","sourceRoot":"","sources":["../../../../../src/language-server/chat/agents/shared/i18n.ts"],"names":[],"mappings":";;AAAA,oBAEC;AAFD,SAAgB,IAAI,CAAC,IAAY;IAC7B,OAAO,IAAI,CAAA;AACf,CAAC"}

package/out/language-server/chat/agents/shared/types.d.ts

@@ -0,0 +1,55 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { WorkspaceFolder } from '@aws/language-server-runtimes/protocol';
+export declare enum FollowUpTypes {
+    ViewDiff = "ViewDiff",
+    AcceptCode = "AcceptCode",
+    RejectCode = "RejectCode",
+    BuildAndExecute = "BuildAndExecute",
+    ModifyCommands = "ModifyCommands",
+    SkipBuildAndFinish = "SkipBuildAndFinish",
+    InstallDependenciesAndContinue = "InstallDependenciesAndContinue",
+    ContinueBuildAndExecute = "ContinueBuildAndExecute",
+    ViewCodeDiffAfterIteration = "ViewCodeDiffAfterIteration",
+    GenerateCode = "GenerateCode",
+    InsertCode = "InsertCode",
+    ProvideFeedbackAndRegenerateCode = "ProvideFeedbackAndRegenerateCode",
+    Retry = "Retry",
+    ModifyDefaultSourceFolder = "ModifyDefaultSourceFolder",
+    DevExamples = "DevExamples",
+    NewTask = "NewTask",
+    CloseSession = "CloseSession",
+    SendFeedback = "SendFeedback",
+    AcceptAutoBuild = "AcceptAutoBuild",
+    DenyAutoBuild = "DenyAutoBuild",
+    GenerateDevFile = "GenerateDevFile",
+    CreateDocumentation = "CreateDocumentation",
+    ChooseFolder = "ChooseFolder",
+    UpdateDocumentation = "UpdateDocumentation",
+    SynchronizeDocumentation = "SynchronizeDocumentation",
+    EditDocumentation = "EditDocumentation",
+    AcceptChanges = "AcceptChanges",
+    RejectChanges = "RejectChanges",
+    MakeChanges = "MakeChanges",
+    ProceedFolderSelection = "ProceedFolderSelection",
+    CancelFolderSelection = "CancelFolderSelection"
+}
+export type DiffTreeFileInfo = {
+    zipFilePath: string;
+    relativePath: string;
+    rejected: boolean;
+    changeApplied: boolean;
+};
+export type NewFileZipContents = {
+    zipFilePath: string;
+    fileContent: string;
+};
+export type DeletedFileInfo = DiffTreeFileInfo & {
+    workspaceFolder: WorkspaceFolder;
+};
+export type NewFileInfo = DiffTreeFileInfo & NewFileZipContents & {
+    virtualMemoryUri: string;
+    workspaceFolder: WorkspaceFolder;
+};

package/out/language-server/chat/agents/shared/types.js

@@ -0,0 +1,46 @@
+"use strict";
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FollowUpTypes = void 0;
+//TODO: May not be needed follow up with flare
+var FollowUpTypes;
+(function (FollowUpTypes) {
+    // UnitTestGeneration
+    FollowUpTypes["ViewDiff"] = "ViewDiff";
+    FollowUpTypes["AcceptCode"] = "AcceptCode";
+    FollowUpTypes["RejectCode"] = "RejectCode";
+    FollowUpTypes["BuildAndExecute"] = "BuildAndExecute";
+    FollowUpTypes["ModifyCommands"] = "ModifyCommands";
+    FollowUpTypes["SkipBuildAndFinish"] = "SkipBuildAndFinish";
+    FollowUpTypes["InstallDependenciesAndContinue"] = "InstallDependenciesAndContinue";
+    FollowUpTypes["ContinueBuildAndExecute"] = "ContinueBuildAndExecute";
+    FollowUpTypes["ViewCodeDiffAfterIteration"] = "ViewCodeDiffAfterIteration";
+    // FeatureDev
+    FollowUpTypes["GenerateCode"] = "GenerateCode";
+    FollowUpTypes["InsertCode"] = "InsertCode";
+    FollowUpTypes["ProvideFeedbackAndRegenerateCode"] = "ProvideFeedbackAndRegenerateCode";
+    FollowUpTypes["Retry"] = "Retry";
+    FollowUpTypes["ModifyDefaultSourceFolder"] = "ModifyDefaultSourceFolder";
+    FollowUpTypes["DevExamples"] = "DevExamples";
+    FollowUpTypes["NewTask"] = "NewTask";
+    FollowUpTypes["CloseSession"] = "CloseSession";
+    FollowUpTypes["SendFeedback"] = "SendFeedback";
+    FollowUpTypes["AcceptAutoBuild"] = "AcceptAutoBuild";
+    FollowUpTypes["DenyAutoBuild"] = "DenyAutoBuild";
+    FollowUpTypes["GenerateDevFile"] = "GenerateDevFile";
+    // Doc
+    FollowUpTypes["CreateDocumentation"] = "CreateDocumentation";
+    FollowUpTypes["ChooseFolder"] = "ChooseFolder";
+    FollowUpTypes["UpdateDocumentation"] = "UpdateDocumentation";
+    FollowUpTypes["SynchronizeDocumentation"] = "SynchronizeDocumentation";
+    FollowUpTypes["EditDocumentation"] = "EditDocumentation";
+    FollowUpTypes["AcceptChanges"] = "AcceptChanges";
+    FollowUpTypes["RejectChanges"] = "RejectChanges";
+    FollowUpTypes["MakeChanges"] = "MakeChanges";
+    FollowUpTypes["ProceedFolderSelection"] = "ProceedFolderSelection";
+    FollowUpTypes["CancelFolderSelection"] = "CancelFolderSelection";
+})(FollowUpTypes || (exports.FollowUpTypes = FollowUpTypes = {}));
+//# sourceMappingURL=types.js.map

package/out/language-server/chat/agents/shared/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/language-server/chat/agents/shared/types.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAIH,8CAA8C;AAC9C,IAAY,aAmCX;AAnCD,WAAY,aAAa;IACrB,qBAAqB;IACrB,sCAAqB,CAAA;IACrB,0CAAyB,CAAA;IACzB,0CAAyB,CAAA;IACzB,oDAAmC,CAAA;IACnC,kDAAiC,CAAA;IACjC,0DAAyC,CAAA;IACzC,kFAAiE,CAAA;IACjE,oEAAmD,CAAA;IACnD,0EAAyD,CAAA;IACzD,aAAa;IACb,8CAA6B,CAAA;IAC7B,0CAAyB,CAAA;IACzB,sFAAqE,CAAA;IACrE,gCAAe,CAAA;IACf,wEAAuD,CAAA;IACvD,4CAA2B,CAAA;IAC3B,oCAAmB,CAAA;IACnB,8CAA6B,CAAA;IAC7B,8CAA6B,CAAA;IAC7B,oDAAmC,CAAA;IACnC,gDAA+B,CAAA;IAC/B,oDAAmC,CAAA;IACnC,MAAM;IACN,4DAA2C,CAAA;IAC3C,8CAA6B,CAAA;IAC7B,4DAA2C,CAAA;IAC3C,sEAAqD,CAAA;IACrD,wDAAuC,CAAA;IACvC,gDAA+B,CAAA;IAC/B,gDAA+B,CAAA;IAC/B,4CAA2B,CAAA;IAC3B,kEAAiD,CAAA;IACjD,gEAA+C,CAAA;AACnD,CAAC,EAnCW,aAAa,6BAAb,aAAa,QAmCxB"}

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

@@ -3,7 +3,7 @@ import { CancellationToken, Chat, ChatParams, ChatResult, EndChatParams, QuickAc
 import { Features, LspHandlers } from '../types';
 import { ChatSessionManagementService } from './chatSessionManagementService';
 import { TelemetryService } from '../telemetryService';
-type ChatHandlers = Omit<LspHandlers<Chat>, 'openTab'>;
+type ChatHandlers = Omit<LspHandlers<Chat>, 'openTab' | 'sendChatUpdate' | 'onFileClicked'>;
 export declare class ChatController implements ChatHandlers {
     #private;
     constructor(chatSessionManagementService: ChatSessionManagementService, features: Features, telemetryService: TelemetryService);

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

@@ -15,6 +15,8 @@ const triggerContext_1 = require("./contexts/triggerContext");
 const constants_1 = require("./constants");
 const qConfigurationServer_1 = require("../configuration/qConfigurationServer");
 const lsp_core_1 = require("@aws/lsp-core");
+const errors_1 = require("../amazonQServiceManager/errors");
+const AmazonQTokenServiceManager_1 = require("../amazonQServiceManager/AmazonQTokenServiceManager");
 class ChatController {
     #features;
     #chatSessionManagementService;
@@ -57,7 +59,8 @@ class ChatController {
         const conversationIdentifier = session?.conversationId ?? 'New conversation';
         try {
             this.#log('Request for conversation id:', conversationIdentifier);
-            requestInput = this.#triggerContext.getChatParamsFromTrigger(params, triggerContext, this.#customizationArn);
+            const profileArn = AmazonQTokenServiceManager_1.AmazonQTokenServiceManager.getInstance(this.#features).getActiveProfileArn();
+            requestInput = this.#triggerContext.getChatParamsFromTrigger(params, triggerContext, this.#customizationArn, profileArn);
             metric.recordStart();
             response = await session.sendMessage(requestInput);
             this.#log('Response for conversation id:', conversationIdentifier, JSON.stringify(response.$metadata));
@@ -67,6 +70,19 @@ class ChatController {
                 metric.setDimension('cwsprChatRepsonseCode', err.statusCode ?? 400);
                 this.#telemetryController.emitMessageResponseError(params.tabId, metric.metric);
             }
+            if (err instanceof errors_1.AmazonQServicePendingSigninError) {
+                this.#log(`Q Chat SSO Connection error: ${(0, utils_2.getErrorMessage)(err)}`);
+                return (0, utils_1.createAuthFollowUpResult)('full-auth');
+            }
+            if (err instanceof errors_1.AmazonQServicePendingProfileError) {
+                this.#log(`Q Chat SSO Connection error: ${(0, utils_2.getErrorMessage)(err)}`);
+                const followUpResult = (0, utils_1.createAuthFollowUpResult)('use-supported-auth');
+                // Access first element in array
+                if (followUpResult.followUp?.options) {
+                    followUpResult.followUp.options[0].pillText = 'Select Q Developer Profile';
+                }
+                return followUpResult;
+            }
             const authFollowType = (0, utils_1.getAuthFollowUpType)(err);
             if (authFollowType) {
                 this.#log(`Q auth error: ${(0, utils_2.getErrorMessage)(err)}`);