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

Ignore items not owned by the user

Browse source
This commit is contained in:
vishnukvmd 2023-10-24 14:12:22 +05:30
parent 53fa9c6830
commit ce1d6c0805
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -56,7 +56,6 @@ class EmbeddingStore {
} }
Future<void> _pushEmbedding(EnteFile file, Embedding embedding) async { Future<void> _pushEmbedding(EnteFile file, Embedding embedding) async {
return;
final encryptionKey = getFileKey(file); final encryptionKey = getFileKey(file);
final embeddingData = final embeddingData =
Uint8List.view(Float64List.fromList(embedding.embedding).buffer); Uint8List.view(Float64List.fromList(embedding.embedding).buffer);
@ -69,7 +68,7 @@ class EmbeddingStore {
final header = CryptoUtil.bin2base64(encryptedEmbeddingData.header!); final header = CryptoUtil.bin2base64(encryptedEmbeddingData.header!);
try { try {
final response = await _dio.put( final response = await _dio.put(
"/embeddings/", "/embeddings",
data: { data: {
"fileID": embedding.fileID, "fileID": embedding.fileID,
"model": embedding.model, "model": embedding.model,

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

@ -7,6 +7,7 @@ import "package:computer/computer.dart";
import "package:flutter/services.dart"; import "package:flutter/services.dart";
import "package:logging/logging.dart"; import "package:logging/logging.dart";
import "package:path_provider/path_provider.dart"; import "package:path_provider/path_provider.dart";
import "package:photos/core/configuration.dart";
import "package:photos/core/event_bus.dart"; import "package:photos/core/event_bus.dart";
import "package:photos/db/files_db.dart"; import "package:photos/db/files_db.dart";
import "package:photos/events/file_indexed_event.dart"; import "package:photos/events/file_indexed_event.dart";
@ -201,6 +202,8 @@ class SemanticSearchService {
return; return;
} }
final files = await FilesDB.instance.getFilesWithoutEmbeddings(); final files = await FilesDB.instance.getFilesWithoutEmbeddings();
final ownerID = Configuration.instance.getUserID();
files.removeWhere((f) => f.ownerID != ownerID);
_logger.info(files.length.toString() + " pending to be embedded"); _logger.info(files.length.toString() + " pending to be embedded");
_queue.addAll(files); _queue.addAll(files);
_pollQueue(); _pollQueue();