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

12
apps/aether-gateway/src/tests/usage/local.rs
View File

@@ -33,7 +33,8 @@ where
T: UsageReadRepository + ?Sized,
{
let mut stored = None;
for _ in 0..50 {
let deadline = tokio::time::Instant::now() + std::time::Duration::from_secs(5);
loop {
stored = repository
.find_by_request_id(request_id)
.await
@@ -44,9 +45,16 @@ where
{
break;
}
if tokio::time::Instant::now() >= deadline {
let observed = stored
.as_ref()
.map(|usage| usage.status.as_str())
.unwrap_or("<missing>");
panic!("usage should reach status {expected_status}, last observed status: {observed}");
}
tokio::time::sleep(std::time::Duration::from_millis(10)).await;
}
stored.unwrap_or_else(|| panic!("usage should reach status {expected_status}"))
stored.expect("usage should be present once the expected status is observed")
}
#[tokio::test]