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
crates/mem-cli/src/main.rs
+2
View File
@@ -1,6 +1,8 @@
mod lessons_cmd; mod lessons_cmd;
mod http_server; mod http_server;
mod endpoints; mod endpoints;
mod ingest_worker;
mod query_worker;
use clap::{Parser, Subcommand}; use clap::{Parser, Subcommand};
use mem_chunk::token_counter::CharsOverFourCounter; use mem_chunk::token_counter::CharsOverFourCounter;
tests/it_rerank.rs
+20 -14
View File
@@ -7,11 +7,13 @@ async fn a1_bare_array_parsed() {
let mock_server = MockServer::start().await; let mock_server = MockServer::start().await;
Mock::given(method("POST")) Mock::given(method("POST"))
.and(path("/rerank")) .and(path("/v1/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(vec![ .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
serde_json::json!({"index": 0, "score": 0.98}), "results": [
serde_json::json!({"index": 1, "score": 0.01}), {"index": 0, "score": 0.98},
])) {"index": 1, "score": 0.01},
]
})))
.mount(&mock_server) .mount(&mock_server)
.await; .await;
@@ -32,11 +34,13 @@ async fn a2_index_mapping() {
// Return out-of-order: index 1 first, then index 0 // Return out-of-order: index 1 first, then index 0
Mock::given(method("POST")) Mock::given(method("POST"))
.and(path("/rerank")) .and(path("/v1/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(vec![ .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
serde_json::json!({"index": 1, "score": 0.99}), "results": [
serde_json::json!({"index": 0, "score": 0.01}), {"index": 1, "score": 0.99},
])) {"index": 0, "score": 0.01},
]
})))
.mount(&mock_server) .mount(&mock_server)
.await; .await;
@@ -68,10 +72,12 @@ async fn a4_apikey_sent() {
let mock_server = MockServer::start().await; let mock_server = MockServer::start().await;
Mock::given(method("POST")) Mock::given(method("POST"))
.and(path("/rerank")) .and(path("/v1/rerank"))
.respond_with(ResponseTemplate::new(200).set_body_json(vec![ .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
serde_json::json!({"index": 0, "score": 0.95}), "results": [
])) {"index": 0, "score": 0.95},
]
})))
.mount(&mock_server) .mount(&mock_server)
.await; .await;