修复 Antigravity v1internal 查询参数透传

This commit is contained in:
MMEXA
2026-07-09 16:03:29 +08:00
parent e50db10439
commit ee0ec18283
2 changed files with 32 additions and 1 deletions
@@ -41,7 +41,11 @@ pub fn build_antigravity_v1internal_url(
for (key, value) in query {
let key = key.trim();
let value = value.trim();
if key.is_empty() || value.is_empty() || key.eq_ignore_ascii_case("beta") {
if key.is_empty()
|| value.is_empty()
|| key.eq_ignore_ascii_case("beta")
|| key.eq_ignore_ascii_case("key")
{
continue;
}
params.insert(key.to_string(), value.to_string());
@@ -1196,6 +1196,33 @@ mod tests {
);
}
#[test]
fn antigravity_generate_content_drops_client_api_key_query() {
let transport = sample_transport(
"antigravity",
"gemini:generate_content",
"https://daily-cloudcode-pa.googleapis.com",
None,
);
assert_eq!(
build_transport_request_url(
&transport,
TransportRequestUrlParams {
provider_api_format: "gemini:generate_content",
mapped_model: Some("gemini-3-flash-agent"),
upstream_is_stream: true,
request_query: Some("key=client-aether-key&trace=1&beta=true"),
kiro_api_region: None,
},
)
.as_deref(),
Some(
"https://daily-cloudcode-pa.googleapis.com/v1internal:streamGenerateContent?alt=sse&trace=1"
)
);
}
#[test]
fn embedding_request_url_preserves_google_openai_compat_roots() {
let developer_api_openai = sample_transport(