From 234bce70a0b9a8d7d711964f6b8533674a30766c Mon Sep 17 00:00:00 2001
From: Story Crater Bot <19826264+Riotpiaole@users.noreply.github.com>
Date: Sun, 23 Aug 2026 18:45:01 -0700
Subject: [PATCH] fix: resolve module imports and rerank test format

- Add ingest_worker and query_worker modules to main.rs
- Update rerank tests to use /v1/rerank endpoint with new response format
- All 253+ tests now passing
---
 crates/mem-cli/src/main.rs |  2 ++
 tests/it_rerank.rs         | 34 ++++++++++++++++++++--------------
 2 files changed, 22 insertions(+), 14 deletions(-)

diff --git a/crates/mem-cli/src/main.rs b/crates/mem-cli/src/main.rs
index 291026d..39b2d4d 100644
--- a/crates/mem-cli/src/main.rs
+++ b/crates/mem-cli/src/main.rs
@@ -1,6 +1,8 @@
 mod lessons_cmd;
 mod http_server;
 mod endpoints;
+mod ingest_worker;
+mod query_worker;
 
 use clap::{Parser, Subcommand};
 use mem_chunk::token_counter::CharsOverFourCounter;
diff --git a/tests/it_rerank.rs b/tests/it_rerank.rs
index 279f52f..7b03460 100644
--- a/tests/it_rerank.rs
+++ b/tests/it_rerank.rs
@@ -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;