fix(amazonq): allow for more iterations on prompt related errors

Author: avi-alpert

packages/core/src/amazonqDoc/errors.ts

@@ -7,16 +7,16 @@ import { ToolkitError } from '../shared/errors'
 import { i18n } from '../shared/i18n-helper'
 
 export class DocServiceError extends ToolkitError {
-    remainingIterations: number
-    constructor(message: string, code: string, remainingIterations: number) {
+    remainingIterations?: number
+    constructor(message: string, code: string, remainingIterations?: number) {
         super(message, { code })
         this.remainingIterations = remainingIterations
     }
 }
 
 export class ReadmeTooLargeError extends DocServiceError {
-    constructor(remainingIterations: number) {
-        super(i18n('AWS.amazonq.doc.error.readmeTooLarge'), ReadmeTooLargeError.name, remainingIterations)
+    constructor() {
+        super(i18n('AWS.amazonq.doc.error.readmeTooLarge'), ReadmeTooLargeError.name)
     }
 }
 
@@ -27,18 +27,14 @@ export class ReadmeUpdateTooLargeError extends DocServiceError {
 }
 
 export class WorkspaceEmptyError extends DocServiceError {
-    constructor(remainingIterations: number) {
-        super(i18n('AWS.amazonq.doc.error.workspaceEmpty'), WorkspaceEmptyError.name, remainingIterations)
+    constructor() {
+        super(i18n('AWS.amazonq.doc.error.workspaceEmpty'), WorkspaceEmptyError.name)
     }
 }
 
 export class NoChangeRequiredException extends DocServiceError {
-    constructor(remainingIterations: number) {
-        super(
-            i18n('AWS.amazonq.doc.error.noChangeRequiredException'),
-            NoChangeRequiredException.name,
-            remainingIterations
-        )
+    constructor() {
+        super(i18n('AWS.amazonq.doc.error.noChangeRequiredException'), NoChangeRequiredException.name)
     }
 }
 
@@ -49,8 +45,8 @@ export class PromptRefusalException extends DocServiceError {
 }
 
 export class ContentLengthError extends DocServiceError {
-    constructor(remainingIterations: number) {
-        super(i18n('AWS.amazonq.doc.error.contentLengthError'), ContentLengthError.name, remainingIterations)
+    constructor() {
+        super(i18n('AWS.amazonq.doc.error.contentLengthError'), ContentLengthError.name)
     }
 }
 

packages/core/src/amazonqDoc/session/sessionState.ts

@@ -148,16 +148,16 @@ abstract class CodeGenBase {
                     messenger.sendUpdatePromptProgress(this.tabID, null)
                     switch (true) {
                         case codegenResult.codeGenerationStatusDetail?.includes('README_TOO_LARGE'): {
-                            throw new ReadmeTooLargeError(codeGenerationRemainingIterationCount)
+                            throw new ReadmeTooLargeError()
                         }
                         case codegenResult.codeGenerationStatusDetail?.includes('README_UPDATE_TOO_LARGE'): {
                             throw new ReadmeUpdateTooLargeError(codeGenerationRemainingIterationCount)
                         }
                         case codegenResult.codeGenerationStatusDetail?.includes('WORKSPACE_TOO_LARGE'): {
-                            throw new ContentLengthError(codeGenerationRemainingIterationCount)
+                            throw new ContentLengthError()
                         }
                         case codegenResult.codeGenerationStatusDetail?.includes('WORKSPACE_EMPTY'): {
-                            throw new WorkspaceEmptyError(codeGenerationRemainingIterationCount)
+                            throw new WorkspaceEmptyError()
                         }
                         case codegenResult.codeGenerationStatusDetail?.includes('PROMPT_UNRELATED'): {
                             throw new PromptUnrelatedError(codeGenerationRemainingIterationCount)
@@ -177,12 +177,11 @@ abstract class CodeGenBase {
                         }
                         case codegenResult.codeGenerationStatusDetail?.includes('EmptyPatch'): {
                             if (codegenResult.codeGenerationStatusDetail?.includes('NO_CHANGE_REQUIRED')) {
-                                throw new NoChangeRequiredException(codeGenerationRemainingIterationCount)
+                                throw new NoChangeRequiredException()
                             }
                             throw new DocServiceError(
                                 i18n('AWS.amazonq.doc.error.docGen.default'),
-                                'EmptyPatchException',
-                                codeGenerationRemainingIterationCount
+                                'EmptyPatchException'
                             )
                         }
                         case codegenResult.codeGenerationStatusDetail?.includes('Throttling'): {