Add Gemini CLI v1internal quota support

2026-10-06 01:17:46 +08:00 · 2026-05-22 16:58:04 +08:00
parent 9533bd7043
commit ce02f1ae8c
17 changed files with 985 additions and 52 deletions
@@ -2085,6 +2085,99 @@ async fn gateway_renders_gemini_cli_account_quota_from_status_snapshot() {
    assert_eq!(keys[0]["account_quota"], json!("Gemini 2.5 Pro 冷却中"));
 }

+#[tokio::test]
+async fn gateway_prefers_gemini_cli_account_credits_over_model_quota_text() {
+    let mut provider = sample_provider("provider-gemini-cli", "gemini_cli", 10)
+        .with_transport_fields(
+            true,
+            false,
+            true,
+            None,
+            None,
+            None,
+            None,
+            None,
+            Some(json!({
+                "pool_advanced": {
+                    "enabled": true,
+                    "skip_exhausted_accounts": true
+                }
+            })),
+        );
+    provider.provider_type = "gemini_cli".to_string();
+
+    let mut key = sample_key(
+        "key-gemini-cli-credits",
+        "provider-gemini-cli",
+        "gemini:generate_content",
+        "oauth-placeholder",
+    );
+    key.name = "gemini cli credits".to_string();
+    key.auth_type = "oauth".to_string();
+    key.status_snapshot = Some(json!({
+        "quota": {
+            "version": 2,
+            "provider_type": "gemini_cli",
+            "code": "ok",
+            "freshness": "fresh",
+            "source": "report_effect",
+            "observed_at": 1_775_553_285u64,
+            "exhausted": false,
+            "usage_ratio": 0.25,
+            "updated_at": 1_775_553_285u64,
+            "plan_type": "g1-pro-tier",
+            "credits": {
+                "remaining": 123.5,
+                "consumed": 7.0,
+                "has_credits": true
+            },
+            "windows": [
+                {
+                    "code": "model:gemini-2.5-pro",
+                    "label": "Gemini 2.5 Pro",
+                    "scope": "model",
+                    "unit": "percent",
+                    "model": "gemini-2.5-pro",
+                    "used_ratio": 0.25,
+                    "remaining_ratio": 0.75,
+                    "is_exhausted": false
+                }
+            ]
+        }
+    }));
+
+    let provider_catalog_repository = Arc::new(InMemoryProviderCatalogReadRepository::seed(
+        vec![provider],
+        Vec::new(),
+        vec![key],
+    ));
+    let state = AppState::new()
+        .expect("gateway should build")
+        .with_data_state_for_tests(GatewayDataState::with_provider_catalog_reader_for_tests(
+            provider_catalog_repository,
+        ));
+
+    let response = local_admin_pool_response(
+        &state,
+        http::Method::GET,
+        "/api/admin/pool/provider-gemini-cli/keys?page=1&page_size=50&status=all",
+        None,
+    )
+    .await;
+
+    assert_eq!(response.status(), StatusCode::OK);
+    let payload: serde_json::Value = serde_json::from_slice(
+        &to_bytes(response.into_body(), usize::MAX)
+            .await
+            .expect("body should read"),
+    )
+    .expect("json body should parse");
+    let keys = payload["keys"].as_array().expect("keys should be array");
+
+    assert_eq!(keys[0]["scheduling_status"], json!("available"));
+    assert_eq!(keys[0]["account_quota"], json!("AI Credits 剩余 123.5"));
+}
+
 #[tokio::test]
 async fn gateway_formats_codex_quota_countdown_from_reset_after_seconds() {
    let mut provider = sample_provider("provider-codex", "codex", 10).with_transport_fields(