luulam/9router
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(antigravity): preserve client identity on model catalog requests

Browse Source 
Restrict the legacy IDE-version override to generation endpoints so
catalog and other passthrough requests keep their original User-Agent
and metadata.ideVersion, letting newer Antigravity releases see current
models like Gemini 3.7 Flash in the MITM selector.

Fixes #3414
This commit is contained in:
Matt Van Horn
2026-09-02 20:14:33 +07:00
committed by decolua
parent ed1bd0c528
commit f68d2f5ee5
3 changed files with 97 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/mitm/antigravityIdeVersion.js
View File

@@ -1,8 +1,8 @@
"use strict";
// Rewrite Antigravity IDE markers so upstream AG 2.x backend accepts the request.
// User-Agent header (antigravity/<old>) and body.metadata.ideVersion are forced
// to a known-good IDE version. Hardcoded MVP — toggle/version configurable later.
// Rewrite Antigravity IDE markers on generation requests so upstream AG 2.x
// backend accepts them. Catalog and other passthrough requests retain the
// client's current identity. Hardcoded MVP — toggle/version configurable later.
const ANTIGRAVITY_IDE_VERSION = "1.23.2";
const ANTIGRAVITY_IDE_VERSION_OVERRIDE_ENABLED = true;
@@ -19,8 +19,10 @@ function rewriteAntigravityUserAgent(userAgent, version) {
return userAgent.replace(/antigravity\/[^\s]+/, `antigravity/${version}`);
}
function applyAntigravityIdeVersionOverride(bodyBuffer, headers) {
if (!ANTIGRAVITY_IDE_VERSION_OVERRIDE_ENABLED) {
function applyAntigravityIdeVersionOverride(bodyBuffer, headers, requestUrl) {
const isGenerationEndpoint = requestUrl?.includes(":generateContent") ||
requestUrl?.includes(":streamGenerateContent");
if (!ANTIGRAVITY_IDE_VERSION_OVERRIDE_ENABLED || !isGenerationEndpoint) {
return { bodyBuffer, headers, applied: false, version: ANTIGRAVITY_IDE_VERSION };
}

2
src/mitm/server.js
View File

@@ -132,7 +132,7 @@ async function passthrough(req, res, bodyBuffer, onResponse) {
const tool = getToolForHost(req.headers.host);
const versionOverride = tool === "antigravity"
? applyAntigravityIdeVersionOverride(bodyBuffer, req.headers)
? applyAntigravityIdeVersionOverride(bodyBuffer, req.headers, req.url)
: { bodyBuffer, headers: req.headers };
const bodyForForwarding = versionOverride.bodyBuffer;
const headersForForwarding = { ...versionOverride.headers, host: targetHost };