From 04518b52c0ddcd5ae1192763c55e472add218b3c Mon Sep 17 00:00:00 2001 From: matt korwel Date: Thu, 19 Jun 2025 16:52:22 -0700 Subject: Auth First Run (#1207) Co-authored-by: Tommaso Sciortino Co-authored-by: N. Taylor Mullen --- packages/core/src/code_assist/codeAssist.ts | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'packages/core/src/code_assist/codeAssist.ts') diff --git a/packages/core/src/code_assist/codeAssist.ts b/packages/core/src/code_assist/codeAssist.ts index 4b46f8b9..1ceb818b 100644 --- a/packages/core/src/code_assist/codeAssist.ts +++ b/packages/core/src/code_assist/codeAssist.ts @@ -4,15 +4,23 @@ * SPDX-License-Identifier: Apache-2.0 */ -import { ContentGenerator } from '../core/contentGenerator.js'; +import { AuthType, ContentGenerator } from '../core/contentGenerator.js'; import { getOauthClient } from './oauth2.js'; import { setupUser } from './setup.js'; import { CodeAssistServer, HttpOptions } from './server.js'; export async function createCodeAssistContentGenerator( httpOptions: HttpOptions, + authType: AuthType, ): Promise { - const authClient = await getOauthClient(); - const projectId = await setupUser(authClient); - return new CodeAssistServer(authClient, projectId, httpOptions); + if ( + authType === AuthType.LOGIN_WITH_GOOGLE_ENTERPRISE || + authType === AuthType.LOGIN_WITH_GOOGLE_PERSONAL + ) { + const authClient = await getOauthClient(); + const projectId = await setupUser(authClient); + return new CodeAssistServer(authClient, projectId, httpOptions); + } + + throw new Error(`Unsupported authType: ${authType}`); } -- cgit v1.2.3