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

Push computed embeddings upstream

Browse source
This commit is contained in:
vishnukvmd 2023-10-03 23:54:09 +05:30
parent 57df32762a
commit 3411d250a7
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
lib/services/semantic_search/semantic_search_service.dart
View file

@ -213,15 +213,6 @@ class SemanticSearchService {
) as List<List<double>>;
imageEmbeddings.addAll(result);
}
for (int i = 0; i < imageEmbeddings.length; i++) {
await FilesDB.instance.insertEmbedding(
Embedding(
files[i].generatedID!,
"c_uq",
imageEmbeddings[i],
),
);
}
final endTime = DateTime.now();
_logger.info(
"createImageEmbeddings took: " +
@ -231,6 +222,16 @@ class SemanticSearchService {
imageEmbeddings.length.toString() +
" items",
);
for (int i = 0; i < imageEmbeddings.length; i++) {
EmbeddingStore.instance.storeEmbedding(
files[i],
Embedding(
files[i].uploadedFileID!,
"c_uq",
imageEmbeddings[i],
),
);
}
}
}