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: align embedding provider request formats

Browse Source
This commit is contained in:
mayrain
2026-05-12 20:10:52 +08:00
parent fca3f24d91
commit 8f082674d7
4 changed files with 25 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
crates/aether-ai-formats/src/protocol/canonical.rs
View File

@@ -4758,10 +4758,13 @@ mod tests {
);
assert!(gemini.get("messages").is_none());
let doubao =
super::canonical_to_embedding_request(&canonical, "doubao-embedding-vision", "doubao")
.expect("doubao embedding request");
assert_eq!(doubao["input"][0], json!({"type": "text", "text": "alpha"}));
let doubao = super::canonical_to_embedding_request(
&canonical,
"doubao-embedding-text-240515",
"doubao",
)
.expect("doubao embedding request");
assert_eq!(doubao["input"], json!(["alpha", "beta"]));
assert!(doubao.get("messages").is_none());
}
@@ -4802,11 +4805,14 @@ mod tests {
assert_eq!(gemini["content"]["parts"][0]["text"], "alpha");
assert!(gemini.get("requests").is_none());
let doubao =
super::canonical_to_embedding_request(&single, "doubao-embedding-vision", "doubao")
.expect("doubao embedding request");
assert_eq!(doubao["model"], "doubao-embedding-vision");
assert_eq!(doubao["input"], json!([{"type": "text", "text": "alpha"}]));
let doubao = super::canonical_to_embedding_request(
&single,
"doubao-embedding-text-240515",
"doubao",
)
.expect("doubao embedding request");
assert_eq!(doubao["model"], "doubao-embedding-text-240515");
assert_eq!(doubao["input"], json!(["alpha"]));
assert_eq!(doubao["dimensions"], 1536);
}