Add Gemini CLI v1internal quota support

This commit is contained in:
Mas0nShi
2026-05-22 16:58:04 +08:00
parent 9533bd7043
commit ce02f1ae8c
17 changed files with 985 additions and 52 deletions
@@ -219,6 +219,23 @@ fn classifies_gemini_generate_content_api_key_without_cli_marker() {
assert!(decision.is_execution_runtime_candidate());
}
#[test]
fn does_not_classify_gemini_options_preflight_as_ai_execution_route() {
let headers = headers(&[
("origin", "http://localhost:3000"),
("access-control-request-method", "POST"),
(
"access-control-request-headers",
"authorization,content-type",
),
]);
let uri: Uri = "/v1beta/models/gemini-3-flash-preview:streamGenerateContent?alt=sse"
.parse()
.expect("uri should parse");
assert!(classify_control_route(&http::Method::OPTIONS, &uri, &headers).is_none());
}
#[test]
fn classifies_gemini_embed_content_as_embedding_route() {
let headers = headers(&[("x-goog-api-key", "gemini-key")]);