From 6876e0e85e8564636b5043378719f2be16dad777 Mon Sep 17 00:00:00 2001
From: vishnukvmd <vishnu@ente.io>
Date: Tue, 9 Jan 2024 00:07:06 +0530
Subject: [PATCH] Execute a single query to fetch files for fetched embeddings

---
 lib/services/semantic_search/embedding_store.dart | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/lib/services/semantic_search/embedding_store.dart b/lib/services/semantic_search/embedding_store.dart
index de10fec60..6aedf8547 100644
--- a/lib/services/semantic_search/embedding_store.dart
+++ b/lib/services/semantic_search/embedding_store.dart
@@ -133,12 +133,12 @@ class EmbeddingStore {
       return;
     }
     final inputs = <EmbeddingsDecoderInput>[];
+    final fileMap = await FilesDB.instance
+        .getFilesFromIDs(remoteEmbeddings.map((e) => e.fileID).toList());
+
     for (final embedding in remoteEmbeddings) {
-      final file = await FilesDB.instance.getAnyUploadedFile(embedding.fileID);
-      if (file == null) {
-        continue;
-      }
-      final fileKey = getFileKey(file);
+      final file = fileMap[embedding.fileID];
+      final fileKey = getFileKey(file!);
       final input = EmbeddingsDecoderInput(embedding, fileKey);
       inputs.add(input);
     }