From ea6f9c8e18351bfd09612f371baf5031a09ecdbb Mon Sep 17 00:00:00 2001
From: perf3ct <jonfuller2012@gmail.com>
Date: Sat, 8 Mar 2025 22:04:31 +0000
Subject: [PATCH] initialize embeddings if option is enabled

---
 src/services/llm/embeddings/init.ts | 26 ++++++++++++++++++++++++++
 src/services/options_interface.ts   | 10 ++++++++--
 2 files changed, 34 insertions(+), 2 deletions(-)
 create mode 100644 src/services/llm/embeddings/init.ts

diff --git a/src/services/llm/embeddings/init.ts b/src/services/llm/embeddings/init.ts
new file mode 100644
index 000000000..edcea6774
--- /dev/null
+++ b/src/services/llm/embeddings/init.ts
@@ -0,0 +1,26 @@
+import log from "../../log.js";
+import options from "../../options.js";
+import vectorStore from "./vector_store.js";
+import providerManager from "./providers.js";
+
+/**
+ * Initialize the embedding system
+ */
+export async function initializeEmbeddings() {
+    try {
+        log.info("Initializing embedding system...");
+
+        // Initialize default embedding providers
+        await providerManager.initializeDefaultProviders();
+
+        // Start the embedding system if AI is enabled
+        if (await options.getOptionBool('aiEnabled')) {
+            await vectorStore.initEmbeddings();
+            log.info("Embedding system initialized successfully.");
+        } else {
+            log.info("Embedding system disabled (AI features are turned off).");
+        }
+    } catch (error: any) {
+        log.error(`Error initializing embedding system: ${error.message || error}`);
+    }
+}
diff --git a/src/services/options_interface.ts b/src/services/options_interface.ts
index e985062a7..b920764b0 100644
--- a/src/services/options_interface.ts
+++ b/src/services/options_interface.ts
@@ -47,20 +47,26 @@ export interface OptionDefinitions extends KeyboardShortcutsOptions<KeyboardActi
     encryptedDataKey: string;
 
     // AI/LLM integration options
-    aiEnabled: string;
+    aiEnabled: boolean;
     openaiApiKey: string;
     openaiDefaultModel: string;
     openaiBaseUrl: string;
     anthropicApiKey: string;
     anthropicDefaultModel: string;
     anthropicBaseUrl: string;
-    ollamaEnabled: string;
+    ollamaEnabled: boolean;
     ollamaBaseUrl: string;
     ollamaDefaultModel: string;
     aiProviderPrecedence: string;
     aiTemperature: string;
     aiSystemPrompt: string;
 
+    // Embedding-related options
+    embeddingAutoUpdateEnabled: boolean;
+    embeddingUpdateInterval: number;
+    embeddingBatchSize: number;
+    embeddingDefaultDimension: number;
+
     lastSyncedPull: number;
     lastSyncedPush: number;
     revisionSnapshotTimeInterval: number;