Merge pull request #2 from hangwin/feat/optional_use_model

change embedding model initial time

Author: hangwin

app/chrome-extension/common/message-types.ts

@@ -20,6 +20,7 @@ export const BACKGROUND_MESSAGE_TYPES = {
   GET_SERVER_STATUS: 'get_server_status',
   REFRESH_SERVER_STATUS: 'refresh_server_status',
   SERVER_STATUS_CHANGED: 'server_status_changed',
+  INITIALIZE_SEMANTIC_ENGINE: 'initialize_semantic_engine',
 } as const;
 
 // Offscreen message types

app/chrome-extension/entrypoints/background/index.ts

@@ -1,8 +1,5 @@
 import { initNativeHostListener } from './native-host';
-import {
-  initSemanticSimilarityListener,
-  initializeDefaultSemanticEngine,
-} from './semantic-similarity';
+import { initSemanticSimilarityListener } from './semantic-similarity';
 import { initStorageManagerListener } from './storage-manager';
 import { vectorSearchTabsContentTool } from './tools/browser/vector-search';
 import { ERROR_MESSAGES } from '@/common/constants';
@@ -25,6 +22,6 @@ export default defineBackground(() => {
     );
   });
 
-  // Initialize semantic similarity engine
-  initializeDefaultSemanticEngine();
+  // Note: Semantic similarity engine initialization is now user-controlled via popup
+  console.log('Background: Semantic similarity engine initialization is now user-controlled');
 });

app/chrome-extension/entrypoints/background/semantic-similarity.ts

@@ -19,6 +19,11 @@ let currentBackgroundModelConfig: ModelConfig | null = null;
  */
 export async function initializeDefaultSemanticEngine(): Promise<void> {
   try {
+    console.log('Background: Initializing default semantic engine...');
+
+    // Update status to initializing
+    await updateModelStatus('initializing', 0);
+
     const result = await chrome.storage.local.get([STORAGE_KEYS.SEMANTIC_MODEL, 'selectedVersion']);
     const defaultModel =
       (result[STORAGE_KEYS.SEMANTIC_MODEL] as ModelPreset) || 'multilingual-e5-small';
@@ -49,11 +54,31 @@ export async function initializeDefaultSemanticEngine(): Promise<void> {
         modelDimension: modelInfo.dimension,
       };
       console.log('Semantic engine initialized successfully:', currentBackgroundModelConfig);
+
+      // Update status to ready
+      await updateModelStatus('ready', 100);
+
+      // Also initialize ContentIndexer now that semantic engine is ready
+      try {
+        const { getGlobalContentIndexer } = await import('@/utils/content-indexer');
+        const contentIndexer = getGlobalContentIndexer();
+        contentIndexer.startSemanticEngineInitialization();
+        console.log('ContentIndexer initialization triggered after semantic engine initialization');
+      } catch (indexerError) {
+        console.warn(
+          'Failed to initialize ContentIndexer after semantic engine initialization:',
+          indexerError,
+        );
+      }
     } else {
-      throw new Error(response?.error || ERROR_MESSAGES.TOOL_EXECUTION_FAILED);
+      const errorMessage = response?.error || ERROR_MESSAGES.TOOL_EXECUTION_FAILED;
+      await updateModelStatus('error', 0, errorMessage, 'unknown');
+      throw new Error(errorMessage);
     }
-  } catch (error) {
+  } catch (error: any) {
     console.error('Background: Failed to initialize default semantic engine:', error);
+    const errorMessage = error?.message || 'Unknown error during semantic engine initialization';
+    await updateModelStatus('error', 0, errorMessage, 'unknown');
     // Don't throw error, let the extension continue running
   }
 }
@@ -314,6 +339,11 @@ export const initSemanticSimilarityListener = () => {
         .then((result: { success: boolean; error?: string }) => sendResponse(result))
         .catch((error: any) => sendResponse({ success: false, error: error.message }));
       return true;
+    } else if (message.type === BACKGROUND_MESSAGE_TYPES.INITIALIZE_SEMANTIC_ENGINE) {
+      initializeDefaultSemanticEngine()
+        .then(() => sendResponse({ success: true }))
+        .catch((error: any) => sendResponse({ success: false, error: error.message }));
+      return true;
     }
   });
 };

app/chrome-extension/entrypoints/background/storage-manager.ts

@@ -13,10 +13,8 @@ export async function handleGetStorageStats(): Promise<{
     const { getGlobalContentIndexer } = await import('@/utils/content-indexer');
     const contentIndexer = getGlobalContentIndexer();
 
-    // Ensure ContentIndexer is initialized
-    if (!contentIndexer.isSemanticEngineReady() && !contentIndexer.isSemanticEngineInitializing()) {
-      contentIndexer.startSemanticEngineInitialization();
-    }
+    // Note: Semantic engine initialization is now user-controlled
+    // ContentIndexer will be initialized when user manually triggers semantic engine initialization
 
     // Get statistics
     const stats = contentIndexer.getStats();

app/chrome-extension/entrypoints/background/tools/browser/vector-search.ts

@@ -41,10 +41,13 @@ class VectorSearchTabsContentTool extends BaseBrowserToolExecutor {
 
   /**
    * Start background initialization (non-blocking)
+   * Note: Semantic engine initialization is now user-controlled
    */
   private startBackgroundInitialization(): void {
-    console.log('VectorSearchTabsContentTool: Starting background initialization...');
-    this.contentIndexer.startSemanticEngineInitialization();
+    console.log(
+      'VectorSearchTabsContentTool: Background initialization skipped - semantic engine initialization is now user-controlled',
+    );
+    // this.contentIndexer.startSemanticEngineInitialization(); // Removed automatic initialization
   }
 
   private async initializeIndexer(): Promise<void> {
@@ -261,7 +264,16 @@ class VectorSearchTabsContentTool extends BaseBrowserToolExecutor {
    */
   public async getIndexStats() {
     if (!this.isInitialized) {
-      await this.initializeIndexer();
+      // Don't automatically initialize - just return basic stats
+      return {
+        totalDocuments: 0,
+        totalTabs: 0,
+        indexSize: 0,
+        indexedPages: 0,
+        isInitialized: false,
+        semanticEngineReady: false,
+        semanticEngineInitializing: false,
+      };
     }
     return this.contentIndexer.getStats();
   }