Add Antigravity v1internal gateway adapter

2026-10-04 16:37:46 +08:00 · 2026-05-25 06:58:15 +08:00
parent 505d9fd8bc
commit 28c3a5dbe4
15 changed files with 978 additions and 33 deletions
@@ -52,6 +52,12 @@ const OPENAI_RERANK_TOP_N_DETAIL: &str = "Rerank request top_n must be a positiv
 const OPENAI_RERANK_CHAT_PAYLOAD_DETAIL: &str =
    "Rerank request must use query/documents, not chat messages";
 const OPENAI_RERANK_STREAM_UNSUPPORTED_DETAIL: &str = "Rerank requests do not support streaming";
+const ANTIGRAVITY_USER_SETTINGS_MISSING_BODY_DETAIL: &str =
+    "Antigravity setUserSettings request body is required";
+const ANTIGRAVITY_USER_SETTINGS_INVALID_JSON_DETAIL: &str =
+    "Antigravity setUserSettings request JSON body is invalid";
+const ANTIGRAVITY_USER_SETTINGS_INVALID_DETAIL: &str =
+    "Antigravity setUserSettings request must include object userSettings";

 #[derive(Clone, Copy, Debug, PartialEq, Eq)]
 enum OpenAiImageOperation {
@@ -103,7 +109,9 @@ pub(crate) fn ai_public_local_requires_buffered_body(
                        && request_context.request_path == "/v1/embeddings")
                    || (decision.route_family.as_deref() == Some("openai")
                        && decision.route_kind.as_deref() == Some("rerank")
-                        && request_context.request_path == "/v1/rerank"))
+                        && request_context.request_path == "/v1/rerank")
+                    || (decision.route_family.as_deref() == Some("antigravity")
+                        && decision.route_kind.as_deref() != Some("stream_generate_content")))
        })
 }

@@ -133,6 +141,12 @@ pub(crate) async fn maybe_build_local_ai_public_response(
        return Some(response);
    }

+    if let Some(response) =
+        maybe_build_local_antigravity_v1internal_response(request_context, request_body)
+    {
+        return Some(response);
+    }
+
    maybe_build_local_gemini_video_operations_response(state, request_context, decision).await
 }

@@ -861,6 +875,140 @@ fn maybe_build_local_claude_count_tokens_response(
    Some(Json(json!({ "input_tokens": input_tokens })).into_response())
 }

