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

fix: repair embedding rerank CI checks

Browse Source
This commit is contained in:
Kayphoon
2026-05-04 11:33:00 +08:00
parent 5abe664d65
commit 6bdd7792eb
2 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4337,12 +4337,7 @@ fn rerank_request_from_raw(body_json: &Value, namespace: &str) -> Option<Canonic
.map(str::trim) .map(str::trim)
.filter(|value| !value.is_empty())? .filter(|value| !value.is_empty())?
.to_string(); .to_string();
let documents = request let documents = request.get("documents").and_then(Value::as_array)?.to_vec();
.get("documents")
.and_then(Value::as_array)?
.iter()
.cloned()
.collect::<Vec<_>>();
let rerank = CanonicalRerankRequest { let rerank = CanonicalRerankRequest {
query, query,
documents, documents,

6
crates/aether-ai-formats/src/request/route.rs
View File

@@ -676,12 +676,13 @@ mod tests {
Some("ai_public"), Some("ai_public"),
Some("openai"), Some("openai"),
Some("embedding"), Some("embedding"),
None,
&Method::POST, &Method::POST,
"/v1/embeddings", "/v1/embeddings",
), ),
Some(OPENAI_EMBEDDING_SYNC_PLAN_KIND) Some(OPENAI_EMBEDDING_SYNC_PLAN_KIND)
); );
assert!(supports_sync_scheduler_decision_kind( assert!(supports_sync_execution_decision_kind(
OPENAI_EMBEDDING_SYNC_PLAN_KIND OPENAI_EMBEDDING_SYNC_PLAN_KIND
)); ));
} }
@@ -693,12 +694,13 @@ mod tests {
Some("ai_public"), Some("ai_public"),
Some("openai"), Some("openai"),
Some("rerank"), Some("rerank"),
None,
&Method::POST, &Method::POST,
"/v1/rerank", "/v1/rerank",
), ),
Some(OPENAI_RERANK_SYNC_PLAN_KIND) Some(OPENAI_RERANK_SYNC_PLAN_KIND)
); );
assert!(supports_sync_scheduler_decision_kind( assert!(supports_sync_execution_decision_kind(
OPENAI_RERANK_SYNC_PLAN_KIND OPENAI_RERANK_SYNC_PLAN_KIND
)); ));
} }