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;