replacing customerFacingMessage with UiMessage

Author: laileni-aws

packages/core/src/amazonqTest/chat/controller/controller.ts

@@ -242,7 +242,7 @@ export class TestController {
         // eslint-disable-next-line unicorn/no-null
         this.messenger.sendUpdatePromptProgress(data.tabID, null)
         const session = this.sessionStorage.getSession()
-        const isCancel = data.error.customerFacingMessage === unitTestGenerationCancelMessage
+        const isCancel = data.error.uiMessage === unitTestGenerationCancelMessage
         telemetry.amazonq_utgGenerateTests.emit({
             cwsprChatProgrammingLanguage: session.fileLanguage ?? 'plaintext',
             jobId: session.listOfTestGenerationJobId[0], // For RIV, UTG does only one StartTestGeneration API call
@@ -255,17 +255,18 @@ export class TestController {
             artifactsUploadDuration: session.artifactsUploadDuration,
             perfClientLatency: performance.now() - session.testGenerationStartTime,
             result: isCancel ? 'Cancelled' : 'Failed',
+            reason: data.error.code,
             reasonDesc: getTelemetryReasonDesc(data.error),
             isSupportedLanguage: true,
             credentialStartUrl: AuthUtil.instance.startUrl,
             httpStatusCode: data.error.statusCode ?? 0, // If status code is 0,  need to investigate where this is originating from.
         })
         if (session.stopIteration) {
             // Error from Science
-            this.messenger.sendMessage(data.error.customerFacingMessage.replaceAll('```', ''), data.tabID, 'answer')
+            this.messenger.sendMessage(data.error.uiMessage.replaceAll('```', ''), data.tabID, 'answer')
         } else {
             isCancel
-                ? this.messenger.sendMessage(data.error.customerFacingMessage, data.tabID, 'answer')
+                ? this.messenger.sendMessage(data.error.uiMessage, data.tabID, 'answer')
                 : this.sendErrorMessage(data)
         }
         await this.sessionCleanUp()
@@ -274,7 +275,7 @@ export class TestController {
     // Client side error messages
     private sendErrorMessage(data: {
         tabID: string
-        error: { customerFacingMessage: string; message: string; code: string; statusCode: string }
+        error: { uiMessage: string; message: string; code: string; statusCode: string }
     }) {
         const { error, tabID } = data
 
@@ -308,7 +309,7 @@ export class TestController {
             }
         } else {
             // other unexpected errors (TODO enumerate all other failure cases)
-            getLogger().error('Other error message: %s', error.customerFacingMessage)
+            getLogger().error('Other error message: %s', error.uiMessage)
             this.messenger.sendErrorMessage('', tabID)
         }
     }
@@ -720,12 +721,7 @@ export class TestController {
         // this.messenger.sendMessage('Accepted', message.tabID, 'prompt')
         telemetry.ui_click.emit({ elementId: 'unitTestGeneration_acceptDiff' })
         getLogger().info(
-            session.fileLanguage ?? 'plaintext',
-            session.listOfTestGenerationJobId[0],
-            session.testGenerationJobGroupName,
-            'Succeeded',
-            AuthUtil.instance.startUrl,
-            '200'
+            `Generated unit tests are accepted for ${session.fileLanguage ?? 'plaintext'} language with jobId: ${session.listOfTestGenerationJobId[0]}, jobGroupName: ${session.testGenerationJobGroupName}, result: Succeeded`
         )
         telemetry.amazonq_utgGenerateTests.emit({
             generatedCount: session.numberOfTestsGenerated,

packages/core/src/amazonqTest/error.ts

@@ -12,7 +12,7 @@ export class TestGenError extends ToolkitError {
         error: string,
         code: string,
         public statusCode: string,
-        public customerFacingMessage: string
+        public uiMessage: string
     ) {
         super(error, { code })
     }