mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-04 16:37:46 +08:00
Add Antigravity v1internal gateway adapter
This commit is contained in:
@@ -149,7 +149,8 @@ pub fn extract_ai_requested_model_from_request_path(
|
||||
) -> Option<String> {
|
||||
match family {
|
||||
AiRequestedModelFamily::Standard => extract_ai_standard_requested_model(body_json),
|
||||
AiRequestedModelFamily::Gemini => extract_ai_gemini_model_from_path(request_path),
|
||||
AiRequestedModelFamily::Gemini => extract_ai_gemini_model_from_path(request_path)
|
||||
.or_else(|| extract_ai_standard_requested_model(body_json)),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -233,4 +234,15 @@ mod tests {
|
||||
Some("gemini-2.5-pro")
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn gemini_requested_model_parser_uses_body_model_when_path_has_no_model() {
|
||||
let requested_model = extract_ai_requested_model_from_request_path(
|
||||
"/v1internal:streamGenerateContent",
|
||||
&serde_json::json!({ "model": " gemini-cli " }),
|
||||
AiRequestedModelFamily::Gemini,
|
||||
);
|
||||
|
||||
assert_eq!(requested_model.as_deref(), Some("gemini-cli"));
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user