working code

Author: thefirehacker

lib/Pages/Playground/playground.js

@@ -183,6 +183,12 @@ class PlaygroundApp {
         console.log('🔄 Playground received shared connection event:', event, data);
 
         if (event === 'connected') {
+          // Check if we already have the same connection to avoid duplicates
+          if (this.aiAssistant === data && this.isConnected) {
+            console.log('⏸️ Same connection already active, skipping duplicate setup');
+            return;
+          }
+          
           this.aiAssistant = data;
           this.aiAssistant.onStatusChange = (status) => this.handleAIStatusChange(status);
 
@@ -344,6 +350,12 @@ class PlaygroundApp {
     console.log('🔄 AI Status Change:', status);
 
     if (status.connected) {
+      // Check if we're already connected to avoid duplicate processing
+      if (this.isConnected && status.provider === this.getCurrentProvider()) {
+        console.log('⏸️ Already connected to same provider, skipping duplicate status change');
+        return;
+      }
+      
       this.isConnected = true;
       this.updateStatus('aiStatus', `Connected (${status.provider})`, 'success');
       this.updateConnectButtonText();

lib/Pages/shared/ai-connection-manager.js

@@ -311,6 +311,12 @@ class SharedAIConnectionManager {
   async restoreAIAssistant() {
     const now = Date.now();
 
+    // Early exit if we already have a working connection
+    if (this.isConnected && this.aiAssistant && this.aiAssistant.isConnected) {
+      console.log('✅ AI connection already active, skipping restoration');
+      return;
+    }
+    
     // Check if restoration is already in progress
     if (this.isRestoring) {
       console.log('⏸️ Restoration already in progress, skipping...');
@@ -345,25 +351,25 @@ class SharedAIConnectionManager {
       // Attempt to restore connection based on provider
       let restored = false;
 
-             if (this.currentProvider === 'ollama') {
-         const baseURL = this.connectionSettings.baseURL || 'http://localhost:11434';
-         await this.connectOllama({ baseURL });
-         restored = this.aiAssistant && this.aiAssistant.isConnected;
-       } else if (this.currentProvider === 'lmstudio') {
-         const baseURL = this.connectionSettings.baseURL || 'http://localhost:1234';
-         await this.connectLMStudio({ baseURL });
-         restored = this.aiAssistant && this.aiAssistant.isConnected;
-       } else if (this.currentProvider === 'openai') {
-         if (this.connectionSettings.apiKey) {
-           await this.connectOpenAI({ apiKey: this.connectionSettings.apiKey });
-           restored = this.aiAssistant && this.aiAssistant.isConnected;
-         } else {
-           console.log('❌ Cannot restore OpenAI - no API key saved');
-         }
-       } else if (this.currentProvider === 'local') {
-         await this.connectLocal();
-         restored = this.aiAssistant && this.aiAssistant.isConnected;
-       }
+      if (this.currentProvider === 'ollama') {
+        const baseURL = this.connectionSettings.baseURL || 'http://localhost:11434';
+        await this.connectOllama({ baseURL });
+        restored = this.aiAssistant && this.aiAssistant.isConnected;
+      } else if (this.currentProvider === 'lmstudio') {
+        const baseURL = this.connectionSettings.baseURL || 'http://localhost:1234';
+        await this.connectLMStudio({ baseURL });
+        restored = this.aiAssistant && this.aiAssistant.isConnected;
+      } else if (this.currentProvider === 'openai') {
+        if (this.connectionSettings.apiKey) {
+          await this.connectOpenAI({ apiKey: this.connectionSettings.apiKey });
+          restored = this.aiAssistant && this.aiAssistant.isConnected;
+        } else {
+          console.log('❌ Cannot restore OpenAI - no API key saved');
+        }
+      } else if (this.currentProvider === 'local') {
+        await this.connectLocal();
+        restored = this.aiAssistant && this.aiAssistant.isConnected;
+      }
 
       if (restored) {
         console.log('✅ AI connection fully restored from saved state');
@@ -414,6 +420,13 @@ class SharedAIConnectionManager {
 
     this.lastRefreshTime = now;
 
+    // If we already have a working connection, just notify listeners
+    if (this.isConnected && this.aiAssistant && this.aiAssistant.isConnected) {
+      console.log('✅ Active connection found, notifying listeners only');
+      this.notifyListeners('connected', this.aiAssistant);
+      return;
+    }
+    
     // Reload connection state from localStorage
     this.loadConnectionState();
 
@@ -434,11 +447,17 @@ class SharedAIConnectionManager {
   async checkAndRestoreConnection() {
     console.log('🔍 Checking for saved AI connection...');
 
+    // If we already have a working connection, don't restore
+    if (this.isConnected && this.aiAssistant && this.aiAssistant.isConnected) {
+      console.log('✅ AI connection already active, skipping restoration');
+      return;
+    }
+    
     if (this.isConnected && this.currentProvider) {
       console.log(`📂 Found saved connection: ${this.currentProvider}`);
 
-      if (!this.aiAssistant) {
-        console.log('🔄 No AI assistant instance, attempting to restore...');
+      if (!this.aiAssistant || !this.aiAssistant.isConnected) {
+        console.log('🔄 No active AI assistant instance, attempting to restore...');
         await this.restoreAIAssistant();
       } else {
         console.log('✅ AI assistant already available, notifying listeners');