fix(amazonqFeatureDev): allow follow up options after finally only

Author: tverney

packages/core/package.nls.json

@@ -296,7 +296,8 @@
     "AWS.amazonq.featureDev.pillText.generatingCode": "Generating code...",
     "AWS.amazonq.featureDev.pillText.requestingChanges": "Requesting changes ...",
     "AWS.amazonq.featureDev.pillText.insertCode": "Accept code",
-    "AWS.amazonq.featureDev.pillText.stoppedCodeGeneration": "You've stopped the code generation.",
+    "AWS.amazonq.featureDev.pillText.stoppingCodeGeneration": "Stopping code generation ...",
+    "AWS.amazonq.featureDev.pillText.stoppedCodeGeneration": "The code generation has stopped.",
     "AWS.amazonq.featureDev.pillText.sendFeedback": "Send feedback",
     "AWS.amazonq.featureDev.pillText.selectFiles": "Select files for context",
     "AWS.amazonq.featureDev.pillText.retry": "Retry",

packages/core/src/amazonqFeatureDev/controllers/chat/controller.ts

@@ -425,6 +425,7 @@ export class FeatureDevController {
 
             if (session?.state?.tokenSource?.token.isCancellationRequested) {
                 this.disposeToken(session)
+                this.workOnNewTask(session)
             } else {
                 this.messenger.sendAsyncEventProgress(tabID, false, undefined)
 
@@ -446,6 +447,11 @@ export class FeatureDevController {
         }
     }
     private workOnNewTask(message: any) {
+        this.messenger.sendAnswer({
+            message: i18n('AWS.amazonq.featureDev.pillText.stoppedCodeGeneration'),
+            type: 'answer-part',
+            tabID: message.tabID,
+        })
         this.messenger.sendAnswer({
             type: 'system-prompt',
             tabID: message.tabID,
@@ -713,11 +719,10 @@ export class FeatureDevController {
 
     private async stopResponse(message: any) {
         this.messenger.sendAnswer({
-            message: i18n('AWS.amazonq.featureDev.pillText.stoppedCodeGeneration'),
+            message: i18n('AWS.amazonq.featureDev.pillText.stoppingCodeGeneration'),
             type: 'answer-part',
             tabID: message.tabID,
         })
-        this.workOnNewTask(message)
         const session = await this.sessionStorage.getSession(message.tabID)
         if (session.state?.tokenSource) {
             session.state?.tokenSource?.cancel()