implement retry mechanism

Author: Hweinstock

packages/core/src/applicationcomposer/messageHandlers/generateResourceHandler.ts

@@ -18,13 +18,14 @@ import {
 } from '../types'
 import globals from '../../shared/extensionGlobals'
 import { getLogger } from '../../shared/logger/logger'
-import * as extApi from '../../amazonq/extApi'
+import { AmazonqNotFoundError, getAmazonqApi } from '../../amazonq/extApi'
 
 const TIMEOUT = 30_000
 
 export async function generateResourceHandler(request: GenerateResourceRequestMessage, context: WebviewContext) {
     try {
         const { chatResponse, references, metadata, isSuccess } = await generateResource(request.cfnType)
+
         const responseMessage: GenerateResourceResponseMessage = {
             command: Command.GENERATE_RESOURCE,
             messageType: MessageType.RESPONSE,
@@ -57,9 +58,9 @@ async function generateResource(cfnType: string) {
     let startTime = globals.clock.Date.now()
 
     try {
-        const amazonqApi = await extApi.getAmazonqApi()
+        const amazonqApi = await getAmazonqApi()
         if (!amazonqApi) {
-            throw new extApi.AmazonqNotFoundError()
+            throw new AmazonqNotFoundError()
         }
         const request: GenerateAssistantResponseRequest = {
             conversationState: {

packages/core/src/test/applicationcomposer/messageHandlers/generateResourceHandler.test.ts

@@ -16,6 +16,7 @@ describe.only('generateResourceHandler', function () {
     })
     for (const _ of Array.from({ length: 1000 }, (i) => i)) {
         it('amazon q is not installed', async () => {
+            this.retries(3)
             const panel = await createTemplate()
             const postMessageSpy = sinon.spy(panel.webview, 'postMessage')
             const context = await createWebviewContext({