fix(amazonq): /doc enable text input in edit mode for retries and use it instead of retry button

Author: Viktor Shesternyak

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/DocConstants.kt

@@ -3,6 +3,11 @@
 
 package software.aws.toolkits.jetbrains.services.amazonqDoc
 
+import software.aws.toolkits.jetbrains.services.amazonqDoc.messages.FollowUp
+import software.aws.toolkits.jetbrains.services.amazonqDoc.messages.FollowUpStatusType
+import software.aws.toolkits.jetbrains.services.amazonqDoc.messages.FollowUpTypes
+import software.aws.toolkits.resources.message
+
 const val FEATURE_EVALUATION_PRODUCT_NAME = "DocGeneration"
 
 const val FEATURE_NAME = "Amazon Q Documentation Generation"
@@ -25,3 +30,17 @@ enum class ModifySourceFolderErrorReason(
 
     override fun toString(): String = reasonText
 }
+
+
+val NEW_SESSION_FOLLOWUPS: List<FollowUp> = listOf(
+    FollowUp(
+        pillText = message("amazonqDoc.prompt.reject.new_task"),
+        type = FollowUpTypes.NEW_TASK,
+        status = FollowUpStatusType.Info
+    ),
+    FollowUp(
+        pillText = message("amazonqDoc.prompt.reject.close_session"),
+        type = FollowUpTypes.CLOSE_SESSION,
+        status = FollowUpStatusType.Info
+    ))
+

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/messages/DocMessagePublisherExtensions.kt

@@ -7,6 +7,7 @@ import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthNeededState
 import software.aws.toolkits.jetbrains.services.amazonq.messages.MessagePublisher
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.messages.ProgressField
 import software.aws.toolkits.jetbrains.services.amazonqCodeTest.messages.PromptProgressMessage
+import software.aws.toolkits.jetbrains.services.amazonqDoc.NEW_SESSION_FOLLOWUPS
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.session.CodeReferenceGenerated
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.session.DeletedFileInfo
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.session.NewFileZipInfo
@@ -160,36 +161,29 @@ suspend fun MessagePublisher.sendErrorToUser(
 ) {
     val conversationIdText = if (conversationId == null) "" else "\n\nConversation ID: **$conversationId**"
 
-    var followUps = listOf(
-        FollowUp(
-            pillText = message("amazonqDoc.prompt.reject.new_task"),
-            type = FollowUpTypes.NEW_TASK,
-            status = FollowUpStatusType.Info
-        ),
-        FollowUp(
-            pillText = message("amazonqDoc.prompt.reject.close_session"),
-            type = FollowUpTypes.CLOSE_SESSION,
-            status = FollowUpStatusType.Info
-        )
-    );
-
-    this.sendChatInputEnabledMessage(tabId, enabled = isEnableChatInput)
-    if (isEnableChatInput) {
-        this.sendUpdatePlaceholder(tabId, message("amazonqDoc.edit.placeholder"))
-        followUps = Collections.emptyList()
-    }
-
     this.sendAnswer(
         tabId = tabId,
         messageType = DocMessageType.Answer,
         message = errMessage + conversationIdText,
     )
 
-    this.sendAnswer(
-        tabId = tabId,
-        messageType = DocMessageType.SystemPrompt,
-        followUp = followUps
-    )
+    if (isEnableChatInput) {
+        this.sendAnswer(
+            tabId = tabId,
+            messageType = DocMessageType.SystemPrompt,
+            followUp = Collections.emptyList()
+        )
+        this.sendUpdatePlaceholder(tabId, message("amazonqDoc.edit.placeholder"))
+    } else {
+        this.sendAnswer(
+            tabId = tabId,
+            messageType = DocMessageType.SystemPrompt,
+            followUp = NEW_SESSION_FOLLOWUPS
+        )
+        this.sendUpdatePlaceholder(tabId, message("amazonqDoc.prompt.placeholder"))
+    }
+
+    this.sendChatInputEnabledMessage(tabId, enabled = isEnableChatInput)
 }
 
 suspend fun MessagePublisher.sendMonthlyLimitError(tabId: String) {