vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-04-30 04:30:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 24

Simplify streaming code for code assist server (#4619)

Browse source
This commit is contained in:
Tommaso Sciortino 2025-07-21 13:44:43 -07:00 • committed by GitHub
parent f95674e646
commit d7a57d85a3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 36 additions and 133 deletions
Download patch file Download diff file Expand all files Collapse all files

12
packages/core/src/code_assist/setup.ts
View file

@ -22,12 +22,17 @@ export class ProjectIdRequiredError extends Error {
}
}
export interface UserData {
projectId: string;
userTier: UserTierId;
}
/**
*
* @param projectId the user's project id, if any
* @returns the user's actual project id
*/
export async function setupUser(client: OAuth2Client): Promise<string> {
export async function setupUser(client: OAuth2Client): Promise<UserData> {
let projectId = process.env.GOOGLE_CLOUD_PROJECT || undefined;
const caServer = new CodeAssistServer(client, projectId);
@ -64,7 +69,10 @@ export async function setupUser(client: OAuth2Client): Promise<string> {
await new Promise((f) => setTimeout(f, 5000));
lroRes = await caServer.onboardUser(onboardReq);
}
return lroRes.response?.cloudaicompanionProject?.id || '';
return {
projectId: lroRes.response?.cloudaicompanionProject?.id || '',
userTier: tier.id,
};
}
function getOnboardTier(res: LoadCodeAssistResponse): GeminiUserTier {