fix(q): handle ServiceQuotaExceededException that is returned from backend when iteration limits reached for plan and code generation of feature dev (#4242)

Author: sarayev

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

@@ -11,6 +11,7 @@ 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.ServiceQuotaExceededException
 import software.amazon.awssdk.services.codewhispererstreaming.model.ThrottlingException
 import software.aws.toolkits.core.utils.debug
 import software.aws.toolkits.core.utils.getLogger
@@ -114,7 +115,9 @@ suspend fun generatePlan(
             errMssg = e.awsErrorDetails().errorMessage()
             logger.warn(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) {
+            if (e is ServiceQuotaExceededException ||
+                (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)
             }
         }
@@ -142,9 +145,11 @@ fun startTaskAssistCodeGeneration(proxyClient: FeatureDevClient, conversationId:
             errMssg = e.awsErrorDetails().errorMessage()
             logger.warn(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
+            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
+                    )
             ) {
                 throw CodeIterationLimitError(message("amazonqFeatureDev.code_generation.iteration_limit.error_text"), e.cause)
             }

plugins/core/sdk-codegen/codegen-resources/codewhispererruntime/service-2.json

@@ -39,6 +39,7 @@
             "output":{"shape":"CreateUploadUrlResponse"},
             "errors":[
                 {"shape":"ThrottlingException"},
+                {"shape":"ServiceQuotaExceededException"},
                 {"shape":"ConflictException"},
                 {"shape":"ResourceNotFoundException"},
                 {"shape":"InternalServerException"},
@@ -63,6 +64,9 @@
                 {
                     "shape": "ThrottlingException"
                 },
+                {
+                    "shape": "ServiceQuotaExceededException"
+                },
                 {
                     "shape": "InternalServerException"
                 },
@@ -258,6 +262,7 @@
             "output":{"shape":"StartTaskAssistCodeGenerationResponse"},
             "errors":[
                 {"shape":"ThrottlingException"},
+                {"shape":"ServiceQuotaExceededException"},
                 {"shape":"ConflictException"},
                 {"shape":"ResourceNotFoundException"},
                 {"shape":"InternalServerException"},
@@ -1560,6 +1565,15 @@
             "exception":true,
             "retryable":{"throttling":true}
         },
+        "ServiceQuotaExceededException":{
+            "type":"structure",
+            "required":["message"],
+            "members":{
+                "message":{"shape":"String"}
+            },
+            "exception":true,
+            "retryable":{"throttling":true}
+        },
         "TimeBetweenChunks": {
             "type": "list",
             "member": { "shape": "Double" },

plugins/core/sdk-codegen/codegen-resources/codewhispererstreaming/service-2.json

@@ -55,6 +55,7 @@
             "output":{"shape":"GenerateTaskAssistPlanResponse"},
             "errors":[
                 {"shape":"ThrottlingException"},
+                {"shape":"ServiceQuotaExceededException"},
                 {"shape":"ConflictException"},
                 {"shape":"ResourceNotFoundException"},
                 {"shape":"InternalServerException"},
@@ -632,6 +633,15 @@
             "exception":true,
             "retryable":{"throttling":true}
         },
+        "ServiceQuotaExceededException":{
+            "type":"structure",
+            "required":["message"],
+            "members":{
+                "message":{"shape":"String"}
+            },
+            "exception":true,
+            "retryable":{"throttling":true}
+        },
         "UploadId":{
             "type":"string",
             "max":128,