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

fix(provider): support unversioned API roots in model fetch

Browse Source
This commit is contained in:
zhefox
2026-05-26 15:39:56 +08:00
parent c4927162b7
commit d28a389a93
11 changed files with 525 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/aether-model-fetch/src/transport.rs
+43
View File
@@ -807,6 +807,49 @@ mod tests {
);
}
#[tokio::test]
async fn builds_bigmodel_coding_models_fetch_plan() {
let runtime = TestRuntime {
oauth_auth: None,
proxy: None,
};
let mut transport = sample_transport("openai", "openai:chat", "api_key");
transport.endpoint.base_url = "https://open.bigmodel.cn/api/coding/paas/v4".to_string();
transport.key.decrypted_auth_config = None;
let plan = build_models_fetch_execution_plan(&runtime, &transport)
.await
.expect("plan");
assert_eq!(
plan.url,
"https://open.bigmodel.cn/api/coding/paas/v4/models/models"
);
assert_eq!(
plan.headers.get("authorization").map(String::as_str),
Some("Bearer secret")
);
}
#[tokio::test]
async fn builds_unversioned_api_root_models_fetch_plan() {
let runtime = TestRuntime {
oauth_auth: None,
proxy: None,
};
let mut transport = sample_transport("openai", "openai:chat", "api_key");
transport.endpoint.base_url = "https://proxy.example.com/api".to_string();
transport.key.decrypted_auth_config = None;
let plan = build_models_fetch_execution_plan(&runtime, &transport)
.await
.expect("plan");
assert_eq!(plan.url, "https://proxy.example.com/api/models");
assert_eq!(
plan.headers.get("authorization").map(String::as_str),
Some("Bearer secret")
);
}
#[tokio::test]
async fn builds_codex_models_fetch_plan_with_account_header() {
let runtime = TestRuntime {