copilot-api-plus 1.0.32 → 1.0.34

package/dist/main.js

@@ -2162,30 +2162,66 @@ function createErrorResponse(type, message, status) {
 * Create Anthropic-compatible message response using Antigravity
 * Note: Both Gemini and Claude models use the same endpoint and Gemini-style format
 */
+const MAX_RETRIES = 5;
 async function createAntigravityMessages(request) {
-	const accessToken = await getValidAccessToken();
-	if (!accessToken) return createErrorResponse("authentication_error", "No valid Antigravity access token available. Please run login first.", 401);
 	const endpoint = request.stream ? ANTIGRAVITY_STREAM_URL : ANTIGRAVITY_NO_STREAM_URL;
 	const body = buildGeminiRequest(request);
-	consola.debug(`Antigravity messages request to ${endpoint} with model ${request.model}`);
-	try {
-		const response = await fetch(endpoint, {
-			method: "POST",
-			headers: {
-				Host: ANTIGRAVITY_API_HOST,
-				"User-Agent": ANTIGRAVITY_USER_AGENT,
-				Authorization: `Bearer ${accessToken}`,
-				"Content-Type": "application/json",
-				"Accept-Encoding": "gzip"
-			},
-			body: JSON.stringify(body)
-		});
-		if (!response.ok) return await handleApiError(response);
-		return request.stream ? transformStreamResponse(response, request.model) : await transformNonStreamResponse(response, request.model);
-	} catch (error) {
-		consola.error("Antigravity messages request error:", error);
-		return createErrorResponse("api_error", `Request failed: ${String(error)}`, 500);
+	for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
+		const accessToken = await getValidAccessToken();
+		if (!accessToken) return createErrorResponse("authentication_error", "No valid Antigravity access token available. Please run login first.", 401);
+		consola.debug(`Antigravity request to ${endpoint} (attempt ${attempt + 1}/${MAX_RETRIES + 1})`);
+		try {
+			const response = await fetch(endpoint, {
+				method: "POST",
+				headers: {
+					Host: ANTIGRAVITY_API_HOST,
+					"User-Agent": ANTIGRAVITY_USER_AGENT,
+					Authorization: `Bearer ${accessToken}`,
+					"Content-Type": "application/json",
+					"Accept-Encoding": "gzip"
+				},
+				body: JSON.stringify(body)
+			});
+			if (response.ok) return request.stream ? transformStreamResponse(response, request.model) : await transformNonStreamResponse(response, request.model);
+			const errorResult = await handleApiError(response);
+			if (errorResult.shouldRetry && attempt < MAX_RETRIES) {
+				consola.info(`Rate limited, retrying in ${errorResult.retryDelayMs}ms...`);
+				await sleep(errorResult.retryDelayMs);
+				continue;
+			}
+			return errorResult.response;
+		} catch (error) {
+			consola.error("Antigravity messages request error:", error);
+			if (attempt < MAX_RETRIES) {
+				await sleep(500);
+				continue;
+			}
+			return createErrorResponse("api_error", `Request failed: ${String(error)}`, 500);
+		}
 	}
+	return createErrorResponse("api_error", "Max retries exceeded", 429);
+}
+/**
+* Parse retry delay from error response
+*/
+function parseRetryDelay(errorText) {
+	try {
+		const details = JSON.parse(errorText).error?.details ?? [];
+		for (const detail of details) {
+			if (detail["@type"]?.includes("RetryInfo") && detail.retryDelay) {
+				const match = /(\d+(?:\.\d+)?)s/.exec(detail.retryDelay);
+				if (match) return Math.ceil(Number.parseFloat(match[1]) * 1e3);
+			}
+			if (detail.quotaResetDelay) {
+				const match = /(\d+(?:\.\d+)?)(?:ms|s)/.exec(detail.quotaResetDelay);
+				if (match) {
+					const value = Number.parseFloat(match[1]);
+					return detail.quotaResetDelay.includes("ms") ? Math.ceil(value) : Math.ceil(value * 1e3);
+				}
+			}
+		}
+	} catch {}
+	return 500;
 }
 /**
 * Handle API error response
@@ -2194,8 +2230,19 @@ async function handleApiError(response) {
 	const errorText = await response.text();
 	consola.error(`Antigravity error: ${response.status} ${errorText}`);
 	if (response.status === 403) await disableCurrentAccount();
-	if (response.status === 429 || response.status === 503) await rotateAccount();
-	return createErrorResponse("api_error", `Antigravity API error: ${response.status}`, response.status);
+	if (response.status === 429 || response.status === 503) {
+		await rotateAccount();
+		return {
+			shouldRetry: true,
+			retryDelayMs: parseRetryDelay(errorText),
+			response: createErrorResponse("api_error", `Antigravity API error: ${response.status}`, response.status)
+		};
+	}
+	return {
+		shouldRetry: false,
+		retryDelayMs: 0,
+		response: createErrorResponse("api_error", `Antigravity API error: ${response.status}`, response.status)
+	};
 }
 /**
 * Emit SSE event to controller based on stream event type
@@ -2272,12 +2319,14 @@ function transformStreamResponse(response, model) {
 */
 async function processStream(reader, decoder, state$1, controller) {
 	const emit = (event) => emitSSEEvent(event, controller, state$1);
-	while (true) {
+	let finished = false;
+	while (!finished) {
 		const { done, value } = await reader.read();
 		if (done) break;
 		const chunk = decoder.decode(value, { stream: true });
 		const lines = processChunk(chunk, state$1);
 		for (const line of lines) {
+			if (finished) break;
 			const data = parseSSELine(line);
 			if (!data) continue;
 			const { candidates, usage } = extractFromData(data);
@@ -2288,7 +2337,11 @@ async function processStream(reader, decoder, state$1, controller) {
 			const candidate = candidates[0];
 			const parts = candidate?.content?.parts ?? [];
 			for (const part of parts) processPart(part, state$1, emit);
-			if (candidate?.finishReason === "STOP") handleFinish(state$1, emit);
+			if (candidate?.finishReason === "STOP") {
+				handleFinish(state$1, emit);
+				finished = true;
+				break;
+			}
 		}
 	}
 }