feat: Added handling for iteration limit exceeded exceptions for approach and code gen in featureDev (#1840)

Author: kumsmrit

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/FeatureDevExceptions.kt

@@ -9,6 +9,10 @@ open class FeatureDevException(override val message: String?, override val cause
 
 class ContentLengthError(override val message: String, override val cause: Throwable?) : RuntimeException()
 
+class PlanIterationLimitError(override val message: String, override val cause: Throwable?) : RuntimeException()
+
+class CodeIterationLimitError(override val message: String, override val cause: Throwable?) : RuntimeException()
+
 internal fun codeGenerationFailedError(): Nothing =
     throw FeatureDevException(message("amazonqFeatureDev.code_generation.failed_generation"))
 

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/controller/FeatureDevController.kt

@@ -30,10 +30,12 @@ import software.aws.toolkits.jetbrains.core.coroutines.EDT
 import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
 import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController
 import software.aws.toolkits.jetbrains.services.amazonq.toolwindow.AmazonQToolWindowFactory
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.CodeIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ContentLengthError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.DEFAULT_RETRY_LIMIT
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FEATURE_NAME
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.InboundAppMessagesHandler
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.PlanIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.createUserFacingErrorMessage
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.FeatureDevMessageType
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.messages.FollowUp
@@ -389,6 +391,37 @@ class FeatureDevController(
                         )
                     ),
                 )
+            } else if (err is PlanIterationLimitError) {
+                messenger.sendError(tabId = tabId, errMessage = err.message, retries = retriesRemaining(session))
+                messenger.sendSystemPrompt(
+                    tabId = tabId,
+                    followUp = listOf(
+                        FollowUp(
+                            pillText = message("amazonqFeatureDev.follow_up.new_plan"),
+                            type = FollowUpTypes.NEW_PLAN,
+                            status = FollowUpStatusType.Info,
+                        ),
+                        FollowUp(
+                            pillText = message("amazonqFeatureDev.follow_up.generate_code"),
+                            type = FollowUpTypes.GENERATE_CODE,
+                            status = FollowUpStatusType.Info,
+                        )
+                    ),
+                )
+                messenger.sendUpdatePlaceholder(tabId = tabId, newPlaceholder = message("amazonqFeatureDev.placeholder.after_code_generation"))
+            } else if (err is CodeIterationLimitError) {
+                messenger.sendError(tabId = tabId, errMessage = err.message, retries = retriesRemaining(session))
+                messenger.sendSystemPrompt(
+                    tabId = tabId,
+                    followUp = listOf(
+                        FollowUp(
+                            pillText = message("amazonqFeatureDev.follow_up.insert_code"),
+                            type = FollowUpTypes.INSERT_CODE,
+                            icon = FollowUpIcons.Ok,
+                            status = FollowUpStatusType.Success,
+                        )
+                    ),
+                )
             } else {
                 val msg = createUserFacingErrorMessage("$FEATURE_NAME request failed: ${err.message ?: err.cause?.message}")
                 messenger.sendError(

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/util/FeatureDevClientUtil.kt

@@ -11,11 +11,14 @@ import software.amazon.awssdk.services.codewhispererruntime.model.GetTaskAssistC
 import software.amazon.awssdk.services.codewhispererruntime.model.StartTaskAssistCodeGenerationResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.ValidationException
 import software.amazon.awssdk.services.codewhispererstreaming.model.CodeWhispererStreamingException
+import software.amazon.awssdk.services.codewhispererstreaming.model.ThrottlingException
 import software.aws.toolkits.core.utils.debug
 import software.aws.toolkits.core.utils.error
 import software.aws.toolkits.core.utils.getLogger
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.CodeIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ContentLengthError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FEATURE_NAME
+import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.PlanIterationLimitError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.apiError
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.clients.FeatureDevClient
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.clients.GenerateTaskAssistPlanResult
@@ -110,6 +113,10 @@ suspend fun generatePlan(
         if (e is CodeWhispererStreamingException) {
             errMssg = e.awsErrorDetails().errorMessage()
             logger.error(e) { "Generate plan failed for request: ${e.requestId()}" }
+
+            if (e is ThrottlingException && e.message?.contains("limit for number of iterations on an implementation plan") == true) {
+                throw PlanIterationLimitError(message("amazonqFeatureDev.approach_gen.iteration_limit.error_text"), e.cause)
+            }
         }
         apiError(errMssg, e.cause)
     }
@@ -134,6 +141,13 @@ fun startTaskAssistCodeGeneration(proxyClient: FeatureDevClient, conversationId:
         if (e is CodeWhispererRuntimeException) {
             errMssg = e.awsErrorDetails().errorMessage()
             logger.error(e) { "StartTaskAssistCodeGeneration failed for request: ${e.requestId()}" }
+
+            if (e is software.amazon.awssdk.services.codewhispererruntime.model.ThrottlingException && e.message?.contains(
+                    "limit for number of iterations on a code generation"
+                ) == true
+            ) {
+                throw CodeIterationLimitError(message("amazonqFeatureDev.code_generation.iteration_limit.error_text"), e.cause)
+            }
         }
         apiError(errMssg, e.cause)
     }

plugins/toolkit/resources/resources/software/aws/toolkits/resources/MessagesBundle.properties

@@ -35,13 +35,15 @@ action.aws.toolkit.toolwindow.sso.signout.text=Sign out of SSO
 action.dynamic.open.text=Open Resource...
 action.q.openchat.text=Chat with Q
 amazonqFeatureDev.approach_gen.error_text=Sorry, we're experiencing an issue on our side. Would you like to try again?
+amazonqFeatureDev.approach_gen.iteration_limit.error_text=You have reached the free tier limit for number of iterations on an implementation plan. Please proceed to generating code or start to discuss a new plan.
 amazonqFeatureDev.chat_message.ask_for_new_task=What change would you like to make?
 amazonqFeatureDev.chat_message.closed_session=Your session is now closed
 amazonqFeatureDev.chat_message.requesting_changes=Requesting changes ...
 amazonqFeatureDev.chat_message.start_code_generation=This may take a few minutes. I will send a notification when it's complete if you navigate away from this panel, but please keep the tab open.
 amazonqFeatureDev.chat_message.uploading_code=Uploading code ...
 amazonqFeatureDev.code_generation.failed_generation=Code generation failed
 amazonqFeatureDev.code_generation.generating_code=Generating code ...
+amazonqFeatureDev.code_generation.iteration_limit.error_text=You have reached the free tier limit for number of iterations on a code generation. Please proceed to accept the code or start a new conversation.
 amazonqFeatureDev.code_generation.no_file_changes=Unable to generate any file changes
 amazonqFeatureDev.code_generation.notification_message=Your code suggestions from Amazon Q are ready to review
 amazonqFeatureDev.code_generation.notification_open_link=Open chat