vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-05-01 05:00:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

Cache oauth credentials (#927)

Browse source
This commit is contained in:
Tommaso Sciortino 2025-06-11 13:26:41 -07:00 • committed by GitHub
parent c0580eaf4b
commit 24c61147b8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 87 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,6 +7,8 @@
import { ClientMetadata, OnboardUserRequest } from './types.js';
import { CcpaServer } from './ccpaServer.js';
import { OAuth2Client } from 'google-auth-library';
import { GaxiosError } from 'gaxios';
import { clearCachedCredentials } from './oauth2.js';
/**
*
@ -38,13 +40,27 @@ export async function setupUser(
cloudaicompanionProject: loadRes.cloudaicompanionProject || '',
metadata: clientMetadata,
};
try {
// Poll onboardUser until long running operation is complete.
let lroRes = await ccpaServer.onboardUser(onboardReq);
while (!lroRes.done) {
await new Promise((f) => setTimeout(f, 5000));
lroRes = await ccpaServer.onboardUser(onboardReq);
}
// Poll onboardUser until long running operation is complete.
let lroRes = await ccpaServer.onboardUser(onboardReq);
while (!lroRes.done) {
await new Promise((f) => setTimeout(f, 5000));
lroRes = await ccpaServer.onboardUser(onboardReq);
return lroRes.response?.cloudaicompanionProject?.id || '';
} catch (e) {
if (e instanceof GaxiosError) {
const detail = e.response?.data?.error?.details[0].detail;
if (detail && detail.includes('projectID is empty')) {
await clearCachedCredentials();
console.log(
'\n\nEnterprise users must specify GOOGLE_CLOUD_PROJECT ' +
'in your environment variables or .env file.\n\n',
);
process.exit(1);
}
}
throw e;
}
return lroRes.response?.cloudaicompanionProject?.id || '';
}