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

feat(embedding): 接入阿里云多模态向量端点

Browse Source
This commit is contained in:
Entropy.Xu
2026-05-28 16:05:36 +08:00
parent 535039c29e
commit 0f6d4b9146
56 changed files with 1605 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/aether-ai-formats/src/formats/matrix.rs
+27 -4
View File
@@ -42,6 +42,7 @@ const EMBEDDING_CANDIDATE_API_FORMATS: &[&str] = &[
"gemini:embedding",
"jina:embedding",
"doubao:embedding",
"aliyun:multimodal_embedding",
];
const RERANK_CANDIDATE_API_FORMATS: &[&str] = &["openai:rerank", "jina:rerank"];
@@ -238,7 +239,11 @@ pub fn is_standard_api_format(api_format: &str) -> bool {
pub fn is_embedding_api_format(api_format: &str) -> bool {
matches!(
normalize_api_format_alias(api_format).as_str(),
"openai:embedding" | "gemini:embedding" | "jina:embedding" | "doubao:embedding"
"openai:embedding"
| "gemini:embedding"
| "jina:embedding"
| "doubao:embedding"
| "aliyun:multimodal_embedding"
)
}
@@ -267,9 +272,11 @@ pub fn api_data_format_id(api_format: &str) -> Option<&'static str> {
"gemini:generate_content" => Some("gemini"),
"openai:chat" => Some("openai_chat"),
"openai:responses" | "openai:responses:compact" => Some("openai_responses"),
"openai:embedding" | "gemini:embedding" | "jina:embedding" | "doubao:embedding" => {
Some("embedding")
}
"openai:embedding"
| "gemini:embedding"
| "jina:embedding"
| "doubao:embedding"
| "aliyun:multimodal_embedding" => Some("embedding"),
"openai:rerank" | "jina:rerank" => Some("rerank"),
_ => None,
}
@@ -442,6 +449,7 @@ mod tests {
"gemini:embedding",
"jina:embedding",
"doubao:embedding",
"aliyun:multimodal_embedding",
]
);
assert_eq!(
@@ -451,6 +459,7 @@ mod tests {
"openai:embedding",
"gemini:embedding",
"doubao:embedding",
"aliyun:multimodal_embedding",
]
);
assert!(!request_candidate_api_formats("openai:embedding", false).contains(&"openai:chat"));
@@ -479,6 +488,7 @@ mod tests {
"openai:embedding",
"jina:embedding",
"doubao:embedding",
"aliyun:multimodal_embedding",
]
);
assert_eq!(
@@ -488,6 +498,17 @@ mod tests {
"openai:embedding",
"gemini:embedding",
"jina:embedding",
"aliyun:multimodal_embedding",
]
);
assert_eq!(
request_candidate_api_formats("aliyun:multimodal_embedding", false),
vec![
"aliyun:multimodal_embedding",
"openai:embedding",
"gemini:embedding",
"jina:embedding",
"doubao:embedding",
]
);
@@ -496,6 +517,7 @@ mod tests {
"gemini:embedding",
"jina:embedding",
"doubao:embedding",
"aliyun:multimodal_embedding",
];
for client_api_format in embedding_formats {
for provider_api_format in embedding_formats {
@@ -520,6 +542,7 @@ mod tests {
"gemini:embedding",
"jina:embedding",
"doubao:embedding",
"aliyun:multimodal_embedding",
];
let standard_formats = [
"openai:chat",