mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-11 19:59:50 +08:00
Merge branch 'pr-390' into aether-rust-pioneer
This commit is contained in:
@@ -30,8 +30,9 @@ use super::{
|
|||||||
WalletLookupKey, WalletMutationOutcome,
|
WalletLookupKey, WalletMutationOutcome,
|
||||||
};
|
};
|
||||||
use aether_data_contracts::repository::usage::{
|
use aether_data_contracts::repository::usage::{
|
||||||
PendingUsageCleanupSummary, StoredUsageDailySummary, UsageAuditListQuery, UsageCleanupSummary,
|
PendingUsageCleanupSummary, ProviderApiKeyWindowUsageRequest,
|
||||||
UsageCleanupWindow, UsageDailyHeatmapQuery,
|
StoredProviderApiKeyWindowUsageSummary, StoredUsageDailySummary, UsageAuditListQuery,
|
||||||
|
UsageCleanupSummary, UsageCleanupWindow, UsageDailyHeatmapQuery,
|
||||||
};
|
};
|
||||||
use aether_video_tasks_core::read_data_backed_video_task_response;
|
use aether_video_tasks_core::read_data_backed_video_task_response;
|
||||||
|
|
||||||
@@ -1366,6 +1367,20 @@ impl GatewayDataState {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(crate) async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<Vec<StoredProviderApiKeyWindowUsageSummary>, DataLayerError> {
|
||||||
|
match &self.usage_reader {
|
||||||
|
Some(repository) => {
|
||||||
|
repository
|
||||||
|
.summarize_usage_by_provider_api_key_windows(requests)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
None => Ok(Vec::new()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub(crate) async fn list_users_by_ids(
|
pub(crate) async fn list_users_by_ids(
|
||||||
&self,
|
&self,
|
||||||
user_ids: &[String],
|
user_ids: &[String],
|
||||||
|
|||||||
@@ -9,7 +9,11 @@ use aether_admin::provider::quota as admin_provider_quota_pure;
|
|||||||
use aether_data_contracts::repository::provider_catalog::{
|
use aether_data_contracts::repository::provider_catalog::{
|
||||||
StoredProviderCatalogEndpoint, StoredProviderCatalogKey,
|
StoredProviderCatalogEndpoint, StoredProviderCatalogKey,
|
||||||
};
|
};
|
||||||
|
use aether_data_contracts::repository::usage::{
|
||||||
|
ProviderApiKeyWindowUsageRequest, StoredProviderApiKeyWindowUsageSummary,
|
||||||
|
};
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
|
use std::collections::BTreeMap;
|
||||||
|
|
||||||
fn admin_pool_string_list(value: Option<&serde_json::Value>) -> Option<Vec<String>> {
|
fn admin_pool_string_list(value: Option<&serde_json::Value>) -> Option<Vec<String>> {
|
||||||
let values = value
|
let values = value
|
||||||
@@ -289,6 +293,124 @@ fn admin_pool_quota_window<'a>(
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(super) type AdminPoolCodexWindowUsageByKey =
|
||||||
|
BTreeMap<(String, String), StoredProviderApiKeyWindowUsageSummary>;
|
||||||
|
|
||||||
|
fn admin_pool_provider_type_is_codex(provider_type: &str) -> bool {
|
||||||
|
provider_type.trim().eq_ignore_ascii_case("codex")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn admin_pool_codex_window_usage_code(
|
||||||
|
window: &serde_json::Map<String, serde_json::Value>,
|
||||||
|
) -> Option<&'static str> {
|
||||||
|
let code = window
|
||||||
|
.get("code")
|
||||||
|
.and_then(serde_json::Value::as_str)
|
||||||
|
.map(str::trim)?;
|
||||||
|
if code.eq_ignore_ascii_case("5h") {
|
||||||
|
Some("5h")
|
||||||
|
} else if code.eq_ignore_ascii_case("weekly") {
|
||||||
|
Some("weekly")
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn admin_pool_codex_window_usage_bounds(
|
||||||
|
window: &serde_json::Map<String, serde_json::Value>,
|
||||||
|
) -> Option<(u64, u64)> {
|
||||||
|
let reset_at = admin_pool_json_to_u64(window.get("reset_at"))?;
|
||||||
|
let window_minutes = admin_pool_json_to_u64(window.get("window_minutes"))?;
|
||||||
|
let window_seconds = window_minutes.checked_mul(60)?;
|
||||||
|
let start = reset_at.checked_sub(window_seconds)?;
|
||||||
|
(start < reset_at).then_some((start, reset_at))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) fn build_admin_pool_codex_window_usage_requests(
|
||||||
|
provider_type: &str,
|
||||||
|
keys: &[StoredProviderCatalogKey],
|
||||||
|
) -> Vec<ProviderApiKeyWindowUsageRequest> {
|
||||||
|
if !admin_pool_provider_type_is_codex(provider_type) {
|
||||||
|
return Vec::new();
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut requests = Vec::new();
|
||||||
|
for key in keys {
|
||||||
|
let status_snapshot = provider_key_status_snapshot_payload(key, provider_type);
|
||||||
|
let Some(windows) = status_snapshot
|
||||||
|
.get("quota")
|
||||||
|
.and_then(serde_json::Value::as_object)
|
||||||
|
.and_then(|quota| quota.get("windows"))
|
||||||
|
.and_then(serde_json::Value::as_array)
|
||||||
|
else {
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
|
for window in windows.iter().filter_map(serde_json::Value::as_object) {
|
||||||
|
let Some(window_code) = admin_pool_codex_window_usage_code(window) else {
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
let Some((start_unix_secs, end_unix_secs)) =
|
||||||
|
admin_pool_codex_window_usage_bounds(window)
|
||||||
|
else {
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
requests.push(ProviderApiKeyWindowUsageRequest {
|
||||||
|
provider_api_key_id: key.id.clone(),
|
||||||
|
window_code: window_code.to_string(),
|
||||||
|
start_unix_secs,
|
||||||
|
end_unix_secs,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
requests
|
||||||
|
}
|
||||||
|
|
||||||
|
fn admin_pool_codex_window_usage_payload(
|
||||||
|
usage: &StoredProviderApiKeyWindowUsageSummary,
|
||||||
|
) -> serde_json::Value {
|
||||||
|
json!({
|
||||||
|
"request_count": usage.request_count,
|
||||||
|
"total_tokens": usage.total_tokens,
|
||||||
|
"total_cost_usd": format!("{:.8}", usage.total_cost_usd),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn admin_pool_attach_codex_window_usage(
|
||||||
|
status_snapshot: &mut serde_json::Value,
|
||||||
|
key_id: &str,
|
||||||
|
usage_by_key: &AdminPoolCodexWindowUsageByKey,
|
||||||
|
) {
|
||||||
|
if usage_by_key.is_empty() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let Some(windows) = status_snapshot
|
||||||
|
.get_mut("quota")
|
||||||
|
.and_then(serde_json::Value::as_object_mut)
|
||||||
|
.and_then(|quota| quota.get_mut("windows"))
|
||||||
|
.and_then(serde_json::Value::as_array_mut)
|
||||||
|
else {
|
||||||
|
return;
|
||||||
|
};
|
||||||
|
|
||||||
|
for window in windows
|
||||||
|
.iter_mut()
|
||||||
|
.filter_map(serde_json::Value::as_object_mut)
|
||||||
|
{
|
||||||
|
let Some(window_code) = admin_pool_codex_window_usage_code(window) else {
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
let lookup_key = (key_id.to_string(), window_code.to_string());
|
||||||
|
if let Some(usage) = usage_by_key.get(&lookup_key) {
|
||||||
|
window.insert(
|
||||||
|
"usage".to_string(),
|
||||||
|
admin_pool_codex_window_usage_payload(usage),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn admin_pool_quota_window_used_percent(
|
fn admin_pool_quota_window_used_percent(
|
||||||
window: &serde_json::Map<String, serde_json::Value>,
|
window: &serde_json::Map<String, serde_json::Value>,
|
||||||
) -> Option<f64> {
|
) -> Option<f64> {
|
||||||
@@ -804,6 +926,7 @@ pub(super) fn build_admin_pool_key_payload(
|
|||||||
key: &StoredProviderCatalogKey,
|
key: &StoredProviderCatalogKey,
|
||||||
runtime: &AdminProviderPoolRuntimeState,
|
runtime: &AdminProviderPoolRuntimeState,
|
||||||
pool_config: Option<AdminProviderPoolConfig>,
|
pool_config: Option<AdminProviderPoolConfig>,
|
||||||
|
codex_window_usage_by_key: &AdminPoolCodexWindowUsageByKey,
|
||||||
) -> serde_json::Value {
|
) -> serde_json::Value {
|
||||||
let cooldown_reason = runtime.cooldown_reason_by_key.get(&key.id).cloned();
|
let cooldown_reason = runtime.cooldown_reason_by_key.get(&key.id).cloned();
|
||||||
let cooldown_ttl_seconds = cooldown_reason
|
let cooldown_ttl_seconds = cooldown_reason
|
||||||
@@ -821,7 +944,14 @@ pub(super) fn build_admin_pool_key_payload(
|
|||||||
admin_pool_derive_oauth_expires_at(provider_type, key, auth_config.as_ref());
|
admin_pool_derive_oauth_expires_at(provider_type, key, auth_config.as_ref());
|
||||||
let oauth_plan_type =
|
let oauth_plan_type =
|
||||||
admin_pool_derive_oauth_plan_type(key, provider_type, auth_config.as_ref());
|
admin_pool_derive_oauth_plan_type(key, provider_type, auth_config.as_ref());
|
||||||
let status_snapshot = provider_key_status_snapshot_payload(key, provider_type);
|
let mut status_snapshot = provider_key_status_snapshot_payload(key, provider_type);
|
||||||
|
if admin_pool_provider_type_is_codex(provider_type) {
|
||||||
|
admin_pool_attach_codex_window_usage(
|
||||||
|
&mut status_snapshot,
|
||||||
|
&key.id,
|
||||||
|
codex_window_usage_by_key,
|
||||||
|
);
|
||||||
|
}
|
||||||
let account_snapshot = status_snapshot
|
let account_snapshot = status_snapshot
|
||||||
.get("account")
|
.get("account")
|
||||||
.and_then(serde_json::Value::as_object);
|
.and_then(serde_json::Value::as_object);
|
||||||
|
|||||||
@@ -253,6 +253,26 @@ pub(super) async fn build_admin_pool_list_keys_response(
|
|||||||
}
|
}
|
||||||
_ => AdminProviderPoolRuntimeState::default(),
|
_ => AdminProviderPoolRuntimeState::default(),
|
||||||
};
|
};
|
||||||
|
let codex_window_usage_requests =
|
||||||
|
pool_payloads::build_admin_pool_codex_window_usage_requests(&provider.provider_type, &keys);
|
||||||
|
let codex_window_usage_by_key: pool_payloads::AdminPoolCodexWindowUsageByKey =
|
||||||
|
if codex_window_usage_requests.is_empty() {
|
||||||
|
pool_payloads::AdminPoolCodexWindowUsageByKey::new()
|
||||||
|
} else {
|
||||||
|
state
|
||||||
|
.app()
|
||||||
|
.summarize_usage_by_provider_api_key_windows(&codex_window_usage_requests)
|
||||||
|
.await?
|
||||||
|
.into_iter()
|
||||||
|
.map(|usage| {
|
||||||
|
(
|
||||||
|
(usage.provider_api_key_id.clone(), usage.window_code.clone()),
|
||||||
|
usage,
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
};
|
||||||
|
|
||||||
let items = keys
|
let items = keys
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|key| {
|
.map(|key| {
|
||||||
@@ -263,6 +283,7 @@ pub(super) async fn build_admin_pool_list_keys_response(
|
|||||||
&key,
|
&key,
|
||||||
&runtime,
|
&runtime,
|
||||||
pool_config.clone(),
|
pool_config.clone(),
|
||||||
|
&codex_window_usage_by_key,
|
||||||
)
|
)
|
||||||
})
|
})
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
|
|||||||
@@ -345,6 +345,16 @@ impl AppState {
|
|||||||
.map_err(|err| GatewayError::Internal(err.to_string()))
|
.map_err(|err| GatewayError::Internal(err.to_string()))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub(crate) async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[usage::ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<Vec<usage::StoredProviderApiKeyWindowUsageSummary>, GatewayError> {
|
||||||
|
self.data
|
||||||
|
.summarize_usage_by_provider_api_key_windows(requests)
|
||||||
|
.await
|
||||||
|
.map_err(|err| GatewayError::Internal(err.to_string()))
|
||||||
|
}
|
||||||
|
|
||||||
pub(crate) async fn list_users_by_ids(
|
pub(crate) async fn list_users_by_ids(
|
||||||
&self,
|
&self,
|
||||||
user_ids: &[String],
|
user_ids: &[String],
|
||||||
|
|||||||
@@ -2,7 +2,9 @@ use std::sync::{Arc, Mutex};
|
|||||||
|
|
||||||
use aether_crypto::{encrypt_python_fernet_plaintext, DEVELOPMENT_ENCRYPTION_KEY};
|
use aether_crypto::{encrypt_python_fernet_plaintext, DEVELOPMENT_ENCRYPTION_KEY};
|
||||||
use aether_data::repository::provider_catalog::InMemoryProviderCatalogReadRepository;
|
use aether_data::repository::provider_catalog::InMemoryProviderCatalogReadRepository;
|
||||||
|
use aether_data::repository::usage::InMemoryUsageReadRepository;
|
||||||
use aether_data_contracts::repository::provider_catalog::ProviderCatalogReadRepository;
|
use aether_data_contracts::repository::provider_catalog::ProviderCatalogReadRepository;
|
||||||
|
use aether_data_contracts::repository::usage::StoredRequestUsageAudit;
|
||||||
use axum::body::{to_bytes, Body, Bytes};
|
use axum::body::{to_bytes, Body, Bytes};
|
||||||
use axum::routing::{any, get, post};
|
use axum::routing::{any, get, post};
|
||||||
use axum::{extract::Request, Router};
|
use axum::{extract::Request, Router};
|
||||||
@@ -21,6 +23,54 @@ use crate::constants::{
|
|||||||
use crate::control::resolve_public_request_context;
|
use crate::control::resolve_public_request_context;
|
||||||
use crate::data::GatewayDataState;
|
use crate::data::GatewayDataState;
|
||||||
|
|
||||||
|
fn sample_pool_usage_row(
|
||||||
|
request_id: &str,
|
||||||
|
provider_api_key_id: &str,
|
||||||
|
created_at_unix_secs: i64,
|
||||||
|
total_tokens: i32,
|
||||||
|
total_cost_usd: f64,
|
||||||
|
) -> StoredRequestUsageAudit {
|
||||||
|
StoredRequestUsageAudit::new(
|
||||||
|
format!("usage-{request_id}"),
|
||||||
|
request_id.to_string(),
|
||||||
|
Some("user-codex".to_string()),
|
||||||
|
Some("api-key-codex".to_string()),
|
||||||
|
Some("codex-user".to_string()),
|
||||||
|
Some("codex-api-key".to_string()),
|
||||||
|
"codex".to_string(),
|
||||||
|
"gpt-5-codex".to_string(),
|
||||||
|
None,
|
||||||
|
Some("provider-codex".to_string()),
|
||||||
|
Some("endpoint-codex".to_string()),
|
||||||
|
Some(provider_api_key_id.to_string()),
|
||||||
|
Some("responses".to_string()),
|
||||||
|
Some("openai:responses".to_string()),
|
||||||
|
Some("openai".to_string()),
|
||||||
|
Some("responses".to_string()),
|
||||||
|
Some("openai:responses".to_string()),
|
||||||
|
Some("openai".to_string()),
|
||||||
|
Some("responses".to_string()),
|
||||||
|
false,
|
||||||
|
false,
|
||||||
|
total_tokens,
|
||||||
|
0,
|
||||||
|
total_tokens,
|
||||||
|
total_cost_usd,
|
||||||
|
total_cost_usd,
|
||||||
|
Some(200),
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
Some(240),
|
||||||
|
Some(80),
|
||||||
|
"completed".to_string(),
|
||||||
|
"settled".to_string(),
|
||||||
|
created_at_unix_secs,
|
||||||
|
created_at_unix_secs + 1,
|
||||||
|
Some(created_at_unix_secs + 2),
|
||||||
|
)
|
||||||
|
.expect("usage row should build")
|
||||||
|
}
|
||||||
|
|
||||||
fn trusted_admin_headers() -> HeaderMap {
|
fn trusted_admin_headers() -> HeaderMap {
|
||||||
let mut headers = HeaderMap::new();
|
let mut headers = HeaderMap::new();
|
||||||
headers.insert(GATEWAY_HEADER, HeaderValue::from_static("rust-phase3b"));
|
headers.insert(GATEWAY_HEADER, HeaderValue::from_static("rust-phase3b"));
|
||||||
@@ -793,6 +843,233 @@ async fn gateway_sorts_admin_pool_keys_by_imported_and_last_used_time() {
|
|||||||
assert_eq!(last_used_names, vec!["active", "old", "fresh"]);
|
assert_eq!(last_used_names, vec!["active", "old", "fresh"]);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn gateway_pool_list_adds_codex_cycle_usage_to_quota_windows() {
|
||||||
|
const RESET_AT: u64 = 1_711_000_000;
|
||||||
|
|
||||||
|
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 usage_key = sample_key(
|
||||||
|
"key-codex-cycle",
|
||||||
|
"provider-codex",
|
||||||
|
"openai:responses",
|
||||||
|
"oauth-placeholder",
|
||||||
|
);
|
||||||
|
usage_key.name = "codex cycle usage".to_string();
|
||||||
|
usage_key.auth_type = "oauth".to_string();
|
||||||
|
usage_key.request_count = Some(4);
|
||||||
|
usage_key.total_tokens = 999;
|
||||||
|
usage_key.total_cost_usd = 9.99;
|
||||||
|
usage_key.status_snapshot = Some(json!({
|
||||||
|
"quota": {
|
||||||
|
"version": 2,
|
||||||
|
"provider_type": "codex",
|
||||||
|
"code": "ok",
|
||||||
|
"label": serde_json::Value::Null,
|
||||||
|
"reason": serde_json::Value::Null,
|
||||||
|
"freshness": "fresh",
|
||||||
|
"source": "response_headers",
|
||||||
|
"observed_at": RESET_AT,
|
||||||
|
"exhausted": false,
|
||||||
|
"usage_ratio": 0.0,
|
||||||
|
"updated_at": RESET_AT,
|
||||||
|
"reset_seconds": serde_json::Value::Null,
|
||||||
|
"plan_type": "plus",
|
||||||
|
"windows": [
|
||||||
|
{
|
||||||
|
"code": "weekly",
|
||||||
|
"label": "周",
|
||||||
|
"scope": "account",
|
||||||
|
"unit": "percent",
|
||||||
|
"used_ratio": 0.0,
|
||||||
|
"remaining_ratio": 1.0,
|
||||||
|
"reset_at": RESET_AT,
|
||||||
|
"reset_seconds": 604_800,
|
||||||
|
"window_minutes": 10_080
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "5h",
|
||||||
|
"label": "5H",
|
||||||
|
"scope": "account",
|
||||||
|
"unit": "percent",
|
||||||
|
"used_ratio": 0.0,
|
||||||
|
"remaining_ratio": 1.0,
|
||||||
|
"reset_at": RESET_AT,
|
||||||
|
"reset_seconds": 18_000,
|
||||||
|
"window_minutes": 300
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
|
||||||
|
let mut zero_key = sample_key(
|
||||||
|
"key-codex-zero",
|
||||||
|
"provider-codex",
|
||||||
|
"openai:responses",
|
||||||
|
"oauth-placeholder",
|
||||||
|
);
|
||||||
|
zero_key.name = "codex zero usage".to_string();
|
||||||
|
zero_key.auth_type = "oauth".to_string();
|
||||||
|
zero_key.status_snapshot = usage_key.status_snapshot.clone();
|
||||||
|
|
||||||
|
let mut invalid_key = sample_key(
|
||||||
|
"key-codex-invalid",
|
||||||
|
"provider-codex",
|
||||||
|
"openai:responses",
|
||||||
|
"oauth-placeholder",
|
||||||
|
);
|
||||||
|
invalid_key.name = "codex invalid window".to_string();
|
||||||
|
invalid_key.auth_type = "oauth".to_string();
|
||||||
|
invalid_key.status_snapshot = Some(json!({
|
||||||
|
"quota": {
|
||||||
|
"version": 2,
|
||||||
|
"provider_type": "codex",
|
||||||
|
"code": "ok",
|
||||||
|
"windows": [
|
||||||
|
{
|
||||||
|
"code": "weekly",
|
||||||
|
"label": "周",
|
||||||
|
"reset_at": serde_json::Value::Null,
|
||||||
|
"window_minutes": 10_080
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "5h",
|
||||||
|
"label": "5H",
|
||||||
|
"reset_at": RESET_AT,
|
||||||
|
"window_minutes": serde_json::Value::Null
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
|
||||||
|
let provider_catalog_repository = Arc::new(InMemoryProviderCatalogReadRepository::seed(
|
||||||
|
vec![provider],
|
||||||
|
Vec::new(),
|
||||||
|
vec![usage_key, zero_key, invalid_key],
|
||||||
|
));
|
||||||
|
let usage_repository = Arc::new(InMemoryUsageReadRepository::seed(vec![
|
||||||
|
sample_pool_usage_row(
|
||||||
|
"req-5h-a",
|
||||||
|
"key-codex-cycle",
|
||||||
|
RESET_AT as i64 - 60,
|
||||||
|
100,
|
||||||
|
0.10,
|
||||||
|
),
|
||||||
|
sample_pool_usage_row(
|
||||||
|
"req-5h-b",
|
||||||
|
"key-codex-cycle",
|
||||||
|
RESET_AT as i64 - 17_999,
|
||||||
|
125,
|
||||||
|
0.20,
|
||||||
|
),
|
||||||
|
sample_pool_usage_row(
|
||||||
|
"req-weekly-only",
|
||||||
|
"key-codex-cycle",
|
||||||
|
RESET_AT as i64 - 18_001,
|
||||||
|
150,
|
||||||
|
0.30,
|
||||||
|
),
|
||||||
|
sample_pool_usage_row(
|
||||||
|
"req-before-weekly",
|
||||||
|
"key-codex-cycle",
|
||||||
|
RESET_AT as i64 - 604_801,
|
||||||
|
200,
|
||||||
|
0.40,
|
||||||
|
),
|
||||||
|
]));
|
||||||
|
let state = AppState::new()
|
||||||
|
.expect("gateway should build")
|
||||||
|
.with_data_state_for_tests(
|
||||||
|
GatewayDataState::with_provider_catalog_and_usage_reader_for_tests(
|
||||||
|
provider_catalog_repository,
|
||||||
|
usage_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");
|
||||||
|
fn key_by_id<'a>(keys: &'a [serde_json::Value], key_id: &str) -> &'a serde_json::Value {
|
||||||
|
keys.iter()
|
||||||
|
.find(|key| key["key_id"] == json!(key_id))
|
||||||
|
.expect("key payload should exist")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn window_by_code<'a>(key_payload: &'a serde_json::Value, code: &str) -> &'a serde_json::Value {
|
||||||
|
key_payload["status_snapshot"]["quota"]["windows"]
|
||||||
|
.as_array()
|
||||||
|
.expect("quota windows should be array")
|
||||||
|
.iter()
|
||||||
|
.find(|window| window["code"] == json!(code))
|
||||||
|
.expect("quota window should exist")
|
||||||
|
}
|
||||||
|
|
||||||
|
let usage_key_payload = key_by_id(keys, "key-codex-cycle");
|
||||||
|
let five_hour_window = window_by_code(usage_key_payload, "5h");
|
||||||
|
let weekly_window = window_by_code(usage_key_payload, "weekly");
|
||||||
|
assert_eq!(five_hour_window["usage"]["request_count"], json!(2));
|
||||||
|
assert_eq!(five_hour_window["usage"]["total_tokens"], json!(225));
|
||||||
|
assert_eq!(
|
||||||
|
five_hour_window["usage"]["total_cost_usd"],
|
||||||
|
json!("0.30000000")
|
||||||
|
);
|
||||||
|
assert_eq!(weekly_window["usage"]["request_count"], json!(3));
|
||||||
|
assert_eq!(weekly_window["usage"]["total_tokens"], json!(375));
|
||||||
|
assert_eq!(
|
||||||
|
weekly_window["usage"]["total_cost_usd"],
|
||||||
|
json!("0.60000000")
|
||||||
|
);
|
||||||
|
assert_eq!(usage_key_payload["request_count"], json!(4));
|
||||||
|
assert_eq!(usage_key_payload["total_tokens"], json!(999));
|
||||||
|
assert_eq!(usage_key_payload["total_cost_usd"], json!("9.99000000"));
|
||||||
|
|
||||||
|
let zero_key_payload = key_by_id(keys, "key-codex-zero");
|
||||||
|
assert_eq!(
|
||||||
|
window_by_code(zero_key_payload, "5h")["usage"]["request_count"],
|
||||||
|
json!(0)
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
window_by_code(zero_key_payload, "weekly")["usage"]["total_tokens"],
|
||||||
|
json!(0)
|
||||||
|
);
|
||||||
|
|
||||||
|
let invalid_key_payload = key_by_id(keys, "key-codex-invalid");
|
||||||
|
assert!(window_by_code(invalid_key_payload, "5h")
|
||||||
|
.get("usage")
|
||||||
|
.is_none());
|
||||||
|
assert!(window_by_code(invalid_key_payload, "weekly")
|
||||||
|
.get("usage")
|
||||||
|
.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn gateway_marks_account_blocked_pool_key_in_list_keys_response() {
|
async fn gateway_marks_account_blocked_pool_key_in_list_keys_response() {
|
||||||
let mut provider = sample_provider("provider-codex", "codex", 10).with_transport_fields(
|
let mut provider = sample_provider("provider-codex", "codex", 10).with_transport_fields(
|
||||||
|
|||||||
@@ -2,7 +2,8 @@ mod types;
|
|||||||
|
|
||||||
pub use types::{
|
pub use types::{
|
||||||
parse_usage_body_ref, usage_body_ref, PendingUsageCleanupSummary,
|
parse_usage_body_ref, usage_body_ref, PendingUsageCleanupSummary,
|
||||||
StoredProviderApiKeyUsageSummary, StoredProviderUsageSummary, StoredProviderUsageWindow,
|
ProviderApiKeyWindowUsageRequest, StoredProviderApiKeyUsageSummary,
|
||||||
|
StoredProviderApiKeyWindowUsageSummary, StoredProviderUsageSummary, StoredProviderUsageWindow,
|
||||||
StoredRequestUsageAudit, StoredUsageAuditAggregation, StoredUsageAuditSummary,
|
StoredRequestUsageAudit, StoredUsageAuditAggregation, StoredUsageAuditSummary,
|
||||||
StoredUsageBreakdownSummaryRow, StoredUsageCacheAffinityHitSummary,
|
StoredUsageBreakdownSummaryRow, StoredUsageCacheAffinityHitSummary,
|
||||||
StoredUsageCacheAffinityIntervalRow, StoredUsageCacheHitSummary, StoredUsageCostSavingsSummary,
|
StoredUsageCacheAffinityIntervalRow, StoredUsageCacheHitSummary, StoredUsageCostSavingsSummary,
|
||||||
|
|||||||
@@ -619,6 +619,23 @@ pub struct StoredProviderApiKeyUsageSummary {
|
|||||||
pub last_used_at_unix_secs: Option<u64>,
|
pub last_used_at_unix_secs: Option<u64>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
|
||||||
|
pub struct ProviderApiKeyWindowUsageRequest {
|
||||||
|
pub provider_api_key_id: String,
|
||||||
|
pub window_code: String,
|
||||||
|
pub start_unix_secs: u64,
|
||||||
|
pub end_unix_secs: u64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||||
|
pub struct StoredProviderApiKeyWindowUsageSummary {
|
||||||
|
pub provider_api_key_id: String,
|
||||||
|
pub window_code: String,
|
||||||
|
pub request_count: u64,
|
||||||
|
pub total_tokens: u64,
|
||||||
|
pub total_cost_usd: f64,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Eq, Default, serde::Serialize, serde::Deserialize)]
|
#[derive(Debug, Clone, PartialEq, Eq, Default, serde::Serialize, serde::Deserialize)]
|
||||||
pub struct UsageAuditListQuery {
|
pub struct UsageAuditListQuery {
|
||||||
pub created_from_unix_secs: Option<u64>,
|
pub created_from_unix_secs: Option<u64>,
|
||||||
@@ -1479,6 +1496,11 @@ pub trait UsageReadRepository: Send + Sync {
|
|||||||
crate::DataLayerError,
|
crate::DataLayerError,
|
||||||
>;
|
>;
|
||||||
|
|
||||||
|
async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<Vec<StoredProviderApiKeyWindowUsageSummary>, crate::DataLayerError>;
|
||||||
|
|
||||||
async fn summarize_provider_usage_since(
|
async fn summarize_provider_usage_since(
|
||||||
&self,
|
&self,
|
||||||
provider_id: &str,
|
provider_id: &str,
|
||||||
|
|||||||
@@ -30,9 +30,10 @@ use super::{
|
|||||||
api_key_usage_contribution, provider_api_key_usage_contribution,
|
api_key_usage_contribution, provider_api_key_usage_contribution,
|
||||||
strip_deprecated_usage_display_fields, usage_can_recover_terminal_failure,
|
strip_deprecated_usage_display_fields, usage_can_recover_terminal_failure,
|
||||||
ApiKeyUsageContribution, ApiKeyUsageDelta, ProviderApiKeyUsageContribution,
|
ApiKeyUsageContribution, ApiKeyUsageDelta, ProviderApiKeyUsageContribution,
|
||||||
ProviderApiKeyUsageDelta, StoredProviderApiKeyUsageSummary, StoredProviderUsageSummary,
|
ProviderApiKeyUsageDelta, ProviderApiKeyWindowUsageRequest, StoredProviderApiKeyUsageSummary,
|
||||||
StoredProviderUsageWindow, StoredRequestUsageAudit, StoredUsageDailySummary, UpsertUsageRecord,
|
StoredProviderApiKeyWindowUsageSummary, StoredProviderUsageSummary, StoredProviderUsageWindow,
|
||||||
UsageAuditListQuery, UsageDailyHeatmapQuery, UsageReadRepository, UsageWriteRepository,
|
StoredRequestUsageAudit, StoredUsageDailySummary, UpsertUsageRecord, UsageAuditListQuery,
|
||||||
|
UsageDailyHeatmapQuery, UsageReadRepository, UsageWriteRepository,
|
||||||
};
|
};
|
||||||
use crate::repository::auth::InMemoryAuthApiKeySnapshotRepository;
|
use crate::repository::auth::InMemoryAuthApiKeySnapshotRepository;
|
||||||
use crate::repository::provider_catalog::InMemoryProviderCatalogReadRepository;
|
use crate::repository::provider_catalog::InMemoryProviderCatalogReadRepository;
|
||||||
@@ -2327,6 +2328,59 @@ impl UsageReadRepository for InMemoryUsageReadRepository {
|
|||||||
Ok(summaries)
|
Ok(summaries)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<Vec<StoredProviderApiKeyWindowUsageSummary>, DataLayerError> {
|
||||||
|
let usage = self.by_request_id.read().expect("usage repository lock");
|
||||||
|
let mut summaries = Vec::with_capacity(requests.len());
|
||||||
|
|
||||||
|
for request in requests {
|
||||||
|
let provider_api_key_id = request.provider_api_key_id.trim();
|
||||||
|
if provider_api_key_id.is_empty() {
|
||||||
|
return Err(DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage provider_api_key_id cannot be empty".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let window_code = request.window_code.trim();
|
||||||
|
if window_code.is_empty() {
|
||||||
|
return Err(DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage window_code cannot be empty".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
if request.start_unix_secs >= request.end_unix_secs {
|
||||||
|
return Err(DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage range must be non-empty".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut summary = StoredProviderApiKeyWindowUsageSummary {
|
||||||
|
provider_api_key_id: provider_api_key_id.to_string(),
|
||||||
|
window_code: window_code.to_string(),
|
||||||
|
..StoredProviderApiKeyWindowUsageSummary::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
for item in usage.values() {
|
||||||
|
if item.provider_api_key_id.as_deref() != Some(provider_api_key_id) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if item.created_at_unix_ms < request.start_unix_secs
|
||||||
|
|| item.created_at_unix_ms >= request.end_unix_secs
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
summary.request_count = summary.request_count.saturating_add(1);
|
||||||
|
summary.total_tokens = summary.total_tokens.saturating_add(item.total_tokens);
|
||||||
|
summary.total_cost_usd += item.total_cost_usd;
|
||||||
|
}
|
||||||
|
|
||||||
|
summaries.push(summary);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(summaries)
|
||||||
|
}
|
||||||
|
|
||||||
async fn summarize_provider_usage_since(
|
async fn summarize_provider_usage_since(
|
||||||
&self,
|
&self,
|
||||||
provider_id: &str,
|
provider_id: &str,
|
||||||
@@ -2934,8 +2988,9 @@ mod tests {
|
|||||||
UsageWriteRepository,
|
UsageWriteRepository,
|
||||||
};
|
};
|
||||||
use aether_data_contracts::repository::usage::{
|
use aether_data_contracts::repository::usage::{
|
||||||
usage_body_ref, UsageAuditAggregationGroupBy, UsageAuditAggregationQuery, UsageBodyField,
|
usage_body_ref, ProviderApiKeyWindowUsageRequest, UsageAuditAggregationGroupBy,
|
||||||
UsageProviderPerformanceQuery, UsageTimeSeriesGranularity,
|
UsageAuditAggregationQuery, UsageBodyField, UsageProviderPerformanceQuery,
|
||||||
|
UsageTimeSeriesGranularity,
|
||||||
};
|
};
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
|
|
||||||
@@ -4533,6 +4588,44 @@ mod tests {
|
|||||||
assert_eq!(item.last_used_at_unix_secs, Some(1_711_000_250));
|
assert_eq!(item.last_used_at_unix_secs, Some(1_711_000_250));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn summarizes_provider_api_key_window_usage_with_zero_rows() {
|
||||||
|
let repository = InMemoryUsageReadRepository::seed(vec![
|
||||||
|
sample_usage("req-1", 1_711_000_000),
|
||||||
|
sample_usage("req-2", 1_711_000_250),
|
||||||
|
]);
|
||||||
|
|
||||||
|
let usage = repository
|
||||||
|
.summarize_usage_by_provider_api_key_windows(&[
|
||||||
|
ProviderApiKeyWindowUsageRequest {
|
||||||
|
provider_api_key_id: "provider-key-1".to_string(),
|
||||||
|
window_code: "5h".to_string(),
|
||||||
|
start_unix_secs: 1_711_000_000,
|
||||||
|
end_unix_secs: 1_711_000_300,
|
||||||
|
},
|
||||||
|
ProviderApiKeyWindowUsageRequest {
|
||||||
|
provider_api_key_id: "provider-key-empty".to_string(),
|
||||||
|
window_code: "weekly".to_string(),
|
||||||
|
start_unix_secs: 1_711_000_000,
|
||||||
|
end_unix_secs: 1_711_000_300,
|
||||||
|
},
|
||||||
|
])
|
||||||
|
.await
|
||||||
|
.expect("window summary should succeed");
|
||||||
|
|
||||||
|
assert_eq!(usage.len(), 2);
|
||||||
|
assert_eq!(usage[0].provider_api_key_id, "provider-key-1");
|
||||||
|
assert_eq!(usage[0].window_code, "5h");
|
||||||
|
assert_eq!(usage[0].request_count, 2);
|
||||||
|
assert_eq!(usage[0].total_tokens, 300);
|
||||||
|
assert_eq!(usage[0].total_cost_usd, 0.24);
|
||||||
|
assert_eq!(usage[1].provider_api_key_id, "provider-key-empty");
|
||||||
|
assert_eq!(usage[1].window_code, "weekly");
|
||||||
|
assert_eq!(usage[1].request_count, 0);
|
||||||
|
assert_eq!(usage[1].total_tokens, 0);
|
||||||
|
assert_eq!(usage[1].total_cost_usd, 0.0);
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn list_usage_audits_applies_second_based_time_filters() {
|
async fn list_usage_audits_applies_second_based_time_filters() {
|
||||||
let repository = InMemoryUsageReadRepository::seed(vec![
|
let repository = InMemoryUsageReadRepository::seed(vec![
|
||||||
|
|||||||
@@ -319,6 +319,17 @@ macro_rules! impl_materialized_usage_read_repository {
|
|||||||
<$crate::repository::usage::InMemoryUsageReadRepository as $crate::repository::usage::UsageReadRepository>::summarize_usage_by_provider_api_key_ids(&repository, provider_api_key_ids).await
|
<$crate::repository::usage::InMemoryUsageReadRepository as $crate::repository::usage::UsageReadRepository>::summarize_usage_by_provider_api_key_ids(&repository, provider_api_key_ids).await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[$crate::repository::usage::ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<
|
||||||
|
Vec<$crate::repository::usage::StoredProviderApiKeyWindowUsageSummary>,
|
||||||
|
$crate::DataLayerError,
|
||||||
|
> {
|
||||||
|
let repository = self.materialize_read_model().await?;
|
||||||
|
<$crate::repository::usage::InMemoryUsageReadRepository as $crate::repository::usage::UsageReadRepository>::summarize_usage_by_provider_api_key_windows(&repository, requests).await
|
||||||
|
}
|
||||||
|
|
||||||
async fn summarize_provider_usage_since(
|
async fn summarize_provider_usage_since(
|
||||||
&self,
|
&self,
|
||||||
provider_id: &str,
|
provider_id: &str,
|
||||||
@@ -352,9 +363,10 @@ mod sqlite;
|
|||||||
|
|
||||||
#[allow(unused_imports)]
|
#[allow(unused_imports)]
|
||||||
pub(crate) use aether_data_contracts::repository::usage::{
|
pub(crate) use aether_data_contracts::repository::usage::{
|
||||||
PendingUsageCleanupSummary, StoredProviderApiKeyUsageSummary, StoredProviderUsageSummary,
|
PendingUsageCleanupSummary, ProviderApiKeyWindowUsageRequest, StoredProviderApiKeyUsageSummary,
|
||||||
StoredProviderUsageWindow, StoredRequestUsageAudit, StoredUsageAuditAggregation,
|
StoredProviderApiKeyWindowUsageSummary, StoredProviderUsageSummary, StoredProviderUsageWindow,
|
||||||
StoredUsageAuditSummary, StoredUsageBreakdownSummaryRow, StoredUsageCacheAffinityHitSummary,
|
StoredRequestUsageAudit, StoredUsageAuditAggregation, StoredUsageAuditSummary,
|
||||||
|
StoredUsageBreakdownSummaryRow, StoredUsageCacheAffinityHitSummary,
|
||||||
StoredUsageCacheAffinityIntervalRow, StoredUsageCacheHitSummary, StoredUsageCostSavingsSummary,
|
StoredUsageCacheAffinityIntervalRow, StoredUsageCacheHitSummary, StoredUsageCostSavingsSummary,
|
||||||
StoredUsageDailySummary, StoredUsageDashboardDailyBreakdownRow,
|
StoredUsageDailySummary, StoredUsageDashboardDailyBreakdownRow,
|
||||||
StoredUsageDashboardProviderCount, StoredUsageDashboardSummary,
|
StoredUsageDashboardProviderCount, StoredUsageDashboardSummary,
|
||||||
|
|||||||
@@ -38,7 +38,8 @@ use super::{
|
|||||||
api_key_usage_contribution, incoming_usage_can_recover_terminal_failure,
|
api_key_usage_contribution, incoming_usage_can_recover_terminal_failure,
|
||||||
model_usage_contribution, provider_api_key_usage_contribution,
|
model_usage_contribution, provider_api_key_usage_contribution,
|
||||||
strip_deprecated_usage_display_fields, ApiKeyUsageDelta, ModelUsageDelta,
|
strip_deprecated_usage_display_fields, ApiKeyUsageDelta, ModelUsageDelta,
|
||||||
PendingUsageCleanupSummary, ProviderApiKeyUsageDelta, StoredProviderApiKeyUsageSummary,
|
PendingUsageCleanupSummary, ProviderApiKeyUsageDelta, ProviderApiKeyWindowUsageRequest,
|
||||||
|
StoredProviderApiKeyUsageSummary, StoredProviderApiKeyWindowUsageSummary,
|
||||||
StoredProviderUsageSummary, StoredRequestUsageAudit, StoredUsageDailySummary,
|
StoredProviderUsageSummary, StoredRequestUsageAudit, StoredUsageDailySummary,
|
||||||
UpsertUsageRecord, UsageAuditListQuery, UsageDailyHeatmapQuery, UsageReadRepository,
|
UpsertUsageRecord, UsageAuditListQuery, UsageDailyHeatmapQuery, UsageReadRepository,
|
||||||
UsageWriteRepository,
|
UsageWriteRepository,
|
||||||
@@ -1314,6 +1315,9 @@ const SUMMARIZE_USAGE_TOTALS_BY_USER_IDS_SQL: &str =
|
|||||||
const SUMMARIZE_USAGE_BY_PROVIDER_API_KEY_IDS_SQL: &str =
|
const SUMMARIZE_USAGE_BY_PROVIDER_API_KEY_IDS_SQL: &str =
|
||||||
include_str!("queries/summarize_usage_by_provider_api_key_ids_sql.sql");
|
include_str!("queries/summarize_usage_by_provider_api_key_ids_sql.sql");
|
||||||
|
|
||||||
|
const SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL: &str =
|
||||||
|
include_str!("queries/summarize_provider_api_key_window_usage_sql.sql");
|
||||||
|
|
||||||
const APPLY_API_KEY_USAGE_DELTA_SQL: &str =
|
const APPLY_API_KEY_USAGE_DELTA_SQL: &str =
|
||||||
include_str!("queries/apply_api_key_usage_delta_sql.sql");
|
include_str!("queries/apply_api_key_usage_delta_sql.sql");
|
||||||
|
|
||||||
@@ -7225,6 +7229,98 @@ ORDER BY "usage".user_id ASC
|
|||||||
Ok(summaries)
|
Ok(summaries)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<Vec<StoredProviderApiKeyWindowUsageSummary>, DataLayerError> {
|
||||||
|
if requests.is_empty() {
|
||||||
|
return Ok(Vec::new());
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut provider_api_key_ids = Vec::with_capacity(requests.len());
|
||||||
|
let mut window_codes = Vec::with_capacity(requests.len());
|
||||||
|
let mut start_unix_secs = Vec::with_capacity(requests.len());
|
||||||
|
let mut end_unix_secs = Vec::with_capacity(requests.len());
|
||||||
|
|
||||||
|
for request in requests {
|
||||||
|
let provider_api_key_id = request.provider_api_key_id.trim();
|
||||||
|
if provider_api_key_id.is_empty() {
|
||||||
|
return Err(DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage provider_api_key_id cannot be empty".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let window_code = request.window_code.trim();
|
||||||
|
if window_code.is_empty() {
|
||||||
|
return Err(DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage window_code cannot be empty".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
if request.start_unix_secs >= request.end_unix_secs {
|
||||||
|
return Err(DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage range must be non-empty".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
provider_api_key_ids.push(provider_api_key_id.to_string());
|
||||||
|
window_codes.push(window_code.to_string());
|
||||||
|
start_unix_secs.push(i64::try_from(request.start_unix_secs).map_err(|_| {
|
||||||
|
DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage start_unix_secs is out of range".to_string(),
|
||||||
|
)
|
||||||
|
})?);
|
||||||
|
end_unix_secs.push(i64::try_from(request.end_unix_secs).map_err(|_| {
|
||||||
|
DataLayerError::InvalidInput(
|
||||||
|
"provider api key window usage end_unix_secs is out of range".to_string(),
|
||||||
|
)
|
||||||
|
})?);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut rows = sqlx::query(SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL)
|
||||||
|
.bind(&provider_api_key_ids)
|
||||||
|
.bind(&window_codes)
|
||||||
|
.bind(&start_unix_secs)
|
||||||
|
.bind(&end_unix_secs)
|
||||||
|
.fetch(&self.pool);
|
||||||
|
|
||||||
|
let mut summaries = Vec::new();
|
||||||
|
while let Some(row) = rows.try_next().await.map_postgres_err()? {
|
||||||
|
let total_cost_usd = row.try_get::<f64, _>("total_cost_usd").map_postgres_err()?;
|
||||||
|
if !total_cost_usd.is_finite() {
|
||||||
|
return Err(DataLayerError::UnexpectedValue(
|
||||||
|
"usage.total_cost_usd window aggregate is not finite".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
summaries.push(StoredProviderApiKeyWindowUsageSummary {
|
||||||
|
provider_api_key_id: row
|
||||||
|
.try_get::<String, _>("provider_api_key_id")
|
||||||
|
.map_postgres_err()?,
|
||||||
|
window_code: row.try_get::<String, _>("window_code").map_postgres_err()?,
|
||||||
|
request_count: row
|
||||||
|
.try_get::<i64, _>("request_count")
|
||||||
|
.map_postgres_err()?
|
||||||
|
.try_into()
|
||||||
|
.map_err(|_| {
|
||||||
|
DataLayerError::UnexpectedValue(
|
||||||
|
"usage.request_count window aggregate is negative".to_string(),
|
||||||
|
)
|
||||||
|
})?,
|
||||||
|
total_tokens: row
|
||||||
|
.try_get::<i64, _>("total_tokens")
|
||||||
|
.map_postgres_err()?
|
||||||
|
.try_into()
|
||||||
|
.map_err(|_| {
|
||||||
|
DataLayerError::UnexpectedValue(
|
||||||
|
"usage.total_tokens window aggregate is negative".to_string(),
|
||||||
|
)
|
||||||
|
})?,
|
||||||
|
total_cost_usd,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(summaries)
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn upsert(
|
pub async fn upsert(
|
||||||
&self,
|
&self,
|
||||||
usage: UpsertUsageRecord,
|
usage: UpsertUsageRecord,
|
||||||
@@ -8044,6 +8140,13 @@ impl UsageReadRepository for SqlxUsageReadRepository {
|
|||||||
Self::summarize_usage_by_provider_api_key_ids(self, provider_api_key_ids).await
|
Self::summarize_usage_by_provider_api_key_ids(self, provider_api_key_ids).await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn summarize_usage_by_provider_api_key_windows(
|
||||||
|
&self,
|
||||||
|
requests: &[ProviderApiKeyWindowUsageRequest],
|
||||||
|
) -> Result<Vec<StoredProviderApiKeyWindowUsageSummary>, DataLayerError> {
|
||||||
|
Self::summarize_usage_by_provider_api_key_windows(self, requests).await
|
||||||
|
}
|
||||||
|
|
||||||
async fn summarize_provider_usage_since(
|
async fn summarize_provider_usage_since(
|
||||||
&self,
|
&self,
|
||||||
provider_id: &str,
|
provider_id: &str,
|
||||||
|
|||||||
+36
@@ -0,0 +1,36 @@
|
|||||||
|
WITH requested AS (
|
||||||
|
SELECT
|
||||||
|
request_row.provider_api_key_id,
|
||||||
|
request_row.window_code,
|
||||||
|
request_row.start_unix_secs,
|
||||||
|
request_row.end_unix_secs,
|
||||||
|
request_row.ordinality
|
||||||
|
FROM UNNEST(
|
||||||
|
$1::TEXT[],
|
||||||
|
$2::TEXT[],
|
||||||
|
$3::BIGINT[],
|
||||||
|
$4::BIGINT[]
|
||||||
|
) WITH ORDINALITY AS request_row(
|
||||||
|
provider_api_key_id,
|
||||||
|
window_code,
|
||||||
|
start_unix_secs,
|
||||||
|
end_unix_secs,
|
||||||
|
ordinality
|
||||||
|
)
|
||||||
|
)
|
||||||
|
SELECT
|
||||||
|
requested.provider_api_key_id,
|
||||||
|
requested.window_code,
|
||||||
|
COUNT("usage".id)::BIGINT AS request_count,
|
||||||
|
COALESCE(SUM("usage".total_tokens), 0)::BIGINT AS total_tokens,
|
||||||
|
CAST(COALESCE(SUM("usage".total_cost_usd), 0) AS DOUBLE PRECISION) AS total_cost_usd
|
||||||
|
FROM requested
|
||||||
|
LEFT JOIN usage_billing_facts AS "usage"
|
||||||
|
ON "usage".provider_api_key_id = requested.provider_api_key_id
|
||||||
|
AND "usage".created_at >= to_timestamp(requested.start_unix_secs::DOUBLE PRECISION)
|
||||||
|
AND "usage".created_at < to_timestamp(requested.end_unix_secs::DOUBLE PRECISION)
|
||||||
|
GROUP BY
|
||||||
|
requested.provider_api_key_id,
|
||||||
|
requested.window_code,
|
||||||
|
requested.ordinality
|
||||||
|
ORDER BY requested.ordinality ASC
|
||||||
@@ -234,6 +234,21 @@ fn usage_sql_summarizes_usage_by_provider_api_key_ids_in_database() {
|
|||||||
assert!(super::SUMMARIZE_USAGE_BY_PROVIDER_API_KEY_IDS_SQL.contains("ANY($1::TEXT[])"));
|
assert!(super::SUMMARIZE_USAGE_BY_PROVIDER_API_KEY_IDS_SQL.contains("ANY($1::TEXT[])"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn usage_sql_summarizes_provider_key_window_usage_from_billing_facts() {
|
||||||
|
assert!(super::SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL.contains("UNNEST"));
|
||||||
|
assert!(super::SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL
|
||||||
|
.contains("LEFT JOIN usage_billing_facts AS \"usage\""));
|
||||||
|
assert!(
|
||||||
|
super::SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL.contains("created_at >= to_timestamp")
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
super::SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL.contains("created_at < to_timestamp")
|
||||||
|
);
|
||||||
|
assert!(super::SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL
|
||||||
|
.contains("COUNT(\"usage\".id)::BIGINT AS request_count"));
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn usage_sql_serializes_request_id_upserts_before_reading_previous_usage() {
|
fn usage_sql_serializes_request_id_upserts_before_reading_previous_usage() {
|
||||||
assert!(super::LOCK_USAGE_REQUEST_ID_SQL.contains("pg_advisory_xact_lock"));
|
assert!(super::LOCK_USAGE_REQUEST_ID_SQL.contains("pg_advisory_xact_lock"));
|
||||||
@@ -455,6 +470,8 @@ fn usage_sql_raw_aggregates_use_canonical_billing_facts() {
|
|||||||
.contains("FROM usage_billing_facts AS \"usage\""));
|
.contains("FROM usage_billing_facts AS \"usage\""));
|
||||||
assert!(super::SUMMARIZE_USAGE_TOTALS_BY_USER_IDS_SQL
|
assert!(super::SUMMARIZE_USAGE_TOTALS_BY_USER_IDS_SQL
|
||||||
.contains("FROM usage_billing_facts AS \"usage\""));
|
.contains("FROM usage_billing_facts AS \"usage\""));
|
||||||
|
assert!(super::SUMMARIZE_PROVIDER_API_KEY_WINDOW_USAGE_SQL
|
||||||
|
.contains("usage_billing_facts AS \"usage\""));
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
|||||||
@@ -18,6 +18,12 @@ export interface AccountStatusSnapshot {
|
|||||||
recoverable?: boolean
|
recoverable?: boolean
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export interface QuotaWindowUsageSnapshot {
|
||||||
|
request_count?: number | null
|
||||||
|
total_tokens?: number | null
|
||||||
|
total_cost_usd?: number | string | null
|
||||||
|
}
|
||||||
|
|
||||||
export interface QuotaWindowSnapshot {
|
export interface QuotaWindowSnapshot {
|
||||||
code: string
|
code: string
|
||||||
label?: string | null
|
label?: string | null
|
||||||
@@ -33,6 +39,7 @@ export interface QuotaWindowSnapshot {
|
|||||||
reset_seconds?: number | null
|
reset_seconds?: number | null
|
||||||
window_minutes?: number | null
|
window_minutes?: number | null
|
||||||
is_exhausted?: boolean | null
|
is_exhausted?: boolean | null
|
||||||
|
usage?: QuotaWindowUsageSnapshot | null
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface QuotaCreditsSnapshot {
|
export interface QuotaCreditsSnapshot {
|
||||||
|
|||||||
@@ -39,6 +39,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: 20,
|
pageSize: 20,
|
||||||
sortBy: 'last_used_at',
|
sortBy: 'last_used_at',
|
||||||
sortOrder: 'asc',
|
sortOrder: 'asc',
|
||||||
|
statsMode: 'account_total',
|
||||||
},
|
},
|
||||||
storage,
|
storage,
|
||||||
)
|
)
|
||||||
@@ -52,6 +53,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: '100',
|
pageSize: '100',
|
||||||
sortBy: 'imported_at',
|
sortBy: 'imported_at',
|
||||||
sortOrder: 'desc',
|
sortOrder: 'desc',
|
||||||
|
statsMode: 'current_cycle',
|
||||||
},
|
},
|
||||||
storage,
|
storage,
|
||||||
)
|
)
|
||||||
@@ -64,6 +66,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: 100,
|
pageSize: 100,
|
||||||
sortBy: 'imported_at',
|
sortBy: 'imported_at',
|
||||||
sortOrder: 'desc',
|
sortOrder: 'desc',
|
||||||
|
statsMode: 'current_cycle',
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -77,6 +80,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: 50,
|
pageSize: 50,
|
||||||
sortBy: 'last_used_at',
|
sortBy: 'last_used_at',
|
||||||
sortOrder: 'asc',
|
sortOrder: 'asc',
|
||||||
|
statsMode: 'account_total',
|
||||||
},
|
},
|
||||||
storage,
|
storage,
|
||||||
)
|
)
|
||||||
@@ -91,6 +95,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: 50,
|
pageSize: 50,
|
||||||
sortBy: 'last_used_at',
|
sortBy: 'last_used_at',
|
||||||
sortOrder: 'asc',
|
sortOrder: 'asc',
|
||||||
|
statsMode: 'account_total',
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -104,6 +109,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: 50,
|
pageSize: 50,
|
||||||
sortBy: null,
|
sortBy: null,
|
||||||
sortOrder: 'desc',
|
sortOrder: 'desc',
|
||||||
|
statsMode: 'current_cycle',
|
||||||
}),
|
}),
|
||||||
).toEqual({
|
).toEqual({
|
||||||
providerId: 'provider-d',
|
providerId: 'provider-d',
|
||||||
@@ -113,6 +119,7 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: undefined,
|
pageSize: undefined,
|
||||||
sortBy: undefined,
|
sortBy: undefined,
|
||||||
sortOrder: undefined,
|
sortOrder: undefined,
|
||||||
|
statsMode: undefined,
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -126,13 +133,36 @@ describe('poolManagementState', () => {
|
|||||||
pageSize: 50,
|
pageSize: 50,
|
||||||
sortBy: 'last_used_at',
|
sortBy: 'last_used_at',
|
||||||
sortOrder: 'asc',
|
sortOrder: 'asc',
|
||||||
|
statsMode: 'account_total',
|
||||||
}),
|
}),
|
||||||
).toMatchObject({
|
).toMatchObject({
|
||||||
sortBy: 'last_used_at',
|
sortBy: 'last_used_at',
|
||||||
sortOrder: 'asc',
|
sortOrder: 'asc',
|
||||||
|
statsMode: 'account_total',
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
it('restores stats mode from storage and lets query override it', () => {
|
||||||
|
writePoolManagementViewState(
|
||||||
|
{
|
||||||
|
providerId: 'provider-f',
|
||||||
|
search: '',
|
||||||
|
status: 'all',
|
||||||
|
page: 1,
|
||||||
|
pageSize: 50,
|
||||||
|
sortBy: null,
|
||||||
|
sortOrder: 'desc',
|
||||||
|
statsMode: 'account_total',
|
||||||
|
},
|
||||||
|
storage,
|
||||||
|
)
|
||||||
|
|
||||||
|
expect(readPoolManagementViewState({}, storage).statsMode).toBe('account_total')
|
||||||
|
expect(
|
||||||
|
readPoolManagementViewState({ statsMode: 'current_cycle' }, storage).statsMode,
|
||||||
|
).toBe('current_cycle')
|
||||||
|
})
|
||||||
|
|
||||||
it('clamps a restored page to the last available page after load', () => {
|
it('clamps a restored page to the last available page after load', () => {
|
||||||
expect(
|
expect(
|
||||||
resolvePoolManagementPageAfterLoad({
|
resolvePoolManagementPageAfterLoad({
|
||||||
|
|||||||
@@ -0,0 +1,115 @@
|
|||||||
|
import { describe, expect, it } from 'vitest'
|
||||||
|
|
||||||
|
import {
|
||||||
|
buildPoolStatsDisplay,
|
||||||
|
type PoolStatsKeyInput,
|
||||||
|
} from '@/features/pool/utils/poolStatsDisplay'
|
||||||
|
|
||||||
|
function metricValues(metrics: Array<{ key: string, value: string }>) {
|
||||||
|
return Object.fromEntries(metrics.map(metric => [metric.key, metric.value]))
|
||||||
|
}
|
||||||
|
|
||||||
|
function createCodexKey(overrides: Partial<PoolStatsKeyInput> = {}): PoolStatsKeyInput {
|
||||||
|
return {
|
||||||
|
request_count: 1234,
|
||||||
|
total_tokens: 5678000,
|
||||||
|
total_cost_usd: '12.3456',
|
||||||
|
status_snapshot: {
|
||||||
|
quota: {
|
||||||
|
windows: [
|
||||||
|
{
|
||||||
|
code: '5h',
|
||||||
|
usage: {
|
||||||
|
request_count: 5,
|
||||||
|
total_tokens: 2500,
|
||||||
|
total_cost_usd: '0.0045',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
code: 'weekly',
|
||||||
|
usage: {
|
||||||
|
request_count: 0,
|
||||||
|
total_tokens: 0,
|
||||||
|
total_cost_usd: '0.00000000',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
...overrides,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
describe('poolStatsDisplay', () => {
|
||||||
|
it('builds Codex current-cycle groups in 5H and weekly order', () => {
|
||||||
|
const display = buildPoolStatsDisplay(createCodexKey(), 'codex', 'current_cycle')
|
||||||
|
|
||||||
|
expect(display.kind).toBe('codex_cycle')
|
||||||
|
if (display.kind !== 'codex_cycle') throw new Error('expected codex cycle display')
|
||||||
|
|
||||||
|
expect(display.groups.map(group => group.label)).toEqual(['5H', '周'])
|
||||||
|
expect(metricValues(display.groups[0].metrics)).toEqual({
|
||||||
|
request_count: '5',
|
||||||
|
total_tokens: '2.5K',
|
||||||
|
total_cost_usd: '$0.0045',
|
||||||
|
})
|
||||||
|
expect(metricValues(display.groups[1].metrics)).toEqual({
|
||||||
|
request_count: '0',
|
||||||
|
total_tokens: '0',
|
||||||
|
total_cost_usd: '0',
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
it('renders missing cycle usage as dashes instead of account-total fallback', () => {
|
||||||
|
const display = buildPoolStatsDisplay(
|
||||||
|
createCodexKey({
|
||||||
|
status_snapshot: {
|
||||||
|
quota: {
|
||||||
|
windows: [{ code: '5h', usage: null }],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
'codex',
|
||||||
|
'current_cycle',
|
||||||
|
)
|
||||||
|
|
||||||
|
expect(display.kind).toBe('codex_cycle')
|
||||||
|
if (display.kind !== 'codex_cycle') throw new Error('expected codex cycle display')
|
||||||
|
|
||||||
|
expect(metricValues(display.groups[0].metrics)).toEqual({
|
||||||
|
request_count: '—',
|
||||||
|
total_tokens: '—',
|
||||||
|
total_cost_usd: '—',
|
||||||
|
})
|
||||||
|
expect(metricValues(display.groups[1].metrics)).toEqual({
|
||||||
|
request_count: '—',
|
||||||
|
total_tokens: '—',
|
||||||
|
total_cost_usd: '—',
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
it('preserves account-total formatting when toggled away from current cycle', () => {
|
||||||
|
const display = buildPoolStatsDisplay(createCodexKey(), 'codex', 'account_total')
|
||||||
|
|
||||||
|
expect(display.kind).toBe('account_total')
|
||||||
|
if (display.kind !== 'account_total') throw new Error('expected account total display')
|
||||||
|
|
||||||
|
expect(metricValues(display.metrics)).toEqual({
|
||||||
|
request_count: '1,234',
|
||||||
|
total_tokens: '5.7M',
|
||||||
|
total_cost_usd: '$12.35',
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
it('keeps non-Codex providers on account totals even in current-cycle mode', () => {
|
||||||
|
const display = buildPoolStatsDisplay(createCodexKey(), 'openai', 'current_cycle')
|
||||||
|
|
||||||
|
expect(display.kind).toBe('account_total')
|
||||||
|
if (display.kind !== 'account_total') throw new Error('expected account total display')
|
||||||
|
expect(metricValues(display.metrics)).toMatchObject({
|
||||||
|
request_count: '1,234',
|
||||||
|
total_tokens: '5.7M',
|
||||||
|
total_cost_usd: '$12.35',
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
export type PoolManagementStatus = 'all' | 'active' | 'cooldown' | 'inactive'
|
export type PoolManagementStatus = 'all' | 'active' | 'cooldown' | 'inactive'
|
||||||
export type PoolManagementSortBy = 'imported_at' | 'last_used_at'
|
export type PoolManagementSortBy = 'imported_at' | 'last_used_at'
|
||||||
export type PoolManagementSortOrder = 'asc' | 'desc'
|
export type PoolManagementSortOrder = 'asc' | 'desc'
|
||||||
|
export type PoolManagementStatsMode = 'current_cycle' | 'account_total'
|
||||||
|
|
||||||
export interface PoolManagementViewState {
|
export interface PoolManagementViewState {
|
||||||
providerId: string | null
|
providerId: string | null
|
||||||
@@ -10,6 +11,7 @@ export interface PoolManagementViewState {
|
|||||||
pageSize: number
|
pageSize: number
|
||||||
sortBy: PoolManagementSortBy | null
|
sortBy: PoolManagementSortBy | null
|
||||||
sortOrder: PoolManagementSortOrder
|
sortOrder: PoolManagementSortOrder
|
||||||
|
statsMode: PoolManagementStatsMode
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface PoolManagementStateSource {
|
export interface PoolManagementStateSource {
|
||||||
@@ -20,6 +22,7 @@ export interface PoolManagementStateSource {
|
|||||||
pageSize?: string
|
pageSize?: string
|
||||||
sortBy?: string
|
sortBy?: string
|
||||||
sortOrder?: string
|
sortOrder?: string
|
||||||
|
statsMode?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface StorageLike {
|
export interface StorageLike {
|
||||||
@@ -28,6 +31,10 @@ export interface StorageLike {
|
|||||||
removeItem(key: string): void
|
removeItem(key: string): void
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type PoolManagementViewStateInput = Partial<{
|
||||||
|
[Key in keyof PoolManagementViewState]: unknown
|
||||||
|
}>
|
||||||
|
|
||||||
export const POOL_MANAGEMENT_VIEW_STORAGE_KEY = 'aether:pool-management:view-state'
|
export const POOL_MANAGEMENT_VIEW_STORAGE_KEY = 'aether:pool-management:view-state'
|
||||||
|
|
||||||
export const DEFAULT_POOL_MANAGEMENT_VIEW_STATE: PoolManagementViewState = {
|
export const DEFAULT_POOL_MANAGEMENT_VIEW_STATE: PoolManagementViewState = {
|
||||||
@@ -38,6 +45,7 @@ export const DEFAULT_POOL_MANAGEMENT_VIEW_STATE: PoolManagementViewState = {
|
|||||||
pageSize: 50,
|
pageSize: 50,
|
||||||
sortBy: null,
|
sortBy: null,
|
||||||
sortOrder: 'desc',
|
sortOrder: 'desc',
|
||||||
|
statsMode: 'current_cycle',
|
||||||
}
|
}
|
||||||
|
|
||||||
function normalizeProviderId(value: unknown): string | null {
|
function normalizeProviderId(value: unknown): string | null {
|
||||||
@@ -75,7 +83,11 @@ function normalizeSortOrder(value: unknown): PoolManagementSortOrder {
|
|||||||
return value === 'asc' ? 'asc' : 'desc'
|
return value === 'asc' ? 'asc' : 'desc'
|
||||||
}
|
}
|
||||||
|
|
||||||
function normalizeViewState(input: Partial<PoolManagementViewState>): PoolManagementViewState {
|
function normalizeStatsMode(value: unknown): PoolManagementStatsMode {
|
||||||
|
return value === 'account_total' ? 'account_total' : 'current_cycle'
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeViewState(input: PoolManagementViewStateInput): PoolManagementViewState {
|
||||||
return {
|
return {
|
||||||
providerId: normalizeProviderId(input.providerId),
|
providerId: normalizeProviderId(input.providerId),
|
||||||
search: normalizeSearch(input.search),
|
search: normalizeSearch(input.search),
|
||||||
@@ -84,6 +96,7 @@ function normalizeViewState(input: Partial<PoolManagementViewState>): PoolManage
|
|||||||
pageSize: normalizePositiveInteger(input.pageSize, DEFAULT_POOL_MANAGEMENT_VIEW_STATE.pageSize),
|
pageSize: normalizePositiveInteger(input.pageSize, DEFAULT_POOL_MANAGEMENT_VIEW_STATE.pageSize),
|
||||||
sortBy: normalizeSortBy(input.sortBy),
|
sortBy: normalizeSortBy(input.sortBy),
|
||||||
sortOrder: normalizeSortOrder(input.sortOrder),
|
sortOrder: normalizeSortOrder(input.sortOrder),
|
||||||
|
statsMode: normalizeStatsMode(input.statsMode),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -106,15 +119,20 @@ export function readPoolManagementViewState(
|
|||||||
): PoolManagementViewState {
|
): PoolManagementViewState {
|
||||||
const stored = normalizeViewState(readStoredState(storage))
|
const stored = normalizeViewState(readStoredState(storage))
|
||||||
|
|
||||||
return normalizeViewState({
|
return {
|
||||||
providerId: source.providerId ?? stored.providerId,
|
providerId: source.providerId !== undefined ? normalizeProviderId(source.providerId) : stored.providerId,
|
||||||
search: source.search ?? stored.search,
|
search: source.search !== undefined ? normalizeSearch(source.search) : stored.search,
|
||||||
status: source.status ?? stored.status,
|
status: source.status !== undefined ? normalizeStatus(source.status) : stored.status,
|
||||||
page: source.page ?? stored.page,
|
page: source.page !== undefined
|
||||||
pageSize: source.pageSize ?? stored.pageSize,
|
? normalizePositiveInteger(source.page, DEFAULT_POOL_MANAGEMENT_VIEW_STATE.page)
|
||||||
sortBy: source.sortBy ?? stored.sortBy,
|
: stored.page,
|
||||||
sortOrder: source.sortOrder ?? stored.sortOrder,
|
pageSize: source.pageSize !== undefined
|
||||||
})
|
? normalizePositiveInteger(source.pageSize, DEFAULT_POOL_MANAGEMENT_VIEW_STATE.pageSize)
|
||||||
|
: stored.pageSize,
|
||||||
|
sortBy: source.sortBy !== undefined ? normalizeSortBy(source.sortBy) : stored.sortBy,
|
||||||
|
sortOrder: source.sortOrder !== undefined ? normalizeSortOrder(source.sortOrder) : stored.sortOrder,
|
||||||
|
statsMode: source.statsMode !== undefined ? normalizeStatsMode(source.statsMode) : stored.statsMode,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export function writePoolManagementViewState(
|
export function writePoolManagementViewState(
|
||||||
@@ -150,6 +168,7 @@ export function buildPoolManagementQueryPatch(
|
|||||||
: String(normalized.pageSize),
|
: String(normalized.pageSize),
|
||||||
sortBy: normalized.sortBy || undefined,
|
sortBy: normalized.sortBy || undefined,
|
||||||
sortOrder: normalized.sortBy ? normalized.sortOrder : undefined,
|
sortOrder: normalized.sortBy ? normalized.sortOrder : undefined,
|
||||||
|
statsMode: normalized.statsMode === 'account_total' ? 'account_total' : undefined,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,178 @@
|
|||||||
|
import type { QuotaWindowUsageSnapshot } from '@/api/endpoints/types/statusSnapshot'
|
||||||
|
import type { PoolManagementStatsMode } from '@/features/pool/utils/poolManagementState'
|
||||||
|
|
||||||
|
export type PoolStatsMetricKey = 'request_count' | 'total_tokens' | 'total_cost_usd'
|
||||||
|
export type PoolStatsDisplayKind = 'account_total' | 'codex_cycle'
|
||||||
|
export type PoolCodexCycleWindowCode = '5h' | 'weekly'
|
||||||
|
|
||||||
|
export interface PoolStatsKeyInput {
|
||||||
|
request_count?: number | null
|
||||||
|
total_tokens?: number | null
|
||||||
|
total_cost_usd?: number | string | null
|
||||||
|
status_snapshot?: {
|
||||||
|
quota?: {
|
||||||
|
windows?: Array<{
|
||||||
|
code?: string | null
|
||||||
|
usage?: QuotaWindowUsageSnapshot | null
|
||||||
|
} | null> | null
|
||||||
|
} | null
|
||||||
|
} | null
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface PoolStatsMetric {
|
||||||
|
key: PoolStatsMetricKey
|
||||||
|
label: string
|
||||||
|
value: string
|
||||||
|
missing: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface PoolAccountTotalStatsDisplay {
|
||||||
|
kind: 'account_total'
|
||||||
|
metrics: PoolStatsMetric[]
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface PoolCodexCycleStatsGroup {
|
||||||
|
code: PoolCodexCycleWindowCode
|
||||||
|
label: string
|
||||||
|
metrics: PoolStatsMetric[]
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface PoolCodexCycleStatsDisplay {
|
||||||
|
kind: 'codex_cycle'
|
||||||
|
groups: PoolCodexCycleStatsGroup[]
|
||||||
|
}
|
||||||
|
|
||||||
|
export type PoolStatsDisplay = PoolAccountTotalStatsDisplay | PoolCodexCycleStatsDisplay
|
||||||
|
|
||||||
|
const MISSING_STAT_VALUE = '—'
|
||||||
|
const CODEX_CYCLE_WINDOWS: Array<{ code: PoolCodexCycleWindowCode, label: string }> = [
|
||||||
|
{ code: '5h', label: '5H' },
|
||||||
|
{ code: 'weekly', label: '周' },
|
||||||
|
]
|
||||||
|
|
||||||
|
export function isCodexProviderType(providerType: string | null | undefined): boolean {
|
||||||
|
return String(providerType || '').trim().toLowerCase() === 'codex'
|
||||||
|
}
|
||||||
|
|
||||||
|
export function formatPoolStatInteger(value: number | null | undefined): string {
|
||||||
|
const n = Number(value ?? 0)
|
||||||
|
if (!Number.isFinite(n) || n <= 0) return '0'
|
||||||
|
return Math.round(n).toLocaleString('en-US')
|
||||||
|
}
|
||||||
|
|
||||||
|
export function formatPoolTokenCount(value: number | null | undefined): string {
|
||||||
|
const n = Number(value ?? 0)
|
||||||
|
if (!Number.isFinite(n) || n <= 0) return '0'
|
||||||
|
if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`
|
||||||
|
if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`
|
||||||
|
return String(Math.round(n))
|
||||||
|
}
|
||||||
|
|
||||||
|
export function formatPoolStatUsd(value: number | string | null | undefined): string {
|
||||||
|
const n = Number(value ?? 0)
|
||||||
|
if (!Number.isFinite(n) || n <= 0) return '$0.00'
|
||||||
|
if (n < 0.01) return `$${n.toFixed(4)}`
|
||||||
|
if (n < 1) return `$${n.toFixed(3)}`
|
||||||
|
if (n < 1000) return `$${n.toFixed(2)}`
|
||||||
|
return `$${n.toLocaleString('en-US', { minimumFractionDigits: 2, maximumFractionDigits: 2 })}`
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatCycleInteger(value: number | null | undefined): string | null {
|
||||||
|
if (value == null) return null
|
||||||
|
const n = Number(value)
|
||||||
|
if (!Number.isFinite(n)) return null
|
||||||
|
if (n <= 0) return '0'
|
||||||
|
return Math.round(n).toLocaleString('en-US')
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatCycleTokenCount(value: number | null | undefined): string | null {
|
||||||
|
if (value == null) return null
|
||||||
|
const n = Number(value)
|
||||||
|
if (!Number.isFinite(n)) return null
|
||||||
|
return formatPoolTokenCount(n)
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatCycleUsd(value: number | string | null | undefined): string | null {
|
||||||
|
if (value == null) return null
|
||||||
|
const n = Number(value)
|
||||||
|
if (!Number.isFinite(n)) return null
|
||||||
|
if (n <= 0) return '0'
|
||||||
|
return formatPoolStatUsd(value)
|
||||||
|
}
|
||||||
|
|
||||||
|
function createMetric(
|
||||||
|
key: PoolStatsMetricKey,
|
||||||
|
label: string,
|
||||||
|
value: string | null,
|
||||||
|
): PoolStatsMetric {
|
||||||
|
return {
|
||||||
|
key,
|
||||||
|
label,
|
||||||
|
value: value ?? MISSING_STAT_VALUE,
|
||||||
|
missing: value == null,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeWindowCode(value: unknown): string {
|
||||||
|
return String(value || '').trim().toLowerCase()
|
||||||
|
}
|
||||||
|
|
||||||
|
function getQuotaWindowUsage(
|
||||||
|
key: PoolStatsKeyInput,
|
||||||
|
code: PoolCodexCycleWindowCode,
|
||||||
|
): QuotaWindowUsageSnapshot | null {
|
||||||
|
const windows = key.status_snapshot?.quota?.windows
|
||||||
|
if (!Array.isArray(windows)) return null
|
||||||
|
|
||||||
|
const window = windows.find(item => normalizeWindowCode(item?.code) === code)
|
||||||
|
return window?.usage ?? null
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildAccountTotalMetrics(key: PoolStatsKeyInput): PoolStatsMetric[] {
|
||||||
|
return [
|
||||||
|
createMetric('request_count', '请求', formatPoolStatInteger(key.request_count)),
|
||||||
|
createMetric('total_tokens', 'Token', formatPoolTokenCount(key.total_tokens)),
|
||||||
|
createMetric('total_cost_usd', '费用', formatPoolStatUsd(key.total_cost_usd)),
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
function buildCycleMetrics(usage: QuotaWindowUsageSnapshot | null): PoolStatsMetric[] {
|
||||||
|
return [
|
||||||
|
createMetric('request_count', '请求', formatCycleInteger(usage?.request_count)),
|
||||||
|
createMetric('total_tokens', 'Token', formatCycleTokenCount(usage?.total_tokens)),
|
||||||
|
createMetric('total_cost_usd', '费用', formatCycleUsd(usage?.total_cost_usd)),
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
export function buildAccountTotalStatsDisplay(
|
||||||
|
key: PoolStatsKeyInput,
|
||||||
|
): PoolAccountTotalStatsDisplay {
|
||||||
|
return {
|
||||||
|
kind: 'account_total',
|
||||||
|
metrics: buildAccountTotalMetrics(key),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function buildCodexCycleStatsDisplay(
|
||||||
|
key: PoolStatsKeyInput,
|
||||||
|
): PoolCodexCycleStatsDisplay {
|
||||||
|
return {
|
||||||
|
kind: 'codex_cycle',
|
||||||
|
groups: CODEX_CYCLE_WINDOWS.map(window => ({
|
||||||
|
...window,
|
||||||
|
metrics: buildCycleMetrics(getQuotaWindowUsage(key, window.code)),
|
||||||
|
})),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function buildPoolStatsDisplay(
|
||||||
|
key: PoolStatsKeyInput,
|
||||||
|
providerType: string | null | undefined,
|
||||||
|
mode: PoolManagementStatsMode,
|
||||||
|
): PoolStatsDisplay {
|
||||||
|
if (isCodexProviderType(providerType) && mode === 'current_cycle') {
|
||||||
|
return buildCodexCycleStatsDisplay(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
return buildAccountTotalStatsDisplay(key)
|
||||||
|
}
|
||||||
@@ -76,6 +76,34 @@
|
|||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
<div
|
||||||
|
v-if="showCodexStatsModeSwitch"
|
||||||
|
class="flex items-center"
|
||||||
|
data-testid="pool-mobile-header-actions"
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
class="group inline-flex min-h-12 flex-col items-center justify-center gap-1 rounded-md border border-border/50 bg-muted/20 px-2.5 py-1.5 text-xs transition-all duration-200 hover:border-primary/40 hover:bg-muted/40"
|
||||||
|
data-testid="pool-stats-mode-control"
|
||||||
|
>
|
||||||
|
<div class="flex items-center gap-1 leading-none">
|
||||||
|
<span
|
||||||
|
class="font-medium transition-colors"
|
||||||
|
:class="!codexCurrentCycleStatsEnabled ? 'text-foreground/90' : 'text-muted-foreground/80'"
|
||||||
|
>累计</span>
|
||||||
|
<span class="text-muted-foreground/50 transition-colors group-hover:text-muted-foreground/80">/</span>
|
||||||
|
<span
|
||||||
|
class="font-medium transition-colors"
|
||||||
|
:class="codexCurrentCycleStatsEnabled ? 'text-foreground/90' : 'text-muted-foreground/80'"
|
||||||
|
>周期</span>
|
||||||
|
</div>
|
||||||
|
<Switch
|
||||||
|
v-model="codexCurrentCycleStatsEnabled"
|
||||||
|
class="shrink-0"
|
||||||
|
aria-label="Codex 统计模式"
|
||||||
|
data-testid="pool-stats-mode-switch"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
<div
|
<div
|
||||||
v-if="selectedProviderId"
|
v-if="selectedProviderId"
|
||||||
class="flex items-center gap-1"
|
class="flex items-center gap-1"
|
||||||
@@ -171,7 +199,10 @@
|
|||||||
</span>
|
</span>
|
||||||
</h3>
|
</h3>
|
||||||
</div>
|
</div>
|
||||||
<div class="flex items-center gap-2">
|
<div
|
||||||
|
class="flex items-center gap-2"
|
||||||
|
data-testid="pool-header-actions"
|
||||||
|
>
|
||||||
<Select
|
<Select
|
||||||
v-model="selectedProviderIdProxy"
|
v-model="selectedProviderIdProxy"
|
||||||
:disabled="providerSelectDisabled"
|
:disabled="providerSelectDisabled"
|
||||||
@@ -228,6 +259,33 @@
|
|||||||
v-if="selectedProviderId"
|
v-if="selectedProviderId"
|
||||||
class="h-4 w-px bg-border"
|
class="h-4 w-px bg-border"
|
||||||
/>
|
/>
|
||||||
|
<div
|
||||||
|
v-if="showCodexStatsModeSwitch"
|
||||||
|
class="group inline-flex min-h-12 flex-col items-center justify-center gap-1 rounded-md border border-border/50 bg-muted/20 px-2.5 py-1.5 text-xs transition-all duration-200 hover:border-primary/40 hover:bg-muted/40"
|
||||||
|
data-testid="pool-stats-mode-control"
|
||||||
|
>
|
||||||
|
<div class="flex items-center gap-1 leading-none">
|
||||||
|
<span
|
||||||
|
class="font-medium transition-colors"
|
||||||
|
:class="!codexCurrentCycleStatsEnabled ? 'text-foreground/90' : 'text-muted-foreground/80'"
|
||||||
|
>累计</span>
|
||||||
|
<span class="text-muted-foreground/50 transition-colors group-hover:text-muted-foreground/80">/</span>
|
||||||
|
<span
|
||||||
|
class="font-medium transition-colors"
|
||||||
|
:class="codexCurrentCycleStatsEnabled ? 'text-foreground/90' : 'text-muted-foreground/80'"
|
||||||
|
>周期</span>
|
||||||
|
</div>
|
||||||
|
<Switch
|
||||||
|
v-model="codexCurrentCycleStatsEnabled"
|
||||||
|
class="shrink-0"
|
||||||
|
aria-label="Codex 统计模式"
|
||||||
|
data-testid="pool-stats-mode-switch"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<div
|
||||||
|
v-if="showCodexStatsModeSwitch"
|
||||||
|
class="h-4 w-px bg-border"
|
||||||
|
/>
|
||||||
<Button
|
<Button
|
||||||
v-if="selectedProviderId"
|
v-if="selectedProviderId"
|
||||||
variant="ghost"
|
variant="ghost"
|
||||||
@@ -571,23 +629,44 @@
|
|||||||
>-</span>
|
>-</span>
|
||||||
</TableCell>
|
</TableCell>
|
||||||
<TableCell class="py-3 px-2 align-middle">
|
<TableCell class="py-3 px-2 align-middle">
|
||||||
<div class="grid grid-rows-3 gap-0.5 w-[136px] mx-auto text-[10px] leading-4">
|
<div
|
||||||
<div class="flex items-center justify-between gap-2">
|
v-if="isPoolKeyCycleStatsDisplay(key)"
|
||||||
<span class="text-muted-foreground">请求</span>
|
class="mx-auto w-[136px] space-y-1.5 text-[10px] leading-4"
|
||||||
<span class="tabular-nums text-foreground/90">
|
data-testid="pool-stats-cycle-groups"
|
||||||
{{ formatStatInteger(key.request_count) }}
|
>
|
||||||
</span>
|
<div
|
||||||
|
v-for="group in getPoolKeyCycleStatsGroups(key)"
|
||||||
|
:key="`${key.key_id}-${group.code}-desktop-stats`"
|
||||||
|
:data-testid="`pool-stats-cycle-group-${group.code}`"
|
||||||
|
>
|
||||||
|
<div class="text-[9px] text-muted-foreground/70 font-medium mb-0.5">{{ group.label }}</div>
|
||||||
|
<div
|
||||||
|
v-for="metric in group.metrics"
|
||||||
|
:key="`${group.code}-${metric.key}`"
|
||||||
|
class="flex items-center justify-between gap-2"
|
||||||
|
>
|
||||||
|
<span class="text-muted-foreground">{{ metric.label }}</span>
|
||||||
|
<span
|
||||||
|
class="tabular-nums text-foreground/90"
|
||||||
|
:class="metric.missing ? 'text-muted-foreground/80' : ''"
|
||||||
|
:data-testid="`pool-stats-${group.code}-${metric.key}`"
|
||||||
|
>{{ metric.value }}</span>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div class="flex items-center justify-between gap-2">
|
</div>
|
||||||
<span class="text-muted-foreground">Token</span>
|
<div
|
||||||
|
v-else
|
||||||
|
class="grid grid-rows-3 gap-0.5 w-[136px] mx-auto text-[10px] leading-4"
|
||||||
|
data-testid="pool-stats-account-total"
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
v-for="metric in getPoolKeyAccountStatsMetrics(key)"
|
||||||
|
:key="`${key.key_id}-${metric.key}-account-total`"
|
||||||
|
class="flex items-center justify-between gap-2"
|
||||||
|
>
|
||||||
|
<span class="text-muted-foreground">{{ metric.label }}</span>
|
||||||
<span class="tabular-nums text-foreground/90">
|
<span class="tabular-nums text-foreground/90">
|
||||||
{{ formatTokenCount(key.total_tokens) }}
|
{{ metric.value }}
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
<div class="flex items-center justify-between gap-2">
|
|
||||||
<span class="text-muted-foreground">费用</span>
|
|
||||||
<span class="tabular-nums text-foreground/90">
|
|
||||||
{{ formatStatUsd(key.total_cost_usd) }}
|
|
||||||
</span>
|
</span>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -787,16 +866,48 @@
|
|||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div class="overflow-x-auto rounded-xl border border-border/50 bg-muted/30 px-3 py-2 text-[11px] text-muted-foreground">
|
<div class="overflow-x-auto rounded-xl border border-border/50 bg-muted/30 px-3 py-2 text-[11px] text-muted-foreground">
|
||||||
<div class="flex min-w-max items-center justify-center whitespace-nowrap text-center">
|
<div class="space-y-1 text-center">
|
||||||
<span class="font-medium text-foreground/90">请求:{{ formatStatInteger(key.request_count) }}</span>
|
<template v-if="isPoolKeyCycleStatsDisplay(key)">
|
||||||
<span class="mx-1.5 text-muted-foreground/40">|</span>
|
<div
|
||||||
<span class="font-medium text-foreground/90">Token:{{ formatTokenCount(key.total_tokens) }}</span>
|
v-for="group in getPoolKeyCycleStatsGroups(key)"
|
||||||
<span class="mx-1.5 text-muted-foreground/40">|</span>
|
:key="`${key.key_id}-${group.code}-mobile-stats`"
|
||||||
<span class="font-medium text-foreground/90">费用:{{ formatStatUsd(key.total_cost_usd) }}</span>
|
class="flex items-start gap-3 text-left"
|
||||||
<span class="mx-1.5 text-muted-foreground/40">|</span>
|
:data-testid="`pool-mobile-stats-cycle-group-${group.code}`"
|
||||||
<span class="font-medium text-foreground/90">导入:{{ keyUiStateMap[key.key_id]?.importedAtRelative || '-' }}</span>
|
>
|
||||||
<span class="mx-1.5 text-muted-foreground/40">|</span>
|
<span class="w-10 shrink-0 pt-0.5 text-[10px] font-semibold text-foreground">{{ group.label }}</span>
|
||||||
<span class="font-medium text-foreground/90">最后使用:{{ keyUiStateMap[key.key_id]?.lastUsedRelative || '-' }}</span>
|
<div class="min-w-0 flex-1 space-y-0.5">
|
||||||
|
<div
|
||||||
|
v-for="metric in group.metrics"
|
||||||
|
:key="`${group.code}-${metric.key}-mobile`"
|
||||||
|
class="flex items-center justify-between gap-2"
|
||||||
|
>
|
||||||
|
<span class="text-muted-foreground">{{ metric.label }}</span>
|
||||||
|
<span
|
||||||
|
class="font-medium text-foreground/90 tabular-nums"
|
||||||
|
:class="metric.missing ? 'text-muted-foreground/80' : ''"
|
||||||
|
>{{ metric.value }}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
<template v-else>
|
||||||
|
<div
|
||||||
|
v-for="metric in getPoolKeyAccountStatsMetrics(key)"
|
||||||
|
:key="`${key.key_id}-${metric.key}-mobile-account-total`"
|
||||||
|
class="flex items-center justify-between gap-2"
|
||||||
|
>
|
||||||
|
<span class="text-muted-foreground">{{ metric.label }}</span>
|
||||||
|
<span class="font-medium text-foreground/90">{{ metric.value }}</span>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
<div class="flex items-center justify-between gap-2 border-t border-border/40 pt-1 mt-1">
|
||||||
|
<span class="text-muted-foreground">导入</span>
|
||||||
|
<span class="font-medium text-foreground/90">{{ keyUiStateMap[key.key_id]?.importedAtRelative || '-' }}</span>
|
||||||
|
</div>
|
||||||
|
<div class="flex items-center justify-between gap-2">
|
||||||
|
<span class="text-muted-foreground">最后使用</span>
|
||||||
|
<span class="font-medium text-foreground/90">{{ keyUiStateMap[key.key_id]?.lastUsedRelative || '-' }}</span>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -1146,6 +1257,7 @@ import {
|
|||||||
SortableTableHead,
|
SortableTableHead,
|
||||||
TableFilterMenu,
|
TableFilterMenu,
|
||||||
TableCell,
|
TableCell,
|
||||||
|
Switch,
|
||||||
Pagination,
|
Pagination,
|
||||||
Popover,
|
Popover,
|
||||||
PopoverTrigger,
|
PopoverTrigger,
|
||||||
@@ -1209,9 +1321,16 @@ import {
|
|||||||
resolvePoolManagementPageAfterLoad,
|
resolvePoolManagementPageAfterLoad,
|
||||||
type PoolManagementSortBy,
|
type PoolManagementSortBy,
|
||||||
type PoolManagementSortOrder,
|
type PoolManagementSortOrder,
|
||||||
|
type PoolManagementStatsMode,
|
||||||
type PoolManagementViewState,
|
type PoolManagementViewState,
|
||||||
writePoolManagementViewState,
|
writePoolManagementViewState,
|
||||||
} from '@/features/pool/utils/poolManagementState'
|
} from '@/features/pool/utils/poolManagementState'
|
||||||
|
import {
|
||||||
|
buildPoolStatsDisplay,
|
||||||
|
type PoolCodexCycleStatsGroup,
|
||||||
|
type PoolStatsDisplay,
|
||||||
|
type PoolStatsMetric,
|
||||||
|
} from '@/features/pool/utils/poolStatsDisplay'
|
||||||
import { getOAuthOrgBadge } from '@/utils/oauthIdentity'
|
import { getOAuthOrgBadge } from '@/utils/oauthIdentity'
|
||||||
import { getOAuthRefreshFeedback } from '@/utils/oauthRefreshFeedback'
|
import { getOAuthRefreshFeedback } from '@/utils/oauthRefreshFeedback'
|
||||||
import {
|
import {
|
||||||
@@ -1253,6 +1372,7 @@ const restoredViewState = readPoolManagementViewState(
|
|||||||
pageSize: getQueryValue('pageSize'),
|
pageSize: getQueryValue('pageSize'),
|
||||||
sortBy: getQueryValue('sortBy'),
|
sortBy: getQueryValue('sortBy'),
|
||||||
sortOrder: getQueryValue('sortOrder'),
|
sortOrder: getQueryValue('sortOrder'),
|
||||||
|
statsMode: getQueryValue('statsMode'),
|
||||||
},
|
},
|
||||||
poolManagementViewStorage,
|
poolManagementViewStorage,
|
||||||
)
|
)
|
||||||
@@ -1476,6 +1596,14 @@ const selectedProviderType = computed(() => {
|
|||||||
return String(fromOverview || '').trim().toLowerCase()
|
return String(fromOverview || '').trim().toLowerCase()
|
||||||
})
|
})
|
||||||
|
|
||||||
|
const showCodexStatsModeSwitch = computed(() => selectedProviderType.value === 'codex')
|
||||||
|
const codexCurrentCycleStatsEnabled = computed({
|
||||||
|
get: () => poolStatsMode.value === 'current_cycle',
|
||||||
|
set: (enabled: boolean) => {
|
||||||
|
poolStatsMode.value = enabled ? 'current_cycle' : 'account_total'
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
const selectedProviderStatusText = computed(() => {
|
const selectedProviderStatusText = computed(() => {
|
||||||
if (!selectedProviderId.value) return ''
|
if (!selectedProviderId.value) return ''
|
||||||
const providerActive = selectedProviderData.value?.is_active
|
const providerActive = selectedProviderData.value?.is_active
|
||||||
@@ -1600,6 +1728,7 @@ const currentPage = ref(restoredViewState.page)
|
|||||||
const pageSize = ref(restoredViewState.pageSize)
|
const pageSize = ref(restoredViewState.pageSize)
|
||||||
const sortBy = ref<PoolManagementSortBy | null>(restoredViewState.sortBy)
|
const sortBy = ref<PoolManagementSortBy | null>(restoredViewState.sortBy)
|
||||||
const sortOrder = ref<PoolManagementSortOrder>(restoredViewState.sortOrder)
|
const sortOrder = ref<PoolManagementSortOrder>(restoredViewState.sortOrder)
|
||||||
|
const poolStatsMode = ref<PoolManagementStatsMode>(restoredViewState.statsMode)
|
||||||
const hasPoolKeyFilters = computed(() => searchQuery.value.trim().length > 0 || statusFilter.value !== 'all')
|
const hasPoolKeyFilters = computed(() => searchQuery.value.trim().length > 0 || statusFilter.value !== 'all')
|
||||||
const MANUAL_QUOTA_REFRESH_COOLDOWN_SECONDS = 5 * 60
|
const MANUAL_QUOTA_REFRESH_COOLDOWN_SECONDS = 5 * 60
|
||||||
const refreshingOAuthKeyId = ref<string | null>(null)
|
const refreshingOAuthKeyId = ref<string | null>(null)
|
||||||
@@ -1681,6 +1810,18 @@ watch(
|
|||||||
{ immediate: true },
|
{ immediate: true },
|
||||||
)
|
)
|
||||||
|
|
||||||
|
watch(
|
||||||
|
() => readPoolManagementViewState(
|
||||||
|
{ statsMode: getQueryValue('statsMode') },
|
||||||
|
poolManagementViewStorage,
|
||||||
|
).statsMode,
|
||||||
|
(value) => {
|
||||||
|
if (poolStatsMode.value === value) return
|
||||||
|
poolStatsMode.value = value
|
||||||
|
},
|
||||||
|
{ immediate: true },
|
||||||
|
)
|
||||||
|
|
||||||
watch(
|
watch(
|
||||||
() => getQueryValue('providerId'),
|
() => getQueryValue('providerId'),
|
||||||
(value) => {
|
(value) => {
|
||||||
@@ -1697,8 +1838,8 @@ watch(
|
|||||||
)
|
)
|
||||||
|
|
||||||
watch(
|
watch(
|
||||||
[selectedProviderId, searchQuery, statusFilter, currentPage, pageSize, sortBy, sortOrder],
|
[selectedProviderId, searchQuery, statusFilter, currentPage, pageSize, sortBy, sortOrder, poolStatsMode],
|
||||||
([providerId, search, status, page, pageSizeValue, sortByValue, sortOrderValue]) => {
|
([providerId, search, status, page, pageSizeValue, sortByValue, sortOrderValue, statsMode]) => {
|
||||||
const nextState: PoolManagementViewState = {
|
const nextState: PoolManagementViewState = {
|
||||||
providerId,
|
providerId,
|
||||||
search,
|
search,
|
||||||
@@ -1707,6 +1848,7 @@ watch(
|
|||||||
pageSize: pageSizeValue,
|
pageSize: pageSizeValue,
|
||||||
sortBy: sortByValue,
|
sortBy: sortByValue,
|
||||||
sortOrder: sortOrderValue,
|
sortOrder: sortOrderValue,
|
||||||
|
statsMode: statsMode as PoolManagementStatsMode,
|
||||||
}
|
}
|
||||||
patchQuery(buildPoolManagementQueryPatch(nextState))
|
patchQuery(buildPoolManagementQueryPatch(nextState))
|
||||||
writePoolManagementViewState(nextState, poolManagementViewStorage)
|
writePoolManagementViewState(nextState, poolManagementViewStorage)
|
||||||
@@ -1739,6 +1881,7 @@ type PoolKeyUiState = {
|
|||||||
quotaTextClass: string
|
quotaTextClass: string
|
||||||
importedAtRelative: string
|
importedAtRelative: string
|
||||||
lastUsedRelative: string
|
lastUsedRelative: string
|
||||||
|
statsDisplay: PoolStatsDisplay
|
||||||
mobileTagItems: PoolMobileTagItem[]
|
mobileTagItems: PoolMobileTagItem[]
|
||||||
mobileActionIds: PoolMobileActionId[]
|
mobileActionIds: PoolMobileActionId[]
|
||||||
}
|
}
|
||||||
@@ -1778,6 +1921,7 @@ const keyUiStateMap = computed<Record<string, PoolKeyUiState>>(() => {
|
|||||||
quotaTextClass: quotaFallbackText ? getQuotaTextClass(quotaFallbackText) : '',
|
quotaTextClass: quotaFallbackText ? getQuotaTextClass(quotaFallbackText) : '',
|
||||||
importedAtRelative: formatPoolKeyImportedAt(key),
|
importedAtRelative: formatPoolKeyImportedAt(key),
|
||||||
lastUsedRelative: key.last_used_at ? formatRelativeTime(key.last_used_at) : '-',
|
lastUsedRelative: key.last_used_at ? formatRelativeTime(key.last_used_at) : '-',
|
||||||
|
statsDisplay: buildPoolStatsDisplay(key, selectedProviderType.value, poolStatsMode.value),
|
||||||
mobileTagItems: getMobileTagItems(key),
|
mobileTagItems: getMobileTagItems(key),
|
||||||
mobileActionIds: splitPoolMobileActions({
|
mobileActionIds: splitPoolMobileActions({
|
||||||
canDownloadOrCopy: true,
|
canDownloadOrCopy: true,
|
||||||
@@ -1791,6 +1935,27 @@ const keyUiStateMap = computed<Record<string, PoolKeyUiState>>(() => {
|
|||||||
return map
|
return map
|
||||||
})
|
})
|
||||||
|
|
||||||
|
function getPoolKeyStatsDisplay(key: PoolKeyDetail): PoolStatsDisplay {
|
||||||
|
return keyUiStateMap.value[key.key_id]?.statsDisplay
|
||||||
|
?? buildPoolStatsDisplay(key, selectedProviderType.value, poolStatsMode.value)
|
||||||
|
}
|
||||||
|
|
||||||
|
function isPoolKeyCycleStatsDisplay(key: PoolKeyDetail): boolean {
|
||||||
|
return getPoolKeyStatsDisplay(key).kind === 'codex_cycle'
|
||||||
|
}
|
||||||
|
|
||||||
|
function getPoolKeyCycleStatsGroups(key: PoolKeyDetail): PoolCodexCycleStatsGroup[] {
|
||||||
|
const display = getPoolKeyStatsDisplay(key)
|
||||||
|
return display.kind === 'codex_cycle' ? display.groups : []
|
||||||
|
}
|
||||||
|
|
||||||
|
function getPoolKeyAccountStatsMetrics(key: PoolKeyDetail): PoolStatsMetric[] {
|
||||||
|
const display = getPoolKeyStatsDisplay(key)
|
||||||
|
return display.kind === 'account_total'
|
||||||
|
? display.metrics
|
||||||
|
: buildPoolStatsDisplay(key, selectedProviderType.value, 'account_total').metrics
|
||||||
|
}
|
||||||
|
|
||||||
const quotaRefreshSupported = computed(() => {
|
const quotaRefreshSupported = computed(() => {
|
||||||
return selectedProviderType.value === 'codex'
|
return selectedProviderType.value === 'codex'
|
||||||
|| selectedProviderType.value === 'kiro'
|
|| selectedProviderType.value === 'kiro'
|
||||||
|
|||||||
@@ -0,0 +1,599 @@
|
|||||||
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
|
||||||
|
import { createApp, nextTick, type App } from 'vue'
|
||||||
|
|
||||||
|
import PoolManagement from '@/views/admin/PoolManagement.vue'
|
||||||
|
import type { PoolKeyDetail, PoolOverviewItem, PoolKeysPageResponse } from '@/api/endpoints/pool'
|
||||||
|
import { POOL_MANAGEMENT_VIEW_STORAGE_KEY } from '@/features/pool/utils/poolManagementState'
|
||||||
|
|
||||||
|
const endpointMocks = vi.hoisted(() => ({
|
||||||
|
getPoolOverview: vi.fn(),
|
||||||
|
getPoolSchedulingPresets: vi.fn(),
|
||||||
|
listPoolKeys: vi.fn(),
|
||||||
|
clearPoolCooldown: vi.fn(),
|
||||||
|
getProvider: vi.fn(),
|
||||||
|
updateProvider: vi.fn(),
|
||||||
|
revealEndpointKey: vi.fn(),
|
||||||
|
exportKey: vi.fn(),
|
||||||
|
deleteEndpointKey: vi.fn(),
|
||||||
|
updateProviderKey: vi.fn(),
|
||||||
|
refreshProviderQuota: vi.fn(),
|
||||||
|
refreshProviderOAuth: vi.fn(),
|
||||||
|
}))
|
||||||
|
|
||||||
|
const routeMocks = vi.hoisted(() => ({
|
||||||
|
query: {} as Record<string, string>,
|
||||||
|
patchQuery: vi.fn((patch: Record<string, string | undefined | null>) => {
|
||||||
|
for (const [key, value] of Object.entries(patch)) {
|
||||||
|
if (value == null || String(value).trim() === '') {
|
||||||
|
delete routeMocks.query[key]
|
||||||
|
} else {
|
||||||
|
routeMocks.query[key] = String(value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
}))
|
||||||
|
|
||||||
|
const proxyStoreMocks = vi.hoisted(() => ({
|
||||||
|
ensureLoaded: vi.fn(),
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/api/endpoints/pool', () => ({
|
||||||
|
getPoolOverview: endpointMocks.getPoolOverview,
|
||||||
|
getPoolSchedulingPresets: endpointMocks.getPoolSchedulingPresets,
|
||||||
|
listPoolKeys: endpointMocks.listPoolKeys,
|
||||||
|
clearPoolCooldown: endpointMocks.clearPoolCooldown,
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/api/endpoints/keys', () => ({
|
||||||
|
revealEndpointKey: endpointMocks.revealEndpointKey,
|
||||||
|
exportKey: endpointMocks.exportKey,
|
||||||
|
deleteEndpointKey: endpointMocks.deleteEndpointKey,
|
||||||
|
updateProviderKey: endpointMocks.updateProviderKey,
|
||||||
|
refreshProviderQuota: endpointMocks.refreshProviderQuota,
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/api/endpoints/provider_oauth', () => ({
|
||||||
|
refreshProviderOAuth: endpointMocks.refreshProviderOAuth,
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/api/endpoints', () => ({
|
||||||
|
getProvider: endpointMocks.getProvider,
|
||||||
|
updateProvider: endpointMocks.updateProvider,
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/composables/useRouteQuery', () => ({
|
||||||
|
useRouteQuery: () => ({
|
||||||
|
getQueryValue: (key: string) => routeMocks.query[key],
|
||||||
|
patchQuery: routeMocks.patchQuery,
|
||||||
|
}),
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/stores/proxy-nodes', () => ({
|
||||||
|
useProxyNodesStore: () => ({
|
||||||
|
nodes: [],
|
||||||
|
ensureLoaded: proxyStoreMocks.ensureLoaded,
|
||||||
|
}),
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/composables/useToast', () => ({
|
||||||
|
useToast: () => ({
|
||||||
|
success: vi.fn(),
|
||||||
|
error: vi.fn(),
|
||||||
|
warning: vi.fn(),
|
||||||
|
}),
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/composables/useConfirm', () => ({
|
||||||
|
useConfirm: () => ({
|
||||||
|
confirm: vi.fn().mockResolvedValue(true),
|
||||||
|
}),
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/composables/useClipboard', () => ({
|
||||||
|
useClipboard: () => ({
|
||||||
|
copyToClipboard: vi.fn().mockResolvedValue(undefined),
|
||||||
|
}),
|
||||||
|
}))
|
||||||
|
|
||||||
|
vi.mock('@/composables/useCountdownTimer', async () => {
|
||||||
|
const { ref } = await import('vue')
|
||||||
|
return {
|
||||||
|
useCountdownTimer: () => ({
|
||||||
|
tick: ref(0),
|
||||||
|
start: vi.fn(),
|
||||||
|
}),
|
||||||
|
getCodexResetCountdown: () => ({
|
||||||
|
isExpired: false,
|
||||||
|
text: '1h',
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
vi.mock('lucide-vue-next', async () => {
|
||||||
|
const { defineComponent, h } = await import('vue')
|
||||||
|
const Icon = defineComponent({
|
||||||
|
name: 'IconStub',
|
||||||
|
setup() {
|
||||||
|
return () => h('span')
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
return {
|
||||||
|
Search: Icon,
|
||||||
|
Upload: Icon,
|
||||||
|
ChevronDown: Icon,
|
||||||
|
RefreshCw: Icon,
|
||||||
|
Power: Icon,
|
||||||
|
Database: Icon,
|
||||||
|
KeyRound: Icon,
|
||||||
|
Download: Icon,
|
||||||
|
Copy: Icon,
|
||||||
|
Shield: Icon,
|
||||||
|
Globe: Icon,
|
||||||
|
SquarePen: Icon,
|
||||||
|
Trash2: Icon,
|
||||||
|
Users: Icon,
|
||||||
|
Settings2: Icon,
|
||||||
|
SlidersHorizontal: Icon,
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
vi.mock('@/components/ui', async () => {
|
||||||
|
const { defineComponent, h } = await import('vue')
|
||||||
|
const passthrough = (name: string, tag = 'div') => defineComponent({
|
||||||
|
name,
|
||||||
|
inheritAttrs: false,
|
||||||
|
setup(_, { attrs, slots }) {
|
||||||
|
return () => h(tag, attrs, slots.default?.())
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const Button = defineComponent({
|
||||||
|
name: 'ButtonStub',
|
||||||
|
inheritAttrs: false,
|
||||||
|
props: {
|
||||||
|
disabled: Boolean,
|
||||||
|
},
|
||||||
|
setup(props, { attrs, slots }) {
|
||||||
|
return () => h('button', { ...attrs, disabled: props.disabled, type: attrs.type ?? 'button' }, slots.default?.())
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const Input = defineComponent({
|
||||||
|
name: 'InputStub',
|
||||||
|
inheritAttrs: false,
|
||||||
|
props: {
|
||||||
|
modelValue: { type: [String, Number], default: '' },
|
||||||
|
},
|
||||||
|
emits: ['update:modelValue'],
|
||||||
|
setup(props, { attrs, emit }) {
|
||||||
|
return () => h('input', {
|
||||||
|
...attrs,
|
||||||
|
value: props.modelValue ?? '',
|
||||||
|
onInput: (event: Event) => emit('update:modelValue', (event.target as HTMLInputElement).value),
|
||||||
|
})
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const Switch = defineComponent({
|
||||||
|
name: 'SwitchStub',
|
||||||
|
inheritAttrs: false,
|
||||||
|
props: {
|
||||||
|
modelValue: Boolean,
|
||||||
|
},
|
||||||
|
emits: ['update:modelValue'],
|
||||||
|
setup(props, { attrs, emit }) {
|
||||||
|
return () => h('input', {
|
||||||
|
...attrs,
|
||||||
|
type: 'checkbox',
|
||||||
|
role: 'switch',
|
||||||
|
checked: props.modelValue,
|
||||||
|
onChange: (event: Event) => emit('update:modelValue', (event.target as HTMLInputElement).checked),
|
||||||
|
})
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const Pagination = defineComponent({
|
||||||
|
name: 'PaginationStub',
|
||||||
|
setup() {
|
||||||
|
return () => h('nav')
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
return {
|
||||||
|
Card: passthrough('CardStub'),
|
||||||
|
Badge: passthrough('BadgeStub', 'span'),
|
||||||
|
Button,
|
||||||
|
Input,
|
||||||
|
Select: passthrough('SelectStub'),
|
||||||
|
SelectTrigger: passthrough('SelectTriggerStub', 'button'),
|
||||||
|
SelectValue: passthrough('SelectValueStub', 'span'),
|
||||||
|
SelectContent: passthrough('SelectContentStub'),
|
||||||
|
SelectItem: passthrough('SelectItemStub'),
|
||||||
|
Table: passthrough('TableStub', 'table'),
|
||||||
|
TableHeader: passthrough('TableHeaderStub', 'thead'),
|
||||||
|
TableBody: passthrough('TableBodyStub', 'tbody'),
|
||||||
|
TableRow: passthrough('TableRowStub', 'tr'),
|
||||||
|
TableHead: passthrough('TableHeadStub', 'th'),
|
||||||
|
SortableTableHead: passthrough('SortableTableHeadStub', 'th'),
|
||||||
|
TableFilterMenu: passthrough('TableFilterMenuStub'),
|
||||||
|
TableCell: passthrough('TableCellStub', 'td'),
|
||||||
|
Switch,
|
||||||
|
Pagination,
|
||||||
|
Popover: passthrough('PopoverStub'),
|
||||||
|
PopoverTrigger: passthrough('PopoverTriggerStub'),
|
||||||
|
PopoverContent: passthrough('PopoverContentStub'),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
vi.mock('@/components/ui/refresh-button.vue', async () => {
|
||||||
|
const { defineComponent, h } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'RefreshButtonStub',
|
||||||
|
setup(_, { attrs }) {
|
||||||
|
return () => h('button', attrs, '刷新')
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
vi.mock('@/features/pool/components/PoolSchedulingDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'PoolSchedulingDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/pool/components/PoolAdvancedDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'PoolAdvancedDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/pool/components/PoolAccountBatchDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'PoolAccountBatchDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/pool/components/ProviderProxyPopover.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'ProviderProxyPopoverStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/providers/components/KeyAllowedModelsEditDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'KeyAllowedModelsEditDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/providers/components/KeyFormDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'KeyFormDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/providers/components/OAuthKeyEditDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'OAuthKeyEditDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/providers/components/OAuthAccountDialog.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'OAuthAccountDialogStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
vi.mock('@/features/providers/components/ProxyNodeSelect.vue', async () => {
|
||||||
|
const { defineComponent } = await import('vue')
|
||||||
|
return {
|
||||||
|
default: defineComponent({
|
||||||
|
name: 'ProxyNodeSelectStub',
|
||||||
|
setup() {
|
||||||
|
return () => null
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
const mountedApps: Array<{ app: App, root: HTMLElement }> = []
|
||||||
|
|
||||||
|
function createOverview(providerType: string): PoolOverviewItem {
|
||||||
|
return {
|
||||||
|
provider_id: `${providerType}-provider`,
|
||||||
|
provider_name: `${providerType} Provider`,
|
||||||
|
provider_type: providerType,
|
||||||
|
total_keys: 1,
|
||||||
|
active_keys: 1,
|
||||||
|
cooldown_count: 0,
|
||||||
|
pool_enabled: true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function createProvider(providerType: string) {
|
||||||
|
return {
|
||||||
|
id: `${providerType}-provider`,
|
||||||
|
name: `${providerType} Provider`,
|
||||||
|
provider_type: providerType,
|
||||||
|
is_active: true,
|
||||||
|
api_formats: ['openai:chat'],
|
||||||
|
proxy: null,
|
||||||
|
pool_advanced: null,
|
||||||
|
claude_code_advanced: null,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function createPoolKey(providerType = 'codex', overrides: Partial<PoolKeyDetail> = {}): PoolKeyDetail {
|
||||||
|
return {
|
||||||
|
key_id: `${providerType}-key-1`,
|
||||||
|
key_name: `${providerType} key`,
|
||||||
|
is_active: true,
|
||||||
|
auth_type: 'api_key',
|
||||||
|
api_formats: ['openai:chat'],
|
||||||
|
internal_priority: 50,
|
||||||
|
account_quota: null,
|
||||||
|
cooldown_reason: null,
|
||||||
|
cooldown_ttl_seconds: null,
|
||||||
|
cost_window_usage: 0,
|
||||||
|
cost_limit: null,
|
||||||
|
request_count: 9876,
|
||||||
|
total_tokens: 4321000,
|
||||||
|
total_cost_usd: '8.7654',
|
||||||
|
sticky_sessions: 0,
|
||||||
|
lru_score: null,
|
||||||
|
created_at: '2026-05-05T00:00:00Z',
|
||||||
|
imported_at: '2026-05-05T00:00:00Z',
|
||||||
|
last_used_at: '2026-05-05T01:00:00Z',
|
||||||
|
status_snapshot: {
|
||||||
|
oauth: { code: 'none' },
|
||||||
|
account: { code: 'ok', blocked: false },
|
||||||
|
quota: {
|
||||||
|
code: 'ok',
|
||||||
|
exhausted: false,
|
||||||
|
provider_type: providerType,
|
||||||
|
windows: providerType === 'codex'
|
||||||
|
? [
|
||||||
|
{
|
||||||
|
code: '5h',
|
||||||
|
remaining_ratio: 0.8,
|
||||||
|
usage: { request_count: 7, total_tokens: 2500, total_cost_usd: '0.0045' },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
code: 'weekly',
|
||||||
|
remaining_ratio: 0.5,
|
||||||
|
usage: { request_count: 0, total_tokens: 0, total_cost_usd: '0.00000000' },
|
||||||
|
},
|
||||||
|
]
|
||||||
|
: [],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
...overrides,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function createKeyPage(key: PoolKeyDetail): PoolKeysPageResponse {
|
||||||
|
return {
|
||||||
|
total: 1,
|
||||||
|
page: 1,
|
||||||
|
page_size: 50,
|
||||||
|
keys: [key],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function resetQuery() {
|
||||||
|
for (const key of Object.keys(routeMocks.query)) {
|
||||||
|
delete routeMocks.query[key]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function mountPoolManagement() {
|
||||||
|
const root = document.createElement('div')
|
||||||
|
document.body.appendChild(root)
|
||||||
|
const app = createApp(PoolManagement)
|
||||||
|
app.mount(root)
|
||||||
|
mountedApps.push({ app, root })
|
||||||
|
return root
|
||||||
|
}
|
||||||
|
|
||||||
|
async function settle() {
|
||||||
|
for (let index = 0; index < 8; index += 1) {
|
||||||
|
await Promise.resolve()
|
||||||
|
await nextTick()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function seedStoredStatsMode(statsMode: 'current_cycle' | 'account_total') {
|
||||||
|
window.sessionStorage.setItem(
|
||||||
|
POOL_MANAGEMENT_VIEW_STORAGE_KEY,
|
||||||
|
JSON.stringify({ statsMode }),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
resetQuery()
|
||||||
|
window.sessionStorage.clear()
|
||||||
|
routeMocks.patchQuery.mockClear()
|
||||||
|
proxyStoreMocks.ensureLoaded.mockClear()
|
||||||
|
|
||||||
|
endpointMocks.getPoolOverview.mockReset()
|
||||||
|
endpointMocks.getPoolSchedulingPresets.mockReset()
|
||||||
|
endpointMocks.listPoolKeys.mockReset()
|
||||||
|
endpointMocks.clearPoolCooldown.mockReset()
|
||||||
|
endpointMocks.getProvider.mockReset()
|
||||||
|
endpointMocks.updateProvider.mockReset()
|
||||||
|
endpointMocks.revealEndpointKey.mockReset()
|
||||||
|
endpointMocks.exportKey.mockReset()
|
||||||
|
endpointMocks.deleteEndpointKey.mockReset()
|
||||||
|
endpointMocks.updateProviderKey.mockReset()
|
||||||
|
endpointMocks.refreshProviderQuota.mockReset()
|
||||||
|
endpointMocks.refreshProviderOAuth.mockReset()
|
||||||
|
|
||||||
|
endpointMocks.getPoolSchedulingPresets.mockResolvedValue([])
|
||||||
|
endpointMocks.clearPoolCooldown.mockResolvedValue({ message: 'ok' })
|
||||||
|
endpointMocks.refreshProviderQuota.mockResolvedValue({ success: 0, failed: 0 })
|
||||||
|
})
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
for (const { app, root } of mountedApps.splice(0)) {
|
||||||
|
app.unmount()
|
||||||
|
root.remove()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
describe('PoolManagement Codex cycle stats mode', () => {
|
||||||
|
it('defaults Codex providers to current-cycle groups and toggles back to account totals', async () => {
|
||||||
|
const codexKey = createPoolKey('codex')
|
||||||
|
endpointMocks.getPoolOverview.mockResolvedValue({ items: [createOverview('codex')] })
|
||||||
|
endpointMocks.listPoolKeys.mockResolvedValue(createKeyPage(codexKey))
|
||||||
|
endpointMocks.getProvider.mockResolvedValue(createProvider('codex'))
|
||||||
|
|
||||||
|
const root = mountPoolManagement()
|
||||||
|
await settle()
|
||||||
|
|
||||||
|
const modeSwitch = root.querySelector<HTMLInputElement>('[data-testid="pool-stats-mode-switch"]')
|
||||||
|
expect(modeSwitch).not.toBeNull()
|
||||||
|
expect(modeSwitch?.checked).toBe(true)
|
||||||
|
expect(root.querySelectorAll('[data-testid="pool-stats-cycle-group-5h"]').length).toBeGreaterThan(0)
|
||||||
|
expect(root.querySelectorAll('[data-testid="pool-stats-cycle-group-weekly"]').length).toBeGreaterThan(0)
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-5h-request_count"]')?.textContent?.trim()).toBe('7')
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-weekly-total_tokens"]')?.textContent?.trim()).toBe('0')
|
||||||
|
|
||||||
|
if (!modeSwitch) throw new Error('expected stats switch')
|
||||||
|
modeSwitch.checked = false
|
||||||
|
modeSwitch.dispatchEvent(new Event('change', { bubbles: true }))
|
||||||
|
await settle()
|
||||||
|
|
||||||
|
expect(routeMocks.query.statsMode).toBe('account_total')
|
||||||
|
expect(window.sessionStorage.getItem(POOL_MANAGEMENT_VIEW_STORAGE_KEY)).toContain('"statsMode":"account_total"')
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-account-total"]')).not.toBeNull()
|
||||||
|
expect(root.textContent).toContain('9,876')
|
||||||
|
expect(root.textContent).toContain('4.3M')
|
||||||
|
expect(root.textContent).toContain('$8.77')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('renders the Codex stats switch in header actions instead of a standalone mode bar', async () => {
|
||||||
|
const codexKey = createPoolKey('codex')
|
||||||
|
endpointMocks.getPoolOverview.mockResolvedValue({ items: [createOverview('codex')] })
|
||||||
|
endpointMocks.listPoolKeys.mockResolvedValue(createKeyPage(codexKey))
|
||||||
|
endpointMocks.getProvider.mockResolvedValue(createProvider('codex'))
|
||||||
|
|
||||||
|
const root = mountPoolManagement()
|
||||||
|
await settle()
|
||||||
|
|
||||||
|
const desktopHeaderActions = root.querySelector('[data-testid="pool-header-actions"]')
|
||||||
|
const mobileHeaderActions = root.querySelector('[data-testid="pool-mobile-header-actions"]')
|
||||||
|
const modeControls = Array.from(root.querySelectorAll('[data-testid="pool-stats-mode-control"]'))
|
||||||
|
|
||||||
|
expect(desktopHeaderActions?.querySelector('[data-testid="pool-stats-mode-control"]')).not.toBeNull()
|
||||||
|
expect(mobileHeaderActions?.querySelector('[data-testid="pool-stats-mode-control"]')).not.toBeNull()
|
||||||
|
expect(modeControls).toHaveLength(2)
|
||||||
|
expect(modeControls.every(control => control.closest('[data-testid="pool-header-actions"], [data-testid="pool-mobile-header-actions"]'))).toBe(true)
|
||||||
|
expect(desktopHeaderActions?.textContent).toContain('累计')
|
||||||
|
expect(desktopHeaderActions?.textContent).toContain('周期')
|
||||||
|
expect(root.textContent).not.toContain('Codex 统计模式')
|
||||||
|
expect(root.textContent).not.toContain('当前周期显示 5H 与周窗口')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('restores stored Codex account-total mode when the query omits statsMode', async () => {
|
||||||
|
seedStoredStatsMode('account_total')
|
||||||
|
const codexKey = createPoolKey('codex')
|
||||||
|
endpointMocks.getPoolOverview.mockResolvedValue({ items: [createOverview('codex')] })
|
||||||
|
endpointMocks.listPoolKeys.mockResolvedValue(createKeyPage(codexKey))
|
||||||
|
endpointMocks.getProvider.mockResolvedValue(createProvider('codex'))
|
||||||
|
|
||||||
|
const root = mountPoolManagement()
|
||||||
|
await settle()
|
||||||
|
|
||||||
|
const modeSwitch = root.querySelector<HTMLInputElement>('[data-testid="pool-stats-mode-switch"]')
|
||||||
|
expect(modeSwitch).not.toBeNull()
|
||||||
|
expect(modeSwitch?.checked).toBe(false)
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-account-total"]')).not.toBeNull()
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-cycle-group-5h"]')).toBeNull()
|
||||||
|
expect(routeMocks.query.statsMode).toBe('account_total')
|
||||||
|
expect(window.sessionStorage.getItem(POOL_MANAGEMENT_VIEW_STORAGE_KEY)).toContain('"statsMode":"account_total"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('lets a current-cycle statsMode query override stored Codex account-total mode', async () => {
|
||||||
|
seedStoredStatsMode('account_total')
|
||||||
|
routeMocks.query.statsMode = 'current_cycle'
|
||||||
|
const codexKey = createPoolKey('codex')
|
||||||
|
endpointMocks.getPoolOverview.mockResolvedValue({ items: [createOverview('codex')] })
|
||||||
|
endpointMocks.listPoolKeys.mockResolvedValue(createKeyPage(codexKey))
|
||||||
|
endpointMocks.getProvider.mockResolvedValue(createProvider('codex'))
|
||||||
|
|
||||||
|
const root = mountPoolManagement()
|
||||||
|
await settle()
|
||||||
|
|
||||||
|
const modeSwitch = root.querySelector<HTMLInputElement>('[data-testid="pool-stats-mode-switch"]')
|
||||||
|
expect(modeSwitch).not.toBeNull()
|
||||||
|
expect(modeSwitch?.checked).toBe(true)
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-cycle-group-5h"]')).not.toBeNull()
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-account-total"]')).toBeNull()
|
||||||
|
expect(routeMocks.query.statsMode).toBeUndefined()
|
||||||
|
expect(window.sessionStorage.getItem(POOL_MANAGEMENT_VIEW_STORAGE_KEY)).toContain('"statsMode":"current_cycle"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('hides the stats mode switch for non-Codex providers and keeps account totals', async () => {
|
||||||
|
const openaiKey = createPoolKey('openai', {
|
||||||
|
request_count: 12,
|
||||||
|
total_tokens: 3456,
|
||||||
|
total_cost_usd: '1.25',
|
||||||
|
})
|
||||||
|
endpointMocks.getPoolOverview.mockResolvedValue({ items: [createOverview('openai')] })
|
||||||
|
endpointMocks.listPoolKeys.mockResolvedValue(createKeyPage(openaiKey))
|
||||||
|
endpointMocks.getProvider.mockResolvedValue(createProvider('openai'))
|
||||||
|
|
||||||
|
const root = mountPoolManagement()
|
||||||
|
await settle()
|
||||||
|
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-mode-switch"]')).toBeNull()
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-mode-control"]')).toBeNull()
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-cycle-group-5h"]')).toBeNull()
|
||||||
|
expect(root.querySelector('[data-testid="pool-stats-account-total"]')).not.toBeNull()
|
||||||
|
expect(root.textContent).toContain('12')
|
||||||
|
expect(root.textContent).toContain('3.5K')
|
||||||
|
expect(root.textContent).toContain('$1.25')
|
||||||
|
})
|
||||||
|
})
|
||||||
Reference in New Issue
Block a user