diff options
Diffstat (limited to 'packages/core/src/code_assist/setup.ts')
| -rw-r--r-- | packages/core/src/code_assist/setup.ts | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/packages/core/src/code_assist/setup.ts b/packages/core/src/code_assist/setup.ts index 0efe5f0c..6ff61523 100644 --- a/packages/core/src/code_assist/setup.ts +++ b/packages/core/src/code_assist/setup.ts @@ -5,7 +5,7 @@ */ import { ClientMetadata, OnboardUserRequest } from './types.js'; -import { CcpaServer } from './ccpaServer.js'; +import { CodeAssistServer } from './server.js'; import { OAuth2Client } from 'google-auth-library'; import { GaxiosError } from 'gaxios'; import { clearCachedCredentials } from './oauth2.js'; @@ -19,7 +19,7 @@ export async function setupUser( oAuth2Client: OAuth2Client, projectId?: string, ): Promise<string> { - const ccpaServer: CcpaServer = new CcpaServer(oAuth2Client, projectId); + const caServer = new CodeAssistServer(oAuth2Client, projectId); const clientMetadata: ClientMetadata = { ideType: 'IDE_UNSPECIFIED', platform: 'PLATFORM_UNSPECIFIED', @@ -30,7 +30,7 @@ export async function setupUser( } // TODO: Support Free Tier user without projectId. - const loadRes = await ccpaServer.loadCodeAssist({ + const loadRes = await caServer.loadCodeAssist({ cloudaicompanionProject: process.env.GOOGLE_CLOUD_PROJECT, metadata: clientMetadata, }); @@ -42,10 +42,10 @@ export async function setupUser( }; try { // Poll onboardUser until long running operation is complete. - let lroRes = await ccpaServer.onboardUser(onboardReq); + let lroRes = await caServer.onboardUser(onboardReq); while (!lroRes.done) { await new Promise((f) => setTimeout(f, 5000)); - lroRes = await ccpaServer.onboardUser(onboardReq); + lroRes = await caServer.onboardUser(onboardReq); } return lroRes.response?.cloudaicompanionProject?.id || ''; |