diff --git a/rust/lancedb/src/connection.rs b/rust/lancedb/src/connection.rs index 8946beee6..83c850652 100644 --- a/rust/lancedb/src/connection.rs +++ b/rust/lancedb/src/connection.rs @@ -1169,6 +1169,7 @@ mod tests { use lance_testing::datagen::{BatchGenerator, IncrementingInt32}; use tempfile::tempdir; + use crate::query::QueryBase; use crate::query::{ExecutableQuery, QueryExecutionOptions}; use super::*; @@ -1296,6 +1297,7 @@ mod tests { // In v1 the row group size will trump max_batch_length let batches = tbl .query() + .limit(20000) .execute_with_options(QueryExecutionOptions { max_batch_length: 50000, ..Default::default() diff --git a/rust/lancedb/src/query.rs b/rust/lancedb/src/query.rs index 54c344e19..4e3059816 100644 --- a/rust/lancedb/src/query.rs +++ b/rust/lancedb/src/query.rs @@ -596,7 +596,7 @@ impl Query { pub(crate) fn new(parent: Arc) -> Self { Self { parent, - limit: None, + limit: Some(DEFAULT_TOP_K), offset: None, filter: None, full_text_search: None, diff --git a/rust/lancedb/src/remote/table.rs b/rust/lancedb/src/remote/table.rs index 11dbee1b3..c8ce2ec4a 100644 --- a/rust/lancedb/src/remote/table.rs +++ b/rust/lancedb/src/remote/table.rs @@ -1227,6 +1227,7 @@ mod tests { "prefilter": true, "distance_type": "l2", "nprobes": 20, + "k": 10, "ef": Option::::None, "refine_factor": null, "version": null,