From c8a75efa274b6f948f05054a6ed2acc7c5405b4a Mon Sep 17 00:00:00 2001
From: litongjava <litongjava@qq.com>
Date: Mon, 3 Feb 2025 09:10:38 -1000
Subject: [PATCH] feat: keep fouce model

---
 ui/components/ChatWindow.tsx | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index 0707457..ec8e3e6 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -5,7 +5,6 @@ import {Document} from '@langchain/core/documents';
 import Navbar from './Navbar';
 import Chat from './Chat';
 import EmptyChat from './EmptyChat';
-import crypto from 'crypto';
 import {toast} from 'sonner';
 import {useSearchParams} from 'next/navigation';
 import {getSuggestions} from '@/lib/actions';
@@ -481,6 +480,18 @@ const ChatWindow = ({id}: { id?: string }) => {
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, []);
 
+  useEffect(() => {
+    const savedFocusMode = localStorage.getItem('focusMode');
+    if (savedFocusMode) {
+      setFocusMode(savedFocusMode);
+    }
+  }, [setFocusMode]);
+
+  const handleFocusModeChange = (mode: string) => {
+    localStorage.setItem('focusMode', mode);
+    setFocusMode(mode);
+  };
+
   const messagesRef = useRef<Message[]>([]);
 
   useEffect(() => {
@@ -515,7 +526,7 @@ const ChatWindow = ({id}: { id?: string }) => {
     ws.send(
       JSON.stringify({
         type: 'message',
-        userId:userId,
+        userId: userId,
         message: {
           messageId: messageId,
           chatId: chatId!,
@@ -701,7 +712,7 @@ const ChatWindow = ({id}: { id?: string }) => {
             focusMode={focusMode}
             copilotEnabled={copilotEnabled}
             setCopilotEnabled={setCopilotEnabled}
-            setFocusMode={setFocusMode}
+            setFocusMode={handleFocusModeChange}
             optimizationMode={optimizationMode}
             setOptimizationMode={setOptimizationMode}
             fileIds={fileIds}