backend error handling

Author: chungjac

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

@@ -284,7 +284,10 @@ export class TestController {
         this.messenger.sendUpdatePromptProgress(data.tabID, null)
         const session = this.sessionStorage.getSession()
         const isCancel = data.error.uiMessage === unitTestGenerationCancelMessage
-        const telemetryErrorMessage = getTelemetryReasonDesc(data.error.uiMessage.replaceAll('```', ''))
+        let telemetryErrorMessage = getTelemetryReasonDesc(data.error)
+        if (session.stopIteration) {
+            telemetryErrorMessage = getTelemetryReasonDesc(data.error.uiMessage.replaceAll('```', ''))
+        }
         TelemetryHelper.instance.sendTestGenerationToolkitEvent(
             session,
             session.isSupportedLanguage,
@@ -306,13 +309,26 @@ export class TestController {
             undefined,
             isCancel ? 'CANCELLED' : 'FAILED'
         )
-        this.messenger.sendMessage(
-            data.error.uiMessage.replaceAll('```', ''),
-            data.tabID,
-            'answer',
-            'testGenErrorMessage',
-            this.getFeedbackButtons()
-        )
+        if (session.stopIteration) {
+            // Error from Science
+            this.messenger.sendMessage(
+                data.error.uiMessage.replaceAll('```', ''),
+                data.tabID,
+                'answer',
+                'testGenErrorMessage',
+                this.getFeedbackButtons()
+            )
+        } else {
+            isCancel
+                ? this.messenger.sendMessage(
+                      data.error.uiMessage,
+                      data.tabID,
+                      'answer',
+                      'testGenErrorMessage',
+                      this.getFeedbackButtons()
+                  )
+                : this.sendErrorMessage(data)
+        }
         await this.sessionCleanUp()
         return
     }
@@ -1382,6 +1398,7 @@ export class TestController {
         session.sourceFilePath = ''
         session.generatedFilePath = ''
         session.projectRootPath = ''
+        session.stopIteration = false
         session.fileLanguage = undefined
         ChatSessionManager.Instance.setIsInProgress(false)
         session.linesOfCodeGenerated = 0

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

@@ -42,6 +42,7 @@ export class Session {
     public generatedFilePath: string = ''
     public projectRootPath: string = ''
     public fileLanguage: string | undefined = 'plaintext'
+    public stopIteration: boolean = false
     public targetFileInfo: TargetFileInfo | undefined
     public jobSummary: string = ''
 

packages/core/src/codewhisperer/service/testGenHandler.ts

@@ -199,6 +199,7 @@ export async function pollTestJobStatus(
             session.numberOfTestsGenerated = 0
             logger.verbose(`Test generation failed.`)
             if (resp.testGenerationJob?.jobStatusReason) {
+                session.stopIteration = true
                 throw new TestGenFailedError(resp.testGenerationJob?.jobStatusReason)
             } else {
                 throw new TestGenFailedError()