Do not set websocket to null if in error state.

ItzCrazyKns · Jul 30, 2024 · 8bd648f · 8bd648f
1 parent f03f665
commit 8bd648f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
@@ -194,7 +194,9 @@ const useSocket = (
 
         ws.onclose = () => {
           clearTimeout(timeoutId);
-          setWs(null);
+          if (!hasError) {
+            setWs(null); // forces websocket to reopen when needed.
+          }
           console.log('[DEBUG] closed');
         };
 
@@ -285,6 +287,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
     process.env.NEXT_PUBLIC_WS_URL!,
     setIsWSReady,
     setHasError,
+    hasError
   );
 
   const [loading, setLoading] = useState(false);