rock/poimen-memory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: resolve module imports and rerank test format (#12)
Build and Push / Test (push) Successful in 3m35s
Details
Build and Push / Build and push image (push) Successful in 2m39s
Details

Browse Source
This commit was merged in pull request #12.
This commit is contained in:
rock
2026-08-24 01:45:47 +00:00
parent e6e39cf6fd
commit b10c0b9c53
2 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/it_rerank.rs
+20 -14
View File
@@ -7,11 +7,13 @@ async fn a1_bare_array_parsed() {
let mock_server = MockServer::start().await;
Mock::given(method("POST"))
.and(path("/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(vec![
serde_json::json!({"index": 0, "score": 0.98}),
serde_json::json!({"index": 1, "score": 0.01}),
]))
.and(path("/v1/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
"results": [
{"index": 0, "score": 0.98},
{"index": 1, "score": 0.01},
]
})))
.mount(&mock_server)
.await;
@@ -32,11 +34,13 @@ async fn a2_index_mapping() {
// Return out-of-order: index 1 first, then index 0
Mock::given(method("POST"))
.and(path("/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(vec![
serde_json::json!({"index": 1, "score": 0.99}),
serde_json::json!({"index": 0, "score": 0.01}),
]))
.and(path("/v1/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
"results": [
{"index": 1, "score": 0.99},
{"index": 0, "score": 0.01},
]
})))
.mount(&mock_server)
.await;
@@ -68,10 +72,12 @@ async fn a4_apikey_sent() {
let mock_server = MockServer::start().await;
Mock::given(method("POST"))
.and(path("/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(vec![
serde_json::json!({"index": 0, "score": 0.95}),
]))
.and(path("/v1/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
"results": [
{"index": 0, "score": 0.95},
]
})))
.mount(&mock_server)
.await;