3e52af35e2
- Send { project, model, request } envelope expected by Cloud Code Assist
- Parse google.rpc.RetryInfo from 429 responses to expose retry hint
Co-authored-by: Cursor <cursoragent@cursor.com>
89 lines
2.9 KiB
JavaScript
89 lines
2.9 KiB
JavaScript
import { BaseExecutor } from "./base.js";
|
|
import { PROVIDERS } from "../config/providers.js";
|
|
import { OAUTH_ENDPOINTS, GEMINI_CLI_API_CLIENT, geminiCLIUserAgent } from "../config/appConstants.js";
|
|
|
|
export class GeminiCLIExecutor extends BaseExecutor {
|
|
constructor() {
|
|
super("gemini-cli", PROVIDERS["gemini-cli"]);
|
|
}
|
|
|
|
buildUrl(model, stream, urlIndex = 0) {
|
|
const action = stream ? "streamGenerateContent?alt=sse" : "generateContent";
|
|
return `${this.config.baseUrl}:${action}`;
|
|
}
|
|
|
|
buildHeaders(credentials, stream = true) {
|
|
return {
|
|
"Content-Type": "application/json",
|
|
"Authorization": `Bearer ${credentials.accessToken}`,
|
|
"User-Agent": geminiCLIUserAgent(this._currentModel),
|
|
"X-Goog-Api-Client": GEMINI_CLI_API_CLIENT,
|
|
"Accept": stream ? "text/event-stream" : "application/json"
|
|
};
|
|
}
|
|
|
|
transformRequest(model, body, stream, credentials) {
|
|
// Store model for use in buildHeaders (called by base.execute after transformRequest)
|
|
this._currentModel = model;
|
|
// Cloud Code Assist wraps the Gemini payload: { project, model, request: <body> }
|
|
if (body && body.request && body.model) return body;
|
|
return {
|
|
project: credentials?.projectId || body?.project,
|
|
model,
|
|
request: body
|
|
};
|
|
}
|
|
|
|
// Parse RetryInfo.retryDelay from Google API 429 body to surface upstream retry hint
|
|
parseError(response, bodyText) {
|
|
const base = super.parseError(response, bodyText);
|
|
if (response.status !== 429 || !bodyText) return base;
|
|
try {
|
|
const parsed = JSON.parse(bodyText);
|
|
const details = parsed?.error?.details;
|
|
if (Array.isArray(details)) {
|
|
for (const d of details) {
|
|
if (d?.["@type"] === "type.googleapis.com/google.rpc.RetryInfo" && d?.retryDelay) {
|
|
base.retryAfter = d.retryDelay;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
} catch {}
|
|
return base;
|
|
}
|
|
|
|
async refreshCredentials(credentials, log) {
|
|
if (!credentials.refreshToken) return null;
|
|
|
|
try {
|
|
const response = await fetch(OAUTH_ENDPOINTS.google.token, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/x-www-form-urlencoded", "Accept": "application/json" },
|
|
body: new URLSearchParams({
|
|
grant_type: "refresh_token",
|
|
refresh_token: credentials.refreshToken,
|
|
client_id: this.config.clientId,
|
|
client_secret: this.config.clientSecret
|
|
})
|
|
});
|
|
|
|
if (!response.ok) return null;
|
|
|
|
const tokens = await response.json();
|
|
log?.info?.("TOKEN", "Gemini CLI refreshed");
|
|
|
|
return {
|
|
accessToken: tokens.access_token,
|
|
refreshToken: tokens.refresh_token || credentials.refreshToken,
|
|
expiresIn: tokens.expires_in,
|
|
projectId: credentials.projectId
|
|
};
|
|
} catch (error) {
|
|
log?.error?.("TOKEN", `Gemini CLI refresh error: ${error.message}`);
|
|
return null;
|
|
}
|
|
}
|
|
}
|
|
|
|
export default GeminiCLIExecutor;
|