[mob] Log when clip embeddings are read from DB

This commit is contained in:
Neeraj Gupta
2024-10-21 15:10:34 +05:30
parent 94d9aeb67a
commit df863d1602
3 changed files with 9 additions and 1 deletions

View File

@@ -1,6 +1,7 @@
import "dart:io";
import "dart:typed_data";
import "package:logging/logging.dart";
import "package:photos/core/event_bus.dart";
import "package:photos/db/ml/db.dart";
import "package:photos/db/ml/db_fields.dart";
@@ -19,6 +20,7 @@ extension ClipDB on MLDataDB {
}
Future<List<EmbeddingVector>> getAllClipVectors() async {
Logger("ClipDB").info("reading all embeddings from DB");
final db = await MLDataDB.instance.asyncDB;
final results = await db.getAll('SELECT * FROM $clipTable');
return _convertToVectors(results);

View File

@@ -30,6 +30,8 @@ class MLDataDB {
static const _databaseName = "ente.ml.db";
static Logger get logger => _logger;
// static const _databaseVersion = 1;
MLDataDB._privateConstructor();

View File

@@ -108,8 +108,12 @@ class SemanticSearchService {
}
Future<List<EmbeddingVector>> getClipVectors() async {
_logger.info("Pulling cached clip embeddings");
if (_cachedImageEmbeddingVectors != null) {
return _cachedImageEmbeddingVectors!;
}
_cachedImageEmbeddingVectors ??= MLDataDB.instance.getAllClipVectors();
_logger.info("read all embeddings from DB");
return _cachedImageEmbeddingVectors!;
}