+fn maybe_build_local_antigravity_v1internal_response(
+    request_context: &GatewayPublicRequestContext,
+    request_body: Option<&Bytes>,
+) -> Option<Response<Body>> {
+    let decision = request_context.control_decision.as_ref()?;
+    if decision.route_family.as_deref() != Some("antigravity")
+        || request_context.request_method != http::Method::POST
+    {
+        return None;
+    }
+
+    match decision.route_kind.as_deref()? {
+        "load_code_assist" => {
+            Some(Json(build_antigravity_load_code_assist_payload()).into_response())
+        }
+        "fetch_available_models" => {
+            Some(Json(build_antigravity_fetch_available_models_payload()).into_response())
+        }
+        "fetch_user_info" => {
+            Some(Json(build_antigravity_fetch_user_info_payload()).into_response())
+        }
+        "fetch_admin_controls" => Some(Json(json!({})).into_response()),
+        "list_experiments" => Some(
+            Json(json!({
+                "experimentIds": [],
+                "flags": {}
+            }))
+            .into_response(),
+        ),
+        "record_code_assist_metrics" => Some(Json(json!({})).into_response()),
+        "set_user_settings" => Some(build_antigravity_set_user_settings_response(request_body)),
+        "stream_generate_content" => None,
+        _ => None,
+    }
+}
+
+fn build_antigravity_set_user_settings_response(request_body: Option<&Bytes>) -> Response<Body> {
+    let Some(request_body) = request_body else {
+        return build_ai_public_error_response(
+            http::StatusCode::BAD_REQUEST,
+            ANTIGRAVITY_USER_SETTINGS_MISSING_BODY_DETAIL,
+        );
+    };
+    let payload = match serde_json::from_slice::<Value>(request_body) {
+        Ok(payload) => payload,
+        Err(_) => {
+            return build_ai_public_error_response(
+                http::StatusCode::BAD_REQUEST,
+                ANTIGRAVITY_USER_SETTINGS_INVALID_JSON_DETAIL,
+            );
+        }
+    };
+    let Some(user_settings) = payload
+        .get("userSettings")
+        .filter(|value| value.is_object())
+        .cloned()
+    else {
+        return build_ai_public_error_response(
+            http::StatusCode::BAD_REQUEST,
+            ANTIGRAVITY_USER_SETTINGS_INVALID_DETAIL,
+        );
+    };
+
+    Json(json!({ "userSettings": user_settings })).into_response()
+}
+
+fn build_antigravity_load_code_assist_payload() -> Value {
+    json!({
+        "allowedTiers": ["free"],
+        "cloudaicompanionProject": "aether-antigravity-local",
+        "currentTier": "free",
+        "gcpManaged": false,
+        "paidTier": false,
+        "upgradeSubscriptionUri": ""
+    })
+}
+
+fn build_antigravity_fetch_user_info_payload() -> Value {
+    json!({
+        "regionCode": "US",
+        "userSettings": build_antigravity_default_user_settings_payload()
+    })
+}
+
+fn build_antigravity_default_user_settings_payload() -> Value {
+    json!({
+        "preferredModelId": "gemini-3.5-flash-low"
+    })
+}
+
+fn build_antigravity_fetch_available_models_payload() -> Value {
+    json!({
+        "models": {
+            "gemini-3.5-flash-low": antigravity_model_payload("gemini-3.5-flash-low", "Gemini 3.5 Flash Low"),
+            "gemini-3-flash-agent": antigravity_model_payload("gemini-3-flash-agent", "Gemini 3 Flash Agent"),
+            "gemini-3.1-flash-lite": antigravity_model_payload("gemini-3.1-flash-lite", "Gemini 3.1 Flash Lite"),
+            "gemini-3.1-pro-low": antigravity_model_payload("gemini-3.1-pro-low", "Gemini 3.1 Pro Low"),
+            "gemini-3-flash": antigravity_model_payload("gemini-3-flash", "Gemini 3 Flash"),
+            "gemini-3.1-flash-image": antigravity_model_payload("gemini-3.1-flash-image", "Gemini 3.1 Flash Image"),
+            "tab_flash_lite_preview": antigravity_model_payload("tab_flash_lite_preview", "Tab Flash Lite Preview"),
+            "tab_jump_flash_lite_preview": antigravity_model_payload("tab_jump_flash_lite_preview", "Tab Jump Flash Lite Preview"),
+            "models/proactive-observer": antigravity_model_payload("models/proactive-observer", "Proactive Observer")
+        },
+        "agentModelSorts": [
+            "gemini-3.5-flash-low",
+            "gemini-3-flash-agent",
+            "gemini-3.1-pro-low",
+            "gemini-3.1-flash-lite"
+        ],
+        "audioTranscriptionModelIds": ["models/proactive-observer"],
+        "commandModelIds": ["gemini-3-flash"],
+        "commitMessageModelIds": ["gemini-3-flash"],
+        "defaultAgentModelId": "gemini-3.5-flash-low",
+        "deprecatedModelIds": [],
+        "experimentIds": [],
+        "imageGenerationModelIds": ["gemini-3.1-flash-image"],
+        "mqueryModelIds": ["gemini-3-flash"],
+        "tabModelIds": ["tab_flash_lite_preview", "tab_jump_flash_lite_preview"],
+        "tieredModelIds": {
+            "flash": "gemini-3-flash-agent",
+            "flashLite": "gemini-3.1-flash-lite",
+            "pro": "gemini-3.1-pro-low"
+        },
+        "webSearchModelIds": ["gemini-3-flash"]
+    })
+}
+
+fn antigravity_model_payload(id: &str, display_name: &str) -> Value {
+    json!({
+        "id": id,
+        "displayName": display_name
+    })
+}
+
 async fn maybe_build_local_gemini_video_operations_response(
    state: &AppState,
    request_context: &GatewayPublicRequestContext,