fix(featureDev): add handling for new limit mechanism

willyyhuang · willyyhuang · commit 6bdaef688c25 · 2024-11-22T12:29:28.000-05:00
diff --git a/packages/amazonq/.changes/next-release/Bug Fix-170bbb4e-ab52-4290-b803-1836b75ed415.json b/packages/amazonq/.changes/next-release/Bug Fix-170bbb4e-ab52-4290-b803-1836b75ed415.json
@@ -0,0 +1,4 @@
+{
+	"type": "Bug Fix",
+	"description": "Amazon Q Feature Dev: display limit reached error message"
+}
diff --git a/packages/core/src/amazonqFeatureDev/client/featureDev.ts b/packages/core/src/amazonqFeatureDev/client/featureDev.ts
@@ -171,6 +171,7 @@ export class FeatureDevClient {
                 isAwsError(e) &&
                 ((e.code === 'ThrottlingException' &&
                     e.message.includes('limit for number of iterations on a code generation')) ||
+                    e.message.includes('StartTaskAssistCodeGeneration reached for this month.') ||
                     e.code === 'ServiceQuotaExceededException')
             ) {
                 throw new CodeIterationLimitError()

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +	"type": "Bug Fix",
 +	"description": "Amazon Q Feature Dev: display limit reached error message"
 +}