From d7a57d85a39535e84bba7e65eb02dcb627b9cb81 Mon Sep 17 00:00:00 2001 From: Tommaso Sciortino Date: Mon, 21 Jul 2025 13:44:43 -0700 Subject: Simplify streaming code for code assist server (#4619) --- packages/core/src/code_assist/codeAssist.ts | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 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 23dfe403..f9c6a7a3 100644 --- a/packages/core/src/code_assist/codeAssist.ts +++ b/packages/core/src/code_assist/codeAssist.ts @@ -21,8 +21,14 @@ export async function createCodeAssistContentGenerator( authType === AuthType.CLOUD_SHELL ) { const authClient = await getOauthClient(authType, config); - const projectId = await setupUser(authClient); - return new CodeAssistServer(authClient, projectId, httpOptions, sessionId); + const userData = await setupUser(authClient); + return new CodeAssistServer( + authClient, + userData.projectId, + httpOptions, + sessionId, + userData.userTier, + ); } throw new Error(`Unsupported authType: ${authType}`); -- cgit v1.2.3