From 883e246f3085e821f408dd71dffbaffc4a5757c6 Mon Sep 17 00:00:00 2001
From: Neeraj Gupta <254676+ua741@users.noreply.github.com>
Date: Mon, 13 May 2024 15:50:27 +0530
Subject: [PATCH] [mob] Keep fetching embeddings till no embedding is returned

---
 .../machine_learning/semantic_search/embedding_store.dart    | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mobile/lib/services/machine_learning/semantic_search/embedding_store.dart b/mobile/lib/services/machine_learning/semantic_search/embedding_store.dart
index 420b8c97f7..485e1f2c91 100644
--- a/mobile/lib/services/machine_learning/semantic_search/embedding_store.dart
+++ b/mobile/lib/services/machine_learning/semantic_search/embedding_store.dart
@@ -145,9 +145,12 @@ class EmbeddingStore {
     }
 
     _logger.info("${remoteEmbeddings.length} embeddings fetched");
+
     return RemoteEmbeddings(
       remoteEmbeddings,
-      remoteEmbeddings.length == limit,
+      // keep fetching until we get all embeddings. Avoid limit check as
+      // some embedding fetch might fail on server
+      remoteEmbeddings.isNotEmpty,
     );
   }