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

Preserve terminal request candidate state

Browse Source
This commit is contained in:
elky
2026-07-02 01:40:57 +08:00
parent 2e5ff32e1a
commit 400b3125a4
5 changed files with 163 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/tests/ai_execute/sync/cli.rs
+9 -7
View File
@@ -1261,21 +1261,23 @@ async fn gateway_executes_openai_responses_sync_after_api_key_concurrency_wait_b
*execution_runtime_hits.lock().expect("mutex should lock") >= 1
})
.await;
let pending_deadline = tokio::time::Instant::now() + std::time::Duration::from_millis(5_000);
let active_deadline = tokio::time::Instant::now() + std::time::Duration::from_millis(5_000);
loop {
let inflight_candidates = request_candidate_repository
.list_by_request_id("trace-openai-cli-local-timeout-inflight-123")
.await
.expect("inflight request candidate trace should read");
if inflight_candidates
.iter()
.any(|candidate| candidate.status == RequestCandidateStatus::Pending)
{
if inflight_candidates.iter().any(|candidate| {
matches!(
candidate.status,
RequestCandidateStatus::Pending | RequestCandidateStatus::Streaming
) && candidate.started_at_unix_ms.is_some()
}) {
break;
}
assert!(
tokio::time::Instant::now() < pending_deadline,
"inflight request candidate did not become pending"
tokio::time::Instant::now() < active_deadline,
"inflight request candidate did not become active: {inflight_candidates:?}"
);
tokio::time::sleep(std::time::Duration::from_millis(10)).await;
}