fix: failing test

0xPlaygrounds · Dec 2, 2024 · a29e37e · a29e37e
1 parent f71fd5d
commit a29e37e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/rig-mongodb/tests/integration_tests.rs b/rig-mongodb/tests/integration_tests.rs
@@ -145,11 +145,14 @@ async fn vector_search_test() {
         .await
         .unwrap();
 
-    if results.is_empty() {
+    let mut i = 1;
+
+    while results.is_empty() && i < 5 {
         results = index
             .top_n::<serde_json::Value>("What is a linglingdong?", 1)
             .await
-            .expect("Failed to query vector index");
+            .unwrap();
+        i += 1;
     }
 
     let (score, _, value) = &results.first().unwrap();