wangxuming/makelore
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix(coding): preserve runtime model and failure contracts

Browse Source
This commit is contained in:
brother7
2026-08-23 21:21:02 +08:00
parent 52b2467d5d
commit 195979d30f
11 changed files with 427 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
electron/api/coding-provider-auth.ts
View File

@@ -8,33 +8,53 @@ import {
const WORKS_SQUARE_AI_GATEWAY_CREDENTIAL_MODE = 'works_square_ai_gateway';
const AUTHENTICATION_ERROR_PATTERN = /\b(?:401|403|unauthori[sz]ed|forbidden|authentication failed|auth failed|invalid (?:api key|credential|access token|bearer token)|(?:access |bearer )?token expired)\b/i;
export class CodingProviderCredentialRefreshError extends Error {
constructor(message: string, options?: ErrorOptions) {
super(message, options);
this.name = 'CodingProviderCredentialRefreshError';
}
}
export function isCodingProviderAuthenticationError(error: unknown): boolean {
return error instanceof Error && AUTHENTICATION_ERROR_PATTERN.test(error.message);
return error instanceof CodingProviderCredentialRefreshError
|| (error instanceof Error && AUTHENTICATION_ERROR_PATTERN.test(error.message));
}
export async function refreshCodingProviderCredential(accountId: string): Promise<void> {
const providerService = getProviderService();
const account = await providerService.getAccount(accountId);
if (!account?.enabled) throw new Error('Provider account is unavailable');
try {
const providerService = getProviderService();
const account = await providerService.getAccount(accountId);
if (!account?.enabled) {
throw new CodingProviderCredentialRefreshError('Provider account is unavailable');
}
if (account.metadata?.worksSquareCredentialMode === WORKS_SQUARE_AI_GATEWAY_CREDENTIAL_MODE) {
markWorksSquareAIGatewayCredentialExpired();
const credential = await getFreshWorksSquareAIGatewayCredential();
if (!credential) throw new Error('Provider credential refresh failed');
await providerService.updateAccount(account.id, {
baseUrl: credential.oneApiBaseUrl,
metadata: {
...account.metadata,
worksSquareCredentialExpiresAt: credential.expiresAt === null
? undefined
: new Date(credential.expiresAt).toISOString(),
},
}, credential.accessToken);
return;
}
if (account.metadata?.worksSquareCredentialMode === WORKS_SQUARE_AI_GATEWAY_CREDENTIAL_MODE) {
markWorksSquareAIGatewayCredentialExpired();
const credential = await getFreshWorksSquareAIGatewayCredential();
if (!credential) {
throw new CodingProviderCredentialRefreshError('Provider credential refresh failed');
}
await providerService.updateAccount(account.id, {
baseUrl: credential.oneApiBaseUrl,
metadata: {
...account.metadata,
worksSquareCredentialExpiresAt: credential.expiresAt === null
? undefined
: new Date(credential.expiresAt).toISOString(),
},
}, credential.accessToken);
return;
}
const current = await resolvePiProviderCredentialFromSecretStore(account);
if (!current && account.authMode !== 'local') {
throw new Error('Provider credential is unavailable');
const current = await resolvePiProviderCredentialFromSecretStore(account);
if (!current && account.authMode !== 'local') {
throw new CodingProviderCredentialRefreshError('Provider credential is unavailable');
}
} catch (error) {
if (error instanceof CodingProviderCredentialRefreshError) throw error;
throw new CodingProviderCredentialRefreshError(
'Provider credential refresh failed',
{ cause: error },
);
}
}