debugging streaming

Author: mrlynn

mongodb-rag-docs/.docusaurus/client-manifest.json

@@ -504,9 +504,9 @@
     "5354": {
       "js": [
         {
-          "file": "assets/js/runtime~main.266052e3.js",
-          "hash": "75a08fe72a9d2a9a",
-          "publicPath": "/mongodb-rag/assets/js/runtime~main.266052e3.js"
+          "file": "assets/js/runtime~main.d5dee0a9.js",
+          "hash": "03a1c33c41e3b718",
+          "publicPath": "/mongodb-rag/assets/js/runtime~main.d5dee0a9.js"
         }
       ]
     },
@@ -621,9 +621,9 @@
     "7522": {
       "js": [
         {
-          "file": "assets/js/e3e8e5bb.b2f6b0e6.js",
-          "hash": "06409c248eb61e81",
-          "publicPath": "/mongodb-rag/assets/js/e3e8e5bb.b2f6b0e6.js"
+          "file": "assets/js/e3e8e5bb.7e7fac55.js",
+          "hash": "4a21ad8447d1be8c",
+          "publicPath": "/mongodb-rag/assets/js/e3e8e5bb.7e7fac55.js"
         }
       ]
     },

mongodb-rag-docs/src/components/ChatbotInterface/index.js

@@ -224,74 +224,75 @@ export default function ChatbotInterface() {
     e.preventDefault();
     if (!input.trim()) return;
 
-    // Add user message
     const userMessage = { role: 'user', content: input };
     setMessages(msgs => [...msgs, userMessage]);
     setInput('');
     setIsLoading(true);
-    setShowSampleQuestions(true);
 
     try {
-      // Try streaming first
-      await handleStreamedChat(input, messages);
-    } catch (error) {
-      console.error('Streaming failed, falling back to regular chat:', error);
-      
-      // Fallback to non-streaming implementation
-      try {
-        const response = await fetch('https://mongodb-rag-docs-backend.vercel.app/api/chat', {
-          method: 'POST',
-          headers: { 
-            'Content-Type': 'application/json',
-            'Accept': 'application/json'  // Request JSON response
-          },
-          body: JSON.stringify({
-            query: input,
-            sessionId: sessionId,
-            history: !sessionId ? messages : undefined,
-            stream: false
-          })
-        });
+      const response = await fetch('https://mongodb-rag-docs-backend.vercel.app/api/chat', {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({
+          query: input,
+          sessionId: sessionId,
+          history: !sessionId ? messages : undefined
+        }),
+        // Add request timeout
+        signal: AbortSignal.timeout(45000) // 45 second timeout
+      });
 
-        if (!response.ok) {
-          throw new Error(`HTTP error! status: ${response.status}`);
-        }
+      if (!response.ok) {
+        throw new Error(`HTTP error! status: ${response.status}`);
+      }
 
-        const text = await response.text();
-        const data = safeJsonParse(text);
-        
-        if (!data) {
-          throw new Error('Invalid response format');
-        }
+      const data = await response.json();
+      
+      if (data.error) {
+        throw new Error(data.error);
+      }
 
-        if (data.sessionId && !sessionId) {
-          setSessionId(data.sessionId);
+      setMessages(msgs => [
+        ...msgs,
+        {
+          role: 'assistant',
+          content: data.answer,
+          sources: data.sources
         }
-        
-        setMessages(msgs => [
-          ...msgs,
-          {
-            role: 'assistant',
-            content: data.answer || 'Sorry, I received an invalid response.',
-            sources: data.sources
-          }
-        ]);
-      } catch (error) {
-        console.error('Chat error:', error);
-        setMessages(msgs => [
-          ...msgs,
-          {
-            role: 'assistant',
-            content: `Sorry, I encountered an error: ${error.message}`
-          }
-        ]);
-      }
+      ]);
+    } catch (error) {
+      handleError(error);
     } finally {
       setIsLoading(false);
-      setCurrentStreamedMessage('');
     }
   };
+
+  const handleError = (error, fallback = true) => {
+    console.error('Chat error:', error);
+    
+    let errorMessage = 'Sorry, I encountered an error. Please try again later.';
+    
+    if (error.response?.status === 504 || error.message?.includes('timed out')) {
+      errorMessage = 'The search is taking longer than expected. You might try:' +
+        '\n1. Rephrasing your question to be more specific' +
+        '\n2. Breaking it into smaller parts' +
+        '\n3. Trying again in a moment';
+    }
+  
+    setMessages(msgs => [
+      ...msgs,
+      {
+        role: 'assistant',
+        content: errorMessage
+      }
+    ]);
 
+    if (fallback && !error.message?.includes('timed out')) {
+      // Optionally try fallback handling for non-timeout errors
+      handleFallbackChat(input, messages);
+    }
+  };
+
   const clearChat = () => {
     // Reset the chat
     setMessages([