i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-10-10 11:19:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(provider): harden Agent Identity OAuth lifecycle

Browse Source
This commit is contained in:
elky
2026-07-23 09:33:00 +08:00
parent e49024d33b
commit 3606290ac8
84 changed files with 8063 additions and 1104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/handlers/admin/model/routing.rs
+8 -5
View File
@@ -190,7 +190,10 @@ pub(crate) async fn build_admin_global_model_routing_payload(
let payload = json!({
"id": key.id,
"name": key.name,
"masked_key": state.masked_catalog_api_key(key),
"masked_key": state.masked_catalog_api_key_for_provider(
key,
&provider.provider_type,
),
"is_active": key.is_active,
"is_adaptive": is_adaptive,
"effective_rpm": effective_rpm,
@@ -252,9 +255,9 @@ pub(crate) async fn build_admin_global_model_routing_payload(
.iter()
.map(|provider| provider.id.clone())
.collect::<Vec<_>>();
let active_provider_name_by_id = active_providers
let active_provider_metadata_by_id = active_providers
.into_iter()
.map(|provider| (provider.id, provider.name))
.map(|provider| (provider.id, (provider.name, provider.provider_type)))
.collect::<BTreeMap<_, _>>();
let active_keys = if active_provider_ids.is_empty() {
Vec::new()
@@ -273,14 +276,14 @@ pub(crate) async fn build_admin_global_model_routing_payload(
if allowed_models.is_empty() {
continue;
}
let provider_name = active_provider_name_by_id
let (provider_name, provider_type) = active_provider_metadata_by_id
.get(&key.provider_id)
.cloned()
.unwrap_or_default();
all_keys_whitelist.push(json!({
"key_id": key.id,
"key_name": key.name,
"masked_key": state.masked_catalog_api_key(&key),
"masked_key": state.masked_catalog_api_key_for_provider(&key, &provider_type),
"provider_id": key.provider_id,
"provider_name": provider_name,
"allowed_models": allowed_models,