mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-01 17:00:21 +08:00
fix(pool): 修复 Codex 配额倒计时不准确 (#297)
* fix(pool): 修复 Codex 配额倒计时不准确 - 后端号池 keys payload 增加 upstream_metadata 透传 - 前端补充 PoolKeyDetail.upstream_metadata 与 Codex reset_after_seconds 类型 - 号池页倒计时优先使用 reset_at/reset_seconds/updated_at 结构化数据计算 - 仅在缺少结构化字段时回退 account_quota 文案解析 * fix(pool): 将已重置的 Codex 配额恢复为 100% - reset_after_seconds/reset_seconds 会按 updated_at 扣减经过时间(不再当作静态值) - 若窗口已重置(剩余秒数 <= 0),该窗口 used_percent 按 0 处理
This commit is contained in:
@@ -243,29 +243,50 @@ fn admin_pool_build_codex_account_quota(
|
||||
reset_seconds_key: &str,
|
||||
reset_after_seconds_key: &str,
|
||||
reset_at_key: &str,
|
||||
now_unix_secs: u64,
|
||||
updated_at_unix_secs: Option<u64>,
|
||||
) -> Option<f64> {
|
||||
admin_pool_json_to_f64(data.get(reset_seconds_key))
|
||||
.or_else(|| admin_pool_json_to_f64(data.get(reset_after_seconds_key)))
|
||||
.or_else(|| {
|
||||
let reset_at = admin_pool_json_to_u64(data.get(reset_at_key))?;
|
||||
let now_unix_secs = chrono::Utc::now().timestamp().max(0) as u64;
|
||||
Some(reset_at.saturating_sub(now_unix_secs) as f64)
|
||||
})
|
||||
if let Some(reset_at) = admin_pool_json_to_u64(data.get(reset_at_key)) {
|
||||
return Some(reset_at.saturating_sub(now_unix_secs) as f64);
|
||||
}
|
||||
|
||||
let remaining = admin_pool_json_to_f64(data.get(reset_seconds_key))
|
||||
.or_else(|| admin_pool_json_to_f64(data.get(reset_after_seconds_key)))?;
|
||||
let elapsed = updated_at_unix_secs
|
||||
.map(|updated_at| now_unix_secs.saturating_sub(updated_at) as f64)
|
||||
.unwrap_or(0.0);
|
||||
Some((remaining - elapsed).max(0.0))
|
||||
}
|
||||
|
||||
fn codex_effective_used_percent(used_percent: f64, reset_seconds: Option<f64>) -> f64 {
|
||||
let normalized = used_percent.clamp(0.0, 100.0);
|
||||
if normalized <= 1e-6 {
|
||||
return 0.0;
|
||||
}
|
||||
if reset_seconds.is_some_and(|value| value <= 0.0) {
|
||||
return 0.0;
|
||||
}
|
||||
normalized
|
||||
}
|
||||
|
||||
let mut parts = Vec::new();
|
||||
let now_unix_secs = chrono::Utc::now().timestamp().max(0) as u64;
|
||||
let updated_at_unix_secs = admin_pool_json_to_u64(data.get("updated_at"));
|
||||
|
||||
let primary_used = admin_pool_json_to_f64(data.get("primary_used_percent"));
|
||||
if let Some(primary_used) = primary_used {
|
||||
let primary_used_raw = admin_pool_json_to_f64(data.get("primary_used_percent"));
|
||||
if let Some(primary_used_raw) = primary_used_raw {
|
||||
let primary_reset_seconds = codex_reset_seconds(
|
||||
data,
|
||||
"primary_reset_seconds",
|
||||
"primary_reset_after_seconds",
|
||||
"primary_reset_at",
|
||||
now_unix_secs,
|
||||
updated_at_unix_secs,
|
||||
);
|
||||
let primary_used = codex_effective_used_percent(primary_used_raw, primary_reset_seconds);
|
||||
let mut part = format!("周剩余 {}", admin_pool_format_percent(100.0 - primary_used));
|
||||
if admin_pool_has_quota_consumption(Some(primary_used)) {
|
||||
if let Some(reset_text) = codex_reset_seconds(
|
||||
data,
|
||||
"primary_reset_seconds",
|
||||
"primary_reset_after_seconds",
|
||||
"primary_reset_at",
|
||||
)
|
||||
.and_then(admin_pool_format_reset_after)
|
||||
if let Some(reset_text) = primary_reset_seconds.and_then(admin_pool_format_reset_after)
|
||||
{
|
||||
part.push_str(&format!(" ({reset_text})"));
|
||||
}
|
||||
@@ -273,20 +294,25 @@ fn admin_pool_build_codex_account_quota(
|
||||
parts.push(part);
|
||||
}
|
||||
|
||||
let secondary_used = admin_pool_json_to_f64(data.get("secondary_used_percent"));
|
||||
if let Some(secondary_used) = secondary_used {
|
||||
let secondary_used_raw = admin_pool_json_to_f64(data.get("secondary_used_percent"));
|
||||
if let Some(secondary_used_raw) = secondary_used_raw {
|
||||
let secondary_reset_seconds = codex_reset_seconds(
|
||||
data,
|
||||
"secondary_reset_seconds",
|
||||
"secondary_reset_after_seconds",
|
||||
"secondary_reset_at",
|
||||
now_unix_secs,
|
||||
updated_at_unix_secs,
|
||||
);
|
||||
let secondary_used =
|
||||
codex_effective_used_percent(secondary_used_raw, secondary_reset_seconds);
|
||||
let mut part = format!(
|
||||
"5H剩余 {}",
|
||||
admin_pool_format_percent(100.0 - secondary_used)
|
||||
);
|
||||
if admin_pool_has_quota_consumption(Some(secondary_used)) {
|
||||
if let Some(reset_text) = codex_reset_seconds(
|
||||
data,
|
||||
"secondary_reset_seconds",
|
||||
"secondary_reset_after_seconds",
|
||||
"secondary_reset_at",
|
||||
)
|
||||
.and_then(admin_pool_format_reset_after)
|
||||
if let Some(reset_text) =
|
||||
secondary_reset_seconds.and_then(admin_pool_format_reset_after)
|
||||
{
|
||||
part.push_str(&format!(" ({reset_text})"));
|
||||
}
|
||||
@@ -754,6 +780,10 @@ pub(super) fn build_admin_pool_key_payload(
|
||||
"model_exclude_patterns".to_string(),
|
||||
json!(admin_pool_string_list(key.model_exclude_patterns.as_ref())),
|
||||
);
|
||||
payload.insert(
|
||||
"upstream_metadata".to_string(),
|
||||
json!(key.upstream_metadata.clone()),
|
||||
);
|
||||
payload.insert("proxy".to_string(), json!(key.proxy.clone()));
|
||||
payload.insert("fingerprint".to_string(), json!(key.fingerprint.clone()));
|
||||
payload.insert(
|
||||
|
||||
@@ -1067,7 +1067,7 @@ async fn gateway_formats_codex_quota_countdown_from_reset_after_seconds() {
|
||||
key.upstream_metadata = Some(json!({
|
||||
"codex": {
|
||||
"plan_type": "plus",
|
||||
"updated_at": 1_775_553_285u64,
|
||||
"updated_at": 4_102_444_800u64,
|
||||
"primary_used_percent": 10.0,
|
||||
"primary_reset_after_seconds": 266_400,
|
||||
"secondary_used_percent": 33.0,
|
||||
@@ -1109,6 +1109,75 @@ async fn gateway_formats_codex_quota_countdown_from_reset_after_seconds() {
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn gateway_codex_quota_resets_to_full_after_countdown_elapsed() {
|
||||
let mut provider = sample_provider("provider-codex", "codex", 10).with_transport_fields(
|
||||
true,
|
||||
false,
|
||||
true,
|
||||
None,
|
||||
None,
|
||||
None,
|
||||
None,
|
||||
None,
|
||||
Some(json!({
|
||||
"pool_advanced": {
|
||||
"enabled": true
|
||||
}
|
||||
})),
|
||||
);
|
||||
provider.provider_type = "codex".to_string();
|
||||
|
||||
let mut key = sample_key(
|
||||
"key-codex-expired",
|
||||
"provider-codex",
|
||||
"openai:cli",
|
||||
"oauth-placeholder",
|
||||
);
|
||||
key.name = "codex expired quota key".to_string();
|
||||
key.auth_type = "oauth".to_string();
|
||||
key.upstream_metadata = Some(json!({
|
||||
"codex": {
|
||||
"plan_type": "plus",
|
||||
"updated_at": 1u64,
|
||||
"primary_used_percent": 42.0,
|
||||
"primary_reset_after_seconds": 60,
|
||||
"secondary_used_percent": 77.0,
|
||||
"secondary_reset_after_seconds": 120
|
||||
}
|
||||
}));
|
||||
|
||||
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-codex/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.len(), 1);
|
||||
assert_eq!(keys[0]["account_quota"], "周剩余 100.0% | 5H剩余 100.0%");
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn gateway_pool_prefers_upstream_plan_type_over_auth_config() {
|
||||
let mut provider = sample_provider("provider-codex", "codex", 10).with_transport_fields(
|
||||
|
||||
Reference in New Issue
Block a user