agentgui 1.0.829 → 1.0.830

package/static/js/client-ws.js

@@ -0,0 +1,161 @@
+Object.assign(AgentGUIClient.prototype, {
+  setupWebSocketListeners() {
+    this.wsManager.on('connected', () => {
+      this._dbg('WebSocket connected');
+      this.updateConnectionStatus('connected');
+      this._subscribeToConversationUpdates();
+      if (this.wsManager.stats.totalReconnects > 0 && this.state.currentConversation?.id) {
+        this.invalidateCache(this.state.currentConversation.id);
+      }
+      this._recoverMissedChunks();
+      this.updateSendButtonState();
+      this.enablePromptArea();
+      if (this.state.currentConversation?.id) {
+        this.updateBusyPromptArea(this.state.currentConversation.id);
+      }
+      this.emit('ws:connected');
+      if (window.__SERVER_VERSION) {
+        fetch((window.__BASE_URL || '') + '/api/version').then(r => r.json()).then(d => {
+          if (d.version && d.version !== window.__SERVER_VERSION) {
+            this._dbg(`Server updated ${window.__SERVER_VERSION} → ${d.version}, reloading`);
+            window.location.reload();
+          }
+        }).catch(() => {});
+      }
+    });
+
+    this.wsManager.on('disconnected', () => {
+      this._dbg('WebSocket disconnected');
+      this.updateConnectionStatus('disconnected');
+      this.updateSendButtonState();
+      this.disablePromptArea();
+      this.emit('ws:disconnected');
+    });
+
+    this.wsManager.on('reconnecting', (data) => {
+      this._dbg('WebSocket reconnecting:', data);
+      this.updateConnectionStatus('reconnecting');
+    });
+
+    this.wsManager.on('message', (data) => {
+      this.handleWebSocketMessage(data);
+    });
+
+    this.wsManager.on('error', (data) => {
+      console.error('WebSocket error:', data);
+    });
+
+    this.wsManager.on('latency_update', (data) => {
+      this._updateConnectionIndicator(data.quality);
+    });
+
+    this.wsManager.on('connection_degrading', () => {
+      const dot = document.querySelector('.connection-dot');
+      if (dot) dot.classList.add('degrading');
+    });
+
+    this.wsManager.on('connection_recovering', () => {
+      const dot = document.querySelector('.connection-dot');
+      if (dot) dot.classList.remove('degrading');
+    });
+
+    window.addEventListener('conversation-selected', (e) => {
+      const convId = e.detail.conversationId;
+      this.saveDraftPrompt();
+
+      const isStreaming = this._convIsStreaming(convId);
+      if (!isStreaming && window.switchView) {
+        window.switchView('chat');
+      }
+
+      this.restoreDraftPrompt(convId);
+    });
+
+    window.addEventListener('view-switched', (e) => {
+      const view = e.detail.view;
+      if (view === 'chat') {
+        const convId = this.state.currentConversation?.id;
+        const isStreaming = this._convIsStreaming(convId);
+        if (isStreaming) {
+          this.disableControls();
+        } else {
+          this.enableControls();
+        }
+      }
+    });
+  },
+
+
+  _convIsStreaming(convId) {
+    if (!convId) return false;
+    if (typeof convMachineAPI !== 'undefined') return convMachineAPI.isStreaming(convId);
+    return this.state.streamingConversations.has(convId);
+  },
+
+
+  _setConvStreaming(convId, streaming, sessionId, agentId) {
+    if (!convId) return;
+    if (streaming) {
+      this.state.streamingConversations.set(convId, true);
+      if (typeof convMachineAPI !== 'undefined') convMachineAPI.send(convId, { type: 'STREAM_START', sessionId, agentId });
+    } else {
+      this.state.streamingConversations.delete(convId);
+      if (typeof convMachineAPI !== 'undefined') convMachineAPI.send(convId, { type: 'COMPLETE' });
+    }
+  },
+
+
+  setupRendererListeners() {
+    this.renderer.on('batch:complete', (data) => {
+      this._dbg('Batch rendered:', data);
+      this.updateMetrics(data.metrics);
+    });
+
+    this.renderer.on('error:render', (data) => {
+      console.error('Render error:', data.error);
+    });
+  },
+
+
+  restoreStateFromUrl() {
+    const pathMatch = window.location.pathname.match(/\/conversations\/([^\/]+)$/);
+    const conversationId = pathMatch ? pathMatch[1] : null;
+
+    const params = new URLSearchParams(window.location.search);
+    const sessionId = params.get('session');
+
+    if (conversationId && this.isValidId(conversationId)) {
+      this.routerState.currentConversationId = conversationId;
+      if (sessionId && this.isValidId(sessionId)) {
+        this.routerState.currentSessionId = sessionId;
+      }
+      this._dbg('Restoring conversation from URL:', conversationId);
+      this._isLoadingConversation = true;
+      if (window.conversationManager) {
+        window.conversationManager.select(conversationId);
+      } else {
+        this.loadConversationMessages(conversationId).catch((err) => {
+          console.warn('Failed to restore conversation from URL, loading latest instead:', err);
+          if (this.state.conversations && this.state.conversations.length > 0) {
+            const latestConv = this.state.conversations[0];
+            this._dbg('Loading latest conversation instead:', latestConv.id);
+            return this.loadConversationMessages(latestConv.id);
+          } else {
+            this._showWelcomeScreen();
+          }
+        }).finally(() => {
+          this._isLoadingConversation = false;
+        });
+      }
+    } else {
+      this._showWelcomeScreen();
+    }
+  },
+
+
+  isValidId(id) {
+    if (!id || typeof id !== 'string') return false;
+    return /^[a-zA-Z0-9_-]+$/.test(id) && id.length < 256;
+  }
+
+});