i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 17:30:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix OpenAI family local auth to use bearer (#302)

Browse Source 
* Fix OpenAI family local auth to use bearer

* test(gateway): fix bearer auth assertions for openai local flows

* test(usage): make local usage status wait resilient

* style(gateway): apply rustfmt to usage test helper

---------

Co-authored-by: fawney19 <elky0401@gmail.com>
This commit is contained in:
RWDai
2026-04-17 11:10:05 +08:00
committed by GitHub
parent 6964729cb7
commit 96a25d058b
15 changed files with 89 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
apps/aether-gateway/src/tests/ai_execute/finalize_local_cli/compact.rs
View File

@@ -87,7 +87,7 @@ async fn gateway_executes_openai_compact_openai_family_upstream_stream_via_local
endpoint_is_active: true,
key_id: "key-openai-compact-finalize-local-1".to_string(),
key_name: "prod".to_string(),
key_auth_type: "bearer".to_string(),
key_auth_type: "api_key".to_string(),
key_is_active: true,
key_api_formats: Some(vec!["openai:compact".to_string()]),
key_allowed_models: None,
@@ -162,7 +162,7 @@ async fn gateway_executes_openai_compact_openai_family_upstream_stream_via_local
"key-openai-compact-finalize-local-1".to_string(),
"provider-openai-compact-finalize-local-1".to_string(),
"prod".to_string(),
"bearer".to_string(),
"api_key".to_string(),
None,
true,
)

10
apps/aether-gateway/src/tests/ai_execute/stream/decision.rs
View File

@@ -449,7 +449,7 @@ async fn gateway_executes_openai_chat_stream_via_local_openai_cli_cross_format_c
model: String,
stream: bool,
accept: String,
api_key: String,
authorization: String,
x_client_request_id: String,
session_id: String,
conversation_id: String,
@@ -717,9 +717,9 @@ async fn gateway_executes_openai_chat_stream_via_local_openai_cli_cross_format_c
.and_then(|value| value.as_str())
.unwrap_or_default()
.to_string(),
api_key: payload
authorization: payload
.get("headers")
.and_then(|value| value.get("x-api-key"))
.and_then(|value| value.get("authorization"))
.and_then(|value| value.as_str())
.unwrap_or_default()
.to_string(),
@@ -871,8 +871,8 @@ async fn gateway_executes_openai_chat_stream_via_local_openai_cli_cross_format_c
assert!(seen_execution_runtime_request.stream);
assert_eq!(seen_execution_runtime_request.accept, "text/event-stream");
assert_eq!(
seen_execution_runtime_request.api_key,
"sk-upstream-openai-chat-cli"
seen_execution_runtime_request.authorization,
"Bearer sk-upstream-openai-chat-cli"
);
assert_eq!(
seen_execution_runtime_request.x_client_request_id,

4
apps/aether-gateway/src/tests/ai_execute/sync/cli.rs
View File

@@ -88,7 +88,7 @@ async fn gateway_executes_openai_cli_sync_via_local_decision_gate_with_local_syn
endpoint_is_active: true,
key_id: "key-openai-cli-local-1".to_string(),
key_name: "prod".to_string(),
key_auth_type: "bearer".to_string(),
key_auth_type: "api_key".to_string(),
key_is_active: true,
key_api_formats: Some(vec!["openai:cli".to_string()]),
key_allowed_models: None,
@@ -171,7 +171,7 @@ async fn gateway_executes_openai_cli_sync_via_local_decision_gate_with_local_syn
"key-openai-cli-local-1".to_string(),
"provider-openai-cli-local-1".to_string(),
"prod".to_string(),
"bearer".to_string(),
"api_key".to_string(),
None,
true,
)