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

Fix/provider query cli model tests (#312)

Browse Source 
* fix(admin): support openai cli provider-query model tests

* fix(admin): support claude and gemini cli provider-query model tests

* fix(admin): preserve provider-query prompts on cli fallback

* test(usage): relax async status wait for local usage checks

* fix(gateway): align provider-query CLI auth and headers

* fix(gateway): resolve provider-query clippy lint

---------

Co-authored-by: fawney19 <elky0401@gmail.com>
This commit is contained in:
RWDai
2026-04-18 23:05:39 +08:00
committed by GitHub
parent f2a3836877
commit 425227509a
3 changed files with 1245 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1011
apps/aether-gateway/src/tests/control/admin/provider_query.rs
View File

File diff suppressed because it is too large Load Diff

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

@@ -34,7 +34,8 @@ where
T: UsageReadRepository + ?Sized,
{
let mut stored = None;
let deadline = tokio::time::Instant::now() + std::time::Duration::from_secs(5);
let timeout = std::time::Duration::from_secs(10);
let deadline = tokio::time::Instant::now() + timeout;
loop {
stored = repository
.find_by_request_id(request_id)
@@ -51,7 +52,10 @@ where
.as_ref()
.map(|usage| usage.status.as_str())
.unwrap_or("<missing>");
panic!("usage should reach status {expected_status}, last observed status: {observed}");
panic!(
"usage should reach status {expected_status} within {:?}, last observed status: {observed}",
timeout
);
}
tokio::time::sleep(std::time::Duration::from_millis(10)).await;
}