feat: hide LLM embedding behind feature flag

matthiasn · May 3, 2024 · d3943d2 · d3943d2
1 parent d6df25d
commit d3943d2
Show file tree

Hide file tree

Showing 6 changed files with 25 additions and 2 deletions.
diff --git a/lib/database/journal_db/config_flags.dart b/lib/database/journal_db/config_flags.dart
@@ -14,7 +14,13 @@ Future<void> initConfigFlags(
       status: true,
     ),
   );
-
+  await db.insertFlagIfNotExists(
+    const ConfigFlag(
+      name: attemptEmbedding,
+      description: 'Create LLM embedding',
+      status: false,
+    ),
+  );
   await db.insertFlagIfNotExists(
     const ConfigFlag(
       name: allowInvalidCertFlag,

diff --git a/lib/logic/ai/ai_logic.dart b/lib/logic/ai/ai_logic.dart
@@ -1,6 +1,9 @@
 import 'package:flutter/foundation.dart';
 import 'package:langchain/langchain.dart';
 import 'package:lotti/classes/journal_entities.dart';
+import 'package:lotti/database/database.dart';
+import 'package:lotti/get_it.dart';
+import 'package:lotti/utils/consts.dart';
 import 'package:lotti/utils/platform.dart';
 import 'package:ollama_dart/ollama_dart.dart';
 
@@ -13,6 +16,13 @@ class AiLogic {
     JournalEntity? journalEntity, {
     String? linkedFromId,
   }) async {
+    final shouldAttemptEmbedding = await getIt<JournalDb>().getConfigFlag(
+      attemptEmbedding,
+    );
+    if (!shouldAttemptEmbedding) {
+      return;
+    }
+
     final markdown = journalEntity?.entryText?.markdown;
     final headline = switch (journalEntity) {
       Task() => journalEntity.data.title,

diff --git a/lib/pages/settings/flags_page.dart b/lib/pages/settings/flags_page.dart
@@ -22,6 +22,7 @@ class FlagsPage extends StatelessWidget {
 
         const displayedItems = {
           privateFlag,
+          attemptEmbedding,
           enableNotificationsFlag,
           autoTranscribeFlag,
           recordLocationFlag,

diff --git a/lib/utils/consts.dart b/lib/utils/consts.dart
@@ -6,3 +6,4 @@ const recordLocationFlag = 'record_location';
 const autoTranscribeFlag = 'auto_transcribe';
 const enableMatrixFlag = 'enable_matrix';
 const resendAttachments = 'resend_attachments';
+const attemptEmbedding = 'attempt_embedding';
diff --git a/pubspec.yaml b/pubspec.yaml
@@ -1,7 +1,7 @@
 name: lotti
 description: Achieve your goals and keep your data private with Lotti.
 publish_to: 'none'
-version: 0.9.458+2493
+version: 0.9.459+2494
 
 msix_config:
   display_name: LottiApp

diff --git a/test/database/database_test.dart b/test/database/database_test.dart
@@ -16,6 +16,11 @@ final expectedFlags = <ConfigFlag>{
     description: 'Show private entries?',
     status: true,
   ),
+  const ConfigFlag(
+    name: attemptEmbedding,
+    description: 'Create LLM embedding',
+    status: false,
+  ),
   const ConfigFlag(
     name: autoTranscribeFlag,
     description: 'Automatically transcribe audio',