fix(amazonq): add handling for new limit mechanism

Author: willyyhuang

.changes/next-release/bugfix-554078f3-042e-4b73-8ac4-96085bfc0b8b.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Amazon Q Feature Dev: display limit reached error message"
+}

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/util/FeatureDevService.kt

@@ -135,9 +135,11 @@ class FeatureDevService(val proxyClient: FeatureDevClient, val project: Project)
                 logger.warn(e) { "StartTaskAssistCodeGeneration failed for request: ${e.requestId()}" }
 
                 if (e is software.amazon.awssdk.services.codewhispererruntime.model.ServiceQuotaExceededException || (
-                        e is software.amazon.awssdk.services.codewhispererruntime.model.ThrottlingException && e.message?.contains(
-                            "limit for number of iterations on a code generation"
-                        ) == true
+                        e is software.amazon.awssdk.services.codewhispererruntime.model.ThrottlingException && (
+                            e.message?.contains(
+                                "limit for number of iterations on a code generation"
+                            ) == true || e.message?.contains("StartTaskAssistCodeGeneration reached for this month.") == true
+                            )
                         )
                 ) {
                     throw CodeIterationLimitException(operation = FeatureDevOperation.StartTaskAssistCodeGeneration.toString(), desc = null, e.cause)

plugins/amazonq/chat/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/util/FeatureDevServiceTest.kt

@@ -195,6 +195,37 @@ class FeatureDevServiceTest : FeatureDevTestBase() {
         )
     }
 
+    @Test
+    fun `test startTaskAssistConversation throws ServiceQuotaExceededException, different case`() {
+        val exampleCWException =
+            ThrottlingException
+                .builder()
+                .awsErrorDetails(
+                    AwsErrorDetails.builder().errorMessage("StartTaskAssistCodeGeneration reached for this month.").build(),
+                ).build()
+        whenever(
+            featureDevClient.startTaskAssistCodeGeneration(
+                testConversationId,
+                testUploadId,
+                userMessage,
+                codeGenerationId = codeGenerationId,
+                currentCodeGenerationId = "EMPTY_CURRENT_CODE_GENERATION_ID",
+            ),
+        ).thenThrow(exampleCWException)
+
+        assertThatThrownBy {
+            featureDevService.startTaskAssistCodeGeneration(
+                testConversationId,
+                testUploadId,
+                userMessage,
+                codeGenerationId = codeGenerationId,
+                currentCodeGenerationId = "EMPTY_CURRENT_CODE_GENERATION_ID",
+            )
+        }.isExactlyInstanceOf(CodeIterationLimitException::class.java).withFailMessage(
+            message("amazonqFeatureDev.code_generation.iteration_limit.error_text"),
+        )
+    }
+
     @Test
     fun `test startTaskAssistConversation throws ServiceQuotaExceededException`() {
         val exampleCWException =
@@ -226,6 +257,7 @@ class FeatureDevServiceTest : FeatureDevTestBase() {
         )
     }
 
+
     @Test
     fun `test startTaskAssistConversation throws another CodeWhispererRuntimeException`() {
         val exampleCWException =