0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Execute a single query to fetch files for fetched embeddings

Browse source
This commit is contained in:
vishnukvmd 2024-01-09 00:07:06 +05:30
parent 20c0d455df
commit 6876e0e85e
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
lib/services/semantic_search/embedding_store.dart
View file

@ -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);
}