managed for small batch size edge case in semantic search

KCaverly created 2 years ago

Change summary

crates/semantic_index/src/semantic_index.rs | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)

Detailed changes

crates/semantic_index/src/semantic_index.rs

@@ -721,7 +721,12 @@ impl SemanticIndex {
             )?;
 
             let batch_n = cx.background().num_cpus();
-            let batch_size = file_ids.clone().len() / batch_n;
+            let ids_len = file_ids.clone().len();
+            let batch_size = if ids_len <= batch_n {
+                ids_len
+            } else {
+                ids_len / batch_n
+            };
 
             let mut result_tasks = Vec::new();
             for batch in file_ids.chunks(batch_size) {