speed up loading the context commands (#12)

Author: leigaol

packages/core/src/amazonq/lsp/lspClient.ts

@@ -40,7 +40,7 @@ import { CodeWhispererSettings } from '../../codewhisperer/util/codewhispererSet
 import { fs } from '../../shared/fs/fs'
 import { getLogger } from '../../shared/logger/logger'
 import globals from '../../shared/extensionGlobals'
-import { sleep, waitUntil } from '../../shared/utilities/timeoutUtils'
+import { waitUntil } from '../../shared/utilities/timeoutUtils'
 
 const localize = nls.loadMessageBundle()
 
@@ -225,13 +225,6 @@ export class LspClient {
     async waitUntilReady() {
         return waitUntil(
             async () => {
-                for (let i = 0; i < 5; i++) {
-                    if (this.client !== undefined) {
-                        break
-                    } else {
-                        await sleep(5_000)
-                    }
-                }
                 if (this.client === undefined) {
                     return false
                 }

packages/core/src/amazonq/lsp/lspController.ts

@@ -24,6 +24,7 @@ import { ToolkitError } from '../../shared/errors'
 import { isWeb } from '../../shared/extensionGlobals'
 import { getUserAgent } from '../../shared/telemetry/util'
 import { isAmazonInternalOs } from '../../shared/vscode/env'
+import { sleep } from '../../shared/utilities/timeoutUtils'
 
 export interface Chunk {
     readonly filePath: string
@@ -399,7 +400,8 @@ export class LspController {
             try {
                 await activateLsp(context)
                 getLogger().info('LspController: LSP activated')
-                void LspController.instance.buildIndex(buildIndexConfig)
+                await vscode.commands.executeCommand(`aws.amazonq.updateContextCommandItems`)
+                await sleep(5_000).then(void LspController.instance.buildIndex(buildIndexConfig))
                 // log the LSP server CPU and Memory usage per 30 minutes.
                 globals.clock.setInterval(
                     async () => {