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

feat(providers): support FedRAMP Codex agent identity registration

Browse Source
This commit is contained in:
ZheFox
2026-07-28 13:29:30 +08:00
parent 3043982486
commit 1b697b1111
3 changed files with 84 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/handlers/admin/provider/oauth/dispatch/import.rs
+8 -1
View File
@@ -343,10 +343,15 @@ async fn resolve_admin_provider_oauth_codex_access_token_agent_identity_import(
request_proxy: Option<ProxySnapshot>,
) -> Result<AdminProviderOAuthSingleImportTokens, Response<Body>> {
let executor = crate::oauth::GatewayOAuthHttpExecutor::new(*state);
let is_fedramp_account = identity_hints
.get("is_fedramp")
.and_then(serde_json::Value::as_bool)
.unwrap_or(false);
let mut auth_config =
aether_provider_transport::register_codex_agent_identity_from_access_token(
&executor,
access_token,
aether_provider_transport::CodexAgentIdentityRegistrationOptions { is_fedramp_account },
OAuthNetworkContext::provider_operation(request_proxy),
)
.await
@@ -1278,7 +1283,8 @@ mod tests {
"https://api.openai.com/auth": {
"chatgpt_account_id": "account-1",
"chatgpt_user_id": "user-1",
"chatgpt_plan_type": "plus"
"chatgpt_plan_type": "plus",
"chatgpt_account_is_fedramp": true
},
"https://api.openai.com/profile": {
"email": "[email protected]"
@@ -1291,6 +1297,7 @@ mod tests {
assert_eq!(hints.get("account_id"), Some(&json!("account-1")));
assert_eq!(hints.get("user_id"), Some(&json!("user-1")));
assert_eq!(hints.get("plan_type"), Some(&json!("plus")));
assert_eq!(hints.get("is_fedramp"), Some(&json!(true)));
assert_eq!(hints.get("email"), Some(&json!("[email protected]")));
assert!(!hints.contains_key("access_token"));
assert!(!hints.contains_key("id_token"));