Add missing functions from index to wrapped_async.rs

diskann-providers/src/index/wrapped_async.rs

@@ -8,12 +8,16 @@ use std::{num::NonZeroUsize, sync::Arc};
 use diskann::{
     ANNResult,
     graph::{
-        self, ConsolidateKind, InplaceDeleteMethod,
+        self, ConsolidateKind, InplaceDeleteMethod, RangeSearchParams, SearchParams,
         glue::{
-            self, AsElement, InplaceDeleteStrategy, InsertStrategy, PruneStrategy, SearchStrategy,
+            self, AsElement, IdIterator, InplaceDeleteStrategy, InsertStrategy, PruneStrategy,
+            SearchStrategy,
+        },
+        index::{
+            DegreeStats, PartitionedNeighbors, QueryLabelProvider, SearchState, SearchStats,
         },
-        index::{DegreeStats, PartitionedNeighbors, SearchState, SearchStats},
         search::Knn,
+        search::record::SearchRecord,
         search_output_buffer,
     },
     neighbor::Neighbor,
@@ -299,6 +303,32 @@ where
         ))
     }
 
+    #[allow(clippy::too_many_arguments)]
+    pub fn multihop_search<S, T, O, OB>(
+        &self,
+        strategy: &S,
+        context: &DP::Context,
+        query: &T,
+        search_params: &SearchParams,
+        output: &mut OB,
+        query_label_evaluator: &dyn QueryLabelProvider<DP::InternalId>,
+    ) -> ANNResult<SearchStats>
+    where
+        T: Sync + ?Sized,
+        S: SearchStrategy<DP, T, O>,
+        O: Send,
+        OB: search_output_buffer::SearchOutputBuffer<O> + Send,
+    {
+        self.handle.block_on(self.inner.multihop_search(
+            strategy,
+            context,
+            query,
+            search_params,
+            output,
+            query_label_evaluator,
+        ))
+    }
+
     pub fn count_reachable_nodes<NA>(
         &self,
         start_points: &[DP::InternalId],