From 930168f908bd95a5f04dbaa3c54836c0e667c764 Mon Sep 17 00:00:00 2001
From: David Hasani <davhasan@amazon.com>
Date: Tue, 31 Dec 2024 09:21:12 -0800
Subject: [PATCH] fix(amazonq): increase maxRetries for GetPlan

---
 packages/core/src/codewhisperer/client/codewhisperer.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/core/src/codewhisperer/client/codewhisperer.ts b/packages/core/src/codewhisperer/client/codewhisperer.ts
index 69ec6ad1b34..7a869a68372 100644
--- a/packages/core/src/codewhisperer/client/codewhisperer.ts
+++ b/packages/core/src/codewhisperer/client/codewhisperer.ts
@@ -318,7 +318,8 @@ export class DefaultCodeWhispererClient {
     public async codeModernizerGetCodeTransformationPlan(
         request: CodeWhispererUserClient.GetTransformationPlanRequest
     ): Promise<PromiseResult<CodeWhispererUserClient.GetTransformationPlanResponse, AWSError>> {
-        return (await this.createUserSdkClient()).getTransformationPlan(request).promise()
+        // instead of the default of 3 retries, use 8 retries for this API which is polled every 5 seconds
+        return (await this.createUserSdkClient(8)).getTransformationPlan(request).promise()
     }
 
     public async startCodeFixJob(